From e1ed22231d955eedf3895adf72fd3fbbeb133376 Mon Sep 17 00:00:00 2001
From: Johannes Löthberg <johannes@kyriasis.com>
Date: Sat, 23 Jul 2016 01:14:27 +0200
Subject: Initial commit
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: Johannes Löthberg <johannes@kyriasis.com>
---
 grid.py | 124 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 124 insertions(+)
 create mode 100644 grid.py

(limited to 'grid.py')

diff --git a/grid.py b/grid.py
new file mode 100644
index 0000000..8cc442a
--- /dev/null
+++ b/grid.py
@@ -0,0 +1,124 @@
+import random
+
+from PIL import Image, ImageDraw, ImageFilter
+
+from cell import Cell
+
+
+class Grid(object):
+    rows = None
+    columns = None
+
+    def __init__(self, rows, columns):
+        self.rows = rows
+        self.columns = columns
+        self.grid = self.prepare_grid()
+        self.configure_cells()
+
+    def prepare_grid(self):
+        grid = []
+        for x in range(0, self.rows):
+            row = []
+            for y in range(0, self.columns):
+                row.append(Cell(x, y))
+            grid.append(row)
+
+        return grid
+
+    def configure_cells(self):
+        for cell in self.each_cell():
+            row, col = cell.row, cell.column
+            cell.north = self[row - 1, col]
+            cell.south = self[row + 1, col]
+            cell.west  = self[row, col - 1]
+            cell.east  = self[row, col + 1]
+
+    def __getitem__(self, key):
+        row, column = key
+        if not 0 <= row <= self.rows - 1:
+            return None
+
+        if not 0 <= column <= len(self.grid[row]) - 1:
+            return None
+
+        return self.grid[row][column]
+
+    def random_cell(self):
+        row = random.randint(0, self.rows - 1)
+        column = random.randint(0, len(self.grid[row]))
+        return self[row, column]
+
+    def size(self):
+        return self.rows * self.columns
+
+    def each_row(self):
+        for row in self.grid:
+            yield row
+
+    def each_cell(self):
+        for row in self.each_row():
+            for cell in row:
+                if cell:
+                    yield cell
+
+    def __str__(self):
+        output = '+' + '---+' * self.columns + '\n'
+
+        for row in self.each_row():
+            top = '|'
+            bottom = '+'
+
+            for cell in row:
+                if not cell:
+                    cell = Cell(-1, -1)
+
+                if cell.is_linked_to(cell.east):
+                    east_boundary = ' '
+                else:
+                    east_boundary = '|'
+                top = top + '   ' + east_boundary
+
+                if cell.is_linked_to(cell.south):
+                    south_boundary = '   '
+                else:
+                    south_boundary = '---'
+                bottom = bottom + south_boundary + '+'
+
+            output += top + '\n'
+            output += bottom + '\n'
+
+        return output
+
+    def to_img(self, cell_size=50):
+        img_width = cell_size * self.columns
+        img_height = cell_size * self.rows
+
+        img = Image.new(mode='RGB', size=(img_width + 1, img_height + 1), color='#FFFFFF')
+        draw = ImageDraw.Draw(img)
+
+        for cell in self.each_cell():
+            x1 = cell.column * cell_size
+            y1 = cell.row * cell_size
+            x2 = (cell.column + 1) * cell_size
+            y2 = (cell.row + 1) * cell_size
+
+            if not cell.north:
+                draw.line([(x1, y1), (x2, y1)], fill='#000000')
+
+            if not cell.west:
+                draw.line([(x1, y1), (x1, y2)], fill='#000000')
+
+            if not cell.is_linked_to(cell.east):
+                r=1
+                draw.line([(x2, y1), (x2, y2)], fill='#000000', width=4)
+                draw.ellipse((x2-r, y1-r, x2+r, y1+r), fill='#000000')
+                draw.ellipse((x2-r, y2-r, x2+r, y2+r), fill='#000000')
+
+            if not cell.is_linked_to(cell.south):
+                r=1
+                draw.line([(x1, y2), (x2, y2)], fill='#000000', width=4)
+                draw.ellipse((x1-r, y2-r, x1+r, y2+r), fill='#000000')
+                draw.ellipse((x2-r, y2-r, x2+r, y2+r), fill='#000000')
+
+        img = img.filter(ImageFilter.GaussianBlur(radius=0.8))
+        img.save('maze.png')
-- 
cgit v1.2.3-54-g00ecf