virson 2

Author: einav242

Visual/binary_tree.py

@@ -0,0 +1,31 @@
+#
+#
+# class Node:
+#    def __init__(self, data):
+#       self.left = None
+#       self.right = None
+#       self.data = data
+#
+#    def insert(self, data):
+# # Compare the new value with the parent node
+#       if self.data:
+#          if data < self.data:
+#             if self.left is None:
+#                self.left = Node(data)
+#             else:
+#                self.left.insert(data)
+#         elif data > self.data:
+#                if self.right is None:
+#                   self.right = Node(data)
+#                else:
+#                   self.right.insert(data)
+#       else:
+#          self.data = data
+#
+# # Print the tree
+#    def PrintTree(self):
+#       if self.left:
+#          self.left.PrintTree()
+#       print( self.data),
+#       if self.right:
+#          self.right.PrintTree()

Visual/test.py

@@ -0,0 +1,169 @@
+import re
+from email.headerregistry import Group
+import cvxpy as cp
+from cvxpy import Expression
+
+import cvxopt
+
+from cvxpy import Minimize, Problem, Variable, quad_form
+from cvxpy.problems.objective import Objective
+
+import operator
+
+priority0 = ['**']
+priority1 = ['@', '*', '/']
+priority2 = ['+', '-']
+
+
+def is_operator(s):
+    try:
+        return hasattr(operator, s)
+    except TypeError:
+        return False
+
+
+def is_float(s):
+    try:
+        float(s)
+        return True
+    except ValueError:
+        return False
+
+
+class Visual:
+    def __init__(self, obj: Objective):
+        self.name = obj.NAME
+        self.expr = str(obj.expr).replace("+ -", " - ")
+        self.parameters = []
+        self.variables = []
+        self.operators = []
+        self.func = []
+
+    def create_lists(self):
+        isFunc = False
+        isMatrix = False
+        func_string = ""
+        matrix_string = ""
+        for s in self.expr.split():
+            # --------func---------
+            if ')' in s:
+                isFunc = False
+                func_string += " " + s
+                self.func.append(func_string)
+                func_string = ""
+                continue
+            if isFunc:
+                func_string += " " + s
+                continue
+            if '(' in s:
+                isFunc = True
+                func_string += s
+                continue
+            # --------func---------
+
+            # --------parameter---------
+            if is_float(s) and not isMatrix:
+                self.parameters.append(s)
+                continue
+            if ']' in s:
+                isMatrix = False
+                matrix_string += " " + s
+                self.parameters.append(matrix_string)
+                matrix_string = ""
+                continue
+            if isMatrix:
+                matrix_string += " " + s
+                continue
+            if '[' in s:
+                isMatrix = True
+                matrix_string += s
+                continue
+            # --------parameter---------
+
+            # --------variables---------
+            if "var" in s:
+                self.variables.append(s)
+                continue
+            # --------variables---------
+
+            self.operators.append(s)
+
+    def split_expr(self):
+        if len(self.operators) == 0:
+            return
+
+    def show(self):
+        pass
+
+    def priority(self):
+        pr = 0
+        ans = ''
+        for op in self.operators:
+            if op in priority2:
+                self.operators.remove(op)
+                return op
+            if op in priority1:
+                pr = 1
+                ans = op
+            if op in priority0 and pr == 0:
+                ans = op
+        self.operators.remove(ans)
+        return ans
+
+
+# def visual(objective):
+#     str_objective = str(objective.expr)
+#     print(str_objective)
+#     split_objective = str_objective.split()
+#     print(objective.NAME)
+#     print(objective.args)
+
+
+# n = 4
+# A = [[1, 2, 3], [4, 5, 6], [7, 8, 9], [10, 11, 12]]
+# b = [1, 2, 3]
+# x = cp.Variable(n)
+# y = cp.Variable()
+#
+# objective = cp.Minimize(cp.sum_squares(A@x+b+y))
+#
+# constraints = [x <= 0, x <= 1, y <= 1]
+# prob = cp.Problem(objective, constraints)
+# prob.solve()
+#
+# visual(objective)
+
+n = 3
+P = cvxopt.matrix([13, 12, -2,
+                   12, 17, 6,
+                   -2, 6, 12], (n, n))
+q = cvxopt.matrix([-22, -14.5, 13], (n, 1))
+r = 1
+x_star = cvxopt.matrix([1, 1 / 2, -1], (n, 1))
+
+# Frame and solve the problem
+
+x = Variable(n)
+y = Variable(n)
+objective = Minimize(0.5 * quad_form(x, P) + cp.sum_squares(x) - q.T @ x + r)
+constraints = [x >= -1, x <= 1]
+
+p = Problem(objective, constraints)
+# The optimal objective is returned by p.solve().
+result = p.solve()
+print(str(objective.expr))
+
+# expression = "2 * math.pow(3, 2) + math.sqrt(16)"
+# tokens = re.findall('[\d\.]+|\(|\)|\+|\-|\@|\/|\^|\w+\(', str(objective.expr))
+#
+# print(tokens)
+
+for s in str(objective.expr).split():
+    print("s= ", s)
+
+v = Visual(objective)
+v.split_expr()
+print(v.func)
+print(v.parameters)
+print(v.variables)
+print(v.operators)

Visual/visual.py

@@ -13,6 +13,8 @@
 from cvxpy.error import DCPError
 import cvxpy as cp
 
+from Visual.visual import prob
+
 
 def get_solvers(problem):
     """Get valid solvers.

analyzer/solvers.py

@@ -27,3 +27,6 @@ def check_version():
     else:
         msg = "The request to pypi returned status code" + str(r.status_code)
         raise RuntimeError(msg)
+
+
+check_version()

analyzer/version.py

@@ -26,3 +26,8 @@ def test_solvers():
     prob = cp.Problem(cp.Minimize(cp.sum_squares(x)), [-1 <= x, x <= 1])
     solvers = get_solvers(prob)
     assert len(solvers) > 0
+
+
+test_solvers()
+test_checker()
+test_version()