|
|
|
|
|
#pragma execution_character_set("utf-8")
|
|
|
|
|
|
#include "SmokeBoxIdentification.h"
|
|
|
|
|
|
|
|
|
|
|
|
SmokeBoxIdentification::SmokeBoxIdentification(QWidget *parent)
|
|
|
|
|
|
: QMainWindow(parent)
|
|
|
|
|
|
{
|
|
|
|
|
|
qRegisterMetaType<QVector<ImageInfo> >("QVector<ImageInfo>");
|
|
|
|
|
|
ui.setupUi(this);
|
|
|
|
|
|
//m_pShowRltImgLabel = ui.label_showImg;
|
|
|
|
|
|
m_pCameraControl = new CameraControl();
|
|
|
|
|
|
m_pCaptureThread = new QThread();
|
|
|
|
|
|
m_pCameraControl->moveToThread(m_pCaptureThread);
|
|
|
|
|
|
|
|
|
|
|
|
connect(this, &SmokeBoxIdentification::sgCapture, m_pCameraControl, &CameraControl::onCapture);
|
|
|
|
|
|
connect(m_pCameraControl, &CameraControl::sgCapturedImage, this, &SmokeBoxIdentification::onCapturedImage);
|
|
|
|
|
|
connect(m_pCaptureThread, &QThread::finished, m_pCameraControl, &QObject::deleteLater);
|
|
|
|
|
|
m_pCaptureThread->start();
|
|
|
|
|
|
|
|
|
|
|
|
QString applicationDirPath = QCoreApplication::applicationDirPath();
|
|
|
|
|
|
QtConcurrent::run([=]()
|
|
|
|
|
|
{
|
|
|
|
|
|
readCameraConfig();
|
|
|
|
|
|
m_bCamerasInit = initCamera();
|
|
|
|
|
|
if (decodeSmokeInfoCsv(applicationDirPath + SMOKEINFO_XLSX_CONFIG))
|
|
|
|
|
|
matchSmokeNameAndTemplateFolderName(applicationDirPath + NUM_TEMPLATE_PATH);
|
|
|
|
|
|
m_pDataProcess = std::make_shared<dataPreProcessing>();
|
|
|
|
|
|
if (m_pDataProcess)
|
|
|
|
|
|
{
|
|
|
|
|
|
dataPreProcessing * t = (dataPreProcessing*)m_pDataProcess.get();
|
|
|
|
|
|
m_pDataProcess->getFileDir(applicationDirPath + NUM_TEMPLATE_PATH);
|
|
|
|
|
|
m_pDataProcess->transmitFilePathForSampleWithItsCorresLabel();
|
|
|
|
|
|
ADModule::instance()->setDataProcessObj(m_pDataProcess);
|
|
|
|
|
|
m_pDataProcess->setWinSize(170);
|
|
|
|
|
|
}
|
|
|
|
|
|
});
|
|
|
|
|
|
|
|
|
|
|
|
QString filePath = applicationDirPath + NET_CONFIG;
|
|
|
|
|
|
m_pNetControl = new NetControl(filePath, true);
|
|
|
|
|
|
connect(m_pNetControl, &NetControl::sgReceiveData, this, &SmokeBoxIdentification::onDecodeMsg);
|
|
|
|
|
|
ADModule::instance()->loadParameters(filePath);
|
|
|
|
|
|
filePath = applicationDirPath + SERIAL_CONFIG;
|
|
|
|
|
|
lpSerialStation::instance()->loadParameters(filePath);
|
|
|
|
|
|
lpSerialStation::instance()->openComs();
|
|
|
|
|
|
m_pCategoryMatcher = new CategoryMatcher();
|
|
|
|
|
|
connect(this, &SmokeBoxIdentification::sgCategoryMatchFinished, this, &SmokeBoxIdentification::onNumStatistic);
|
|
|
|
|
|
connect(lpSerialStation::instance(), &lpSerialStation::sgMonitorSensorTriggerStart, this
|
|
|
|
|
|
, &SmokeBoxIdentification::onMonitorSensorTriggerStart);
|
|
|
|
|
|
connect(lpSerialStation::instance(), &lpSerialStation::sgMonitorSensorTriggerStop, this
|
|
|
|
|
|
, &SmokeBoxIdentification::onMonitorSensorTriggerStop);
|
|
|
|
|
|
connect(this, &SmokeBoxIdentification::sgMonitorSensorTriggerStart, ADModule::instance()
|
|
|
|
|
|
, &ADModule::onDataRequestStart);
|
|
|
|
|
|
connect(this, &SmokeBoxIdentification::sgMonitorSensorTriggerStop, ADModule::instance()
|
|
|
|
|
|
, &ADModule::onDataRequestStop);
|
|
|
|
|
|
connect(this, &SmokeBoxIdentification::sgSendData2AndAlgo, ADModule::instance(), &ADModule::onSendData2Algo);
|
|
|
|
|
|
connect(this, &SmokeBoxIdentification::sgSaveSensorData2Local, ADModule::instance(), &ADModule::onSaveData2File);
|
|
|
|
|
|
connect(this, &SmokeBoxIdentification::sgSendStockCheckMsg, this, &SmokeBoxIdentification::onSendStockCheckMsg, Qt::QueuedConnection);
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
SmokeBoxIdentification::~SmokeBoxIdentification()
|
|
|
|
|
|
{
|
|
|
|
|
|
DELETE_POINTER(m_pNetControl);
|
|
|
|
|
|
m_pCaptureThread->quit();
|
|
|
|
|
|
m_pCaptureThread->wait();
|
|
|
|
|
|
DELETE_POINTER(m_pCaptureThread);
|
|
|
|
|
|
DELETE_POINTER(m_pCameraControl);
|
|
|
|
|
|
DELETE_POINTER(m_pCategoryMatcher);
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
bool SmokeBoxIdentification::readCameraConfig()
|
|
|
|
|
|
{
|
|
|
|
|
|
m_mapIdCameraInfoSide.clear();
|
|
|
|
|
|
m_mapIdCameraInfoTop.clear();
|
|
|
|
|
|
m_mapSerialNumberIdSide.clear();
|
|
|
|
|
|
m_mapSerialNumberIdTop.clear();
|
|
|
|
|
|
QString filePath = QCoreApplication::applicationDirPath() + CAMERA_CONFIG;
|
|
|
|
|
|
QFile file(filePath);
|
|
|
|
|
|
if (!file.open(QIODevice::ReadOnly | QIODevice::Text))
|
|
|
|
|
|
{
|
|
|
|
|
|
qWarning() << "<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ļ<EFBFBD>ʧ<EFBFBD>ܣ<EFBFBD>";
|
|
|
|
|
|
return false;
|
|
|
|
|
|
}
|
|
|
|
|
|
QByteArray arr = file.readAll();
|
|
|
|
|
|
file.close();
|
|
|
|
|
|
if (arr.isEmpty())
|
|
|
|
|
|
{
|
|
|
|
|
|
qWarning() << "<EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ϊ<EFBFBD><EFBFBD>";
|
|
|
|
|
|
return false;
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
QJsonParseError err;
|
|
|
|
|
|
QJsonDocument doc = QJsonDocument::fromJson(arr, &err);
|
|
|
|
|
|
if (doc.isEmpty())
|
|
|
|
|
|
{
|
|
|
|
|
|
qWarning() << err.errorString(); //<2F><>ӡʧ<D3A1><CAA7><EFBFBD><EFBFBD>Ϣ
|
|
|
|
|
|
return false;
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
QJsonObject jsonObj = doc.object();
|
|
|
|
|
|
auto getCameraType = [](int type)-> CameraType
|
|
|
|
|
|
{
|
|
|
|
|
|
CameraType cameraType;
|
|
|
|
|
|
switch (type)
|
|
|
|
|
|
{
|
|
|
|
|
|
case 100:
|
|
|
|
|
|
cameraType = CameraType::Virtual;
|
|
|
|
|
|
break;
|
|
|
|
|
|
case 140:
|
|
|
|
|
|
cameraType = CameraType::Hik;
|
|
|
|
|
|
break;
|
|
|
|
|
|
case 190:
|
|
|
|
|
|
cameraType = CameraType::Basler;
|
|
|
|
|
|
break;
|
|
|
|
|
|
default:
|
|
|
|
|
|
break;
|
|
|
|
|
|
}
|
|
|
|
|
|
return cameraType;
|
|
|
|
|
|
};
|
|
|
|
|
|
QJsonObject topObj = jsonObj.value("top").toObject();
|
|
|
|
|
|
int topStartId, topEndId;
|
|
|
|
|
|
if (!topObj.isEmpty())
|
|
|
|
|
|
{
|
|
|
|
|
|
topStartId = topObj.value("startID").toInt();
|
|
|
|
|
|
topEndId = topObj.value("endID").toInt();
|
|
|
|
|
|
}
|
|
|
|
|
|
else
|
|
|
|
|
|
{
|
|
|
|
|
|
qWarning() << "top id error";
|
|
|
|
|
|
return false;
|
|
|
|
|
|
}
|
|
|
|
|
|
QJsonObject sideObj = jsonObj.value("side").toObject();
|
|
|
|
|
|
int sideStartId, sideEndId;
|
|
|
|
|
|
if (!sideObj.isEmpty())
|
|
|
|
|
|
{
|
|
|
|
|
|
sideStartId = sideObj.value("startID").toInt();
|
|
|
|
|
|
sideEndId = sideObj.value("endID").toInt();
|
|
|
|
|
|
}
|
|
|
|
|
|
else
|
|
|
|
|
|
{
|
|
|
|
|
|
qWarning() << "side id error";
|
|
|
|
|
|
return false;
|
|
|
|
|
|
}
|
|
|
|
|
|
QJsonObject devicesObj = jsonObj.value("devices").toObject();
|
|
|
|
|
|
if (!devicesObj.isEmpty()) {
|
|
|
|
|
|
CameraInfo cameraInfo;
|
|
|
|
|
|
QStringList devicesLabelList = devicesObj.keys();
|
|
|
|
|
|
for each (QString deviceLabel in devicesLabelList)
|
|
|
|
|
|
{
|
|
|
|
|
|
cameraInfo.cameraLabel = deviceLabel;
|
|
|
|
|
|
QStringList strList = deviceLabel.split(' ');
|
|
|
|
|
|
if (strList.length() != 2)
|
|
|
|
|
|
{
|
|
|
|
|
|
qWarning() << "Device label: " << deviceLabel << " syntax error!";
|
|
|
|
|
|
continue;
|
|
|
|
|
|
}
|
|
|
|
|
|
cameraInfo.modelName = strList.at(0);
|
|
|
|
|
|
cameraInfo.serialNumber = strList.at(1);
|
|
|
|
|
|
QJsonObject deviceObj = devicesObj.value(deviceLabel).toObject();
|
|
|
|
|
|
if (!deviceObj.isEmpty())
|
|
|
|
|
|
{
|
|
|
|
|
|
cameraInfo.cameraType = getCameraType(deviceObj.value("type").toInt());
|
|
|
|
|
|
cameraInfo.exposure = deviceObj.value("exposure").toDouble();
|
|
|
|
|
|
cameraInfo.imageHeight = deviceObj.value("height").toInt();
|
|
|
|
|
|
cameraInfo.imageWidth = deviceObj.value("width").toInt();
|
|
|
|
|
|
cameraInfo.saveImgSuffix = deviceObj.value("save_img_suffix").toString();
|
|
|
|
|
|
int id = deviceObj.value("id").toInt();
|
|
|
|
|
|
if (id >= topStartId && id <= topEndId)
|
|
|
|
|
|
{
|
|
|
|
|
|
m_mapIdCameraInfoTop[id] = cameraInfo;
|
|
|
|
|
|
m_mapSerialNumberIdTop[cameraInfo.serialNumber] = id;
|
|
|
|
|
|
}
|
|
|
|
|
|
else if (id >= sideStartId && id <= sideEndId)
|
|
|
|
|
|
{
|
|
|
|
|
|
m_mapIdCameraInfoSide[id] = cameraInfo;
|
|
|
|
|
|
m_mapSerialNumberIdSide[cameraInfo.serialNumber] = id;
|
|
|
|
|
|
}
|
|
|
|
|
|
}
|
|
|
|
|
|
}
|
|
|
|
|
|
}
|
|
|
|
|
|
return true;
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
bool SmokeBoxIdentification::initCamera()
|
|
|
|
|
|
{
|
|
|
|
|
|
QList<CameraInfo> cameraInfoList = m_mapIdCameraInfoSide.values();
|
|
|
|
|
|
cameraInfoList.append(m_mapIdCameraInfoTop.values());
|
|
|
|
|
|
if (!m_pCameraControl->loadCamera(cameraInfoList))
|
|
|
|
|
|
{
|
|
|
|
|
|
qWarning() << "Load camera failed";
|
|
|
|
|
|
return false;
|
|
|
|
|
|
}
|
|
|
|
|
|
else {
|
|
|
|
|
|
QStringList serialNumberList = m_mapSerialNumberIdSide.keys();
|
|
|
|
|
|
serialNumberList.append(m_mapSerialNumberIdTop.keys());
|
|
|
|
|
|
if (!m_pCameraControl->openCamera(serialNumberList))
|
|
|
|
|
|
{
|
|
|
|
|
|
qWarning() << "Open camera failed";
|
|
|
|
|
|
return false;
|
|
|
|
|
|
}
|
|
|
|
|
|
else {
|
|
|
|
|
|
if (!m_pCameraControl->startCamera(serialNumberList))
|
|
|
|
|
|
{
|
|
|
|
|
|
qWarning() << "Start camera failed";
|
|
|
|
|
|
return false;
|
|
|
|
|
|
}
|
|
|
|
|
|
else {
|
|
|
|
|
|
if (!m_pCameraControl->setSoftwareTrigger(serialNumberList))
|
|
|
|
|
|
{
|
|
|
|
|
|
qWarning() << "Set software trigger failed";
|
|
|
|
|
|
}
|
|
|
|
|
|
}
|
|
|
|
|
|
}
|
|
|
|
|
|
}
|
|
|
|
|
|
return true;
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
void SmokeBoxIdentification::captureTop()
|
|
|
|
|
|
{
|
|
|
|
|
|
emit sgCapture(m_mapSerialNumberIdTop.keys());
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
void SmokeBoxIdentification::captureSide()
|
|
|
|
|
|
{
|
|
|
|
|
|
emit sgCapture(m_mapSerialNumberIdSide.keys());
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
bool SmokeBoxIdentification::getSaveImgNameInfo(QString serialNumber, FileInfo& fileInfo)
|
|
|
|
|
|
{
|
|
|
|
|
|
QString imgSaveBaseName;
|
|
|
|
|
|
auto iter = m_mapSerialNumberIdSide.find(serialNumber);
|
|
|
|
|
|
if (iter == m_mapSerialNumberIdSide.end())
|
|
|
|
|
|
{
|
|
|
|
|
|
iter = m_mapSerialNumberIdTop.find(serialNumber);
|
|
|
|
|
|
if (iter != m_mapSerialNumberIdTop.end())
|
|
|
|
|
|
{
|
|
|
|
|
|
int id = iter.value();
|
|
|
|
|
|
QString suffix = m_mapIdCameraInfoTop.value(id).saveImgSuffix;
|
|
|
|
|
|
fileInfo.suffix = suffix;
|
|
|
|
|
|
fileInfo.baseName = QString::number(id);
|
|
|
|
|
|
fileInfo.name = fileInfo.baseName + suffix;
|
|
|
|
|
|
}
|
|
|
|
|
|
else
|
|
|
|
|
|
{
|
|
|
|
|
|
qWarning() << "Can not find " << serialNumber;
|
|
|
|
|
|
return false;
|
|
|
|
|
|
}
|
|
|
|
|
|
}
|
|
|
|
|
|
else
|
|
|
|
|
|
{
|
|
|
|
|
|
int id = iter.value();
|
|
|
|
|
|
QString suffix = m_mapIdCameraInfoSide.value(id).saveImgSuffix;
|
|
|
|
|
|
fileInfo.suffix = suffix;
|
|
|
|
|
|
fileInfo.baseName = QString::number(id);
|
|
|
|
|
|
fileInfo.name = fileInfo.baseName + suffix;
|
|
|
|
|
|
}
|
|
|
|
|
|
return true;
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
void SmokeBoxIdentification::saveImage(const cv::Mat& imgMat, const QString& filePath)
|
|
|
|
|
|
{
|
|
|
|
|
|
if (imgMat.empty())
|
|
|
|
|
|
{
|
|
|
|
|
|
qWarning() << "Image is empty";
|
|
|
|
|
|
return;
|
|
|
|
|
|
}
|
|
|
|
|
|
cv::imwrite(filePath.toStdString(), imgMat);
|
|
|
|
|
|
qDebug() << "Save image: " << filePath;
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
void SmokeBoxIdentification::saveImage(const QVector<ImageInfo>& vecImageInfo, int workMode)
|
|
|
|
|
|
{
|
|
|
|
|
|
QString dataSaveFolderPath;
|
|
|
|
|
|
switch (workMode)
|
|
|
|
|
|
{
|
|
|
|
|
|
case WorkMode::StockCheck:
|
|
|
|
|
|
dataSaveFolderPath = QString("%1\\%2\\%3").arg(FILE_STORAGE_PATH).arg(m_currtStockCheckInfo.stockNum)
|
|
|
|
|
|
.arg(m_currtStockCheckInfo.taskNum);
|
|
|
|
|
|
break;
|
|
|
|
|
|
case WorkMode::IntoStock:
|
|
|
|
|
|
dataSaveFolderPath =QString("%1\\%2\\%3\\%4").arg(FILE_STORAGE_PATH).arg(INTOSTOCK_IMAGE_FOLDER)
|
|
|
|
|
|
.arg(m_currtIntoStockInfo.stockNum).arg(m_currtIntoStockInfo.taskNum);
|
|
|
|
|
|
break;
|
|
|
|
|
|
default:
|
|
|
|
|
|
qDebug() << "Mode: " << workMode << "do not save Image";
|
|
|
|
|
|
return;
|
|
|
|
|
|
}
|
|
|
|
|
|
QDir folder(dataSaveFolderPath);
|
|
|
|
|
|
if (!folder.exists())
|
|
|
|
|
|
{
|
|
|
|
|
|
QDir dir;
|
|
|
|
|
|
if (dir.mkpath(dataSaveFolderPath))
|
|
|
|
|
|
{
|
|
|
|
|
|
qDebug() << "Create folder: " << dataSaveFolderPath << " successfully.";
|
|
|
|
|
|
}
|
|
|
|
|
|
else
|
|
|
|
|
|
{
|
|
|
|
|
|
qDebug() << "Create folder:" << dataSaveFolderPath << " failed.";
|
|
|
|
|
|
return;
|
|
|
|
|
|
}
|
|
|
|
|
|
}
|
|
|
|
|
|
for (int i = 0; i < vecImageInfo.size(); ++i)
|
|
|
|
|
|
{
|
|
|
|
|
|
FileInfo fileInfo;
|
|
|
|
|
|
getSaveImgNameInfo(vecImageInfo.at(i).cameraSn, fileInfo);
|
|
|
|
|
|
QString filePath = QString("%1\\%2").arg(dataSaveFolderPath).arg(fileInfo.name);
|
|
|
|
|
|
if (workMode == WorkMode::IntoStock)
|
|
|
|
|
|
{
|
|
|
|
|
|
QString filePathSend = QString("%1/%2/%3/%4").arg(INTOSTOCK_IMAGE_FOLDER)
|
|
|
|
|
|
.arg(m_currtIntoStockInfo.stockNum).arg(m_currtIntoStockInfo.taskNum).arg(fileInfo.name);
|
|
|
|
|
|
m_currtIntoStockInfo.imagesPathList.push_back(filePathSend);
|
|
|
|
|
|
}
|
|
|
|
|
|
saveImage(vecImageInfo.at(i).image, filePath);
|
|
|
|
|
|
}
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
Q_SLOT void SmokeBoxIdentification::onCapturedImage(const QVector<ImageInfo>& vecImageInfo)
|
|
|
|
|
|
{
|
|
|
|
|
|
QString serialNumber = vecImageInfo.last().cameraSn;
|
|
|
|
|
|
if (m_mapSerialNumberIdTop.contains(serialNumber))
|
|
|
|
|
|
{
|
|
|
|
|
|
qDebug() << "top captured";
|
|
|
|
|
|
QtConcurrent::run([=] {
|
|
|
|
|
|
saveImage(vecImageInfo, m_workMode);
|
|
|
|
|
|
if (m_workMode == WorkMode::StockCheck)
|
|
|
|
|
|
{
|
|
|
|
|
|
emit sgSendStockCheckMsg(MsgType::DataCollectionEndSC);
|
|
|
|
|
|
}
|
|
|
|
|
|
else if (m_workMode = WorkMode::IntoStock)
|
|
|
|
|
|
{
|
|
|
|
|
|
emit sgSendStockCheckMsg(MsgType::IntoStockResult);
|
|
|
|
|
|
}
|
|
|
|
|
|
m_workMode = WorkMode::Idle;
|
|
|
|
|
|
});
|
|
|
|
|
|
}
|
|
|
|
|
|
else if (m_mapSerialNumberIdSide.contains(serialNumber))
|
|
|
|
|
|
{
|
|
|
|
|
|
qDebug() << "side captured";
|
|
|
|
|
|
QtConcurrent::run([=] {
|
|
|
|
|
|
saveImage(vecImageInfo, m_workMode);
|
|
|
|
|
|
bool bRlt = false;
|
|
|
|
|
|
QString typeNum = m_currtStockCheckInfo.smokeTypeNum;
|
|
|
|
|
|
cv::Mat imgRltMat;
|
|
|
|
|
|
QString dataSaveFolder = QString("%1\\%2\\%3").arg(FILE_STORAGE_PATH)
|
|
|
|
|
|
.arg(m_currtStockCheckInfo.stockNum).arg(m_currtStockCheckInfo.taskNum);
|
|
|
|
|
|
for each (const ImageInfo& imgInfo in vecImageInfo)
|
|
|
|
|
|
{
|
|
|
|
|
|
if (m_pCategoryMatcher->smokeMatch(typeNum, imgInfo.image, imgRltMat))
|
|
|
|
|
|
{
|
|
|
|
|
|
bRlt = true;
|
|
|
|
|
|
FileInfo fileInfo;
|
|
|
|
|
|
getSaveImgNameInfo(imgInfo.cameraSn, fileInfo);
|
|
|
|
|
|
QString fileName = QString("%1_%2_rlt%3").arg(fileInfo.baseName)
|
|
|
|
|
|
.arg(m_currtStockCheckInfo.smokeTypeNum).arg(fileInfo.suffix);
|
|
|
|
|
|
QString filePath = QString("%1\\%2").arg(dataSaveFolder).arg(fileName);
|
|
|
|
|
|
saveImage(imgRltMat, filePath);
|
|
|
|
|
|
qDebug() << "Match successful";
|
|
|
|
|
|
break;
|
|
|
|
|
|
}
|
|
|
|
|
|
}
|
|
|
|
|
|
if (!bRlt)
|
|
|
|
|
|
{
|
|
|
|
|
|
QString filePath = dataSaveFolder + QString("\\%1_error.PNG")
|
|
|
|
|
|
.arg(m_currtStockCheckInfo.smokeTypeNum);
|
|
|
|
|
|
saveImage(imgRltMat, filePath);
|
|
|
|
|
|
}
|
|
|
|
|
|
QString typeNumRlt = bRlt ? typeNum : "0";
|
|
|
|
|
|
m_currtStockCheckInfo.smokeTypeNumRlt = typeNumRlt;
|
|
|
|
|
|
qDebug() << "typeNumRlt: " << typeNumRlt;
|
|
|
|
|
|
emit sgCategoryMatchFinished(typeNumRlt);
|
|
|
|
|
|
//if (!imgRltMat.empty())
|
|
|
|
|
|
//{
|
|
|
|
|
|
// cv::cvtColor(imgRltMat, imgRltMat, cv::COLOR_BGR2RGB);
|
|
|
|
|
|
// QImage resultQImg((const unsigned char*)imgRltMat.data, imgRltMat.cols, imgRltMat.rows, imgRltMat.step, QImage::Format_RGB888);
|
|
|
|
|
|
// emit sgShowImage(m_pShowRltImgLabel, resultQImg);
|
|
|
|
|
|
//}
|
|
|
|
|
|
});
|
|
|
|
|
|
}
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
Q_SLOT void SmokeBoxIdentification::onDecodeMsg(const QString& msg)
|
|
|
|
|
|
{
|
|
|
|
|
|
QString msgContent = msg;
|
|
|
|
|
|
if (msgContent.right(1) == END_SYMBOL)
|
|
|
|
|
|
{
|
|
|
|
|
|
msgContent.chop(1);
|
|
|
|
|
|
}
|
|
|
|
|
|
qDebug() << "msg: " << msg << " content: " << msgContent;
|
|
|
|
|
|
QStringList strList = msgContent.split(SPLIT_RULE);
|
|
|
|
|
|
qDebug() << "strlist length: " << strList.length() << " head: " << strList.at(0);
|
|
|
|
|
|
if (strList.length() == 7 && strList.at(0) == STOCK_CHECK_HEAD)
|
|
|
|
|
|
{
|
|
|
|
|
|
qDebug() << "<EFBFBD><EFBFBD>ʼ<EFBFBD><EFBFBD>λ<EFBFBD>̵<EFBFBD>";
|
|
|
|
|
|
m_workMode = WorkMode::StockCheck;
|
|
|
|
|
|
StockCheckInfo stockCheckInfo;
|
|
|
|
|
|
stockCheckInfo.streetName = strList.at(1);
|
|
|
|
|
|
stockCheckInfo.stockNum = strList.at(2);
|
|
|
|
|
|
stockCheckInfo.taskNum = strList.at(3);
|
|
|
|
|
|
stockCheckInfo.smokeTypeNum = strList.at(4);
|
|
|
|
|
|
stockCheckInfo.smokeQuantity = strList.at(5);
|
|
|
|
|
|
m_currtStockCheckInfo = stockCheckInfo;
|
|
|
|
|
|
}
|
|
|
|
|
|
else if (strList.length() == 3 && strList.at(0) == STOCK_CHECK_HEAD)
|
|
|
|
|
|
{
|
|
|
|
|
|
qDebug() << "ȡ<EFBFBD><EFBFBD><EFBFBD><EFBFBD>λ";
|
|
|
|
|
|
QString taskNum = strList.at(2);
|
|
|
|
|
|
QString currentTaskNum = m_currtStockCheckInfo.taskNum;
|
|
|
|
|
|
if (m_workMode == WorkMode::StockCheck)
|
|
|
|
|
|
{
|
|
|
|
|
|
captureSide();
|
|
|
|
|
|
captureTop();
|
|
|
|
|
|
}
|
|
|
|
|
|
}
|
|
|
|
|
|
else if (strList.length() == 4 && strList.at(0) == INTO_STOCK_HEAD)
|
|
|
|
|
|
{
|
|
|
|
|
|
qDebug() << "<EFBFBD><EFBFBD><EFBFBD>ⶥ<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>";
|
|
|
|
|
|
m_workMode = WorkMode::IntoStock;
|
|
|
|
|
|
IntoStockInfo intoStockInfo;
|
|
|
|
|
|
intoStockInfo.streetName = strList.at(1);
|
|
|
|
|
|
intoStockInfo.taskNum = strList.at(2);
|
|
|
|
|
|
intoStockInfo.stockNum = strList.at(3);
|
|
|
|
|
|
m_currtIntoStockInfo = intoStockInfo;
|
|
|
|
|
|
captureTop();
|
|
|
|
|
|
}
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
Q_SLOT void SmokeBoxIdentification::onNumStatistic(QString typeNum)
|
|
|
|
|
|
{
|
|
|
|
|
|
qDebug() << "start num statistic";
|
|
|
|
|
|
if (!m_bDataCollection)
|
|
|
|
|
|
{
|
|
|
|
|
|
qDebug() << "typeNum: " << typeNum;
|
|
|
|
|
|
QString dataSaveFolder = QString("%1\\%2\\%3\\%4").arg(FILE_STORAGE_PATH)
|
|
|
|
|
|
.arg(m_currtStockCheckInfo.stockNum).arg(m_currtStockCheckInfo.taskNum)
|
|
|
|
|
|
.arg(m_currtStockCheckInfo.smokeQuantity);
|
|
|
|
|
|
dataPreProcessing::mkDir(dataSaveFolder);
|
|
|
|
|
|
emit sgSaveSensorData2Local(dataSaveFolder);
|
|
|
|
|
|
if (typeNum == "0")
|
|
|
|
|
|
{
|
|
|
|
|
|
m_currtStockCheckInfo.smokeQuantityRlt = "0";
|
|
|
|
|
|
m_currtStockCheckInfo.correctness = false;
|
|
|
|
|
|
emit sgSendStockCheckMsg(MsgType::StockCheckResult);
|
|
|
|
|
|
}
|
|
|
|
|
|
else
|
|
|
|
|
|
{
|
|
|
|
|
|
QString smokeCnName = m_mapSmokeCode2Name.value(typeNum);
|
|
|
|
|
|
QString templateName;
|
|
|
|
|
|
if (m_mapSmokeName2TemplateFolderName.contains(smokeCnName))
|
|
|
|
|
|
{
|
|
|
|
|
|
templateName = m_mapSmokeName2TemplateFolderName.value(smokeCnName);
|
|
|
|
|
|
QtConcurrent::run([=]() {
|
|
|
|
|
|
dataPreProcessing::QStandardnormFileMapLabelSsample m = m_pDataProcess->getDataStructure();
|
|
|
|
|
|
int matchLabel = -2;
|
|
|
|
|
|
if (m.contains(templateName))
|
|
|
|
|
|
{
|
|
|
|
|
|
qDebug() << "template find " << templateName;
|
|
|
|
|
|
matchLabel = m_pDataProcess->dataRepeatMatch(templateName);
|
|
|
|
|
|
qDebug() << "match label: " << matchLabel;
|
|
|
|
|
|
m_pDataProcess->clearComData();
|
|
|
|
|
|
}
|
|
|
|
|
|
else
|
|
|
|
|
|
{
|
|
|
|
|
|
qWarning() << "template not find " << templateName;
|
|
|
|
|
|
m_pDataProcess->clearComData();
|
|
|
|
|
|
matchLabel = -1;
|
|
|
|
|
|
}
|
|
|
|
|
|
int quantityReceive = m_currtStockCheckInfo.smokeQuantity.toInt();
|
|
|
|
|
|
if (matchLabel < 0 || matchLabel != quantityReceive)
|
|
|
|
|
|
{
|
|
|
|
|
|
m_currtStockCheckInfo.smokeQuantityRlt = "0";
|
|
|
|
|
|
m_currtStockCheckInfo.correctness = false;
|
|
|
|
|
|
}
|
|
|
|
|
|
else
|
|
|
|
|
|
{
|
|
|
|
|
|
m_currtStockCheckInfo.smokeQuantityRlt = QString::number(matchLabel);
|
|
|
|
|
|
m_currtStockCheckInfo.correctness = true;
|
|
|
|
|
|
}
|
|
|
|
|
|
emit sgSendStockCheckMsg(MsgType::StockCheckResult);
|
|
|
|
|
|
});
|
|
|
|
|
|
}
|
|
|
|
|
|
else
|
|
|
|
|
|
{
|
|
|
|
|
|
qWarning() << "Template folder not find: " << smokeCnName;
|
|
|
|
|
|
m_currtStockCheckInfo.smokeQuantityRlt = "0";
|
|
|
|
|
|
m_currtStockCheckInfo.correctness = false;
|
|
|
|
|
|
emit sgSendStockCheckMsg(MsgType::StockCheckResult);
|
|
|
|
|
|
}
|
|
|
|
|
|
}
|
|
|
|
|
|
}
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
void SmokeBoxIdentification::onSendStockCheckMsg(int mode)
|
|
|
|
|
|
{
|
|
|
|
|
|
QString resultInfoMsg;
|
|
|
|
|
|
switch (mode)
|
|
|
|
|
|
{
|
|
|
|
|
|
case EmptyCheckResult:
|
|
|
|
|
|
break;
|
|
|
|
|
|
case DataCollectionEndSC:
|
|
|
|
|
|
resultInfoMsg.append(STOCK_CHECK_HEAD).append(SPLIT_RULE).append(m_currtStockCheckInfo.streetName)
|
|
|
|
|
|
.append(SPLIT_RULE).append(m_currtStockCheckInfo.taskNum).append(END_SYMBOL);
|
|
|
|
|
|
break;
|
|
|
|
|
|
case StockCheckResult:
|
|
|
|
|
|
resultInfoMsg.append(STOCK_CHECK_HEAD).append(SPLIT_RULE)
|
|
|
|
|
|
.append(m_currtStockCheckInfo.streetName).append(SPLIT_RULE)
|
|
|
|
|
|
.append(m_currtStockCheckInfo.stockNum).append(SPLIT_RULE)
|
|
|
|
|
|
.append(m_currtStockCheckInfo.taskNum).append(SPLIT_RULE)
|
|
|
|
|
|
.append(m_currtStockCheckInfo.smokeTypeNum).append(SPLIT_RULE)
|
|
|
|
|
|
.append(m_currtStockCheckInfo.smokeQuantity).append(SPLIT_RULE)
|
|
|
|
|
|
.append(m_currtStockCheckInfo.smokeTypeNumRlt).append(SPLIT_RULE)
|
|
|
|
|
|
.append(m_currtStockCheckInfo.smokeQuantityRlt).append(SPLIT_RULE)
|
|
|
|
|
|
.append(QString::number(m_currtStockCheckInfo.correctness)).append(END_SYMBOL);
|
|
|
|
|
|
break;
|
|
|
|
|
|
case IntoStockResult:
|
|
|
|
|
|
{
|
|
|
|
|
|
resultInfoMsg.append(INTO_STOCK_HEAD).append(SPLIT_RULE)
|
|
|
|
|
|
.append(m_currtIntoStockInfo.streetName).append(SPLIT_RULE)
|
|
|
|
|
|
.append(m_currtIntoStockInfo.taskNum).append(SPLIT_RULE)
|
|
|
|
|
|
.append(m_currtIntoStockInfo.stockNum).append(SPLIT_RULE);
|
|
|
|
|
|
QStringList imagesPathList = m_currtIntoStockInfo.imagesPathList;
|
|
|
|
|
|
for (int i = 0; i < imagesPathList.length() - 1; ++i)
|
|
|
|
|
|
{
|
|
|
|
|
|
resultInfoMsg.append(imagesPathList.at(i)).append(CONTENT_SPLIT);
|
|
|
|
|
|
}
|
|
|
|
|
|
resultInfoMsg.append(imagesPathList.last()).append(END_SYMBOL);
|
|
|
|
|
|
}
|
|
|
|
|
|
break;
|
|
|
|
|
|
default:
|
|
|
|
|
|
break;
|
|
|
|
|
|
}
|
|
|
|
|
|
m_pNetControl->sendMsg(resultInfoMsg);
|
|
|
|
|
|
qDebug() << "send msg: " << resultInfoMsg;
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
Q_SLOT void SmokeBoxIdentification::onShowImage(QLabel* labelPtr, const QImage& qImage)
|
|
|
|
|
|
{
|
|
|
|
|
|
qDebug() << "Showing image";;
|
|
|
|
|
|
QRect rect = labelPtr->rect();
|
|
|
|
|
|
if (qImage.isNull())
|
|
|
|
|
|
{
|
|
|
|
|
|
qWarning() << "QImage is empty";
|
|
|
|
|
|
return;
|
|
|
|
|
|
}
|
|
|
|
|
|
QImage imgShow = qImage.scaled(QSize(rect.size()));
|
|
|
|
|
|
labelPtr->setPixmap(QPixmap::fromImage(imgShow));
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
Q_SLOT void SmokeBoxIdentification::onMonitorSensorTriggerStart(eSideID sideId)
|
|
|
|
|
|
{
|
|
|
|
|
|
if (m_workMode == WorkMode::StockCheck)
|
|
|
|
|
|
{
|
|
|
|
|
|
qDebug() << "start collect data";
|
|
|
|
|
|
emit sgMonitorSensorTriggerStart(sideId);
|
|
|
|
|
|
m_bDataCollection = true;
|
|
|
|
|
|
}
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
Q_SLOT void SmokeBoxIdentification::onMonitorSensorTriggerStop(eSideID)
|
|
|
|
|
|
{
|
|
|
|
|
|
if (m_workMode == WorkMode::StockCheck)
|
|
|
|
|
|
{
|
|
|
|
|
|
qDebug() << "stop collect data";
|
|
|
|
|
|
emit sgMonitorSensorTriggerStop();
|
|
|
|
|
|
m_bDataCollection = false;
|
|
|
|
|
|
emit sgSendData2AndAlgo();
|
|
|
|
|
|
}
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
bool SmokeBoxIdentification::decodeSmokeInfoCsv(const QString& filePath)
|
|
|
|
|
|
{
|
|
|
|
|
|
m_mapSmokeCode2Name.clear();
|
|
|
|
|
|
QXlsx::Document* pXlsxDoc = new QXlsx::Document(filePath);
|
|
|
|
|
|
if (pXlsxDoc == NULL)
|
|
|
|
|
|
{
|
|
|
|
|
|
qWarning() << "Load " << filePath << " error";
|
|
|
|
|
|
return false;
|
|
|
|
|
|
}
|
|
|
|
|
|
QString currentSheetName;
|
|
|
|
|
|
QStringList sheetNames = pXlsxDoc->sheetNames();
|
|
|
|
|
|
if (sheetNames.length() == 0)
|
|
|
|
|
|
{
|
|
|
|
|
|
qWarning() << "The quantity of xlsx's sheet is 0";
|
|
|
|
|
|
return false;
|
|
|
|
|
|
}
|
|
|
|
|
|
QXlsx::AbstractSheet* currentSheet = pXlsxDoc->sheet(sheetNames.at(0));
|
|
|
|
|
|
if (currentSheet == NULL)
|
|
|
|
|
|
return false;
|
|
|
|
|
|
|
|
|
|
|
|
// get full cells of current sheet
|
|
|
|
|
|
currentSheet->workbook()->setActiveSheet(0);
|
|
|
|
|
|
QXlsx::Worksheet* wsheet = (QXlsx::Worksheet*)currentSheet->workbook()->activeSheet();
|
|
|
|
|
|
if (wsheet == NULL)
|
|
|
|
|
|
return false;
|
|
|
|
|
|
|
|
|
|
|
|
QXlsx::CellRange cellRange = wsheet->dimension();
|
|
|
|
|
|
int rowStart = cellRange.firstRow();
|
|
|
|
|
|
int colStart = cellRange.firstColumn();
|
|
|
|
|
|
|
|
|
|
|
|
int rowEnd = cellRange.lastRow();
|
|
|
|
|
|
int colEnd = cellRange.lastColumn();
|
|
|
|
|
|
for (int i = rowStart + 2; i <= rowEnd; ++i)
|
|
|
|
|
|
{
|
|
|
|
|
|
QString smokeCode = wsheet->read(i, colStart).toString();
|
|
|
|
|
|
QString smokeName = wsheet->read(i, colStart + 1).toString();
|
|
|
|
|
|
m_mapSmokeCode2Name.insert(smokeCode, smokeName);
|
|
|
|
|
|
}
|
|
|
|
|
|
qDebug() << "Decode smokeInfo from " << filePath << " successed";
|
|
|
|
|
|
return true;
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
bool SmokeBoxIdentification::matchSmokeNameAndTemplateFolderName(const QString& folderPath)
|
|
|
|
|
|
{
|
|
|
|
|
|
m_mapSmokeName2TemplateFolderName.clear();
|
|
|
|
|
|
QDir fileDir(folderPath);
|
|
|
|
|
|
if (!fileDir.exists())
|
|
|
|
|
|
{
|
|
|
|
|
|
qWarning() << folderPath << " is not exist";
|
|
|
|
|
|
return false;
|
|
|
|
|
|
}
|
|
|
|
|
|
if (m_mapSmokeCode2Name.empty())
|
|
|
|
|
|
{
|
|
|
|
|
|
qWarning() << "Decode smokeInfo csv first!";
|
|
|
|
|
|
return false;
|
|
|
|
|
|
}
|
|
|
|
|
|
QStringList smokeCNNameList = m_mapSmokeCode2Name.values();
|
|
|
|
|
|
QFileInfoList listInfoList = fileDir.entryInfoList(QDir::Dirs | QDir::NoDotAndDotDot);
|
|
|
|
|
|
if (listInfoList.empty())
|
|
|
|
|
|
{
|
|
|
|
|
|
qDebug() << folderPath << " has not folder";
|
|
|
|
|
|
return false;
|
|
|
|
|
|
}
|
|
|
|
|
|
for each (QFileInfo typeFolder in listInfoList)
|
|
|
|
|
|
{
|
|
|
|
|
|
QString typeFolderName = typeFolder.fileName();
|
|
|
|
|
|
bool bFind = false;
|
|
|
|
|
|
for (int i = 0; i < smokeCNNameList.length(); ++i)
|
|
|
|
|
|
{
|
|
|
|
|
|
QString smokeCNName = smokeCNNameList.at(i);
|
|
|
|
|
|
if (typeFolderName.contains(smokeCNName))
|
|
|
|
|
|
{
|
|
|
|
|
|
m_mapSmokeName2TemplateFolderName.insert(smokeCNName, typeFolderName);
|
|
|
|
|
|
smokeCNNameList.removeAt(i);
|
|
|
|
|
|
bFind = true;
|
|
|
|
|
|
break;
|
|
|
|
|
|
}
|
|
|
|
|
|
}
|
|
|
|
|
|
if (!bFind)
|
|
|
|
|
|
{
|
|
|
|
|
|
qDebug() << "Not find: " << typeFolderName << "in excel";
|
|
|
|
|
|
}
|
|
|
|
|
|
}
|
|
|
|
|
|
qDebug() << "match smokeName and templateFolderName successed";
|
|
|
|
|
|
return true;
|
|
|
|
|
|
}
|