def next_file(self):
self.__file_number += 1
if self.__file_number == len(self.__file_list) - 1:
wx.PostEvent(self.__gui, ResultEvent(Events.EVT_NEXT_DONE_ID, 1))
if self.__file_number == len(self.__file_list):
wx.PostEvent(self.__gui, ResultEvent(Events.EVT_NEXT_DONE_ID, 0))
self.__file_number = 0
try:
self.__file_path = self.__file_list[self.__file_number]
except ImportError:
pass
def run(self):
while self.__running:
if self.__start:
try:
img = cv2.imread(self.__file_path)
if self.__enable:
img_res = processing_image(img)
else:
img_res = img
if self.__save:
self._save_image(self._get_save_path(self.__file_path),
img_res)
else:
pass
height, width = img_res.shape[:2]
img_res = cv2.cvtColor(img_res, cv2.COLOR_BGR2RGB)
bmp = wx.ImageFromBuffer(width, height, img_res)
wx.PostEvent(self.__gui,
ResultEvent(Events.EVT_DISPLAY_ID, bmp))
except Exception:
pass
self.Stop()
else:
pass
time.sleep(0.1)
def run(self):
while self.__running:
if self.__start:
if self.__cap.isOpened():
ret, frame = self.__cap.read()
if ret:
if self.__enable:
frame_res = processing_image(frame)
else:
frame_res = frame
if self.__save:
self.__out.write(frame_res)
else:
pass
height, width = frame_res.shape[:2]
frame_res = cv2.cvtColor(frame_res, cv2.COLOR_BGR2RGB)
bmp = wx.ImageFromBuffer(width, height, frame_res)
wx.PostEvent(self.__gui,
ResultEvent(Events.EVT_DISPLAY_ID, bmp))
else:
self.Stop()
else:
pass
time.sleep(1 / int(self.__fps))
else:
time.sleep(0.1)
def Stop(self):
self.__start = False
wx.PostEvent(self.__gui,
ResultEvent(Events.EVT_START_STATE_ID, 0))
def Start(self):
self.__start = True
wx.PostEvent(self.__gui,
ResultEvent(Events.EVT_START_STATE_ID, 1))
def Stop(self):
self._stop_out()
self._stop_cap()
wx.PostEvent(self.__gui, ResultEvent(Events.EVT_START_STATE_ID, 0))
self.__start = False
def Start(self):
self._start_cap()
if self.__save:
self._start_out()
wx.PostEvent(self.__gui, ResultEvent(Events.EVT_START_STATE_ID, 1))
self.__start = True