wip

learn_sql_model/game/map.py

@@ -0,0 +1,99 @@
+from learn_sql_model.optional import _optional_import_
+import pydantic
+from rich.console import Console
+
+snoise2 = _optional_import_("noise", "snoise2", group="game")
+pygame = _optional_import_("pygame", group="game")
+
+
+console = Console()
+
+
+class Point(pydantic.BaseModel):
+    x: int
+    y: int
+
+
+class Map:
+    def __init__(self, game):
+        self.game = game
+        # self.grass = pygame.image.load("grass.webp").convert_alpha()
+        # self.rock = pygame.image.load("rock.jpg").convert_alpha()
+        # self.dirt = pygame.image.load("dirt.jpg").convert_alpha()
+        self.brown = (204, 153, 102)
+        self.grey = (128, 128, 128)
+        self.green = (0, 255, 0)
+        self.white = (255, 255, 255)
+        self.resolution = 16
+        self.scale = 0.14  # Determines the "smoothness" of the terrain
+        self.scale = 0.05  # Determines the "smoothness" of the terrain
+        self.offset = Point(x=0, y=0)
+        self.last_offset = self.offset
+        self.screen_width = self.game.screen.get_width()
+        self.screen_height = self.game.screen.get_height()
+        self.octaves = 2  # Number of layers of noise to combine
+        self.persistence = 0.05  # Amplitude of each octave
+        self.lacunarity = 1.0  # Frequency of each octave
+        self.thresh = 125
+        self.pre_draw()
+
+    def refresh_surf(self):
+        self.surf = pygame.Surface((self.screen_width, self.screen_height))
+
+    def get_noise(self, x, y):
+        value = snoise2(
+            (x + self.offset.x) * self.scale,
+            (y + self.offset.y) * self.scale,
+            self.octaves,
+            self.persistence,
+            self.lacunarity,
+        )
+        value = (value + 1) / 2 * 255
+        return value
+
+    def render(self):
+        self.game.screen.blit(
+            pygame.transform.scale(self.surf, (self.screen_width, self.screen_height)),
+            (0, 0),
+        )
+
+    def point_check_collision(self, x, y, thresh=None):
+        return self.get_noise(x / self.resolution, y / self.resolution) < (
+            thresh or self.thresh
+        )
+
+    def pre_draw(self):
+        self.refresh_surf()
+        for x in range(int(self.screen_width)):
+            for y in range(int(self.screen_height)):
+                if not self.point_check_collision(x, y):
+                    pygame.draw.rect(
+                        self.surf,
+                        self.white,
+                        (
+                            x,
+                            y,
+                            1,
+                            1,
+                        ),
+                    )
+        pygame.image.save(self.surf, "map.png")
+        # av1 = (
+        #     Image.open("rock.jpg")
+        #     .convert("RGB")
+        #     .resize((self.screen_width, self.screen_height))
+        # )
+        # av2 = (
+        #     Image.open("dirt.jpg")
+        #     .convert("RGB")
+        #     .resize((self.screen_width, self.screen_height))
+        # )
+        # mask = (
+        #     Image.open("map.png")
+        #     .convert("L")
+        #     .resize((self.screen_width, self.screen_height))
+        #     .filter(ImageFilter.GaussianBlur(3))
+        # )
+        # Image.composite(av2, av1, mask).save("result.png")
+        # result = pygame.image.load("result.png")
+        # self.surf.blit(result, (0, 0))