civil-tools-v 0.0.1__py3-none-any.whl → 0.0.3__py3-none-any.whl

CivilTools/Const/CAD.py

@@ -0,0 +1,2 @@
+class CADConst:
+    BY_LAYER = 256

CivilTools/Const/Concrete.py

@@ -0,0 +1,144 @@
+class ConcreteLevel:
+    def __init__(
+        self,
+        level: int,
+        fck: float,
+        ftk: float,
+        fc: float,
+        ft: float,
+        elastic_module: float,
+    ):
+        self.__level = level
+        self.__fck = fck
+        self.__ftk = ftk
+        self.__fc = fc
+        self.__ft = ft
+        self.__elastic_module = elastic_module
+
+    @property
+    def name(self):
+        return f"C{self.__level}"
+
+    @property
+    def fck(self):
+        """混凝土轴心抗压强度标准值，MPa"""
+        return self.__fck
+
+    @property
+    def ftk(self):
+        """混凝土轴心抗拉强度标准值，MPa"""
+        return self.__ftk
+
+    @property
+    def fc(self):
+        """混凝土轴心抗压强度设计值，MPa"""
+        return self.__fc
+
+    @property
+    def ft(self):
+        """混凝土轴心抗拉强度设计值，MPa"""
+        return self.__ft
+
+    @property
+    def elastic_module(self):
+        """混凝土弹性模量，MPa"""
+        return self.__elastic_module
+
+
+fck = {
+    15: 10.0,
+    20: 13.4,
+    25: 16.7,
+    30: 20.1,
+    35: 23.4,
+    40: 26.8,
+    45: 29.6,
+    50: 32.4,
+    55: 35.5,
+    60: 38.5,
+    65: 41.5,
+    70: 44.5,
+    75: 47.4,
+    80: 50.2,
+}
+ftk = {
+    15: 1.27,
+    20: 1.54,
+    25: 1.78,
+    30: 2.01,
+    35: 2.2,
+    40: 2.39,
+    45: 2.51,
+    50: 2.64,
+    55: 2.74,
+    60: 2.85,
+    65: 2.93,
+    70: 2.99,
+    75: 3.05,
+    80: 3.11,
+}
+fc = {
+    15: 7.2,
+    20: 9.6,
+    25: 11.9,
+    30: 14.3,
+    35: 16.7,
+    40: 19.1,
+    45: 21.1,
+    50: 23.1,
+    55: 25.3,
+    60: 27.5,
+    65: 29.7,
+    70: 31.8,
+    75: 33.8,
+    80: 35.9,
+}
+ft = {
+    15: 0.91,
+    20: 1.1,
+    25: 1.27,
+    30: 1.43,
+    35: 1.57,
+    40: 1.71,
+    45: 1.8,
+    50: 1.89,
+    55: 1.96,
+    60: 2.04,
+    65: 2.09,
+    70: 2.14,
+    75: 2.18,
+    80: 2.22,
+}
+e = {
+    15: 22000,
+    20: 25500,
+    25: 28000,
+    30: 30000,
+    35: 31500,
+    40: 32500,
+    45: 33500,
+    50: 34500,
+    55: 35500,
+    60: 36000,
+    65: 36500,
+    70: 37000,
+    75: 37500,
+    80: 38000,
+}
+
+
+class Concrete:
+    C15 = ConcreteLevel(15, fck[15], ftk[15], fc[15], ft[15], e[15])
+    C20 = ConcreteLevel(20, fck[20], ftk[20], fc[20], ft[20], e[20])
+    C25 = ConcreteLevel(25, fck[25], ftk[25], fc[25], ft[25], e[25])
+    C30 = ConcreteLevel(30, fck[30], ftk[30], fc[30], ft[30], e[30])
+    C35 = ConcreteLevel(35, fck[35], ftk[35], fc[35], ft[35], e[35])
+    C40 = ConcreteLevel(40, fck[40], ftk[40], fc[40], ft[40], e[40])
+    C45 = ConcreteLevel(45, fck[45], ftk[45], fc[45], ft[45], e[45])
+    C50 = ConcreteLevel(50, fck[50], ftk[50], fc[50], ft[50], e[50])
+    C55 = ConcreteLevel(55, fck[55], ftk[55], fc[55], ft[55], e[55])
+    C60 = ConcreteLevel(60, fck[60], ftk[60], fc[60], ft[60], e[60])
+    C65 = ConcreteLevel(65, fck[65], ftk[65], fc[65], ft[65], e[65])
+    C70 = ConcreteLevel(70, fck[70], ftk[70], fc[70], ft[70], e[70])
+    C75 = ConcreteLevel(75, fck[75], ftk[75], fc[75], ft[75], e[75])
+    C80 = ConcreteLevel(80, fck[80], ftk[80], fc[80], ft[80], e[80])

