node_data.cpp 6.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300
  1. #include <assert.h>
  2. #include <iterator>
  3. #include <sstream>
  4. #include "yaml-cpp/exceptions.h"
  5. #include "yaml-cpp/node/detail/memory.h"
  6. #include "yaml-cpp/node/detail/node.h" // IWYU pragma: keep
  7. #include "yaml-cpp/node/detail/node_data.h"
  8. #include "yaml-cpp/node/detail/node_iterator.h"
  9. #include "yaml-cpp/node/ptr.h"
  10. #include "yaml-cpp/node/type.h"
  11. namespace YAML {
  12. namespace detail {
  13. std::string node_data::empty_scalar;
  14. node_data::node_data()
  15. : m_isDefined(false),
  16. m_mark(Mark::null_mark()),
  17. m_type(NodeType::Null),
  18. m_style(EmitterStyle::Default),
  19. m_seqSize(0) {}
  20. void node_data::mark_defined() {
  21. if (m_type == NodeType::Undefined)
  22. m_type = NodeType::Null;
  23. m_isDefined = true;
  24. }
  25. void node_data::set_mark(const Mark& mark) { m_mark = mark; }
  26. void node_data::set_type(NodeType::value type) {
  27. if (type == NodeType::Undefined) {
  28. m_type = type;
  29. m_isDefined = false;
  30. return;
  31. }
  32. m_isDefined = true;
  33. if (type == m_type)
  34. return;
  35. m_type = type;
  36. switch (m_type) {
  37. case NodeType::Null:
  38. break;
  39. case NodeType::Scalar:
  40. m_scalar.clear();
  41. break;
  42. case NodeType::Sequence:
  43. reset_sequence();
  44. break;
  45. case NodeType::Map:
  46. reset_map();
  47. break;
  48. case NodeType::Undefined:
  49. assert(false);
  50. break;
  51. }
  52. }
  53. void node_data::set_tag(const std::string& tag) { m_tag = tag; }
  54. void node_data::set_style(EmitterStyle::value style) { m_style = style; }
  55. void node_data::set_null() {
  56. m_isDefined = true;
  57. m_type = NodeType::Null;
  58. }
  59. void node_data::set_scalar(const std::string& scalar) {
  60. m_isDefined = true;
  61. m_type = NodeType::Scalar;
  62. m_scalar = scalar;
  63. }
  64. // size/iterator
  65. std::size_t node_data::size() const {
  66. if (!m_isDefined)
  67. return 0;
  68. switch (m_type) {
  69. case NodeType::Sequence:
  70. compute_seq_size();
  71. return m_seqSize;
  72. case NodeType::Map:
  73. compute_map_size();
  74. return m_map.size() - m_undefinedPairs.size();
  75. default:
  76. return 0;
  77. }
  78. return 0;
  79. }
  80. void node_data::compute_seq_size() const {
  81. while (m_seqSize < m_sequence.size() && m_sequence[m_seqSize]->is_defined())
  82. m_seqSize++;
  83. }
  84. void node_data::compute_map_size() const {
  85. kv_pairs::iterator it = m_undefinedPairs.begin();
  86. while (it != m_undefinedPairs.end()) {
  87. kv_pairs::iterator jt = std::next(it);
  88. if (it->first->is_defined() && it->second->is_defined())
  89. m_undefinedPairs.erase(it);
  90. it = jt;
  91. }
  92. }
  93. const_node_iterator node_data::begin() const {
  94. if (!m_isDefined)
  95. return const_node_iterator();
  96. switch (m_type) {
  97. case NodeType::Sequence:
  98. return const_node_iterator(m_sequence.begin());
  99. case NodeType::Map:
  100. return const_node_iterator(m_map.begin(), m_map.end());
  101. default:
  102. return const_node_iterator();
  103. }
  104. }
  105. node_iterator node_data::begin() {
  106. if (!m_isDefined)
  107. return node_iterator();
  108. switch (m_type) {
  109. case NodeType::Sequence:
  110. return node_iterator(m_sequence.begin());
  111. case NodeType::Map:
  112. return node_iterator(m_map.begin(), m_map.end());
  113. default:
  114. return node_iterator();
  115. }
  116. }
  117. const_node_iterator node_data::end() const {
  118. if (!m_isDefined)
  119. return const_node_iterator();
  120. switch (m_type) {
  121. case NodeType::Sequence:
  122. return const_node_iterator(m_sequence.end());
  123. case NodeType::Map:
  124. return const_node_iterator(m_map.end(), m_map.end());
  125. default:
  126. return const_node_iterator();
  127. }
  128. }
  129. node_iterator node_data::end() {
  130. if (!m_isDefined)
  131. return node_iterator();
  132. switch (m_type) {
  133. case NodeType::Sequence:
  134. return node_iterator(m_sequence.end());
  135. case NodeType::Map:
  136. return node_iterator(m_map.end(), m_map.end());
  137. default:
  138. return node_iterator();
  139. }
  140. }
  141. // sequence
  142. void node_data::push_back(node& node, shared_memory_holder /* pMemory */) {
  143. if (m_type == NodeType::Undefined || m_type == NodeType::Null) {
  144. m_type = NodeType::Sequence;
  145. reset_sequence();
  146. }
  147. if (m_type != NodeType::Sequence)
  148. throw BadPushback();
  149. m_sequence.push_back(&node);
  150. }
  151. void node_data::insert(node& key, node& value, shared_memory_holder pMemory) {
  152. switch (m_type) {
  153. case NodeType::Map:
  154. break;
  155. case NodeType::Undefined:
  156. case NodeType::Null:
  157. case NodeType::Sequence:
  158. convert_to_map(pMemory);
  159. break;
  160. case NodeType::Scalar:
  161. throw BadSubscript();
  162. }
  163. insert_map_pair(key, value);
  164. }
  165. // indexing
  166. node* node_data::get(node& key, shared_memory_holder /* pMemory */) const {
  167. if (m_type != NodeType::Map) {
  168. return NULL;
  169. }
  170. for (node_map::const_iterator it = m_map.begin(); it != m_map.end(); ++it) {
  171. if (it->first->is(key))
  172. return it->second;
  173. }
  174. return NULL;
  175. }
  176. node& node_data::get(node& key, shared_memory_holder pMemory) {
  177. switch (m_type) {
  178. case NodeType::Map:
  179. break;
  180. case NodeType::Undefined:
  181. case NodeType::Null:
  182. case NodeType::Sequence:
  183. convert_to_map(pMemory);
  184. break;
  185. case NodeType::Scalar:
  186. throw BadSubscript();
  187. }
  188. for (node_map::const_iterator it = m_map.begin(); it != m_map.end(); ++it) {
  189. if (it->first->is(key))
  190. return *it->second;
  191. }
  192. node& value = pMemory->create_node();
  193. insert_map_pair(key, value);
  194. return value;
  195. }
  196. bool node_data::remove(node& key, shared_memory_holder /* pMemory */) {
  197. if (m_type != NodeType::Map)
  198. return false;
  199. for (node_map::iterator it = m_map.begin(); it != m_map.end(); ++it) {
  200. if (it->first->is(key)) {
  201. m_map.erase(it);
  202. return true;
  203. }
  204. }
  205. return false;
  206. }
  207. void node_data::reset_sequence() {
  208. m_sequence.clear();
  209. m_seqSize = 0;
  210. }
  211. void node_data::reset_map() {
  212. m_map.clear();
  213. m_undefinedPairs.clear();
  214. }
  215. void node_data::insert_map_pair(node& key, node& value) {
  216. m_map.emplace_back(&key, &value);
  217. if (!key.is_defined() || !value.is_defined())
  218. m_undefinedPairs.emplace_back(&key, &value);
  219. }
  220. void node_data::convert_to_map(shared_memory_holder pMemory) {
  221. switch (m_type) {
  222. case NodeType::Undefined:
  223. case NodeType::Null:
  224. reset_map();
  225. m_type = NodeType::Map;
  226. break;
  227. case NodeType::Sequence:
  228. convert_sequence_to_map(pMemory);
  229. break;
  230. case NodeType::Map:
  231. break;
  232. case NodeType::Scalar:
  233. assert(false);
  234. break;
  235. }
  236. }
  237. void node_data::convert_sequence_to_map(shared_memory_holder pMemory) {
  238. assert(m_type == NodeType::Sequence);
  239. reset_map();
  240. for (std::size_t i = 0; i < m_sequence.size(); i++) {
  241. std::stringstream stream;
  242. stream << i;
  243. node& key = pMemory->create_node();
  244. key.set_scalar(stream.str());
  245. insert_map_pair(key, *m_sequence[i]);
  246. }
  247. reset_sequence();
  248. m_type = NodeType::Map;
  249. }
  250. }
  251. }