diff --git a/Horizontal.py b/Horizontal.py index d8a9ae6..5d73b94 100644 --- a/Horizontal.py +++ b/Horizontal.py @@ -7,3 +7,13 @@ class Horizontal(Constraint): self.pt1 = pt1 self.shape2 = shape2 self.pt2 = pt2 + + def toEqu(self): + varrefs1 = self.shape1.toEqu(pt1)[1] + varrefs2 = self.shape2.toEqu(pt2)[1] + coeffs = [] + for i in range(len(varrefs1)): + coeffs.append((1 * varrefs1[i], (self.shape1, i))) + for i in range(len(varrefs2)): + coeffs.append((-1 * varrefs2[i], (self.shape2, i)) + return [(coeffs, 0)] diff --git a/Vertical.py b/Vertical.py index 58b8d34..ad0dfaa 100644 --- a/Vertical.py +++ b/Vertical.py @@ -7,3 +7,13 @@ class Vertical(Constraint): self.pt1 = pt1 self.shape2 = shape2 self.pt2 = pt2 + + def toEqu(self): + varrefs1 = self.shape1.toEqu(pt1)[0] + varrefs2 = self.shape2.toEqu(pt2)[0] + coeffs = [] + for i in range(len(varrefs1)): + coeffs.append((1 * varrefs1[i], (self.shape1, i))) + for i in range(len(varrefs2)): + coeffs.append((-1 * varrefs2[i], (self.shape2, i)) + return [(coeffs, 0)]