QuLab 2.0.0__py3-none-any.whl → 2.10.11__py3-none-any.whl

qulab/scan/expression.py

@@ -1,360 +0,0 @@
-from __future__ import annotations
-
-import operator
-
-import numpy as np
-from pyparsing import (CaselessKeyword, Combine, Forward, Group, Keyword,
-                       Literal, Optional, ParserElement, Suppress, Word,
-                       alphanums, alphas, delimitedList, nums, oneOf, opAssoc,
-                       pyparsing_common, restOfLine, srange, stringEnd,
-                       stringStart)
-
-LPAREN, RPAREN, LBRACK, RBRACK, LBRACE, RBRACE, DOT, TILDE, BANG, PLUS, MINUS = map(
-    Suppress, "()[]{}.~!+-")
-
-INT = Combine(srange("[1-9]") +
-              Optional(Word(nums))).set_parse_action(lambda t: int(t[0]))
-OCT = Combine("0" + Word("01234567")).set_parse_action(lambda t: int(t[0], 8))
-HEX = Combine("0x" + Word("0123456789abcdefABCDEF")).set_parse_action(
-    lambda t: int(t[0], 16))
-FLOAT = Combine(Word(nums) + DOT + Word(nums)) | \
-        Combine(DOT + Word(nums)) | \
-        Combine(Word(nums) + DOT) | \
-        Combine(Word(nums) + DOT + Word(nums) + CaselessKeyword("e") + Word("+-") + Word(nums)) | \
-        Combine(Word(nums) + DOT + CaselessKeyword("e") + Word("+-") + Word(nums)) | \
-        Combine(DOT + Word(nums) + CaselessKeyword("e") + Word("+-") + Word(nums)) | \
-        Combine(Word(nums) + CaselessKeyword("e") + Word("+-") + Word(nums))
-FLOAT.set_parse_action(lambda t: float(t[0]))
-SYMBOL = Word(alphas, alphanums + "_")
-SYMBOL.set_parse_action(lambda t: Symbol(t[0]))
-
-expr = Forward()
-unary = Forward()
-binary = Forward()
-atom = Forward()
-
-atom << (INT | OCT | HEX | FLOAT | SYMBOL | (LPAREN + expr + RPAREN) |
-         (LBRACK + expr + RBRACK) | (LBRACE + expr + RBRACE) | (MINUS + atom) |
-         (PLUS + atom) | (TILDE + atom) | (BANG + atom) | (nums + DOT + nums))
-
-unary << (atom | (MINUS + unary) | (PLUS + unary) | (TILDE + unary) |
-          (BANG + unary))
-
-ConstType = (int, float, complex)
-_empty = object()
-
-
-class Ref():
-    __slots__ = ['name']
-
-    def __init__(self, name):
-        self.name = name
-
-    def __repr__(self) -> str:
-        return f"Ref({self.name!r})"
-
-
-class Env():
-
-    def __init__(self):
-        self.consts = {}
-        self.variables = {}
-        self.refs = {}
-
-    def __contains__(self, key):
-        return key in self.consts or key in self.variables or key in self.refs
-
-    def __getitem__(self, key):
-        if key in self.consts:
-            return self.consts[key]
-        if key in self.variables:
-            return self.variables[key]
-        if key in self.refs:
-            return self[self.refs[key]]
-        raise KeyError(f"Key {key} not found")
-
-    def __setitem__(self, key, value):
-        if key in self.consts:
-            raise KeyError(f"Key {key:r} is const")
-        elif isinstance(value, Ref):
-            self.create_ref(key, value.name)
-        elif key in self.refs:
-            self[self.refs[key]] = value
-        else:
-            self.variables[key] = value
-
-    def __delitem__(self, key):
-        if key in self.consts:
-            raise KeyError(f"Key {key:r} is const")
-        elif key in self.refs:
-            del self[self.refs[key]]
-        else:
-            del self.variables[key]
-
-    def ref(self, key):
-        if key in self:
-            return Ref(key)
-        else:
-            raise KeyError(f"Key {key!r} not found")
-
-    def create_ref(self, key, name):
-        if name in self.refs:
-            if key in self.refs[name]:
-                raise ValueError(f"Key {key!r} already exists in ref {name!r}")
-            else:
-                self.refs[key] = [name, *self.refs[name]]
-        else:
-            self.refs[key] = [name]
-
-    def is_const(self, key):
-        return key in self.consts
-
-
-class Expression():
-
-    def __init__(self):
-        self.cache = _empty
-
-    def d(self, x: str | Symbol):
-        if isinstance(x, Symbol):
-            x = x.name
-        if x in self.symbols():
-            return self.derivative(x)
-        else:
-            return 0
-
-    def derivative(self, x):
-        raise NotImplementedError
-
-    def __add__(self, other):
-        return BinaryExpression(self, other, operator.add)
-
-    def __radd__(self, other):
-        return BinaryExpression(other, self, operator.add)
-
-    def __sub__(self, other):
-        return BinaryExpression(self, other, operator.sub)
-
-    def __rsub__(self, other):
-        return BinaryExpression(other, self, operator.sub)
-
-    def __mul__(self, other):
-        return BinaryExpression(self, other, operator.mul)
-
-    def __rmul__(self, other):
-        return BinaryExpression(other, self, operator.mul)
-
-    def __truediv__(self, other):
-        return BinaryExpression(self, other, operator.truediv)
-
-    def __rtruediv__(self, other):
-        return BinaryExpression(other, self, operator.truediv)
-
-    def __pow__(self, other):
-        return BinaryExpression(self, other, operator.pow)
-
-    def __rpow__(self, other):
-        return BinaryExpression(other, self, operator.pow)
-
-    def __neg__(self):
-        return UnaryExpression(self, operator.neg)
-
-    def __pos__(self):
-        return UnaryExpression(self, operator.pos)
-
-    def __eq__(self, other):
-        return BinaryExpression(self, other, operator.eq)
-
-    def __ne__(self, other):
-        return BinaryExpression(self, other, operator.ne)
-
-    def __lt__(self, other):
-        return BinaryExpression(self, other, operator.lt)
-
-    def __le__(self, other):
-        return BinaryExpression(self, other, operator.le)
-
-    def __gt__(self, other):
-        return BinaryExpression(self, other, operator.gt)
-
-    def __ge__(self, other):
-        return BinaryExpression(self, other, operator.ge)
-
-    def __getitem__(self, other):
-        return ObjectMethod(self, '__getitem__', other)
-
-    def __getattr__(self, other):
-        return ObjectMethod(self, '__getattr__', other)
-
-    def __call__(self, *args):
-        return ObjectMethod(self, '__call__', *args)
-    
-    def __round__(self, n=None):
-        return self
-
-    def eval(self, env):
-        raise NotImplementedError
-
-    def symbols(self) -> list[str]:
-        raise NotImplementedError
-
-    def changed(self, env) -> bool:
-        return True
-
-    def is_const(self, env) -> bool:
-        return False
-
-    def value(self, env):
-        if self.changed(env):
-            self.cache = self.eval(env)
-        return self.cache
-
-
-class UnaryExpression(Expression):
-
-    def __init__(self, a, op):
-        super().__init__()
-        self.a = a
-        self.op = op
-
-    def symbols(self) -> list[str]:
-        if isinstance(self.a, Expression):
-            return self.a.symbols()
-        else:
-            return []
-
-    def changed(self, env) -> bool:
-        if isinstance(self.a, ConstType):
-            return False
-        return self.cache is _empty or isinstance(
-            self.a, Expression) and self.a.changed(env)
-
-    def is_const(self, env) -> bool:
-        return isinstance(self.a,
-                          Expression) and self.a.is_const(env) or isinstance(
-                              self.a, ConstType)
-
-    def eval(self, env):
-        a = self.a.value(env) if isinstance(self.a, Expression) else self.a
-        return self.op(a)
-
-    def derivative(self, x):
-        if isinstance(self.a, Expression):
-            return self.op(self.a.d(x))
-        else:
-            return 0
-        
-    def __repr__(self) -> str:
-        return f"{self.op.__name__}({self.a!r})"
-
-
-class BinaryExpression(Expression):
-
-    def __init__(self, a, b, op):
-        super().__init__()
-        self.a = a
-        self.b = b
-        self.op = op
-
-    def symbols(self) -> list[str]:
-        symbs = set()
-        if isinstance(self.a, Expression):
-            symbs.update(self.a.symbols())
-        if isinstance(self.b, Expression):
-            symbs.update(self.b.symbols())
-        return list(symbs)
-
-    def eval(self, env):
-        a = self.a.value(env) if isinstance(self.a, Expression) else self.a
-        b = self.b.value(env) if isinstance(self.b, Expression) else self.b
-        return self.op(a, b)
-
-    def derivative(self, x):
-        if isinstance(self.a, Expression):
-            da = self.a.d(x)
-        else:
-            da = 0
-        if isinstance(self.b, Expression):
-            db = self.b.d(x)
-        else:
-            db = 0
-
-        if self.op is operator.add:
-            return da + db
-        elif self.op is operator.sub:
-            return da - db
-        elif self.op is operator.mul:
-            return self.a * db + da * self.b
-        elif self.op is operator.truediv:
-            return (da * self.b - self.a * db) / self.b**2
-        elif self.op is operator.pow:
-            if isinstance(self.a, Expression) and isinstance(
-                    self.b, Expression):
-                return self.a**self.b * (self.b * da / self.a +
-                                         ObjectMethod(np, 'log', self.a) * db)
-            elif isinstance(self.a, Expression):
-                return self.b * self.a**(self.b - 1) * da
-            elif isinstance(self.b, Expression):
-                return np.log(self.a) * db * self.a**self.b
-            else:
-                return 0
-        else:
-            return 0
-        
-    def __repr__(self) -> str:
-        return f"({self.a!r} {self.op.__name__} {self.b!r})"
-
-
-class ObjectMethod(Expression):
-
-    def __init__(self, obj, method: str, *args):
-        super().__init__()
-        self.obj = obj
-        self.method = method
-        self.args = args
-
-    def symbols(self) -> list[str]:
-        symbs = set()
-        if isinstance(self.obj, Expression):
-            symbs.update(self.obj.symbols())
-        for a in self.args:
-            if isinstance(a, Expression):
-                symbs.update(a.symbols())
-        return list(symbs)
-
-    def eval(self, env):
-        obj = self.obj.value(env) if isinstance(self.obj,
-                                                Expression) else self.obj
-        args = [
-            a.value(env) if isinstance(a, Expression) else a for a in self.args
-        ]
-        if isinstance(obj, Expression) or any(
-                isinstance(x, Expression) for x in args):
-            return ObjectMethod(obj, self.method, *args)
-        else:
-            return getattr(obj, self.method)(*args)
-
-
-class Symbol(Expression):
-
-    def __init__(self, name):
-        super().__init__()
-        self.name = name
-
-    def symbols(self) -> list[str]:
-        return [self.name]
-
-    def eval(self, env):
-        if self.name in env:
-            return env[self.name]
-        else:
-            return self
-
-    def derivative(self, x):
-        if x == self.name:
-            return 1
-        else:
-            return 0
-    
-    def __repr__(self) -> str:
-        return self.name

