lafrite/Mapytex
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Feat(API): factory build every tokens!

Browse Source
This commit is contained in:
Bertrand Benjamin
2018-12-07 11:23:05 +01:00
parent 21cc3b68a3
commit bcf589c607
3 changed files with 54 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

53
mapytex/calculus/API/tokens/__init__.py
View File

@@ -23,27 +23,70 @@ __all__ = ["factory"]
def factory(exp, name="", ancestor=None):
""" Transform a Expression with on MathObject (from core) to a appropriate token (from API)
:example:
>>> from ..expression import Expression
>>> a = Expression(MOnumber(2))
>>> factory(a)
<Integer 2>
>>> a = Expression(MOnumber(2.5))
>>> factory(a)
<Decimal 2.5>
>>> a = Expression(MOFraction(2, 5))
>>> factory(a)
<Fraction 2 / 5>
>>> a = Expression(MOstr('x'))
>>> factory(a)
<Linear x>
>>> a = Expression(MOstrPower('x', 2))
>>> factory(a)
<Quadratic x^2>
>>> a = Expression(MOstrPower('x', 3))
>>> factory(a)
<Polynomial x^3>
>>> a = Expression(MOMonomial(3, 'x', 1))
>>> factory(a)
<Linear 3x>
>>> a = Expression(MOMonomial(3, 'x', 2))
>>> factory(a)
<Quadratic 3x^2>
>>> a = Expression(MOMonomial(3, 'x', 3))
>>> factory(a)
<Polynomial 3x^3>
"""
mo = exp._tree
if not isinstance(mo, MO):
raise TypeError(f"Can't build Token from not computed Expression (got {mo})")
if isinstance(mo, MOnumber):
if isinstance(mo.value, int):
return Integer(mo, name, ancestor)
elif isinstance(mo.value, _Decimal):
return Decimal(mo, name, ancestor)
else:
raise TypeError(f"Can't build from MOnumber ({mo}) neither int nor decimal")
raise TypeError(f"Can't build from MOnumber ({mo}) neither int nor decimal")
elif isinstance(mo, MOFraction):
if isinstance(mo._denominator, MOnumber) and \
isinstance(mo._numerator, MOnumber):
return Fraction(mo, name, ancestor)
else:
raise TypeError(f"Can't build from MOFraction ({mo}) numerator and denominator are not MOnumber")
raise TypeError(f"Can't build from MOFraction ({mo}) numerator and denominator are not MOnumber")
elif isinstance(mo, (MOstr, MOstrPower, MOMonomial, MOpolynomial)):
raise TypeError(f"Can't build Polynom yet")
if not isinstance(mo._variable, (MOstr, str)):
raise TypeError(f"Can't build Polynom over something else than a letter (got {mo._variable})")
if isinstance(mo, MOstr) or \
(isinstance(mo, MOMonomial) and mo.power.value == 1) or \
(isinstance(mo, MOpolynomial) and mo.power.value == 1):
return Linear(mo, name, ancestor)
elif (isinstance(mo, MOstrPower) and mo.power.value == 2) or \
(isinstance(mo, MOMonomial) and mo.power.value == 2) or \
(isinstance(mo, MOpolynomial) and mo.power.value == 2):
return Quadratic(mo, name, ancestor)
else:
return Polynomial(mo, name, ancestor)
else:
raise TypeError(f"{type(mo)} is unknown MathObject")

12
mapytex/calculus/API/tokens/polynomial.py
View File

@@ -19,10 +19,6 @@ class Polynomial(Token):
""" Token representing a polynomial """
def __init__(self, mo, name="", ancestor=None):
if not isinstance(mo, MOpolynomial):
raise TypeError
if not isinstance(mo.value, int):
raise TypeError
Token.__init__(self, mo, name, ancestor)
self._mathtype = 'polynome'
@@ -48,10 +44,6 @@ class Linear(Token):
""" Token representing a linear """
def __init__(self, mo, name="", ancestor=None):
if not isinstance(mo, MOpolynomial):
raise TypeError
if not isinstance(mo.value, int):
raise TypeError
Token.__init__(self, mo, name, ancestor)
self._mathtype = 'affine'
@@ -65,10 +57,6 @@ class Quadratic(Token):
""" Token representing a quadratic """
def __init__(self, mo, name="", ancestor=None):
if not isinstance(mo, MOpolynomial):
raise TypeError
if not isinstance(mo.value, int):
raise TypeError
Token.__init__(self, mo, name, ancestor)
self._mathtype = 'polynome du 2nd degré'