class Camera():
def __init__(self,
capture=cv2.VideoCapture,
width=800,
height=600,
label=None,
label_name=None):
# Get the capture object from the MainWindow init.
self.capture = capture
self.capture.set(3, int(width))
self.capture.set(4, int(height))
self.width = width
self.height = height
self.label = label
self.label_name = label_name
self.currentFrame = np.array([])
self.logger = MyLogging(logger_name='user').logger
self.logger.info('Create [Capture]: %s [Label_name]: %s' %
(self.capture, self.label_name))
def getFrame(self):
try:
# Get frame and convert it to PixMap
ret, img = self.capture.read()
img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
# Add the hisself Cap object's label name in pic top.
if self.label_name:
cv2.putText(img, self.label_name, (18, 56), 0, 1,
(129, 216, 207), 3)
# Get the height, width, byserPer from this img.
height, width, bytesPer = img.shape
# bytesPerLine = bytesPer*3
# Convert the img to QPixmap type, because the QtLabel just
# accept this type.
img = QImage(img, width, height, QImage.Format_RGB888)
img = QPixmap.fromImage(img)
# Set label show img. And update it.
self.label.setPixmap(img)
except:
self.logger.error('Some error happened in <cameraModule.py | '
'getFrame>.')
sys.exit(0)
def refresh(self):
# 1. Create a new Thread Class and Init it.
self.cap_timer = Camera_Timer()
self.logger.info("Camera_Timer: [%s] has created." % self.cap_timer)
# 2. Connect the thing let signal to do it.
self.cap_timer.update.connect(self.getFrame)
# 3. Start Thread.
try:
self.cap_timer.start()
except BaseException:
self.cap_timer.stop()
self.cap_timer.quit()
def quit(self):
self.cap_timer.stop()
self.cap_timer.quit()
class Save_img_Timer(QThread):
"""
Send_msg will emit the capture process to capWindow's textBroswer.
Creat_dirs_msg will emit the new creating dirs to capWindow's textBroswer.
:param
parent: It is the who create or init The Save_img_Timer class.
cap_Objects: a list of all avaliable Cameras on PC.
time: The time of capturing 32 pics to spend.
save_dirs: a list of to save dirs to save img by needing.
img_names: a list of how to save img with your name.
"""
send_msg = pyqtSignal(str)
creat_dirs_msg = pyqtSignal(str)
def __init__(self, parent=None, cap_Objects=None, time=float):
super(Save_img_Timer, self).__init__()
self.parent = parent
self.cap_Objects = cap_Objects
self.time = time / 32
self.save_dirs = []
self.img_names = []
self.logger = MyLogging(logger_name='user').logger
for cap_Object in self.cap_Objects:
self.save_dirs.append(self.parent.save_path + self.parent.cvid + '/'
+ cap_Object.label_name + '/' + \
self.parent.direction + '/')
self.img_names.append(self.parent.cvid + '_' + self.parent.char + '_' + \
self.parent.date + '_' + cap_Object.label_name +\
'_' + self.parent.direction)
for save_dir in self.save_dirs:
if not os.path.exists(save_dir):
os.makedirs(save_dir)
self.logger.info('Create dir [%s]' % save_dir)
msg = 'Create dir [%s]' % save_dir
self.creat_dirs_msg.connect(self.parent.update_textBrowser)
self.creat_dirs_msg.emit(msg)
self.index = 1
self.isStop = False
def run(self):
msg = '[CVID]: %s [DIRECTION]: %s [ACTIVED]: Start.' % (
self.parent.cvid, self.parent.direction)
self.logger.info(msg)
self.send_msg.emit(msg)
num = len(self.cap_Objects)
while not self.isStop:
self.imgs = []
self.file_names = []
for i in range(num):
ret, img = self.cap_Objects[i].capture.read()
file_name = self.img_names[i] + '_' + str(self.index) + '.jpg'
self.imgs.append(img)
self.file_names.append(file_name)
for i in range(num):
cv2.imwrite(self.save_dirs[i] + self.file_names[i],
self.imgs[i])
msg = '[Cvid]: %s [Direction]: %s | ----------> %d\n' % (
self.parent.cvid, self.parent.direction, self.index)
self.send_msg.emit(msg)
self.index += 1
time.sleep(self.time)
if self.index > 32:
self.isStop = True
msg = '[CVID]: %s [DIRECTION]: %s [ACTIVED]: End.' % (
self.parent.cvid, self.parent.direction)
self.logger.info(msg)
self.send_msg.emit(msg)
self.parent.ui.cap_ok_Button.setEnabled(True)
break