vent_DXF_2/drawer/duct_drawer.py

import math

from ezdxf.enums import TextEntityAlignment

import core
from drawer.BaseDrawer import BaseDrawer, VentGraphDrawer
from entity.primitives import Window, AirDuct


class DuctDrawer(BaseDrawer):
    WIDTH_MULTIPLIER = 1.5
    HEIGHT_MULTIPLIER = 1 / 6

    def __init__(self, obj, msp, style):

        super().__init__(obj, msp, style)

        if self.style == 1:
            self.obj.color = (0, 255, 255)
            self.line_height = 50

        if self.style == 0:
            self.line_height = 0

        self.duct_drawers = {
            0: self.draw_duct_style_0,
            1: self.draw_duct_style_1,
        }

    def draw_duct_style_0(self, duct_type):
        """
        绘制风管路径（支持聚合线和分散线）

        :param duct_type: 绘制类型，可选 'agg'（聚合线）或 'div'（分散线）
        """
        assert isinstance(self.obj, AirDuct)
        assert duct_type in {'agg', 'div'}, "Invalid duct type. Must be 'agg' or 'div'."

        # 根据类型选择对应的线段属性
        line_attr = 'agg_line' if duct_type == 'agg' else 'div_line'

        # 遍历路径列表并绘制
        for i in range(len(self.obj.path_list)):
            path = self.obj.path_list[i]
            start_point, end_point = getattr(path, line_attr)  # 获取起点和终点
            self.draw_air_duct_list_0(start_point, end_point)  # 绘制当前线段

            # 如果不是最后一段，绘制连接线
            if i + 1 < len(self.obj.path_list):
                next_start_point = getattr(self.obj.path_list[i + 1], line_attr)[0]
                connection_line = self.msp.add_line(end_point, next_start_point)
                connection_line.rgb = self.obj.color  # 设置连接线颜色

    def draw_duct_style_1(self, duct_type):
        assert isinstance(self.obj, AirDuct)
        assert duct_type in {'agg', 'div'}, "Invalid duct type. Must be 'agg' or 'div'."

        # 根据类型选择对应的线段属性
        line_attr = 'agg_line' if duct_type == 'agg' else 'div_line'

        # 遍历路径列表并绘制
        for i in range(len(self.obj.path_list)):
            path = self.obj.path_list[i]
            start_point, end_point = getattr(path, line_attr)  # 获取起点和终点
            self.draw_air_duct_list_2(start_point, end_point)  # 绘制当前线段

            # 如果不是最后一段，绘制连接线
            if i + 1 < len(self.obj.path_list):
                next_start_point = getattr(self.obj.path_list[i + 1], line_attr)[0]
                connection_line = self.msp.add_line(end_point, next_start_point,
                                                    dxfattribs={"layer": f"图层{self.obj.layer_id}", "lineweight": 50})
                connection_line.rgb = self.obj.color  # 设置连接线颜色

    def draw_agg(self):
        self.duct_drawers[self.style]('agg')

    def draw_div(self):
        self.duct_drawers[self.style]('div')

    def initialize_data(self):
        pass

    #   d1           a1 b1
    # c1|----------c2|F|c3
    #   d2   center a2  b2

    def draw_air_duct_list_0(self, from_, to_):

        distance = core.distance(from_, to_)

        ad_len = self.get_draw_air_duct_len()

        ad_height = self.obj.width * 0.3

        ad_width = self.obj.width * 3

        # if from_[0] < to_[0]:
        #     route_ = core.calculate_angle_with_x_axis(from_, to_)
        # else:
        #     route_ = core.calculate_angle_with_x_axis(to_, from_)
        #
        route_ = core.calculate_angle_with_x_axis(from_, to_)
        seg_from = from_

        while distance > ad_len:
            seg_to = core.find_point_on_segment(seg_from, to_, ad_len)

            center_ = core.find_point_on_line(seg_from, seg_to, 1 / 8)

            self.draw_single_duct_0(center_, route_, ad_width, ad_height)

            seg_from = seg_to

            distance = distance - ad_len

        self.draw_single_duct_0(seg_from, route_, distance, ad_height)

    def draw_air_duct_list_2(self, from_, to_):
        distance = core.distance(from_, to_)

        ad_len = self.get_draw_air_duct_len()

        ad_height = self.obj.width * 0.3

        ad_width = self.obj.width * 3

        route_ = core.calculate_angle_with_x_axis(from_, to_)

        seg_from = from_

        while distance > ad_len:

            seg_to = core.find_point_on_segment(seg_from, to_, ad_len)

            self.draw_single_duct_1(seg_from, route_, ad_width, ad_height)

            seg_from = seg_to

            distance = distance - ad_len

        self.draw_single_duct_1(seg_from, route_, distance, ad_height)

    def draw_line(self, start, end):
        dxfattribs = {"layer": f"图层{self.obj.layer_id}", "lineweight": self.line_height}
        line = self.msp.add_line(start, end, dxfattribs=dxfattribs)
        line.rgb = self.obj.color

        return line

    def draw_air_duct(self, center, route):
        c1 = center[0] - self.obj.width * self.WIDTH_MULTIPLIER, center[1]
        c2 = center[0] + self.obj.width, center[1]

        c1 = core.rotate_point_around_another(c1, center, route)
        c2 = core.rotate_point_around_another(c2, center, route)

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

        d1 = center[0] - self.obj.width * 1.5, center[1] + self.obj.width * 1 / 6
        d2 = center[0] - self.obj.width * 1.5, center[1] - self.obj.width * 1 / 6

        d1 = core.rotate_point_around_another(d1, center, route)
        d2 = core.rotate_point_around_another(d2, center, route)

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

        a1 = center[0] + self.obj.width * 1, center[1] + self.obj.width * 1 / 6
        a2 = center[0] + self.obj.width * 1, center[1] - self.obj.width * 1 / 6

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

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

        f = center[0] + self.obj.width * 1.25, center[1]
        f = core.rotate_point_around_another(f, center, route)
        if route < 0: route = route + math.pi
        angle = math.degrees(route)
        dxfattribs = {
            'insert': f, 'style': 'msyh',
            "layer": f"图层{self.obj.layer_id}",
        }

        text = self.msp.add_text("F", rotation=angle, dxfattribs=dxfattribs, height=self.obj.width / 6).set_placement(f,
                                                                                                                      align=TextEntityAlignment.MIDDLE_CENTER)
        text.rgb = self.obj.color
        b1 = center[0] + self.obj.width * 1.5, center[1] + self.obj.width * 1 / 6
        b2 = center[0] + self.obj.width * 1.5, center[1] - self.obj.width * 1 / 6

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

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

    def get_draw_air_duct_len(self):
        return self.obj.width * 3.5

    #  a1   a3
    # c1|---|c2
    #  a2   a4
    def draw_single_duct_1(self, center, route, ad_len, ad_height):
        c1 = center
        c2 = center[0] + ad_len, center[1]
        a1 = center[0], center[1] + ad_height
        a2 = center[0], center[1] - ad_height
        a3 = c2[0], c2[1] + ad_height
        a4 = c2[0], c2[1] - ad_height

        c2 = core.rotate_point_around_another(c2, center, route)
        a1 = core.rotate_point_around_another(a1, center, route)
        a2 = core.rotate_point_around_another(a2, center, route)
        a3 = core.rotate_point_around_another(a3, center, route)
        a4 = core.rotate_point_around_another(a4, center, route)

        self.draw_line(c1, c2)

        self.draw_line(a1, a2)

        self.draw_line(a3, a4)

    # a1a3
    # |F|c1---c2
    # a2a4
    def draw_single_duct_0(self, center, route, ad_len, ad_height):

        f_p = center

        angle = math.degrees(route)
        if 180 < angle: angle = angle + 180
        dxfattribs = {
            'insert': f_p, 'style': 'msyh',
            "layer": f"图层{self.obj.layer_id}",
        }

        text = self.msp.add_text("F", rotation=angle, dxfattribs=dxfattribs, height=self.obj.width / 6).set_placement(
            f_p,
            align=TextEntityAlignment.MIDDLE_CENTER)
        text.rgb = self.obj.color
        c1 = center[0] + 1 / 8 * ad_len, center[1]
        c2 = center[0] + ad_len, center[1]
        a1 = center[0] - 1 / 8 * ad_len, center[1] + ad_height
        a2 = center[0] - 1 / 8 * ad_len, center[1] - ad_height
        a3 = center[0] + 1 / 8 * ad_len, c2[1] + ad_height
        a4 = center[0] + 1 / 8 * ad_len, c2[1] - ad_height
        c1 = core.rotate_point_around_another(c1, center, route)
        c2 = core.rotate_point_around_another(c2, center, route)
        a1 = core.rotate_point_around_another(a1, center, route)
        a2 = core.rotate_point_around_another(a2, center, route)
        a3 = core.rotate_point_around_another(a3, center, route)
        a4 = core.rotate_point_around_another(a4, center, route)

        self.draw_line(c1, c2)

        self.draw_line(a1, a2)

        self.draw_line(a3, a4)