Add Matrix, improve Polynomial

Author: eanorambuena

eggdriver/resources/__pycache__/utils.cpython-39.pyc

@@ -13,7 +13,7 @@ def plus(a, b, autoExpand = False):
         if autoExpand:
             dualExpand(a, b)
         else:
-            raise Exception("Vectors have to be of the same size")
+            raise LinearError
     if len(a) != 0 and len(a) == len(b):
         for i in range(0, a.size):
             result.append(a[i] + b[i])
@@ -25,7 +25,7 @@ def dot(a, b, autoExpand = False):
         if autoExpand:
             dualExpand(a, b)
         else:
-            raise Exception("Vectors have to be of the same size")
+            raise LinearError
     if len(a) != 0 and len(a) == len(b):
         for i in range(0, a.size):
             result += a[i] * b[i]
@@ -39,6 +39,7 @@ def scale(vector, scalar):
     return result
 
 def vectorize(poly: str):
+    """Transform a string polynomial into a coordinates vector"""
     p = poly.split(" ")
     result = Vector()
     coefs = List()
@@ -77,3 +78,22 @@ def scale(self, scalar):
     def expand(self, scalar):
         [self.append(0) for i in range(0, scalar)]
 
+class Matrix(Vector):
+    def __init__(self, n, m):
+        vectorOfVectors = Vector()
+        for i in range(m):
+            v = Vector()
+            v.expand(n)
+            vectorOfVectors.append(v)
+        super().__init__(vectorOfVectors)
+    def display(self):
+        for i in self:
+            print(i.display(True, ["(", ")"]))
+
+class LinearError(Exception):
+    def __init__(self, type = 0):
+        message = {
+            0: "Vectors have to be of the same size"
+        }
+        super().__init__(message[type])
+

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

@@ -40,6 +40,12 @@ def times(a, b):
             result = result.plus(product)
     return result
 
+def fromZeros(zeros = []):
+    result = Polynomial(1)
+    for i in zeros:
+        result = Polynomial([-i, 1]).times(result)
+    return result
+
 class Polynomial(Vector):
     def __init__(self, poly = [], variable = "x"):
         if type(poly) != list:

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

@@ -18,11 +18,14 @@ def removeLast(self):
         self.reverse()
     def contains(self, item):
         return item in self
-    def display(self):
-        text = "[ "
+    def display(self, returnText = False, symbols = ["[", "]"]):
+        text = symbols[0] + " "
         for i in self:
             text += str(i) + " "
-        print(text + "]")
+        if returnText:
+            return text + symbols[1]
+        else:
+            print(text + "]")
     def Iterator(self):
         return Iterator(self)
     def iterate(self, function):

eggdriver/resources/math/linear.py

@@ -11,3 +11,4 @@ def colour(id: int, bgcolor: int = 0):
 
 def indexes(x):
     return range(0, len(x))
+

eggdriver/resources/math/polynomial.py

@@ -1,6 +1,7 @@
-from eggdriver import Polynomial, vectorize
-
-
-u = Polynomial("7 +x -49x^2 -2x^9 +x^40 +3x^2")
-u.display()
+from eggdriver import Matrix, Vector
 
+v = Vector()
+v.expand(3)
+v.display()
+u = Matrix(3, 3)
+u.display()