import core
from drawer.BaseDrawer import BaseDrawer, VentGraphDrawer


class SealedDrawer(VentGraphDrawer):


    def initialize_data(self):
        self.obj.color = (255, 255, 255)

    # a1   b1
    # |    |
    # |    |
    # a2   b2
    def draw_obj(self,center,route):

        center_x = center[0]
        center_y = center[1]

        a1 = center_x-self.obj.width/2,center_y+3/2 *self.obj.width
        a2 = center_x-self.obj.width/2,center_y- self.obj.width
        b1 = center_x+self.obj.width/2,center_y+3/2 *self.obj.width
        b2 = center_x+self.obj.width/2,center_y-self.obj.width


        a1 = core.rotate_point_around_another(a1, center, route)
        a2 = core.rotate_point_around_another(a2, center, route)
        b1 = core.rotate_point_around_another(b1, center, route)
        b2 = core.rotate_point_around_another(b2, center, route)

        line = self.msp.add_line(a1, a2, dxfattribs={"layer": f"图层{self.obj.layer_id}"})
        line.color = self.obj.color

        line = self.msp.add_line(b1, b2, dxfattribs={"layer": f"图层{self.obj.layer_id}"})
        line.color = self.obj.color