46 lines
1.6 KiB
Python
46 lines
1.6 KiB
Python
"""优化路由文件：只保留在控制区域内行驶的车辆"""
|
||
import xml.etree.ElementTree as ET
|
||
|
||
# 控制区域的边（来自config）
|
||
control_edges = {
|
||
"G1523_AM3_4.1", "G1523_AM3_4.2", "G1523_AM3_4.3",
|
||
"G1523_AM3_4.4", "G1523_AM3_4.5", "G1523_AM3_4.6",
|
||
"G1523_AM4", "G1523_AM5", "G1523_AM6",
|
||
"G1523_AM6.1", "G1523_AM6.2", "G1523_AM6.3", "G1523_AM6.4", "G1523_AM6.5", "G1523_AM6.6",
|
||
"G1523_AM7", "G1523_AM7.1", "G1523_AM7.2", "G1523_AM8", "G1523_AM8.1", "G1523_AM8.2"
|
||
}
|
||
|
||
print(f"控制区域边数: {len(control_edges)}")
|
||
|
||
# 解析路由文件
|
||
route_file = "sumo_resource/routes_filtered.xml"
|
||
tree = ET.parse(route_file)
|
||
root = tree.getroot()
|
||
|
||
# 分析每条路由
|
||
routes_info = {}
|
||
for route in root.findall('route'):
|
||
route_id = route.get('id')
|
||
edges = route.get('edges').split()
|
||
|
||
# 计算在控制区域内的边数量
|
||
control_edges_in_route = [e for e in edges if e in control_edges]
|
||
non_control_edges = [e for e in edges if e not in control_edges]
|
||
|
||
routes_info[route_id] = {
|
||
'total': len(edges),
|
||
'control': len(control_edges_in_route),
|
||
'non_control': len(non_control_edges),
|
||
'edges': edges
|
||
}
|
||
|
||
print(f"\n路由分析:")
|
||
for rid, info in routes_info.items():
|
||
print(f" {rid}: 总边数={info['total']}, 控制区={info['control']}, 非控制区={info['non_control']}")
|
||
|
||
# 只保留主要在控制区域内的路由（控制区边数 >= 总边数的50%）
|
||
valid_routes = {rid for rid, info in routes_info.items() if info['control'] >= info['total'] * 0.5}
|
||
|
||
print(f"\n保留的路由: {valid_routes}")
|
||
print(f"删除的路由: {set(routes_info.keys()) - valid_routes}")
|