infix-math

2021-10-21

An extensible infix syntax for math in Common Lisp.

Upstream URL

github.com/ruricolist/infix-math

Author

Paul M. Rodriguez <pmr@ruricolist.com>

License

MIT
README

Infix-Math

Infix-Math is a library that provides a special-purpose syntax for transcribing mathematical formulas into Lisp.

Bitter experience has taught me that the more the formula on screen resembles the formula on paper, the better. The more the formula on screen resembles the formula on paper, the easier it is to prevent bugs from transcription errors. The easier it is to prevent transcription errors, the easier it is to trace the source of any bugs that do occur – because sometimes the formula is wrong.

(Having to transcribe formulas from crooked, blurry scans of ancient pre-LaTeX typescripts is bad enough without having to parse operator precedence in your head.)

Even if you end up rewriting the formula for speed or numerical stability, having the specification in an executable form is invaluable for reference and testing.

Examples

The macro $ is the entry point into Infix-Math.

($ 2 + 2)     => 4
($ 1 + 2 * 3) => 7

Operator precedence parsing in Infix-Math is reliable – it uses Dijkstra’s shunting yard algorithm.

The parser automatically descends into function argument lists, which means that the total number of parentheses is never greater than it would be in a purely infix language.

($ (tan pi * (p - 1/2)))
≡ (tan (* pi (- p 1/2)))
≅ tan(pi*(p-0.5))

Common subexpression elimination is automatic and aggressive. All forms are assumed to be pure. Math does not have side effects.

(macroexpand '($ 2 ^ 2x * 2 ^ 2x)
=> ‘(let ((#:subexp11325 (^ 2 (* 2 x))))
      (* #:subexp11325 #:subexp11325))

Infix-Math knows about the following arithmetic and bitwise operators, in descending order of precedence.

  • unary -, sqrt
  • expt, log
  • *, /, rem, mod, floor, ffloor, ceiling, fceiling, truncate, ftruncate, round, fround, scale-float, gcd, lcm, atan
  • +, -
  • ash
  • logand, logandc1, logandc2, lognand
  • logxor, logeqv
  • logior, logorc1, logorc2, lognor
  • min, max
  • over

Operations at the same level of precedence are always evaluated left-to-right.

(+ 0.1d0 (+ 0.2d0 0.3d0)) => 0.6d0
(+ (+ 0.1d0 0.2d0) 0.3d0) => 0.6000000000000001D0
($ 0.1d0 + 0.2d0 + 0.3d0) => 0.6000000000000001D0

Parentheses can be used for grouping.

($ 0.1d0 + (0.2d0 + 0.3d0)) => 0.6d0

Variables can be written with literal numbers as coefficients.

($ 2x)  => 10
($ -2x) => 10

Literal coefficients have very high priority.

($ 2 ^ 2 * x) ≡ (* (expt 2 2) x)     => 20
($ 2 ^ 2x)    ≡ (expt 2 (* 2 x))     => 1024

A literal coefficient of 1 can be omitted.

($ -x) ≡ ($ -1x) ≡ (* -1 x)

Literal coefficients are parsed as decimals, rather than floats.

($ 1.5x) ≡ (* 3/2 x)

You can also use fractions as literal coefficients.

($ 1/3x) ≡ (* 1/3 x)

Among other things, literal coefficients are very convenient for units of measurement.

(The idea for literal coefficients comes from Julia.)

Symbols

Infix-Math exports only five symbols: $, ^, over, and two macros for declaring operators: declare-unary-operator and declare-binary-operator.

The symbol ^ is just a shorthand for expt.

($ 1 + 2 * 3 ^ 4) => 163

(^ is from Dylan.)

The symbol over represents the same operation as /, but at a much lower priority. Using over lets you avoid introducing parentheses for grouping when transcribing fractions.

(setf x 5)
($ x * 2 / x * 3)     ≡ (* (/ (* x 2) x) 3) => 6
($ (x * 2) / (x * 3)) ≡ (/ (* x 2) (* x 3)) => 2/3
($ x * 2 over x * 3)  ≡ (/ (* x 2) (* x 3)) => 2/3

You can also spell over with a series of dashes or underscores.

($ x * 2
   -----
   x * 3)
=> 2/3

If you want more math symbols, the package infix-math/symbols provides a few more.

Calculator

You can use Infix-Math to turn your REPL into a calculator.

First, load the infix-math/calc system:

(asdf:load-system "infix-math/calc")

Then, at the REPL, start the calculator:

(infix-math/calc:calc)

This will put you at a calculator prompt. You can type in mathematical expressions directly:

$> 2 + 2
4

A single form entered at the REPL is interpreted as ordinary CL.

$> *package*
:infix-math/calc-user

You can assign to variables using the <- operator.

$> x <- 2 + 2
4
$> x
4

Certain one-letter variables are provided for you to assign to, such as x, y, and z. You can see the full list by evaluating :v at the calculator prompt.

To quit, use :q. The value of the last expression evaluated will be returned.

$> 2 + 2
4
$> :q
4
CL-USER> *
4

Extending

Infix-Math is easily to extend. In fact, you may not even need to extend it.

Any symbol that consists entirely of operator characters is interpreted as an infix operator, with the highest non-unary priority. Operator characters are anything but dashes, underscores, whitespace or alphanumeric characters.

(defun <*> (x y)
  "Matrix multiplication, maybe."
  ...)

(macroexpand '($ x * y <*> z)) => (* x (<*> y z))

(This approach is taken from Haskell.)

You can use any function as an infix operator by surrounding its name with dots.

(defun choose (n k)
  "Binomial coefficient, maybe."
  ...)

(macroexpand '($ n .choose. k)) => '(choose n k)

Again, the operator has the highest non-unary priority.

(This approach is taken from Haskell and Fortran.)

If you need more flexibility, declare the operators using declare-binary-operator or declare-unary-operator.

To declare a unary operator:

(declare-unary-operator √)

To copy the precedence of another operator:

(declare-binary-operator <*> :from *)

To declare an operator right-associative:

(declare-binary-operator ?
  :from *
  :right-associative t)

Dependencies (5)

  • alexandria
  • asdf-package-system
  • parse-number
  • serapeum
  • wu-decimal

Dependents (2)

  • GitHub
  • Quicklisp