241 lines
7.3 KiB
Python
241 lines
7.3 KiB
Python
#!/usr/bin/env python
|
|
# encoding: utf-8
|
|
|
|
|
|
from .expression import Expression
|
|
from .operator import op
|
|
from .generic import isNumerand
|
|
from .random_expression import RdExpression
|
|
from .abstract_polynom import AbstractPolynom
|
|
|
|
from functools import wraps
|
|
import inspect
|
|
|
|
__all__ = ["Polynom"]
|
|
|
|
def polynom_factory(func):
|
|
""" Decorator which specify the type of polynom that the function returns """
|
|
@wraps(func)
|
|
def wrapper(*args, **kwrds):
|
|
P = func(*args, **kwrds)
|
|
if issubclass(type(P),AbstractPolynom) and P.degree == 2:
|
|
from .polynomDeg2 import Polynom_deg2
|
|
new_P = Polynom_deg2(poly=P)
|
|
new_P.steps = P.steps
|
|
return new_P
|
|
elif issubclass(type(P),AbstractPolynom):
|
|
new_P = Polynom(poly=P)
|
|
new_P.steps = P.steps
|
|
return new_P
|
|
else:
|
|
return P
|
|
return wrapper
|
|
|
|
class Polynom(AbstractPolynom):
|
|
|
|
"""Polynom view as a function.
|
|
|
|
It can be initiate like a AbstractPolynom
|
|
# Put example
|
|
Randomly
|
|
# Put example
|
|
It can be evaluate
|
|
# Put example
|
|
And derivate
|
|
# Put example
|
|
|
|
"""
|
|
|
|
@classmethod
|
|
def random(self, coefs_form=[], conditions=[], letter = "x", degree = 0, name = "P"):
|
|
""" Create a random polynom from coefs_form and conditions
|
|
|
|
:param coefs_form: list of forms (one by coef) (ascending degree sorted)
|
|
:param conditions: condition on variables
|
|
:param letter: the letter for the polynom
|
|
:param degree: degree of the polynom (can't be used with coefs_form, it will be overwrite) - can't be higher than 26 (number of letters in alphabet)
|
|
|
|
/!\ variables need to be in brackets {}
|
|
|
|
>>> Polynom.random(["{b}", "{a}"]) # doctest:+ELLIPSIS
|
|
< <class 'pymath.polynom.Polynom'> ...
|
|
>>> Polynom.random(degree = 2) # doctest:+ELLIPSIS
|
|
< <class 'pymath.polynomDeg2.Polynom_deg2'> ...
|
|
>>> Polynom.random(degree = 3) # doctest:+ELLIPSIS
|
|
< <class 'pymath.polynom.Polynom'> ...
|
|
>>> Polynom.random(degree = 2, conditions=["{b**2-4*a*c}>0"]) # Polynom deg 2 with positive Delta (ax^2 + bx + c)
|
|
< <class 'pymath.polynomDeg2.Polynom_deg2'> ...
|
|
>>> Polynom.random(["{c}", "{b}", "{a}"], conditions=["{b**2-4*a*c}>0"]) # Same as above
|
|
< <class 'pymath.polynomDeg2.Polynom_deg2'> ...
|
|
|
|
"""
|
|
if (degree > 0 and degree < 26):
|
|
# Générer assez de lettre pour les coefs
|
|
coefs_name = map(chr, range(97, 98+degree))
|
|
coefs_form = ["{" + i + "}" for i in coefs_name][::-1]
|
|
|
|
form = str(coefs_form)
|
|
# On créé les valeurs toutes concaténées dans un string
|
|
coefs = RdExpression(form, conditions)()
|
|
# On "parse" ce string pour créer les coefs
|
|
coefs = [eval(i) if type(i)==str else i for i in eval(coefs)]
|
|
# Création du polynom
|
|
return Polynom(coefs = coefs, letter = letter, name = name)
|
|
|
|
def __init__(self, coefs = [1], letter = "x", name = "P", poly = 0):
|
|
"""Initiate the polynom
|
|
|
|
:param coef: coefficients of the polynom (ascending degree sorted)
|
|
3 possibles type of coefficent:
|
|
- a : simple "number". [1,2] designate 1 + 2x
|
|
- [a,b,c]: list of coeficient for same degree. [1,[2,3],4] designate 1 + 2x + 3x + 4x^2
|
|
- a: a Expression. [1, Expression("2+3"), 4] designate 1 + (2+3)x + 4x^2
|
|
:param letter: the string describing the unknown
|
|
:param name: Name of the polynom
|
|
|
|
>>> P = Polynom([1, 2, 3])
|
|
>>> P.mainOp
|
|
'+'
|
|
>>> P.name
|
|
'P'
|
|
>>> P._letter
|
|
'x'
|
|
>>> Polynom([1]).mainOp
|
|
'*'
|
|
>>> Polynom([0, 0, 3]).mainOp
|
|
'*'
|
|
>>> Polynom([1, 2, 3])._letter
|
|
'x'
|
|
>>> Polynom([1, 2, 3], "y")._letter
|
|
'y'
|
|
>>> Polynom([1, 2, 3], name = "Q").name
|
|
'Q'
|
|
"""
|
|
if poly:
|
|
coefs = poly._coef
|
|
letter = poly._letter
|
|
name = poly.name
|
|
|
|
super(Polynom, self).__init__(coefs, letter, name)
|
|
|
|
def __call__(self, value):
|
|
""" Evaluate the polynom in value
|
|
|
|
:returns: Expression ready to be simplify
|
|
|
|
>>> P = Polynom([1, 2, 3])
|
|
>>> P(2)
|
|
17
|
|
>>> for i in P(2).explain():
|
|
... print(i)
|
|
3 \\times 2^{ 2 } + 2 \\times 2 + 1
|
|
3 \\times 4 + 4 + 1
|
|
12 + 4 + 1
|
|
16 + 1
|
|
17
|
|
>>> Q = P("1+h")
|
|
>>> print(Q)
|
|
3 h^{ 2 } + 8 h + 6
|
|
>>> R = P(Q)
|
|
"""
|
|
if isNumerand(value) or Expression.isExpression(value):
|
|
postfix_exp = [value if i==self._letter else i for i in self.postfix_tokens]
|
|
else:
|
|
postfix_exp = [Expression(value) if i==self._letter else i for i in self.postfix_tokens]
|
|
|
|
return Expression(postfix_exp).simplify()
|
|
|
|
def derivate(self):
|
|
""" Return the derivated polynom
|
|
|
|
>>> P = Polynom([1, 2, 3])
|
|
>>> Q = P.derivate()
|
|
>>> Q
|
|
< <class 'pymath.polynom.Polynom'> [2, 6]>
|
|
>>> print(Q.name)
|
|
P'
|
|
>>> for i in Q.explain():
|
|
... print(i)
|
|
2 \\times 3 x + 1 \\times 2
|
|
6 x + 2
|
|
"""
|
|
derv_coefs = []
|
|
for (i,c) in enumerate(self._coef):
|
|
derv_coefs += [Expression([i, c, op.mul])]
|
|
|
|
ans = Polynom(derv_coefs[1:]).simplify()
|
|
ans.name = self.name + "'"
|
|
return ans
|
|
|
|
# Decorate methods which may return Polynoms
|
|
methods_list = ["__add__", "__call__", "__mul__", "__neg__", "__pow__",
|
|
"__radd__", "__rmul__", "__rsub__", "__sub__", "derivate",
|
|
"reduce", "simplify", "random"]
|
|
for name, func in inspect.getmembers(Polynom):
|
|
if name in methods_list:
|
|
setattr(Polynom, name, polynom_factory(func))
|
|
|
|
|
|
def test(p,q):
|
|
print("---------------------")
|
|
print("---------------------")
|
|
print("p : ",p)
|
|
print("q : ",q)
|
|
|
|
print("\n Plus ------")
|
|
print(p, "+", q)
|
|
for i in (p + q):
|
|
#print(repr(i))
|
|
#print("\t", str(i.postfix_tokens))
|
|
print(i)
|
|
|
|
print("\n Moins ------")
|
|
for i in (p - q):
|
|
#print(repr(i))
|
|
#print("\t", str(i.postfix_tokens))
|
|
print(i)
|
|
|
|
print("\n Multiplier ------")
|
|
for i in (p * q):
|
|
#print(repr(i))
|
|
#print("\t", str(i.postfix_tokens))
|
|
print(i)
|
|
|
|
print("\n Evaluer p ------")
|
|
for i in p(3).simplify():
|
|
print(i)
|
|
|
|
print("\n Evaluer q ------")
|
|
for i in q(3).simplify():
|
|
print(i)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
#from .fraction import Fraction
|
|
#with Expression.tmp_render(txt):
|
|
# p = Polynom([1, 2, 3])
|
|
# q = Polynom([4, 5, 6])
|
|
# for i in (p*q).explain():
|
|
# print(i)
|
|
# r = Polynom([0,1])
|
|
# for i in (r*3).explain():
|
|
# print(i)
|
|
# print("q = ", q)
|
|
# r = q.reduce()
|
|
# print("r = ", r)
|
|
# for i in r.explain():
|
|
# print("q = ", i)
|
|
# print(p-q)
|
|
# for i in p-q:
|
|
# print(i)
|
|
#Polynom.random(degree = 2, conditions=["{b**2-4*a*c}>0"]) # Polynom deg 2 with positive Delta (ax^2 + bx + c)
|
|
|
|
import doctest
|
|
doctest.testmod(optionflags=doctest.ELLIPSIS)
|
|
|
|
|
|
# -----------------------------
|
|
# Reglages pour 'vim'
|
|
# vim:set autoindent expandtab tabstop=4 shiftwidth=4:
|
|
# cursor: 16 del
|