- def get_neighbors(self, pos):
- # DOWNLEFT, DOWNRIGHT, LEFT, RIGHT, UPLEFT, UPRIGHT (alphabetically)
- neighbors = [None, None, None, None, None, None] # e, d, c, x, s, w
- if pos[1] > 0:
- neighbors[2] = [pos[0], pos[1] - 1]
- if pos[1] < self.size[1] - 1:
- neighbors[3] = [pos[0], pos[1] + 1]
- # x, c, s, d, w, e # 3->0, 2->1, 5->4, 0->5
- if pos[0] % 2 == 1:
- if pos[0] > 0 and pos[1] > 0:
- neighbors[4] = [pos[0] - 1, pos[1] - 1]
- if pos[0] < self.size[0] - 1 and pos[1] > 0:
- neighbors[0] = [pos[0] + 1, pos[1] - 1]
- if pos[0] > 0:
- neighbors[5] = [pos[0] - 1, pos[1]]
- if pos[0] < self.size[0] - 1:
- neighbors[1] = [pos[0] + 1, pos[1]]
- else:
- if pos[0] > 0 and pos[1] < self.size[1] - 1:
- neighbors[5] = [pos[0] - 1, pos[1] + 1]
- if pos[0] < self.size[0] - 1 and pos[1] < self.size[1] - 1:
- neighbors[1] = [pos[0] + 1, pos[1] + 1]
- if pos[0] > 0:
- neighbors[4] = [pos[0] - 1, pos[1]]
- if pos[0] < self.size[0] - 1:
- neighbors[0] = [pos[0] + 1, pos[1]]
- return neighbors
-