scalarfloat support

1 files changed, 77 insertions, 0 deletions

diff --git a/scalarfloat.py b/scalarfloat.py
new file mode 100644
index 0000000..01d7080
--- /dev/null
+++ b/scalarfloat.py
@@ -0,0 +1,77 @@
+# coding: utf-8
+
+from __future__ import print_function, absolute_import, division, unicode_literals
+
+if False:  # MYPY
+    from typing import Text, Any, Dict, List  # NOQA
+
+__all__ = ["ScalarFloat", "ExponentialFloat", "ExponentialCapsFloat"]
+
+from .compat import no_limit_int  # NOQA
+
+
+class ScalarFloat(float):
+    def __new__(cls, *args, **kw):
+        # type: (Any, Any, Any) -> Any
+        width = kw.pop('width', None)            # type: ignore
+        prec = kw.pop('prec', None)              # type: ignore
+        m_sign = kw.pop('m_sign', None)          # type: ignore
+        exp = kw.pop('exp', None)                # type: ignore
+        e_width = kw.pop('e_width', None)        # type: ignore
+        e_sign = kw.pop('e_sign', None)          # type: ignore
+        underscore = kw.pop('underscore', None)  # type: ignore
+        v = float.__new__(cls, *args, **kw)      # type: ignore
+        v._width = width
+        v._prec = prec
+        v._m_sign = m_sign
+        v._exp = exp
+        v._e_width = e_width
+        v._e_sign = e_sign
+        v._underscore = underscore
+        return v
+
+    def __iadd__(self, a):  # type: ignore
+        # type: (Any) -> Any
+        x = type(self)(self + a)
+        x._width = self._width  # type: ignore
+        x._underscore = self._underscore[:] if self._underscore is not None else None  # type: ignore  # NOQA
+        return x
+
+    def __ifloordiv__(self, a):  # type: ignore
+        # type: (Any) -> Any
+        x = type(self)(self // a)
+        x._width = self._width  # type: ignore
+        x._underscore = self._underscore[:] if self._underscore is not None else None  # type: ignore  # NOQA
+        return x
+
+    def __imul__(self, a):  # type: ignore
+        # type: (Any) -> Any
+        x = type(self)(self * a)
+        x._width = self._width  # type: ignore
+        x._underscore = self._underscore[:] if self._underscore is not None else None  # type: ignore  # NOQA
+        return x
+
+    def __ipow__(self, a):  # type: ignore
+        # type: (Any) -> Any
+        x = type(self)(self ** a)
+        x._width = self._width  # type: ignore
+        x._underscore = self._underscore[:] if self._underscore is not None else None  # type: ignore  # NOQA
+        return x
+
+    def __isub__(self, a):  # type: ignore
+        # type: (Any) -> Any
+        x = type(self)(self - a)
+        x._width = self._width  # type: ignore
+        x._underscore = self._underscore[:] if self._underscore is not None else None  # type: ignore  # NOQA
+        return x
+
+
+class ExponentialFloat(ScalarFloat):
+    def __new__(cls, value, width=None, underscore=None):
+        # type: (Any, Any, Any) -> Any
+        return ScalarFloat.__new__(cls, value, width=width, underscore=underscore)
+
+class ExponentialCapsFloat(ScalarFloat):
+    def __new__(cls, value, width=None, underscore=None):
+        # type: (Any, Any, Any) -> Any
+        return ScalarFloat.__new__(cls, value, width=width, underscore=underscore)