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

140 lines
4.1 KiB
Python
Raw Normal View History

2018-03-14 08:42:43 +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
2018-03-14 08:42:43 +00:00
__all__ = ["MOMonomial"]
class MOstrPower(MO):
2018-03-14 08:42:43 +00:00
""" Power of a MOstr """
2018-03-14 08:42:43 +00:00
def __init__(self, variable, power):
""" Initiate a MOstrPower
2018-03-14 08:42:43 +00:00
:param variable: variable of the monomial (a MOstr or later a MOSqrt)
2018-03-14 08:42:43 +00:00
:param power: non negative interger (MOnumber type)
>>> MOstrPower("x", 2)
<MOstrPower x^2>
>>> MOstrPower(3, 1)
2018-03-14 08:42:43 +00:00
Traceback (most recent call last):
...
mapytex.calculus.core.MO.exceptions.MOError: The variable of a monomial should be convertible into MOstr
>>> MOstrPower("x", 0)
2018-03-14 08:42:43 +00:00
Traceback (most recent call last):
...
mapytex.calculus.core.MO.exceptions.MOError: The power of a MOstrPower should be greater than 1
>>> MOstrPower("x", 1)
2018-03-14 08:42:43 +00:00
Traceback (most recent call last):
...
mapytex.calculus.core.MO.exceptions.MOError: The power of a MOstrPower should be greater than 1
>>> MOstrPower("x", -2)
2018-03-14 08:42:43 +00:00
Traceback (most recent call last):
...
mapytex.calculus.core.MO.exceptions.MOError: The power of a MOstrPower should be greater than 1
>>> MOstrPower("x", 2.4)
2018-03-14 08:42:43 +00:00
Traceback (most recent call last):
...
mapytex.calculus.core.MO.exceptions.MOError: The power of a monomial should be a integer
2018-03-14 08:42:43 +00:00
"""
2018-03-14 08:42:43 +00:00
_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 <= 1:
raise MOError("The power of a MOstrPower should be greater than 1")
2018-03-14 08:42:43 +00:00
elif not isinstance(_power.value, int):
raise MOError("The power of a monomial should be a integer")
self._power = _power
value = Tree("^",
self._variable,
self._power
2018-03-14 08:42:43 +00:00
)
MO.__init__(self, value)
@property
def variable(self):
return self._variable
@property
def power(self):
return self._power
class MOMonomial(MO):
""" Monomial math object"""
def __init__(self, coefficient, variable):
""" Initiate the MOMonomial
:param coefficient: coefficient of the monomial (a non zero constant)
:param variable: variable of the monomial (a MOstr or later a MOSqrt)
>>> x = MOstr('x')
>>> MOMonomial(4, x)
<MOMonomial 4x>
>>> x = MOstrPower('x', 2)
>>> MOMonomial(4, x)
<MOMonomial 4x^2>
>>> MOMonomial(0, x)
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
if not isinstance(variable, (MOstr, MOstrPower)):
raise
self._variable = variable
value = Tree("*",
self._coefficient,
self._variable
2018-03-14 08:42:43 +00:00
)
MO.__init__(self, value)
@property
def coefficient(self):
return self._coefficient
@property
def strpower(self):
if isinstance(self._variable, MOstr):
return self._variable
return self._variable
@property
def variable(self):
if isinstance(self._variable, MOstr):
return self._variable
return self._variable.variable
@property
def power(self):
if isinstance(self._variable, MOstr):
return 1
return self._variable.power
2018-03-14 08:42:43 +00:00
# -----------------------------
# Reglages pour 'vim'
# vim:set autoindent expandtab tabstop=4 shiftwidth=4:
# cursor: 16 del