orgsyncd/generate_ta.py

139 lines
5.9 KiB
Python

from file_object import SFileObject, CordinationObject, DFileAsSingle
TA_ENTRY_TEMPLATE = (
"{tower_number},{suspend_or_tension},{mileage}, {conductor_suspension_poit},0.3212,0.000,"
"{altitude},{pile_number},{tower_name}_{tower_height},{tower_height},{string_name},"
"{string_length}, 0.000,0.1142,0.000000,0.000000,,15,15.500000,,{angel},2.5,15.5,2.500,3.000,1,"
"610.4,2.000,0,2,0,2,{conductor_type},GJ-50,0,0,0,0.000,,0,0.000,9,750,1.000,,0,0.000,0,750,"
"1.000,ZHCH_1,28,1.000,9.5,750,1.000,9.5,750,1.000,{string_name},1,{string_length},{string_name},"
"1,{string_length},,0,0.000,,0,0.000,0,0,0.000, 0.000,0,0,0.000,0,0,4.000,0,1,1.000,,0,0.000,,"
"0,0.000,,0,0.000,1,1,0.000,0,0,0.000, 0.000,0,0, 0.000,1.000,1.000,0.000,0.000,0.000,,0,\n "
)
TA_NC_STRING_NAME = "750-NC"
TA_NC_STRING_LENGTH = 14
TA_NC_TOWER_NAME = "WNSJ"
TA_XC_STRING_NAME = "750-FXBW"
TA_XC_STRING_LENGTH = 9.5
TA_XC_TOWER_NAME = "WNSZ"
TA_CONDUCTOR_TYPE = "LGJ-400/50"
S_NC_STRING_NAME = "2*42-50"
# 通过串型来判别是不是耐张塔
def is_tension_tower_by_string(string_name):
if string_name == S_NC_STRING_NAME:
return 2
return 1
def prepare_TA_entry_without_altitude(SFiles):
visited_tower_number = {} # 记录访问过的塔号
last_S_mileage = 0
ta_entry = []
for S in SFiles:
SObj = SFileObject(S)
S_content = SObj.content()
for S_entry in S_content:
_entry = {}
if S_entry[0] == "首端转角号" or S_entry[0] == "塔号":
continue
tower_number = S_entry[0]
_entry["tower_number"] = tower_number
if tower_number not in visited_tower_number:
visited_tower_number[tower_number] = 0
else:
continue
mileage = int(float(S_entry[1])) + last_S_mileage
_entry["mileage"] = mileage
tower_height = float(S_entry[7])
_entry["tower_height"] = tower_height
angel = S_entry[5]
_entry["angel"] = angel
string_name = S_entry[8]
suspend_or_tension = is_tension_tower_by_string(string_name)
_entry["suspend_or_tension"] = suspend_or_tension
conductor_type = TA_CONDUCTOR_TYPE
_entry["conductor_type"] = conductor_type
# 1是悬垂 2是耐张
if suspend_or_tension == 1:
string_length = TA_XC_STRING_LENGTH
_entry["string_length"] = string_length
conductor_suspension_point = tower_height - string_length
_entry["conductor_suspension_point"] = conductor_suspension_point
pile_number = tower_number
_entry["pile_number"] = pile_number
ta_string_name = TA_XC_STRING_NAME
_entry["ta_string_name"] = ta_string_name
tower_name = TA_XC_TOWER_NAME
_entry["tower_name"] = tower_name
else:
string_length = TA_NC_STRING_LENGTH
_entry["string_length"] = string_length
conductor_suspension_point = tower_height
_entry["conductor_suspension_point"] = conductor_suspension_point
pile_number = tower_number
_entry["pile_number"] = pile_number
ta_string_name = TA_NC_STRING_NAME
_entry["ta_string_name"] = ta_string_name
tower_name = TA_NC_TOWER_NAME
_entry["tower_name"] = tower_name
ta_entry.append(_entry)
last_S_mileage = float(S_content[-1][1]) + last_S_mileage
return ta_entry
def write_to_TA_file(TA_file, ta_entry):
with open(TA_file, "w") as file:
for entry in ta_entry:
tower_number = entry["tower_number"]
suspend_or_tension = entry["suspend_or_tension"]
mileage = entry["mileage"]
conductor_suspension_point = entry["conductor_suspension_point"]
altitude = entry["altitude"]
pile_number = entry["pile_number"]
tower_name = entry["tower_name"]
tower_height = entry["tower_height"]
ta_string_name = entry["ta_string_name"]
string_length = entry["string_length"]
angel = entry["angel"]
conductor_type = entry["conductor_type"]
file.write(
TA_ENTRY_TEMPLATE.format(
tower_number=tower_number,
suspend_or_tension=suspend_or_tension,
mileage=mileage,
conductor_suspension_poit=conductor_suspension_point,
altitude=altitude,
pile_number=pile_number,
tower_name=tower_name,
tower_height=tower_height,
string_name=ta_string_name,
string_length=string_length,
angel=angel,
conductor_type=conductor_type,
)
)
# SFiles中的文件必须是按顺序排列的
def generate_ta_from_D(TA_file, SFiles, D_files_paths):
# 准备要写入的文件
ta_entry_without_altitude = prepare_TA_entry_without_altitude(SFiles)
d_file_object = DFileAsSingle(D_files_paths)
ta_entry = list(ta_entry_without_altitude)
for entry in ta_entry:
mileage = entry["mileage"]
entry["altitude"] = d_file_object.get_altitude(mileage)
write_to_TA_file(TA_file, ta_entry)
# SFiles中的文件必须是按顺序排列的
def generate_ta_from_csv(TA_file, SFiles, cordination_file_path):
ta_entry_without_altitude = prepare_TA_entry_without_altitude(SFiles)
cord_file_object = CordinationObject(cordination_file_path)
ta_entry = list(ta_entry_without_altitude)
for entry in ta_entry:
pile_number = entry["tower_number"]
entry["altitude"] = cord_file_object.get_altitude(pile_number)
write_to_TA_file(TA_file, ta_entry)