untitled0.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Created on Fri May 17 21:05:29 2019

@author: kazzastic
"""

from mazelib import *
import numpy as np
import matplotlib.patches as mpatches
import matplotlib.pyplot as plt
from copy import deepcopy

ile = 'mazegreen.jpg'
img = cv2.imread(file)
cv2.imshow('The Initial', img)
cv2.waitKey(0)
cv2.destroyAllWindows()
    
gray_img = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
cv2.imshow('Gray', gray_img)
cv2.waitKey(0)
cv2.destroyAllWindows()
    
blurred = cv2.GaussianBlur(gray_img, (5,5), 0)
cv2.imshow('Blur', blurred)
cv2.waitKey(0)
cv2.destroyAllWindows()
    
(t, thresh) = cv2.threshold(blurred, 155, 255, cv2.THRESH_BINARY)
cv2.imshow('Threshold Binary', thresh)
cv2.waitKey(0)
cv2.destroyAllWindows()

arr = np.array(thresh)

for i in range(0, 300):
    for j in range(0, 264):    
        if(arr[i][j] == 255):        
            arr[i][j] = 1


for i in range(0, 300):
    for j in range(0, 264):    
        if(arr[i][j] == 1):        
            arr[i][j] = 120

cv2.imshow('Result', arr)
cv2.waitKey(0)
cv2.destroyAllWindows()


m = Maze()
m.generator = Prims(27, 34)
m.generate()

m.solver = WallFollower()
m.generate_entrances()
m.solve()
sol = m.solutions
maze = m.grid
'''
def showPNG(maze):
    """Generate a simple image of the maze."""
    gridCopy = deepcopy(maze.grid)

    start_row , start_col = m.start
    end_row, end_col = m.end
    
    
    gridCopy[start_row][start_col]=2.5 #random values to change colormap
    gridCopy[end_row][end_col]=3.5

    solution = maze.solutions[0]

    if solution:

        for r,c in solution:
            gridCopy[r][c]=4.0

        start_patch = mpatches.Patch(color='tab:blue', label='Start')
        end_patch = mpatches.Patch(color='turquoise', label='End')



    plt.figure(figsize=(10, 5))

    plt.legend(handles = [start_patch,end_patch], \
               loc='center left', bbox_to_anchor=(1, 0.5))

    plt.imshow(gridCopy, cmap=plt.cm.ocean,interpolation = "nearest")
    plt.xticks([]), plt.yticks([])
    plt.show()

if __name__ == '__main__':
    
    m = Maze()
    m.generator = BacktrackingGenerator(25, 35)
    m.generate()
    m.generate_entrances()    
    m.solver = WallFollower()
    showPNG(m)
'''