civil-tools-v 0.0.2__tar.gz → 0.0.3__tar.gz

civil_tools_v-0.0.3/CivilTools/FigureGenerator/BasicPltPlotter.py

@@ -0,0 +1,117 @@
+from typing import List
+import matplotlib.pyplot as plt
+from matplotlib.axes import Axes
+import io
+import numpy as np
+
+plt.rcParams["font.sans-serif"] = ["SimHei"]  # 用来正常显示中文标签
+plt.rcParams["axes.unicode_minus"] = False
+plt.rcParams["font.size"] = 14
+
+
+def GetTicks(n_max=1):
+    if n_max == 0:
+        n_max = 100
+    m = int(np.log10(n_max))
+    if m <= 0 and n_max <= 1:
+        m -= 1
+    p = n_max / (10**m)
+    if p <= 2.2:
+        result = [i * 1 * 10**m for i in range(3)]
+    elif p <= 3.5:
+        result = [i * 2 * 10**m for i in range(3)]
+    elif p <= 5:
+        result = [i * 2.5 * 10**m for i in range(3)]
+    elif p <= 6.9:
+        result = [i * 3 * 10**m for i in range(4)]
+    elif p <= 8:
+        result = [i * 2 * 10**m for i in range(5)]
+    else:
+        result = [i * 2 * 10**m for i in range(6)]
+    return result
+
+
+class BasicPltPlotter:
+    def __init__(self, fig_num: int = 1, fig_size=(10, 10)):
+        self.fig, _axes = plt.subplots(1, fig_num, figsize=fig_size)
+        if fig_num == 1:
+            self.axes = [_axes]
+        else:
+            self.axes = _axes
+        self.fig.patch.set_facecolor("none")
+        for ax in self.axes:
+            ax.patch.set_facecolor("none")
+        self.axes: List[Axes]
+        self.__remove_border()
+
+    def __remove_border(self):
+        for ax in self.axes:
+            ax.spines["right"].set_color("none")
+            ax.spines["top"].set_color("none")
+
+    def save_to_stream(self):
+        # 将图片保存到内存中的 BytesIO 对象
+        img_buffer = io.BytesIO()
+        self.fig.savefig(img_buffer, format="png", dpi=150)  # 保存为 PNG 格式
+        plt.close()  # 关闭图形，释放内存
+        # 将指针重置到流的开头，以便后续读取
+        img_buffer.seek(0)
+        return img_buffer
+
+
+class SeismicPlotter(BasicPltPlotter):
+    def __init__(self, fig_num=2, floor_num: int = 8):
+        if fig_num != 1 and fig_num != 2:
+            raise ValueError("Only 1 or 2 is accepted for fig_num.")
+        if fig_num == 1:
+            fig_size = (3, 5)
+        else:
+            fig_size = (6, 5)
+        super().__init__(fig_num, fig_size)
+        self.kwargs_x = {
+            "label": "X",
+            "ls": "-",
+            "color": "k",
+            "marker": "o",
+            "ms": 3,
+        }
+        self.kwargs_y = {
+            "label": "X",
+            "ls": "-",
+            "color": "r",
+            "marker": "o",
+            "ms": 3,
+        }
+        self.floor_num = floor_num
+        self.y_label = "层号"
+        self._y_values = [i + 1 for i in range(self.floor_num)]
+        self._y_major_ticks = self.__create_y_ticks()
+        self._y_minor_ticks = [i + 1 for i in range(self.floor_num)]
+        self._ax1_x = [i for i in range(self.floor_num)]
+        self._ax1_y = [i * 0.5 for i in range(self.floor_num)]
+        self._ax2_x = [i for i in range(self.floor_num)]
+        self._ax2_y = [i * 0.5 for i in range(self.floor_num)]
+
+    def test_plot(self):
+        self.__plot()
+
+    def __plot(self):
+        self.axes[0].plot(self._ax1_x, self._y_values, **self.kwargs_x)
+        self.axes[0].plot(self._ax1_y, self._y_values, **self.kwargs_y)
+        self.axes[1].plot(self._ax2_x, self._y_values, **self.kwargs_x)
+        self.axes[1].plot(self._ax2_y, self._y_values, **self.kwargs_y)
+
+    def __create_y_ticks(self):
+        floor_num = self.floor_num
+        return range(0, int(floor_num) + 1, int(floor_num // 5) + 1)
+
+    def _validate_list_length(self, data_list, name):
+        """
+        验证列表长度是否与楼层数相等
+        :param data_list: 待验证的列表
+        :param name: 列表的名称，用于异常信息
+        """
+        if len(data_list) != self.floor_num:
+            raise ValueError(
+                f"Length of {name} is not equal to floor number: {self.floor_num}!"
+            )

civil_tools_v-0.0.3/CivilTools/FigureGenerator/SeismicReport/ShearMassRatio.py

@@ -0,0 +1,73 @@
+from ..BasicPltPlotter import SeismicPlotter, GetTicks
+from typing import List
+import numpy as np
+import matplotlib.pyplot as plt
+
+
+class ShearMassRatioPlotter(SeismicPlotter):
+    def __init__(self, fig_num=2, floor_num=8):
+        super().__init__(fig_num, floor_num)
+        self.__limit = None
+        self.type = "剪重比"
+
+    def set_data(self, shear_x: List[float], shear_y: List[float], mass: List[float]):
+        if len(shear_x) != self.floor_num:
+            raise ValueError(
+                f"Lenght of shear_x is not equal to floor number: {self.floor_num}!"
+            )
+        if len(shear_y) != self.floor_num:
+            raise ValueError(
+                f"Lenght of shear_y is not equal to floor number: {self.floor_num}!"
+            )
+        if len(mass) != self.floor_num:
+            raise ValueError(
+                f"Lenght of mass is not equal to floor number: {self.floor_num}!"
+            )
+
+        self._ax1_x = np.array(shear_x) / np.array(mass)
+        self._ax2_x = np.array(shear_y) / np.array(mass)
+
+    def set_limit(self, limit: float):
+        self.__limit = limit
+
+    def plot(self):
+        if self.__limit:
+            self.__plot_limit()
+        kwargs_x = self.kwargs_x.copy()
+        kwargs_x["label"] = "X"
+        kwargs_y = self.kwargs_x.copy()
+        kwargs_y["label"] = "Y"
+        self.axes[0].plot(self._ax1_x, self._y_values, **kwargs_x)
+        self.axes[1].plot(self._ax2_x, self._y_values, **kwargs_y)
+        self.__adjust_lim()
+        self.__add_titles()
+
+    def __plot_limit(self):
+        limitation = self.__limit
+        for ax in self.axes:
+            ax.vlines(
+                x=limitation,
+                ymin=0,
+                ymax=self.floor_num,
+                color="r",
+                linewidth=3,
+                ls="--",
+                label=f"限值{limitation*100:.1f}%",
+            )
+
+    def __adjust_lim(self):
+        xmaxs = [self._ax1_x.max(), self._ax2_x.max()]
+        for i in range(2):
+            self.axes[i].set_xlim(left=0, right=xmaxs[i] * 1.2)
+            self.axes[i].set_yticks(self._y_major_ticks)
+            self.axes[i].set_yticks(self._y_minor_ticks, minor=True)
+            x_ticks = GetTicks(xmaxs[i])
+            self.axes[i].set_xticks(x_ticks)
+            self.axes[i].set_xticklabels([f"{i*100:.1f}%" for i in x_ticks])
+
+    def __add_titles(self):
+        self.axes[0].set_ylabel(self.y_label)
+        self.axes[0].set_xlabel(f"X小震下{self.type}")
+        self.axes[1].set_xlabel(f"Y小震下{self.type}")
+        self.axes[0].legend(framealpha=0, fontsize=12, loc=4)
+        self.axes[1].legend(framealpha=0, fontsize=12, loc=4)

civil_tools_v-0.0.3/CivilTools/FigureGenerator/SeismicReport/ShearMoment.py

@@ -0,0 +1,71 @@
+from ..BasicPltPlotter import SeismicPlotter, GetTicks
+from typing import List
+import numpy as np
+import matplotlib.pyplot as plt
+
+
+class ShearMomentPlotter(SeismicPlotter):
+    def __init__(self, floor_num=8, type="Shear", fig_num=2):
+        super().__init__(fig_num, floor_num)
+        self.__limit = None
+        if type.lower() == "shear":
+            self.type = "楼层剪力 kN"
+        elif type.lower() == "moment":
+            self.type = "倾覆力矩 MN·m"
+        else:
+            raise ValueError("Only shear or moment is supported for type.")
+
+    def set_data(
+        self,
+        wind_x: List[float],
+        wind_y: List[float],
+        seismic_x: List[float],
+        seismic_y: List[float],
+    ):
+        # 验证数据长度
+        data_dict = {
+            "wind_x": wind_x,
+            "wind_y": wind_y,
+            "seismic_x": seismic_x,
+            "seismic_y": seismic_y,
+        }
+        for name, data_list in data_dict.items():
+            self._validate_list_length(data_list, name)
+
+        self._ax1_x = np.array(wind_x)
+        self._ax1_y = np.array(wind_y)
+        self._ax2_x = np.array(seismic_x)
+        self._ax2_y = np.array(seismic_y)
+
+    def plot(self):
+        if self.__limit:
+            self.__plot_limit()
+        kwargs_x = self.kwargs_x.copy()
+        kwargs_x["label"] = "X风"
+        kwargs_y = self.kwargs_y.copy()
+        kwargs_y["label"] = "Y风"
+        self.axes[0].plot(self._ax1_x, self._y_values, **kwargs_x)
+        self.axes[0].plot(self._ax1_y, self._y_values, **kwargs_y)
+        kwargs_x["label"] = "X小震"
+        kwargs_y["label"] = "Y小震"
+        self.axes[1].plot(self._ax2_x, self._y_values, **kwargs_x)
+        self.axes[1].plot(self._ax2_y, self._y_values, **kwargs_y)
+        self.__adjust_lim()
+        self.__add_titles()
+
+    def __adjust_lim(self):
+        xmaxs = [self._ax1_x.max(), self._ax2_x.max()]
+        for i in range(2):
+            self.axes[i].set_xlim(left=0, right=xmaxs[i] * 1.2)
+            self.axes[i].set_yticks(self._y_major_ticks)
+            self.axes[i].set_yticks(self._y_minor_ticks, minor=True)
+            x_ticks = GetTicks(xmaxs[i])
+            self.axes[i].set_xticks(x_ticks)
+            # self.axes[i].set_xticklabels([f"{i*100:.1f}%" for i in x_ticks])
+
+    def __add_titles(self):
+        self.axes[0].set_ylabel(self.y_label)
+        self.axes[0].set_xlabel(f"风下{self.type}")
+        self.axes[1].set_xlabel(f"小震下{self.type}")
+        self.axes[0].legend(framealpha=0, fontsize=12, loc=1)
+        self.axes[1].legend(framealpha=0, fontsize=12, loc=1)

civil_tools_v-0.0.3/CivilTools/FigureGenerator/SeismicReport/__init__.py

@@ -0,0 +1,2 @@
+from .ShearMassRatio import ShearMassRatioPlotter
+from .ShearMoment import ShearMomentPlotter

{civil_tools_v-0.0.2 → civil_tools_v-0.0.3}/CivilTools/YDBLoader/BuildingDefine/GlobalResult/BasicResult.py

@@ -12,17 +12,25 @@ class SingleMassResult:
     ):
         self.floor_num = floor_num
         self.tower_num = tower_num
-        self.dead_load = dead_load
+        self.dead_load = round(dead_load, 4)
         """单层恒载，单位kN，没有折减"""
-        self.live_load = live_load
+        self.live_load = round(live_load, 4) * 2
         """单层活载，单位kN，没有折减"""
-        self.slab_area = slab_area
+        self.slab_area = round(slab_area)
         """单层楼板面积，单位m2"""
 
     @property
     def total_load(self):
         """单层质量，恒+0.5活"""
-        return self.dead_load + 0.5 * self.live_load
+        return round(self.dead_load + 0.5 * self.live_load, 4)
+
+    def to_json(self):
+        return {
+            "floor_num": self.floor_num,
+            "tower_num": self.tower_num,
+            "dead_load": self.dead_load,
+            "live_load": self.live_load,
+        }
 
 
 class MassResult:
@@ -52,3 +60,6 @@ class MassResult:
             mass_list.append(SingleMassResult(i + 1, 1, 5000, 2000, 350))
 
         return MassResult(mass_list)
+
+    def to_json(self):
+        return {"mass_list": [i.to_json() for i in self.mass_list]}

{civil_tools_v-0.0.2 → civil_tools_v-0.0.3}/CivilTools/YDBLoader/BuildingDefine/GlobalResult/SeismicResult.py

@@ -64,6 +64,9 @@ class SinglePeriod:
     def __repr__(self):
         return str(self)
 
+    def to_json(self):
+        return {"index": self.index, "time": self.time}
+
 
 class Period:
     def __init__(self, periods: List[SinglePeriod], model_type=None):
@@ -144,6 +147,9 @@ class Period:
             )
         return Period(single_period_list)
 
