#!/usr/bin/python3
import pathlib
import pygubu
from tkinter import *
from PIL import ImageTk, Image
import numpy as np
import cv2

PROJECT_PATH = pathlib.Path(__file__).parent
PROJECT_UI = "./src/gui/main.ui"


class MainApp:
    def __init__(self, master=None):
        self.builder = builder = pygubu.Builder()
        builder.add_resource_path(PROJECT_PATH)
        builder.add_from_file(PROJECT_UI)
        # Main widget
        self.mainwindow = builder.get_object("main", master)

        self.canvas = builder.get_object("output_canvas")
        self.tk_imgs = []
        
        self.canny_thr1 = None
        self.canny_thr2 = None
        self.img_path = None
        self.blur_rate = None
        self.img_size = None
        self.sobel_select = None
        builder.import_variables(self,
                                ['canny_thr1',
                                'canny_thr2',
                                'img_path',
                                'blur_rate',
                                'img_size',
                                'sobel_select'])

        builder.connect_callbacks(self)

    def run(self):
        self.mainwindow.mainloop()

    def update(self, event=None):
        path = self.img_path.get()
        
        if path != None and path != "":
            # Get all user vars
            ct1 = self.canny_thr1.get()
            ct2 = self.canny_thr2.get()
            br = self.blur_rate.get()
            sxy = self.sobel_select.get()
            size = self.img_size.get()
            
            # Keep array of cv2 images to output
            output = []
            
            # Clear previously printed images
            self.tk_imgs = []
            
            # Import and resize image
            img = cv2.imread(path)
            img = cv2.resize(img, (size, size), interpolation = cv2.INTER_AREA)
            output.append(img) # First output image is original
            
            # Set grayscale
            img_gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
            output.append(img_gray) # Second output is grayscale
            
            # Blurred edition
            if (br == 0):
                # Disable
                img_blur = img_gray
            elif (br % 2 == 1):
                img_blur = cv2.GaussianBlur(img_gray, (br, br), 0)
            else:
                print(f"Blur rate changed to {br - 1}")
                img_blur = cv2.GaussianBlur(img_gray, (br-1, br-1), 0)
            output.append(img_blur)
            
            # Sobel edge
            if sxy in ['x', 'y', 'both']:
                if sxy == 'x':
                    dx = 1
                    dy = 0
                elif sxy == 'y':
                    dx = 0
                    dy = 1
                elif sxy == 'both':
                    dx = 1
                    dy = 1
                    
                img_sobel = cv2.Sobel(src=img_blur, ddepth=cv2.CV_8U, dx=dx, dy=dy, ksize=5)
            else:
                img_sobel = img_gray
            output.append(img_sobel)
            
            # Canny edge
            img_canny = cv2.Canny(image=img_blur,threshold1=ct1,threshold2=ct2)
            output.append(img_canny)
            
            # Draw all output images
            for idx, data in enumerate(output):
                # Create ui image
                tk_img = cv2.cvtColor(data, cv2.COLOR_BGR2RGB)
                tk_img = ImageTk.PhotoImage(image=Image.fromarray(tk_img))
                self.tk_imgs.append(tk_img)
            
                self.canvas.create_image((size * idx),0,anchor=NW,image=self.tk_imgs[idx],tags="og")

            self.canvas.draw()

if __name__ == "__main__":
    app = MainApp()
    app.run()