124 lines
3.9 KiB
Python
124 lines
3.9 KiB
Python
#! /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__ = ["MOMonomial"]
|
|
|
|
class MOMonomial(MO):
|
|
|
|
""" Monomial math object"""
|
|
|
|
def __init__(self, coefficient, variable, power = 1):
|
|
""" Initiate the MOMonomial
|
|
|
|
:param coefficient: coefficient of the monomial (a non zero constant)
|
|
:param variable: varaible of the monomial (a MOstr or later a MOSqrt)
|
|
:param power: non negative interger (MOnumber type)
|
|
|
|
>>> MOMonomial(4, "x", 2)
|
|
<MOMonomial 4x^2>
|
|
>>> MOMonomial(4, "x", 1)
|
|
<MOMonomial 4x>
|
|
>>> MOMonomial(1, "x", 2)
|
|
<MOMonomial x^2>
|
|
>>> MOMonomial(1, "x", 1)
|
|
<MOMonomial x>
|
|
>>> MOMonomial(4, 3, 1)
|
|
Traceback (most recent call last):
|
|
...
|
|
mapytex.calculus.core.MO.exceptions.MOError: The variable of a monomial should be convertible into MOstr
|
|
>>> MOMonomial(4, "x", 0)
|
|
Traceback (most recent call last):
|
|
...
|
|
mapytex.calculus.core.MO.exceptions.MOError: The power of a monomial should be greater than 0
|
|
>>> MOMonomial(4, "x", -2)
|
|
Traceback (most recent call last):
|
|
...
|
|
mapytex.calculus.core.MO.exceptions.MOError: The power of a monomial should be greater than 0
|
|
>>> MOMonomial(4, "x", 2.4)
|
|
Traceback (most recent call last):
|
|
...
|
|
mapytex.calculus.core.MO.exceptions.MOError: The power of a monomial should be a integer
|
|
>>> MOMonomial(0, "x", 1)
|
|
Traceback (most recent call last):
|
|
...
|
|
mapytex.calculus.core.MO.exceptions.MOError: The coefficient of a monomial should not be 0
|
|
"""
|
|
_coefficient = MO.factory(coefficient)
|
|
if coefficient == 0:
|
|
raise MOError("The coefficient of a monomial should not be 0")
|
|
self._coefficient = _coefficient
|
|
|
|
_variable = MO.factory(variable)
|
|
if not isinstance(_variable, MOstr):
|
|
raise MOError("The variable of a monomial should be convertible into MOstr")
|
|
self._variable = _variable
|
|
|
|
_power = MO.factory(power)
|
|
if power <= 0:
|
|
raise MOError("The power of a monomial should be greater than 0")
|
|
elif not isinstance(_power.value, int):
|
|
raise MOError("The power of a monomial should be a integer")
|
|
self._power = _power
|
|
|
|
if self._power > 1:
|
|
self.value = Tree("*",
|
|
self._coefficient,
|
|
Tree("^",
|
|
self._variable,
|
|
self._power
|
|
)
|
|
)
|
|
else:
|
|
self.value = Tree("*",
|
|
self._coefficient,
|
|
self._variable,
|
|
)
|
|
|
|
@property
|
|
def __txt__(self):
|
|
try:
|
|
if self._coefficient == 1:
|
|
try:
|
|
return tree2txt(self.value.right_value)
|
|
except AttributeError:
|
|
return str(self.value.right_value)
|
|
except TypeError:
|
|
pass
|
|
|
|
try:
|
|
return tree2txt(self.value)
|
|
except AttributeError:
|
|
return str(self.value)
|
|
|
|
@property
|
|
def __tex__(self):
|
|
try:
|
|
if self._coefficient == 1:
|
|
try:
|
|
return tree2tex(self.value.right_value)
|
|
except AttributeError:
|
|
return str(self.value.right_value)
|
|
except TypeError:
|
|
pass
|
|
|
|
try:
|
|
return tree2tex(self.value)
|
|
except AttributeError:
|
|
return str(self.value)
|
|
|
|
|
|
# -----------------------------
|
|
# Reglages pour 'vim'
|
|
# vim:set autoindent expandtab tabstop=4 shiftwidth=4:
|
|
# cursor: 16 del
|