# -*- coding: utf-8 -*-
# @File : main.py
# @Author: CSD
# @Date : 2019/9/7 0007 20:48
# @Software: PyCharm
import gc
import os
import sys
import time
import cv2
import qimage2ndarray
from PyQt5.QtCore import QTimer, Qt, QThread, pyqtSignal
from PyQt5.QtGui import QPixmap
from PyQt5.QtWidgets import QApplication, QMainWindow, QDialog, QAbstractItemView, QTableWidgetItem, QMessageBox
from Mainwindow import Ui_MainWindow
from db import mysql_conn
from face_register import register_handler
from face_search import search_handler
from late import Ui_Table
from register import Ui_Register
from cnn_net import CnnNet
from sklearn.model_selection import train_test_split
import numpy as np
from getimgdata import GetImgData
import captureface
from captureface import CaptureFace
import threading
from cv2ImgAddText import cv2ImgAddText
imgs, labels, number_name = GetImgData().readimg() # 讀取數(shù)據(jù)
train_x, test_x, train_y, test_y = train_test_split(imgs, labels, test_size=0.1, random_state=10) # 訓(xùn)練集測(cè)試集數(shù)據(jù)劃分
cnnnet = CnnNet() # 調(diào)用CNN算法類
# 主窗口
class parentWindow(QMainWindow, Ui_MainWindow):
 def __del__(self):
 try:
 self.camera.release() # 釋放資源
 except:
 return
 def __init__(self, parent=None):
 super(parentWindow, self).__init__(parent)
 self.setupUi(self)
 self.Latebt.setEnabled(False)
 # self.PrepCamera()
 self.CallBackFunctions()
 self.Timer = QTimer() # 實(shí)例化一個(gè)類
 self.Timer.timeout.connect(self.TimerOutFun) # 定時(shí)刷新
 self.flagThread = FlagThread()
 # 當(dāng)獲得循環(huán)完畢的信號(hào)時(shí)，停止計(jì)數(shù)
 self.flagThread.trigger.connect(self.fun_timer)
 # 初始化攝像頭
 def PrepCamera(self):
 try:
 self.camera = cv2.VideoCapture(0)
 self.camera.set(cv2.CAP_PROP_FOURCC, cv2.VideoWriter_fourcc('M', 'J', 'P', 'G'))
 self.MsgTE.clear()
 self.MsgTE.append('攝像頭已連接！')
 except Exception as e:
 self.MsgTE.clear()
 self.MsgTE.append(str(e))
 # 回調(diào)函數(shù)
 def CallBackFunctions(self):
 self.Showbt.clicked.connect(self.StartCamera)
 self.About.clicked.connect(self.about)
 # 開(kāi)始按鈕函數(shù)
 def StartCamera(self):
 self.Showbt.setEnabled(False)
 self.Loginbt.setEnabled(False)
 self.Latebt.setEnabled(True)
 self.PrepCamera()
 self.Timer.start(1) # 每隔1ms刷新一次
 self.timelb = time.clock()
 self.flagThread.start()
 # 從攝像頭讀取圖像
 def TimerOutFun(self):
 success, img = self.camera.read()
 if success:
 self.Image = img
 self.DispImg()
 else:
 self.MsgTE.clear()
 self.MsgTE.setPlainText('攝像頭讀取圖像已暫停！')
 def fun_timer(self, flag):
 self.MsgTE.setPlainText(flag)
 # 檢測(cè)人臉、色彩空間及格式轉(zhuǎn)換
 def DispImg(self):
 results = captureface.detect_face(self.Image)
 if results is not ():
 faceboxes = results # 提取人臉位置信息
 for (x, y, w, h) in faceboxes:
 face = self.Image[y:y + h, x:x + w] # 截取圖片中的人臉圖像
 face = cv2.cvtColor(face, cv2.COLOR_BGR2GRAY) # 轉(zhuǎn)為灰度圖片
 face = cv2.resize(face, (64, 64)) # 壓縮成指定大小
 face = face.reshape([1, 64, 64, 1])
 cnnNet = CnnNet(modelfile='./temp/train-model') # 注意這步很關(guān)鍵，起到了重置計(jì)算圖的作用，否則多次導(dǎo)入訓(xùn)練好的計(jì)算圖會(huì)出現(xiàn)tensor重復(fù)的問(wèn)題
 res, pre = cnnNet.predict(test_x=face) # 調(diào)用已訓(xùn)練好的模型進(jìn)行預(yù)測(cè)
 if np.max(pre) < 0.8: # 通過(guò)調(diào)整閾值為threshold，當(dāng)返回?cái)?shù)組最大值小于threshold是即視為unknown
  self.name = "unknown"
 else:
  self.name = number_name[res[0]]
  self.flagThread.get_name(self.name)
 # cv2.putText(self.Image, self.name, (int(x), int(y) - 20), cv2.FONT_HERSHEY_SIMPLEX, 1, 255, 2)
 cv2.rectangle(self.Image, (int(x), int(y)), (int(x + w), int(y + h)), (255, 0, 0), 3) # 將name顯示出來(lái)
 img = cv2ImgAddText(self.Image, self.name, int(x+25), int(y-50))
 img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
 qimg = qimage2ndarray.array2qimage(img) # 調(diào)用array2qimage函數(shù)將其轉(zhuǎn)為QImage格式
 self.DispLb.setPixmap(QPixmap(qimg)) # 再通過(guò)QPixmap函數(shù)轉(zhuǎn)為QPixmap格式進(jìn)行顯示。
 self.DispLb.show() # 圖像顯示
 gc.collect()
 else:
 img = cv2.cvtColor(self.Image, cv2.COLOR_BGR2RGB)
 qimg = qimage2ndarray.array2qimage(img) # 調(diào)用array2qimage函數(shù)將其轉(zhuǎn)為QImage格式
 self.DispLb.setPixmap(QPixmap(qimg)) # 再通過(guò)QPixmap函數(shù)轉(zhuǎn)為QPixmap格式進(jìn)行顯示。
 self.DispLb.show() # 圖像顯示
 # 幫助
 def about(self):
 reply = QMessageBox.information(self,
   "關(guān)于",
   "1.點(diǎn)擊開(kāi)始按鈕播放視頻流,進(jìn)行簽到\n"
   "2.點(diǎn)擊注冊(cè)按鈕可跳往注冊(cè)窗口進(jìn)行注冊(cè)\n"
   "3.點(diǎn)擊缺勤按鈕可查看缺勤學(xué)生名單，狀態(tài)0代表缺勤\n"
   "4.點(diǎn)擊幫助按鈕可查看幫助，同時(shí)關(guān)閉主窗口視頻流,進(jìn)行人臉注冊(cè)",
   QMessageBox.Close)
 try:
 self.camera.release() # 釋放資源
 self.Showbt.setEnabled(True)
 self.Loginbt.setEnabled(True)
 except:
 return
