Data commit

Task/Imaginary-base-numbers/Python/imaginary-base-numbers.py

@@ -0,0 +1,100 @@
+import math
+import re
+
+def inv(c):
+    denom = c.real * c.real + c.imag * c.imag
+    return complex(c.real / denom, -c.imag / denom)
+
+class QuaterImaginary:
+    twoI = complex(0, 2)
+    invTwoI = inv(twoI)
+
+    def __init__(self, str):
+        if not re.match("^[0123.]+$", str) or str.count('.') > 1:
+            raise Exception('Invalid base 2i number')
+        self.b2i = str
+
+    def toComplex(self):
+        pointPos = self.b2i.find('.')
+        posLen = len(self.b2i) if (pointPos < 0) else pointPos
+        sum = complex(0, 0)
+        prod = complex(1, 0)
+        for j in xrange(0, posLen):
+            k = int(self.b2i[posLen - 1 - j])
+            if k > 0:
+                sum += prod * k
+            prod *= QuaterImaginary.twoI
+        if pointPos != -1:
+            prod = QuaterImaginary.invTwoI
+            for j in xrange(posLen + 1, len(self.b2i)):
+                k = int(self.b2i[j])
+                if k > 0:
+                    sum += prod * k
+                prod *= QuaterImaginary.invTwoI
+        return sum
+
+    def __str__(self):
+        return str(self.b2i)
+
+def toQuaterImaginary(c):
+    if c.real == 0.0 and c.imag == 0.0:
+        return QuaterImaginary("0")
+
+    re = int(c.real)
+    im = int(c.imag)
+    fi = -1
+    ss = ""
+    while re != 0:
+        re, rem = divmod(re, -4)
+        if rem < 0:
+            rem += 4
+            re += 1
+        ss += str(rem) + '0'
+    if im != 0:
+        f = c.imag / 2
+        im = int(math.ceil(f))
+        f = -4 * (f - im)
+        index = 1
+        while im != 0:
+            im, rem = divmod(im, -4)
+            if rem < 0:
+                rem += 4
+                im += 1
+            if index < len(ss):
+                ss[index] = str(rem)
+            else:
+                ss += '0' + str(rem)
+            index = index + 2
+        fi = int(f)
+    ss = ss[::-1]
+    if fi != -1:
+        ss += '.' + str(fi)
+    ss = ss.lstrip('0')
+    if ss[0] == '.':
+        ss = '0' + ss
+    return QuaterImaginary(ss)
+
+for i in xrange(1,17):
+    c1 = complex(i, 0)
+    qi = toQuaterImaginary(c1)
+    c2 = qi.toComplex()
+    print "{0:8} -> {1:>8} -> {2:8}     ".format(c1, qi, c2),
+
+    c1 = -c1
+    qi = toQuaterImaginary(c1)
+    c2 = qi.toComplex()
+    print "{0:8} -> {1:>8} -> {2:8}".format(c1, qi, c2)
+print
+
+for i in xrange(1,17):
+    c1 = complex(0, i)
+    qi = toQuaterImaginary(c1)
+    c2 = qi.toComplex()
+    print "{0:8} -> {1:>8} -> {2:8}     ".format(c1, qi, c2),
+
+    c1 = -c1
+    qi = toQuaterImaginary(c1)
+    c2 = qi.toComplex()
+    print "{0:8} -> {1:>8} -> {2:8}".format(c1, qi, c2)
+
+print "done"