231 lines
9.1 KiB
C++
231 lines
9.1 KiB
C++
#include "VrConfig.h"
|
|
#include <iostream>
|
|
#include <vector>
|
|
#include <string>
|
|
|
|
using namespace tinyxml2;
|
|
|
|
CVrConfig::CVrConfig()
|
|
{
|
|
// 构造函数
|
|
}
|
|
|
|
CVrConfig::~CVrConfig()
|
|
{
|
|
// 析构函数
|
|
}
|
|
|
|
ConfigResult CVrConfig::LoadConfig(const std::string& filePath)
|
|
{
|
|
ConfigResult result;
|
|
|
|
// 使用tinyxml2库加载XML文件
|
|
XMLDocument doc;
|
|
XMLError err = doc.LoadFile(filePath.c_str());
|
|
if (err != XML_SUCCESS)
|
|
{
|
|
std::cerr << "open config file failed: " << filePath << std::endl;
|
|
return result;
|
|
}
|
|
|
|
// 获取根元素
|
|
XMLElement* root = doc.RootElement();
|
|
if (!root || std::string(root->Name()) != "VrConfig")
|
|
{
|
|
std::cerr << "config file format error: root element is not VrConfig" << std::endl;
|
|
return result;
|
|
}
|
|
|
|
// 解析摄像头列表
|
|
XMLElement* camerasElement = root->FirstChildElement("Cameras");
|
|
if (camerasElement)
|
|
{
|
|
XMLElement* cameraElement = camerasElement->FirstChildElement("Camera");
|
|
while (cameraElement)
|
|
{
|
|
DeviceInfo camera;
|
|
if (cameraElement->Attribute("name"))
|
|
camera.name = cameraElement->Attribute("name");
|
|
|
|
if (cameraElement->Attribute("ip"))
|
|
camera.ip = cameraElement->Attribute("ip");
|
|
|
|
result.cameraList.push_back(camera);
|
|
cameraElement = cameraElement->NextSiblingElement("Camera");
|
|
}
|
|
}
|
|
|
|
// 解析设备列表
|
|
XMLElement* devicesElement = root->FirstChildElement("Devices");
|
|
if (devicesElement)
|
|
{
|
|
XMLElement* deviceElement = devicesElement->FirstChildElement("Device");
|
|
while (deviceElement)
|
|
{
|
|
DeviceInfo device;
|
|
if (deviceElement->Attribute("name"))
|
|
device.name = deviceElement->Attribute("name");
|
|
|
|
if (deviceElement->Attribute("ip"))
|
|
device.ip = deviceElement->Attribute("ip");
|
|
|
|
result.deviceList.push_back(device);
|
|
deviceElement = deviceElement->NextSiblingElement("Device");
|
|
}
|
|
}
|
|
|
|
// 解析算法参数
|
|
XMLElement* algoParamsElement = root->FirstChildElement("AlgorithmParams");
|
|
if (algoParamsElement)
|
|
{
|
|
// 解析编织袋参数
|
|
XMLElement* bagParamElement = algoParamsElement->FirstChildElement("BagParam");
|
|
if (bagParamElement)
|
|
{
|
|
if (bagParamElement->Attribute("bagL"))
|
|
result.algorithmParams.bagParam.bagL = bagParamElement->DoubleAttribute("bagL");
|
|
if (bagParamElement->Attribute("bagW"))
|
|
result.algorithmParams.bagParam.bagW = bagParamElement->DoubleAttribute("bagW");
|
|
if (bagParamElement->Attribute("bagH"))
|
|
result.algorithmParams.bagParam.bagH = bagParamElement->DoubleAttribute("bagH");
|
|
}
|
|
|
|
// 解析滤波参数
|
|
XMLElement* filterParamElement = algoParamsElement->FirstChildElement("FilterParam");
|
|
if (filterParamElement)
|
|
{
|
|
if (filterParamElement->Attribute("continuityTh"))
|
|
result.algorithmParams.filterParam.continuityTh = filterParamElement->DoubleAttribute("continuityTh");
|
|
if (filterParamElement->Attribute("outlierTh"))
|
|
result.algorithmParams.filterParam.outlierTh = filterParamElement->IntAttribute("outlierTh");
|
|
}
|
|
|
|
// 解析角点特征参数
|
|
XMLElement* cornerParamElement = algoParamsElement->FirstChildElement("CornerParam");
|
|
if (cornerParamElement)
|
|
{
|
|
if (cornerParamElement->Attribute("minEndingGap"))
|
|
result.algorithmParams.cornerParam.minEndingGap = cornerParamElement->DoubleAttribute("minEndingGap");
|
|
if (cornerParamElement->Attribute("scale"))
|
|
result.algorithmParams.cornerParam.scale = cornerParamElement->DoubleAttribute("scale");
|
|
if (cornerParamElement->Attribute("cornerTh"))
|
|
result.algorithmParams.cornerParam.cornerTh = cornerParamElement->DoubleAttribute("cornerTh");
|
|
if (cornerParamElement->Attribute("jumpCornerTh_1"))
|
|
result.algorithmParams.cornerParam.jumpCornerTh_1 = cornerParamElement->DoubleAttribute("jumpCornerTh_1");
|
|
if (cornerParamElement->Attribute("jumpCornerTh_2"))
|
|
result.algorithmParams.cornerParam.jumpCornerTh_2 = cornerParamElement->DoubleAttribute("jumpCornerTh_2");
|
|
}
|
|
|
|
// 解析斜率参数
|
|
XMLElement* slopeParamElement = algoParamsElement->FirstChildElement("SlopeParam");
|
|
if (slopeParamElement)
|
|
{
|
|
if (slopeParamElement->Attribute("LSlopeZWin"))
|
|
result.algorithmParams.slopeParam.LSlopeZWin = slopeParamElement->DoubleAttribute("LSlopeZWin");
|
|
if (slopeParamElement->Attribute("validSlopeH"))
|
|
result.algorithmParams.slopeParam.validSlopeH = slopeParamElement->DoubleAttribute("validSlopeH");
|
|
if (slopeParamElement->Attribute("minLJumpH"))
|
|
result.algorithmParams.slopeParam.minLJumpH = slopeParamElement->DoubleAttribute("minLJumpH");
|
|
if (slopeParamElement->Attribute("minEndingGap"))
|
|
result.algorithmParams.slopeParam.minEndingGap = slopeParamElement->DoubleAttribute("minEndingGap");
|
|
}
|
|
|
|
// 解析山谷参数
|
|
XMLElement* valleyParamElement = algoParamsElement->FirstChildElement("ValleyParam");
|
|
if (valleyParamElement)
|
|
{
|
|
if (valleyParamElement->Attribute("valleyMinH"))
|
|
result.algorithmParams.valleyParam.valleyMinH = valleyParamElement->DoubleAttribute("valleyMinH");
|
|
if (valleyParamElement->Attribute("valleyMaxW"))
|
|
result.algorithmParams.valleyParam.valleyMaxW = valleyParamElement->DoubleAttribute("valleyMaxW");
|
|
}
|
|
|
|
// 解析增长参数
|
|
XMLElement* growParamElement = algoParamsElement->FirstChildElement("GrowParam");
|
|
if (growParamElement)
|
|
{
|
|
if (growParamElement->Attribute("maxLineSkipNum"))
|
|
result.algorithmParams.growParam.maxLineSkipNum = growParamElement->IntAttribute("maxLineSkipNum");
|
|
if (growParamElement->Attribute("yDeviation_max"))
|
|
result.algorithmParams.growParam.yDeviation_max = growParamElement->DoubleAttribute("yDeviation_max");
|
|
if (growParamElement->Attribute("maxSkipDistance"))
|
|
result.algorithmParams.growParam.maxSkipDistance = growParamElement->DoubleAttribute("maxSkipDistance");
|
|
if (growParamElement->Attribute("zDeviation_max"))
|
|
result.algorithmParams.growParam.zDeviation_max = growParamElement->DoubleAttribute("zDeviation_max");
|
|
if (growParamElement->Attribute("minLTypeTreeLen"))
|
|
result.algorithmParams.growParam.minLTypeTreeLen = growParamElement->DoubleAttribute("minLTypeTreeLen");
|
|
if (growParamElement->Attribute("minVTypeTreeLen"))
|
|
result.algorithmParams.growParam.minVTypeTreeLen = growParamElement->DoubleAttribute("minVTypeTreeLen");
|
|
}
|
|
|
|
// 解析平面校准参数
|
|
XMLElement* planeCalibParamElement = algoParamsElement->FirstChildElement("PlaneCalibParam");
|
|
if (planeCalibParamElement)
|
|
{
|
|
if (planeCalibParamElement->Attribute("planeHeight"))
|
|
result.algorithmParams.planeCalibParam.planeHeight = planeCalibParamElement->DoubleAttribute("planeHeight");
|
|
}
|
|
}
|
|
|
|
return result;
|
|
}
|
|
|
|
bool CVrConfig::SaveConfig(const std::string& filePath, ConfigResult& configResult)
|
|
{
|
|
// 创建XML文档
|
|
XMLDocument doc;
|
|
|
|
// 添加声明
|
|
XMLDeclaration* declaration = doc.NewDeclaration("xml version=\"1.0\" encoding=\"UTF-8\"");
|
|
doc.InsertFirstChild(declaration);
|
|
|
|
// 创建根元素
|
|
XMLElement* root = doc.NewElement("VrConfig");
|
|
doc.InsertEndChild(root);
|
|
|
|
// 添加摄像头列表
|
|
XMLElement* camerasElement = doc.NewElement("Cameras");
|
|
root->InsertEndChild(camerasElement);
|
|
|
|
for (const auto& camera : configResult.cameraList)
|
|
{
|
|
XMLElement* cameraElement = doc.NewElement("Camera");
|
|
cameraElement->SetAttribute("name", camera.name.c_str());
|
|
cameraElement->SetAttribute("ip", camera.ip.c_str());
|
|
camerasElement->InsertEndChild(cameraElement);
|
|
}
|
|
|
|
// 添加设备列表
|
|
XMLElement* devicesElement = doc.NewElement("Devices");
|
|
root->InsertEndChild(devicesElement);
|
|
|
|
for (const auto& device : configResult.deviceList)
|
|
{
|
|
XMLElement* deviceElement = doc.NewElement("Device");
|
|
deviceElement->SetAttribute("name", device.name.c_str());
|
|
deviceElement->SetAttribute("ip", device.ip.c_str());
|
|
devicesElement->InsertEndChild(deviceElement);
|
|
}
|
|
|
|
// 保存到文件
|
|
XMLError err = doc.SaveFile(filePath.c_str());
|
|
if (err != XML_SUCCESS)
|
|
{
|
|
std::cerr << "无法保存配置文件: " << filePath << std::endl;
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
|
|
/**
|
|
* @brief 创建实例
|
|
* @return 实例
|
|
*/
|
|
bool IVrConfig::CreateInstance(IVrConfig** ppVrConfig)
|
|
{
|
|
*ppVrConfig = new CVrConfig() ;
|
|
return *ppVrConfig != nullptr;
|
|
}
|