Python创建XML文件，C++解析XML文件

最近在做一些代码移植工作，为了测试正确性，要拿同样的输入数据测试，于是打算使用XML来存储数据，C++进行解析。简单做个记录。

数据源

这里我选择用10帧图像中的人体2维关节点骨架作为数据源（可以在此下载 提取码 ersp）
shape为 10 x image_number x person_number x 17 x 2 , 分别表示帧，图像数量， 人数，关节点数，关节点图像坐标(x,y)

使用python生成xml

目标是生成下面的样子，直接看代码吧，python这个xml库用起来比较简单，看个例子就会了。

import numpy as np
from xml.dom.minidom import Document

pose_mat = np.load('pose_mat.npy', allow_pickle=True)

doc = Document()
DOCUMENT = doc.createElement('pose_mat')
doc.appendChild(DOCUMENT)

for index, pose in enumerate(pose_mat):
    frame = doc.createElement('frame')
    person_number = doc.createElement('person_number')
    person_number.appendChild(doc.createTextNode(str(len(pose))))
    frame.appendChild(person_number)
    persons = doc.createElement('persons')
    frame.appendChild(persons)
    for i, person in enumerate(pose):
        p = doc.createElement('person')

        rows = doc.createElement('rows')
        rows.appendChild(doc.createTextNode(str(person.shape[0])))
        cols = doc.createElement('cols')
        cols.appendChild(doc.createTextNode(str(person.shape[1])))
        p.appendChild(rows)
        p.appendChild(cols)
        data = doc.createElement('joints_data')
        data.appendChild(doc.createTextNode(str(person.reshape(-1).tolist())[1:-1]))
        p.appendChild(data)
        persons.appendChild(p)
        frame.appendChild(persons)

    DOCUMENT.appendChild(frame)
with open('pose_mat.xml', 'w') as f:
    doc.writexml(f)

C++解析xml (Qt QDomDocument)

刚好机器里有Qt5的环境，索性就用这个解析（挺方便），先说明一下解析之后的数据结构。这里Point2f是为了我后面其他操作做准备，也可以换成别的。

// vector 帧 vector 视角图像 vector 人 vector joints
std::vector<std::vector<std::vector<std::vector<cv::Point2f>>>>   pointsSet;

#include <QFile>
#include <QtXml/QDomDocument>
#include <iostream>
#include <opencv2/core/core.hpp>
#include <opencv2/opencv.hpp>
#include <vector>

std::vector<std::vector<std::vector<std::vector<cv::Point2f>>>>   pointsSet;
// 两个索引，方便在递归过程中定位pointsSet中的vector
int poseMatWorkingFrameIndex = -1;
int poseMatWorkingPersonIndex = -1;

// 使用递归
void iterPoseMatXMLDom(QDomElement &docElem) {
  QDomNode node = docElem.firstChild();
  if (node.toElement().isNull()) return;
  while (!node.isNull()) {
    QDomElement element = node.toElement();
    if (!element.isNull()) {
      //      std::cout << element.tagName().toStdString() << std::endl;
      if (element.tagName() == "frame") {
        std::vector<std::vector<std::vector<cv::Point2f>>>  new_frame;
        pointsSet.emplace_back(new_frame);
        poseMatWorkingFrameIndex++;
        poseMatWorkingPersonIndex = -1;
      } else if (element.tagName() == "person") {
        std::vector<std::vector<cv::Point2f>>   new_person;
        pointsSet.at(poseMatWorkingFrameIndex).emplace_back(new_person);
        poseMatWorkingPersonIndex++;

      } else if (element.tagName() == "joints_data") {
        std::vector<cv::Point2f> joints;
        QStringList paramList = element.text().split(", ");
        QStringList::const_iterator itParam = paramList.begin();
        while (itParam != paramList.end()) {
          float x = (*itParam).toFloat();
          itParam++;
          float y = (*itParam).toFloat();
          itParam++;
          joints.emplace_back(cv::Point2f(x, y));
        }
        pointsSet.at(poseMatWorkingFrameIndex)
            .at(poseMatWorkingPersonIndex).emplace_back(joints);
      }
      
      iterPoseMatXMLDom(element);
    }
    node = node.nextSibling();
  }
}

void readPoseMatXml(const QString &&filename) {
  QDomDocument doc;
  QFile file(filename);
  if (!file.open(QIODevice::ReadOnly)) return;
  if (!doc.setContent(&file)) {
    file.close();
    return;
  }
  QDomElement docElem = doc.documentElement();
  iterPoseMatXMLDom(docElem);
}

int main() {
  readPoseMatXml("/home/shuai/Desktop/pose_mat.xml");
  
  return 0;
}

如图，正常解析