+    def to_json(self):
+        return {"periods": [i.to_json() for i in self.periods]}
+
 
 class ValuePeer:
     def __init__(self, x: float, y: float):
@@ -158,6 +164,9 @@ class ValuePeer:
         else:
             return f"X:{self.x:.3f}\tY:{self.y:.3f}"
 
+    def to_json(self):
+        return {"X": self.x, "Y": self.y}
+
 
 class FloorSeismicResult:
     def __init__(
@@ -186,6 +195,13 @@ class FloorSeismicResult:
     def __repr__(self):
         return self.__str__()
 
+    def to_json(self):
+        return {
+            "floor_num": self.floor_num,
+            "tower_num": self.tower_num,
+            "shear": self.shear.to_json(),
+        }
+
 
 class SeismicResult:
     def __init__(self, floor_result: List[FloorSeismicResult]):
@@ -214,6 +230,9 @@ class SeismicResult:
     def __repr__(self):
         return self.__str__()
 
+    def to_json(self):
+        return {"floor_result": [i.to_json() for i in self.floor_result]}
+
 
 if __name__ == "__main__":
     p_list = []

{civil_tools_v-0.0.2 → civil_tools_v-0.0.3}/CivilTools/YDBLoader/YDBLoader.py

@@ -153,8 +153,8 @@ class YDBLoader:
         row_data = self.connector.extract_table_by_columns(table_name, useful_columns)
         mass_list = []
         for temp_mass in row_data:
-            floor_num = RowDataFactory.convert_to_int(temp_mass, 0)
-            tower_num = RowDataFactory.convert_to_int(temp_mass, 1)
+            floor_num = RowDataFactory.convert_to_int(temp_mass[0])
+            tower_num = RowDataFactory.convert_to_int(temp_mass[1])
             mass_info = RowDataFactory.extract_list(temp_mass, 2)
             dead_load = RowDataFactory.convert_to_float(mass_info[1])
             live_load = RowDataFactory.convert_to_float(mass_info[2])
@@ -225,7 +225,6 @@ class YDBLoader:
             force = ValuePeer(force_x, force_y)
             shear = ValuePeer(shear_x, shear_y)
             moment = ValuePeer(moment_x, moment_y)
-
             temp_floor_result = FloorSeismicResult(
                 floor_num, tower_num, force, shear, moment
             )

{civil_tools_v-0.0.2/civil_tools_v.egg-info → civil_tools_v-0.0.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: civil_tools_v
-Version: 0.0.2
+Version: 0.0.3
 Summary: A powerful tool for civil engineer in their work.
 Home-page: https://github.com/VincentXGao/civil-tools
 Author: Xinyu Gao (Vincent)

{civil_tools_v-0.0.2 → civil_tools_v-0.0.3/civil_tools_v.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: civil_tools_v
-Version: 0.0.2
+Version: 0.0.3
 Summary: A powerful tool for civil engineer in their work.
 Home-page: https://github.com/VincentXGao/civil-tools
 Author: Xinyu Gao (Vincent)

{civil_tools_v-0.0.2 → civil_tools_v-0.0.3}/civil_tools_v.egg-info/SOURCES.txt

@@ -15,6 +15,9 @@ CivilTools/FigureGenerator/BasicPNGPlotter.py
 CivilTools/FigureGenerator/BasicPltPlotter.py
 CivilTools/FigureGenerator/StairCalculationSheetPNGPlotter.py
 CivilTools/FigureGenerator/__init__.py
+CivilTools/FigureGenerator/SeismicReport/ShearMassRatio.py
+CivilTools/FigureGenerator/SeismicReport/ShearMoment.py
+CivilTools/FigureGenerator/SeismicReport/__init__.py
 CivilTools/ReportGenerator/BasicGenerator.py
 CivilTools/ReportGenerator/DocParagraph.py
 CivilTools/ReportGenerator/DocPicture.py

{civil_tools_v-0.0.2 → civil_tools_v-0.0.3}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="civil_tools_v",  # 包的名称
-    version="0.0.2",  # 包的版本号
+    version="0.0.3",  # 包的版本号
     author="Xinyu Gao (Vincent)",  # 作者姓名
     author_email="just_gxy@163.com",  # 作者邮箱
     description="A powerful tool for civil engineer in their work.",  # 包的简短描述

civil_tools_v-0.0.2/CivilTools/FigureGenerator/BasicPltPlotter.py

@@ -1,58 +0,0 @@
-from typing import List
-import matplotlib.pyplot as plt
-from matplotlib.axes import Axes
-import io
-
-
-class BasicPltPlotter:
-    def __init__(self, fig_num: int = 1, fig_size=(10, 10)):
-        self.fig, _axes = plt.subplots(1, fig_num, figsize=fig_size)
-        if fig_num == 1:
-            self.axes = [_axes]
-        else:
-            self.axes = _axes
-        self.axes: List[Axes]
-        self.__remove_border()
-
-    def __remove_border(self):
-        for ax in self.axes:
-            ax.spines["right"].set_color("none")
-            ax.spines["top"].set_color("none")
-
-    def save_to_stream(self):
-        # 将图片保存到内存中的 BytesIO 对象
-        img_buffer = io.BytesIO()
-        self.fig.savefig(img_buffer, format="png", dpi=300)  # 保存为 PNG 格式
-        plt.close()  # 关闭图形，释放内存
-        # 将指针重置到流的开头，以便后续读取
-        img_buffer.seek(0)
-        return img_buffer
-
-
-class SeismicPlotter(BasicPltPlotter):
-    def __init__(self, fig_num=2, floor_num: int = 8):
-        if fig_num != 1 and fig_num != 2:
-            raise ValueError("Only 1 or 2 is accepted for fig_num.")
-        if fig_num == 1:
-            fig_size = (3, 5)
-        else:
-            fig_size = (6, 5)
-        super().__init__(fig_num, fig_size)
-        self.floor_num = floor_num
-        self.__y_values = [i + 1 for i in range(self.floor_num)]
-        self.__ax1_x = [i for i in range(self.floor_num)]
-        self.__ax1_y = [i * 0.5 for i in range(self.floor_num)]
-        self.__ax2_x = [i for i in range(self.floor_num)]
-        self.__ax2_y = [i * 0.5 for i in range(self.floor_num)]
-
-    def test_plot(self):
-        self.__plot()
-
-    def __plot(self):
-        kwargs_x = {"label": "X", "ls": "-", "color": "k", "marker": "o", "ms": 3}
-        kwargs_y = {"label": "X", "ls": "-", "color": "r", "marker": "o", "ms": 3}
-
-        self.axes[0].plot(self.__ax1_x, self.__y_values, **kwargs_x)
-        self.axes[0].plot(self.__ax1_y, self.__y_values, **kwargs_y)
-        self.axes[1].plot(self.__ax2_x, self.__y_values, **kwargs_x)
-        self.axes[1].plot(self.__ax2_y, self.__y_values, **kwargs_y)