481 lines
21 KiB
Python
481 lines
21 KiB
Python
from tkinter.messagebox import NO
|
||
import xlwings as xw
|
||
from PWFile import SFile, Fitting,ColorEnume
|
||
from Apyautocad import Apyautocad, APoint
|
||
import os
|
||
import re
|
||
import numpy as np
|
||
from time import sleep
|
||
from attrs import define
|
||
from typing import List
|
||
|
||
# 读取Z文件，找到Z断面第一个点的坐标
|
||
def plane_z_origin(z_file_path):
|
||
with open(z_file_path) as zfile:
|
||
content = zfile.read()
|
||
norm_content = re.sub("\s+", ",", content)
|
||
sep = norm_content.split(",")
|
||
return np.array([float(sep[0]), float(sep[1])])
|
||
|
||
|
||
def deduce_zfile_from_cad_path(cad_file_path):
|
||
dwg_file_name = os.path.split(cad_file_path)
|
||
dwg_prefix = dwg_file_name[1].split(".")[0]
|
||
return os.path.join(dwg_file_name[0], f"Z{dwg_prefix}")
|
||
|
||
|
||
def deduce_fit_db_from_cad_path(cad_file_path):
|
||
dwg_file_name = os.path.split(cad_file_path)
|
||
dwg_prefix = dwg_file_name[1].split(".")[0]
|
||
return os.path.join(dwg_file_name[0], "Fit.db")
|
||
|
||
|
||
def curve_fun(x, span, k, gaocha):
|
||
return x * gaocha / span - x * (span - x) * k
|
||
|
||
|
||
def np2d_to_array(np2d): # 把2维numpy数组转换成cad可以用的数组
|
||
t = np.hstack((np2d, np.zeros((np2d.shape[0], 1)))).reshape(1, np2d.shape[0] * 3)
|
||
return t[0]
|
||
|
||
|
||
class StringImpactExcel:
|
||
def __init__(self) -> None:
|
||
# self._wb=None
|
||
pass
|
||
|
||
def read(self, wb, gaocha, span, tension):
|
||
pos代表档距 = "F13"
|
||
pos档距 = "L13"
|
||
pos高差 = "R13"
|
||
pos张力 = "AB18"
|
||
pos总串长 = "C8"
|
||
sheet = wb.sheets["3225-3226"]
|
||
sheet.range(pos高差).value = gaocha
|
||
sheet.range(pos档距).value = span
|
||
sheet.range(pos张力).value = tension
|
||
string_length = sheet.range(pos总串长).value
|
||
# print(sheet.range("V25:V46").value)
|
||
x = np.linspace(string_length, span, int(span / 5), endpoint=True)
|
||
x[0] = sheet.range("E23").value
|
||
x[1]=sheet.range("E24").value
|
||
sheet.range(f"E25:E{25+len(x)-3}").value = x[2:].reshape(len(x[2:]), 1)
|
||
y = np.array(sheet.range(f"V23:V{23+len(x)-1}").value)/2#表格是乘以了2的，为了和x保持一致，没有乘比例。
|
||
return (x, y)
|
||
|
||
|
||
def set_true_color(object, r, g, b):
|
||
true_color = object.TrueColor
|
||
true_color.SetRGB(r, g, b)
|
||
object.TrueColor = true_color
|
||
|
||
|
||
@define
|
||
class StringImpactPlate:
|
||
_dwg_file_path: str
|
||
_s_file_path: str
|
||
_draw_tower_name: List[str]
|
||
# _tension_section:int#耐张段数量
|
||
_continouse_tension_excel: str
|
||
_string_impact_curve_excel: str
|
||
|
||
def _find_target_towe_index(self, tower_name_list: List[str]):
|
||
index = []
|
||
for foo in self._draw_tower_name:
|
||
index.append(tower_name_list.index(foo))
|
||
return index
|
||
|
||
def draw(self):
|
||
# 计算代表档距
|
||
s_file = SFile()
|
||
s_file.open(self._s_file_path)
|
||
tower_dict = s_file.tower_dic
|
||
tower_key_list = list(tower_dict.keys())
|
||
draw_tower_index = self._find_target_towe_index(tower_key_list)
|
||
fitting_file_path = deduce_fit_db_from_cad_path(self._dwg_file_path)
|
||
fitting = Fitting(fitting_file_path)
|
||
z_file_path = deduce_zfile_from_cad_path(self._dwg_file_path)
|
||
plate_origin = plane_z_origin(z_file_path)
|
||
with xw.App(visible=True) as excel_app, Apyautocad(
|
||
create_if_not_exists=True, visible=True, auto_close=False
|
||
) as cad:
|
||
continouse_wb = excel_app.books.open("张力计算(临界档距公式法V20090602)-送电室版).xls")
|
||
continouse_sheet = continouse_wb.sheets["2710导线-单回1250-70"]
|
||
wb_string_impact = excel_app.books.open(
|
||
"特高压耐张串影响弧垂计算（送电室2018人员版）1122-1218.xls"
|
||
)
|
||
stringImpactExcel = StringImpactExcel()
|
||
cad.app.Documents.Open(self._dwg_file_path)
|
||
sleep(1)
|
||
for draw_tower_foo in draw_tower_index:
|
||
draw_tower_key = tower_key_list[draw_tower_foo]
|
||
tower_info = tower_dict[draw_tower_key]
|
||
forth_reprtv_span = tower_info.forth_representive_span
|
||
continouse_sheet.range("B69").value = forth_reprtv_span
|
||
high_temperature_tension = continouse_sheet.range("L69").value
|
||
forth_tower_info = tower_dict[tower_key_list[draw_tower_foo + 1]]
|
||
gaocha = (
|
||
(
|
||
forth_tower_info.tower_height
|
||
- forth_tower_info.foundation_low
|
||
- fitting.fitting_length_dic[forth_tower_info.fitting]
|
||
)
|
||
- (
|
||
tower_info.tower_height
|
||
- tower_info.foundation_low
|
||
- fitting.fitting_length_dic[tower_info.fitting]
|
||
)
|
||
+ forth_tower_info.altitude_off
|
||
)
|
||
span = forth_tower_info.mileage_in_s - tower_info.mileage_in_s
|
||
(x, y) = stringImpactExcel.read(
|
||
wb_string_impact, gaocha, span, high_temperature_tension
|
||
)
|
||
# TODO: 计算累计里程和高差
|
||
plot_x = (plate_origin[0] + x / 5).reshape(len(x), 1)
|
||
plot_y = (plate_origin[1] + (tower_info.tower_height-tower_info.foundation_low+ y)*2 ).reshape(len(x), 1)
|
||
plot_vector = np2d_to_array(np.hstack((plot_x, plot_y)))
|
||
added_curve=cad.model.AddPolyLine(plot_vector)
|
||
set_true_color(added_curve,*ColorEnume.wire_color_rgb)
|
||
plot_ground_y=plot_y-18*2
|
||
plot_ground_vector=np2d_to_array(np.hstack((plot_x, plot_ground_y)))
|
||
added_ground_curve = cad.model.AddPolyLine(plot_ground_vector)
|
||
set_true_color(added_ground_curve, *ColorEnume.ground_color_rgb)
|
||
plot_tree_y = plot_y - 13.5 * 2
|
||
plot_tree_vector = np2d_to_array(np.hstack((plot_x, plot_tree_y)))
|
||
added_tree_curve = cad.model.AddPolyLine(plot_tree_vector)
|
||
set_true_color(added_tree_curve, *ColorEnume.tree_color_rgb)
|
||
pass
|
||
|
||
# can_start=False
|
||
# tower_dict_iterator=iter(list(tower_dict.keys()))
|
||
# for tower_info_key in tower_dict_iterator:
|
||
# tower_info=tower_dict[tower_info_key]
|
||
# if tower_info.tower_name==self._draw_tower_name:
|
||
# if tower_info.is_tension_tower==False:
|
||
# raise Exception('耐张段影响弧垂第1基塔不是耐张塔！')
|
||
# can_start=True
|
||
# if can_start:
|
||
|
||
|
||
@define
|
||
class ContinuousPlate:
|
||
_dwg_file_path: str
|
||
_s_file_path: str
|
||
_from_tower_name: str
|
||
_draw_tower_count: int
|
||
|
||
def draw(self, save_to: str):
|
||
s_file = SFile()
|
||
s_file_path = self._s_file_path
|
||
s_file.open(s_file_path)
|
||
# with xw.App(visible=False) as app:
|
||
# wb_string_impact = app.books.open("特高压耐张串影响弧垂计算（送电室2018人员版）1122-1218.xls")
|
||
# stringImpactExcel = StringImpactExcel()
|
||
# stringImpactExcel.read(wb_string_impact)
|
||
dwg_file_path = self._dwg_file_path
|
||
# with xw.App(visible=False) as app:
|
||
# wb = app.books.open('张力计算(临界档距公式法V20090602)-送电室版).xls')
|
||
# sheet = wb.sheets['2710导线-单回1250-70']
|
||
# sheet.range('B69').value=120
|
||
# print(sheet.range('L69').value)
|
||
with Apyautocad(
|
||
create_if_not_exists=True, visible=False, auto_close=False
|
||
) as cad:
|
||
doc = cad.app.Documents.Open(dwg_file_path)
|
||
sleep(1)
|
||
z_file_path = deduce_zfile_from_cad_path(dwg_file_path)
|
||
z_point = plane_z_origin(z_file_path)
|
||
fitting_file_path = deduce_fit_db_from_cad_path(dwg_file_path)
|
||
fitting = Fitting(fitting_file_path)
|
||
fitting_length_dict = fitting.fitting_length_dic
|
||
first_tower_point = z_point
|
||
last_tower_info = None # 上一个塔位信息
|
||
accu_mileage = 0 # 累计档距
|
||
accu_altitude_off = 0 # 累计高差
|
||
is_first_tower = True
|
||
can_start_draw = -1
|
||
start_tower_name = self._from_tower_name
|
||
draw_count = 0
|
||
for tower in s_file.tower_dic:
|
||
tower_info = s_file.tower_dic[tower]
|
||
if tower_info.tower_name == start_tower_name:
|
||
can_start_draw = 0
|
||
if can_start_draw < 0 or draw_count > self._draw_tower_count - 1:
|
||
break
|
||
if not last_tower_info:
|
||
last_tower_info = tower_info
|
||
foundation_low = tower_info.foundation_low
|
||
accu_mileage = (
|
||
accu_mileage
|
||
+ tower_info.mileage_in_s
|
||
- last_tower_info.mileage_in_s
|
||
)
|
||
accu_altitude_off = accu_altitude_off + tower_info.altitude_off # 中心桩高程
|
||
tower_start = APoint(
|
||
*(
|
||
(
|
||
first_tower_point
|
||
+ np.array(
|
||
[
|
||
accu_mileage / 5,
|
||
(accu_altitude_off - foundation_low) * 2,
|
||
]
|
||
)
|
||
).tolist()
|
||
)
|
||
)
|
||
tower_height = tower_info.tower_height
|
||
tower_end = APoint(
|
||
*(
|
||
first_tower_point
|
||
+ np.array(
|
||
[
|
||
accu_mileage / 5,
|
||
(accu_altitude_off + tower_height - foundation_low) * 2,
|
||
]
|
||
)
|
||
).tolist()
|
||
)
|
||
# 画杆高
|
||
cad.model.AddLine(tower_start, tower_end)
|
||
draw_count += 1
|
||
# 画弧垂
|
||
if not is_first_tower: # 从第二基塔开始画
|
||
draw_k = tower_info.back_k
|
||
span = tower_info.mileage_in_s - last_tower_info.mileage_in_s
|
||
last_tower_fiting = last_tower_info.fitting
|
||
last_tower_fitting_length = fitting_length_dict[last_tower_fiting]
|
||
if last_tower_info.is_tension_tower:
|
||
last_tower_fitting_length = 0
|
||
tower_fitting = tower_info.fitting
|
||
tower_fitting_length = fitting_length_dict[tower_fitting]
|
||
if tower_info.is_tension_tower:
|
||
tower_fitting_length = 0
|
||
last_tower_height = last_tower_info.tower_height
|
||
last_foundation_low = last_tower_info.foundation_low
|
||
# 挂点高差
|
||
fiting_altitude_off = (
|
||
tower_info.altitude_off
|
||
+ (tower_height - foundation_low - tower_fitting_length)
|
||
- (
|
||
last_tower_height
|
||
- last_foundation_low
|
||
- last_tower_fitting_length
|
||
)
|
||
) # 前侧高为正
|
||
# 画导线弧垂
|
||
x = np.linspace(0, span, int(span), endpoint=True)
|
||
curve = curve_fun(x, span, draw_k, fiting_altitude_off)
|
||
draw_curve_x = (first_tower_point[0]) + (
|
||
x + accu_mileage - span
|
||
) / 5
|
||
draw_curve_y = (
|
||
first_tower_point[1]
|
||
+ (
|
||
+curve
|
||
+ accu_altitude_off
|
||
- tower_info.altitude_off
|
||
- last_tower_info.foundation_low
|
||
+ last_tower_info.tower_height
|
||
- last_tower_fitting_length
|
||
)
|
||
* 2
|
||
)
|
||
draw_curve_x = draw_curve_x.reshape(len(draw_curve_x), 1)
|
||
draw_curve_y = draw_curve_y.reshape(len(draw_curve_y), 1)
|
||
draw_ground_curve_y = draw_curve_y - 18 * 2 # 切地线
|
||
draw_tree_curve_y = draw_curve_y - 13.5 * 2 # 切树线
|
||
draw_point = np.hstack(
|
||
(draw_curve_x, draw_curve_y, np.zeros((len(draw_curve_x), 1)))
|
||
)
|
||
draw_ground_curve_point = np.hstack(
|
||
(
|
||
draw_curve_x,
|
||
draw_ground_curve_y,
|
||
np.zeros((len(draw_curve_x), 1)),
|
||
)
|
||
)
|
||
draw_tree_curve_point = np.hstack(
|
||
(
|
||
draw_curve_x,
|
||
draw_tree_curve_y,
|
||
np.zeros((len(draw_curve_x), 1)),
|
||
)
|
||
)
|
||
added_curve = cad.model.AddPolyLine(
|
||
draw_point.reshape(1, draw_curve_x.shape[0] * 3)[0]
|
||
)
|
||
set_true_color(added_curve, Color)
|
||
added_ground_curve = cad.model.AddPolyLine(
|
||
draw_ground_curve_point.reshape(
|
||
1, draw_ground_curve_y.shape[0] * 3
|
||
)[0]
|
||
)
|
||
set_true_color(added_ground_curve, 122, 219, 245)
|
||
added_tree_curve = cad.model.AddPolyLine(
|
||
draw_tree_curve_point.reshape(
|
||
1, draw_tree_curve_y.shape[0] * 3
|
||
)[0]
|
||
)
|
||
set_true_color(added_tree_curve, 116, 230, 165)
|
||
is_first_tower = False
|
||
last_tower_info = tower_info
|
||
doc.SaveAs(save_to)
|
||
|
||
|
||
def main1():
|
||
s_file = SFile()
|
||
s_file.open(r"d:\工程\金上线\排位\定位完排位\PW.0706\J33-J49\S033.DAT")
|
||
# with xw.App(visible=False) as app:
|
||
# wb_string_impact = app.books.open("特高压耐张串影响弧垂计算（送电室2018人员版）1122-1218.xls")
|
||
# stringImpactExcel = StringImpactExcel()
|
||
# stringImpactExcel.read(wb_string_impact)
|
||
with Apyautocad(create_if_not_exists=True, visible=False, auto_close=False) as cad:
|
||
doc = cad.app.Documents.Open(r"d:\工程\金上线\排位\定位完排位\PW.0706\J33-J49\033A.dwg")
|
||
sleep(1)
|
||
z_file_path = deduce_zfile_from_cad_path(
|
||
r"d:\工程\金上线\排位\定位完排位\PW.0706\J33-J49\033A.dwg"
|
||
)
|
||
z_point = plane_z_origin(z_file_path)
|
||
fitting_file_path = deduce_fit_db_from_cad_path(
|
||
r"d:\工程\金上线\排位\定位完排位\PW.0706\J33-J49\033A.dwg"
|
||
)
|
||
fitting = Fitting(fitting_file_path)
|
||
fitting_length_dict = fitting.fitting_length_dic
|
||
first_tower_point = z_point
|
||
last_tower_info = None # 上一个塔位信息
|
||
accu_mileage = 0 # 累计档距
|
||
accu_altitude_off = 0 # 累计高差
|
||
is_first_tower = True
|
||
for tower in s_file.tower_dic:
|
||
tower_info = s_file.tower_dic[tower]
|
||
if not last_tower_info:
|
||
last_tower_info = tower_info
|
||
foundation_low = tower_info.foundation_low
|
||
accu_mileage = (
|
||
accu_mileage + tower_info.mileage_in_s - last_tower_info.mileage_in_s
|
||
)
|
||
accu_altitude_off = accu_altitude_off + tower_info.altitude_off # 中心桩高程
|
||
tower_start = APoint(
|
||
*(
|
||
(
|
||
first_tower_point
|
||
+ np.array(
|
||
[accu_mileage / 5, (accu_altitude_off - foundation_low) * 2]
|
||
)
|
||
).tolist()
|
||
)
|
||
)
|
||
tower_height = tower_info.tower_height
|
||
tower_end = APoint(
|
||
*(
|
||
first_tower_point
|
||
+ np.array(
|
||
[
|
||
accu_mileage / 5,
|
||
(accu_altitude_off + tower_height - foundation_low) * 2,
|
||
]
|
||
)
|
||
).tolist()
|
||
)
|
||
# 画杆高
|
||
cad.model.AddLine(tower_start, tower_end)
|
||
# 画弧垂
|
||
if not is_first_tower: # 从第二基塔开始画
|
||
draw_k = tower_info.back_k
|
||
span = tower_info.mileage_in_s - last_tower_info.mileage_in_s
|
||
last_tower_fiting = last_tower_info.fitting
|
||
last_tower_fitting_length = fitting_length_dict[last_tower_fiting]
|
||
if last_tower_info.is_tension_tower:
|
||
last_tower_fitting_length = 0
|
||
tower_fitting = tower_info.fitting
|
||
tower_fitting_length = fitting_length_dict[tower_fitting]
|
||
if tower_info.is_tension_tower:
|
||
tower_fitting_length = 0
|
||
last_tower_height = last_tower_info.tower_height
|
||
last_foundation_low = last_tower_info.foundation_low
|
||
# 挂点高差
|
||
fiting_altitude_off = (
|
||
tower_info.altitude_off
|
||
+ (tower_height - foundation_low - tower_fitting_length)
|
||
- (
|
||
last_tower_height
|
||
- last_foundation_low
|
||
- last_tower_fitting_length
|
||
)
|
||
) # 前侧高为正
|
||
# 画导线弧垂
|
||
x = np.linspace(0, span, int(span), endpoint=True)
|
||
curve = curve_fun(x, span, draw_k, fiting_altitude_off)
|
||
draw_curve_x = (first_tower_point[0]) + (x + accu_mileage - span) / 5
|
||
draw_curve_y = (
|
||
first_tower_point[1]
|
||
+ (
|
||
+curve
|
||
+ accu_altitude_off
|
||
- tower_info.altitude_off
|
||
- last_tower_info.foundation_low
|
||
+ last_tower_info.tower_height
|
||
- last_tower_fitting_length
|
||
)
|
||
* 2
|
||
)
|
||
draw_curve_x = draw_curve_x.reshape(len(draw_curve_x), 1)
|
||
draw_curve_y = draw_curve_y.reshape(len(draw_curve_y), 1)
|
||
draw_ground_curve_y = draw_curve_y - 18 * 2 # 切地线
|
||
draw_tree_curve_y = draw_curve_y - 13.5 * 2 # 切树线
|
||
draw_point = np.hstack(
|
||
(draw_curve_x, draw_curve_y, np.zeros((len(draw_curve_x), 1)))
|
||
)
|
||
draw_ground_curve_point = np.hstack(
|
||
(
|
||
draw_curve_x,
|
||
draw_ground_curve_y,
|
||
np.zeros((len(draw_curve_x), 1)),
|
||
)
|
||
)
|
||
draw_tree_curve_point = np.hstack(
|
||
(draw_curve_x, draw_tree_curve_y, np.zeros((len(draw_curve_x), 1)))
|
||
)
|
||
added_curve = cad.model.AddPolyLine(
|
||
draw_point.reshape(1, draw_curve_x.shape[0] * 3)[0]
|
||
)
|
||
set_true_color(added_curve, *ColorEnume.wire_color_rgb)
|
||
added_ground_curve = cad.model.AddPolyLine(
|
||
draw_ground_curve_point.reshape(
|
||
1, draw_ground_curve_y.shape[0] * 3
|
||
)[0]
|
||
)
|
||
set_true_color(added_ground_curve, *ColorEnume.ground_color_rgb)
|
||
added_tree_curve = cad.model.AddPolyLine(
|
||
draw_tree_curve_point.reshape(1, draw_tree_curve_y.shape[0] * 3)[0]
|
||
)
|
||
set_true_color(added_tree_curve, *ColorEnume.tree_color_rgb)
|
||
is_first_tower = False
|
||
last_tower_info = tower_info
|
||
doc.SaveAs(r"d:\工程\金上线\code\考虑耐张串重弧垂\T033A.dwg")
|
||
|
||
|
||
def main():
|
||
# continousePlate = ContinuousPlate(
|
||
# r"d:\工程\金上线\排位\定位完排位\PW.0706\J33-J49\033A.dwg",
|
||
# r"d:\工程\金上线\排位\定位完排位\PW.0706\J33-J49\S033.DAT",
|
||
# "N3339",
|
||
# 7,
|
||
# )
|
||
# continousePlate.draw(r"d:\工程\金上线\code\考虑耐张串重弧垂\T033A.dwg")
|
||
string_impact_plate = StringImpactPlate(
|
||
r"d:\工程\金上线\排位\定位完排位\PW.0706\J33-J49\033A.dwg",
|
||
r"d:\工程\金上线\排位\定位完排位\PW.0706\J33-J49\S033.DAT",
|
||
["N3339"],
|
||
"",
|
||
"",
|
||
)
|
||
string_impact_plate.draw()
|
||
|
||
|
||
if __name__ == "__main__":
|
||
main()
|
||
print("Finished.")
|