def Event(self, event, values): # Dobotの接続を行う if event == "-Connect-": # self.connection = self.Connect_Disconnect_click(self.connection, self.api) self.connection = Connect_Disconnect(self.connection, self.api, self.CON_STR) if self.connection == 0: # Dobotの現在の姿勢を画面上に表示 self.current_pose = self.GetPose_UpdateWindow() elif event == "-SetJointPose-": # 移動後の関節角度を指定 DestPose = [ float(values["-JointPoseInput_1-"]), float(values["-JointPoseInput_2-"]), float(values["-JointPoseInput_3-"]), float(values["-JointPoseInput_4-"]), ] response = self.SetJointPose_click() print(response) # ------------------------------------ # # WebCamを選択&接続するイベント # # ------------------------------------ # elif event == "-SetWebCam-": # Webカメラの番号を取得する cam_num = WebCamOption(values["-WebCam_Name-"]) # webカメラの番号が取得できなかった場合 if cam_num is None: sg.popup("選択したデバイスは存在しません。", title="カメラ接続エラー") return # ----------------------------# # カメラを接続するイベント # # --------------------------- # # カメラを初めて接続する場合 if (cam_num != None) and (self.cam_num == None): response, self.cam = WebCam_OnOff(cam_num, cam=self.cam) if response == -1: # カメラが接続されていない場合 sg.popup("WebCameraに接続できません.", title="カメラ接続エラー") elif response == 0: # カメラを開放した場合 sg.popup("WebCameraを開放しました。", title="Camの接続") else: sg.popup("WebCameraに接続しました。", title="Camの接続") # 接続したいカメラが接続していカメラと同じ場合 elif (cam_num != None) and (self.cam_num == cam_num): response, self.cam = WebCam_OnOff(cam_num, cam=self.cam) if response == -1: # カメラが接続されていない場合 sg.popup("WebCameraに接続できません.", title="カメラ接続エラー") elif response == 0: # カメラを開放した場合 sg.popup("WebCameraを開放しました。", title="Camの接続") else: sg.popup("WebCameraに接続しました。", title="Camの接続") # 接続したいカメラと接続しているカメラが違う場合 elif (cam_num != None) and (self.cam_num != cam_num): # まず接続しているカメラを開放する. ch_1, self.cam = WebCam_OnOff(cam_num, cam=self.cam) # 開放できた場合 if ch_1 == 0: sg.popup("WebCameraを開放しました。", title="Camの接続") # 次に新しいカメラを接続する. ch_2, self.cam = WebCam_OnOff(cam_num, cam=self.cam) # カメラが接続されていない場合 if ch_2 == -1: sg.popup("WebCameraに接続できません.", title="カメラ接続エラー") else: sg.popup("WebCameraに接続しました.", title="Camの接続") # 新しいカメラを接続した場合 elif ch_1 == 1: sg.popup("新しくWebCameraに接続しました.", title="Camの接続") # カメラの接続に失敗した場合 else: self.cam = None self.cam_num = cam_num #---------------------------------# # プレビューを表示するイベント # #-------------------------------- # elif event == "-Preview-": window_name = "frame" while True: if type(self.cam) == cv2.VideoCapture: # カメラが接続されている場合 response, img = Snapshot(self.cam) if response: #----------------------------------------# # 画像サイズなどをダイアログ上に表示 # #----------------------------------------# self.Window["-IMAGE_width-"].update(str(img.shape[0])) self.Window["-IMAGE_height-"].update(str(img.shape[1])) self.Window["-IMAGE_channel-"].update(str( img.shape[2])) print("プレビューを表示します。") response = Preview(img, window_name=window_name) if cv2.waitKey(0) & 0xFF == ord("e"): cv2.destroyWindow(window_name) break else: sg.popup("SnapShotを撮影できませんでした.", title="撮影エラー") break else: sg.popup("カメラが接続されていません.", title="カメラ接続エラー") break # ---------------------------------------- # # スナップショットを撮影するイベント # # ---------------------------------------- # elif event == '-Snapshot-': response, img = Snapshot(self.cam) if response != 1: sg.popup("スナップショットを撮影できませんでした。", title="撮影エラー") #----------------------------------------# # 画像サイズなどをダイアログ上に表示 # #----------------------------------------# self.Window["-IMAGE_width-"].update(str(img.shape[0])) self.Window["-IMAGE_height-"].update(str(img.shape[1])) self.Window["-IMAGE_channel-"].update(str(img.shape[2])) # --------------------------- # 撮影した画像を変換する。 # --------------------------- # ノイズ除去 # 濃度変換 if values['-Color_Density-'] != 'なし': img = Contrast_cvt(img, values['-Color_Density-']) # ----------------------------------- # 画面上に撮影した画像を表示する # ----------------------------------- img = scale_box(img, self.Image_width, self.Image_height) imgbytes = cv2.imencode('.png', img)[1].tobytes() self.Window['-IMAGE-'].update(data=imgbytes) # ------------------------------------------------ # 画面上に撮影した画像のヒストグラムを表示する # ------------------------------------------------ canvas_elem = self.Window['-CANVAS-'] canvas = canvas_elem.TKCanvas ticks = [0, 42, 84, 127, 169, 211, 255] fig, ax = plt.subplots(figsize=(3, 2)) ax = Image_hist(img, ax, ticks) if self.fig_agg: # ** IMPORTANT ** Clean up previous drawing before drawing again delete_figure_agg(self.fig_agg) self.fig_agg = draw_figure(canvas, fig) self.fig_agg.draw()
def SnapshotBtn(self, values: list) -> np.ndarray: """スナップショットの撮影から一連の画像処理を行う関数。 Args: values (list): ウインドウ上のボタンの状態などを記録している変数 Returns: dst_org (np.ndarray): オリジナルのスナップショット dst_bin (np.ndarray): 二値化処理後の画像 """ dst_org = dst_bin = None response, img = Snapshot(self.cam) if response != 3: sg.popup(self.WebCam_err[response], title="撮影エラー") return dst_org = img.copy() #--------------------------- # # 画像サイズなどをダイアログ上に表示 # #--------------------------- # self.Window["-IMAGE_width-"].update(str(img.shape[0])) self.Window["-IMAGE_height-"].update(str(img.shape[1])) self.Window["-IMAGE_channel-"].update(str(img.shape[2])) # --------------------------- # 撮影した画像を変換する。 # --------------------------- # 色空間変換 if values['-Color_Space-'] != 'RGB': img = Color_cvt(img, values['-Color_Space-']) # ノイズ除去 # 濃度変換 if values['-Color_Density-'] != 'なし': img = Contrast_cvt(img, values['-Color_Density-']) # ------------ # 画像処理 # ------------ if values['-Binarization-'] != 'なし': # 二値化処理 if values['-Binarization-'] == 'Global': # 大域的二値化処理 img = GlobalThreshold(img, threshold=int( values["-LowerThreshold-"])) elif values['-Binarization-'] == 'Otsu': # 大津の二値化処理 img = GlobalThreshold(img, Type='Otsu') elif values['-Binarization-'] == 'Adaptive': img = AdaptiveThreshold( img=img, method=str(values['-AdaptiveThreshold_type-']), block_size=int(values['-AdaptiveThreshold_BlockSize-']), C=int(values['-AdaptiveThreshold_Constant-'])) elif values['-Binarization-'] == 'Two': # 2つの閾値を用いた二値化処理 # ピックアップする色を番号に変換 if values['-color_R-']: color = 0 elif values['-color_G-']: color = 1 elif values['-color_B-']: color = 2 elif values['-color_W-']: color = 3 elif values['-color_Bk-']: color = 4 img = TwoThreshold( img=img, LowerThreshold=int(values["-LowerThreshold-"]), UpperThreshold=int(values["-UpperThreshold-"]), PickupColor=color) dst_bin = img.copy() # ----------------------- # # 画面上に撮影した画像を表示する # # ----------------------- # img = scale_box(img, self.Image_width, self.Image_height) imgbytes = cv2.imencode('.png', img)[1].tobytes() self.Window['-IMAGE-'].update(data=imgbytes) # --------------------------------- # # 画面上に撮影した画像のヒストグラムを表示する # # --------------------------------- # canvas_elem = self.Window['-CANVAS-'] canvas = canvas_elem.TKCanvas ticks = [0, 42, 84, 127, 169, 211, 255] fig, ax = plt.subplots(figsize=(3, 2)) ax = Image_hist(img, ax, ticks) if self.fig_agg: # ** IMPORTANT ** Clean up previous drawing before drawing again delete_figure_agg(self.fig_agg) self.fig_agg = draw_figure(canvas, fig) self.fig_agg.draw() return dst_org, dst_bin