#ifndef RAPIDXML_ITERATORS_HPP_INCLUDED
#define RAPIDXML_ITERATORS_HPP_INCLUDED

// Copyright (C) 2006, 2009 Marcin Kalicinski
// Version 1.13
// Revision $DateTime: 2009/05/13 01:46:17 $
//! \file rapidxml_iterators.hpp This file contains rapidxml iterators

#include "rapidxml.hpp"

namespace rapidxml {

  //! Iterator of child nodes of xml_node

  template<class Ch>
  class node_iterator {
  public:

    typedef typename xml_node<Ch> value_type;
    typedef typename xml_node<Ch> &reference;
    typedef typename xml_node<Ch> *pointer;
    typedef std::ptrdiff_t difference_type;
    typedef std::bidirectional_iterator_tag iterator_category;

    node_iterator()
    : m_node(0) {
    }

    node_iterator(xml_node<Ch> *node)
    : m_node(node->first_node()) {
    }

    reference operator *() const {
      assert(m_node);
      return *m_node;
    }

    pointer operator->() const {
      assert(m_node);
      return m_node;
    }

    node_iterator& operator++() {
      assert(m_node);
      m_node = m_node->next_sibling();
      return *this;
    }

    node_iterator operator++(int) {
      node_iterator tmp = *this;
      ++this;
      return tmp;
    }

    node_iterator& operator--() {
      assert(m_node && m_node->previous_sibling());
      m_node = m_node->previous_sibling();
      return *this;
    }

    node_iterator operator--(int) {
      node_iterator tmp = *this;
      ++this;
      return tmp;
    }

    bool operator ==(const node_iterator<Ch> &rhs) {
      return m_node == rhs.m_node;
    }

    bool operator !=(const node_iterator<Ch> &rhs) {
      return m_node != rhs.m_node;
    }

  private:

    xml_node<Ch> *m_node;

  };

  //! Iterator of child attributes of xml_node

  template<class Ch>
  class attribute_iterator {
  public:

    typedef typename xml_attribute<Ch> value_type;
    typedef typename xml_attribute<Ch> &reference;
    typedef typename xml_attribute<Ch> *pointer;
    typedef std::ptrdiff_t difference_type;
    typedef std::bidirectional_iterator_tag iterator_category;

    attribute_iterator()
    : m_attribute(0) {
    }

    attribute_iterator(xml_node<Ch> *node)
    : m_attribute(node->first_attribute()) {
    }

    reference operator *() const {
      assert(m_attribute);
      return *m_attribute;
    }

    pointer operator->() const {
      assert(m_attribute);
      return m_attribute;
    }

    attribute_iterator& operator++() {
      assert(m_attribute);
      m_attribute = m_attribute->next_attribute();
      return *this;
    }

    attribute_iterator operator++(int) {
      attribute_iterator tmp = *this;
      ++this;
      return tmp;
    }

    attribute_iterator& operator--() {
      assert(m_attribute && m_attribute->previous_attribute());
      m_attribute = m_attribute->previous_attribute();
      return *this;
    }

    attribute_iterator operator--(int) {
      attribute_iterator tmp = *this;
      ++this;
      return tmp;
    }

    bool operator ==(const attribute_iterator<Ch> &rhs) {
      return m_attribute == rhs.m_attribute;
    }

    bool operator !=(const attribute_iterator<Ch> &rhs) {
      return m_attribute != rhs.m_attribute;
    }

  private:

    xml_attribute<Ch> *m_attribute;

  };

}

#endif
