Add polynomials

Author: eanorambuena

eggdriver/resources/math/__init__.py

@@ -1,4 +1,6 @@
 from eggdriver.resources.math.algorithms import *
 from eggdriver.resources.math.theoric import *
 from eggdriver.resources.math.float import *
-from eggdriver.resources.math.functions import *
+from eggdriver.resources.math.functions import *
+from eggdriver.resources.math.linear import *
+from eggdriver.resources.math.polynomial import *

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

@@ -2,9 +2,9 @@
 from eggdriver.resources.math.algorithms.solver.dichotomy import root
 from eggdriver.resources.math.constants import R
 
-def solutions(function, bias = 0, domain = R,  accurancy = 16, grade = 1):
+def solutions(function, bias = 0, domain = R,  accurancy = 16, degree = 1):
     roots = Set()
-    while roots.size < grade:
+    while roots.size < degree:
         r = root(function, bias, domain, accurancy)
         roots.insert(r)
     return roots

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

@@ -0,0 +1,51 @@
+from eggdriver.resources.structures.lists import List
+
+def dualExpand(a, b):
+    if a.size < b.size:
+        a.expand(b.size - a.size)
+    else:
+        b.expand(a.size - b.size)
+
+def plus(a, b, autoExpand = False):
+    result = Vector()
+    if len(a) != len(b):
+        if autoExpand:
+            dualExpand(a, b)
+        else:
+            raise Exception("Vectors have to be of the same size")
+    if len(a) != 0 and len(a) == len(b):
+        for i in range(0, a.size):
+            result.append(a[i] + b[i])
+    return result
+
+def dot(a, b, autoExpand = False):
+    result = 0
+    if len(a) != len(b):
+        if autoExpand:
+            dualExpand(a, b)
+        else:
+            raise Exception("Vectors have to be of the same size")
+    if len(a) != 0 and len(a) == len(b):
+        for i in range(0, a.size):
+            result += a[i] * b[i]
+    return result
+
+def scale(vector, scalar):
+    result = Vector()
+    if vector.size != 0:
+        for i in vector:
+            result.append(scalar * i)
+    return result
+
+class Vector(List):
+    def __init__(self, list = []):
+        super().__init__(list)
+    def plus(self, vector):
+        return plus(self, vector)
+    def dot(self, vector):
+        return dot(self, vector)
+    def scale(self, scalar):
+        return scale(self, scalar)
+    def expand(self, scalar):
+        [self.append(0) for i in range(0, scalar)]
+

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

@@ -0,0 +1,82 @@
+from eggdriver.resources.math.linear import Vector, dualExpand
+
+def x_(i):
+    if i==0 :
+        return ""
+    elif i == 1:
+        return "x"
+    return f"x^{i}"
+
+def plusPoly(a, b):
+    if type(a) != Polynomial:
+        a = Polynomial([a])
+    if type(b) != Polynomial:
+        b = Polynomial([b])
+    result = Polynomial()
+    if len(a) != len(b):
+        dualExpand(a, b)
+    if len(a) != 0 and len(a) == len(b):
+        for i in range(0, a.size):
+            result.append(a[i] + b[i])
+    return result
+
+def scalePoly(polynomial, scalar):
+    result = Vector()
+    if polynomial.size != 0:
+        for i in polynomial:
+            result.append(scalar * i)
+    return result
+
+def times(a, b):
+    result = Polynomial()
+    if len(a) != 0:
+        for i in range(0, len(a)):
+            product = Polynomial()
+            for j in range(0, i):
+                product.append(0)
+            for j in b.scale(a[i]):
+                product.append(j)
+            result = result.plus(product)
+    return result
+
+def vectorize(poly: str):
+    pass
+
+class Polynomial(Vector):
+    def __init__(self, poly = []):
+        if type(poly) != list:
+            if type(poly) != str:
+                poly = [poly]
+            else:
+                poly = vectorize(poly)
+        super().__init__(poly)
+    def display(self):
+        result = Polynomial()
+        nonZeros = 0
+        if self.size != 0:
+            for i in range(0, self.size):
+                if self[i] > 0:
+                    nonZeros += 1
+                if self[i] > 0 and nonZeros > 1:
+                    result.append("+ " + str(self[i]) + x_(i))
+                elif self[i] != 0:
+                    result.append(str(self[i]) + x_(i))
+        text = ""
+        for i in result:
+            text += i + " "
+        print(text)
+    def plus(self, vector):
+        return plusPoly(self, vector)
+    def times(self, vector):
+        return times(self, vector)
+    def power(self, exponent):
+        result = Polynomial(1)
+        for i in range(0, exponent):
+            result = result.times(self)
+        return result
+    def eval(self, x):
+        result = 0
+        for i in range(0, self.size):
+            f = self[i]
+            result += f * (x ** i)
+        return result