284 lines
8.8 KiB
C++
284 lines
8.8 KiB
C++
/*
|
||
* @Author: your name
|
||
* @Date: 2022-02-08 15:59:33
|
||
* @LastEditors: your name
|
||
* @LastEditTime: 2022-02-16 10:39:05
|
||
* @Description: file content
|
||
* @FilePath: \lirs\code\MyYaml\MyYaml.h
|
||
*
|
||
* Copyright © 2022 <Shandong Matrix Software Engineering Co., Ltd>
|
||
*/
|
||
#ifndef MYYAML_H_
|
||
#define MYYAML_H_
|
||
|
||
|
||
#include <mutex>
|
||
#include <fstream>
|
||
#include "yaml-cpp/yaml.h"
|
||
#include "Log.h"
|
||
|
||
namespace ai_matrix
|
||
{
|
||
|
||
struct RunConfig
|
||
{
|
||
// 识别状态
|
||
bool bRun;
|
||
|
||
};
|
||
|
||
// 基础控制参数
|
||
struct BaseConfig
|
||
{
|
||
// 股道名称
|
||
std::string strTrackName;
|
||
// 测试模式
|
||
bool bTestModel;
|
||
// Api 监听端口
|
||
int iApiPort;
|
||
// 日志文件目录
|
||
std::string strLogPath;
|
||
// 识别结果目录
|
||
std::string strResultPath;
|
||
// 调试结果目录
|
||
std::string strDebugResultPath;
|
||
// 日志存储天数
|
||
int iResultSaveDays;
|
||
};
|
||
|
||
// 日志参数
|
||
struct LogConfig {
|
||
// 输出日志级别[DEBUG, INFO, WARN, ERROR, FATAL]
|
||
std::string strOutLevel;
|
||
// 保存日志级别[DEBUG, INFO, WARN, ERROR, FATAL]
|
||
std::string strSaveLevel;
|
||
};
|
||
|
||
// 数据源参数
|
||
struct DataSourceConfig {
|
||
// 数据源地址
|
||
std::string strUrl;
|
||
// 跳帧数
|
||
int iSkipInterval;
|
||
// 行驶方向 0-自动识别 1-向左 2-向右 （与“首位信息”成对存在，形成例如向左就编号在前，向右就属性在前的对应）
|
||
int iDirection;
|
||
// 0-向左编号在前 1-向左属性在前 (向右行驶的情况：2-向右编号在前 3-向右属性在前)
|
||
int iLeftFirst;
|
||
// (向左行驶的情况：0-向左编号在前 1-向左属性在前) 2-向右编号在前 3-向右属性在前
|
||
int iRightFirst;
|
||
// 识别区域
|
||
std::vector<int> vecIdentifyAreas;
|
||
};
|
||
|
||
// 识别参数
|
||
struct IdentifyConfig {
|
||
// 运行方式
|
||
std::string strRunModel;
|
||
// 是否开启动态检测
|
||
bool bNeedMoveDetectFlag;
|
||
// 识别方向 [LEFT,RIGHT,ALL]
|
||
std::string strIdentifyDirection;
|
||
// 大框帧跨度（比一个大框从出现到消失的跨度稍大一点, 跟跳帧有关系）
|
||
int iPartitionFrameSpan;
|
||
// 大框帧跨度的位置像素差异
|
||
int iSplitFrameSpanPx;
|
||
// 集装箱大框帧跨度的像素差异值
|
||
int iContainerSplitFrameSpanPx;
|
||
// 每帧大框位置差异最小值 (持续小于此值，则可能停车)
|
||
int iChkstopPx;
|
||
// 持续X次续位置差异小于gc_chkstop_px，则判断为停车。
|
||
int iChkstopCount;
|
||
// 过滤最小大框高度（不需要的话就写个很小的值）
|
||
int iNumFrameHeight;
|
||
int iProFrameHeight;
|
||
// 过滤最大框宽度（不需要的话就写个很大的值）
|
||
int iSpaceFrameWidth;
|
||
// 是否识别车头
|
||
bool bTrainHeardDetect;
|
||
// 是否识别集装箱
|
||
bool bContainerDetect;
|
||
};
|
||
|
||
// websocket_server 的服务端参数
|
||
struct WSocketConfig {
|
||
// 是否启用
|
||
bool bIsUse;
|
||
// 端口
|
||
int iPort;
|
||
// 最大链接队列
|
||
int iMaxQueueLen;
|
||
};
|
||
|
||
// web服务器参数
|
||
struct HttpServerConfig
|
||
{
|
||
// 使用状态
|
||
bool bIsUse;
|
||
// 获取接口授权地址
|
||
std::string strIp;
|
||
// 通讯端口
|
||
int iPort;
|
||
// 获取接口授权地址
|
||
std::string strTokenUrl;
|
||
// 识别结果上传地址
|
||
std::string strUpResultUrl;
|
||
// 设备状态上传地址
|
||
std::string strUpDeviceStatusUrl;
|
||
// 接口用户名
|
||
std::string strUserName;
|
||
// 接口密码
|
||
std::string strPassword;
|
||
};
|
||
|
||
struct MinioConfig
|
||
{
|
||
// 使用状态
|
||
bool bIsUse;
|
||
// 服务地址
|
||
std::string strUrl;
|
||
// 通讯授权
|
||
std::string strAccessKey;
|
||
// 密钥授权
|
||
std::string strSecretKey;
|
||
// 容器名
|
||
std::string strBucket;
|
||
};
|
||
|
||
struct ModelConfig
|
||
{
|
||
std::string strModelPath;
|
||
float fScoreThreshold;
|
||
std::vector<std::string> vecClass;
|
||
};
|
||
|
||
|
||
class Config final
|
||
{
|
||
public:
|
||
static Config *getins();
|
||
|
||
// 读yaml文件
|
||
int readYaml(std::string &strPath);
|
||
int readModelYaml(std::string &strPath);
|
||
// 写yaml文件
|
||
int writeYaml();
|
||
std::string getStringValue(const char *pszKey, const YAML::Node *pConfig = nullptr) const;
|
||
int getIntValue(const char *pszKey, const YAML::Node *pConfig = nullptr) const;
|
||
bool getBoolValue(const char *pszKey, const YAML::Node *pConfig = nullptr) const;
|
||
float getFloatValue(const char *pszKey, const YAML::Node *pConfig = nullptr) const;
|
||
std::string getPathValue(const char *pszKey, const YAML::Node *pConfig =nullptr) const;
|
||
|
||
|
||
RunConfig getRunConfig() const;
|
||
void setRunConfig(const RunConfig runConfig);
|
||
|
||
// 获取控制参数
|
||
BaseConfig getBaseConfig() const;
|
||
void setBaseConfig(const BaseConfig baseConfig);
|
||
// 获取日志参数
|
||
LogConfig getLogConfig() const;
|
||
void setLogConfig(const LogConfig logConfig);
|
||
// 获取数据源参数
|
||
DataSourceConfig getDataSourceConfig() const;
|
||
void setDataSourceConfig(const DataSourceConfig dataSourceConfig);
|
||
// 获取辅助数据源参数
|
||
DataSourceConfig getSecondaryDataSourceConfig() const;
|
||
void setSecondaryDataSourceConfig(const DataSourceConfig dataSourceConfig);
|
||
// 获取识别参数
|
||
IdentifyConfig getIdentifyConfig() const;
|
||
void setIdentifyConfig(const IdentifyConfig identifyConfig);
|
||
// 获取websocket server 参数
|
||
WSocketConfig getWSocketConfig() const;
|
||
void setWSocketConfig(const WSocketConfig wSocketConfig);
|
||
// 获取web服务器参数
|
||
HttpServerConfig getHttpServerConfig() const;
|
||
void setHttpServerConfig(const HttpServerConfig httpServerConfig);
|
||
// 获取minio服务器参数
|
||
MinioConfig getMinioConfig() const;
|
||
void setMinioConfig(const MinioConfig minioCConfig);
|
||
// 获取来车检测模型参数
|
||
ModelConfig getModelByMoveConfig() const;
|
||
void setModelByMoveConfig(const ModelConfig modelConfig);
|
||
// 获取来车检测模型参数
|
||
ModelConfig getModelByTrainStep1Config() const;
|
||
void setModelByTrainStep1Config(const ModelConfig modelConfig);
|
||
// 获取来车检测模型参数
|
||
ModelConfig getModelByTrainStep2Config() const;
|
||
void setModelByTrainStep2Config(const ModelConfig modelConfig);
|
||
// 获取来车检测模型参数
|
||
ModelConfig getModelByContainerStep1Config() const;
|
||
void setModelByContainerStep1Config(const ModelConfig modelConfig);
|
||
// 获取来车检测模型参数
|
||
ModelConfig getModelByContainerStep2Config() const;
|
||
void setModelByContainerStep2Config(const ModelConfig modelConfig);
|
||
|
||
|
||
YAML::Node config_;
|
||
|
||
private:
|
||
Config() = default;
|
||
Config(const Config &) = delete;
|
||
Config(Config &&) = delete;
|
||
Config &operator=(const Config &) = delete;
|
||
Config &operator=(Config &&) = delete;
|
||
~Config() = default;
|
||
|
||
static Config *pInstance_;
|
||
static std::mutex mx_; //锁，保证线程安全
|
||
std::string strConfigYamlPath_;
|
||
|
||
// （全局）运行实时变量
|
||
RunConfig runConfig_;
|
||
|
||
// 控制参数
|
||
BaseConfig baseConfig_;
|
||
|
||
// 日志参数
|
||
LogConfig logConfig_;
|
||
|
||
// 数据源参数
|
||
DataSourceConfig dataSourceConfig_;
|
||
DataSourceConfig dataSourceConfig_secondary_;
|
||
|
||
// 识别参数
|
||
IdentifyConfig identifyConfig_;
|
||
|
||
// websocket server 服务端参数
|
||
WSocketConfig wSocketConfig_;
|
||
|
||
// web服务器参数
|
||
HttpServerConfig httpServerConfig_;
|
||
|
||
// MinIO服务器参数
|
||
MinioConfig minioConfig_;
|
||
|
||
// 来车检测
|
||
ModelConfig modelConfig_move_;
|
||
|
||
// 车厢第一步识别
|
||
ModelConfig modelConfig_trainStep1_;
|
||
// 车厢第二步识别
|
||
ModelConfig modelConfig_trainStep2_;
|
||
// 集装箱第一步识别
|
||
ModelConfig modelConfig_containerStep1_;
|
||
// 集装箱第二步识别
|
||
ModelConfig modelConfig_containerStep2_;
|
||
|
||
//定义一个嵌套类，负责释放内存，操作系统自动完成，不用担心内存泄露
|
||
class GarbageCollector
|
||
{
|
||
public:
|
||
~GarbageCollector()
|
||
{
|
||
if (Config::pInstance_)
|
||
{
|
||
delete Config::pInstance_;
|
||
Config::pInstance_ = nullptr;
|
||
}
|
||
}
|
||
};
|
||
static GarbageCollector gc_;
|
||
};
|
||
}
|
||
|
||
#endif |