summaryrefslogtreecommitdiffhomepage
path: root/pyecsca/misc/cache.py
blob: 5f388ad23ea1235ae39e2f2a6eee9b91024c7c17 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
"""Cache some things."""
from functools import lru_cache
from sympy import sympify as _orig_sympify, simplify as _orig_simplify, count_ops
from public import public


@public
@lru_cache(maxsize=256, typed=True)
def sympify(
    a, locals=None, convert_xor=True, strict=False, rational=False, evaluate=None
):
    return _orig_sympify(a, locals, convert_xor, strict, rational, evaluate)


@public
@lru_cache(maxsize=256, typed=True)
def simplify(expr, ratio=1.7, measure=count_ops, rational=False, inverse=False, doit=True, **kwargs):
    return _orig_simplify(expr, ratio=ratio, measure=measure, rational=rational, inverse=inverse, doit=doit, **kwargs)