# 注冊(cè)窗口
class childWindow(QDialog, Ui_Register):
 def __init__(self, parent=None):
 super(childWindow, self).__init__(parent)
 self.setupUi(self)
 self.PhotoBt.setEnabled(False)
 self.RegisterBt.setEnabled(False)
 # self.PrepCamera()
 self.CallBackFunctions()
 self.workThread = WorkThread()
 self.photoThread = PhotoThread()
 # 當(dāng)獲得循環(huán)完畢的信號(hào)時(shí)，停止計(jì)數(shù)
 self.workThread.trigger.connect(self.timeStop)
 self.photoThread.trigger.connect(self.runover)
 self.Timer = QTimer() # 實(shí)例化一個(gè)類
 self.Timer.timeout.connect(self.TimerOutFun) # 定時(shí)刷新
 # 初始化攝像頭并打開(kāi)
 def PrepCamera(self):
 try:
 self.camera = cv2.VideoCapture(0)
 self.MsgLb.clear()
 self.MsgLb.setText('請(qǐng)輸入學(xué)號(hào)、姓名,點(diǎn)擊拍照!')
 except Exception as e:
 self.MsgLb.clear()
 self.MsgLb.setText(str(e))
 # 回調(diào)函數(shù)
 def CallBackFunctions(self):
 self.PhotoBt.clicked.connect(self.PhotoCamera)
 self.StartBt.clicked.connect(self.StartCamera)
 self.RegisterBt.clicked.connect(self.RegisterCamera)
 self.ModelBt.clicked.connect(self.ModelTrain)
 # 開(kāi)始按鈕函數(shù)
 def StartCamera(self):
 self.PhotoBt.setEnabled(True)
 self.RegisterBt.setEnabled(True)
 self.PrepCamera()
 self.Timer.start(1) # 每隔1ms刷新一次
 self.timelb = time.clock()
 # 注冊(cè)
 def RegisterCamera(self):
 Msg = register_handler(self.stu_id.text(), self.stu_name.text())
 self.MsgLb.setText(Msg)
 # 拍照按鈕函數(shù)
 def PhotoCamera(self):
 if self.stu_id.text().isdigit():
 self.RecordCamera() # 保存照片
 self.stu_id.clear()
 self.stu_name.clear()
 else:
 self.MsgLb.setText('請(qǐng)檢查學(xué)號(hào)是否正確')
 # 從攝像頭讀取圖像
 def TimerOutFun(self):
 success, img = self.camera.read()
 if success:
 self.Image = img
 self.DispImg()
 else:
 self.MsgLb.clear()
 self.MsgLb.setText('攝像頭讀取圖像已暫停！')
 # 色彩空間及格式轉(zhuǎn)換
 def DispImg(self):
 img = cv2.cvtColor(self.Image, cv2.COLOR_BGR2RGB)
 qimg = qimage2ndarray.array2qimage(img) # 調(diào)用array2qimage函數(shù)將其轉(zhuǎn)為QImage格式
 self.ShowLb.setPixmap(QPixmap(qimg)) # 再通過(guò)QPixmap函數(shù)轉(zhuǎn)為QPixmap格式進(jìn)行顯示。
 self.ShowLb.show() # 圖像顯示
 # 采集照片
 def RecordCamera(self):
 self.MsgLb.setText('正在采集圖片')
 self.photoThread.img(self.Image, self.stu_name.text())
 self.photoThread.start()
 def runover(self, msg, i):
 self.MsgLb.setText(msg)
 self.progressBar.setValue(i)
 # 模型訓(xùn)練
 def ModelTrain(self):
 self.MsgLb.setText('請(qǐng)等待一段時(shí)間進(jìn)行模型訓(xùn)練!')
 self.workThread.start()
 def timeStop(self, msg, i):
 self.MsgLb.setText(msg)
 self.progressBar.setValue(i)
 def closeEvent(self, event):
 self.camera.release()
