Fix more copyright year for Org files.

[gnu-emacs] / lisp / org / ob-calc.el

diff --git a/lisp/org/ob-calc.el b/lisp/org/ob-calc.el
index ce752008b4ce5d1cbecbccc331c0461a4e31dbee..6686c2754bf45058ccef914d4f3c7f31cc87c3b4 100644 (file)
--- a/lisp/org/ob-calc.el
+++ b/lisp/org/ob-calc.el
@@ -1,11 +1,10 @@
 ;;; ob-calc.el --- org-babel functions for calc code evaluation
 
-;; Copyright (C) 2010  Free Software Foundation, Inc
+;; Copyright (C) 2010-2012  Free Software Foundation, Inc
 
 ;; Author: Eric Schulte
 ;; Keywords: literate programming, reproducible research
 ;; Homepage: http://orgmode.org
-;; Version: 7.3
 
 ;; This file is part of GNU Emacs.
 
@@ -29,9 +28,15 @@
 ;;; Code:
 (require 'ob)
 (require 'calc)
-(require 'calc-trail)
+(unless (featurep 'xemacs)
+  (require 'calc-trail)
+  (require 'calc-store))
 (eval-when-compile (require 'ob-comint))
 
+(declare-function calc-store-into    "calc-store" (&optional var))
+(declare-function calc-recall        "calc-store" (&optional var))
+(declare-function math-evaluate-expr "calc-ext"   (x))
+
 (defvar org-babel-default-header-args:calc nil
   "Default arguments for evaluating an calc source block.")
 
@@ -40,28 +45,64 @@
 
 (defun org-babel-execute:calc (body params)
   "Execute a block of calc code with Babel."
-  (mapcar
-   (lambda (line)
-     (when (> (length line) 0)
-       (if (string= "'" (substring line 0 1))
-          (funcall (lookup-key calc-mode-map (substring line 1)) nil)
-        (calc-push-list
-         (list ((lambda (res)
-                  (cond
-                   ((numberp res) res)
-                   ((listp res) (error "calc error \"%s\" on input \"%s\""
-                                       (cadr res) line))
-                   (t res))
-                  (if (numberp res) res (math-read-number res)))
-                (calc-eval line)))))))
-   (mapcar #'org-babel-trim
-          (split-string (org-babel-expand-body:calc body params) "[\n\r]")))
+  (unless (get-buffer "*Calculator*")
+    (save-window-excursion (calc) (calc-quit)))
+  (let* ((vars (mapcar #'cdr (org-babel-get-header params :var)))
+        (var-syms (mapcar #'car vars))
+        (var-names (mapcar #'symbol-name var-syms)))
+    (mapc
+     (lambda (pair)
+       (calc-push-list (list (cdr pair)))
+       (calc-store-into (car pair)))
+     vars)
+    (mapc
+     (lambda (line)
+       (when (> (length line) 0)
+        (cond
+         ;; simple variable name
+         ((member line var-names) (calc-recall (intern line)))
+         ;; stack operation
+         ((string= "'" (substring line 0 1))
+          (funcall (lookup-key calc-mode-map (substring line 1)) nil))
+         ;; complex expression
+         (t
+          (calc-push-list
+           (list ((lambda (res)
+                    (cond
+                     ((numberp res) res)
+                     ((math-read-number res) (math-read-number res))
+                     ((listp res) (error "calc error \"%s\" on input \"%s\""
+                                         (cadr res) line))
+                     (t (replace-regexp-in-string
+                         "'\\[" "["
+                         (calc-eval
+                          (math-evaluate-expr
+                           ;; resolve user variables, calc built in
+                           ;; variables are handled automatically
+                           ;; upstream by calc
+                           (mapcar #'ob-calc-maybe-resolve-var
+                                   ;; parse line into calc objects
+                                   (car (math-read-exprs line)))))))))
+                  (calc-eval line))))))))
+     (mapcar #'org-babel-trim
+            (split-string (org-babel-expand-body:calc body params) "[\n\r]"))))
   (save-excursion
-    (set-buffer (get-buffer "*Calculator*"))
-    (calc-eval (calc-top 1))))
+    (with-current-buffer (get-buffer "*Calculator*")
+      (calc-eval (calc-top 1)))))
+
+(defvar var-syms) ; Dynamically scoped from org-babel-execute:calc
+(defun ob-calc-maybe-resolve-var (el)
+  (if (consp el)
+      (if (and (equal 'var (car el)) (member (cadr el) var-syms))
+         (progn
+           (calc-recall (cadr el))
+           (prog1 (calc-top 1)
+             (calc-pop 1)))
+       (mapcar #'ob-calc-maybe-resolve-var el))
+    el))
 
 (provide 'ob-calc)
 
-;; arch-tag: 5c57a3b7-5818-4c6c-acda-7a94831a6449
+
 
 ;;; ob-calc.el ends here