#! /usr/bin/env python # -*- coding: utf-8 -*- # vim:fenc=utf-8 # # Copyright © 2017 lafrite # # 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 from .monomial import MOMonomial __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]) >>> MOpolynomial('x', [1, 0, 3]) >>> MOpolynomial('x', {0: 1, 1: 2, 2: 3}) >>> MOpolynomial('x', {0: 1, 3: 4}) >>> MOpolynomial('x', {0: 1, 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.items() if c != 0 } elif isinstance(coefs, list): _coefs = {MO.factory(d): MO.factory(c) for (d, c) in enumerate(coefs) if c != 0 } else: raise TypeError("Coefs needs to be a dictionnary or a list") self._coefs = _coefs monomials = {} for deg, coef in self._coefs.items(): if deg == 0: monomials[deg] = coef else: monomials[deg] = MOMonomial(coef, self._variable, deg) self._monomials = monomials tree = Tree.from_list("+", list(self._monomials.values())[::-1]) MO.__init__(self, tree) @property def variable(self): return self._variable @property def degree(self): return self._power @property def coefficients(self): return self._coefs @property def monomials(self): return self._monomials # ----------------------------- # Reglages pour 'vim' # vim:set autoindent expandtab tabstop=4 shiftwidth=4: # cursor: 16 del