Add unit tests for UnitFactory functionality and initialize units package

2025-08-13 20:36:51 +02:00
parent 088ae02080
commit 689b21bf65
10 changed files with 181 additions and 203 deletions
@@ -0,0 +1,43 @@
+import os
+
+class Graphics():
+    def load_assets(self):
+        self.tunnel = self.render_engine.load_image("Rat/BMP_TUNNEL.png", surface=True)
+        self.grasses = [self.render_engine.load_image(f"Rat/BMP_1_GRASS_{i+1}.png", surface=True) for i in range(4)]
+        self.rat_assets = {}
+        self.bomb_assets = {}
+        for sex in ["MALE", "FEMALE", "BABY"]:
+            self.rat_assets[sex] = {}
+            for direction in ["UP", "DOWN", "LEFT", "RIGHT"]:
+                self.rat_assets[sex][direction] = self.render_engine.load_image(f"Rat/BMP_{sex}_{direction}.png", transparent_color=(128, 128, 128))
+        for n in range(5):
+            self.bomb_assets[n] = self.render_engine.load_image(f"Rat/BMP_BOMB{n}.png", transparent_color=(128, 128, 128))
+        self.assets = {}
+        for file in os.listdir("assets/Rat"):
+            if file.endswith(".png"):
+                self.assets[file[:-4]] = self.render_engine.load_image(f"Rat/{file}")
+
+
+    # ==================== RENDERING ====================
+
+    def draw_maze(self):
+        if self.background_texture is None:
+            
+            texture_tiles = []
+            for y, row in enumerate(self.map.matrix):
+                for x, cell in enumerate(row):
+                    variant = x*y % 4
+                    tile = self.grasses[variant] if cell else self.tunnel
+                    texture_tiles.append((tile, x*self.cell_size, y*self.cell_size))
+            self.background_texture = self.render_engine.create_texture(texture_tiles)
+        self.render_engine.draw_background(self.background_texture)
+        
+    def scroll_cursor(self, x=0, y=0):
+        if self.pointer[0] + x > self.map.width or self.pointer[1] + y > self.map.height:
+            return
+        
+        self.pointer = (
+            max(1, min(self.map.width-2, self.pointer[0] + x)),
+            max(1, min(self.map.height-2, self.pointer[1] + y))
+        )
+        self.render_engine.scroll_view(self.pointer)