import core
from drawer.BaseDrawer import BaseDrawer
from entity.primitives import Tun2d


class Tun2dDrawer(BaseDrawer):

    def __init__(self, tun_2d, msp, style):
        super().__init__(tun_2d, msp,style)

    def initialize_data(self):
        assert isinstance(self.obj, Tun2d)

    def draw_agg(self):
        assert isinstance(self.obj, Tun2d)
        vec_list = [vec.agg_line for vec in self.obj.vec_list]
        self.draw_tun(vec_list, self.obj.middle_line.agg_line)

    def draw_div(self):
        assert isinstance(self.obj, Tun2d)
        vec_list = [vec.div_line for vec in self.obj.vec_list]
        self.draw_tun(vec_list, self.obj.middle_line.div_line)

    def draw_tun(self, vec_list, middle_line):
        assert isinstance(self.obj, Tun2d)
        for vec in vec_list:
            line = self.msp.add_line(vec[0], vec[1], dxfattribs={
                'layer': f'图层{self.obj.layer_id}'
            })
            line.rgb = self.obj.color

        # line = self.msp.add_line(middle_line[0], middle_line[1], dxfattribs={
        #     'layer': f'图层{self.obj.layer_id}'
        # })