;; gnuplot.lisp: routines for Maxima's interface to gnuplot
;; Copyright (C) 2007-2013 J. Villate
;; 
;; This program is free software; you can redistribute it and/or
;; modify it under the terms of the GNU General Public License
;; as published by the Free Software Foundation; either version 2
;; of the License, or (at your option) any later version.
;; 
;; This program is distributed in the hope that it will be useful,
;; but WITHOUT ANY WARRANTY; without even the implied warranty of
;; MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
;; GNU General Public License for more details.
;; 
;; You should have received a copy of the GNU General Public License
;; along with this program; if not, write to the Free Software
;; Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
;; MA  02110-1301, USA

(in-package :maxima)

;; Checks that color is a six-digit hexadecimal number with the prefix #,
;; or a symbol for one of the 12 pre-defined colors, in which case the
;; hexadecimal code for that color will be returned. Unknown colors are
;; converted into black.
(defun rgb-color (color)
  (if (plotcolorp color)
      (case color
	($red "#ff0000")
        ($green "#00ff00")
        ($blue "#0000ff")
	($magenta "#ff00ff")
        ($cyan "#00ffff")
        ($yellow "#ffff00")
        ($orange "#ffa500")
        ($violet "#ee82ee")
        ($brown "#a52a2a")
        ($gray "#bebebe")
        ($black "#000000")
        ($white "#ffffff")
        (t color))
      "#000000"))

;; Given a list of valid colors (see rgb-color function) and an object c
;; that can be a real number or a string, produces a gnuplot color
;; specification for c; when c is real, its nearest integer is assigned
;; to one of the numbers in the list, using modulo length of the list.
(defun gnuplot-color (colors c)
  (unless (listp colors) (setq colors (list colors)))
  (when (realp c)
    (unless (integerp c) (setq c (round c)))
    (setq c (nth (mod (1- c) (length colors)) colors)))
  (format nil "rgb ~s" (rgb-color c)))

(defun gnuplot-pointtype (type)
  (case type
    ($bullet 7) ($circle 6) ($plus 1) ($times 2) ($asterisk 3) ($box 5)
    ($square 4) ($triangle 9) ($delta 8) ($wedge 11) ($nabla 10)
    ($diamond 13) ($lozenge 12) (t 7)))

(defun gnuplot-pointtypes (types n)
  (unless (integerp n) (setq n (round n)))
  (unless (listp types) (setq types (list types)))
  (gnuplot-pointtype (nth (mod (- n 1) (length types)) types)))

;; style is a list starting with one of the symbols: points, lines,
;; linespoints or dots,
;; The meaning of the numbers that follow the symbol are:
;;
;;   lines, linewidth, color
;;   points, radius, color, pointtype
;;   linespoints, linewidth, radius, color, pointtype
;;   dots, color
;;
;; linewidth and radius are measured in the same units and can be
;; floating-point numbers.
;;
;; type must be an integer
;; color can be an integer, used as index to get one of the colors defined
;; by the color option, or a 6-digit hexadecimal number #rrggbb

(defun gnuplot-curve-style (style colors types i)
  (unless (listp style) (setq style (list style)))
  (unless (listp colors) (setq colors (list colors)))
  (with-output-to-string
    (st)
    (case (first style)
      ($dots
       (format st "with dots")
       (if (second style)
         (format st " lt ~d" (gnuplot-color colors (second style)))
         (format st " lt ~d" (gnuplot-color colors i))))
      ($impulses
       (format st "with impulses")
       (if (integerp (second style))
         (format st " lt ~d" (gnuplot-color colors (second style)))
         (format st " lt ~d" (gnuplot-color colors i))))
      ($lines
       (format st "with lines")
       (if (realp (second style))
         (format st " lw ~,2f" (second style)))
       (if (third style)
         (format st " lt ~d" (gnuplot-color colors (third style)))
         (format st " lt ~d" (gnuplot-color colors i))))
      ($points
       (format st "with points")
       (if (realp (second style))
         (format st " ps ~,2f" (/ (second style) 2))
         (format st " ps 1.5"))
       (if (third style)
         (format st " lt ~d" (gnuplot-color colors (third style)))
         (format st " lt ~d" (gnuplot-color colors i)))
       (if (integerp (fourth style))
         (format st " pt ~d" (gnuplot-pointtypes types (fourth style)))
         (format st " pt ~d" (gnuplot-pointtypes types i))))
      ($linespoints
       (format st "with linespoints")
       (if (realp (second style))
         (format st " lw ~,2f" (second style)))
       (if (realp (third style))
         (format st " ps ~,2f" (/ (third style) 2))
         (format st " ps 1.5"))
       (if (fourth style)
         (format st " lt ~d" (gnuplot-color colors (fourth style)))
         (format st " lt ~d" (gnuplot-color colors i)))
       (if (integerp (fifth style))
         (format st " pt ~d" (gnuplot-pointtypes types (fifth style)))
         (format st " pt ~d" (gnuplot-pointtypes types i))))
      (t (format st "with lines lt ~d" (gnuplot-color colors i))))))

