370 lines
15 KiB
Python
370 lines
15 KiB
Python
from tkinter.messagebox import NO
|
||
import xlwings as xw
|
||
from PWFile import SFile, Fitting
|
||
from Apyautocad import Apyautocad, APoint
|
||
import os
|
||
import re
|
||
import numpy as np
|
||
from time import sleep
|
||
from attrs import define
|
||
|
||
# 读取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
|
||
|
||
|
||
class StringImpactExcel:
|
||
def __init__(self) -> None:
|
||
# self._wb=None
|
||
pass
|
||
|
||
def read(self, wb):
|
||
# with xw.App(visible=True) as app:
|
||
pos代表档距 = "F13"
|
||
pos档距 = "L13"
|
||
pos高差 = "R13"
|
||
pos张力 = "AB18"
|
||
pos档距21 = "E25"
|
||
sheet = wb.sheets["3225-3226"]
|
||
print(sheet.range("V25:V46").value)
|
||
|
||
|
||
def set_true_color(object, r, g, b):
|
||
true_color = object.TrueColor
|
||
true_color.SetRGB(r, g, b)
|
||
object.TrueColor = true_color
|
||
|
||
|
||
@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 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, 122, 219, 245)
|
||
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():
|
||
# 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)
|
||
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, 122, 219, 245)
|
||
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(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")
|
||
|
||
|
||
if __name__ == "__main__":
|
||
main()
|
||
print("Finished.")
|