env.py 904 B

123456789101112131415161718192021222324252627282930313233343536373839
  1. class Env:
  2. def __init__(self):
  3. self.x_range = 51 # size of background
  4. self.y_range = 31
  5. self.motions = [(1, 0), (-1, 0), (0, 1), (0, -1)]
  6. self.obs = self.obs_map()
  7. def obs_map(self):
  8. """
  9. Initialize obstacles' positions
  10. :return: map of obstacles
  11. """
  12. x = self.x_range
  13. y = self.y_range
  14. obs = []
  15. for i in range(x):
  16. obs.append((i, 0))
  17. for i in range(x):
  18. obs.append((i, y - 1))
  19. for i in range(y):
  20. obs.append((0, i))
  21. for i in range(y):
  22. obs.append((x - 1, i))
  23. for i in range(10, 21):
  24. obs.append((i, 15))
  25. for i in range(15):
  26. obs.append((20, i))
  27. for i in range(15, 30):
  28. obs.append((30, i))
  29. for i in range(16):
  30. obs.append((40, i))
  31. return obs