Pattern matching

Introduction

Pattern matching is an integral part of a computation toolkit. It allows to isolate and modify parts of an expression with replacement rules. A properly broad pattern matcher is effectively its own Turing complete programming language!

For example, Conway’s game of life can be completely emulated by pattern matching, where each cell together with its surrounding cells determined the future value of the cell.

Game of life

Other common uses of pattern matching is in regular expression (regex), widely used for matching patterns in text. The following pattern

^[a-zA-Z0-9]+@gmail\.com$

matches e-mail addresses with at least one alphanumerical character and ending in @gmail.com.

To faithfully parse an e-mail address, a horrendous pattern match is required!

Pattern matching is also ubiquitous in mathematics. For example, the recursive definition of the factorial function

\[ \begin{align} f(0) &= 1\\ f(n) &= n f(n - 1);\quad n > 0 \end{align} \]

is a pattern match. First, the pattern \(f(0)\) is tested. If it is matched, it is replaced by 1. Then the pattern \(f(n)\) is matched, where \(n\) needs to be positive and it is replaced by \(n f(n-1)\), where the \(n\) takes the value of the \(n\) matched in the pattern.

Symbolica has rich support for pattern matching. Let us see some examples where we match and replace all literal parts of the expression.

In this part we use the Python API. The Rust library examples will follow later. We match \(x\) in the expression \(x+f(x) + 5\) and replace it by 6.

from symbolica import Expression
x = Expression.symbol('x')
f = Expression.symbol('f')
e = x + f(x) + 5
e = e.replace_all(x, 6)
print(e)

We get 11 + f(6)

To replace f(x) by 6, we change the pattern:

e = e.replace_all(f(x), 6)

which yields 11 + x.

Wildcards

Let us now try to replicate the factorial function pattern match mentioned above. For that we need to be able to match to any numerical value in the function f and we do not know that value in advance. Writing out an explicit rule for every desired number, e.g., f(40) = 40*f(39) is possible, but tedious. Instead, we use a wildcard that can match any subexpression. The specific rules will be explained later.

A wildcard is defined as a variable ending in a _ (note that the name of the Python variable does not matter). We define a wildcard n_ and write the rule:

from symbolica import Expression
n_ = Expression.symbol('n_')
f = Expression.symbol('f')
e = f(4)
e = e.replace_all(f(0), 1)
e = e.replace_all(f(n_), n_*f(n_-1))
print(e)

This is almost perfect; the pattern still matches to negative numbers, and non-numbers. We can restrict the pattern by adding a third argument to replace_all:

e = e.replace_all(f(n_), n_*f(n_-1), n_.req_gt(0))

Rules

We now set out all the rules for when patterns match.

Syntactic matching

Patterns match syntactically / structurally, which means that the patten must appear explicitly in the tree representation of the expression. For example, the pattern \(x^2\) is not found in the expression \((x+1)^2\), even though it appears in the mathematically equivalent (but structurally different) expression \(x^2+2x+1\).

The pattern does not necessarily have to appear verbatim in the tree structure though. Symbolica uses mathematical properties such as the symmetry of the multiplication and addition operator, and of symmetric functions, to match atoms in unordered fashion. For example \(x + z\) matches \(x+y+z\), even though \(y\) is “in between” in the normal form.

Since matches on products, sums and symmetric functions are unordered, in order to establish that there is no match, in the worst case all permutations or subsets have to be tested. This could be expensive. Providing constraints on the pattern, especially length constraints, will help reduce the cost.

Wildcards

Wildcards are variables ending in one or more underscores _. The number of underscores signify various restrictions on the wildcard length. Wildcards can match any subexpression and will always include the operator it matches. For example, matching x___ (no length restrictions) on a*b*c yields the following options:

flowchart TD
    Mul --> a
    Mul --> b
    Mul --> c

flowchart TD
    Mul --> a
    Mul --> b

flowchart TD
    Mul --> a
    Mul --> c

flowchart TD
    Mul --> b
    Mul --> c

flowchart TD
    Mul --> a

flowchart TD
    Mul --> b

flowchart TD
    Mul --> c

flowchart TD
    Mul

Note that the empty set is one of the options. Because of the rule of syntactic matching, the pattern y + x___ is not present in the expression y even though x_ can be the empty set.

The cases where one or zero sub-atoms are matched is subject to downcasting. For example,

flowchart LR
    Mul --> a

is transformed to

flowchart LR
    a

which means that the atom type of x___ is no longer Mul. The atom type of the empty match is Empty. Downcasting ensures that repeated patterns such as x_*f(x_) match to x*f(x) even though the first x_ matches Mul Var x and the second one Arg Var x.

Function arguments behave analogously to the case of Mul described above, except that the order is preserved if the function is non-symmetric. The pattern matcher interprets function arguments as an operator of type Arg. For example, when matching f(x_) to f(a,b,c) the pattern matcher views the arguments as

