#!/home/m/anaconda3/bin/python3
#https://hackaday.io/project/179946-image-to-svg-and-g-code-converter
import pkg_resources
import subprocess
import sys
import os

REQUIRED = {
  'tk', 'opencv-python', 'numpy', 'pathlib', 
  'Pillow'
}

installed = {pkg.key for pkg in pkg_resources.working_set}
missing = REQUIRED - installed

# Automatically installing missing packages

# Comment these out if want to install them manually

if missing:
    python = sys.executable
    subprocess.check_call([python, '-m', 'pip', 'install', *missing], stdout=subprocess.DEVNULL)


from tkinter import *    # from tkinter import Tk for Python 3.x
from tkinter.filedialog import askopenfilename
from tkinter import messagebox as MB
import cv2
from pathlib import Path
import math
import numpy as np
from PIL import Image, ImageTk
import image_process as IP
from tkinter import ttk
from tkinter import scrolledtext


#------------- Main window creation ----------------

root = Tk()
root.title('SVG From Camera')

tabControl = ttk.Notebook(root)
tab1 = ttk.Frame(tabControl)
tabControl.add(tab1, text="Image upload")
tabControl.pack(expand=1, fill="both")

root.geometry('1020x900')

#-------------Contour detection variables ---------------

image_path = StringVar()
image_name = StringVar()
##gcode_var = StringVar()
thresh_min = IntVar()
thresh_min.set(60)
thresh_max = IntVar()
thresh_max.set(255)
inv_contour = BooleanVar()
choosen_blur = StringVar()
choices = ['Gaussian Blur', 'Simple Blur', 'Median Blur', 'Bilateral Filter', '2D Filter', 'Laplacian', 'No Blur']
choosen_blur.set(choices[5])
max_image_area = IntVar()
min_image_area = IntVar()
conture_threshold = BooleanVar()
min_contour_area = IntVar()
max_contour_area = IntVar()

vertical_mirror = BooleanVar()
vertical_mirror.set(False)
horizontal_mirror = BooleanVar()
horizontal_mirror.set(False)

#------------- Image variables --------------------

preparedImage = None
blurred_process = None
blurred_show = None
contourCanvas = None
tkImageOri = None
tkImageGray = None
tkImageBlurred = None
tkImageContour = None
tkImageContourZoom = None
final_contours = None
isFirstRun=True
cv2Image = None
filteredContourArray = None



# ---------------- Functions -----------------------

# Start when clicking Upload image or New Image buttons, let's you choose a file and reads it
def uploadImage():
    #uploadImage
	cam = cv2.VideoCapture(0)   
	s, img = cam.read()
	if s:
		cv2.imwrite("filename.jpg",img)
	filepath=("filename.jpg")
	path = Path(filepath)
	##path = Path(filepath)
	image_name.set(path.stem)

	
	# Set global variable: Filename
	image_path.set(filepath)

	print(image_name.get())

	# Prepare image for edge detection

	if filepath == "":
		print("No image selected")
		return None

	
	# Refresh label widgets
	setImages()
	
# Creates the different images types from the uploaded image
def createContoures(isInv,thresh_min):
	global tkImageContour
	global final_contours

	if blurred_process is None:
		return None
	
	#global thres_min
	# Big image find contours	
	thresh_min = thres_min_scale.get()
	contours_big = IP.detectContours(blurred_process, isInv,thresh_min)

	if contours_big != None:
		gcode_gen_button.configure(state="active")
		final_contours = contours_big

	# Small image find contours and draw
	#contours_show = IP.detectContours(blurred_process)
	contourCanvas = np.zeros(contourCanvas.shape, np.uint8)
	contourImage, approx = IP.drawContoursOnCanvas(contourCanvas, contours_big)

	# Convert OpenCV canvas image to tKinter image 
	tkim = Image.fromarray(resizedCanvas)
	imgtk_contour = ImageTk.PhotoImage(image=tkim)
	tkImageContour = imgtk_contour

