460 lines
18 KiB
C++
460 lines
18 KiB
C++
#include "Application.h"
|
|
#include "AsyncEvent.h"
|
|
#include "BoostLog.h"
|
|
#include "CategoryLogSinkBackend.h"
|
|
#include "CdcUpdater.h"
|
|
#include "Configuration.h"
|
|
#include "Database.h"
|
|
#include "DateTime.h"
|
|
#include "DeviceDiscovery.h"
|
|
#include "ImageDecoder.h"
|
|
#include "StringUtility.h"
|
|
#include "VideoFrameProvider.h"
|
|
#include "VideoPlayer.h"
|
|
#include <QApplication>
|
|
#include <QFile>
|
|
#include <QFont>
|
|
#include <QImage>
|
|
#include <QQmlApplicationEngine>
|
|
#include <QSerialPortInfo>
|
|
#include <QTimer>
|
|
#include <filesystem>
|
|
#include <fstream>
|
|
#include <mbedtls/md5.h>
|
|
|
|
constexpr uint32_t ImageSliceSize = (4000 - 32);
|
|
|
|
Application::Application(int &argc, char **argv) : m_app(std::make_shared<QApplication>(argc, argv)) {
|
|
m_app->setApplicationName(APPLICATION_NAME);
|
|
m_app->setApplicationVersion(QString("v%1_%2 build: %3 %4").arg(APP_VERSION, GIT_COMMIT_ID, __DATE__, __TIME__));
|
|
QFont font;
|
|
font.setPointSize(16);
|
|
m_app->setFont(font);
|
|
|
|
m_verifyTimer = new QTimer(this);
|
|
m_verifyTimer->setSingleShot(true);
|
|
connect(m_verifyTimer, &QTimer::timeout, this, &Application::onVerifyTimeout);
|
|
|
|
m_database = std::make_shared<Database>();
|
|
QTimer::singleShot(0, this, [this]() {
|
|
if (!m_database->open("database.db")) {
|
|
LOG(error) << "open database failed.";
|
|
}
|
|
});
|
|
|
|
m_videoFrameProvider = new VideoFrameProvider();
|
|
|
|
qmlRegisterSingletonInstance("Analyser", 1, 0, "App", this);
|
|
if (!std::filesystem::exists(JpgPath)) {
|
|
std::filesystem::create_directory(JpgPath);
|
|
}
|
|
if (!std::filesystem::exists(YuvPath)) {
|
|
std::filesystem::create_directory(YuvPath);
|
|
}
|
|
}
|
|
|
|
void Application::onNewEnrollResult(uint16_t userid) {
|
|
m_palmId = userid;
|
|
QTimer::singleShot(0, this, [this, userid]() {
|
|
emit newStatusTip(Info, QString("%1,录入成功,ID:%2").arg(m_palmUsername).arg(userid));
|
|
});
|
|
}
|
|
|
|
void Application::onNewVerifyResult(uint16_t userid, const QString &username, uint16_t elapsed) {
|
|
m_palmUsername = username;
|
|
m_palmId = userid;
|
|
QTimer::singleShot(0, this, [this, userid, username, elapsed]() {
|
|
emit newStatusTip(Info, QString("%1,识别耗时: %2ms")
|
|
.arg(userid == ModuleCommunication::InvalidUserId ? "未录入用户" : username)
|
|
.arg(elapsed));
|
|
});
|
|
}
|
|
|
|
void Application::initializeLogger() {
|
|
auto backend = boost::make_shared<CategoryLogSinkBackend>("GUI", [this](const std::string &log) {
|
|
Amass::executeAtObjectThread(this, [this, log]() { emit newLog(QString::fromStdString(log)); });
|
|
});
|
|
using SynchronousCategorySink = boost::log::sinks::synchronous_sink<CategoryLogSinkBackend>;
|
|
auto sink = boost::make_shared<SynchronousCategorySink>(backend);
|
|
boost::log::core::get()->add_sink(sink);
|
|
}
|
|
|
|
int Application::exec() {
|
|
QQmlApplicationEngine engine;
|
|
engine.addImageProvider("videoframe", m_videoFrameProvider);
|
|
const QUrl url(QStringLiteral("qrc:/Analyser/qml/main.qml"));
|
|
QObject::connect(
|
|
&engine, &QQmlApplicationEngine::objectCreationFailed, m_app.get(), []() { QCoreApplication::exit(-1); },
|
|
Qt::QueuedConnection);
|
|
engine.load(url);
|
|
|
|
return m_app->exec();
|
|
}
|
|
|
|
QStringList Application::availableSerialPorts() const {
|
|
QStringList ret;
|
|
auto ports = QSerialPortInfo::availablePorts();
|
|
for (auto &port : ports) {
|
|
if (port.description() == "蓝牙链接上的标准串行") continue;
|
|
ret << port.portName();
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
QStringList Application::availableUsbVideoCameras() const {
|
|
QStringList ret;
|
|
DeviceDiscovery d;
|
|
auto devices = d.devices();
|
|
for (auto &device : devices) {
|
|
ret << QString::fromStdString(device);
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
bool Application::open(const QString &portName, int baudRate) {
|
|
m_communication = std::make_shared<ModuleCommunication>();
|
|
connect(m_communication.get(), &ModuleCommunication::commandStarted, this, &Application::onCommandStarted);
|
|
connect(m_communication.get(), &ModuleCommunication::commandFinished, this, &Application::onCommandFinished);
|
|
connect(m_communication.get(), &ModuleCommunication::newEnrollResult, this, &Application::onNewEnrollResult);
|
|
connect(m_communication.get(), &ModuleCommunication::newVerifyResult, this, &Application::onNewVerifyResult);
|
|
connect(m_communication.get(), &ModuleCommunication::newPalmFeature, this, &Application::onNewPalmFeature);
|
|
connect(m_communication.get(), &ModuleCommunication::newImageInfo, this, &Application::onNewImageInfo);
|
|
connect(m_communication.get(), &ModuleCommunication::newImageSliceData, this, &Application::onNewImageSliceData);
|
|
connect(m_communication.get(), &ModuleCommunication::errorOccurred, this, &Application::onErrorOccurred);
|
|
emit connectedChanged();
|
|
|
|
auto status = m_communication->open(portName, baudRate);
|
|
emit newStatusTip(status ? Tip : Error, status ? "串口打开成功" : "串口打开失败");
|
|
if (status) {
|
|
QTimer::singleShot(0, this, [this]() { m_communication->requestVersion(); });
|
|
}
|
|
return status;
|
|
}
|
|
|
|
bool Application::openUVC(const QString &deviceName) {
|
|
m_videoPlayer = std::make_shared<VideoPlayer>();
|
|
m_videoPlayer->setFrameCallback([this](const QImage &image) {
|
|
Amass::executeAtObjectThread(this, [this, image]() {
|
|
m_videoFrameProvider->setImage(image);
|
|
emit newVideoFrame();
|
|
});
|
|
});
|
|
m_videoPlayer->setErrorCallback([this](int error, const std::string &message) {
|
|
LOG_CAT(info, GUI) << "UVC错误: " << message;
|
|
Amass::executeAtObjectThread(this, [this, error]() {
|
|
if (error == -EIO) {
|
|
closeUVC();
|
|
}
|
|
});
|
|
});
|
|
|
|
bool status = m_videoPlayer->open(deviceName.toStdString());
|
|
if (!status) {
|
|
m_videoPlayer.reset();
|
|
}
|
|
emit uvcOpenedChanged();
|
|
emit newStatusTip(status ? Tip : Error, status ? "UVC打开成功" : "UVC打开失败");
|
|
return status;
|
|
}
|
|
|
|
void Application::close() {
|
|
m_communication.reset();
|
|
emit connectedChanged();
|
|
|
|
m_persistenceModeStarted = false;
|
|
m_verifyTimer->stop();
|
|
emit isVerifyingChanged();
|
|
}
|
|
|
|
void Application::closeUVC() {
|
|
m_videoFrameProvider->reset();
|
|
emit newVideoFrame();
|
|
|
|
m_videoPlayer.reset();
|
|
emit uvcOpenedChanged();
|
|
}
|
|
|
|
void Application::verify(bool captureImage, uint8_t timeout) {
|
|
if (m_communication->currentMessageId() != ModuleCommunication::Idle) {
|
|
m_communication->reset();
|
|
}
|
|
if (captureImage) {
|
|
m_communication->verifyExtended(captureImage, timeout);
|
|
} else {
|
|
m_communication->verify(timeout);
|
|
}
|
|
m_verifyExtendedMode = captureImage;
|
|
}
|
|
|
|
void Application::enroll(const QString &username, bool persistence, uint8_t timeout) {
|
|
if (m_communication->currentMessageId() != ModuleCommunication::Idle) {
|
|
m_communication->reset();
|
|
}
|
|
m_communication->enroll(username.toStdString(), persistence, timeout);
|
|
m_palmUsername = username;
|
|
}
|
|
|
|
void Application::deleteUser(uint16_t userid) {
|
|
if (m_communication->currentMessageId() != ModuleCommunication::Idle) {
|
|
m_communication->reset();
|
|
}
|
|
m_communication->deleteUser(userid);
|
|
}
|
|
|
|
void Application::deleteAll() {
|
|
if (m_communication->currentMessageId() != ModuleCommunication::Idle) {
|
|
m_communication->reset();
|
|
}
|
|
m_communication->deleteAll();
|
|
}
|
|
|
|
void Application::enrollExtended(const QString &username, bool persistence, uint8_t timeout) {
|
|
if (m_communication->currentMessageId() != ModuleCommunication::Idle) {
|
|
m_communication->reset();
|
|
}
|
|
m_communication->enrollExtended(username.toStdString(), persistence, timeout);
|
|
m_palmUsername = username;
|
|
}
|
|
|
|
void Application::uploadImage(const QString &path, int operation) {
|
|
m_uploadImageSendedSize = 0;
|
|
ModuleCommunication::UploadImageInformation request;
|
|
if (path.endsWith(".jpg")) {
|
|
auto image = ImageDecoder::extractJpegYComponent(Amass::StringUtility::UTF8ToGBK(path.toStdString()));
|
|
if (!image) {
|
|
LOG(error) << "decode failed.";
|
|
return;
|
|
}
|
|
m_uploadBuffer = image->data;
|
|
LOG(info) << "width: " << image->width << ", height: " << image->height;
|
|
request.width = image->width;
|
|
request.height = image->height;
|
|
} else if (path.endsWith(".yuv")) {
|
|
std::ifstream ifs(Amass::StringUtility::UTF8ToGBK(path.toStdString()), std::ofstream::binary);
|
|
m_uploadBuffer = std::vector<uint8_t>((std::istreambuf_iterator<char>(ifs)), std::istreambuf_iterator<char>());
|
|
if (m_uploadBuffer.empty()) {
|
|
LOG(error) << "file is empty.";
|
|
}
|
|
request.width = 600;
|
|
request.height = 800;
|
|
} else {
|
|
LOG(error) << "not supported format.";
|
|
return;
|
|
}
|
|
|
|
mbedtls_md5_context context;
|
|
mbedtls_md5_init(&context);
|
|
mbedtls_md5_starts(&context);
|
|
mbedtls_md5_update(&context, m_uploadBuffer.data(), m_uploadBuffer.size());
|
|
mbedtls_md5_finish(&context, request.md5);
|
|
mbedtls_md5_free(&context);
|
|
|
|
request.operation = operation;
|
|
request.size = m_uploadBuffer.size();
|
|
strncpy(request.username, "下发测试", sizeof(request.username));
|
|
m_communication->uploadImageInfo(request);
|
|
LOG(info) << "upload image, md5: "
|
|
<< ModuleCommunication::protocolDataFormatString(request.md5, sizeof(request.md5));
|
|
m_imageUploadling = true;
|
|
m_startUploadTime = std::chrono::system_clock::now();
|
|
}
|
|
|
|
void Application::timerEvent(QTimerEvent *event) {
|
|
static int i = 0;
|
|
if (i <= 4000 && !m_imageUploadling) {
|
|
uploadImage("./1722829247678.jpg", 0);
|
|
LOG(info) << "register times: " << i++;
|
|
}
|
|
}
|
|
|
|
ModuleCommunication *Application::module() const {
|
|
return m_communication.get();
|
|
}
|
|
|
|
bool Application::connected() const {
|
|
return static_cast<bool>(m_communication);
|
|
}
|
|
|
|
bool Application::uvcOpened() const {
|
|
return static_cast<bool>(m_videoPlayer);
|
|
}
|
|
|
|
bool Application::persistenceMode() const {
|
|
return m_persistenceMode;
|
|
}
|
|
|
|
void Application::setPersistenceMode(bool enabled) {
|
|
if (m_persistenceMode != enabled) {
|
|
m_persistenceMode = enabled;
|
|
emit persistenceModeChanged();
|
|
}
|
|
}
|
|
|
|
int Application::persistenceVerifyInterval() const {
|
|
return m_persistenceVerifyInterval;
|
|
}
|
|
|
|
void Application::setPersistenceVerifyInterval(int interval) {
|
|
if (m_persistenceVerifyInterval != interval) {
|
|
m_persistenceVerifyInterval = interval;
|
|
emit persistenceVerifyIntervalChanged();
|
|
}
|
|
}
|
|
|
|
bool Application::isVerifying() const {
|
|
if (!m_communication) {
|
|
return false;
|
|
}
|
|
return (m_persistenceMode && m_persistenceModeStarted) ||
|
|
(m_communication->currentMessageId() == ModuleCommunication::Verify) ||
|
|
(m_communication->currentMessageId() == ModuleCommunication::VerifyExtended);
|
|
}
|
|
|
|
void Application::onNewPalmFeature(const PalmFeature &feature) {
|
|
auto palms = m_database->palmFeatures();
|
|
if (std::find(palms.cbegin(), palms.cend(), feature) != palms.cend()) {
|
|
LOG(warning) << "本地数据库已有相同特征数据。";
|
|
return;
|
|
}
|
|
if (!m_database->addPalmFeature(feature)) {
|
|
LOG(error) << "add palm feature failed.";
|
|
}
|
|
}
|
|
|
|
void Application::onErrorOccurred(const QString &error) {
|
|
QTimer::singleShot(0, this, [this]() { close(); });
|
|
}
|
|
|
|
void Application::onNewImageInfo(ModuleCommunication::MessageId messageId, uint32_t size, const uint8_t *md5) {
|
|
using namespace std::chrono;
|
|
m_palmImageId = messageId;
|
|
m_palmImageSize = size;
|
|
m_communication->requestEnrolledImage(0, ImageSliceSize);
|
|
m_palmYImageBuffer.clear();
|
|
m_startUploadTime = system_clock::now();
|
|
if (messageId == ModuleCommunication::Note) {
|
|
emit newStatusTip(Error, "活体未通过照片");
|
|
}
|
|
}
|
|
|
|
void Application::onNewImageSliceData(const std::vector<uint8_t> &data) {
|
|
using namespace std::chrono;
|
|
// LOG(info) << "onNewImageSliceData:" << data.size() << ", already received: " << m_palmYImageBuffer.size()
|
|
// << ", total size: " << m_palmImageSize;
|
|
m_palmYImageBuffer.append(reinterpret_cast<const char *>(data.data()), data.size());
|
|
if (m_palmYImageBuffer.size() < m_palmImageSize) {
|
|
m_communication->requestEnrolledImage(m_palmYImageBuffer.size(), ImageSliceSize);
|
|
} else {
|
|
auto username = m_palmUsername.toStdString();
|
|
auto way = (m_palmImageId == ModuleCommunication::VerifyExtended) ? "verify" : "enroll";
|
|
if (m_palmImageId == ModuleCommunication::Note) {
|
|
way = "no_alive";
|
|
}
|
|
LOG(info) << "request image finished, username: " << username << ", userid: " << m_palmId
|
|
<< ", elapsed: " << duration_cast<milliseconds>(system_clock::now() - m_startUploadTime);
|
|
std::ostringstream oss;
|
|
oss << YuvPath << "/" << username << "_" << m_palmId << "_" << way << "_"
|
|
<< DateTime::toString(std::chrono::system_clock::now(), "%Y%m%d%H%M%S") << ".yuv";
|
|
std::ofstream ofs(Amass::StringUtility::UTF8ToGBK(oss.str()), std::ofstream::binary);
|
|
ofs.write(m_palmYImageBuffer.data(), m_palmYImageBuffer.size());
|
|
|
|
QImage image(reinterpret_cast<const uint8_t *>(m_palmYImageBuffer.data()), 600, 800, QImage::Format_Grayscale8);
|
|
oss.str("");
|
|
oss << JpgPath << "/" << username << "_" << m_palmId << "_" << way << "_"
|
|
<< DateTime::toString(std::chrono::system_clock::now(), "%Y%m%d%H%M%S") << ".jpg";
|
|
image.save(QString::fromStdString(oss.str()), "jpg", 100);
|
|
}
|
|
}
|
|
|
|
void Application::onCommandStarted(ModuleCommunication::MessageId messageId) {
|
|
using namespace std::chrono;
|
|
emit isVerifyingChanged();
|
|
}
|
|
|
|
void Application::onCommandFinished(ModuleCommunication::MessageId messageId,
|
|
ModuleCommunication::MessageStatus status) {
|
|
// LOG(info) << m_persistenceMode << " " << m_persistenceModeStarted << " " << m_persistenceVerifyInterval;
|
|
using namespace std::chrono;
|
|
if (messageId == ModuleCommunication::UploadImageInfo) {
|
|
m_communication->uploadImageData(m_uploadImageSendedSize, m_uploadBuffer.data() + m_uploadImageSendedSize,
|
|
ImageSliceSize);
|
|
} else if (messageId == ModuleCommunication::UploadImageData) {
|
|
m_uploadImageSendedSize += ImageSliceSize;
|
|
if (m_uploadImageSendedSize < m_uploadBuffer.size()) {
|
|
auto remainSize = m_uploadBuffer.size() - m_uploadImageSendedSize;
|
|
m_communication->uploadImageData(m_uploadImageSendedSize,
|
|
(const uint8_t *)m_uploadBuffer.data() + m_uploadImageSendedSize,
|
|
remainSize < ImageSliceSize ? remainSize : ImageSliceSize);
|
|
m_imageUploadling = true;
|
|
} else {
|
|
LOG(info) << "upload finished, elapsed: "
|
|
<< duration_cast<milliseconds>(system_clock::now() - m_startUploadTime);
|
|
m_imageUploadling = false;
|
|
}
|
|
}
|
|
|
|
if (((messageId == ModuleCommunication::Verify) || (messageId == ModuleCommunication::VerifyExtended)) &&
|
|
m_persistenceMode) { // 持续识别逻辑
|
|
m_persistenceModeStarted = true;
|
|
} else if (messageId == ModuleCommunication::Reset) {
|
|
m_persistenceModeStarted = false;
|
|
m_verifyTimer->stop();
|
|
}
|
|
|
|
if (m_persistenceMode && m_persistenceModeStarted &&
|
|
((messageId == ModuleCommunication::Verify) || (messageId == ModuleCommunication::VerifyExtended)) &&
|
|
((status == ModuleCommunication::Success) || (status == ModuleCommunication::Failed4UnknownUser) ||
|
|
(status == ModuleCommunication::Failed4Timeout) || (status == ModuleCommunication::Failed4UnknownReason))) {
|
|
m_verifyTimer->start(m_persistenceVerifyInterval * 1000);
|
|
}
|
|
|
|
if (((messageId == ModuleCommunication::EnrollSingle) || (messageId == ModuleCommunication::EnrollExtended)) &&
|
|
(status == ModuleCommunication::Success)) {
|
|
emit newStatusTip(Info, "录入成功。");
|
|
}
|
|
emit isVerifyingChanged();
|
|
}
|
|
|
|
void Application::onVerifyTimeout() {
|
|
if (m_verifyExtendedMode) {
|
|
m_communication->verifyExtended(m_verifyExtendedMode, 120);
|
|
} else {
|
|
m_communication->verify(120);
|
|
}
|
|
}
|
|
|
|
bool Application::startOta(const QString &path) {
|
|
if (!QFile::exists(path)) {
|
|
emit otaMessage("文件不存在");
|
|
return false;
|
|
}
|
|
auto device = CdcUpdater::searchDevice();
|
|
if (device) {
|
|
LOG(info) << "device already in ota mode.";
|
|
} else {
|
|
if (m_communication) {
|
|
m_communication->startOta();
|
|
} else {
|
|
emit otaMessage("请先打开设备");
|
|
return false;
|
|
}
|
|
}
|
|
LOG(info) << "start ota, ota path: " << path.toStdString();
|
|
m_updater = std::make_shared<CdcUpdater>();
|
|
connect(m_updater.get(), &CdcUpdater::deviceDiscovered, this, &Application::onCdcDeviceDiscovered);
|
|
connect(m_updater.get(), &CdcUpdater::updateFinished, this, &Application::updateFinished);
|
|
connect(m_updater.get(), &CdcUpdater::progressChanged, this, &Application::otaProgressChanged);
|
|
connect(m_updater.get(), &CdcUpdater::message, this, &Application::otaMessage);
|
|
|
|
m_updater->start(path, device ? *device : QSerialPortInfo());
|
|
return true;
|
|
}
|
|
|
|
void Application::onCdcDeviceDiscovered(const QSerialPortInfo &info) {
|
|
auto status = m_updater->open(info);
|
|
LOG(info) << "open cdc port: " << info.portName().toStdString() << ", status: " << status;
|
|
if (!status) {
|
|
QTimer::singleShot(0, this, [this]() { m_updater->startSearchDevice(); });
|
|
}
|
|
}
|