| 12345678910111213141516171819202122232425262728293031323334 |
- 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}'
- # })
|
