First commit

Author: lsk-china

.gitignore

@@ -50,3 +50,5 @@ compile_commands.json
 
 # QtCreator local machine specific files for imported projects
 *creator.user*
+
+db.ini

FaceRec.pro

@@ -0,0 +1,45 @@
+QT       += core gui multimedia multimediawidgets
+
+greaterThan(QT_MAJOR_VERSION, 4): QT += widgets
+
+CONFIG += c++17 link_pkgconfig
+
+PKGCONFIG += opencv4
+
+LIBS += -lmysqlcppconn
+# You can make your code fail to compile if it uses deprecated APIs.
+# In order to do so, uncomment the following line.
+#DEFINES += QT_DISABLE_DEPRECATED_BEFORE=0x060000    # disables all the APIs deprecated before Qt 6.0.0
+
+SOURCES += \
+    database.cpp \
+    facepickframe.cpp \
+    facerecframe.cpp \
+    main.cpp \
+    mainwindow.cpp \
+    recognizer.cpp \
+    traindialog.cpp \
+    trainthread.cpp
+
+HEADERS += \
+    database.h \
+    facepickframe.h \
+    facerecframe.h \
+    mainwindow.h \
+    recognizer.h \
+    traindialog.h \
+    trainthread.h
+
+FORMS += \
+    facepickframe.ui \
+    facerecframe.ui \
+    mainwindow.ui \
+    traindialog.ui
+
+# Default rules for deployment.
+qnx: target.path = /tmp/$${TARGET}/bin
+else: unix:!android: target.path = /opt/$${TARGET}/bin
+!isEmpty(target.path): INSTALLS += target
+
+DISTFILES += \
+    db.ini

database.cpp

@@ -0,0 +1,52 @@
+#include "database.h"
+
+Database::Database()
+{
+    if (access("db.ini", F_OK) != 0) {
+        throw std::runtime_error("db.ini not found!");
+        exit(-1);
+    }
+    QSettings *dbSettings = new QSettings("db.ini", QSettings::IniFormat);
+    const string host = dbSettings->value("DB/Host").toString().toStdString();
+    const string username = dbSettings->value("DB/Username").toString().toStdString();
+    const string password = dbSettings->value("DB/Password").toString().toStdString();
+    const string database = dbSettings->value("DB/Database").toString().toStdString();
+    Driver *driver = mysql::get_driver_instance();
+    this->conn = driver->connect(host, username, password);
+    this->conn->setSchema(database);
+    delete dbSettings;
+}
+
+void Database::incrTimes(int sid) {
+    unique_ptr<PreparedStatement> stmt(this->conn->prepareStatement("update student set times=times+1 where id=?"));
+    stmt->setInt(1, sid);
+    stmt->executeUpdate();
+    unique_ptr<PreparedStatement> stmt2(this->conn->prepareStatement("insert into attendance(name, sid, time) values((select name from student where id=?), ?, ?)"));
+    stmt2->setInt(1, sid);
+    stmt2->setInt(2, sid);
+    time_t curtime = 0;
+    time(&curtime);
+    char timeStr[100];
+    strftime(timeStr, 100, "%Y-%m-%d %H:%M:%S", localtime(&curtime));
+    stmt2->setDateTime(3, timeStr);
+    stmt2->executeUpdate();
+}
+
+int Database::newStudent(student student) {
+    unique_ptr<PreparedStatement> stmt(this->conn->prepareStatement("select count(id) as c from student where id=?"));
+    unique_ptr<PreparedStatement> stmt2(this->conn->prepareStatement("insert into student(id, name) values(?,?)"));
+    stmt->setInt(1, student.sid);
+    unique_ptr<ResultSet> res(stmt->executeQuery());
+    if (res->getInt("c") > 0) {
+        return -1;
+    }
+    stmt2->setInt(1, student.sid);
+    stmt2->setString(2, student.name);
+    stmt2->execute();
+    return 0;
+}
+
+Database::~Database() {
+    this->conn->close();
+    delete conn;
+}

database.h

@@ -0,0 +1,41 @@
+#ifndef DATABASE_H
+#define DATABASE_H
+
+#include <QDebug>
+#include <QSettings>
+#include <QMessageBox>
+
+#include <mysql/jdbc.h>
+
+#include <time.h>
+#include <string.h>
+#include <unistd.h>
+
+using namespace sql;
+using namespace std;
+
+typedef struct {
+    int sid;
+    string name;
+    int times;
+} student;
+
+typedef struct {
+    int id;
+    string name;
+    int sid;
+    tm time;
+} attendance;
+
+class Database
+{
+public:
+    Database();
+    ~Database();
+    int newStudent(student student);
+    void incrTimes(int sid);
+private:
+    Connection *conn;
+};
+
+#endif // DATABASE_H

facepickframe.cpp

@@ -0,0 +1,39 @@
+#include "facepickframe.h"
+#include "ui_facepickframe.h"
+
+FacePickFrame::FacePickFrame(QWidget *mainWindow, QWidget *parent) : QFrame(parent), ui(new Ui::FacePickFrame) {
+    this->ui->setupUi(this);
+    this->mainWindow = mainWindow;
+
+    connect(this->ui->pushButton, &QPushButton::clicked, this, [this]() {
+        QString nameInput = this->ui->lineEdit->text();
+        QString sidInput = this->ui->lineEdit_2->text();
+        if (nameInput == "" || sidInput == "") {
+            QMessageBox::information(this, "Error", "姓名和学号不能为空");
+            return;
+        }
+        student stu = {sidInput.toInt(), nameInput.toStdString(), 0};
+
+    });
+}
+
+FacePickFrame::~FacePickFrame()
+{
+    this->camera->stop();
+    this->camera->unload();
+    this->mainWindow->show();
+    delete ui;
+}
+
+void FacePickFrame::closeEvent(QCloseEvent *e) {
+    this->camera->stop();
+    this->camera->unload();
+    this->mainWindow->show();
+}
+
+void FacePickFrame::showEvent(QShowEvent *e) {
+    this->camera = new QCamera(QCameraInfo::defaultCamera());
+    this->camera->setCaptureMode(QCamera::CaptureMode::CaptureViewfinder);
+    this->camera->setViewfinder(this->ui->widget_2);
+    this->camera->start();
+}

facepickframe.h

@@ -0,0 +1,40 @@
+#ifndef FACEPICKFRAME_H
+#define FACEPICKFRAME_H
+
+#include <QFrame>
+#include <QPushButton>
+#include <QMessageBox>
+#include <QCloseEvent>
+#include <QShowEvent>
+#include <QCameraInfo>
+#include <QCamera>
+#include <QVideoWidget>
+#include <QCameraViewfinderSettings>
+
+#include "database.h"
+#include "traindialog.h"
+#include "trainthread.h"
+
+namespace Ui {
+class FacePickFrame;
+}
+
+class FacePickFrame : public QFrame
+{
+    Q_OBJECT
+
+public:
+    explicit FacePickFrame(QWidget *mainWindow, QWidget *parent = nullptr);
+    ~FacePickFrame();
+
+private:
+    Ui::FacePickFrame *ui;
+    QWidget *mainWindow;
+    QCamera *camera;
+
+protected:
+    void closeEvent(QCloseEvent *e) override;
+    void showEvent(QShowEvent *e) override;
+};
+
+#endif // FACEPICKFRAME_H