(defun gnuplot-palette (palette)
;; palette should be a list starting with one of the symbols: hue,
;; saturation, value, gray or gradient.
;;
;; If the symbol is gray, it should be followed by two floating point
;; numbers that indicate the initial gray level and the interval of 
;; gray values.
;;
;; If the symbol is one of hue, saturation or value, it must be followed
;; by three numbers that specify the hue, saturation and value for the
;; initial color, and a fourth number that gives the range of values for
;; the increment of hue, saturation or value.
;; The values for the initial hue, saturation, value and grayness should
;; be within 0 and 1, while the range can be higher or even negative.
;;
;; If the symbol is gradient, it must be followed by either a list of valid
;; colors or by a list of lists with two elements, a number and a valid color.

  (unless (listp palette) (setq palette (list palette)))
  (let (hue sat val gray range fun)
    (case (first palette)
      ($gray
       (case (length (rest palette))
         (2 (setq gray (second palette)) (setq range (third palette)))
         (t (merror
             (intl:gettext
              "palette: gray must be followed by two numbers."))))
       (when (or (< gray 0) (> gray 1))
         (setq gray (- gray (floor gray)))))
      (($hue $saturation $value)
       (case (length (rest palette))
         (4 (setq hue (second palette))
            (setq sat (third palette))
            (setq val (fourth palette))
            (setq range (fifth palette)))
         (t (merror
             (intl:gettext
              "palette: ~M must be followed by four numbers.")
              (first palette))))
       (when (or (< hue 0) (> hue 1)) (setq hue (- hue (floor hue))))
       (when (or (< sat 0) (> sat 1)) (setq sat (- sat (floor sat))))
       (when (or (< val 0) (> val 1)) (setq val (- val (floor val))))))
    (with-output-to-string (st)
      (case (first palette)
        ($hue
         (if (or (< (+ hue range) 0) (> (+ hue range) 1))
             (setq fun (format nil "~,3f+~,3f*gray-floor(~,3f+~,3f*gray)"
                               hue range hue range))
             (setq fun (format nil "~,3f+~,3f*gray" hue range)))
         (format st "model HSV functions ~a, ~,3f, ~,3f" fun sat val))
        ($saturation
         (if (or (< (+ sat range) 0) (> (+ sat range) 1))
             (setq fun (format nil "~,3f+~,3f*gray-floor(~,3f+~,3f*gray)"
                               sat range sat range))
             (setq fun (format nil "~,3f+~,3f*gray" sat range)))
         (format st "model HSV functions ~,3f, ~a, ~,3f" hue fun val))
        ($value
         (if (or (< (+ val range) 0) (> (+ val range) 1))
             (setq fun (format nil "~,3f+~,3f*gray" val range))
             (setq fun (format nil "~,3f+~,3f*gray-floor(~,3f+~,3f*gray)"
                               val range val range)))
         (format st "model HSV functions ~,3f, ~,3f, ~a" hue sat fun))
        ($gray
         (if (or (< (+ gray range) 0) (> (+ gray range) 1))
             (setq fun (format nil "~,3f+~,3f*gray" gray range))
             (setq fun (format nil "~,3f+~,3f*gray-floor(~,3f+~,3f*gray)"
                               gray range gray range)))
         (format st "model RGB functions ~a, ~a, ~a" fun fun fun))

        ($gradient
         (let* ((colors (rest palette)) (n (length colors)) (map nil))
           ;; map is constructed as (n1 c1 n2 c2 ... nj cj) where ni is a
           ;; decreasing sequence of numbers (n1=1, nj=0) and ci are colors
           (cond
             ;; Maxima list of numbers and colors (((mlist) ni ci) ...)
             ((listp (first colors))
              (setq colors (sort colors #'< :key #'cadr))
              (dotimes (i n)
                (setq map (cons (rgb-color (third (nth i colors))) ;; color
                                (cons
                                 (/ (- (second (nth i colors))   ;; ni minus
                                       (second (first colors)))  ;; smallest ni
                                    (- (second (nth (- n 1) colors));; biggest
                                       (second (first colors)))) ;; - smallest
                                 map)))))
             ;; list of only colors
             (t (dotimes (i n)
                  (setq map (cons (rgb-color (nth i colors))  ;; color i
                                  (cons (/ i (1- n)) map))))))    ;; number i

           ;; prints map with the format:  nj, "cj", ...,n1, "c1"  
           (setq fun (format nil "~{~,8f ~s~^, ~}" (reverse map)))
           ;; outputs the string: defined (nj, "cj", ...,n1, "c1")
           (format st "defined (~a)" fun)))
        (t
         (merror
          (intl:gettext
           "palette: wrong keyword ~M. Must be hue, saturation, value, gray or gradient.")
          (first palette)))))))

(defun gnuplot-print-header (dest plot-options)
  (let (terminal-file (palette (getf plot-options :palette))
        (meshcolor (if (member :mesh_lines_color plot-options)
                       (getf plot-options :mesh_lines_color)
                       '$black)))
    (when (and (member :gnuplot_pm3d plot-options)
               (not (getf plot-options :gnuplot_pm3d)))
      (setq palette nil))
    (when (find 'mlist palette :key #'car) (setq palette (list palette)))
    ;; user's preamble
    (when (and (getf plot-options :gnuplot_preamble)
               (> (length  (getf plot-options :gnuplot_preamble)) 0))
      (format dest "~a~%" (getf plot-options :gnuplot_preamble)))

    ;; sets-up terminal command and output file name
    (setq terminal-file (gnuplot-terminal-and-file plot-options))

    ;; prints terminal and output commands
    (when (first terminal-file)
      (format dest "~a~%" (first terminal-file)))
    (when (second terminal-file)
      (format dest "set output ~s~%" (second terminal-file)))

    ;; options specific to plot3d
    (when (string= (getf plot-options :type) "plot3d")
      (format dest "set xyplane relative 0~%")
      (if palette
          (progn
            (if meshcolor
                (progn
                  (format dest "set style line 100 lt rgb ~s lw 1~%"
                          (rgb-color meshcolor))
                  (format dest "set pm3d hidden3d 100~%")
                  (unless (getf plot-options :gnuplot_4_0)
                    (format dest "set pm3d depthorder~%")))
                (format dest "set pm3d~%"))
            (format dest "unset hidden3d~%")
            (format dest "set palette ~a~%"
                    (gnuplot-palette (rest (first palette)))))
          (format dest "set hidden3d~%"))
      (let ((elev (getf plot-options :elevation))
            (azim (getf plot-options :azimuth)))
        (when (or elev azim)
          (if elev
              (format dest "set view ~d" elev)
              (format dest "set view "))
          (when azim (format dest ", ~d" azim))
          (format dest "~%"))))

    ;; color_bar can be used by plot3d or plot2d
    (unless (getf plot-options :color_bar)
      (format dest "unset colorbox~%"))

    ;; ----- BEGIN GNUPLOT 4.0 WORK-AROUND -----
    ;; When the expression to be plotted is a constant, Gnuplot fails
    ;; with a division by 0.  Explicitly assigning cbrange prevents
    ;; the error. Also set zrange to match cbrange.
    ;; When the bug is fixed in Gnuplot (maybe 4.1 ?) this hack can go away.
    (when (floatp (getf plot-options :const_expr))
      (format
       dest "set cbrange [~a : ~a]~%"
       (1- (getf plot-options :const_expr))
       (1+ (getf plot-options :const_expr)))
      (format
       dest "set zrange [~a : ~a]~%"
       (1- (getf plot-options :const_expr))
       (1+ (getf plot-options :const_expr))))
    ;; -----  END GNUPLOT 4.0 WORK-AROUND  -----
    
    ;; logarithmic plots
    (when (getf plot-options :logx) (format dest "set log x~%"))
    (when (getf plot-options :logy) (format dest "set log y~%"))

    ;; axes labels and legend
    (when (getf plot-options :xlabel)
      (format dest "set xlabel ~s~%" (getf plot-options :xlabel)))
    (when (getf plot-options :ylabel)
      (format dest "set ylabel ~s~%" (getf plot-options :ylabel)))
    (when (getf plot-options :zlabel)
      (format dest "set zlabel ~s~%" (getf plot-options :zlabel)))
    (when (and (member :legend plot-options) 
               (null (getf plot-options :legend)))
      (format dest "unset key~%"))

    ;; plotting box
    (when (and (member :box plot-options) (not (getf plot-options :box)))
      (format dest "unset border~%")
      (if (and (getf plot-options :axes)
               (string= (getf plot-options :type) "plot2d"))
          (format dest "set xtics axis~%set ytics axis~%set ztics axis~%")
          (format dest "unset xtics~%unset ytics~%unset ztics~%")))

    (when (string= (getf plot-options :type) "plot2d")

      ;; 2d grid (specific to plot2d)
      (format dest "set grid front~%")
      (if (getf plot-options :grid2d)
          (format dest "set grid~%")
          (format dest "unset grid~%"))

      ;; plot size and aspect ratio for plot2d
      (if (getf plot-options :same_xy)
          (format dest "set size ratio -1~%")
          (if (getf plot-options :yx_ratio)
              (format dest "set size ratio ~,8f~%" (getf plot-options :yx_ratio))
              (format dest "set size ratio 0.75~%")))
      (if (and (getf plot-options :xy_scale)
               (listp (getf plot-options :xy_scale)))
          (format dest "set size ~{~,8f~^, ~}~%" (getf plot-options :xy_scale))))

    ;; plot size and aspect ratio for plot3d
    (when (string= (getf plot-options :type) "plot3d")
      (when (getf plot-options :same_xy)
        (format dest "set view equal xy~%"))
      (when (getf plot-options :same_xyz)
        (format dest "set view equal xyz~%"))
      (when (getf plot-options :zmin)
        (format dest "set xyplane at ~,8f~%" (getf plot-options :zmin))))

    ;; axes tics
    (when (member :xtics plot-options)
      (let ((xtics (getf plot-options :xtics)))
        (if (consp xtics)
            (format dest "set xtics ~{~,8f~^, ~}~%" xtics)
            (if xtics
                (format dest "set xtics ~,8f~%" xtics)
                (format dest "unset xtics~%")))))
    (when (member :ytics plot-options)
      (let ((ytics (getf plot-options :ytics)))
        (if (consp ytics)
            (format dest "set ytics ~{~,8f~^, ~}~%" ytics)
            (if ytics
                (format dest "set ytics ~,8f~%" ytics)
                (format dest "unset ytics~%")))))
    (when (member :ztics plot-options)
      (let ((ztics (getf plot-options :ztics)))
        (if (consp ztics)
            (format dest "set ztics ~{~,8f~^, ~}~%" ztics)
            (if ztics
                (format dest "set ztics ~,8f~%" ztics)
                (format dest "unset ztics~%")))))
    (when (member :color_bar_tics plot-options)
      (let ((cbtics (getf plot-options :color_bar_tics)))
        (if (consp cbtics)
            (format dest "set cbtics ~{~,8f~^, ~}~%" cbtics)
            (if cbtics
                (format dest "set cbtics ~,8f~%" cbtics)
                (format dest "unset cbtics~%")))))

    ;; axes ranges and style
    (when (and (getf plot-options :x) (listp (getf plot-options :x)))
      (format dest "set xrange [~{~,8f~^ : ~}]~%" (getf plot-options :x)))
    (when (and (getf plot-options :y) (listp (getf plot-options :y)))
      (format dest "set yrange [~{~,8f~^ : ~}]~%" (getf plot-options :y)))
    (when (and (getf plot-options :z) (listp (getf plot-options :z)))
      (format dest "set zrange [~{~,8f~^ : ~}]~%" (getf plot-options :z)))
    (when (and (string= (getf plot-options :type) "plot2d")
               (member :axes plot-options))
      (if (getf plot-options :axes)
          (case (getf plot-options :axes)
            ($x (format dest "set xzeroaxis~%"))
            ($y (format dest "set yzeroaxis~%"))
            ($solid (format dest "set zeroaxis lt -1~%"))
            (t (format dest "set zeroaxis~%")))))

    ;; title and labels
    (when (getf plot-options :title)
      (format dest "set title ~s~%" (getf plot-options :title)))
    (when (getf plot-options :label)
      (dolist (label (getf plot-options :label))
        (when (and (listp label) (= (length label) 4))
          (format dest "set label ~s at ~{~,8f~^, ~}~%"
                  (cadr label) (cddr label)))))

    ;; identifier for missing data
    (format dest "set datafile missing ~s~%" *missing-data-indicator*)

    ;; user's commands; may overule any of the previous settings
    (when (and (getf plot-options :gnuplot_postamble)
               (> (length  (getf plot-options :gnuplot_postamble)) 0))
      (format dest "~a~%" (getf plot-options :gnuplot_postamble)))

    ;;returns a list with the name of the file created, or nil
    (if (null (second terminal-file)) nil (list (second terminal-file)))))

(defun gnuplot-plot3d-command (file palette gstyles colors titles n) 
(let (title (style "with pm3d"))
  (with-output-to-string (out)
    (format out "splot ")
  (do ((i 1 (+ i 1))) ((> i n) (format out "~%"))
    (unless palette
      (if gstyles
	  (setq style (ensure-string (nth (mod i (length gstyles)) gstyles)))
	  (setq style
                (format nil "with lines lt ~a" (gnuplot-color colors i)))))
    (when (> i 1) (format out ", "))
    (if titles
        (setq title (nth (mod i (length titles)) titles))
        (setq title ""))
    (format out "~s title ~s ~a" file title style)))))

(defun gnuplot-terminal-and-file (plot-options)
(let (terminal-command out-file)
  (cond
    ((getf plot-options :svg_file)
     (if (getf plot-options :gnuplot_svg_term_command)
         (setq terminal-command
               (getf plot-options :gnuplot_svg_term_command))
         (setq terminal-command "set term svg font \",14\""))
     (setq out-file (getf plot-options :svg_file)))
    ((getf plot-options :png_file)
     (if (getf plot-options :gnuplot_png_term_command)
         (setq terminal-command
               (getf plot-options :gnuplot_png_term_command))
         (setq terminal-command "set term pngcairo font \",12\""))
     (setq out-file (getf plot-options :png_file)))
    ((getf plot-options :pdf_file)
     (if (getf plot-options :gnuplot_pdf_term_command)
         (setq terminal-command
               (getf plot-options :gnuplot_pdf_term_command))
         (setq terminal-command "set term pdfcairo color solid lw 3 size 17.2 cm, 12.9 cm font \",18\""))
     (setq out-file (getf plot-options :pdf_file)))
    ((getf plot-options :ps_file)
     (if (getf plot-options :gnuplot_ps_term_command)
         (setq terminal-command
               (getf plot-options :gnuplot_ps_term_command))
         (setq terminal-command "set term postscript eps color solid lw 2 size 16.4 cm, 12.3 cm font \",24\""))
     (setq out-file (getf plot-options :ps_file)))
    ((eq (getf plot-options :gnuplot_term) '$ps)
     (if (getf plot-options :gnuplot_ps_term_command)
         (setq terminal-command
               (getf plot-options :gnuplot_ps_term_command))
         (setq terminal-command "set term postscript eps color solid lw 2 size 16.4 cm, 12.3 cm font \",24\""))
     (if (getf plot-options :gnuplot_out_file)
         (setq out-file (getf plot-options :gnuplot_out_file))
         (setq out-file "maxplot.ps")))
    ((eq (getf plot-options :gnuplot_term) '$dumb)
     (if (getf plot-options :gnuplot_dumb_term_command)
         (setq terminal-command
               (getf plot-options :gnuplot_ps_term_command))
         (setq terminal-command "set term dumb 79 22"))
     (if (getf plot-options :gnuplot_out_file)
         (setq out-file (getf plot-options :gnuplot_out_file))
         (setq out-file "maxplot.txt")))
    ((eq (getf plot-options :gnuplot_term) '$default)
     (if (getf plot-options :gnuplot_default_term_command)
         (setq terminal-command
               (getf plot-options :gnuplot_default_term_command))
         (setq terminal-command
               "set term pop")))
    ((getf plot-options :gnuplot_term)
     (setq
      terminal-command
          (format nil "set term ~(~a~)"
           (ensure-string (getf plot-options :gnuplot_term))))
     (if (getf plot-options :gnuplot_out_file)
         (setq out-file (getf plot-options :gnuplot_out_file))
         (setq
          out-file
          (format nil "maxplot.~(~a~)"
                  (get-gnuplot-term (getf plot-options :gnuplot_term)))))))

  (unless (null out-file) (setq out-file (plot-file-path out-file)))
  (list terminal-command out-file)))
