# Copyright (C) 2015 Chintalagiri Shashank
#
# This file is part of Tendril.
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
"""
Base Unit Types (:mod:`tendril.utils.types.unitbase`)
=====================================================
The Types provided in this module are not intended for direct use. Instead,
they provide reusable primitives which can be sub-classed to provide
functional Types.
Ideally, all Unit classes should derive from the :class:`UnitBase` class
provided here. In practice, only the newer Types follow this inheritance,
while the older ones still need to be migrated to this form.
.. rubric:: Module Contents
.. autosummary::
UnitBase
NumericalUnitBase
DummyUnit
parse_none
parse_percent
Percentage
.. seealso:: :mod:`tendril.utils.types`, for an overview applicable to
most types defined in Tendril.
"""
import re
import six
import numbers
from math import log10
from math import floor
from decimal import Decimal
from decimal import InvalidOperation
from fractions import Fraction
from . import ParseException
[docs]def round_to_n(x, n):
if x:
return round(x, -int(floor(log10(x))) + (n - 1))
return 0
[docs]def remove_exponent(num):
return num.to_integral() if num == num.to_integral() else num.normalize()
[docs]class TypedComparisonMixin(object):
"""
This mixin allows implementing comparison operators in a Python 3
compatible way.
Two instances of a class are compared using their :func:`_cmpkey`
methods. If the instances have a different ``__class__``, the
comparison is not implemented. A single exception is implemented,
for when the other instance is of a numerical type, with value 0.
"""
[docs] def _compare(self, other, method):
if self.__class__ != other.__class__:
if other is None:
return method(self._cmpkey(), 0)
if not isinstance(other, numbers.Number) or other != 0:
raise TypeError(
"Comparison of : " + repr(self) + ", " + repr(other)
)
else:
return method(self._cmpkey(), other)
return method(self._cmpkey(), other._cmpkey())
def __lt__(self, other):
return self._compare(other, lambda s, o: s < o)
def __le__(self, other):
return self._compare(other, lambda s, o: s <= o)
[docs] def __eq__(self, other):
return self._compare(other, lambda s, o: s == o)
def __ge__(self, other):
return self._compare(other, lambda s, o: s >= o)
def __gt__(self, other):
return self._compare(other, lambda s, o: s > o)
def __ne__(self, other):
return self._compare(other, lambda s, o: s != o)
[docs] def _cmpkey(self):
raise NotImplementedError
[docs]class UnitBase(object):
"""
The base class for all :mod:`tendril.utils.types` units.
When instantiated, the `value` param is processed as follows :
- :class:`str` value is passed though `_parse_func`, and whatever it
returns is stored.
- :class:`numbers.Number` values are first converted into
:class:`decimal.Decimal` and then stored.
- All other `value` types are simply stored as is.
:param value: The core value to be stored.
:param _dostr: The canonical unit / order string to use.
:param _parse_func: The function used to parse string values into an
actual value in the canonical unit.
.. rubric:: Arithmetic Operations
This class supports no arithmetic operations.
"""
_dostr = None
_parse_func = None
def __init__(self, value):
if isinstance(value, (six.text_type, six.string_types)):
try:
value = self._parse_func(value)
except Exception as e:
raise ParseException(value, e)
if isinstance(value, numbers.Number):
if isinstance(value, Fraction):
value = float(value)
if not isinstance(value, Decimal):
value = Decimal(value)
self._value = value
# def __float__(self):
# return float(self._value)
#
# def __int__(self):
# return int(self._value)
@property
def value(self):
"""
:return: The core value of the Unit instance, in it's canonical unit.
"""
return self._value
def __repr__(self):
return str(self._value) + self._dostr
[docs]class NumericalUnitBase(TypedComparisonMixin, UnitBase):
"""
The base class for all :mod:`tendril.utils.types` numerical units.
Provides the patterns used by the various Numerical Units to provide
their functionality. This class represents and implements the core
ideas that remain valid across Units (for the most part). The various
methods and functions implemented here establish the minimum required
functionality and behaviour expected of all numerical units.
Specific numerical unit classes may override the methods present here
to tweak the implementation and/or the interface as per the
requirements of the quantity they represent, as long as they stay true
to the spirit of the architecture.
:param value: The core value to be stored.
.. seealso:: :class:`UnitBase`
.. rubric:: The `_orders` / `_ostrs` Class Variables
Supported units can be provided as a list of strings (`_ostrs`) or a
list of tuples (`_orders`).
- In case `_ostrs` is provided, it is assumed that each string
represents a unit value 1000 times smaller than the next.
- In case `_orders` is provided, it is assumed that the first
element of each tuple is the string representation of the order,
and the second element is the multiplicative factor relative to
the default order string.
- In both cases, note that first order within which the unit value's
representation lies between 1 and 1000 is used to produce the unit's
string representation. As such, you should place higher priority or
more 'standard' units towards the beginning of the list.
.. rubric:: Arithmetic Operations
.. autosummary::
__add__
__sub__
__mul__
__div__
_cmpkey
"""
_orders = None
_ostrs = None
_osuffix = None
_parse_func = None
_regex_std = None
_allow_nOr = True
_order_type = None
_order_dict = None
_has_bare_order = False
_rorders = None
_pluralize_ostr = False
_separate_unit = False
def __init__(self, value):
if isinstance(value, bytes):
value = value.decode()
if not self._orders:
doidx = self._ostrs.index(self._dostr)
self._orders = [(ostr, (3 * (idx - doidx)))
for idx, ostr in enumerate(self._ostrs)]
self._order_type = 0
elif not self._ostrs:
self._ostrs = [order[0] for order in self._orders]
self._order_type = 1
if not self._order_dict:
self._order_dict = {o[0]: o[1] for o in self._orders}
if self._parse_func is None:
self._parse_func = self._standard_parser
if not self._has_bare_order:
try:
value = Decimal(value)
except InvalidOperation:
pass
super(NumericalUnitBase, self).__init__(value)
[docs] def _standard_parser(self, value):
if not self._regex_std:
raise Exception("Standard parser requires a defined regex")
match = self._regex_std.match(value)
n = Decimal(match.group('numerical'))
r = match.group('residual')
if r:
if not self._allow_nOr:
raise ValueError("nOr structure not allowed for this unit")
r = Decimal(r)
n += r/10
o = match.group('order')
if self._osuffix and not o.endswith(self._osuffix):
o += self._osuffix
try:
if self._order_type:
f = self._order_dict[o]
if isinstance(f, numbers.Number):
return n * f
else:
return f(n)
else:
exp = self._order_dict[o]
if exp >= 0:
return n * (10 ** exp)
else:
# TODO Significant error in precision here
return n / (10 ** abs(exp))
except KeyError:
raise ValueError('Order unrecognized : {0} for {1}'
''.format(o, self.__class__))
def __float__(self):
return float(self._value)
[docs] def __add__(self, other):
"""
Addition of two Unit class instances of the same type returns a
Unit class instance of the same type, with the sum of the two
operands as it's value.
If the other operand is a numerical type and evaluates to 0, this
object is simply returned unchanged.
Addition with all other Types / Classes is not supported.
"""
if isinstance(other, numbers.Number) and other == 0:
return self
elif self.__class__ == other.__class__:
return self.__class__(self._value + other._value)
else:
return NotImplemented
def __radd__(self, other):
if other == 0:
return self
else:
return self.__add__(other)
[docs] def __mul__(self, other):
"""
Multiplication of one Unit type class instance with a numerical type
results in a Unit type class instance of the same type, whose value
is the Unit type operand's value multiplied by the numerical operand's
value.
Multiplication with all other Types / Classes is not supported.
"""
if isinstance(other, numbers.Number):
if isinstance(self, FactorBase):
return self.__class__(other * self.value)
if isinstance(other, Decimal):
return self.__class__(self.value * other)
return self.__class__(self.value * Decimal(other))
if isinstance(other, Percentage):
return self.__class__(self.value * other.value / 100)
if isinstance(other, GainBase):
if isinstance(self, GainBase):
if self._gtype != other._gtype:
raise TypeError("Gain is of a different type.")
return self.__class__(self.value * other.value)
if other._gtype and not isinstance(self, other._gtype[1]):
raise TypeError("Gain is of a different type.")
if isinstance(self, other._gtype[0]):
return self.__class__(self.value * other.value)
else:
return other._gtype[0](self.value * other.value)
else:
return NotImplemented
def __rmul__(self, other):
return self.__mul__(other)
[docs] def __div__(self, other):
"""
Division of one Unit type class instance with a numerical type
results in a Unit type class instance of the same type, whose value
is the Unit type operand's value divided by the numerical operand's
value.
In this case, the first operand must be a Unit type class instance,
and not the reverse.
Division of one Unit type class instance by another of the same type
returns a numerical value, which is obtained by performing the
division with the operands' value.
Division with all other Types / Classes is not supported.
"""
if isinstance(other, numbers.Number):
if isinstance(other, Decimal):
return self.__class__(self.value / other)
else:
return self.__class__(self.value / Decimal(other))
elif isinstance(other, Percentage):
return self.__class__(self.value / other.value * 100)
elif isinstance(other, self.__class__):
return self.value / other.value
else:
return NotImplemented
def __rdiv__(self, other):
return NotImplemented
def __truediv__(self, other):
return self.__div__(other)
def __rtruediv__(self, other):
return self.__rdiv__(other)
[docs] def __sub__(self, other):
"""
Subtraction of two Unit class instances of the same type returns a
Unit class instance of the same type, with the difference of the two
operands as it's value.
If the other operand is a numerical type and evaluates to 0, this
object is simply returned unchanged.
Subtraction with all other Types / Classes is not supported.
"""
if isinstance(other, numbers.Number) and other == 0:
return self
else:
return self.__add__(other.__mul__(-1))
def __rsub__(self, other):
return other.__sub__(self)
def __abs__(self):
if self._value < 0:
return self.__class__(self._value * -1)
else:
return self
[docs] def _cmpkey(self):
"""
The comparison of two Unit type class instances of the
same type behaves identically to the comparison of the
operands' values.
Comparison with all other Types / Classes is not supported.
"""
return self._value
[docs] def _pluralize(self, value, ostr):
if value != 1 and self._pluralize_ostr:
return value, ostr + 's'
else:
return value, ostr
@property
def natural_repr(self):
if self._order_type == 1:
return self._pluralize(self._value, self._dostr)
ostr = self._dostr
value = self._value
done = False
while not done:
ostri = self._ostrs.index(ostr)
if 1 <= abs(value) < 1000:
done = True
elif abs(value) >= 1000:
if ostri < len(self._ostrs) - 1:
ostr = self._ostrs[ostri + 1]
value /= Decimal(1000)
else:
done = True
elif abs(value) < 1:
if ostri > 0:
ostr = self._ostrs[ostri - 1]
value *= Decimal(1000)
else:
done = True
return self._pluralize(value, ostr)
[docs] def _pack_str(self, num, unit):
if not isinstance(num, six.string_types):
num = str(num)
if self._separate_unit:
return ' '.join([num, unit])
else:
return num + unit
@property
def quantized_repr(self):
num, unit = self.natural_repr
neg = False
if num < 0:
neg = True
num = str(round_to_n(float(abs(num)), 5))
if neg is True:
num = '-' + num
return self._pack_str(num, unit)
@property
def integral_repr(self):
num, unit = self.natural_repr
neg = False
if num < 0:
neg = True
if num == 0:
return str(num) + unit
num = str(round_to_n(float(abs(num)), 2))
if neg is True:
num = '-' + num
return self._pack_str(num, unit)
@property
def integral_autoscaled_repr(self):
value = self._value
rval = None
for unit, factor in self._rorders:
uv, r = divmod(value, factor)
value = r
if not uv:
continue
part = self._pack_str(*self._pluralize(uv, unit))
if rval:
rval = ', '.join([rval, part])
else:
rval = part
if not rval:
unit, factor = self._rorders[-1]
return self._pack_str(*self._pluralize(0, unit))
return rval
[docs] def fmt_repr(self, fmt):
num, unit = self.natural_repr
parts = []
idx = 0
if '%v' in fmt:
fmt = fmt.replace('%v', '{' + str(idx) + '}')
# TODO Consider applying remove_exponent across the board
parts.append(remove_exponent(num))
idx += 1
if '%u' in fmt:
fmt = fmt.replace('%u', '{' + str(idx) + '}')
parts.append(unit)
idx += 1
return fmt.format(*parts)
def __repr__(self):
if self._rorders:
return self.integral_autoscaled_repr
return self._pack_str(*self.natural_repr)
[docs]class DummyUnit(UnitBase):
"""
This class provides a type for placeholder objects. The original
use case is for handling wave boundaries in streaming protocols.
Does not support any arithmetic operations.
"""
def __repr__(self):
return "Dummy Unit"
[docs]def parse_none(value):
"""
A placeholder parse function which can be used if the Unit
requires / supports no parsing.
"""
return value
[docs]class FactorBase(NumericalUnitBase):
pass
[docs]class Percentage(FactorBase):
"""
A base Unit class which provides support for Types that are essentially
percentages.
The contribution this base class makes is to be able to parse percentage
strings so that the Descendant class need not.
Only the standard :class:`NumericalUnitBase` Arithmetic is supported by
this class at this time.
"""
_has_bare_order = True
_orders = [('%', Decimal('1')), ('pc', Decimal('1')), ('', 100)]
_dostr = '%'
_allow_nOr = False
_regex_std = re.compile(r"^(?P<numerical>[\d]+\.?[\d]*)\s?(?P<order>(pc)?%?)(?P<residual>)$") # noqa
[docs]class Tolerance(Percentage):
pass
[docs]class GainBase(FactorBase):
_inverse_class = None
_gtype = None
def __rdiv__(self, other):
"""
Division of one Unit type class instance with a numerical type
results in a Unit type class instance of the same type, whose value
is the numerical operand's value divided by the unit type operand's
value.
In this case, the second operand must be a Unit type class instance,
and not the reverse.
"""
if isinstance(other, numbers.Number):
if not self._inverse_class:
if not self._gtype[0] == self._gtype[1]:
raise TypeError
inv_cls = self.__class__
else:
inv_cls = self._inverse_class
if isinstance(other, Decimal):
return inv_cls(other / self.value)
else:
return inv_cls(Decimal(other) / self.value)
else:
return NotImplemented
[docs] def in_db(self):
return 20 * log10(self._value)