flowchart TD
    Arg --> a
    Arg --> b
    Arg --> c

When a wildcard that matches multiple arguments is used outside of a function context on the right-hand side of a replacement, the matched arguments will be wrapped in the internal function arg. For example:

e = f(1,2).replace_all(f(x___), x___)

yields arg(1,2). An arg function that is placed into a function context again, is flattened during normalization, e.g., f(arg(1,2)) = f(1,2).

Repeated wildcards

A repeated wildcard means that the matched value must evaluate to exactly the same atoms. For example:

e = x*y*f(x*y)
e = e.replace_all(x___*f(x___), 1)

matches because x___ matches to x*y outside of the function and inside the function. See Examples for more complicated examples.

Precedence rules

The expression that is tested first is the root node with all its children. If that does not match, the subset of the root node with one sub-atoms removed is tested. If no match is found, two sub-atoms are removed, etc. Once all subsets are tested, the pattern matcher repeats the process for each of the sub-atoms as the new root.

For example, the pattern x_ will match any subexpression in any expression, so it will match the root first:

e = x*y*z
e = e.replace_all(x_, 1)

will yield 1, since x_ matches x*y*z.

In the following case the pattern matches at the root level, so a match is returned, even though the child would have matched too:

e = f(f(x))
e = e.replace_all(f(x_), f(1))

will yield f(1), since x_ matches f(x).

In the following case the root did not match, so the sub-atom is matched:

e = g(f(x))
e = e.replace_all(f(x_), f(1))

will yield g(f(1)), since x_ matches f(x).

replace_all will always replace the first match found for every non-overlapping match. To iterate matches, see Match iteration.

Ordering

A wildcard will match to a given set of atoms only once and it will have the order of the normal form. This means that for a symmetric function f(1,2,3), f(x___) will generate one match, x___ = 1,2,3, and it will never permute through all options (e.g. x___ = 2,3,1). Consequently,

f = Expression.symbol('f')
f_sym = Expression.symbol('fsym', is_symmetric=True)
e = f(3,2,1)*fsym(1,2,3)
e = e.replace_all(f(x___)*fsym(x___), f(1))

will not match.

Conditions

Wildcards can be restricted based on arbitrary conditions. Below are some built-in options.

Wildcard length

It is possible to filter on the length of the wildcard before downcasting. The length is defined as the number of sub-atoms for Add, Mul, and Arg and is 1 for any other atom.

Note

A function has a length of 1 (it counts as one atom) regardless of the number of arguments. This way x_ with a wildcard length of 1 matches f(...) regardless of the number arguments.

The number of underscores at the end of the wildcard name determines pre-set length restrictions. We have:

  • x_: must match a single atom
  • x__: must match one or more atoms
  • x___: must match zero or more atoms

The length of x__ and x___ can be further restricted. For example, let us restrict the wildcard x__ to a length between 2 and 3

e1 = f(1).replace_all(f(x__), 1, x_.req_len(2, 3))
e2 = f(1,2).replace_all(f(x__), 1, x_.req_len(2, 3))
e3 = f(1,2,3).replace_all(f(x__), 1, x_.req_len(2, 3))

then only e2 and e3 will yield 1.

Even though the wildcard length of x_ is restricted to 1 and x*y has a length of 2,

e1 = f(x*y).replace_all(f(x_, 1))

we get the match x_ = x*y. This is because the length constraint is applied before downcasting.

A wildcard length restriction check is applied at any point that a wildcard is used and therefore

x*y*f(x*y).replace_all(x__*f(x__), 1, x__.req_len(2, 2))

does not match, since x__ has wildcard length if 1 inside the function argument. To restrict the length of the atom matched to x__ after downcasting, use a filter function.

Type restriction

A wildcard can be restricted based on the atom type.

For example, x_ can only match a variable:

e1 = f(x).replace_all(f(x_), 1, x_.req_type(AtomType.Var))
Filter function

Custom filters can be added to restrict any wildcard. This is a function receives the wildcard name and its match expression as an argument and should return a boolean. If true, the match is accepted.

from symbolica import Expression
x_ = Expression.symbol('x_')
f = Expression.symbol('f')
e = f(1)*f(2)*f(3)
e = e.replace_all(f(x_), 1, x_.req(lambda m: m == 2 or m == 3))
Tip

Since a Python function is called, this operation has significant overhead. For high-performance code, consider implementing the filter in Rust, or use one of the built-in filters, such as req_gt and reg_cmp_gt that does numerical comparisons in Rust.

Comparison function

A more complicated class of custom filters is based on using the values of two wildcards.

