from tkinter import *
from tkinter import filedialog
import tkinter.messagebox
import cv2 as cv
from cv2 import *
import inspection as myModule
import sys
def nothing(x):
x
def do_nothing():
print("Doing absolutely nothing")
def set_ROI(filepath):
global ref_point
ref_point = []
def shape_selection(event, x, y, flags, param):
global ref_point
if event == cv2.EVENT_LBUTTONDOWN:
ref_point = [(x, y)]
elif event == cv2.EVENT_LBUTTONUP:
ref_point.append((x, y))
cv2.rectangle(img, ref_point[0], ref_point[1], (0, 255, 0), 2)
img = cv.imread(filepath)
cv.namedWindow("Set ROI")
cv.setMouseCallback("Set ROI", shape_selection)
while 1:
k = cv.waitKey(1)
if k!=-1:
break
cv.imshow("Set ROI", img)
cv.destroyAllWindows()
global x_min, x_max, y_min, y_max
if ref_point!=[]:
x_min = ref_point[0][0]
x_max = ref_point[1][0]
y_min = ref_point[0][1]
y_max = ref_point[1][1]
else:
x_min = -1
x_max = -1
y_min = -1
y_max = -1
def canny_threshold(smooth_size, filepath):
img = cv.imread(filepath)
imgray = cv.cvtColor(img, cv.COLOR_BGR2GRAY)
filter_img = cv.medianBlur(imgray, smooth_size)
cv.namedWindow("CANNY")
cv.createTrackbar("MIN", "CANNY", 0, 255, nothing)
cv.createTrackbar("MAX", "CANNY", 0, 255, nothing)
while 1:
k = cv.waitKey(1)
if k != -1:
break
t1 = cv.getTrackbarPos("MIN", "CANNY")
t2 = cv.getTrackbarPos("MAX", "CANNY")
current_img = cv.Canny(filter_img, t1, t2)
cv.imshow("CANNY", current_img)
cv.destroyAllWindows()
global min_threshold
global max_threshold
min_threshold = t1
max_threshold = t2
def read_settings():
with open("settings\\program_settings.txt", "r") as f:
settings = f.readlines()
settings = [x.strip() for x in settings]
output = []
for line in settings:
parts = line.split(';')
output.append(parts)
return output
def load_image(path):
global filepath
filepath=path
print("Loading image: "+filepath)
img = PhotoImage(file=filepath)
display_image.my_image=img
root_width=img.width()+40
root_height=img.height()+40+24+20
root_posX=int(round(root.winfo_screenwidth()/2))-int(round(root_width/2))
root_posY=int(round(root.winfo_screenheight()/2))-int(round(root_height/2))-20
root_geometry = str(root_width)+"x"+str(root_height)+"+"+str(root_posX)+"+"+str(root_posY)
root.geometry(root_geometry)
display_image.config(image=display_image.my_image)
button_settings.config(state=NORMAL)
button_run.config(state=NORMAL)
# def tool_load_image():
# window_load_image = Tk()
# window_load_image.geometry("240x100+550+325")
# window_load_image.title("Load image")
#
# path = 'images\\'
# path_display = Label(window_load_image, text=path)
# path_display.grid(row=0, column=0, pady=15, sticky=E)
#
# entry_load_image = Entry(window_load_image)
# entry_load_image.insert(END, '')
# entry_load_image.grid(row=0, column=1, pady=15, sticky=W)
#
# button_load_function = Button(window_load_image, text="Load",
# command=lambda: load_image(path + entry_load_image.get()), width=15)
# button_load_function.grid(columnspan=2, pady=5, padx=65)
#
# status.config(text="Adjust settings or run inspection")
# window_load_image.mainloop()
def tool_load_image():
path = filedialog.askopenfilename(initialdir="images\\", title = "Select TOFD image",
filetypes=[("Image", "*.png")])
load_image(path)
status.config(text="Adjust settings or run inspection")
def tool_run():
print("Inspection running...")
settings = read_settings()
settings_names=[]
settings_values=[]
for i in range(len(settings)):
settings_names.append(settings[i][0])
for j in range(1,len(settings[i])):
settings_values.append(settings[i][j])
for i in range(len(settings_values)):
settings_values[i]=int(settings_values[i])
global x, y, w, h, images
acceptance, x,y,w,h, filepath_inspected, images = \
myModule.inspect(filepath, settings_values, 3, 12)
if images == 0:
status.config(text="Try loading file again")
else:
status.config(text=" Main defect size: "+str(w)+" x "+str(h) + " / Coordinates: ("+str(x)+";"+str(y)+")"
+ " / Acceptance: " + str(acceptance))
load_image(filepath_inspected)
button_display_process.config(state=NORMAL)
def tool_settings():
def save_on_closing():
if_save = tkinter.messagebox.askquestion("Quit", "Do you want to save these settings?")
if if_save=="yes":
print("Saving settings...")
with open("settings\\program_settings.txt", "w") as f:
f.write(label_0["text"]+";"+str(setting_0_return.get())+"\n")
try:
f.write(label_1["text"] + ";" + str(min_threshold) + ";" + str(max_threshold) + "\n")
except:
print(sys.exc_info()[0])
print(label_1["text"]+": Setting default value: 20 - 77")
f.write(label_1["text"] + ";" + "20;77\n")
f.write(label_2["text"] + ";" + str(setting_2_return.get()) + "\n")
f.write(label_3["text"] + ";" + setting_3.get() + "\n")
try:
f.write(label_4["text"] + ";" + str(x_min) + ";" + str(x_max) + ";" + str(y_min) + ";" + str(y_max) + "\n")
except:
print(sys.exc_info()[0])
print(label_4["text"]+": Setting dynamic region of interest")
f.write(label_4["text"] + ";" + "-1;-1;-1;-1\n")
tkinter.messagebox.showinfo("Quit", "Settngs saved in:\nsettings\\program_settings.txt")
window_settings.destroy()
print("Accessing settings...")
settings = read_settings()
settings_names = []
settings_values = []
for i in range(len(settings)):
settings_names.append(settings[i][0])
for j in range(1, len(settings[i])):
settings_values.append(settings[i][j])
print(settings_values)
window_settings = Tk()
window_settings.geometry("400x200+450+280")
window_settings.title("Settings")
label_title_default = Label(window_settings, text="Default settings", font='Helvetica 10 bold')
label_0 = Label(window_settings, text="Filter size")
label_1 = Label(window_settings, text="Canny threshold")
label_2 = Label(window_settings, text="Morphology - kernel size")
label_3 = Label(window_settings, text="Morphology - iterations")
label_4 = Label(window_settings, text="Region of interest")
label_title_current = Label(window_settings, text="Current settings", font='Helvetica 10 bold')
label_0_1 = Label(window_settings, text=str(settings_values[0]))
label_1_1 = Label(window_settings, text=str(settings_values[1])+"-"+str(settings_values[2]))
label_2_1 = Label(window_settings, text=str(settings_values[3]))
label_3_1 = Label(window_settings, text=str(settings_values[4]))
if settings_values[5]=="-1" and settings_values[6]=="-1" and settings_values[7]=="-1" and settings_values[8]=="-1":
label_4_1 = Label(window_settings,text="Dynamic")
else:
label_4_1 = Label(window_settings, text="x: "+str(settings_values[5])+" - "+str(settings_values[6])+" ; y: "+str(settings_values[7])+" - "+str(settings_values[8]))
label_title_default.grid(row=0,columnspan=2)
label_0.grid(row=1, column=0, sticky=E)
label_1.grid(row=2, column=0, sticky=E)
label_2.grid(row=3, column=0, sticky=E)
label_3.grid(row=4, column=0, sticky=E)
label_4.grid(row=5, column=0, sticky=E)
label_title_current.grid(row=0, column=2)
label_0_1.grid(row=1, column=2)
label_1_1.grid(row=2, column=2)
label_2_1.grid(row=3, column=2)
label_3_1.grid(row=4, column=2)
label_4_1.grid(row=5, column=2)
filterType = [3, 5, 7, 9]
setting_0_return = IntVar(window_settings)
setting_0_return.set(filterType[3])
setting_0 = OptionMenu(window_settings, setting_0_return, *filterType)
setting_0.grid(row=1, column=1)
setting_1 = Button(window_settings, text="SET", width=15, command=lambda: canny_threshold(setting_0_return.get(), filepath))
setting_1.grid(row=2, column=1)
morphologyKernel = [3, 5, 7, 9]
setting_2_return = IntVar(window_settings)
setting_2_return.set(morphologyKernel[0])
setting_2 = OptionMenu(window_settings, setting_2_return, *morphologyKernel)
setting_2.grid(row=3, column=1, sticky=W)
setting_3 = Entry(window_settings)
setting_3.insert(END, "3")
setting_3.grid(row=4, column=1, sticky=W)
setting_4_current = Label(window_settings, text="Current ROI:")
setting_4_current.grid(row=5, column=1)
setting_4 = Button(window_settings, text="SET", width=15, command=lambda:set_ROI(filepath))
setting_4.grid(row=6, column=1, pady=5)
window_settings.protocol("WM_DELETE_WINDOW", save_on_closing)
window_settings.mainloop()
root = Tk()
root.title("Program")
root.geometry("500x500+400+100")
toolbar = Frame(root, bd=1, relief=RIDGE)
toolbar.pack(side=TOP, fill=X)
inspection_frame = Frame(root)
inspection_frame.pack(side=BOTTOM, fill=X)
status = Label(inspection_frame, text="Load image to inspect...", bd=1, relief=SUNKEN, anchor=W, font="Calibri 12", height=2) # bd-border, relief-how border appears, anchor-where
status.pack(side=BOTTOM, fill=X)
button_load = Button(toolbar, text="Load image", command=tool_load_image)
button_load.pack(side=LEFT, padx=2, pady=2)
button_settings = Button(toolbar, text="Settings", command=tool_settings)
button_settings.pack(side=LEFT, padx=2,pady=2)
button_run = Button(toolbar, text="Run", state=DISABLED, command=tool_run)
button_run.pack(side=LEFT, padx=2, pady=2)
button_display_process = Button(toolbar, text="Show process", state=DISABLED, command=lambda:myModule.display_process(images))
button_display_process.pack(side=LEFT, padx=2, pady=2)
display_image = Label(inspection_frame)
display_image.pack(padx=20, pady=20)
root.mainloop()
