mirror of
https://github.com/zhm-real/PathPlanning.git
synced 2026-08-29 16:40:46 +08:00
zhm-real
65 lines
1.4 KiB
Python
65 lines
1.4 KiB
Python
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
"""
|
|
@author: huiming zhou
|
|
"""
|
|
|
|
import matplotlib.pyplot as plt
|
|
|
|
x_range, y_range = 51, 31 # size of background
|
|
|
|
def obs_map():
|
|
"""
|
|
Initialize obstacles' positions
|
|
|
|
:return: map of obstacles
|
|
"""
|
|
|
|
obs = []
|
|
for i in range(x_range):
|
|
obs.append((i, 0))
|
|
for i in range(x_range):
|
|
obs.append((i, y_range - 1))
|
|
|
|
for i in range(y_range):
|
|
obs.append((0, i))
|
|
for i in range(y_range):
|
|
obs.append((x_range - 1, i))
|
|
|
|
for i in range(10, 21):
|
|
obs.append((i, 15))
|
|
for i in range(15):
|
|
obs.append((20, i))
|
|
|
|
for i in range(15, 30):
|
|
obs.append((30, i))
|
|
for i in range(16):
|
|
obs.append((40, i))
|
|
|
|
return obs
|
|
|
|
|
|
def lose_map():
|
|
lose = []
|
|
for i in range(27, 34):
|
|
lose.append((i, 13))
|
|
return lose
|
|
|
|
|
|
def show_map(xI, xG, obs_map, lose_map, name):
|
|
obs_x = [obs_map[i][0] for i in range(len(obs_map))]
|
|
obs_y = [obs_map[i][1] for i in range(len(obs_map))]
|
|
|
|
lose_x = [lose_map[i][0] for i in range(len(lose_map))]
|
|
lose_y = [lose_map[i][1] for i in range(len(lose_map))]
|
|
|
|
plt.plot(xI[0], xI[1], "bs")
|
|
plt.plot(xG[0], xG[1], "gs")
|
|
plt.plot(obs_x, obs_y, "sk")
|
|
plt.plot(lose_x, lose_y, marker = 's', color = '#A52A2A')
|
|
plt.title(name, fontdict=None)
|
|
plt.grid(True)
|
|
plt.axis("equal")
|
|
plt.show()
|
|
|