351 lines
10 KiB
Python
351 lines
10 KiB
Python
import json
|
||
import uuid
|
||
|
||
from scipy.constants import point
|
||
from shapely.creation import points
|
||
|
||
from touying.ImageReproject_python.img_types import Point
|
||
|
||
|
||
# from Ai_tottle.touying.ImageReproject_python import Point
|
||
|
||
|
||
def generate_unique_id():
|
||
"""生成唯一ID"""
|
||
return str(uuid.uuid4())
|
||
|
||
|
||
def process_geojson_data(file_path):
|
||
try:
|
||
with open(file_path, 'r', encoding='utf-8') as f:
|
||
data = json.load(f)
|
||
|
||
# # 处理数据
|
||
# processed_data = process_geojson_data(data)
|
||
# print(processed_data)
|
||
# 写入输出文件
|
||
# with open(output_file, 'w', encoding='utf-8') as f:
|
||
# json.dump(processed_data, f, ensure_ascii=False, indent=2)
|
||
#
|
||
# print(f"数据处理完成，结果已保存到 {output_file}")
|
||
|
||
except FileNotFoundError:
|
||
print(f"错误：文件 {file_path} 未找到")
|
||
except json.JSONDecodeError:
|
||
print("错误：文件内容不是有效的JSON格式")
|
||
except Exception as e:
|
||
print(f"发生未知错误: {str(e)}")
|
||
|
||
"""处理GeoJSON数据，添加唯一ID"""
|
||
processed_data = {
|
||
"Point": [],
|
||
"LineString": [],
|
||
"Polygon": []
|
||
}
|
||
|
||
count = 0
|
||
# 处理Point数据
|
||
for point in data.get("Point", []):
|
||
new_point = point.copy()
|
||
# new_point["id"] = generate_unique_id()
|
||
new_point["coordinates"].append(count)
|
||
count=count+1
|
||
processed_data["Point"].append(new_point)
|
||
|
||
# 处理LineString数据
|
||
for line in data.get("LineString", []):
|
||
new_line = line.copy()
|
||
new_line["id"] = generate_unique_id()
|
||
for point in new_line["coordinates"]:
|
||
point.append(count)
|
||
count = count + 1
|
||
processed_data["LineString"].append(new_line)
|
||
|
||
# 处理Polygon数据
|
||
for polygon in data.get("Polygon", []):
|
||
new_polygon = polygon.copy()
|
||
new_polygon["id"] = generate_unique_id()
|
||
for point in new_polygon["coordinates"]:
|
||
point.append(count)
|
||
count = count + 1
|
||
processed_data["Polygon"].append(new_polygon)
|
||
|
||
return processed_data
|
||
|
||
|
||
def convert_processed_to_points(processed_data):
|
||
local_points = []
|
||
count=0
|
||
if processed_data["Point"] is not None and len(processed_data["Point"]) > 0:
|
||
for v in processed_data["Point"]:
|
||
if v["coordinates"] is not None and len(v["coordinates"]) > 0:
|
||
y = float(v["coordinates"][0]) # 103.99211191
|
||
x = float(v["coordinates"][1]) # 30.76381946
|
||
z = float(v["coordinates"][2]) # 高程
|
||
point_id = v["coordinates"][3]
|
||
point1 = Point(x, y, z, point_id)
|
||
count=count+1
|
||
local_points.append(point1)
|
||
if processed_data["LineString"] is not None and len(processed_data["LineString"]) > 0:
|
||
for v in processed_data["LineString"]:
|
||
if v["coordinates"] is not None and len(v["coordinates"]) > 0:
|
||
for vv in v["coordinates"]:
|
||
y = float(vv[0]) # 103.99211191
|
||
x = float(vv[1]) # 30.76381946
|
||
z = float(vv[2]) # 高程
|
||
point_id = vv[3]
|
||
point1 = Point(x, y, z, point_id)
|
||
count = count + 1
|
||
local_points.append(point1)
|
||
|
||
if processed_data["Polygon"] is not None and len(processed_data["Polygon"]) > 0:
|
||
for v in processed_data["Polygon"]:
|
||
if v["coordinates"] is not None and len(v["coordinates"]) > 0:
|
||
for vv in v["coordinates"]:
|
||
y = float(vv[0]) # 103.99211191
|
||
x = float(vv[1]) # 30.76381946
|
||
z = float(vv[2]) # 高程
|
||
point_id = vv[3]
|
||
point1 = Point(x, y, z, point_id)
|
||
count = count + 1
|
||
local_points.append(point1)
|
||
return local_points
|
||
|
||
|
||
def add_result_to_prcessdata(processed_data, result):
|
||
if processed_data["Point"] is not None and len(processed_data["Point"]) > 0:
|
||
for v in processed_data["Point"]:
|
||
if v["coordinates"] is not None and len(v["coordinates"]) >0:
|
||
# y = float(v["coordinates"][0]) # 103.99211191
|
||
# x = float(v["coordinates"][1]) # 30.76381946
|
||
# z = float(v["coordinates"][2]) # 高程
|
||
# point_id = v["id"]
|
||
# point1 = Point(x, y, z, point_id)
|
||
# local_points.append(point1)
|
||
for re in result:
|
||
if v["coordinates"][3] == re["id"]:
|
||
v["re"] = re
|
||
#
|
||
# point1 = {
|
||
# "u": 10,
|
||
# "v": 10,
|
||
# "id": 41
|
||
# }
|
||
# point2 = {
|
||
# "u": 1910,
|
||
# "v": 10,
|
||
# "id": 42
|
||
# }
|
||
# point3 = {
|
||
# "u": 10,
|
||
# "v": 540,
|
||
# "id": 41
|
||
# }
|
||
# point4 = {
|
||
# "u": 1910,
|
||
# "v": 540,
|
||
# "id": 41
|
||
# }
|
||
# point5 = {
|
||
# "u": 0,
|
||
# "v": 1070,
|
||
# "id": 41
|
||
# }
|
||
# point6 = {
|
||
# "u": 1910,
|
||
# "v": 1070,
|
||
# "id": 41
|
||
# }
|
||
# point7 = {
|
||
# "u": 960,
|
||
# "v": 540,
|
||
# "id": 41
|
||
# }
|
||
# processed_data["Point"].append({
|
||
# "re":point1
|
||
# })
|
||
# processed_data["Point"].append({
|
||
# "re":point2
|
||
# })
|
||
# processed_data["Point"].append({
|
||
# "re":point3
|
||
# })
|
||
# processed_data["Point"].append({
|
||
# "re":point4
|
||
# })
|
||
# processed_data["Point"].append({
|
||
# "re":point5
|
||
# })
|
||
# processed_data["Point"].append({
|
||
# "re":point6
|
||
# })
|
||
# processed_data["Point"].append(
|
||
# {
|
||
# "coordinates": [
|
||
# 103.9925256,
|
||
# 30.76395247,
|
||
# 471.192,
|
||
# 5
|
||
# ],
|
||
# "fillStyle": "red",
|
||
# "radius": 7,
|
||
# "comment": "point6",
|
||
# "re": {
|
||
# "u": 0,
|
||
# "v": 1070,
|
||
# "id": 41
|
||
# }
|
||
# }
|
||
# )
|
||
#
|
||
# processed_data["Point"].append(
|
||
# {
|
||
# "coordinates": [
|
||
# 103.9925256,
|
||
# 30.76395247,
|
||
# 471.192,
|
||
# 5
|
||
# ],
|
||
# "fillStyle": "red",
|
||
# "radius": 7,
|
||
# "comment": "point6",
|
||
# "re": {
|
||
# "u": 10,
|
||
# "v": 10,
|
||
# "id": 42
|
||
# }
|
||
# }
|
||
# )
|
||
#
|
||
# processed_data["Point"].append(
|
||
# {
|
||
# "coordinates": [
|
||
# 103.9925256,
|
||
# 30.76395247,
|
||
# 471.192,
|
||
# 5
|
||
# ],
|
||
# "fillStyle": "red",
|
||
# "radius": 7,
|
||
# "comment": "point6",
|
||
# "re": {
|
||
# "u": 1910,
|
||
# "v": 10,
|
||
# "id": 43
|
||
# }
|
||
# }
|
||
# )
|
||
#
|
||
# processed_data["Point"].append(
|
||
# {
|
||
# "coordinates": [
|
||
# 103.9925256,
|
||
# 30.76395247,
|
||
# 471.192,
|
||
# 5
|
||
# ],
|
||
# "fillStyle": "red",
|
||
# "radius": 7,
|
||
# "comment": "point6",
|
||
# "re": {
|
||
# "u": 10,
|
||
# "v": 540,
|
||
# "id": 44
|
||
# }
|
||
# }
|
||
# )
|
||
#
|
||
# processed_data["Point"].append(
|
||
# {
|
||
# "coordinates": [
|
||
# 103.9925256,
|
||
# 30.76395247,
|
||
# 471.192,
|
||
# 5
|
||
# ],
|
||
# "fillStyle": "red",
|
||
# "radius": 7,
|
||
# "comment": "point6",
|
||
# "re": {
|
||
# "u": 1910,
|
||
# "v": 540,
|
||
# "id": 45
|
||
# }
|
||
# }
|
||
# )
|
||
#
|
||
# processed_data["Point"].append(
|
||
# {
|
||
# "coordinates": [
|
||
# 103.9925256,
|
||
# 30.76395247,
|
||
# 471.192,
|
||
# 5
|
||
# ],
|
||
# "fillStyle": "red",
|
||
# "radius": 7,
|
||
# "comment": "point6",
|
||
# "re": {
|
||
# "u": 1910,
|
||
# "v": 1070,
|
||
# "id": 46
|
||
# }
|
||
# }
|
||
# )
|
||
#
|
||
# processed_data["Point"].append(
|
||
# {
|
||
# "coordinates": [
|
||
# 103.9925256,
|
||
# 30.76395247,
|
||
# 471.192,
|
||
# 5
|
||
# ],
|
||
# "fillStyle": "red",
|
||
# "radius": 7,
|
||
# "comment": "point6",
|
||
# "re": {
|
||
# "u": 960,
|
||
# "v": 540,
|
||
# "id": 47
|
||
# }
|
||
# }
|
||
# )
|
||
|
||
|
||
|
||
|
||
|
||
|
||
if processed_data["LineString"] is not None and len(processed_data["LineString"]) > 0:
|
||
for v in processed_data["LineString"]:
|
||
if v["coordinates"] is not None and len(v["coordinates"]) > 0:
|
||
v["re"] = []
|
||
#
|
||
# y = float(vv[0]) # 103.99211191
|
||
# x = float(vv[1]) # 30.76381946
|
||
# z = float(vv[2]) # 高程
|
||
# point_id = v["id"]
|
||
# point1 = Point(x, y, z, point_id)
|
||
# local_points.append(point1)
|
||
for re in result:
|
||
for vv in v["coordinates"]:
|
||
if vv[3]== re["id"]:
|
||
v["re"].append(re)
|
||
# if v["coordinates"][3] == re["id"]:
|
||
# v["re"].append(re)
|
||
|
||
|
||
if processed_data["Polygon"] is not None and len(processed_data["Polygon"]) > 0:
|
||
for v in processed_data["Polygon"]:
|
||
if v["coordinates"] is not None and len(v["coordinates"]) > 0:
|
||
v["re"] = []
|
||
for re in result:
|
||
for vv in v["coordinates"]:
|
||
if vv[3]== re["id"]:
|
||
v["re"].append(re)
|
||
# if v["coordinates"][3] == re["id"]:
|
||
# v["re"].append(re)
|
||
|
||
return processed_data
|