from Sketch import Sketch
from shapes import *
from constraints import *

class SampleSketch(Sketch):
    def __init__(self):
        Sketch.__init__(self)

        l1 = Line(0, 1, 1, 1)
        self.shapes.append(l1)
        l2 = Line(1, 1, 1, 0)
        self.shapes.append(l2)
        l3 = Line(1, 0, 0, 0)
        self.shapes.append(l3)
        l4 = Line(0, 0, 0, 1)
        self.shapes.append(l4)
        c = Circle(0, 0, 0.3)
        self.shapes.append(c)
        l5 = Line(0, 0, 0, 0)
        self.shapes.append(l5)

        self.constraints.append(Horizontal(l1, 0, l1, 1))
        self.constraints.append(Connect(l1, 1, l2, 0))
        self.constraints.append(Vertical(l2, 0, l2, 1))
        self.constraints.append(Connect(l2, 1, l3, 0))
        self.constraints.append(Horizontal(l3, 0, l3, 1))
        self.constraints.append(Connect(l3, 1, l4, 0))
        self.constraints.append(Vertical(l4, 0, l4, 1))
        self.constraints.append(Connect(l4, 1, l1, 0))

        self.constraints.append(Connect(c, 1, l2, 0))
        self.constraints.append(Horizontal(c, 4, l3, 1))
#        self.constraints.append(Connect(c, 4, l3, 1))

        self.constraints.append(XDistance(l1, 0, l1, 1, 1.2))
        self.constraints.append(YDistance(l4, 0, l4, 1, 0.8))

        self.constraints.append(XFix(l1, 0, 0))
        self.constraints.append(YFix(l1, 0, 0))

        self.constraints.append(Connect(l5, 0, l2, 1))
        self.constraints.append(Connect(l5, 2, l1, 0))