61 lines
2.0 KiB
Python
61 lines
2.0 KiB
Python
import matplotlib.pyplot as plt
|
||
import matplotlib.patches as patches
|
||
import json
|
||
|
||
def visualize_solution(row_boundaries, col_boundaries, car_paths, W, H):
|
||
plt.rcParams['font.family'] = ['sans-serif']
|
||
plt.rcParams['font.sans-serif'] = ['SimHei']
|
||
fig, ax = plt.subplots()
|
||
ax.set_xlim(0, W)
|
||
ax.set_ylim(0, H)
|
||
ax.set_title("区域划分与车-机-巢系统覆盖")
|
||
ax.set_xlabel("区域宽度")
|
||
ax.set_ylabel("区域高度")
|
||
|
||
# 定义若干颜色以区分不同系统（系统编号从0开始）
|
||
colors = ['red', 'blue', 'green', 'orange', 'purple', 'cyan', 'magenta']
|
||
|
||
# 绘制区域中心
|
||
region_center = (H / 2.0, W / 2.0) # 注意：x对应宽度，y对应高度
|
||
ax.plot(region_center[1], region_center[0],
|
||
'ko', markersize=8, label="区域中心")
|
||
|
||
# 绘制行分割边界
|
||
for row in row_boundaries:
|
||
ax.axhline(y=row * H, color='black', linestyle='--')
|
||
|
||
# 绘制列分割边界
|
||
for col in col_boundaries:
|
||
ax.axvline(x=col * W, color='black', linestyle='--')
|
||
|
||
# 绘制每辆车的轨迹
|
||
for system_id, path in car_paths.items():
|
||
path = [(region_center[0], region_center[1])] + path + [(region_center[0], region_center[1])]
|
||
y, x = zip(*path)
|
||
ax.plot(x, y, marker='o', color=colors[int(system_id) % len(colors)], label=f"系统 {system_id}")
|
||
|
||
# 添加图例
|
||
ax.legend()
|
||
# 反转 y 轴使得行号从上到下递增（如需，可取消）
|
||
ax.invert_yaxis()
|
||
plt.show()
|
||
|
||
if __name__ == "__main__":
|
||
import yaml
|
||
|
||
# 读取参数
|
||
with open('params.yml', 'r', encoding='utf-8') as file:
|
||
params = yaml.safe_load(file)
|
||
|
||
H = params['H']
|
||
W = params['W']
|
||
|
||
# 读取最佳方案的JSON文件
|
||
with open('./solutions/best_solution_mtkl.json', 'r', encoding='utf-8') as f:
|
||
best_solution = json.load(f)
|
||
|
||
row_boundaries = best_solution['row_boundaries']
|
||
col_boundaries = best_solution['col_boundaries']
|
||
car_paths = best_solution['car_paths']
|
||
|
||
visualize_solution(row_boundaries, col_boundaries, car_paths, W, H) |