CivilTools/Const/Steel.py

@@ -0,0 +1,21 @@
+class SteelLevel:
+    def __init__(
+        self,
+        name: str,
+        fy: float,
+        elastic_module: float,
+    ):
+        self.name = name
+        self.fy = fy
+        self.elastic_module = elastic_module
+
+
+class Steel:
+    HRB300 = SteelLevel("HRB300", 300, 2000000)
+    HRB355 = SteelLevel("HRB355", 300, 2000000)
+    HRB400 = SteelLevel("HRB400", 300, 2000000)
+    HRB500 = SteelLevel("HRB500", 300, 2000000)
+    Q235 = SteelLevel("Q235", 111, 2000000)
+    Q345 = SteelLevel("Q345", 111, 2000000)
+    Q355 = SteelLevel("Q355", 111, 2000000)
+    Q420 = SteelLevel("Q420", 111, 2000000)

CivilTools/Const/__init__.py

@@ -0,0 +1,3 @@
+from .Concrete import Concrete, ConcreteLevel
+from .Steel import Steel, SteelLevel
+from .CAD import CADConst

CivilTools/DXFGenerator/BasicDXF.py

@@ -1,3 +1,240 @@
+import ezdxf
+from ezdxf.enums import TextEntityAlignment
+from ezdxf.addons.drawing import RenderContext, Frontend
+from ezdxf.addons.drawing.matplotlib import MatplotlibBackend
+import matplotlib.pyplot as plt
+import warnings
+from math import inf
+from typing import List, Iterable, Tuple
+from .LayerManager import CADLayer, CADColor, CADLineType
+from .DrawingAttribs import DrawingAttribs, PolylineAttribs, TextAttribs
+from CivilTools.Const import CADConst
+
 
 class BasicDXF:
-    pass
+    file_extension = ".dxf"
+    DXF2007 = "AC1021"
+    line_type_patterns = {
+        CADLineType.Continuous: [1, 1],
+        CADLineType.CENTER: [1, 0.4, -0.2, 0.1, -0.3],
+        CADLineType.DASHDOT: [1, 0.6, -0.2, 0, -0.2],
+        CADLineType.DASHED: [1, 0.7, -0.3],
+    }
+    """线型及其Pattern定义，Pattern定义可参考ezdxf内的定义
+    The simple line type pattern is a list of
+    floats :code:`[total_pattern_length, elem1, elem2, ...]`
+    where an element > 0 is a line, an element < 0 is a gap and  an
+    element == 0.0 is a dot.
+    """
+    DIM_STYLE_PREFIX = "CT"
+    TEXT_STYLE_NAME = "CT2025"
+    TEXT_SHX = "xd-hzs.shx"
+    TEXT_BIG_SHX = "xd-hztxt.shx"
+
+    def __init__(self):
+        self.doc = ezdxf.new(BasicDXF.DXF2007)
+        self.model_space = self.doc.modelspace()
+        self.__loaded_line_types = []
+        self.__min_point = [inf, inf]
+        self.__max_point = [-inf, -inf]
+        self.__load_text_type()
+        self.__load_dim_type(100)
+
+    def init_layers(self, layer_list: List[CADLayer]):
+        layers = self.doc.layers
+        for my_layer in layer_list:
+            # 如果图层名称已存在，则不进行新增
+            if my_layer.name in [l.dxf.name for l in layers]:
+                warnings.warn(f"Layer {my_layer.name} already existed.")
+                continue
+            temp_layer = layers.new(name=my_layer.name)
+            temp_layer.color = my_layer.color.value
+            self.__load_line_type(
+                my_layer.line_type.name, BasicDXF.line_type_patterns[my_layer.line_type]
+            )
+            temp_layer.dxf.linetype = my_layer.line_type.name
+            self.doc.header["$CLAYER"] = my_layer.name
+
+    def _add_horizental_line(
+        self, start_x, start_y, length: float, attribs: PolylineAttribs | None = None
+    ):
+        self._add_polyline([[start_x, start_y], [start_x + length, start_y]], attribs)
+
+    def _add_vertical_line(
+        self, start_x, start_y, length: float, attribs: PolylineAttribs | None = None
+    ):
+        self._add_polyline([[start_x, start_y], [start_x, start_y + length]], attribs)
+
+    def _add_rectangle(
+        self,
+        start_x,
+        start_y,
+        width: float,
+        height: float,
+        attribs: PolylineAttribs | None = None,
+    ):
+        pts = [
+            [start_x, start_y],
+            [start_x + width, start_y],
+            [start_x + width, start_y + height],
+            [start_x, start_y + height],
+        ]
+        attribs.close = True
+        self._add_polyline(pts, attribs)
+
+    def _add_polyline(
+        self,
+        points: Iterable[Tuple[float, float]],
+        attribs: PolylineAttribs | None = None,
+    ):
+        polyline = self.model_space.add_lwpolyline(points, close=attribs.close)
+        if attribs != None:
+            polyline.dxf.layer = attribs.layer
+
+        max_x = max([pt[0] for pt in points])
+        max_y = max([pt[1] for pt in points])
+        min_x = min([pt[0] for pt in points])
+        min_y = min([pt[1] for pt in points])
+        self.__update_boundary(max_x, max_y)
+        self.__update_boundary(min_x, min_y)
+
+    def _add_circle(
+        self,
+        center_point: Iterable[float],
+        radius: float,
+        attribs: DrawingAttribs | None = None,
+    ):
+        circle = self.model_space.add_circle(center_point, radius)
+        if attribs != None:
+            circle.dxf.layer = attribs.layer
+
+    def _add_dimension(
+        self,
+        start_point: Iterable[float],
+        end_point: Iterable[float],
+        attribs: DrawingAttribs | None = None,
+    ):
+        dimension = self.model_space.add_linear_dim(
+            (0, 0), start_point, end_point, dimstyle="CT-100"
+        ).render()
+
+        if attribs != None:
+            # 这里.dimension才是Dimension对象，才有dxf属性
+            dimension.dimension.dxf.layer = attribs.layer
+
+    def _add_text(self, context: str, insert_point, attribs: TextAttribs):
+        text = self.model_space.add_text(context, height=attribs.text_height)
+        text.dxf.style = BasicDXF.TEXT_STYLE_NAME
+        # width就是宽度因子系数
+        text.dxf.width = attribs.text_width_factor
+        text.dxf.layer = attribs.layer
+        text.dxf.color = attribs.color_index
+        text.set_placement(insert_point, align=attribs.text_align)
+        max_x = insert_point[0]
+        min_x = insert_point[0]
+        max_y = insert_point[1] + attribs.text_height
+        min_y = insert_point[1] - attribs.text_height
+        self.__update_boundary(max_x, max_y)
+        self.__update_boundary(min_x, min_y)
+
+    def _save(self, path: str):
+        self.__change_view()
+        if not path.endswith(BasicDXF.file_extension):
+            path += BasicDXF.file_extension
+        for _ in range(10):
+            try:
+                self.doc.saveas(path)
+            except Exception:
+                path = path.replace(
+                    BasicDXF.file_extension, "1" + BasicDXF.file_extension
+                )
+
+    def __change_view(self):
+        if inf in self.__max_point or -inf in self.__min_point:
+            return
+
+        y_range = self.__max_point[1] - self.__min_point[1]
+        x_range = self.__max_point[0] - self.__min_point[0]
+
+        y_middle = (self.__max_point[1] + self.__min_point[1]) / 2
+        # 为了使得内容靠右些，避免左侧panel占位的视觉影响
+        x_middle = (self.__max_point[0] + self.__min_point[0] * 3) / 4
+        # 乘以1.1的系数，增加了一些margin
+        self.doc.set_modelspace_vport(
+            max(x_range * 1.1, y_range * 1.1), (x_middle, y_middle)
+        )
+
+    def __update_boundary(self, x, y):
+        temp_x1 = self.__min_point[0]
+        temp_y1 = self.__min_point[1]
+        self.__min_point = [min(temp_x1, x), min(temp_y1, y)]
+
+        temp_x2 = self.__max_point[0]
+        temp_y2 = self.__max_point[1]
+        self.__max_point = [max(temp_x2, x), max(temp_y2, y)]
+
+    def __load_line_type(self, name: str, pattern: List[float]):
+        if name in self.__loaded_line_types:
+            return
+        self.doc.linetypes.add(name, pattern)
+        self.__loaded_line_types.append(name)
+
+    def __load_dim_type(self, scale: int):
+        # 获取标注样式表
+        dimstyles = self.doc.dimstyles
+        # 定义新标注样式的名称
+        new_dimstyle_name = f"{BasicDXF.DIM_STYLE_PREFIX}-{scale}"
+        # 创建新的标注样式
+        new_dimstyle = dimstyles.new(new_dimstyle_name)
+        # 设置标注文字样式
+        new_dimstyle.dxf.dimtxsty = BasicDXF.TEXT_STYLE_NAME
+        # 设置标注文字高度
+        new_dimstyle.dxf.dimtxt = 2.5
+        # 设置箭头大小
+        new_dimstyle.dxf.dimasz = 1
+        # 设置标注线超出尺寸界线的长度
+        new_dimstyle.dxf.dimexo = 0.625
+        # 设置尺寸界线起点偏移量
+        new_dimstyle.dxf.dimse1 = 0.625
+        #  设置全局比例
+        new_dimstyle.dxf.dimscale = scale
+        # 保留至整数
+        new_dimstyle.dxf.dimrnd = 1
+        # 改为建筑箭头
+        new_dimstyle.dxf.dimblk = "ARCHTICK"
+        new_dimstyle.dxf.dimblk1 = "ARCHTICK"
+        new_dimstyle.dxf.dimblk2 = "ARCHTICK"
+        self.doc.header["$DIMSTYLE"] = new_dimstyle_name
+
+    def __load_text_type(self):
+
+        text_styles = self.doc.styles
+        # 定义新文字样式的名称
+        new_style_name = BasicDXF.TEXT_STYLE_NAME
+        # 创建新的文字样式
+        new_style = text_styles.new(new_style_name)
+        # 设置文字样式的属性
+        # 指定字体文件，例如 Arial 字体
+        new_style.dxf.font = BasicDXF.TEXT_SHX
+        new_style.dxf.bigfont = BasicDXF.TEXT_BIG_SHX
+        # 设置文字高度
+        new_style.dxf.height = 0
+        new_style.dxf.width = 0.7
+        # 修改当前选择的文字样式
+        self.doc.header["$TEXTSTYLE"] = new_style_name
+
+    def _remove_all_entities(self):
+        self.model_space.delete_all_entities()
+        self.__min_point = [inf, inf]
+        self.__max_point = [-inf, -inf]
+
+    def _repr_png_(self):
+        """可以在jupyter中实时显示绘制的图像情况"""
+        ctx = RenderContext(self.doc)
+        fig, ax = plt.subplots()
+        fig.patch.set_facecolor("black")
+        ax.patch.set_facecolor("black")
+        plt.axis("off")
+        backend = MatplotlibBackend(ax)
+        Frontend(ctx, backend).draw_layout(self.model_space)
+        plt.show()