Extract renderer

2 files changed, 59 insertions, 43 deletions

diff --git a/libsolve.py b/libsolve.py
index d932dd6..61d7f40 100644
--- a/libsolve.py
+++ b/libsolve.py
@@ -7,31 +7,28 @@ from itertools import zip_longest
 def is_scalar(obj):
     return isinstance(obj, (Fraction, int))
 
+
 POLY_COLOR = None
 # POLY_COLOR = '36'  # Uncomment to have color
 
 
-# may not work. see
-# https://en.wikipedia.org/wiki/Unicode_subscripts_and_superscripts
-# to test with your font
-superscripts = '⁰¹²³⁴⁵⁶⁷⁸⁹' # must be <sup>0123456789</sup>
-
+class PolyRenderBase:
+    def render(self, poly: Poly) -> str:
+        raise NotImplementedError()
 
-def int2sup(n):
-    res = []
-    while n > 0:
-        res.append(superscripts[n % 10])
-        n //= 10
-    return ''.join(reversed(res))
+    def hint(self, poly: Poly) -> int:
+        raise NotImplementedError()
 
 
 class Poly:
-    def __init__(self, vec: Iterable, letter='x'):
+    def __init__(self, vec: Iterable, letter='x', renderer=None):
         '''
         vec: big-endian coefficients
         '''
+        from render import UnicodeRender
         self.vec = list(vec)
         self.letter = letter
+        self.renderer = renderer or UnicodeRender()
         self._trim_zeros()
 
     def _trim_zeros(self):
@@ -106,36 +103,10 @@ class Poly:
             del self.vec[0]
 
     def __repr__(self):
-        return self.get_hint()
+        return self.renderer.render(self)
 
-    def get_hint(self):
-        '''
-        Get raw text hints for outer structures to align items
-        '''
-        rev = list(reversed(self.vec))
-        if all(r == 0 for r in rev):
-            return '0'
-        d = self.deg()
-        res = []
-        for i in range(len(rev)):
-            if rev[i] == 0:
-                continue
-            if i == 0:
-                if rev[i] < 0:
-                    res.append('-')
-            else:
-                if rev[i] < 0:
-                    res.append(' - ')
-                else:
-                    res.append(' + ')
-
-            if abs(rev[i]) != 1 or i == d:
-                res.append(str(abs(rev[i])))
-            if i != d:
-                res.append(self.letter)
-            if d - i != 1:
-                res.append(int2sup(d - i))
-        return ''.join(res)
+    def hint(self):
+        return self.renderer.hint(self)
 
 
 class Row:
@@ -197,7 +168,7 @@ class Row:
         '''
         Get hints for other structures to align items
         '''
-        return [len(el.get_hint()) for el in self.lst]
+        return [el.hint() for el in self.lst]
 
     def set_hints(self, hints):
         '''
@@ -274,7 +245,7 @@ class Matrix:
             raise e
 
         return res
-    
+
     def __sub__(self, oth):
         return self + (-oth)
 
diff --git a/render.py b/render.py
new file mode 100644
index 0000000..6cd3c7d
--- /dev/null
+++ b/render.py
@@ -0,0 +1,45 @@
+from libsolve import Poly, PolyRenderBase
+
+
+class UnicodeRender(PolyRenderBase):
+    def __init__(self):
+        # may not work. see
+        # https://en.wikipedia.org/wiki/Unicode_subscripts_and_superscripts
+        # to test with your font
+        self.sup = '⁰¹²³⁴⁵⁶⁷⁸⁹'  # must be <sup>0123456789</sup>
+
+    def render(self, poly: Poly) -> str:
+        rev = list(reversed(poly.vec))
+        if all(r == 0 for r in rev):
+            return '0'
+        d = poly.deg()
+        res = []
+        for i in range(len(rev)):
+            if rev[i] == 0:
+                continue
+            if i == 0:
+                if rev[i] < 0:
+                    res.append('-')
+            else:
+                if rev[i] < 0:
+                    res.append(' - ')
+                else:
+                    res.append(' + ')
+
+            if abs(rev[i]) != 1 or i == d:
+                res.append(str(abs(rev[i])))
+            if i != d:
+                res.append(poly.letter)
+            if d - i != 1:
+                res.append(self._int2sup(d - i))
+        return ''.join(res)
+
+    def hint(self, poly: Poly) -> int:
+        return len(self.render(poly))
+
+    def _int2sup(self, n):
+        res = []
+        while n > 0:
+            res.append(self.sup[n % 10])
+            n //= 10
+        return ''.join(reversed(res))