Grid can now have integer values

Piratmac · Piratmac · commit a386be1c1e12 · 2021-12-11T13:20:37.000+01:00
diff --git a/2021/dot.py b/2021/dot.py
@@ -129,17 +129,17 @@ def __ge__(self, other):
 
     def __repr__(self):
         if self.grid.is_isotropic:
-            return self.terrain + "@" + complex(self.position).__str__()
+            return str(self.terrain) + "@" + complex(self.position).__str__()
         else:
             return (
-                self.terrain
+                str(self.terrain)
                 + "@"
                 + complex(self.position).__str__()
                 + direction_to_text[self.source_direction]
             )
 
     def __str__(self):
-        return self.terrain
+        return str(self.terrain)
 
     def __add__(self, direction):
         if not direction in self.allowed_directions:
diff --git a/2021/grid.py b/2021/grid.py
@@ -78,7 +78,7 @@ def reset_pathfinding(self):
             for dot in self.dots.values():
                 dot.neighbors_obsolete = True
 
-    def text_to_dots(self, text, ignore_terrain=""):
+    def text_to_dots(self, text, ignore_terrain="", convert_to_int=False):
         """
         Converts a text to a set of dots
 
@@ -94,14 +94,18 @@ def text_to_dots(self, text, ignore_terrain=""):
         for line in text.splitlines():
             for x in range(len(line)):
                 if line[x] not in ignore_terrain:
+                    if convert_to_int:
+                        value = int(line[x])
+                    else:
+                        value = line[x]
                     if self.is_isotropic:
-                        self.dots[x - y * 1j] = Dot(self, x - y * 1j, line[x])
+                        self.dots[x - y * 1j] = Dot(self, x - y * 1j, value)
                     else:
                         for dir in self.possible_source_directions.get(
-                            line[x], self.direction_default
+                            value, self.direction_default
                         ):
                             self.dots[(x - y * 1j, dir)] = Dot(
-                                self, x - y * 1j, line[x], dir
+                                self, x - y * 1j, value, dir
                             )
             y += 1
 
@@ -150,7 +154,7 @@ def dots_to_text(self, mark_coords={}, void=" "):
         for y in range(max_y, min_y - 1, -1):
             for x in range(min_x, max_x + 1):
                 try:
-                    text += mark_coords[x + y * 1j]
+                    text += str(mark_coords[x + y * 1j])
                 except (KeyError, TypeError):
                     if x + y * 1j in mark_coords:
                         text += "X"
