首页 > 解决方案 > Yaml-cpp:在发射器中的每个新地图之前禁用自动 begindoc (---)

问题描述

我正在使用 yaml-cpp 生成一个结构日志文件,稍后我可以再次读取该文件。它可能不是这项工作的最佳工具,但现在我只需要一些东西来开始。

但是,我遇到了一些问题,如此示例输出所示:

---
0: 42
---
1: 42
---
2: 42
---
3: 42
---
4: 42
...

我想要的是这样的:

---
0: 42
1: 42
2: 42
3: 42
4: 42
...

这是生成它的代码:

YAML::Emitter res;
res << YAML::BeginDoc;
for (int i = 0; i < 5; i++)
{
    res << YAML::BeginMap
        << YAML::Key << i
        << YAML::Value << 42
        << YAML::EndMap;
}
res << YAML::EndDoc;
std::cout << res.c_str() << std::endl;

我查看了源代码(https://github.com/jbeder/yaml-cpp/blob/master/src/emitter.cpp)并发现了这一点:

// EmitBeginDoc
void Emitter::EmitBeginDoc() {
  if (!good())
    return;

  if (m_pState->CurGroupType() != GroupType::NoType) {
    m_pState->SetError("Unexpected begin document");
    return;
  }

  if (m_pState->HasAnchor() || m_pState->HasTag()) {
    m_pState->SetError("Unexpected begin document");
    return;
  }

  if (m_stream.col() > 0)
    m_stream << "\n";
  m_stream << "---\n";

  m_pState->StartedDoc();
}

但我从这里所做的超出了我的范围。一方面,我可以做一个小技巧,然后在将输出写入文件之前从输出中删除所有 ---(当然,第一个除外),但另一方面,我只是假设我一定忽略了一些细节可以更优雅地解决这个问题。

编辑: 我的例子不能证明地图是合理的,所以我会进一步解释。我正在为大量图像存储一些元信息。当我读回它们时,我想直接访问每个图像的信息,即 myyaml[frame_id]["algorithm_2"]["result_x"] 其中 frame_id 可以是例如 3。我不知道该怎么做如果每个图像在文件中都有自己的文档,那么每次查找都不需要遍历整个文件。图像不一定是从 0 到 number_of_images 的完整序列。

编辑 2: 参考 João Augusto 编辑的答案,这里有一个更详细的示例,它也不起作用:

YAML::Emitter res;
res << YAML::BeginDoc;
for (int i = 3; i < 7; i++)
{
    res << YAML::BeginMap
        << YAML::Key << i
        << YAML::Value
            << YAML::BeginSeq
            << YAML::BeginMap
                << YAML::Key << "Algorithm 1"
                << YAML::Value
                << YAML::BeginSeq
                    << YAML::Flow
                    << YAML::BeginSeq << 54*i << 42/i << 10+i << 17-i << YAML::EndSeq
                    << YAML::Flow
                    << YAML::BeginSeq << 6*i << 3/i << 87+i << 33-i << YAML::EndSeq
                << YAML::EndSeq
            << YAML::EndMap
            << YAML::BeginMap
                << YAML::Key << "Algorithm 2"
                << YAML::Value
                << YAML::BeginSeq
                    << YAML::Flow
                    << YAML::BeginSeq << 65*i << 27/i << 54+i << 76-i << YAML::EndSeq
                    << YAML::Flow
                    << YAML::BeginSeq << 45*i << 66/i << 98+i << 34-i << YAML::EndSeq
                << YAML::EndSeq
            << YAML::EndMap
            << YAML::EndSeq
        << YAML::EndMap;
}
res << YAML::EndDoc;
std::cout << res.c_str() << std::endl;

这是输出:

---
3:
  - Algorithm 1:
      - [162, 14, 13, 14]
      - [18, 1, 90, 30]
  - Algorithm 2:
      - [195, 9, 57, 73]
      - [135, 22, 101, 31]
---
4:
  - Algorithm 1:
      - [216, 10, 14, 13]
      - [24, 0, 91, 29]
  - Algorithm 2:
      - [260, 6, 58, 72]
      - [180, 16, 102, 30]
---
5:
  - Algorithm 1:
      - [270, 8, 15, 12]
      - [30, 0, 92, 28]
  - Algorithm 2:
      - [325, 5, 59, 71]
      - [225, 13, 103, 29]
---
6:
  - Algorithm 1:
      - [324, 7, 16, 11]
      - [36, 0, 93, 27]
  - Algorithm 2:
      - [390, 4, 60, 70]
      - [270, 11, 104, 28]
...

其中仍然有不需要的新文档语句。我需要在我的代码中进行哪些更改才能使它们不出现?

标签: c++yamlyaml-cpp

解决方案


您应该阅读 yaml 格式的文档。您正在为每个项目创建一个地图,您想要的是这样的。

YAML::Emitter res;
res << YAML::BeginDoc << YAML::BeginMap;
for (int i = 0; i < 5; i++)
{
    res << YAML::Key << i
        << YAML::Value << 42;

}
res << YAML::EndMap << YAML::EndDoc;

编辑...

您只需要以您想要访问数据的方式编写 yaml,所以如果您执行以下操作:

---
frame_0:
  algorithm_1: 1
  algorithm_2: 2
frame_1:
  algorithm_1: 10
  algorithm_2: 2
frame_2:
  algorithm_1: 8
  algorithm_2: 22
frame_3:
  algorithm_1: 1
  algorithm_2: 23
frame_4:
  algorithm_1: 12
  algorithm_2: 21
...

例如,您可以通过执行以下操作访问 frame_3 中 algorithm_2 的值:

YAML::Node yaml = YAML::Load(res.c_str());
int value = yaml["frame_3"]["algorithm_2"].as<int>();

编辑...

YAML::Emitter res;
res << YAML::BeginDoc;
res << YAML::BeginMap;
for (int i = 3; i < 7; i++)
{

    res << YAML::Key << i
        << YAML::Value
        << YAML::BeginSeq
        << YAML::BeginMap
        << YAML::Key << "Algorithm 1"
        << YAML::Value
        << YAML::BeginSeq
        << YAML::Flow
        << YAML::BeginSeq << 54 * i << 42 / i << 10 + i << 17 - i << YAML::EndSeq
        << YAML::Flow
        << YAML::BeginSeq << 6 * i << 3 / i << 87 + i << 33 - i << YAML::EndSeq
        << YAML::EndSeq
        << YAML::EndMap
        << YAML::BeginMap
        << YAML::Key << "Algorithm 2"
        << YAML::Value
        << YAML::BeginSeq
        << YAML::Flow
        << YAML::BeginSeq << 65 * i << 27 / i << 54 + i << 76 - i << YAML::EndSeq
        << YAML::Flow
        << YAML::BeginSeq << 45 * i << 66 / i << 98 + i << 34 - i << YAML::EndSeq
        << YAML::EndSeq
        << YAML::EndMap
        << YAML::EndSeq;
}
res << YAML::EndMap;
res << YAML::EndDoc;

这将满足您的要求,您的问题是(假设您想要一个带有 3 到 7 键的映射)您正在为 3、4、5、6 和 7 创建一个映射,而不是一个带有键 3 的映射, 4,...,7


推荐阅读