rapidxml_iterators.hpp 7.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295
  1. #ifndef RAPIDXML_ITERATORS_HPP_INCLUDED
  2. #define RAPIDXML_ITERATORS_HPP_INCLUDED
  3. // Copyright (C) 2006, 2009 Marcin Kalicinski
  4. // Version 1.13
  5. // Revision $DateTime: 2009/05/15 23:02:39 $
  6. //! \file rapidxml_iterators.hpp This file contains rapidxml iterators
  7. #include "rapidxml.hpp"
  8. namespace rapidxml
  9. {
  10. const unsigned int iterate_check_name = 1 << 0;
  11. const unsigned int iterate_case_sensitive = 1 << 1;
  12. //! Iterator of child nodes of xml_node
  13. template <class Ch = char>
  14. class node_iterator
  15. {
  16. public:
  17. typedef xml_node<Ch> *value_type;
  18. typedef const value_type& reference;
  19. typedef xml_node<Ch> *pointer;
  20. typedef std::ptrdiff_t difference_type;
  21. typedef std::bidirectional_iterator_tag iterator_category;
  22. node_iterator()
  23. : m_cur(0)
  24. , m_prev(0)
  25. , m_flags(0)
  26. {
  27. }
  28. node_iterator(xml_node<Ch>* node, xml_node<Ch>* prev,
  29. unsigned char flags)
  30. : m_cur(node)
  31. , m_prev(prev)
  32. , m_flags(flags)
  33. {
  34. }
  35. reference operator*() const
  36. {
  37. assert(m_cur);
  38. return m_cur;
  39. }
  40. pointer operator->() const
  41. {
  42. assert(m_cur);
  43. return m_cur;
  44. }
  45. node_iterator& operator++()
  46. {
  47. increment();
  48. return *this;
  49. }
  50. node_iterator operator++(int)
  51. {
  52. node_iterator tmp = *this;
  53. increment();
  54. return tmp;
  55. }
  56. node_iterator& operator--()
  57. {
  58. decrement();
  59. return *this;
  60. }
  61. node_iterator operator--(int)
  62. {
  63. node_iterator tmp = *this;
  64. decrement();
  65. return tmp;
  66. }
  67. bool operator==(const node_iterator<Ch> &rhs) const
  68. {
  69. return m_cur == rhs.m_cur;
  70. }
  71. bool operator!=(const node_iterator<Ch> &rhs) const
  72. {
  73. return m_cur != rhs.m_cur;
  74. }
  75. private:
  76. void increment()
  77. {
  78. assert(m_cur && "Attempted to increment end iterator");
  79. m_prev = m_cur;
  80. if (m_flags & iterate_check_name)
  81. m_cur = m_cur->next_sibling(
  82. m_cur->name(), m_cur->name_size(),
  83. !!(m_flags & iterate_case_sensitive));
  84. else
  85. m_cur = m_cur->next_sibling();
  86. }
  87. void decrement()
  88. {
  89. assert(m_prev && "Attempted to decrement begin iterator");
  90. m_cur = m_prev;
  91. if (m_flags & iterate_check_name)
  92. m_prev = m_prev->previous_sibling(
  93. m_prev->name(), m_prev->name_size(),
  94. !!(m_flags & iterate_case_sensitive));
  95. else
  96. m_prev = m_prev->previous_sibling();
  97. }
  98. xml_node<Ch> *m_cur;
  99. xml_node<Ch> *m_prev;
  100. unsigned char m_flags;
  101. };
  102. //! Iterator of child attributes of xml_node
  103. template <class Ch = char>
  104. class attribute_iterator
  105. {
  106. public:
  107. typedef xml_attribute<Ch> *value_type;
  108. typedef const value_type& reference;
  109. typedef xml_attribute<Ch> *pointer;
  110. typedef std::ptrdiff_t difference_type;
  111. typedef std::bidirectional_iterator_tag iterator_category;
  112. attribute_iterator()
  113. : m_cur(0)
  114. , m_prev(0)
  115. , m_flags(0)
  116. {
  117. }
  118. attribute_iterator(xml_attribute<Ch>* attr, xml_attribute<Ch>* prev,
  119. unsigned char flags)
  120. : m_cur(attr)
  121. , m_prev(prev)
  122. , m_flags(flags)
  123. {
  124. }
  125. reference operator*() const
  126. {
  127. assert(m_cur);
  128. return m_cur;
  129. }
  130. pointer operator->() const
  131. {
  132. assert(m_cur);
  133. return m_cur;
  134. }
  135. attribute_iterator& operator++()
  136. {
  137. increment();
  138. return *this;
  139. }
  140. attribute_iterator operator++(int)
  141. {
  142. attribute_iterator tmp = *this;
  143. increment();
  144. return tmp;
  145. }
  146. attribute_iterator& operator--()
  147. {
  148. decrement();
  149. return *this;
  150. }
  151. attribute_iterator operator--(int)
  152. {
  153. attribute_iterator tmp = *this;
  154. decrement();
  155. return tmp;
  156. }
  157. bool operator==(const attribute_iterator<Ch> &rhs) const
  158. {
  159. return m_cur == rhs.m_cur;
  160. }
  161. bool operator!=(const attribute_iterator<Ch> &rhs) const
  162. {
  163. return m_cur != rhs.m_cur;
  164. }
  165. private:
  166. void increment()
  167. {
  168. assert(m_cur && "Attempted to increment end iterator");
  169. m_prev = m_cur;
  170. if (m_flags & iterate_check_name)
  171. m_cur = m_cur->next_attribute(
  172. m_cur->name(), m_cur->name_size(),
  173. !!(m_flags & iterate_case_sensitive));
  174. else
  175. m_cur = m_cur->next_attribute();
  176. }
  177. void decrement()
  178. {
  179. assert(m_prev && "Attempted to decrement begin iterator");
  180. m_cur = m_prev;
  181. if (m_flags & iterate_check_name)
  182. m_prev = m_prev->previous_attribute(
  183. m_prev->name(), m_prev->name_size(),
  184. !!(m_flags & iterate_case_sensitive));
  185. else
  186. m_prev = m_prev->previous_attribute();
  187. }
  188. xml_attribute<Ch>* m_cur;
  189. xml_attribute<Ch>* m_prev;
  190. unsigned char m_flags;
  191. };
  192. // Range-based for loop support
  193. template <typename Iterator>
  194. class iterator_range
  195. {
  196. public:
  197. typedef Iterator const_iterator;
  198. typedef Iterator iterator;
  199. iterator_range(Iterator first, Iterator last)
  200. : m_first(first)
  201. , m_last(last)
  202. {
  203. }
  204. Iterator begin() const { return m_first; }
  205. Iterator end() const { return m_last; }
  206. private:
  207. Iterator m_first;
  208. Iterator m_last;
  209. };
  210. template <class Ch>
  211. iterator_range<node_iterator<Ch>> nodes(const xml_node<Ch>* node,
  212. const Ch* name = 0,
  213. std::size_t name_size = 0,
  214. bool case_sensitive = true)
  215. {
  216. unsigned char flags = 0;
  217. if (name)
  218. flags |= iterate_check_name;
  219. if (case_sensitive)
  220. flags |= iterate_case_sensitive;
  221. xml_node<Ch>* first =
  222. node->first_node(name, name_size, case_sensitive);
  223. xml_node<Ch>* last = first ?
  224. node->last_node(name, name_size, case_sensitive) : nullptr;
  225. node_iterator<Ch> begin(first, 0, flags);
  226. node_iterator<Ch> end(0, last, flags);
  227. return iterator_range<node_iterator<Ch>>(begin, end);
  228. }
  229. template <class Ch>
  230. iterator_range<attribute_iterator<Ch>> attributes(const xml_node<Ch>* node,
  231. const Ch *name = 0,
  232. std::size_t name_size = 0,
  233. bool case_sensitive = true)
  234. {
  235. unsigned char flags = 0;
  236. if (name)
  237. flags |= iterate_check_name;
  238. if (case_sensitive)
  239. flags |= iterate_case_sensitive;
  240. xml_attribute<Ch>* first =
  241. node->first_attribute(name, name_size, case_sensitive);
  242. xml_attribute<Ch>* last =
  243. node->last_attribute(name, name_size, case_sensitive);
  244. attribute_iterator<Ch> begin(first, 0, flags);
  245. attribute_iterator<Ch> end(0, last, flags);
  246. return iterator_range<attribute_iterator<Ch>>(begin, end);
  247. }
  248. }
  249. #endif