完成了初步功能。
This commit is contained in:
n3040
parent
057c199b18
commit
34b7dfb9e0
|
|
@ -0,0 +1,155 @@
|
|||
from xml.etree.ElementTree import Element
|
||||
from xml.etree.ElementTree import ElementTree
|
||||
from xml.etree.ElementTree import SubElement
|
||||
|
||||
|
||||
def indent(elem, level=0):
|
||||
i = "\n" + level * "\t"
|
||||
if len(elem):
|
||||
if not elem.text or not elem.text.strip():
|
||||
elem.text = i + "\t"
|
||||
if not elem.tail or not elem.tail.strip():
|
||||
elem.tail = i
|
||||
for elem in elem:
|
||||
indent(elem, level + 1)
|
||||
if not elem.tail or not elem.tail.strip():
|
||||
elem.tail = i
|
||||
else:
|
||||
if level and (not elem.tail or not elem.tail.strip()):
|
||||
elem.tail = i
|
||||
|
||||
|
||||
def AddEleElevationCurve(elevationCurves, id, vec_list, curve_side=None):
|
||||
if curve_side:
|
||||
elevationCurve_attribs = {"ID": id, "CurveSide": curve_side}
|
||||
else:
|
||||
elevationCurve_attribs = {"ID": id}
|
||||
elevationCurve = SubElement(
|
||||
elevationCurves, "EleElevationCurve", attrib=elevationCurve_attribs
|
||||
) # 中线
|
||||
SubElement(elevationCurve, "GrowthHeight").text = "0"
|
||||
SubElement(elevationCurve, "LinkedSectionObjectID_Group")
|
||||
SubElement(elevationCurve, "wallheight").text = "0"
|
||||
segmentList = SubElement(elevationCurve, "SegmentList")
|
||||
eleElevationCurveSegment = SubElement(
|
||||
segmentList,
|
||||
"EleElevationCurveSegment",
|
||||
# TODO 这个ID其实依据中、边导线会变化。
|
||||
attrib={"ID": "4", "StartPointClass": "NotSet", "EndPointClass": "NotSet"},
|
||||
)
|
||||
SubElement(eleElevationCurveSegment, "GrowthHeight").text = "0"
|
||||
SubElement(eleElevationCurveSegment, "LinkedSectionObjectID_Group")
|
||||
vec_list_element = SubElement(eleElevationCurveSegment, "VecList")
|
||||
for vec in vec_list:
|
||||
SubElement(vec_list_element, "Vector3", attrib={"xyz": ",".join(vec)})
|
||||
vec_list_rendering_element = SubElement(
|
||||
eleElevationCurveSegment, "VecListRendering"
|
||||
)
|
||||
for vec in vec_list:
|
||||
SubElement(vec_list_rendering_element, "Vector3", attrib={"xyz": ",".join(vec)})
|
||||
return elevationCurve
|
||||
|
||||
|
||||
def AddAnnotationElements(sectionData, pole_attrib, xyz):
|
||||
pdmGraphicsObject = SubElement(sectionData, "PdmGraphicsObject", attrib=pole_attrib)
|
||||
SubElement(pdmGraphicsObject, "GrowthHeight").text = "0"
|
||||
SubElement(pdmGraphicsObject, "LinkedSectionObjectID_Group")
|
||||
SubElement(pdmGraphicsObject, "Position", attrib={"xyz": xyz})
|
||||
|
||||
|
||||
section_name = "002"
|
||||
licheng = "100"
|
||||
remarks_start = "002"
|
||||
pole_name_start = "4001"
|
||||
remarks_end = "003"
|
||||
pole_name_end = "4002"
|
||||
verticalExtent2 = "20" # 边线宽度
|
||||
root = Element(
|
||||
"OverheadTransmissionLineDrawing",
|
||||
attrib={
|
||||
"Name": section_name,
|
||||
"xmlns:xsi": "http://www.w3.org/2001/XMLSchema-instance",
|
||||
"xmlns:xsd": "http://www.w3.org/2001/XMLSchema",
|
||||
"Description": "",
|
||||
"Version": "1.0",
|
||||
"StartAngle": "0",
|
||||
"EndAngle": "0",
|
||||
"NextID": "9",
|
||||
},
|
||||
)
|
||||
SubElement(root, "lstCrossingCable")
|
||||
drawingFrames = SubElement(root, "DrawingFrames", attrib={"PaperSizeId": "0"})
|
||||
SubElement(drawingFrames, "GrowthHeight").text = "0"
|
||||
SubElement(drawingFrames, "LinkedSectionObjectID_Group")
|
||||
frames = SubElement(drawingFrames, "Frames")
|
||||
frame = SubElement(
|
||||
frames,
|
||||
"Frame",
|
||||
attrib={
|
||||
"Name": "完整图纸",
|
||||
"SplitterPosition": "0",
|
||||
"PaperOrientation": "Landscape",
|
||||
"UserVerticalOffset": "0",
|
||||
},
|
||||
)
|
||||
SubElement(frame, "GrowthHeight").text = "0"
|
||||
SubElement(frame, "LinkedSectionObjectID_Group")
|
||||
elevAdjSections = SubElement(frame, "ElevAdjSections")
|
||||
SubElement(elevAdjSections, "GrowthHeight").text = "0"
|
||||
SubElement(elevAdjSections, "LinkedSectionObjectID_Group")
|
||||
sections = SubElement(elevAdjSections, "Sections")
|
||||
SubElement(
|
||||
sections, "ElevationAdjSection", attrib={"Position": "0", "ElevationAjusting": "0"}
|
||||
)
|
||||
SubElement(root, "GPSPointGroups")
|
||||
SubElement(root, "Orthoimages")
|
||||
SubElement(root, "GPSXianGao")
|
||||
SubElement(root, "HouseSwing")
|
||||
sectionData = SubElement(root, "SectionData")
|
||||
SubElement(sectionData, "licheng").text = licheng
|
||||
SubElement(sectionData, "lstlicheng")
|
||||
SubElement(sectionData, "lstSctnRcd")
|
||||
SubElement(sectionData, "ForestElevationCurves")
|
||||
elevationCurves = SubElement(sectionData, "ElevationCurves")
|
||||
AddEleElevationCurve(elevationCurves, "1", [["0", "0"], ["100", "200"]])
|
||||
AddEleElevationCurve(elevationCurves, "2", [["0", "0"], ["100", "202"]], "Left")
|
||||
AddEleElevationCurve(elevationCurves, "3", [["0", "0"], ["100", "205"]], "Right")
|
||||
annotationElements = SubElement(sectionData, "AnnotationElements")
|
||||
AddAnnotationElements(
|
||||
annotationElements,
|
||||
{
|
||||
"xsi:type": "ElePowerLinePole",
|
||||
"ID": "7",
|
||||
"CurveSide": "All",
|
||||
"Remarks": remarks_start,
|
||||
"PoleType": "AnglePole",
|
||||
"LeftElevation": "0",
|
||||
"RightElevation": "0",
|
||||
"CableAngle": "0",
|
||||
"PoleName": pole_name_start,
|
||||
},
|
||||
xyz="0,0",
|
||||
)
|
||||
AddAnnotationElements(
|
||||
annotationElements,
|
||||
{
|
||||
"xsi:type": "ElePowerLinePole",
|
||||
"ID": "8",
|
||||
"CurveSide": "All",
|
||||
"Remarks": remarks_end,
|
||||
"PoleType": "AnglePole",
|
||||
"LeftElevation": "0",
|
||||
"RightElevation": "0",
|
||||
"CableAngle": "0",
|
||||
"PoleName": pole_name_end,
|
||||
},
|
||||
xyz="100,200",
|
||||
)
|
||||
mapData = SubElement(root, "MapData")
|
||||
SubElement(mapData, "GrowthHeight").text = "0"
|
||||
SubElement(mapData, "LinkedSectionObjectID_Group")
|
||||
SubElement(mapData, "verticalExtent2").text = verticalExtent2
|
||||
SubElement(mapData, "MapElements")
|
||||
tree = ElementTree(root)
|
||||
indent(root)
|
||||
tree.write("result.nwed", encoding="utf-8", xml_declaration=True)
|
||||
Loading…
Reference in New Issue