# Sets the global image variable to the previously created images, and show them on the UI
def setImages():

	global isFirstRun
	try:
		tkimage_ori, tkimage_gray, tkimage_blurred, tkimage_contour, tkimage_contour_zoom, max_area, min_area, cv2image, filteredContours = IP.createPreviewImages(image_path.get(), thresh_min.get(), thresh_max.get(), inv_contour.get(), choosen_blur.get(), min_contour_area.get(), max_contour_area.get(), conture_threshold.get(), horizontal_mirror.get(), vertical_mirror.get())
	except Exception as e: 
		MB.showerror(title="Error", message="Image format is not supported")
		print(e)
		return
		
	im_upload_button.place_forget()	
	max_image_area.set(round(max_area))
	min_image_area.set(round(min_area))

	area_thresh_min_scale.configure(from_=min_area,to=max_area, variable=min_contour_area)
	area_thresh_max_scale.configure(from_=min_area,to=max_area, variable=max_contour_area)

	if isFirstRun:
		area_thresh_max_scale.set(max_image_area.get())
		isFirstRun = False

	area_thresh_frame.grid(row=4, column=0)

	global contourCanvas
	contourCanvas = tkimage_contour

	global tkImageOri
	tkImageOri = tkimage_ori

	global tkImageGray
	tkImageGray = tkimage_gray

	global tkImageBlurred
	tkImageBlurred = tkimage_blurred

	global tkImageContour
	tkImageContour = tkimage_contour

	global tkImageContourZoom
	tkImageContourZoom = tkimage_contour_zoom

	global cv2Image
	cv2Image = cv2image

	global filteredContourArray
	filteredContourArray = filteredContours

	# Update image labels with global variables
	
	im_ori_show_label.configure(image=tkImageOri)
	
	im_gray_show_label.configure(image=tkImageGray)
	
	im_blurred_show_label.configure(image=tkImageBlurred)
	
	im_contour_show_label.configure(image=tkImageContour)

	im_zoom_label.configure(image=tkImageContourZoom)

	all_contour_settings_frame.grid(row=0, column=1, padx=10)

# When 'Contour area filter' is off, this hides the corresponding sliders
def hideContourThreshold():
	if conture_threshold.get():
		area_thresh_frame.configure(height=140)
		area_thresh_min_scale.grid(row=1, column=0)
		area_thresh_max_scale.grid(row=2, column=0)
		
	else:
		
		area_thresh_min_scale.grid_forget()
		area_thresh_max_scale.grid_forget()
		area_thresh_frame.configure(height=40)

# When clicking on any of the images it zooms in the contour (svg type) image
def zoomImage(imageName):
	
	im_ori_show_label.grid_forget()
	im_gray_show_label.grid_forget()
	im_blurred_show_label.grid_forget()
	im_contour_show_label.grid_forget()

	im_zoom_label.grid(row=0, column = 0, sticky=W)

# When clicking on the zoomed image, this zooms it out
def closeZoomImage():
	
	im_zoom_label.grid_forget()
	im_ori_show_label.grid(row=0, column = 0, sticky=W)
	im_gray_show_label.grid(row=0, column = 1, sticky=W)
	im_blurred_show_label.grid(row=1, column =0, sticky=W)
	im_contour_show_label.grid(row=1, column = 1, sticky=W)

# Starts the SVG creating process
def getSVG():
	success = IP.createSVG(image_name.get())
	if success:
		MB.showinfo(title="Success", message="SVG has been created and saved in the 'output' directory")
	else:
		MB.showerror(title="Error", message="An error occured during SVG creation")	



#---------------------------- Image upload tab ------------------------------

# Left column on the tab, holds the images

image_frame = Frame(tab1, width=660, height=900, relief="flat")
image_frame.grid(row=0, column=0)
image_frame.rowconfigure(0, weight=1)
image_frame.columnconfigure(0, weight=1)
image_frame.grid_propagate(0)

