facat
/
transsimision_line_section
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

准备解决中间有耐张段的情况。

This commit is contained in:
n3040 2023-01-26 02:03:23 +08:00
parent ce1ff065d1
commit 2109957df1
1 changed files with 47 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

52
PWFile.py
View File

@ -12,6 +12,17 @@ from time import sleep
from attrs import define from attrs import define
from typing import List from typing import List
import error import error
from array import array
def get_layer_if_not_exist_create_it(doc, layer_name):
layers = doc.Layers
for foo in range(layers.Count):
layer = layers.Item(foo)
if layer.Name == layer_name:
return layer
return layers.Add(layer_name)
@define @define
class SEntry: class SEntry:
@ -218,20 +229,25 @@ class StringImpactPlate:
return index return index
def _plot(self, cad, plot_x, plot_y): def _plot(self, cad, plot_x, plot_y):
custom_layer = get_layer_if_not_exist_create_it(cad.doc, "123custom_layer")
plot_vector = np2d_to_array(np.hstack((plot_x, plot_y))) plot_vector = np2d_to_array(np.hstack((plot_x, plot_y)))
added_curve = cad.model.AddPolyLine(plot_vector) added_curve = cad.model.AddPolyLine(plot_vector)
added_curve.Layer = custom_layer.Name
set_true_color(added_curve, *ColorEnume.wire_color_rgb) set_true_color(added_curve, *ColorEnume.wire_color_rgb)
plot_ground_y = plot_y - 18 * 2 # TODO 应该读规程的
plot_ground_y = plot_y - 16 * 2
plot_ground_vector = np2d_to_array(np.hstack((plot_x, plot_ground_y))) plot_ground_vector = np2d_to_array(np.hstack((plot_x, plot_ground_y)))
added_ground_curve = cad.model.AddPolyLine(plot_ground_vector) added_ground_curve = cad.model.AddPolyLine(plot_ground_vector)
added_ground_curve.Layer = custom_layer.Name
set_true_color(added_ground_curve, *ColorEnume.ground_color_rgb) set_true_color(added_ground_curve, *ColorEnume.ground_color_rgb)
plot_tree_y = plot_y - 13.5 * 2 plot_tree_y = plot_y - 13.5 * 2
plot_tree_vector = np2d_to_array(np.hstack((plot_x, plot_tree_y))) plot_tree_vector = np2d_to_array(np.hstack((plot_x, plot_tree_y)))
added_tree_curve = cad.model.AddPolyLine(plot_tree_vector) added_tree_curve = cad.model.AddPolyLine(plot_tree_vector)
added_tree_curve.Layer = custom_layer.Name
set_true_color(added_tree_curve, *ColorEnume.tree_color_rgb) set_true_color(added_tree_curve, *ColorEnume.tree_color_rgb)
def _draw_action(self, excel_app, cad): def _draw_action(self, excel_app, cad):
# 计算代表档距 # TODO 计算代表档距
s_file = SFile() s_file = SFile()
s_file.open(self._s_file_path) s_file.open(self._s_file_path)
tower_dict = s_file.tower_dic tower_dict = s_file.tower_dic
@ -390,6 +406,7 @@ class ContinuousPlate:
self.cad = cad self.cad = cad
doc = cad.app.Documents.Open(dwg_file_path) doc = cad.app.Documents.Open(dwg_file_path)
sleep(1) sleep(1)
custom_layer = get_layer_if_not_exist_create_it(doc, "123custom_layer")
tower_dict = s_file.tower_dic tower_dict = s_file.tower_dic
z_file_path = deduce_zfile_from_cad_path(dwg_file_path) z_file_path = deduce_zfile_from_cad_path(dwg_file_path)
z_point = plane_z_origin(z_file_path) z_point = plane_z_origin(z_file_path)
@ -441,6 +458,7 @@ class ContinuousPlate:
added_represented_span_text, added_represented_span_text,
ColorEnume.representive_span_text_color_rgb, ColorEnume.representive_span_text_color_rgb,
) )
added_represented_span_text.Layer = custom_layer.Name
if is_first_tower: # 是否是开始画的第一个塔。 if is_first_tower: # 是否是开始画的第一个塔。
accu_altitude_off = 0 accu_altitude_off = 0
else: else:
@ -479,14 +497,16 @@ class ContinuousPlate:
tower_pole = cad.model.AddPolyLine( tower_pole = cad.model.AddPolyLine(
np2d_to_array(np.vstack((np_tower_start, np_tower_end))) np2d_to_array(np.vstack((np_tower_start, np_tower_end)))
) )
tower_pole.Layer = custom_layer.Name
tower_pole.SetWidth(0, 0.8, 0.8) tower_pole.SetWidth(0, 0.8, 0.8)
set_true_color(tower_pole, 0, 255, 255) set_true_color(tower_pole, 0, 255, 255)
# 画塔名和呼高 # 画塔名和呼高
cad.model.AddText( added_tower_name = cad.model.AddText(
f"{tower_info.tower_name}", f"{tower_info.tower_name}",
APoint(*(np_tower_end + np.array([-5, 13])).tolist()), APoint(*(np_tower_end + np.array([-5, 13])).tolist()),
5, 5,
) )
added_tower_name.Layer = custom_layer.Name
if ( if (
abs(math.floor(tower_info.tower_height) - tower_info.tower_height) abs(math.floor(tower_info.tower_height) - tower_info.tower_height)
< 0.1 < 0.1
@ -494,11 +514,12 @@ class ContinuousPlate:
draw_tower_height_str = f"{tower_info.tower_height:.0f}" draw_tower_height_str = f"{tower_info.tower_height:.0f}"
else: else:
draw_tower_height_str = f"{tower_info.tower_height:.1f}" draw_tower_height_str = f"{tower_info.tower_height:.1f}"
cad.model.AddText( added_hugao = cad.model.AddText(
f"{tower_info.tower_type}-{draw_tower_height_str}", f"{tower_info.tower_type}-{draw_tower_height_str}",
APoint(*(np_tower_end + np.array([-5, 5])).tolist()), APoint(*(np_tower_end + np.array([-5, 5])).tolist()),
5, 5,
) )
added_hugao.Layer = custom_layer.Name
draw_count += 1 draw_count += 1
# 画弧垂 # 画弧垂
if not is_first_tower: # 从第二基塔开始画 if not is_first_tower: # 从第二基塔开始画
@ -544,7 +565,8 @@ class ContinuousPlate:
) )
draw_curve_x = draw_curve_x.reshape(len(draw_curve_x), 1) 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_curve_y = draw_curve_y.reshape(len(draw_curve_y), 1)
draw_ground_curve_y = draw_curve_y - 18 * 2 # 切地线 # TODO 应该读规程的
draw_ground_curve_y = draw_curve_y - 16 * 2 # 切地线
draw_tree_curve_y = draw_curve_y - 13.5 * 2 # 切树线 draw_tree_curve_y = draw_curve_y - 13.5 * 2 # 切树线
draw_point = np.hstack( draw_point = np.hstack(
(draw_curve_x, draw_curve_y, np.zeros((len(draw_curve_x), 1))) (draw_curve_x, draw_curve_y, np.zeros((len(draw_curve_x), 1)))
@ -566,18 +588,21 @@ class ContinuousPlate:
added_curve = cad.model.AddPolyLine( added_curve = cad.model.AddPolyLine(
draw_point.reshape(1, draw_curve_x.shape[0] * 3)[0] draw_point.reshape(1, draw_curve_x.shape[0] * 3)[0]
) )
added_curve.Layer = custom_layer.Name
set_true_color(added_curve, *ColorEnume.wire_color_rgb) set_true_color(added_curve, *ColorEnume.wire_color_rgb)
added_ground_curve = cad.model.AddPolyLine( added_ground_curve = cad.model.AddPolyLine(
draw_ground_curve_point.reshape( draw_ground_curve_point.reshape(
1, draw_ground_curve_y.shape[0] * 3 1, draw_ground_curve_y.shape[0] * 3
)[0] )[0]
) )
added_ground_curve.Layer = custom_layer.Name
set_true_color(added_ground_curve, *ColorEnume.ground_color_rgb) set_true_color(added_ground_curve, *ColorEnume.ground_color_rgb)
added_tree_curve = cad.model.AddPolyLine( added_tree_curve = cad.model.AddPolyLine(
draw_tree_curve_point.reshape( draw_tree_curve_point.reshape(
1, draw_tree_curve_y.shape[0] * 3 1, draw_tree_curve_y.shape[0] * 3
)[0] )[0]
) )
added_tree_curve.Layer = custom_layer.Name
set_true_color(added_tree_curve, *ColorEnume.tree_color_rgb) set_true_color(added_tree_curve, *ColorEnume.tree_color_rgb)
# 画档距 # 画档距
span_text_insert_point = np.array( span_text_insert_point = np.array(
@ -586,7 +611,23 @@ class ContinuousPlate:
added_span_text = cad.model.AddText( added_span_text = cad.model.AddText(
f"{span:.0f}", APoint(*span_text_insert_point.tolist()), 3 f"{span:.0f}", APoint(*span_text_insert_point.tolist()), 3
) )
# 画档距分割线
add_span_splitter = cad.model.AddPolyLine(
array(
"d",
[
(accu_mileage ) / 5 + 50,
5,
0,
(accu_mileage ) / 5 + 50,
10,
0,
],
)
)
add_span_splitter.Layer = custom_layer.Name
set_true_color(added_span_text, *ColorEnume.span_text_color_rgb) set_true_color(added_span_text, *ColorEnume.span_text_color_rgb)
added_span_text.Layer = custom_layer.Name
is_first_tower = False is_first_tower = False
last_tower_info = tower_info last_tower_info = tower_info
@ -653,6 +694,7 @@ class ControlFile:
string_impact_plate.draw() string_impact_plate.draw()
cad = continuous_plate.cad cad = continuous_plate.cad
cad.doc.SaveAs(self.get_zt_dwg_file_path()) cad.doc.SaveAs(self.get_zt_dwg_file_path())
cad.doc.Utility.Prompt("断面已生成。\n")
# # 画完后再打开 # # 画完后再打开
# cad = None # cad = None
# continousePlate = None # continousePlate = None