Remove chat QUERY.

[plomrogue2] / plomrogue / mapping.py

diff --git a/plomrogue/mapping.py b/plomrogue/mapping.py
index 4f2f5f9e3b7c465bc05ed1409af9b318befb8d71..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,11 +167,12 @@ 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
         self.size = self.source_map.size
-        self.fov_radius = (self.size.y / 2) - 0.5
+        self.fov_radius = 12 # (self.size.y / 2) - 0.5
         self.start_indented = True  #source_map.start_indented
         self.terrain = '?' * self.size_i
         self.center = center