1 ;;; org-plot.el --- Support for plotting from Org-mode
3 ;; Copyright (C) 2008-2014 Free Software Foundation, Inc.
5 ;; Author: Eric Schulte <schulte dot eric at gmail dot com>
6 ;; Keywords: tables, plotting
7 ;; Homepage: http://orgmode.org
9 ;; This file is part of GNU Emacs.
11 ;; GNU Emacs is free software: you can redistribute it and/or modify
12 ;; it under the terms of the GNU General Public License as published by
13 ;; the Free Software Foundation, either version 3 of the License, or
14 ;; (at your option) any later version.
16 ;; GNU Emacs is distributed in the hope that it will be useful,
17 ;; but WITHOUT ANY WARRANTY; without even the implied warranty of
18 ;; MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
19 ;; GNU General Public License for more details.
21 ;; You should have received a copy of the GNU General Public License
22 ;; along with GNU Emacs. If not, see <http://www.gnu.org/licenses/>.
26 ;; Borrows ideas and a couple of lines of code from org-exp.el.
28 ;; Thanks to the org-mode mailing list for testing and implementation
29 ;; and feature suggestions
37 (declare-function gnuplot-delchar-or-maybe-eof
"ext:gnuplot" (arg))
38 (declare-function gnuplot-mode
"ext:gnuplot" ())
39 (declare-function gnuplot-send-buffer-to-gnuplot
"ext:gnuplot" ())
41 (defvar org-plot
/gnuplot-default-options
45 "Default options to gnuplot used by `org-plot/gnuplot'.")
47 (defvar org-plot-timestamp-fmt nil
)
49 (defun org-plot/add-options-to-plist
(p options
)
50 "Parse an OPTIONS line and set values in the property list P.
51 Returns the resulting property list."
54 (let ((op '(("type" .
:plot-type
)
65 ("timeind" .
:timeind
)
66 ("timefmt" .
:timefmt
)))
67 (multiples '("set" "line"))
68 (regexp ":\\([\"][^\"]+?[\"]\\|[(][^)]+?[)]\\|[^ \t\n\r;,.]*\\)")
71 (while (setq o
(pop op
))
72 (if (member (car o
) multiples
) ;; keys with multiple values
74 (concat (regexp-quote (car o
)) regexp
)
76 (setq start
(match-end 0))
77 (setq p
(plist-put p
(cdr o
)
78 (cons (car (read-from-string
79 (match-string 1 options
)))
80 (plist-get p
(cdr o
)))))
82 (if (string-match (concat (regexp-quote (car o
)) regexp
)
84 (setq p
(plist-put p
(cdr o
)
85 (car (read-from-string
86 (match-string 1 options
)))))))))))
89 (defun org-plot/goto-nearest-table
()
90 "Move the point forward to the beginning of nearest table.
91 Return value is the point at the beginning of the table."
92 (interactive) (move-beginning-of-line 1)
93 (while (not (or (org-at-table-p) (< 0 (forward-line 1)))))
94 (goto-char (org-table-begin)))
96 (defun org-plot/collect-options
(&optional params
)
97 "Collect options from an org-plot '#+Plot:' line.
98 Accepts an optional property list PARAMS, to which the options
99 will be added. Returns the resulting property list."
101 (let ((line (thing-at-point 'line
)))
102 (if (string-match "#\\+PLOT: +\\(.*\\)$" line
)
103 (org-plot/add-options-to-plist params
(match-string 1 line
))
106 (defun org-plot-quote-timestamp-field (s)
107 "Convert field S from timestamp to Unix time and export to gnuplot."
108 (format-time-string org-plot-timestamp-fmt
(org-time-string-to-time s
)))
110 (defun org-plot-quote-tsv-field (s)
111 "Quote field S for export to gnuplot."
112 (if (or (string-match org-table-number-regexp s
)
113 (plist-get params
:timeind
)) s
114 (if (string-match org-ts-regexp3 s
)
115 (org-plot-quote-timestamp-field s
)
116 (concat "\"" (mapconcat 'identity
(split-string s
"\"") "\"\"") "\""))))
118 (defun org-plot/gnuplot-to-data
(table data-file params
)
119 "Export TABLE to DATA-FILE in a format readable by gnuplot.
120 Pass PARAMS through to `orgtbl-to-generic' when exporting TABLE."
123 (make-local-variable 'org-plot-timestamp-fmt
)
124 (setq org-plot-timestamp-fmt
(or
125 (plist-get params
:timefmt
)
126 "%Y-%m-%d-%H:%M:%S"))
127 (insert (orgtbl-to-generic
130 '(:sep
"\t" :fmt org-plot-quote-tsv-field
)
134 (defun org-plot/gnuplot-to-grid-data
(table data-file params
)
135 "Export the data in TABLE to DATA-FILE for gnuplot.
136 This means in a format appropriate for grid plotting by gnuplot.
137 PARAMS specifies which columns of TABLE should be plotted as independent
138 and dependant variables."
140 (let* ((ind (- (plist-get params
:ind
) 1))
141 (deps (if (plist-member params
:deps
)
142 (mapcar (lambda (val) (- val
1)) (plist-get params
:deps
))
144 (dotimes (col (length (first table
)))
145 (setf collector
(cons col collector
)))
149 (when (>= ind
0) ;; collect values of ind col
150 (setf row-vals
(mapcar (lambda (row) (setf counter
(+ 1 counter
))
151 (cons counter
(nth ind row
))) table
)))
152 (when (or deps
(>= ind
0)) ;; remove non-plotting columns
153 (setf deps
(delq ind deps
))
154 (setf table
(mapcar (lambda (row)
155 (dotimes (col (length row
))
156 (unless (memq col deps
)
157 (setf (nth col row
) nil
)))
160 ;; write table to gnuplot grid datafile format
161 (with-temp-file data-file
162 (let ((num-rows (length table
)) (num-cols (length (first table
)))
163 (gnuplot-row (lambda (col row value
)
164 (setf col
(+ 1 col
)) (setf row
(+ 1 row
))
165 (format "%f %f %f\n%f %f %f\n"
166 col
(- row
0.5) value
;; lower edge
167 col
(+ row
0.5) value
))) ;; upper edge
168 front-edge back-edge
)
169 (dotimes (col num-cols
)
170 (dotimes (row num-rows
)
173 (funcall gnuplot-row
(- col
1) row
174 (string-to-number (nth col
(nth row table
))))))
177 (funcall gnuplot-row col row
178 (string-to-number (nth col
(nth row table
)))))))
179 ;; only insert once per row
180 (insert back-edge
) (insert "\n") ;; back edge
181 (insert front-edge
) (insert "\n") ;; front edge
182 (setf back-edge
"") (setf front-edge
""))))
185 (defun org-plot/gnuplot-script
(data-file num-cols params
&optional preface
)
186 "Write a gnuplot script to DATA-FILE respecting the options set in PARAMS.
187 NUM-COLS controls the number of columns plotted in a 2-d plot.
188 Optional argument PREFACE returns only option parameters in a
189 manner suitable for prepending to a user-specified script."
190 (let* ((type (plist-get params
:plot-type
))
191 (with (if (equal type
'grid
)
193 (plist-get params
:with
)))
194 (sets (plist-get params
:set
))
195 (lines (plist-get params
:line
))
196 (map (plist-get params
:map
))
197 (title (plist-get params
:title
))
198 (file (plist-get params
:file
))
199 (ind (plist-get params
:ind
))
200 (time-ind (plist-get params
:timeind
))
201 (timefmt (plist-get params
:timefmt
))
202 (text-ind (plist-get params
:textind
))
203 (deps (if (plist-member params
:deps
) (plist-get params
:deps
)))
204 (col-labels (plist-get params
:labels
))
205 (x-labels (plist-get params
:xlabels
))
206 (y-labels (plist-get params
:ylabels
))
207 (plot-str "'%s' using %s%d%s with %s title '%s'")
213 ; ats = add-to-script
214 (ats (lambda (line) (setf script
(format "%s\n%s" script line
))))
216 (when file
;; output file
217 (funcall ats
(format "set term %s" (file-name-extension file
)))
218 (funcall ats
(format "set output '%s'" file
)))
221 ('3d
(if map
(funcall ats
"set map")))
222 ('grid
(if map
(funcall ats
"set pm3d map")
223 (funcall ats
"set pm3d"))))
224 (when title
(funcall ats
(format "set title '%s'" title
))) ;; title
225 (when lines
(mapc (lambda (el) (funcall ats el
)) lines
)) ;; line
227 (mapc (lambda (el) (funcall ats
(format "set %s" el
))) sets
))
228 (when x-labels
;; x labels (xtics)
230 (format "set xtics (%s)"
231 (mapconcat (lambda (pair)
232 (format "\"%s\" %d" (cdr pair
) (car pair
)))
234 (when y-labels
;; y labels (ytics)
236 (format "set ytics (%s)"
237 (mapconcat (lambda (pair)
238 (format "\"%s\" %d" (cdr pair
) (car pair
)))
240 (when time-ind
;; timestamp index
241 (funcall ats
"set xdata time")
242 (funcall ats
(concat "set timefmt \""
243 (or timefmt
;; timefmt passed to gnuplot
244 "%Y-%m-%d-%H:%M:%S") "\"")))
246 (case type
;; plot command
247 ('2d
(dotimes (col num-cols
)
248 (unless (and (equal type
'2d
)
249 (or (and ind
(equal (+ 1 col
) ind
))
250 (and deps
(not (member (+ 1 col
) deps
)))))
253 (format plot-str data-file
254 (or (and ind
(> ind
0)
256 (format "%d:" ind
)) "")
258 (if text-ind
(format ":xticlabel(%d)" ind
) "")
260 (or (nth col col-labels
) (format "%d" (+ 1 col
))))
263 (setq plot-lines
(list (format "'%s' matrix with %s title ''"
266 (setq plot-lines
(list (format "'%s' with %s title ''"
269 (concat plot-cmd
" " (mapconcat 'identity
(reverse plot-lines
) ",\\\n "))))
272 ;;-----------------------------------------------------------------------------
275 (defun org-plot/gnuplot
(&optional params
)
276 "Plot table using gnuplot. Gnuplot options can be specified with PARAMS.
277 If not given options will be taken from the +PLOT
278 line directly before or after the table."
281 (save-window-excursion
282 (delete-other-windows)
283 (when (get-buffer "*gnuplot*") ;; reset *gnuplot* if it already running
284 (with-current-buffer "*gnuplot*"
285 (goto-char (point-max))
286 (gnuplot-delchar-or-maybe-eof nil
)))
287 (org-plot/goto-nearest-table
)
288 ;; set default options
291 (unless (plist-member params
(car pair
))
292 (setf params
(plist-put params
(car pair
) (cdr pair
)))))
293 org-plot
/gnuplot-default-options
)
294 ;; collect table and table information
295 (let* ((data-file (make-temp-file "org-plot"))
296 (table (org-table-to-lisp))
297 (num-cols (length (if (eq (first table
) 'hline
) (second table
)
299 (while (equal 'hline
(first table
)) (setf table
(cdr table
)))
300 (when (equal (second table
) 'hline
)
301 (setf params
(plist-put params
:labels
(first table
))) ;; headers to labels
302 (setf table
(delq 'hline
(cdr table
)))) ;; clean non-data from table
304 (save-excursion (while (and (equal 0 (forward-line -
1))
305 (looking-at "[[:space:]]*#\\+"))
306 (setf params
(org-plot/collect-options params
))))
307 ;; dump table to datafile (very different for grid)
308 (case (plist-get params
:plot-type
)
309 ('2d
(org-plot/gnuplot-to-data table data-file params
))
310 ('3d
(org-plot/gnuplot-to-data table data-file params
))
311 ('grid
(let ((y-labels (org-plot/gnuplot-to-grid-data
312 table data-file params
)))
313 (when y-labels
(plist-put params
:ylabels y-labels
)))))
314 ;; check for timestamp ind column
315 (let ((ind (- (plist-get params
:ind
) 1)))
316 (when (and (>= ind
0) (equal '2d
(plist-get params
:plot-type
))
317 (not (or (plist-get params
:timeind
)
318 (plist-get params
:textind
))))
322 (if (string-match org-ts-regexp3 el
)
324 (mapcar (lambda (row) (nth ind row
)) table
)))) 0)
325 (plist-put params
:timeind t
)
326 ;; check for text ind column
327 (if (or (string= (plist-get params
:with
) "hist")
331 (if (string-match org-table-number-regexp el
)
333 (mapcar (lambda (row) (nth ind row
)) table
)))) 0))
334 (plist-put params
:textind t
)))))
337 (if (plist-get params
:script
) ;; user script
339 (org-plot/gnuplot-script data-file num-cols params t
))
341 (insert-file-contents (plist-get params
:script
))
342 (goto-char (point-min))
343 (while (re-search-forward "$datafile" nil t
)
344 (replace-match data-file nil nil
)))
346 (org-plot/gnuplot-script data-file num-cols params
)))
349 (gnuplot-send-buffer-to-gnuplot))
351 (bury-buffer (get-buffer "*gnuplot*"))
352 (run-with-idle-timer 0.1 nil
(lambda () (delete-file data-file
))))))
357 ;; generated-autoload-file: "org-loaddefs.el"
360 ;;; org-plot.el ends here