# 缺勤窗口
class childWindow_late(QDialog, Ui_Table):
 def __init__(self, parent=None):
 super(childWindow_late, self).__init__(parent)
 self.setupUi(self)
 self.CallBackFunctions()
 self.tableWidget.setEditTriggers(QAbstractItemView.NoEditTriggers)
 # 回調(diào)函數(shù)
 def CallBackFunctions(self):
 self.ResetBt.clicked.connect(self.ResetTable)
 self.ShowBt.clicked.connect(self.ShowTable)
 # 顯示缺勤表格
 def ShowTable(self):
 sql = 'select * from users where state=0'
 print(sql)
 cursor = mysql_conn.cursor()
 cursor.execute(sql)
 results = cursor.fetchall()
 print(results)
 if results:
 row = cursor.rowcount
 vol = len(results[0])
 self.tableWidget.setRowCount(row)
 self.tableWidget.setColumnCount(3)
 for i in range(row):
 for j in range(3):
  temp_data = results[i][j + 1] # 臨時(shí)記錄，不能直接插入表格
  print(temp_data)
  data = QTableWidgetItem(str(temp_data)) # 轉(zhuǎn)換后可插入表格
  data.setTextAlignment(Qt.AlignHCenter | Qt.AlignVCenter)
  self.tableWidget.setItem(i, j, data)
 # 重置
 def ResetTable(self):
 reply = QMessageBox.question(self,
   "消息",
   "重置后所有的學(xué)生的狀態(tài)將更改為0，是否修改？",
   QMessageBox.Yes | QMessageBox.No)
 if reply == QMessageBox.Yes:
 sql = 'update users set state = 0'
 print(sql)
 cursor = mysql_conn.cursor()
 cursor.execute(sql)
 mysql_conn.commit()
 else:
 return
