add MapGenerator for ddnet checker, use twmap-py for it!

Author: mwinkens

requirements.txt

@@ -4,4 +4,5 @@ pyqt6==6.7.0
 flake8==7.0.0
 pyinstaller
 pyyaml
-coverage
+coverage
+twmap

src/backend/map_generator.py

@@ -0,0 +1,59 @@
+import twmap
+import numpy as np
+from src.config.app_state import AppState
+from pathlib import Path
+from src.dialogs.dialog_check_map import CheckMapDialog
+
+
+class MapGenerator:
+    def __init__(self, map_name):
+        self._map = twmap.Map.empty("DDNet06")
+
+        solid_map = np.loadtxt("data/debroijn_torus.txt", dtype=np.uint8)
+        solid_map = np.pad(solid_map, 1, mode='constant')  # add gap to borders
+        height, width = solid_map.shape
+
+        # add layers
+        physics_group = self._map.groups.new_physics()
+        physics_layer = physics_group.layers.new_game(width, height)
+
+        # add image
+        assert AppState.imagePath()
+        path = Path(AppState.imagePath()).absolute()
+        self._map.images.new_from_file(str(path))
+
+        # set physic tiles
+        zeros = np.zeros(solid_map.shape, dtype=np.uint8)
+        solid_map_stacked = np.stack([solid_map, zeros], axis=2)
+        physics_layer.tiles = solid_map_stacked
+
+        # add tile layer
+        tile_layer = physics_group.layers.new_tiles(width, height)
+        tile_layer.name = "test"
+        tile_layer.image = 0
+        tile_layer_map = np.zeros(solid_map_stacked.shape, dtype=np.uint8)
+
+        # TODO twmap doesn't have automappers in the python version yet, so I have to do it manually
+        for y in range(1, height - 1):
+            for x in range(1, width - 1):
+                tile, status = CheckMapDialog.getMapTile(solid_map, x, y)
+                if tile:
+                    tile_layer_map[y, x, 0] = tile.tile_id
+                if status:
+                    tile_layer_map[y, x, 1] = MapGenerator._encodeBits(status.h_flip, status.v_flip, status.rot)
+        tile_layer.tiles = tile_layer_map
+
+        # save map
+        self._map.save(map_name)
+
+    def __del__(self):
+        # TODO delete map
+        pass
+
+    @staticmethod
+    def _encodeBits(h_flip: bool, v_flip: bool, rot: bool) -> int:
+        ret: int = 0
+        ret += (rot << 3)
+        ret += (h_flip << 1)
+        ret += v_flip
+        return ret

src/dialogs/dialog_check_map.py

@@ -23,7 +23,7 @@ def __init__(self, parent, title, cancel, *args, **kwargs):
         self.layout.setVerticalSpacing(0)
 
         # load torus
-        solid_map = np.loadtxt("data/debroijn_torus.txt", dtype=np.int8)
+        solid_map = np.loadtxt("data/debroijn_torus.txt", dtype=np.uint8)
         height, width = solid_map.shape
 
         # add stretch in front and after map for central alignment
@@ -45,7 +45,7 @@ def __init__(self, parent, title, cancel, *args, **kwargs):
                 grid_y = map_y + 1
                 grid_x = map_x + 1
 
-                tile, status = CheckMapDialog._getMapTile(solid_map, map_x, map_y)
+                tile, status = CheckMapDialog.getMapTile(solid_map, map_x, map_y)
                 if not tile:
                     tile = BaseTile(-1)  # Empty tile
                     if solid_map[map_y, map_x] == 0:
@@ -98,7 +98,7 @@ def _getMapNeighborhood(solid_map: np.ndarray, x: int, y: int) -> List[int]:
         return neighbors
 
     @staticmethod
-    def _getMapTile(solid_map: np.ndarray, x, y) -> Tuple[Optional[BaseTile], Optional[TileStatus]]:
+    def getMapTile(solid_map: np.ndarray, x, y) -> Tuple[Optional[BaseTile], Optional[TileStatus]]:
         if solid_map[y, x] == 0:
             return None, None  # TODO this isn't true
         neighbors = CheckMapDialog._getMapNeighborhood(solid_map, x, y)

src/widgets/widget_mapper_generator.py

@@ -8,6 +8,7 @@
 from src.config.app_state import AppState, ApplicationStatusEnum
 from src.dialogs.dialog_check_map import CheckMapDialog
 from src.config.config_manager import ConfigManager
+from src.backend.map_generator import MapGenerator
 
 
 class MapperGeneratorWidget(QWidget):
@@ -137,13 +138,15 @@ def startDDNetCheck(self):
         # add tmp mapping rule
         # automap map with debroijn torus
         # open map with ddnet
-        map_name = ""
+        map_name = "tmp_map.map"
+        MapGenerator(map_name)
+
         client_path = ConfigManager.instance().config()["client_path"]
         if not client_path:
             self.ddnet_push_button.setDisabled(True)
             return
-        cmd = [client_path, map_name]
-        subprocess.Popen(cmd, start_new_session=True)
+        #cmd = [client_path, map_name]
+        #subprocess.Popen(cmd, start_new_session=True)
 
     def rulesLoaded(self):
         rules = AppState.ruleManager().getRules()