Mapytex/mapytex/calculus/core/MO/polynomial.py

76 lines
2.0 KiB
Python
Raw Normal View History

2018-11-12 08:41:39 +00:00
#! /usr/bin/env python
# -*- coding: utf-8 -*-
# vim:fenc=utf-8
#
# Copyright © 2017 lafrite <lafrite@Poivre>
#
# Distributed under terms of the MIT license.
from mapytex.calculus.core.tree import Tree
from .mo import MO, MOnumber, MOstr
from .exceptions import MOError
from ..renders import tree2txt, tree2tex
__all__ = ["MOpolynomial"]
class MOpolynomial(MO):
""" MO polynomial"""
def __init__(self, variable, coefs):
""" Initiate a MOpolynomial
:param variable: variable of the monomial (a MOstr or later a MOSqrt)
:param coefs: dictionnary {deg: coef} or a list [coef0, coef1...]
:example:
>>> MOpolynomial('x', [1, 2, 3])
3x^2 + 2x + 1
>>> MOpolynomial('x', [1, 0, 3])
3x^2 + 1
>>> MOpolynomial('x', {0: 1, 1: 2, 2: 3})
3x^2 + 2x + 1
>>> MOpolynomial('x', {0: 1, 3: 4})
4x^3 + 1
"""
_variable = MO.factory(variable)
if not isinstance(_variable, MOstr):
raise MOError("The variable of a monomial should be convertible into MOstr")
self._variable = _variable
if isinstance(coefs, dict):
_coefs = {MO.factory(d): MO.factory(c) for (d, c) in coefs}
elif isinstance(coefs, list):
_coefs = {MO.factory(d): MO.factory(c) for (d, c) in enumerate(coefs)}
else:
raise TypeError("Coefs needs to be a dictionnary or a list")
self._coefs = _coefs
monomials = []
for deg, coef in self._coefs:
if deg == 0:
monomials.append(coef)
elif deg == 1:
monomials.append(coef)
value = Tree("^",
self._variable,
self._power
)
MO.__init__(self, value)
@property
def variable(self):
return self._variable
@property
def degree(self):
return self._power
# -----------------------------
# Reglages pour 'vim'
# vim:set autoindent expandtab tabstop=4 shiftwidth=4:
# cursor: 16 del