Commit | Line | Data |
---|---|---|
3ab2c837 BG |
1 | ;;; ob-awk.el --- org-babel functions for awk evaluation |
2 | ||
ba318903 | 3 | ;; Copyright (C) 2011-2014 Free Software Foundation, Inc. |
3ab2c837 BG |
4 | |
5 | ;; Author: Eric Schulte | |
6 | ;; Keywords: literate programming, reproducible research | |
7 | ;; Homepage: http://orgmode.org | |
3ab2c837 BG |
8 | |
9 | ;; This file is part of GNU Emacs. | |
10 | ||
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. | |
15 | ||
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. | |
20 | ||
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/>. | |
23 | ||
24 | ;;; Commentary: | |
25 | ||
3ab2c837 | 26 | ;; Babel's awk can use special header argument: |
14e1337f | 27 | ;; |
3ab2c837 | 28 | ;; - :in-file takes a path to a file of data to be processed by awk |
14e1337f | 29 | ;; |
3ab2c837 BG |
30 | ;; - :stdin takes an Org-mode data or code block reference, the value |
31 | ;; of which will be passed to the awk process through STDIN | |
32 | ||
33 | ;;; Code: | |
34 | (require 'ob) | |
8223b1d2 | 35 | (require 'org-compat) |
3ab2c837 BG |
36 | (eval-when-compile (require 'cl)) |
37 | ||
38 | (declare-function org-babel-ref-resolve "ob-ref" (ref)) | |
39 | (declare-function orgtbl-to-generic "org-table" (table params)) | |
40 | ||
41 | (defvar org-babel-tangle-lang-exts) | |
42 | (add-to-list 'org-babel-tangle-lang-exts '("awk" . "awk")) | |
43 | ||
44 | (defvar org-babel-awk-command "awk" | |
45 | "Name of the awk executable command.") | |
46 | ||
271672fa | 47 | (defun org-babel-expand-body:awk (body params) |
3ab2c837 BG |
48 | "Expand BODY according to PARAMS, return the expanded body." |
49 | (dolist (pair (mapcar #'cdr (org-babel-get-header params :var))) | |
50 | (setf body (replace-regexp-in-string | |
e66ba1df | 51 | (regexp-quote (format "$%s" (car pair))) (cdr pair) body))) |
3ab2c837 BG |
52 | body) |
53 | ||
54 | (defun org-babel-execute:awk (body params) | |
55 | "Execute a block of Awk code with org-babel. This function is | |
56 | called by `org-babel-execute-src-block'" | |
57 | (message "executing Awk source code block") | |
58 | (let* ((result-params (cdr (assoc :result-params params))) | |
59 | (cmd-line (cdr (assoc :cmd-line params))) | |
60 | (in-file (cdr (assoc :in-file params))) | |
61 | (full-body (org-babel-expand-body:awk body params)) | |
666ffc7e SM |
62 | (code-file (let ((file (org-babel-temp-file "awk-"))) |
63 | (with-temp-file file (insert full-body)) file)) | |
64 | (stdin (let ((stdin (cdr (assoc :stdin params)))) | |
3ab2c837 BG |
65 | (when stdin |
66 | (let ((tmp (org-babel-temp-file "awk-stdin-")) | |
67 | (res (org-babel-ref-resolve stdin))) | |
68 | (with-temp-file tmp | |
69 | (insert (org-babel-awk-var-to-awk res))) | |
666ffc7e | 70 | tmp)))) |
3ab2c837 BG |
71 | (cmd (mapconcat #'identity (remove nil (list org-babel-awk-command |
72 | "-f" code-file | |
73 | cmd-line | |
74 | in-file)) | |
75 | " "))) | |
76 | (org-babel-reassemble-table | |
666ffc7e SM |
77 | (let ((results |
78 | (cond | |
79 | (stdin (with-temp-buffer | |
80 | (call-process-shell-command cmd stdin (current-buffer)) | |
81 | (buffer-string))) | |
82 | (t (org-babel-eval cmd ""))))) | |
83 | (when results | |
84 | (org-babel-result-cond result-params | |
85 | results | |
86 | (let ((tmp (org-babel-temp-file "awk-results-"))) | |
87 | (with-temp-file tmp (insert results)) | |
88 | (org-babel-import-elisp-from-file tmp))))) | |
3ab2c837 BG |
89 | (org-babel-pick-name |
90 | (cdr (assoc :colname-names params)) (cdr (assoc :colnames params))) | |
91 | (org-babel-pick-name | |
92 | (cdr (assoc :rowname-names params)) (cdr (assoc :rownames params)))))) | |
93 | ||
94 | (defun org-babel-awk-var-to-awk (var &optional sep) | |
95 | "Return a printed value of VAR suitable for parsing with awk." | |
8223b1d2 | 96 | (let ((echo-var (lambda (v) (if (stringp v) v (format "%S" v))))) |
3ab2c837 BG |
97 | (cond |
98 | ((and (listp var) (listp (car var))) | |
8223b1d2 | 99 | (orgtbl-to-generic var (list :sep (or sep "\t") :fmt echo-var))) |
3ab2c837 | 100 | ((listp var) |
8223b1d2 BG |
101 | (mapconcat echo-var var "\n")) |
102 | (t (funcall echo-var var))))) | |
3ab2c837 BG |
103 | |
104 | (defun org-babel-awk-table-or-string (results) | |
105 | "If the results look like a table, then convert them into an | |
106 | Emacs-lisp table, otherwise return the results as a string." | |
107 | (org-babel-script-escape results)) | |
108 | ||
109 | (provide 'ob-awk) | |
110 | ||
5b409b39 | 111 | |
3ab2c837 BG |
112 | |
113 | ;;; ob-awk.el ends here |