Added custom district generator and old GDMC functions and classes (to get the height, water and tree map)

world_maker/District.py

@@ -0,0 +1,90 @@
+from Position import Position
+
+
+class District:
+    """
+    The District class represents a district in the world.
+    A district can be characterized by its type and its unique id.
+
+    Attributes:
+        tile_id (int): The unique id of the district.
+        type (str): The type of the district. Can be "Forest", "City", "Mountain" or "Villa".
+    """
+
+    def __init__(self, tile_id: int):
+        """
+        The constructor for the District class.
+
+        :param tile_id: Unique id of the district (Must be greater than 0)
+        """
+        if tile_id <= 0:
+            raise ValueError("Tile id must be greater than 0")
+        self.tile_id = tile_id
+        self.type = "" #Forest, City, Montain, Villa
+
+
+def verify_point(point: Position, point_new: Position, map_data: list[list[int]], height_map: list[list[int]]):
+    """
+    Function to verify if a new point can be added to a district extend area list.
+
+    :param point: The current point.
+    :param point_new: The new point to be verified.
+    :param map_data: The 2D list representing the map.
+    :param height_map: The 2D list representing the height map.
+    :return: True if the new point can be added, False otherwise.
+    """
+    return (0 <= point_new.x < len(map_data[0]) and
+            0 <= point_new.y < len(map_data) and
+            map_data[point_new.y][point_new.x] == 0 and
+            abs(height_map[point_new.y][point_new.x] - height_map[point.y][point.x]) < 2)
+
+
+class CustomDistrict(District):
+    """
+    The CustomDistrict class represents a district that can be expanded.
+
+    Attributes:
+        center_expend (Position): The center position from which the district expands.
+        area (list): The list of positions that are part of the district.
+        area_expend_from_point (list): The list of positions from which the district can expand.
+        area_expend (list): The list of positions to which the district will maybe expand.
+    """
+    def __init__(self, tile_id: int, center: Position):
+        """
+        The constructor for the CustomDistrict class.
+
+        :param tile_id: Unique id of the district (Must be greater than 0)
+        :param center: The center position from which the district expands.
+        """
+        super().__init__(tile_id)
+        self.center_expend = center
+        self.area = [center]
+        self.area_expend_from_point = [center]
+        self.area_expend = []
+
+    def update_expend_points(self, point: Position, map_data: list[list[int]], height_map: list[list[int]]):
+        """
+        Update the points to which the district can expand.
+
+        :param point: The current point.
+        :param map_data: The 2D list representing the map.
+        :param height_map: The 2D list representing the height map.
+        """
+        for pos in [Position(1, 0), Position(-1, 0), Position(0, 1), Position(0, -1)]:
+            if verify_point(point, point + pos, map_data, height_map):
+                if point + pos not in self.area_expend:
+                    self.area_expend.append(point + pos)
+        self.area_expend_from_point.remove(point)
+
+
+class Edge: #I'm Edging rn
+    def __init__(self, point1, point2):
+        self.point1 = point1
+        self.point2 = point2
+
+
+class VoronoiDistrict(District):
+    def __init__(self, tile_id: int, center: Position):
+        super().__init__(tile_id)
+        self.center = center
+        self.edges = []