Downloads: 1
Diff of
/program.py
[000000]
..
[ec4a1d]
Switch to side-by-side view
--- a +++ b/program.py @@ -0,0 +1,277 @@ +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()