class WorkThread(QThread):
 trigger = pyqtSignal(str, int)
 def __int__(self):
 super(WorkThread, self).__init__()
 def run(self):
 time.sleep(2)
 path = './small_img_gray' # 灰度圖路徑
 imgs, labels, number_name = GetImgData(dir=path).readimg()
 x_train, x_test, y_train, y_test = train_test_split(imgs, labels, test_size=0.2)
 cnnNet = CnnNet(modelfile='./temp/train-model',
  imgs=x_train, labels=y_train)
 train_class = cnnNet.cnnTrain(maxiter=1000, # 最大迭代次數(shù)
   accu=0.99, ) # 指定正確率（499次之后）
 for index, out in enumerate(train_class):
 # 循環(huán)完畢后發(fā)出信號(hào)
 self.trigger.emit(out, index * 10)
 self.trigger.emit("訓(xùn)練完成", 100)
 def kill_thread(self):
 self.terminate()
class PhotoThread(QThread, Ui_Register):
 trigger = pyqtSignal(str, int)
 def __int__(self):
 super(PhotoThread, self).__init__()
 def img(self, img, stu_name):
 self.Image = img
 self.stu_name = stu_name
 def run(self):
 filepath = os.path.join('./test_img/', self.stu_name) # 路徑拼接
 if not os.path.exists(filepath): # 看是否需要?jiǎng)?chuàng)建路徑
 os.makedirs(filepath)
 for i in range(100): # 開(kāi)始拍照
 savePath = (filepath + "/%d.jpg" % i)
 cv2.imencode('.jpg', self.Image, [cv2.IMWRITE_JPEG_QUALITY, 100])[1].tofile(savePath) # 保存圖片
 # 無(wú)法寫入中文 搜了搜發(fā)現(xiàn)OpenCV的imwrite不支持 所以換成了cv2.imencode
 # picturepath = os.path.join(filepath, str(i)) + '.jpg' # 圖片的完整路徑名 無(wú)法讀取中文路徑
 # cv2.imwrite(picturepath, self.Image, [cv2.IMWRITE_JPEG_QUALITY, 100]) # 將圖片寫入指定d路徑
 self.trigger.emit('正在采集圖片~', i)
 picture = CaptureFace(
 imgdir='./test_img/', # 采集到的圖片存放位置
 grayfacedir='./small_img_gray' # 灰度處理后的圖片存放位置
 )
 picture.facetogray(someone=self.stu_name, waitkey=100, size=64)
 self.trigger.emit('圖片采集完畢！點(diǎn)擊注冊(cè)', 100)
class FlagThread(QThread):
 trigger = pyqtSignal(str)
 def __int__(self):
 super(FlagThread, self).__init__()
 def get_name(self, name=''):
 self.name = name
 def log_in(self):
 if self.name != '':
 user_id, user_name, user_state = search_handler(self.name)
 if user_state == 1:
 self.Msg = '%s %s 簽到成功.' % (user_id, user_name)
 elif user_id is None:
 self.Msg = '簽到失敗'
 else:
 self.Msg = '未檢測(cè)到人臉'
 def run(self):
 time.sleep(2)
 while True:
 self.get_name()
 timer = threading.Timer(2, self.log_in) # 等待5s鐘調(diào)用一次fun_timer() 函數(shù)
 timer.start()
 timer.join()
 self.trigger.emit(self.Msg)
 print(self.Msg)
if __name__ == '__main__':
 app = QApplication(sys.argv)
 ui = parentWindow()
 ui_child = childWindow()
 ui_child_late = childWindow_late()
 login_btn = ui.Loginbt
 late_btn = ui.Latebt
 login_btn.clicked.connect(ui_child.show)
 late_btn.clicked.connect(ui_child_late.show)
 if mysql_conn is None:
 QMessageBox.warning(ui,
  "警告",
  "請(qǐng)先聯(lián)網(wǎng)，以便進(jìn)行操作",
  QMessageBox.Close)
 sys.exit(app.exec_())
 else:
 ui.show()
 sys.exit(app.exec_())