Added Building zone

world_maker/pack_rectangle.py

@@ -0,0 +1,115 @@
+from PIL import Image
+import numpy as np
+from typing import Union
+from data_analysis import handle_import_image
+
+class Rectangle:
+    def __init__(self, width, height):
+        self.width = width
+        self.height = height
+
+
+class Bin:
+    def __init__(self, grid):
+        self.grid = grid
+        self.rectangles = []
+
+    def place_rectangle(self, rectangle):
+        best_spot = None
+        best_spot_empty_area = float('inf')
+
+        for i in range(len(self.grid[0]) - rectangle.width + 1):  # Swap usage of x and y
+            for j in range(len(self.grid) - rectangle.height + 1):
+                if self.can_place(rectangle, i, j):
+                    empty_area = self.calculate_empty_area(rectangle, i, j)
+                    if empty_area < best_spot_empty_area:
+                        best_spot = (i, j)
+                        best_spot_empty_area = empty_area
+
+        if best_spot is not None:
+            self.rectangles.append((best_spot, (best_spot[0]+rectangle.width, best_spot[1]+rectangle.height)))
+            self.update_grid(rectangle, *best_spot)
+            return True
+
+        return False
+
+    def calculate_empty_area(self, rectangle, x, y):
+        empty_area = 0
+        for rect_x in range(x, x + rectangle.width):
+            for rect_y in range(y, y + rectangle.height):
+                if self.grid[rect_y][rect_x]:  # Swap usage of x and y
+                    empty_area += 1
+        return empty_area
+
+    def can_place(self, rectangle, x, y):
+        for rect_x in range(x, x + rectangle.width):
+            for rect_y in range(y, y + rectangle.height):
+                if not self.grid[rect_y][rect_x]:  # Swap usage of x and y
+                    return False
+        return True
+
+    def update_grid(self, rectangle, x, y):
+        for rect_x in range(x, x + rectangle.width):
+            for rect_y in range(y, y + rectangle.height):
+                self.grid[rect_y][rect_x] = False  # Swap usage of x and y
+
+
+def pack_rectangles(rectangles, grid):
+    rectangles = sorted(rectangles, key=lambda r: r.width * r.height, reverse=True)
+    bins = [Bin(grid)]
+
+    for rectangle in rectangles:
+        for bin in bins:
+            if bin.place_rectangle(rectangle):
+                break
+        else:  # No break, meaning rectangle couldn't be placed in any bin
+            new_bin = Bin(grid)
+            if new_bin.place_rectangle(rectangle):
+                bins.append(new_bin)
+            else:
+                return False  # If a rectangle can't be placed even in a new bin, return False
+
+    return True  # If all rectangles can be placed, return True
+
+
+import random
+
+
+def generate_rectangle(max_width, max_height):
+    width = random.randint(6, 20)
+    height = random.randint(6, 20)
+    return Rectangle(width, height)
+
+
+def pack_rectangles(grid):
+    max_width = len(grid[0])
+    max_height = len(grid)
+    bin = Bin(grid)
+
+    while True:
+        rectangle = generate_rectangle(max_width // 2, max_height // 2)
+        if not bin.place_rectangle(rectangle):
+            break  # Stop when a rectangle can't be placed
+        print(len(bin.rectangles))
+    return bin.rectangles  # Return the list of rectangles that were placed
+
+
+def draw_rectangles(rectangles, grid):
+    image = Image.new('RGB', (len(grid[0]), len(grid)), (0, 0, 0))
+    for rectangle in rectangles:
+        start, end = rectangle
+        for x in range(start[0], end[0]):
+            for y in range(start[1], end[1]):
+                image.putpixel((x, y), (144, 255, 144))
+    return image
+
+
+def generate_building(image: Union[str, Image] = './data/roadmap2.png'):
+    image = handle_import_image(image).convert('L')
+    grid = np.array(image)
+    rectangles = pack_rectangles(grid)
+    draw_rectangles(rectangles, grid).save('./data/building.png')
+    return rectangles
+
+if __name__ == '__main__':
+    generate_building()