在yaml-cpp中操作节点

时间:2018-03-27 13:37:02

标签: c++ yaml yaml-cpp

我目前正在开发一个项目,我想使用库yaml-cpp进行配置。现在我试图编写一个给出标签/字符串向量(例如“A”,“B”,“C”)和一些参数值(例如31)的函数。然后,该函数应为每个标记插入一个映射节点(如果此类节点尚不存在),将其映射到后续节点,最后创建一个带有参数值的标量节点。在所述示例调用之后,配置应该如此读取

A:
  B:
    C: 31

下面你可以看到一个干净简单的实现尝试:

Config.hpp

class Config {

public:

    template<typename T>
    static void setConfigParam(const std::vector<std::string> &tags,
            T value) {
        assert(tags.size() > 0);
        assert(config);
        YAML::Node parentNode = config;
        for(int i=0; i < tags.size()-1; i++) {
            const std::string & tag = tags.at(i);
            if(parentNode[tag]) {
                YAML::Node childNode = parentNode[tag];
                parentNode = childNode;
            }
            else {
                YAML::Node childNode(YAML::NodeType::Map);
                parentNode[tag] = childNode;
                parentNode = childNode;
            }
        }
        parentNode[tags.back()] = value;
        std::cout << "Config:\n" << config << "\n\n";
    }

private:

    static YAML::Node config;

};

YAML::Node Config::config(YAML::NodeType::Map);

Test.cpp的

#include "Config.hpp"

int main(int argc, char **argv) {
    int val1 = 3;
    vector<string> tags1 = {"A1","B1","C1","D1"};
    Config::setConfigParam(tags1, val1);
    string val2 = "test";
    vector<string> tags2 = {"A1","B1","C1","D2"};
    Config::setConfigParam(tags2, val2);
    bool val3 = true;
    vector<string> tags3 = {"A1","B1","C1"};
    Config::setConfigParam(tags3, val3);
    double val4 = 2.385;
    vector<string> tags4 = {"A1","B2","C1"};
    Config::setConfigParam(tags4, val4);
}

函数setConfigParam()的问题似乎是行

YAML::Node parentNode = config;

parentNode = childNode

根据我对tutorial的理解,第一行显然创建了节点parentNode作为config的别名。在每次函数调用时,第二行的第一次执行然后用新创建的config替换现有的childNode对象。但是,在剩余的迭代中,config不再完全重置,而是按预期进行调整。

如果一切正常,输出应为:

Config:
A1:
  B1:
    C1:
      D1: 3

Config:
A1:
  B1:
    C1:
      D1: 3
      D2: test

Config:
A1:
  B1:
    C1: true

Config:
A1:
  B1:
    C1: true
  B2:
    C1: 2.385

上面代码的实际输出是

Config:
B1:
  C1:
    D1: 3

Config:
B1:
  C1:
    D2: test

Config:
B1:
  C1: true

Config:
B2:
  C1: 2.385

我试图通过使用指针来防止上述问题,最终设法提出以下实现,至少在测试代码中正常工作:

template<typename T>
static void setConfigParam(const std::vector<std::string> &tags,
        T value) {
    int numTags = tags.size();
    // Reads root node.
    assert(numTags > 0);
    assert(config);
    YAML::Node* parentNode = &config;
    // Determines index of first non-existing node.
    int i=0;
    for(; i < numTags-1; i++) {
        const std::string & tag = tags.at(i);
        if((*parentNode)[tag]) {
            auto childNode = (*parentNode)[tag];
            parentNode = &childNode;
        }
        else {
            break;
        }
    }
    // Note: necessary because *parentNode will later point
    // to a different node due to lib behavior .
    YAML::Node lastExistingNode = *parentNode;
    // Sets node value and creates missing nodes.
    if(i == numTags - 1) {
        lastExistingNode[tags.back()] = value;
    }
    else {
        YAML::Node newNode;
        newNode = value;
        for(int j = tags.size()-1; j>i; j--) {
            YAML::Node tmpNode;
            tmpNode[tags.at(j)] = newNode;
            newNode = tmpNode;
        }
        // Inserts missing nodes.
        lastExistingNode[tags.at(i)] = newNode;
    }
    std::cout << "Config:\n" << config << "\n\n";
}

但是,我认为必须有一个更简单,更清洁的解决方案。因此,我非常感谢有关如何通过更简单的实现使函数正常工作的任何想法,特别是因为没有太多关于库的文档。

1 个答案:

答案 0 :(得分:1)

在第一个示例中,分配给现有节点的行为与您的意图不同。有关详细信息,请参阅this issue

你的指针示例解决了这个问题,但它有另一个问题:它取消引用堆栈变量。

相反,使用递归解决方案:

template<typename T, typename Iter>
void setConfigParam(YAML::Node node, Iter begin, Iter end, T value) {
  if (begin == end) {
    return;
  }
  const auto& tag = *begin;
  if (std::next(begin) == end) {
    node[tag] = value;
    return;
  }
  if (!node[tag]) {
    node[tag] = YAML::Node(YAML::NodeType::Map);
  }
  setConfigParam(node[tag], std::next(begin), end, value);
}