123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708 |
- ;;; org-plot.el --- Support for Plotting from Org -*- lexical-binding: t; -*-
- ;; Copyright (C) 2008-2021 Free Software Foundation, Inc.
- ;;
- ;; Author: Eric Schulte <schulte dot eric at gmail dot com>
- ;; Maintainer: TEC <tecosaur@gmail.com>
- ;; Keywords: tables, plotting
- ;; Homepage: https://orgmode.org
- ;;
- ;; This file is part of GNU Emacs.
- ;;
- ;; GNU Emacs 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 3 of the License, or
- ;; (at your option) any later version.
- ;; GNU Emacs 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 GNU Emacs. If not, see <https://www.gnu.org/licenses/>.
- ;;; Commentary:
- ;; Borrows ideas and a couple of lines of code from org-exp.el.
- ;; Thanks to the Org mailing list for testing and implementation and
- ;; feature suggestions
- ;;; Code:
- (require 'cl-lib)
- (require 'org)
- (require 'org-table)
- (declare-function gnuplot-delchar-or-maybe-eof "ext:gnuplot" (arg))
- (declare-function gnuplot-mode "ext:gnuplot" ())
- (declare-function gnuplot-send-buffer-to-gnuplot "ext:gnuplot" ())
- (defvar org-plot/gnuplot-default-options
- '((:plot-type . 2d)
- (:with . lines)
- (:ind . 0))
- "Default options to gnuplot used by `org-plot/gnuplot'.")
- (defvar org-plot-timestamp-fmt nil)
- (defun org-plot/add-options-to-plist (p options)
- "Parse an OPTIONS line and set values in the property list P.
- Returns the resulting property list."
- (when options
- (let ((op '(("type" . :plot-type)
- ("script" . :script)
- ("line" . :line)
- ("set" . :set)
- ("title" . :title)
- ("ind" . :ind)
- ("deps" . :deps)
- ("with" . :with)
- ("file" . :file)
- ("labels" . :labels)
- ("map" . :map)
- ("timeind" . :timeind)
- ("timefmt" . :timefmt)
- ("min" . :ymin)
- ("ymin" . :ymin)
- ("max" . :ymax)
- ("ymax" . :ymax)
- ("xmin" . :xmin)
- ("xmax" . :xmax)
- ("ticks" . :ticks)
- ("trans" . :transpose)
- ("transpose" . :transpose)))
- (multiples '("set" "line"))
- (regexp ":\\([\"][^\"]+?[\"]\\|[(][^)]+?[)]\\|[^ \t\n\r;,.]*\\)")
- (start 0))
- (dolist (o op)
- (if (member (car o) multiples) ;; keys with multiple values
- (while (string-match
- (concat (regexp-quote (car o)) regexp)
- options start)
- (setq start (match-end 0))
- (setq p (plist-put p (cdr o)
- (cons (car (read-from-string
- (match-string 1 options)))
- (plist-get p (cdr o)))))
- p)
- (if (string-match (concat (regexp-quote (car o)) regexp)
- options)
- (setq p (plist-put p (cdr o)
- (car (read-from-string
- (match-string 1 options))))))))))
- p)
- (defun org-plot/goto-nearest-table ()
- "Move the point forward to the beginning of nearest table.
- Return value is the point at the beginning of the table."
- (interactive) (move-beginning-of-line 1)
- (while (not (or (org-at-table-p) (< 0 (forward-line 1)))))
- (goto-char (org-table-begin)))
- (defun org-plot/collect-options (&optional params)
- "Collect options from an org-plot `#+Plot:' line.
- Accepts an optional property list PARAMS, to which the options
- will be added. Returns the resulting property list."
- (interactive)
- (let ((line (thing-at-point 'line)))
- (if (string-match "#\\+PLOT: +\\(.*\\)$" line)
- (org-plot/add-options-to-plist params (match-string 1 line))
- params)))
- (defun org-plot-quote-timestamp-field (s)
- "Convert field S from timestamp to Unix time and export to gnuplot."
- (format-time-string org-plot-timestamp-fmt (org-time-string-to-time s)))
- (defun org-plot-quote-tsv-field (s)
- "Quote field S for export to gnuplot."
- (if (string-match org-table-number-regexp s) s
- (if (string-match org-ts-regexp3 s)
- (org-plot-quote-timestamp-field s)
- (concat "\"" (mapconcat 'identity (split-string s "\"") "\"\"") "\""))))
- (defun org-plot/gnuplot-to-data (table data-file params)
- "Export TABLE to DATA-FILE in a format readable by gnuplot.
- Pass PARAMS through to `orgtbl-to-generic' when exporting TABLE."
- (with-temp-file
- data-file
- (setq-local org-plot-timestamp-fmt (or
- (plist-get params :timefmt)
- "%Y-%m-%d-%H:%M:%S"))
- (insert (orgtbl-to-generic
- table
- (org-combine-plists
- '(:sep "\t" :fmt org-plot-quote-tsv-field)
- params))))
- nil)
- (defun org-plot/gnuplot-to-grid-data (table data-file params)
- "Export the data in TABLE to DATA-FILE for gnuplot.
- This means in a format appropriate for grid plotting by gnuplot.
- PARAMS specifies which columns of TABLE should be plotted as independent
- and dependent variables."
- (interactive)
- (let* ((ind (- (plist-get params :ind) 1))
- (deps (if (plist-member params :deps)
- (mapcar (lambda (val) (- val 1)) (plist-get params :deps))
- (let (collector)
- (dotimes (col (length (nth 0 table)))
- (setf collector (cons col collector)))
- collector)))
- (counter 0)
- row-vals)
- (when (>= ind 0) ;; collect values of ind col
- (setf row-vals (mapcar (lambda (row) (setf counter (+ 1 counter))
- (cons counter (nth ind row)))
- table)))
- (when (or deps (>= ind 0)) ;; remove non-plotting columns
- (setf deps (delq ind deps))
- (setf table (mapcar (lambda (row)
- (dotimes (col (length row))
- (unless (memq col deps)
- (setf (nth col row) nil)))
- (delq nil row))
- table)))
- ;; write table to gnuplot grid datafile format
- (with-temp-file data-file
- (let ((num-rows (length table)) (num-cols (length (nth 0 table)))
- (gnuplot-row (lambda (col row value)
- (setf col (+ 1 col)) (setf row (+ 1 row))
- (format "%f %f %f\n%f %f %f\n"
- col (- row 0.5) value ;; lower edge
- col (+ row 0.5) value))) ;; upper edge
- front-edge back-edge)
- (dotimes (col num-cols)
- (dotimes (row num-rows)
- (setf back-edge
- (concat back-edge
- (funcall gnuplot-row (- col 1) row
- (string-to-number (nth col (nth row table))))))
- (setf front-edge
- (concat front-edge
- (funcall gnuplot-row col row
- (string-to-number (nth col (nth row table)))))))
- ;; only insert once per row
- (insert back-edge) (insert "\n") ;; back edge
- (insert front-edge) (insert "\n") ;; front edge
- (setf back-edge "") (setf front-edge ""))))
- row-vals))
- (defun org--plot/values-stats (nums &optional hard-min hard-max)
- "From a list of NUMS return a plist containing some rudamentry statistics on the
- values, namely regarding the range."
- (let* ((minimum (or hard-min (apply #'min nums)))
- (maximum (or hard-max (apply #'max nums)))
- (range (- maximum minimum))
- (rangeOrder (if (= range 0) 0
- (ceiling (- 1 (log range 10)))))
- (range-factor (expt 10 rangeOrder))
- (nice-min (if (= range 0) (car nums)
- (/ (float (floor (* minimum range-factor))) range-factor)))
- (nice-max (if (= range 0) (car nums)
- (/ (float (ceiling (* maximum range-factor))) range-factor))))
- `(:min ,minimum :max ,maximum :range ,range
- :range-factor ,range-factor
- :nice-min ,nice-min :nice-max ,nice-max :nice-range ,(- nice-max nice-min))))
- (defun org--plot/sensible-tick-num (table &optional hard-min hard-max)
- "From a the values in a TABLE of data, attempt to guess an appropriate number of ticks."
- (let* ((row-data
- (mapcar (lambda (row) (org--plot/values-stats
- (mapcar #'string-to-number (cdr row))
- hard-min
- hard-max)) table))
- (row-normalised-ranges (mapcar (lambda (r-data)
- (let ((val (round (*
- (plist-get r-data :range-factor)
- (plist-get r-data :nice-range)))))
- (if (= (% val 10) 0) (/ val 10) val)))
- row-data))
- (range-prime-decomposition (mapcar #'org--plot/prime-factors row-normalised-ranges))
- (weighted-factors (sort (apply #'org--plot/merge-alists #'+ 0
- (mapcar (lambda (factors) (org--plot/item-frequencies factors t))
- range-prime-decomposition))
- (lambda (a b) (> (cdr a) (cdr b))))))
- (apply #'* (org--plot/nice-frequency-pick weighted-factors))))
- (defun org--plot/nice-frequency-pick (frequencies)
- "From a list of frequences, try to sensibly pick a sample of the most frequent."
- ;; TODO this mosly works decently, but counld do with some tweaking to work more consistently.
- (cl-case (length frequencies)
- (1 (list (car (nth 0 frequencies))))
- (2 (if (<= 3 (/ (cdr (nth 0 frequencies))
- (cdr (nth 1 frequencies))))
- (make-list 2
- (car (nth 0 frequencies)))
- (list (car (nth 0 frequencies))
- (car (nth 1 frequencies)))))
- (t
- (let* ((total-count (apply #'+ (mapcar #'cdr frequencies)))
- (n-freq (mapcar (lambda (freq) `(,(car freq) . ,(/ (float (cdr freq)) total-count))) frequencies))
- (f-pick (list (car (car n-freq))))
- (1-2-ratio (/ (cdr (nth 0 n-freq))
- (cdr (nth 1 n-freq))))
- (2-3-ratio (/ (cdr (nth 1 n-freq))
- (cdr (nth 2 n-freq))))
- (1-3-ratio (* 1-2-ratio 2-3-ratio))
- (1-val (car (nth 0 n-freq)))
- (2-val (car (nth 1 n-freq)))
- (3-val (car (nth 2 n-freq))))
- (when (> 1-2-ratio 4) (push 1-val f-pick))
- (when (and (< 1-2-ratio 2-val)
- (< (* (apply #'* f-pick) 2-val) 30))
- (push 2-val f-pick))
- (when (and (< 1-3-ratio 3-val)
- (< (* (apply #'* f-pick) 3-val) 30))
- (push 3-val f-pick))
- f-pick))))
- (defun org--plot/merge-alists (function default alist1 alist2 &rest alists)
- "Using FUNCTION, combine the elements of all given ALISTS. When an element is
- only present in one alist, DEFAULT is used as the second argument for the FUNCTION."
- (when (> (length alists) 0)
- (setq alist2 (apply #'org--plot/merge-alists function default alist2 alists)))
- (cl-flet ((keys (alist) (mapcar #'car alist))
- (lookup (key alist) (or (cdr (assoc key alist)) default)))
- (cl-loop with keys = (cl-union (keys alist1) (keys alist2) :test 'equal)
- for k in keys collect
- (cons k (funcall function (lookup k alist1) (lookup k alist2))))))
- (defun org--plot/item-frequencies (values &optional normalise)
- "Return an alist indicating the frequency of values in VALUES list."
- (let ((normaliser (if normalise (float (length values)) 1)))
- (cl-loop for (n . m) in (seq-group-by #'identity values)
- collect (cons n (/ (length m) normaliser)))))
- (defun org--plot/prime-factors (value)
- "Return the prime decomposition of VALUE, e.g. for 12, '(3 2 2)"
- (let ((factors '(1)) (i 1))
- (while (/= 1 value)
- (setq i (1+ i))
- (when (eq 0 (% value i))
- (push i factors)
- (setq value (/ value i))
- (setq i (1- i))
- ))
- (cl-subseq factors 0 -1)))
- (defcustom org-plot/gnuplot-script-preamble ""
- "String or function which provides content to be inserted into the GNUPlot
- script before the plot command. Not that this is in addition to, not instead of
- other content generated in `org-plot/gnuplot-script'.
- If a function, it is called with the plot type as the argument."
- :group 'org-plot
- :type '(choice string function))
- (defcustom org-plot/preset-plot-types
- '((2d :plot-cmd "plot"
- :check-ind-type t
- :plot-func
- (lambda (_table data-file num-cols params plot-str)
- (let* ((type (plist-get params :plot-type))
- (with (if (eq type 'grid) 'pm3d (plist-get params :with)))
- (ind (plist-get params :ind))
- (deps (if (plist-member params :deps) (plist-get params :deps)))
- (text-ind (plist-get params :textind))
- (col-labels (plist-get params :labels))
- res)
- (dotimes (col num-cols res)
- (unless (and (eq type '2d)
- (or (and ind (equal (1+ col) ind))
- (and deps (not (member (1+ col) deps)))))
- (setf res
- (cons
- (format plot-str data-file
- (or (and ind (> ind 0)
- (not text-ind)
- (format "%d:" ind)) "")
- (1+ col)
- (if text-ind (format ":xticlabel(%d)" ind) "")
- with
- (or (nth col col-labels)
- (format "%d" (1+ col))))
- res)))))))
- (3d :plot-cmd "splot"
- :plot-pre (lambda (_table _data-file _num-cols params _plot-str)
- (if (plist-get params :map) "set map"))
- :plot-func
- (lambda (_table data-file _num-cols params _plot-str)
- (let* ((type (plist-get params :plot-type))
- (with (if (eq type 'grid) 'pm3d (plist-get params :with))))
- (list (format "'%s' matrix with %s title ''"
- data-file with)))))
- (grid :plot-cmd "splot"
- :plot-pre (lambda (_table _data-file _num-cols params _plot-str)
- (if (plist-get params :map) "set pm3d map" "set map"))
- :data-dump (lambda (table data-file params _num-cols)
- (let ((y-labels (org-plot/gnuplot-to-grid-data
- table data-file params)))
- (when y-labels (plist-put params :ylabels y-labels))))
- :plot-func
- (lambda (table data-file _num-cols params _plot-str)
- (let* ((type (plist-get params :plot-type))
- (with (if (eq type 'grid) 'pm3d (plist-get params :with))))
- (list (format "'%s' with %s title ''"
- data-file with)))))
- (radar :plot-func
- (lambda (table _data-file _num-cols params plot-str)
- (list (org--plot/radar table params)))))
- "List of plists describing the avalible plot types.
- The car is the type name, and the property :plot-func must be
- set. The value of :plot-func is a lambda which yields plot-lines
- (a list of strings) as the cdr.
- All lambda functions have the parameters of
- `org-plot/gnuplot-script' and PLOT-STR passed to them. i.e. they
- are called with the following signature: (TABLE DATA-FILE
- NUM-COLS PARAMS PLOT-STR)
- Potentially useful parameters in PARAMS include:
- :set :line :map :title :file :ind :timeind :timefmt :textind
- :deps :labels :xlabels :ylabels :xmin :xmax :ymin :ymax :ticks
- In addition to :plot-func, the following optional properties may
- be set.
- - :plot-cmd - A gnuplot command appended to each plot-line.
- Accepts string or nil. Default value: nil.
- - :check-ind-type - Whether the types of ind values should be checked.
- Accepts boolean.
- - :plot-str - the formula string passed to :plot-func as PLOT-STR
- Accepts string. Default value: \"'%s' using %s%d%s with %s title '%s'\"
- - :data-dump - Function to dump the table to a datafile for ease of
- use.
- Accepts lambda function. Default lambda body:
- (org-plot/gnuplot-to-data table data-file params)
- - :plot-pre - Gnuplot code to be inserted early into the script, just
- after term and output have been set.
- Accepts string, nil, or lambda function which returns string
- or nil. Defaults to nil."
- :group 'org-plot
- :type '(alist :value-type (symbol group)))
- (defvar org--plot/radar-template
- "### spider plot/chart with gnuplot
- # also known as: radar chart, web chart, star chart, cobweb chart,
- # radar plot, web plot, star plot, cobweb plot, etc. ...
- set datafile separator ' '
- set size square
- unset tics
- set angles degree
- set key bmargin center horizontal
- unset border
- # Load data and settup
- load \"%s\"
- # General settings
- DataColCount = words($Data[1])-1
- AxesCount = |$Data|-HeaderLines-1
- AngleOffset = 90
- Max = 1
- d=0.1*Max
- Direction = -1 # counterclockwise=1, clockwise = -1
- # Tic settings
- TicCount = %s
- TicOffset = 0.1
- TicValue(axis,i) = real(i)*(word($Settings[axis],3)-word($Settings[axis],2)) \\
- / word($Settings[axis],4)+word($Settings[axis],2)
- TicLabelPosX(axis,i) = PosX(axis,i/TicCount) + PosY(axis, TicOffset)
- TicLabelPosY(axis,i) = PosY(axis,i/TicCount) - PosX(axis, TicOffset)
- TicLen = 0.03
- TicdX(axis,i) = 0.5*TicLen*cos(alpha(axis)-90)
- TicdY(axis,i) = 0.5*TicLen*sin(alpha(axis)-90)
- # Label
- LabOffset = 0.10
- LabX(axis) = PosX(axis+1,Max+2*d) + PosY(axis, LabOffset)
- LabY(axis) = PosY($0+1,Max+2*d)
- # Functions
- alpha(axis) = (axis-1)*Direction*360.0/AxesCount+AngleOffset
- PosX(axis,R) = R*cos(alpha(axis))
- PosY(axis,R) = R*sin(alpha(axis))
- Scale(axis,value) = real(value-word($Settings[axis],2))/(word($Settings[axis],3)-word($Settings[axis],2))
- # Spider settings
- set style arrow 1 dt 1 lw 1.0 @fgal head filled size 0.06,25 # style for axes
- set style arrow 2 dt 2 lw 0.5 @fgal nohead # style for weblines
- set style arrow 3 dt 1 lw 1 @fgal nohead # style for axis tics
- set samples AxesCount
- set isosamples TicCount
- set urange[1:AxesCount]
- set vrange[1:TicCount]
- set style fill transparent solid 0.2
- set xrange[-Max-4*d:Max+4*d]
- set yrange[-Max-4*d:Max+4*d]
- plot \\
- '+' u (0):(0):(PosX($0,Max+d)):(PosY($0,Max+d)) w vec as 1 not, \\
- $Data u (LabX($0)): \\
- (LabY($0)):1 every ::HeaderLines w labels center enhanced @fgt not, \\
- for [i=1:DataColCount] $Data u (PosX($0+1,Scale($0+1,column(i+1)))): \\
- (PosY($0+1,Scale($0+1,column(i+1)))) every ::HeaderLines w filledcurves lt i title word($Data[1],i+1), \\
- %s
- # '++' u (PosX($1,$2/TicCount)-TicdX($1,$2/TicCount)): \\
- # (PosY($1,$2/TicCount)-TicdY($1,$2/TicCount)): \\
- # (2*TicdX($1,$2/TicCount)):(2*TicdY($1,$2/TicCount)) \\
- # w vec as 3 not, \\
- ### end of code
- ")
- (defvar org--plot/radar-ticks
- " '++' u (PosX($1,$2/TicCount)):(PosY($1,$2/TicCount)): \\
- (PosX($1+1,$2/TicCount)-PosX($1,$2/TicCount)): \\
- (PosY($1+1,$2/TicCount)-PosY($1,$2/TicCount)) w vec as 2 not, \\
- '++' u (TicLabelPosX(%s,$2)):(TicLabelPosY(%s,$2)): \\
- (sprintf('%%g',TicValue(%s,$2))) w labels font ',8' @fgat not")
- (defvar org--plot/radar-setup-template
- "# Data
- $Data <<HEREHAVESOMEDATA
- %s
- HEREHAVESOMEDATA
- HeaderLines = 1
- # Settings for scale and offset adjustments
- # axis min max tics axisLabelXoff axisLabelYoff
- $Settings <<EOD
- %s
- EOD
- ")
- (defun org--plot/radar (table params)
- (let* ((data
- (concat "\"" (mapconcat #'identity (plist-get params :labels) "\" \"") "\""
- "\n"
- (mapconcat (lambda (row)
- (format
- "\"%s\" %s"
- (car row)
- (mapconcat #'identity (cdr row) " ")))
- (append table (list (car table)))
- "\n")))
- (ticks (or (plist-get params :ticks)
- (org--plot/sensible-tick-num table
- (plist-get params :ymin)
- (plist-get params :ymax))))
- (settings
- (mapconcat (lambda (row)
- (let ((data (org--plot/values-stats
- (mapcar #'string-to-number (cdr row)))))
- (format
- "\"%s\" %s %s %s"
- (car row)
- (or (plist-get params :ymin)
- (plist-get data :nice-min))
- (or (plist-get params :ymax)
- (plist-get data :nice-max))
- (if (eq ticks 0) 2 ticks)
- )))
- (append table (list (car table)))
- "\n"))
- (setup-file (make-temp-file "org-plot-setup")))
- (let ((coding-system-for-write 'utf-8))
- (write-region (format org--plot/radar-setup-template data settings) nil setup-file nil :silent))
- (format org--plot/radar-template
- setup-file
- (if (eq ticks 0) 2 ticks)
- (if (eq ticks 0) ""
- (apply #'format org--plot/radar-ticks
- (make-list 3 (if (and (plist-get params :ymin)
- (plist-get params :ymax))
- ;; FIXME multi-drawing of tick labels with "1"
- "1" "$1")))))))
- (defcustom org-plot/gnuplot-term-extra ""
- "String or function which provides the extra term options.
- E.g. a value of \"size 1050,650\" would cause
- \"set term ... size 1050,650\" to be used.
- If a function, it is called with the plot type as the argument."
- :group 'org-plot
- :type '(choice string function))
- (defun org-plot/gnuplot-script (table data-file num-cols params &optional preface)
- "Write a gnuplot script to DATA-FILE respecting the options set in PARAMS.
- NUM-COLS controls the number of columns plotted in a 2-d plot.
- Optional argument PREFACE returns only option parameters in a
- manner suitable for prepending to a user-specified script."
- (let* ((type-name (plist-get params :plot-type))
- (type (cdr (assoc type-name org-plot/preset-plot-types))))
- (unless type
- (user-error "Org-plot type `%s' is undefined." type-name))
- (let* ((sets (plist-get params :set))
- (lines (plist-get params :line))
- (title (plist-get params :title))
- (file (plist-get params :file))
- (time-ind (plist-get params :timeind))
- (timefmt (plist-get params :timefmt))
- (x-labels (plist-get params :xlabels))
- (y-labels (plist-get params :ylabels))
- (plot-str (or (plist-get type :plot-str)
- "'%s' using %s%d%s with %s title '%s'"))
- (plot-cmd (plist-get type :plot-cmd))
- (plot-pre (plist-get type :plot-pre))
- (script "reset")
- ;; ats = add-to-script
- (ats (lambda (line) (when line (setf script (concat script "\n" line)))))
- plot-lines)
- ;; handle output file, background, and size
- (funcall ats (format "set term %s %s"
- (if file (file-name-extension file) "GNUTERM")
- (if (stringp org-plot/gnuplot-term-extra)
- org-plot/gnuplot-term-extra
- (funcall org-plot/gnuplot-term-extra type))))
- (when file ; output file
- (funcall ats (format "set output '%s'" file)))
- (when plot-pre
- (funcall ats (funcall plot-pre table data-file num-cols params plot-str)))
- (funcall ats
- (if (stringp org-plot/gnuplot-script-preamble)
- org-plot/gnuplot-script-preamble
- (funcall org-plot/gnuplot-script-preamble type)))
- (when title (funcall ats (format "set title '%s'" title))) ; title
- (mapc ats lines) ; line
- (dolist (el sets) (funcall ats (format "set %s" el))) ; set
- ;; Unless specified otherwise, values are TAB separated.
- (unless (string-match-p "^set datafile separator" script)
- (funcall ats "set datafile separator \"\\t\""))
- (when x-labels ; x labels (xtics)
- (funcall ats
- (format "set xtics (%s)"
- (mapconcat (lambda (pair)
- (format "\"%s\" %d" (cdr pair) (car pair)))
- x-labels ", "))))
- (when y-labels ; y labels (ytics)
- (funcall ats
- (format "set ytics (%s)"
- (mapconcat (lambda (pair)
- (format "\"%s\" %d" (cdr pair) (car pair)))
- y-labels ", "))))
- (when time-ind ; timestamp index
- (funcall ats "set xdata time")
- (funcall ats (concat "set timefmt \""
- (or timefmt ; timefmt passed to gnuplot
- "%Y-%m-%d-%H:%M:%S") "\"")))
- (unless preface
- (let ((type-func (plist-get type :plot-func)))
- (when type-func
- (setq plot-lines
- (funcall type-func table data-file num-cols params plot-str))))
- (funcall ats
- (concat plot-cmd
- (when plot-cmd " ")
- (mapconcat #'identity
- (reverse plot-lines)
- ",\\\n "))))
- script)))
- ;;-----------------------------------------------------------------------------
- ;; facade functions
- ;;;###autoload
- (defun org-plot/gnuplot (&optional params)
- "Plot table using gnuplot. Gnuplot options can be specified with PARAMS.
- If not given options will be taken from the +PLOT
- line directly before or after the table."
- (interactive)
- (require 'gnuplot)
- (save-window-excursion
- (delete-other-windows)
- (when (get-buffer "*gnuplot*") ; reset *gnuplot* if it already running
- (with-current-buffer "*gnuplot*"
- (goto-char (point-max))))
- (org-plot/goto-nearest-table)
- ;; Set default options.
- (dolist (pair org-plot/gnuplot-default-options)
- (unless (plist-member params (car pair))
- (setf params (plist-put params (car pair) (cdr pair)))))
- ;; Collect options.
- (save-excursion (while (and (equal 0 (forward-line -1))
- (looking-at "[[:space:]]*#\\+"))
- (setf params (org-plot/collect-options params))))
- ;; collect table and table information
- (let* ((data-file (make-temp-file "org-plot"))
- (table (let ((tbl (org-table-to-lisp)))
- (when (pcase (plist-get params :transpose)
- (`y t)
- (`yes t)
- (`t t))
- (if (not (memq 'hline tbl))
- (setq tbl (apply #'cl-mapcar #'list tbl))
- ;; When present, remove hlines as they can't (currentily) be easily transposed.
- (setq tbl (apply #'cl-mapcar #'list
- (remove 'hline tbl)))
- (push 'hline (cdr tbl))))
- tbl))
- (num-cols (length (if (eq (nth 0 table) 'hline) (nth 1 table)
- (nth 0 table))))
- (type (assoc (plist-get params :plot-type)
- org-plot/preset-plot-types)))
- (unless type
- (user-error "Org-plot type `%s' is undefined." (plist-get params :plot-type)))
- (run-with-idle-timer 0.1 nil #'delete-file data-file)
- (when (eq (cadr table) 'hline)
- (setf params
- (plist-put params :labels (car table))) ; headers to labels
- (setf table (delq 'hline (cdr table)))) ; clean non-data from table
- ;; Collect options.
- (save-excursion (while (and (equal 0 (forward-line -1))
- (looking-at "[[:space:]]*#\\+"))
- (setf params (org-plot/collect-options params))))
- ;; Dump table to datafile
- (if-let ((dump-func (plist-get type :data-dump)))
- (funcall dump-func table data-file num-cols params)
- (org-plot/gnuplot-to-data table data-file params))
- ;; Check type of ind column (timestamp? text?)
- (when (plist-get params :check-ind-type)
- (let* ((ind (1- (plist-get params :ind)))
- (ind-column (mapcar (lambda (row) (nth ind row)) table)))
- (cond ((< ind 0) nil) ; ind is implicit
- ((cl-every (lambda (el)
- (string-match org-ts-regexp3 el))
- ind-column)
- (plist-put params :timeind t)) ; ind holds timestamps
- ((or (string= (plist-get params :with) "hist")
- (cl-notevery (lambda (el)
- (string-match org-table-number-regexp el))
- ind-column))
- (plist-put params :textind t))))) ; ind holds text
- ;; Write script.
- (with-temp-buffer
- (if (plist-get params :script) ; user script
- (progn (insert
- (org-plot/gnuplot-script table data-file num-cols params t))
- (insert "\n")
- (insert-file-contents (plist-get params :script))
- (goto-char (point-min))
- (while (re-search-forward "\\$datafile" nil t)
- (replace-match data-file nil nil)))
- (insert (org-plot/gnuplot-script table data-file num-cols params)))
- ;; Graph table.
- (gnuplot-mode)
- (gnuplot-send-buffer-to-gnuplot))
- ;; Cleanup.
- (bury-buffer (get-buffer "*gnuplot*")))))
- (provide 'org-plot)
- ;; Local variables:
- ;; generated-autoload-file: "org-loaddefs.el"
- ;; End:
- ;;; org-plot.el ends here
|