# This is kinda the same as the previous one
image_preview_frame = Frame(image_frame, width=660, height=900, relief="flat")
image_preview_frame.grid(row=0, column=0)


	# Original image
im_ori_show_label = Label(image_preview_frame, image = None, pady = 20)
im_ori_show_label.grid(row=0, column = 0, sticky=W)
im_ori_show_label.bind("", lambda e: zoomImage("originalImage"))

	# Gray Image 
im_gray_show_label = Label(image_preview_frame, image = None, pady = 20)
im_gray_show_label.grid(row=0, column = 1, sticky=W)
im_gray_show_label.bind("", lambda e: zoomImage("grayImage"))

	# Blurred Image
im_blurred_show_label = Label(image_preview_frame, image = None, pady = 20)
im_blurred_show_label.grid(row=1, column =0, sticky=W)
im_blurred_show_label.bind("", lambda e: zoomImage("blurredImage"))

	# Contour image
im_contour_show_label = Label(image_preview_frame, image = None, pady = 20)
im_contour_show_label.grid(row=1, column = 1, sticky=W)
im_contour_show_label.bind("", lambda e: zoomImage("contourImage"))

	# Zoomed image, it's hidden by default
im_zoom_label = Label(image_frame, image = tkImageContourZoom, pady = 20, width=660, height=800)
im_zoom_label.bind("", lambda e: closeZoomImage())
	

	# Image upload button, it hides after an image is uploaded
im_upload_button = Button(tab1, text="Upload image", command=uploadImage)
im_upload_button.place(relx=0.5, rely=0.5,anchor=CENTER)


# Contour detection parameter sliders and checkboxes (second column on the tab)


	# The frame holding all the paramaters
all_contour_settings_frame = Frame(tab1, width=340, height=800)
all_contour_settings_frame.grid_propagate(0)
all_contour_settings_frame.columnconfigure(0, weight=1)


# --------------- New image and refresh button frame ------------------------------
newimage_frame = Frame(all_contour_settings_frame, width=320, height=40, highlightbackground="black", highlightthickness=1, pady=5)
newimage_frame.grid(column=0, row=0)
newimage_frame.grid_propagate(0)
#Lovejoy
new_image_button = Button(newimage_frame,text="New image", command=uploadImage)
new_image_button.grid(column=0, row=0, padx=40)

recontour_button = Button(newimage_frame,text="Recontour image", command=setImages)
recontour_button.grid(column=1, row=0,padx=40)


# ----------- Spacing Frame ----------

spacing_frame = Frame(all_contour_settings_frame, width=320, height=20)
spacing_frame.grid(column=0, row=1)

#------------  Detection settings frame ------------------
detection_settings_frame = Frame(all_contour_settings_frame, width=320, height=370, highlightthickness=1, highlightbackground="black")
detection_settings_frame.grid(row=2, column=0)
detection_settings_frame.grid_propagate(0)
detection_settings_frame.columnconfigure(0, weight=1)

 	# Title of the group
detection_settings_label = Label(detection_settings_frame, text="Contour detection parameters")
detection_settings_label.grid(row=0, column=0, pady=10)
	
	# Detection minimum frame # row 0,
thresh_min_frame = Frame(detection_settings_frame, width=340, height=70)
thresh_min_frame.grid(row=1, column=0,padx=0, pady=10)
thresh_min_frame.grid_propagate(0)
thresh_min_frame.columnconfigure(0, weight=1)

thresh_min_label = Label(thresh_min_frame, text="Threshold minimum value")
	
thres_min_scale = Scale(thresh_min_frame, from_=1, to=254, orient=HORIZONTAL,length=300, variable=thresh_min, label="Detection minimum")
thres_min_scale.grid(row=1, column=0,padx=5, pady=0)

	#Detection maximum frame # row 1
thresh_max_frame = Frame(detection_settings_frame, width=340, height=70)
thresh_max_frame.grid(row=2, column=0, padx=0, pady=10)
thresh_max_frame.columnconfigure(0, weight=1)

thresh_max_label = Label(thresh_max_frame, text="Threshold maximum")