from symbolica import Expression
x_, y_ = Expression.symbols('x_', 'y_')
f = Expression.symbol('f')
e = f(1)*f(2)*f(3)
e = e.replace_all(f(x_)*f(y_), 1, x_.req_cmp(y_, lambda m1, m2: m1 + m2 == 4))

Logical conditions

Multiply wildcard conditions can be chained using the and (&), or (|) and not (~ in Python and ! in Rust) operators. For example:

x, x___, x___, y___ = Expression.symbols('x', 'x_', 'x___', 'y___')
f = Expression.symbol('f')
e = f(3, x, 2).replace_all(f(x___, x___, y___),
                           f(x___), x___.req_lt(3) | x___.req_type(AtomType.Var))

matches if x___ is smaller than 3 or is a variable. The example yields f(2).

Level restrictions

Sometimes a pattern should only match in a function or at a certain depth of the expression tree. This can be set with level_range, which defines the inclusive range for which the pattern should match. For example:

x = Expression.symbol('x')
f = Expression.symbol('f')
r = x+x**2*f(x**2, x, f(x))
print(r.replace_all(x, 1, level_range=(1, 1)))

yields x+x^2*f(1,1,f(x)).

By default, the level signifies the function depth. It can be set to signify the depth in the expression tree by the optional argument level_is_tree_depth=True.

Function symbols

To add a condition on a function symbol (its name), use get_symbol. For example:

f, f_ = Expression.symbol('f', 'f_')
e = f(1).replace_all(f_(2), 1, f_.get_symbol().req(lambda x: str(x) == 'f'))

Additional settings

Wildcards are greedy by default, i.e. they try to be as large as possible. In the example of the previous section we see that the result is f(2) instead of f(x), because x___ is greedy. By passing the setting non_greedy_wildcards we can change this behaviour:

e = f(3, x, 2).replace_all(f(x___, x_, y___),
                           f(x_), x_.req_lt(3) | x_.req_type(AtomType.Var),
                           non_greedy_wildcards=[x___])

This code yields f(x).

Match iteration

It is possible to iterate through all matches without modifying the expression.

For example, here f(x_) is matched:

from symbolica import Expression
x_ = Expression.symbol('x_')
f = Expression.symbol('f')
e = f(1)+f(2)+f(3)

for m in e.match(f(x_)):
    print(m[x_])

which yields

1
2
3

The match function returns a dictionary with the wildcard as keys.

The match function also yields nested matches:

from symbolica import Expression
x_ = Expression.symbol('x_')
f = Expression.symbol('f')
e = f(f(1))

for m in e.match(f(x_)):
    print(m[x_])

which yields

f(1)
1

Replacements

We have already seen many examples of replacing all occurrences of a pattern in this section. An iterator over single replacements can be generated as well:

from symbolica import Expression
x_ = Expression.symbol('x_')
f = Expression.symbol('f')
e = f(1)*f(2)*f(3)
for r in e.replace(f(x_), f(x_ + 1)):
    print(r)

which yields

f(2)*f(2)*f(3)
f(1)*f(3)*f(3)
f(1)*f(2)*f(4)

Transformers

Often the goal of a pattern match and replacement is to isolate a part of the expression and to modify it to something else. Rather than simply allowing one atom to be replaced by another, functions can be executed for the replacement instead. These are called transformers (see here for more information). A transformer takes a pattern with wildcards as an argument and executes an operation on it after the wildcards have been substituted.

For example:

from symbolica import Expression, Transformer

x, x_ = Expression.symbols('x','x_')
f = Expression.symbol('f')
e = f((x+1)**2)
print(e.replace_all(f(x_), f(x_.transform().expand())))

yields

f(x^2+2*x+1)

Examples

Multiple matches

from symbolica import Expression
x_, y_, z_, w_, a_ = Expression.symbols('x__', 'y__', 'z__', 'w__', 'a__')
f, g = Expression.symbols('f', 'g')
e = f(1,2,3,4)*g(5,2,3,6,7)
for m in e.match(f(x_,y_,z_)*g(w_,y_,a_)):
    print('{}\t{}\t{}\t{}\t{}'.format(m[x_], m[y_], m[z_], m[w_], m[a_]))

yields:

1        2        arg(3,4) 5        arg(3,6,7)
1        arg(2,3) 4        5        arg(6,7)
arg(1,2) 3        4        arg(5,2) arg(6,7)

Coefficient extraction

The coefficients of the polynomial x^2*(y+z) + x^3*(y+z^2) in x are matched:

from symbolica import Expression

x, y, z, n_, x_ = Expression.symbols('x', 'y', 'z', 'n_', 'x_')
e = x**2*(y+z) + x**3*(y+z**2)

# match returns a dictionary that maps every wildcard to its value
coeff_list = [(m[n_], m[x_]) for m in e.match(x**n_*x_)]
for (pow, content) in coeff_list:
    print(x**pow, content)