Add terrain descriptions.

[plomrogue2] / plomrogue / mapping.py

diff --git a/plomrogue/mapping.py b/plomrogue/mapping.py
index 10f9b70a5c1dbe9975e11d4233ba56aad80368d6..01c10e893c872322c44e7dab1c8bc571d270bf16 100644 (file)
--- a/plomrogue/mapping.py
+++ b/plomrogue/mapping.py
@@ -20,6 +20,7 @@ class MapGeometry():
 
     def __init__(self, size):
         self.size = size
+        self.neighbors_i = {}
 
     def get_directions(self):
         directions = []
@@ -34,6 +35,21 @@ class MapGeometry():
             neighbors[direction] = self.move(pos, direction)
         return neighbors
 
+    def get_neighbors_i(self, i):
+        if i in self.neighbors_i:
+            return self.neighbors_i[i]
+        pos = YX(i // self.size.x, i % self.size.x)
+        neighbors_pos = self.get_neighbors(pos)
+        neighbors_i = {}
+        for direction in neighbors_pos:
+            pos = neighbors_pos[direction]
+            if pos is None:
+                neighbors_i[direction] = None
+            else:
+                neighbors_i[direction] = pos.y * self.size.x + pos.x
+        self.neighbors_i[i] = neighbors_i
+        return self.neighbors_i[i]
+
     def move(self, start_pos, direction):
         mover = getattr(self, 'move_' + direction)
         target = mover(start_pos)
@@ -119,6 +135,12 @@ class Map():
         else:
             self.terrain[pos_i] = c
 
+    def __iter__(self):
+        """Iterate over YX position coordinates."""
+        for y in range(self.size.y):
+            for x in range(self.size.x):
+                yield YX(y, x)
+
     @property
     def size_i(self):
         return self.size.y * self.size.x
@@ -145,6 +167,7 @@ class Map():
 
 
 class FovMap(Map):
+    # FIXME: player visibility asymmetrical (A can see B when B can't see A)
 
     def __init__(self, source_map, center):
         self.source_map = source_map