garbage.py

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

@author: kazzastic
"""
import cv2
import numpy as np
import threading 
import colorsys

class Point(object):
    
    def __init__(self, x = 0, y = 0):
        self.x = x
        self.y = y
    
    def __add__(self, other):
        return Point(self.x+other.x, self.y+other.y)
    
    def __eq__(self, other):
        return self.x == other.x and self.y == other.y
    
rw = 2
p = 0 
start = Point()
end = Point()

dir4 = [Point(0, 1), Point(0, -1), Point(1, 0), Point(-1, 0)]

def mouse_mov(event, pX, pY, flags, param):
    global img, start, end, p
    
    if event == cv2.EVENT_LBUTTONUP:
        if p == 0:
            cv2.rectangle(img, (pX - rw, pY - rw), (pX + rw, pY + rw), (0,0,255), -1)
            start = Point(pX, pY)
            print("start = ", start.x, start.y)
            p +=1
        elif p == 1:
            cv2.rectangle(img, (pX - rw, pY - rw), (pX + rw, pY + rw), (0, 200, 50), -1)
            end = Point(pX, pY)
            print("end = ", end.x, end.y)
            p+=1
            
def display():
    global img
    cv2.imshow("Image", img)
    cv2.setMouseCallback('Image', mouse_mov)
    while True:
        cv2.imshow("Image", img)
        cv2.waitKey(1)
 
def BFS(s, e):
    global img, h, w
    const = 10000

    found = False
    q = []
    v = [[0 for j in range(w)] for i in range(h)]
    parent = [[Point() for j in range(w)] for i in range(h)]

    q.append(s)
    v[s.y][s.x] = 1
    while len(q) > 0:
        p = q.pop(0)
        for d in dir4:
            cell = p + d
            if (cell.x >= 0 and cell.x < w and cell.y >= 0 and cell.y < h and v[cell.y][cell.x] == 0 and
                    (img[cell.y][cell.x][0] != 0 or img[cell.y][cell.x][1] != 0 or img[cell.y][cell.x][2] != 0)):
                q.append(cell)
                v[cell.y][cell.x] = v[p.y][p.x] + 1  # Later

                img[cell.y][cell.x] = list(reversed(
                    [i * 120 for i in colorsys.hsv_to_rgb(v[cell.y][cell.x] / const, 1, 1)])
                )
                parent[cell.y][cell.x] = p
                if cell == e:
                    found = True
                    del q[:]
                    break

    path = []
    if found:
        p = e
        while p != s:
            path.append(p)
            p = parent[p.y][p.x]
        path.append(p)
        path.reverse()

        for p in path:
            img[p.y][p.x] = [255, 255, 255]
        print("Path Found")
    else:
        print("Path Not Found")
    
    
if __name__ == "__main__":
    
    img = cv2.imread('MyMaze.png', cv2.IMREAD_GRAYSCALE)
    t, img = cv2.threshold(img, 120, 255, cv2.THRESH_BINARY)
    img = cv2.cvtColor(img, cv2.COLOR_GRAY2BGR)
    h, w= img.shape[:2]
    
    print("Select start and end points : ")
    t = threading.Thread(target=display, args=())
    t.daemon = True
    t.start()
    while p<2:
        pass
    
    BFS(start, end)
    cv2.waitKey(0)