graphbuilderadapter.cpp 2.7 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394
  1. #include "graphbuilderadapter.h"
  2. #include "yaml-cpp/contrib/graphbuilder.h"
  3. namespace YAML {
  4. struct Mark;
  5. int GraphBuilderAdapter::ContainerFrame::sequenceMarker;
  6. void GraphBuilderAdapter::OnNull(const Mark &mark, anchor_t anchor) {
  7. void *pParent = GetCurrentParent();
  8. void *pNode = m_builder.NewNull(mark, pParent);
  9. RegisterAnchor(anchor, pNode);
  10. DispositionNode(pNode);
  11. }
  12. void GraphBuilderAdapter::OnAlias(const Mark &mark, anchor_t anchor) {
  13. void *pReffedNode = m_anchors.Get(anchor);
  14. DispositionNode(m_builder.AnchorReference(mark, pReffedNode));
  15. }
  16. void GraphBuilderAdapter::OnScalar(const Mark &mark, const std::string &tag,
  17. anchor_t anchor, const std::string &value) {
  18. void *pParent = GetCurrentParent();
  19. void *pNode = m_builder.NewScalar(mark, tag, pParent, value);
  20. RegisterAnchor(anchor, pNode);
  21. DispositionNode(pNode);
  22. }
  23. void GraphBuilderAdapter::OnSequenceStart(const Mark &mark,
  24. const std::string &tag,
  25. anchor_t anchor,
  26. EmitterStyle::value /* style */) {
  27. void *pNode = m_builder.NewSequence(mark, tag, GetCurrentParent());
  28. m_containers.push(ContainerFrame(pNode));
  29. RegisterAnchor(anchor, pNode);
  30. }
  31. void GraphBuilderAdapter::OnSequenceEnd() {
  32. void *pSequence = m_containers.top().pContainer;
  33. m_containers.pop();
  34. DispositionNode(pSequence);
  35. }
  36. void GraphBuilderAdapter::OnMapStart(const Mark &mark, const std::string &tag,
  37. anchor_t anchor,
  38. EmitterStyle::value /* style */) {
  39. void *pNode = m_builder.NewMap(mark, tag, GetCurrentParent());
  40. m_containers.push(ContainerFrame(pNode, m_pKeyNode));
  41. m_pKeyNode = nullptr;
  42. RegisterAnchor(anchor, pNode);
  43. }
  44. void GraphBuilderAdapter::OnMapEnd() {
  45. void *pMap = m_containers.top().pContainer;
  46. m_pKeyNode = m_containers.top().pPrevKeyNode;
  47. m_containers.pop();
  48. DispositionNode(pMap);
  49. }
  50. void *GraphBuilderAdapter::GetCurrentParent() const {
  51. if (m_containers.empty()) {
  52. return nullptr;
  53. }
  54. return m_containers.top().pContainer;
  55. }
  56. void GraphBuilderAdapter::RegisterAnchor(anchor_t anchor, void *pNode) {
  57. if (anchor) {
  58. m_anchors.Register(anchor, pNode);
  59. }
  60. }
  61. void GraphBuilderAdapter::DispositionNode(void *pNode) {
  62. if (m_containers.empty()) {
  63. m_pRootNode = pNode;
  64. return;
  65. }
  66. void *pContainer = m_containers.top().pContainer;
  67. if (m_containers.top().isMap()) {
  68. if (m_pKeyNode) {
  69. m_builder.AssignInMap(pContainer, m_pKeyNode, pNode);
  70. m_pKeyNode = nullptr;
  71. } else {
  72. m_pKeyNode = pNode;
  73. }
  74. } else {
  75. m_builder.AppendToSequence(pContainer, pNode);
  76. }
  77. }
  78. } // namespace YAML