Improve polynomials

Author: eanorambuena

eggdriver/resources/math/__init__.py

@@ -3,4 +3,5 @@
 from eggdriver.resources.math.float import *
 from eggdriver.resources.math.functions import *
 from eggdriver.resources.math.linear import *
-from eggdriver.resources.math.polynomial import *
+from eggdriver.resources.math.polynomial import *
+from eggdriver.resources.math.constants import *

eggdriver/resources/math/__pycache__/__init__.cpython-39.pyc

@@ -1,10 +1,17 @@
 from eggdriver.resources.structures.lists import Set
 from eggdriver.resources.math.algorithms.solver.dichotomy import root
-from eggdriver.resources.math.constants import R
+from eggdriver.resources.math.constants import R, inf
 
-def solutions(function, bias = 0, domain = R,  accurancy = 16, degree = 1):
+def solve(function, bias = 0, domain = R,  accurancy = 16, degree = 1, alerts = True):
     roots = Set()
-    while roots.size < degree:
+    count = 0
+    customDomain = domain != R
+    while roots.size < degree and count < inf:
+        if not customDomain:
+            domain = [- 2.0 ** int(count), 2.0 ** int(count)]
         r = root(function, bias, domain, accurancy)
-        roots.insert(r)
+        roots.insert(str(r))
+        if alerts:
+            print(degree - roots.size, "solutions remaining")
+        count += 1 / (2 * degree)
     return roots

eggdriver/resources/math/__pycache__/constants.cpython-39.pyc

@@ -4,7 +4,7 @@
 
 def root(function, bias = 0, domain = R,  accurancy = 16):
     """Gives a root of function(x) = bias, in a certain domain"""
-    error = 10 ** (- accurancy - 5)
+    error = 10 ** (- accurancy - 100)
     while True:
         bound = [random.uniform(domain[0], domain[1]), random.uniform(domain[0], domain[1])]
         bound.sort()
@@ -17,7 +17,7 @@ def root(function, bias = 0, domain = R,  accurancy = 16):
         mean = (a + b) / 2
         fm = function(mean) - bias
         if abs(fm) < error:
-            return truncate(round(mean, accurancy + 3), accurancy)
+            return round(mean, accurancy)
         if fa * fm < 0:
             b = mean
         else:

eggdriver/resources/math/__pycache__/polynomial.cpython-39.pyc

@@ -1,4 +1,4 @@
-inf = 1000000
+inf = 10 ** 9
 R = [-inf, inf]
 
 e =  2.7182818284590452353602874713526624977572470936999595749669676277240766303

eggdriver/resources/math/algorithms/solver/__init__.py

@@ -1,11 +1,12 @@
 from eggdriver.resources.math.linear import Vector, dualExpand
+from eggdriver.resources.math.algorithms.solver import solve
 
-def x_(i):
-    if i==0 :
+def x_(i, variable = "x"):
+    if i == 0 :
         return ""
     elif i == 1:
-        return "x"
-    return f"x^{i}"
+        return variable
+    return f"{variable}^{i}"
 
 def plusPoly(a, b):
     if type(a) != Polynomial:
@@ -43,13 +44,14 @@ def vectorize(poly: str):
     pass
 
 class Polynomial(Vector):
-    def __init__(self, poly = []):
+    def __init__(self, poly = [], variable = "x"):
         if type(poly) != list:
             if type(poly) != str:
                 poly = [poly]
             else:
                 poly = vectorize(poly)
         super().__init__(poly)
+        self.var = variable
     def display(self):
         result = Polynomial()
         nonZeros = 0
@@ -58,9 +60,9 @@ def display(self):
                 if self[i] > 0:
                     nonZeros += 1
                 if self[i] > 0 and nonZeros > 1:
-                    result.append("+ " + str(self[i]) + x_(i))
+                    result.append("+ " + str(self[i]) + x_(i, self.var))
                 elif self[i] != 0:
-                    result.append(str(self[i]) + x_(i))
+                    result.append(str(self[i]) + x_(i, self.var))
         text = ""
         for i in result:
             text += i + " "
@@ -79,4 +81,14 @@ def eval(self, x):
         for i in range(0, self.size):
             f = self[i]
             result += f * (x ** i)
-        return result
+        return result
+    @property
+    def zeros(self):
+        return solve(self.eval, degree = self.degree)
+    @property
+    def degree(self):
+        deg = 0
+        for i in range(0, self.size):
+            if self[i] != 0:
+                deg = i
+        return deg