thres_max_scale = Scale(thresh_max_frame, from_=1, to=254, orient=HORIZONTAL,length=300, variable=thresh_max, label="Detection maximum")
thres_max_scale.grid(row=1, column=0,padx=5)

	# Inverse contour checkkox # row 2
inv_contour_checkbox = Checkbutton(detection_settings_frame, text="Inverse binary contours detection", variable=inv_contour, onvalue=True, offvalue=False)
inv_contour_checkbox.grid(row=3, column=0, pady=10)

	# Blur mode dropdrown # row 3
blur_mode_frame = Frame(detection_settings_frame, width=340, height=50)
blur_mode_frame.grid(row=4, column=0, padx=0, pady=10)
blur_mode_frame.columnconfigure(0, weight=1)

blur_mode_label = Label(blur_mode_frame, text="Blur type")
blur_mode_label.grid(row=0, column=0,padx=30)

blur_mode_dropdown = OptionMenu(blur_mode_frame, choosen_blur, *choices)
blur_mode_dropdown.grid(row=1, column=0, pady=0)

#----------- Spacing frame --------------

spacing_frame_2 = Frame(all_contour_settings_frame, width=320, height=20)
spacing_frame_2.grid(column=0, row=3)

#----------------------------------------



#----------- Contour area filters frame ----------------

area_thresh_frame = Frame(all_contour_settings_frame, width=320, height=40, highlightbackground="black", highlightthickness=1, pady=5, padx=5)
area_thresh_frame.columnconfigure(0, weight=1)
area_thresh_frame.grid_propagate(0)

	# Contour area filter ON/OFF chechbox
area_thresh_checkbutton = Checkbutton(area_thresh_frame, text="Contour area filter", offvalue=False, onvalue=True, variable=conture_threshold, command=hideContourThreshold)
area_thresh_checkbutton.grid(row=0, column=0,padx=5)

	# Contour area filter Maximum slider
area_thresh_min_scale = Scale(area_thresh_frame, from_=0, to=1, orient=HORIZONTAL,length=300, variable=min_contour_area, label="Min area", showvalue=False)
area_thresh_min_scale.bind("", lambda e: area_thresh_min_scale.set(area_thresh_min_scale.get() + 9 ))
area_thresh_min_scale.bind("", lambda e: area_thresh_min_scale.set(area_thresh_min_scale.get() - 9 ))
	
	# Contour area filter Minimum slider
area_thresh_max_scale = Scale(area_thresh_frame, from_=0, to=1, orient=HORIZONTAL,length=300, variable=max_contour_area, label="Max area", showvalue=False)
area_thresh_max_scale.bind("", lambda e: area_thresh_max_scale.set(area_thresh_max_scale.get() + 9 ))
area_thresh_max_scale.bind("", lambda e: area_thresh_max_scale.set(area_thresh_max_scale.get() - 9 ))

#----------- Spacing frame --------------

spacing_frame_2 = Frame(all_contour_settings_frame, width=320, height=20)
spacing_frame_2.grid(column=0, row=5)

#----------------------------------------

# --------------- Create SVG and Recontour button frame ------------------------------
separate_window_frame = Frame(all_contour_settings_frame, width=320, height=40, highlightbackground="black", highlightthickness=1, pady=5)
separate_window_frame.grid(column=0, row=6)
separate_window_frame.grid_propagate(0)
separate_window_frame.columnconfigure(0, weight=0)
separate_window_frame.rowconfigure(0, weight=1)

create_SVG_button = Button(separate_window_frame,text="Generate SVG", command=getSVG)
create_SVG_button.grid(column=0, row=0, padx=40)

recontour_button_2 = Button(separate_window_frame,text="Recontour image", command=setImages)
recontour_button_2.grid(column=1, row=0,padx=40)




#----Close button # don't really think is neccessary------------

close_button = Button(tab1, text="Close", command=root.destroy)
#close_button.grid(row=3, column=3)


# Starts program
root.mainloop()