Mapytex/mapytex/calculus/API/tokens/polynomial.py

112 lines
2.4 KiB
Python
Raw Normal View History

#! /usr/bin/env python
# -*- coding: utf-8 -*-
# vim:fenc=utf-8
#
# Copyright © 2017 lafrite <lafrite@Poivre>
#
# Distributed under terms of the MIT license.
"""
Tokens representing polynomials functions
"""
from .token import Token
from ...core.MO import MO
__all__ = ["Polynomial", "Quadratic", "Linear"]
2019-05-14 04:55:56 +00:00
class Polynomial(Token):
""" Token representing a polynomial
:examples:
>>> from ...core.MO.polynomial import MOpolynomial
>>> P = Polynomial(MOpolynomial('x', [1, 2, 3]))
>>> P
"""
def __init__(self, a, name="", ancestor=None):
if not isinstance(a, MO):
if isinstance(a, str):
raise TypeError
else:
raise TypeError
else:
mo = a
Token.__init__(self, mo, name, ancestor)
2019-05-14 04:55:56 +00:00
self._mathtype = "polynome"
@classmethod
def from_mo(cls, mo, name="", ancestor=None):
return cls(mo, name, ancestor)
@classmethod
def random(cls):
raise NotImplemented
def __setitem__(self, key, item):
""" Use Polynomial like if they were a dictionnary to set coefficients """
pass
def __getitem__(self, key):
""" Use Polynomial like if they were a dictionnary to get coefficients """
pass
def __call__(self, value):
""" Call a Polynomial to evaluate itself on value
:examples:
>>> from ...core.MO.polynomial import MOpolynomial
>>> P = Polynomial(MOpolynomial('x', [1, 2, 3]))
>>> P(2)
"""
pass
2019-05-14 04:55:56 +00:00
class Linear(Polynomial):
""" Token representing a linear """
def __init__(self, mo, name="", ancestor=None):
Polynomial.__init__(self, mo, name, ancestor)
2019-05-14 04:55:56 +00:00
self._mathtype = "affine"
@classmethod
def random(cls):
raise NotImplemented
2019-05-14 04:55:56 +00:00
class Quadratic(Polynomial):
""" Token representing a quadratic """
def __init__(self, mo, name="", ancestor=None):
Polynomial.__init__(self, mo, name, ancestor)
2019-05-14 04:55:56 +00:00
self._mathtype = "polynome du 2nd degré"
@classmethod
def random(cls):
raise NotImplemented
def extract_variable(leaf):
try:
return leaf.variable
except AttributeError:
return None
def replace(leaf, origin, dest):
# map_on_leaf sur leaf?
if leaf == origin:
return dest
return leaf
# -----------------------------
# Reglages pour 'vim'
# vim:set autoindent expandtab tabstop=4 shiftwidth=4:
# cursor: 16 del