qulab/scan/scanner.py

@@ -1,244 +0,0 @@
-import ast
-
-import numpy as np
-
-from .base import scan_iters
-from .expression import Env, Expression, Symbol, _empty
-
-
-def is_valid_identifier(s: str) -> bool:
-    try:
-        ast.parse(f"f({s}=0)")
-        return True
-    except SyntaxError:
-        return False
-
-
-class Atom():
-    __slots__ = ('value', )
-
-    def __init__(self, value):
-        self.value = value
-
-
-class MappedSymbol(Symbol):
-    pass
-
-
-class OptimizeSpace():
-
-    def __init__(self, optimizer, space):
-        self.optimizer = optimizer
-        self.space = space
-        self.name = None
-
-
-class Optimizer():
-
-    def __init__(self, cls, *args, **kwds):
-        self.cls = cls
-        self.args = args
-        self.kwds = kwds
-        self.dimensions = {}
-        self.function = None
-
-    def Categorical(self, *args, **kwds):
-        from skopt.space import Categorical
-        return OptimizeSpace(self, Categorical(*args, **kwds))
-
-    def Integer(self, *args, **kwds):
-        from skopt.space import Integer
-        return OptimizeSpace(self, Integer(*args, **kwds))
-
-    def Real(self, *args, **kwds):
-        from skopt.space import Real
-        return OptimizeSpace(self, Real(*args, **kwds))
-
-    @property
-    def target(self):
-        return None
-
-    @target.setter
-    def target(self, fun):
-        if isinstance(fun, Symbol):
-            self.function = fun.name
-        elif isinstance(fun, Expression):
-            self.function = fun
-        else:
-            raise ValueError("Invalid function")
-
-    def create_optimizer(self):
-        dimensions = list(self.dimensions.values())
-        return tuple(self.dimensions.keys()), self.cls(dimensions, *self.args,
-                                                       **self.kwds)
-
-
-class Scan():
-
-    def __new__(cls, *args, mixin=None, **kwds):
-        if mixin is None:
-            return super().__new__(cls)
-        for k in dir(mixin):
-            if not hasattr(cls, k):
-                try:
-                    setattr(cls, k, getattr(mixin, k))
-                except:
-                    pass
-        return super().__new__(cls)
-
-    def __init__(self, name, *args, env=None, mixin=None, **kwds):
-        super().__init__(*args, **kwds)
-        self._name = name.replace(' ', '_')
-        self.env = Env() if env is None else env
-        self.functions = {}
-        self.consts = {}
-        self.loops = {}
-        self.mapping = {}
-        self.optimizers = {}
-        self._mapping_i = 0
-        self.filter = None
-        self.scan_info = {'loops': {}}
-        self._tmp = {}
-
-    @property
-    def name(self):
-        return f"Scan.{self._name}"
-
-    def get(self, key, default=_empty):
-        if key in self.consts:
-            return self.consts[key]
-        if key in self.functions:
-            return self.functions[key]
-        if default is _empty:
-            raise KeyError(f"Key {key} not found")
-        return default
-
-    def set(self, key, value):
-        print(f"Set {key} to {value}")
-
-    def _mapping(self, key, value):
-        tmpkey = f"__tmp_{self._mapping_i}__"
-        self._mapping_i += 1
-        self[tmpkey] = value
-        self.mapping[key] = tmpkey
-
-    def __setitem__(self, key, value):
-        if is_valid_identifier(key):
-            if isinstance(value, Atom):
-                self.consts[key] = value.value
-                return
-            elif isinstance(value, (str, int, float, complex, tuple)):
-                self.consts[key] = value
-                return
-
-        if isinstance(value, Expression):
-            env = Env()
-            env.consts = self.consts
-            value = value.value(env)
-            if not isinstance(value, Expression):
-                self.__setitem__(key, value)
-                return
-
-        self._tmp[key] = value
-
-    def __setitem(self, key, value):
-        if not is_valid_identifier(key):
-            self._mapping(key, value)
-            return
-        if isinstance(value, Expression) or callable(value):
-            self.functions[key] = value
-        elif isinstance(value, OptimizeSpace):
-            self.optimizers[key] = value.optimizer
-            value.name = key
-            value.optimizer.dimensions[key] = value.space
-            self.loops[key] = value.optimizer
-        elif isinstance(value, (np.ndarray, list, range)):
-            self.loops[key] = value
-        elif isinstance(value, Atom):
-            self.consts[key] = value.value
-        else:
-            self.consts[key] = value
-
-    def __getitem__(self, key):
-        if key in self.consts:
-            return self.consts[key]
-        if is_valid_identifier(key):
-            return Symbol(key)
-        else:
-            if key in self.mapping:
-                return Symbol(self.mapping[key])
-            return MappedSymbol(key)
-
-    def assemble(self):
-        for key, value in self._tmp.items():
-            self.__setitem(key, value)
-
-        variables = {}
-        loops = {}
-
-        for k, v in self.functions.items():
-            if isinstance(v, MappedSymbol):
-                variables[k] = eval(
-                    f"lambda {self.mapping[k]}: {self.mapping[k]}")
-            elif isinstance(v, Expression):
-                args = v.symbols()
-                for x in args:
-                    if x in self.mapping:
-                        args.remove(x)
-                        v = v.value({x: Symbol(self.mapping[x])})
-                        x = self.mapping[x]
-                    if x in self.consts:
-                        args.remove(x)
-                        v = v.value({x: self.consts[x]})
-                if args:
-                    variables[k] = eval(
-                        f"lambda {','.join(args)}: expr.value({{{','.join([f'{x!r}:{x}' for x in args])}}})",
-                        {'expr': v})
-                else:
-                    self.consts[k] = v
-            else:
-                variables[k] = v
-
-        for key, value in self.loops.items():
-            if isinstance(value, Optimizer):
-                #variables[key] = value.create_optimizer()
-                pass
-            else:
-                loops[key] = value
-
-        self.scan_info = {
-            'loops': loops,
-            'functions': variables,
-            'constants': self.consts
-        }
-
-        if self.filter is not None:
-            self.scan_info['filter'] = self.filter
-
-    def main(self):
-        self.assemble()
-        for step in self.scan():
-            for k, v in self.mapping.items():
-                self.set(k, step.kwds[v])
-            self.process(step)
-
-    def process(self, step):
-        print(step.kwds)
-
-    def scan(self):
-        for step in scan_iters(**self.scan_info):
-            for k, v in self.mapping.items():
-                step.kwds[k] = step.kwds[v]
-            yield step
-
-    def run(self, dry_run=False):
-        pass
-
-    def plot(self,
-             result=None,
-             fig=None,
-             axis=None,
-             data='population',
-             T=False,
-             **kwds):
-        pass

qulab/scan/transforms.py

@@ -1,16 +0,0 @@
-class Aggregation():
-
-    def __init__(self):
-        pass
-
-    def __call__(self, data):
-        raise NotImplementedError()
-
-
-class Transform():
-
-    def __init__(self):
-        pass
-
-    def __call__(self, data):
-        raise NotImplementedError()