124 lines
2.8 KiB
Python
124 lines
2.8 KiB
Python
|
#导入测量的轨道数据
|
|||
|
|
|
|||
|
|
import re
|
|||
|
|
from PyQt6.QtWidgets import QFileDialog
|
|||
|
|
import color
|
|||
|
|
|
|||
|
|
def get_filename(filename = None):
|
|||
|
|
if not filename:
|
|||
|
|
if hasattr(get_filename, "filename"):
|
|||
|
|
path_hint = os.path.dirname(get_filename.filename)
|
|||
|
|
else:
|
|||
|
|
path_hint = ""
|
|||
|
|
filename, _ = QFileDialog.getOpenFileName(window, "从DAT文件导入数据", path_hint, "DAT数据文件(*.dat *.DAT)")
|
|||
|
|
if not filename:
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
if len(filename) < 4 or filename[-4:].lower() != ".dat":
|
|||
|
|
filename += ".dat"
|
|||
|
|
|
|||
|
|
filename = filename.replace("\\", "\\\\")
|
|||
|
|
filename = filename.replace("\"", "\\\"")
|
|||
|
|
return filename
|
|||
|
|
|
|||
|
|
def load_measure_data(file_name):
|
|||
|
|
with open(file_name, "r") as fobj:
|
|||
|
|
raw = fobj.readlines()
|
|||
|
|
|
|||
|
|
flag = False
|
|||
|
|
r1 = []
|
|||
|
|
|
|||
|
|
for line in raw:
|
|||
|
|
line = line.strip()
|
|||
|
|
match = re.match(r"For\s+M5\|Adr\s+", line)
|
|||
|
|
if match is None:
|
|||
|
|
raise RuntimeError("Unrecognized data: {line}")
|
|||
|
|
m = match.group(0)
|
|||
|
|
pos = len(m)
|
|||
|
|
line = line[pos:]
|
|||
|
|
|
|||
|
|
match = re.match(r"\d+\|([0-9A-Z]+)\s+", line)
|
|||
|
|
if match is None:
|
|||
|
|
raise RuntimeError("Unrecognized data: {line}")
|
|||
|
|
m = match.group(0)
|
|||
|
|
pos = len(m)
|
|||
|
|
line = line[pos:]
|
|||
|
|
|
|||
|
|
match = re.match(r"[0-9a-zA-Z\.\-]+", line)
|
|||
|
|
if match is None:
|
|||
|
|
raise RuntimeError("Unrecognized data: {line}")
|
|||
|
|
label = match.group(0)
|
|||
|
|
pos = len(label)
|
|||
|
|
line = line[pos:]
|
|||
|
|
|
|||
|
|
match = re.match(r"KZ\d*", label)
|
|||
|
|
if not match is None:
|
|||
|
|
label = match.group(0)
|
|||
|
|
pos = len(label)
|
|||
|
|
line = line[pos:]
|
|||
|
|
#print("控制点", label, line)
|
|||
|
|
continue
|
|||
|
|
|
|||
|
|
if label == "Intermediate":
|
|||
|
|
flag = True
|
|||
|
|
continue
|
|||
|
|
|
|||
|
|
if label == "End":
|
|||
|
|
flag = False
|
|||
|
|
continue
|
|||
|
|
|
|||
|
|
if not flag:
|
|||
|
|
continue
|
|||
|
|
|
|||
|
|
match = re.match(r"[A-Z]*\d+[\.\-]\d+", label)
|
|||
|
|
if match is None:
|
|||
|
|
continue
|
|||
|
|
|
|||
|
|
data = line.split("|")
|
|||
|
|
data = data[3]
|
|||
|
|
match = re.match(r"[0-9a-zA-Z]+\s+(\d+\.\d+)", data)
|
|||
|
|
if match is None:
|
|||
|
|
raise RuntimeError(f"找不到所需数据: {data}")
|
|||
|
|
data = match.group(1)
|
|||
|
|
|
|||
|
|
#print("轨道点", label, data)
|
|||
|
|
r1.append([label, data])
|
|||
|
|
|
|||
|
|
r2 = []
|
|||
|
|
for label, data in r1:
|
|||
|
|
match = re.match(r"([A-Z]*)(\d+)[\.\-](\d+)", label)
|
|||
|
|
l = match.group(1)
|
|||
|
|
d1 = int(match.group(2))
|
|||
|
|
d2 = int(match.group(3))
|
|||
|
|
d3 = float(data)
|
|||
|
|
r2.append([l, d1, d2, d3])
|
|||
|
|
|
|||
|
|
r2.sort(key = lambda x: (x[1], x[2]))
|
|||
|
|
# for line in r2:
|
|||
|
|
# print(f"{line[0]}{line[1]}.{line[2]} {line[3]}")
|
|||
|
|
return r2
|
|||
|
|
|
|||
|
|
filename = get_filename()
|
|||
|
|
if filename:
|
|||
|
|
#轨道数据表
|
|||
|
|
sheet_measure = sheet("measure_data")
|
|||
|
|
|
|||
|
|
data = load_measure_data(filename)
|
|||
|
|
sheet_measure.clear_content()
|
|||
|
|
with sheet_measure:
|
|||
|
|
for line in data:
|
|||
|
|
label = "%s%d.%d" % tuple(line[:3])
|
|||
|
|
row = line[1] - 1
|
|||
|
|
flag = line[2] > 1
|
|||
|
|
if line[2] > 2:
|
|||
|
|
print(f"{color.yellow}警告{color.nc}:编号{label}第2个数字大于2，当作右轨处理")
|
|||
|
|
data = line[3]
|
|||
|
|
if flag:
|
|||
|
|
cell(row, 2).value = label
|
|||
|
|
cell(row, 3).value = data
|
|||
|
|
else:
|
|||
|
|
cell(row, 0).value = label
|
|||
|
|
cell(row, 1).value = data
|
|||
|
|
|
|||
|
|
sheet_measure.fit_content()
|