SUMO仿真中自定义边缘数据的可视化方法

在SUMO交通仿真系统中，边缘数据(edgedata)是分析路网性能的重要信息来源。虽然SUMO提供了丰富的内置边缘指标，但实际项目中我们经常需要可视化一些自定义指标，如停车位占用率等。本文将详细介绍如何在SUMO中实现自定义边缘数据的可视化。

自定义边缘数据的基本原理

SUMO的plot_net_dump.py工具能够可视化边缘数据XML文件中的各种指标。该文件的标准结构包含多个时间间隔(interval)元素，每个interval下又包含多个edge元素，edge元素中可以定义任意属性作为可视化指标。

方法一：直接生成标准格式的XML文件

最直接的方法是按照SUMO的edgeData格式要求生成XML文件。文件结构需要满足以下要求：

1. 根元素为edgeData
2. 包含多个interval子元素，每个interval需定义begin和end属性表示时间范围
3. 每个interval下包含多个edge子元素，每个edge需定义id属性
4. edge元素中可以添加任意自定义属性作为可视化指标

示例代码框架如下：

with open('custom_edgedata.xml', 'w') as f:
    f.write('<edgeData>\n')
    for time in simulation_times:
        f.write(f'<interval begin="{time}" end="{time+step}">\n')
        for edge in network_edges:
            occupancy = calculate_parking_occupancy(edge)
            f.write(f'<edge id="{edge.id}" parkingOccupancy="{occupancy}"/>\n')
        f.write('</interval>\n')
    f.write('</edgeData>')

方法二：扩展现有edgeData文件

如果已有edgeData文件，可以使用sumolib库进行扩展：

from sumolib import xml

# 假设已有停车位占用数据
parking_data = {
    "edge1": 0.75,
    "edge2": 0.32,
    # 其他边缘数据...
}

with open('extended_edgedata.xml', 'w') as outf:
    outf.write('<edgeData>\n')
    # 解析现有edgeData文件
    for interval in xml.parse('original_edgedata.xml', 'interval'):
        for edge in interval.edge:
            # 为每个edge添加自定义属性
            edge.setAttribute('parkingOccupancy', parking_data.get(edge.id, 0))
        outf.write(interval.toXML())
    outf.write('</edgeData>\n')

可视化自定义数据

生成文件后，使用plot_net_dump.py工具进行可视化：

python plot_net_dump.py -n your_network.net.xml -i custom_edgedata.xml --attributes parkingOccupancy

注意事项

1. 属性名称应避免使用SUMO保留关键字
2. 时间间隔设置应与仿真设置一致
3. 数值型属性应确保格式正确
4. 对于大量数据，建议分批处理以避免内存问题

通过上述方法，用户可以灵活地将各种自定义指标集成到SUMO的可视化分析中，大大扩展了SUMO在交通仿真研究中的应用范围。