320 lines
11 KiB
Python
320 lines
11 KiB
Python
"""
|
||
@Project ：microproduct
|
||
@File ：OnePlantHeight.PY
|
||
@Function ：主函数
|
||
|
||
@Author ：LMM
|
||
@Date ：2021/10/19 14:39
|
||
@Version ：1.0.0
|
||
"""
|
||
from xml.dom import minidom
|
||
from xml.etree.ElementTree import ElementTree, Element
|
||
import xml.dom.minidom
|
||
from lxml import etree
|
||
import codecs
|
||
import xml.etree.cElementTree as ET
|
||
|
||
|
||
class CreateMetafile:
|
||
"""
|
||
生成元文件案例
|
||
"""
|
||
def __init__(self, input_image_path, input_para_file, an_li_path, path):
|
||
"""
|
||
input_image_path: 影像头文件
|
||
input_para_file: 配置文件
|
||
an_li_path：案例路径
|
||
path：保存路径
|
||
"""
|
||
self.input_image_path = input_image_path
|
||
self.input_para_file = input_para_file
|
||
self.an_li_path= an_li_path
|
||
self.path = path
|
||
pass
|
||
|
||
def create_xml(self):
|
||
"""
|
||
读取元文件（只保留从头文件到sensor节点的部分）
|
||
输出sensor的节点位置
|
||
"""
|
||
tree = ElementTree()
|
||
tree.parse(self.input_image_path) # 影像头文件
|
||
root = tree.getroot()
|
||
# 1、只保留从头文件到sensor节点的部分
|
||
element_trees = list(root)
|
||
count = 0
|
||
count_01=1
|
||
for element in element_trees:
|
||
count = count+1
|
||
if element.tag == "sensor":
|
||
element.tail = "\n\n\t"
|
||
count_01 = count-1
|
||
for i in range(0, len(element_trees)):
|
||
if i > count_01:
|
||
root.remove(element_trees[i])
|
||
# 2、只保留"satellite", "orbitType", "attiType", "Direction", "ReceiveTime", "sensor"的部分
|
||
element_trees2 = list(root)
|
||
for i in element_trees2:
|
||
if i.tag not in ["satellite", "orbitType", "attiType", "Direction", "ReceiveTime", "sensor"]:
|
||
root.remove(i)
|
||
# 3、获取"sensor"节点的位置，并输出
|
||
count2 = 0
|
||
count2_01=1
|
||
element_trees3 = list(root)
|
||
for element in element_trees3:
|
||
count2 = count2+1
|
||
if element.tag == "sensor":
|
||
element.tail = "\n\n\t"
|
||
count2_01 = count2-1
|
||
tree.write(self.path, encoding="utf-8", xml_declaration=True)
|
||
return count2_01
|
||
|
||
@staticmethod
|
||
def create_node(tag, property_map, content):
|
||
"""
|
||
fun: 新造一个节点
|
||
para: tag:节点标签
|
||
para: property_map:属性及属性值map
|
||
para: content: 节点闭合标签里的文本内容
|
||
para: return 新节点
|
||
"""
|
||
element = Element(tag, property_map)
|
||
element.text = content
|
||
element.tail = "\n\t"
|
||
return element
|
||
|
||
def add_standard_xml(self, num):
|
||
"""
|
||
模板直接写入到元文件中
|
||
"""
|
||
tree = ElementTree()
|
||
tree.parse(self.path) # 影像头文件
|
||
root = tree.getroot()
|
||
|
||
tree2 = ElementTree()
|
||
tree2.parse(self.an_li_path) # 影像头文件
|
||
root2 = tree2.getroot()
|
||
|
||
productinfo = root2.find("productinfo")
|
||
root.insert(num + 1, productinfo)
|
||
processinfo = root2.find("processinfo")
|
||
root.insert(num + 2, processinfo)
|
||
tree.write(self.path, encoding="utf-8", xml_declaration=True)
|
||
|
||
def add_img_xml(self, num,SrcImageName):
|
||
"""添加影像信息"""
|
||
tree = ElementTree()
|
||
tree.parse(self.path)
|
||
root = tree.getroot()
|
||
|
||
a = self.create_node("SrcImageName", {"desc": "原始影像名称"}, SrcImageName)
|
||
root.insert(num+1, a)
|
||
# root.append(a)
|
||
b = self.create_node("AlgCompt", {"desc": "算法信息"}, "\n\t\t")
|
||
b.tail = "\n\n\t"
|
||
# root.append(b)
|
||
root.insert(num+2, b)
|
||
tree.write(self.path, encoding="utf-8", xml_declaration=True)
|
||
|
||
def add_info_xml(self):
|
||
"""
|
||
向元文件中添加配置文件的部分节（"AlgorithmName", "ChsName", "AlgorithmDesc", "Version",
|
||
"AlgorithmClass", "AlgorithmLevel", "AlgoirthmID", "Author"）
|
||
"""
|
||
tree = ElementTree()
|
||
tree.parse(self.input_para_file) # 配置文件
|
||
root = tree.getroot()
|
||
|
||
tree2 = ElementTree()
|
||
tree2.parse(self.path)
|
||
root2 = tree2.getroot()
|
||
AlgCompt = root2.find("AlgCompt")
|
||
|
||
a = root.find("AlgCompt")
|
||
|
||
element_trees = list(a)
|
||
for element in element_trees:
|
||
if element.tag in ["AlgorithmName", "ChsName", "AlgorithmDesc", "Version",
|
||
"AlgorithmClass", "AlgorithmLevel", "AlgoirthmID", "Author"]:
|
||
element.tail = "\n\t\t"
|
||
AlgCompt.append(element)
|
||
if element.tag == "Author":
|
||
element.tail = "\n\t"
|
||
|
||
tree2.write(self.path, encoding="utf-8", xml_declaration=True)
|
||
|
||
def add_class_info(self, type_id_name, type_id_parent=None):
|
||
"""
|
||
向元文件中：
|
||
1.删除productinfo-productType信息；
|
||
2.加入地物类别信息；
|
||
输出：
|
||
<Class1>
|
||
<parent_id>1</parent_id>
|
||
<id>101</id>
|
||
<covernm>耕地</covernm>
|
||
</Class1>
|
||
<Class2>
|
||
<parent_id>5</parent_id>
|
||
<id>502</id>
|
||
<covernm>草地</covernm>
|
||
</Class2>
|
||
"""
|
||
tree = ElementTree()
|
||
tree.parse(self.path) # 配置文件
|
||
root = tree.getroot()
|
||
productinfo = root.find("productinfo")
|
||
# element_trees = list(productinfo)
|
||
# for element in element_trees:
|
||
# if element.tag == "productType":
|
||
# productinfo.remove(element) # 移除 "productType"
|
||
productinfo.find("productConsumeTime").tail = "\n\t\t" # 定位到productConsumeTime，设置好位置
|
||
b = self.create_node("LandCoverClass", {}, "\n\t\t\t")
|
||
b.tail = "\n\t\t"
|
||
productinfo_count=0
|
||
for i in list(productinfo):
|
||
productinfo_count=productinfo_count+1
|
||
if i.tag=="productConsumeTime":
|
||
break
|
||
productinfo.insert(productinfo_count, b)
|
||
|
||
# productinfo.insert(num, b) # 插入LandCoverClass
|
||
class_num = 1
|
||
for key, value in type_id_name.items():
|
||
|
||
LandCoverClass = productinfo.find("LandCoverClass")
|
||
name="Class"+str(class_num)
|
||
# name = "Class"
|
||
c = self.create_node(name, {}, "\n\t\t\t\t")
|
||
if class_num!=(len(type_id_name.keys())):
|
||
c.tail = "\n\t\t\t"
|
||
else:
|
||
c.tail = "\n\t\t"
|
||
LandCoverClass.append(c) # 插入LandCoverClass
|
||
|
||
# LandCoverClass.find("Class")[num].tail = "\n\t\t\t"
|
||
aaa=LandCoverClass.find(name)
|
||
|
||
if type_id_parent is not None:
|
||
parent_id = self.create_node("parent_id", {}, type_id_parent[key])
|
||
parent_id.tail="\n\t\t\t\t"
|
||
LandCoverClass.find(name).append(parent_id)
|
||
id = self.create_node("id", {}, str(key))
|
||
id.tail = "\n\t\t\t\t"
|
||
LandCoverClass.find(name).append(id)
|
||
covernm = self.create_node("covernm", {}, value)
|
||
covernm.tail = "\n\t\t\t"
|
||
LandCoverClass.find(name).append(covernm)
|
||
class_num=class_num+1
|
||
tree.write(self.path, encoding="utf-8", xml_declaration=True)
|
||
|
||
def rewrite_name(self):
|
||
"""
|
||
修改class的名字:
|
||
class1->class
|
||
class2->class
|
||
"""
|
||
tree = ElementTree()
|
||
tree.parse(self.path) # 配置文件
|
||
root = tree.getroot()
|
||
productinfo = root.find("productinfo")
|
||
LandCoverClass=productinfo.find("LandCoverClass")
|
||
element_trees = list(LandCoverClass)
|
||
for element in element_trees:
|
||
element.tag="Class"
|
||
tree.write(self.path, encoding="utf-8", xml_declaration=True)
|
||
|
||
def OrthoInsertNode(self):
|
||
"""正射算法专用，插入节点<l1aInfo>"""
|
||
tree = ElementTree()
|
||
tree.parse(self.path) # 影像头文件
|
||
root = tree.getroot()
|
||
|
||
# 插入节点<l1aInfo>
|
||
count2 = 0
|
||
count2_01=1
|
||
element_trees3 = list(root)
|
||
for element in element_trees3:
|
||
count2 = count2+1
|
||
if element.tag == "sensor":
|
||
element.tail = "\n\n\t"
|
||
count2_01 = count2-1
|
||
b = self.create_node("l1aInfo", {}, "\n\t\t")
|
||
b.tail = "\n\n\t"
|
||
root.insert(count2_01+1, b)
|
||
|
||
# 查询节点位置<l1aInfo>
|
||
node_l1aInfo=root.find("l1aInfo")
|
||
|
||
img_tree = ElementTree()
|
||
img_tree.parse(self.input_image_path) # 影像头文件
|
||
img_root = img_tree.getroot()
|
||
|
||
node_imageinfo = img_root.find("imageinfo")
|
||
node_processinfo=img_root.find("processinfo")
|
||
|
||
ele_node_imageinfo = list(node_imageinfo)
|
||
ele_node_processinfo= list(node_processinfo)
|
||
|
||
for i in ele_node_imageinfo:
|
||
if i.tag == "QualifyValue":
|
||
i.tail = "\n\t\t"
|
||
node_l1aInfo.append(i)
|
||
|
||
for j in ele_node_processinfo:
|
||
if j.tag == "CalibrationConst":
|
||
j.tail = "\n\t" #后一个节点的位置
|
||
node_l1aInfo.append(j)
|
||
tree.write(self.path, encoding="utf-8", xml_declaration=True)
|
||
|
||
def process(self,SrcImageName):
|
||
"""
|
||
不涉及到地表覆盖调用此函数
|
||
"""
|
||
if self.input_image_path is None:
|
||
import xml.etree.cElementTree as ET
|
||
product = ET.Element("product") # 根节点tag= "product"
|
||
product.text = "\n\t"
|
||
tree = ET.ElementTree(product)
|
||
tree.write(self.path)
|
||
count = 0
|
||
count_2 = -1
|
||
else:
|
||
count = self.create_xml()
|
||
count_2 = count
|
||
self.add_standard_xml(count)
|
||
self.add_img_xml(count_2, SrcImageName)
|
||
self.add_info_xml()
|
||
|
||
def process2(self, type_id_name, type_id_parent,SrcImageName):
|
||
"""
|
||
涉及到地表覆盖的调用此函数
|
||
type_id_name={"101":"耕地","502":"草地"}
|
||
type_id_parent={"101":"1"，"502":"5"}
|
||
"""
|
||
count = self.create_xml()
|
||
self.add_standard_xml(count)
|
||
self.add_img_xml(count,SrcImageName)
|
||
self.add_info_xml()
|
||
self.add_class_info(type_id_name, type_id_parent)
|
||
self.rewrite_name()
|
||
|
||
def process3(self,SrcImageName):
|
||
"""
|
||
正射调用此函数
|
||
"""
|
||
if self.input_image_path is None:
|
||
import xml.etree.cElementTree as ET
|
||
product = ET.Element("product") # 根节点tag= "product"
|
||
product.text = "\n\t"
|
||
tree = ET.ElementTree(product)
|
||
tree.write(self.path)
|
||
count = 0
|
||
else:
|
||
count = self.create_xml()
|
||
self.add_standard_xml(count)
|
||
self.add_img_xml(count, SrcImageName)
|
||
self.add_info_xml()
|
||
self.OrthoInsertNode()
|
||
|