elem_center_iterator.hpp

Go to the documentation of this file.

 
#ifndef NIHU_ELEM_CENTER_ITERATOR_HPP_INCLUDED
#define NIHU_ELEM_CENTER_ITERATOR_HPP_INCLUDED
 
#include <iterator>
 
namespace NiHu
{
namespace fmm
{
 
template <class It>
class elem_center_iterator
    : public std::iterator<std::forward_iterator_tag, typename It::value_type::x_t>
    , private It
{
    typedef typename It::value_type elem_t;
    typedef typename elem_t::x_t x_t;
 
public:
    elem_center_iterator(It it = It())
        : It(it)
    {
    }
 
    It const &base() const
    {
        return static_cast<It const &>(*this);
    }
 
    It &base()
    {
        return static_cast<It &>(*this);
    }
 
    ptrdiff_t operator-(elem_center_iterator const &other) const
    {
        return base() - other;
    }
 
    elem_center_iterator operator-(ptrdiff_t diff) const
    {
        return base() - diff;
    }
 
    elem_center_iterator operator+(ptrdiff_t diff) const
    {
        return base() + diff;
    }
 
    elem_center_iterator &operator++()
    {
        return static_cast<elem_center_iterator &>(++base());
    }
 
    elem_center_iterator operator++(int)
    {
        return base()++;
    }
 
    x_t const &operator*() const
    {
        return (*base()).get_center();
    }
 
    x_t const &operator[](size_t idx) const
    {
        return base()[idx].get_center();
    }
 
    bool operator != (elem_center_iterator const& other) const
    {
        return base() != other;
    }
 
    bool operator == (elem_center_iterator const& other) const
    {
        return base() == other;
    }
};
 
template <class It>
auto create_elem_center_iterator(It it)
{
    return elem_center_iterator<It>(it);
}
 
 
template <class It>
class field_center_iterator
    : public std::iterator<std::forward_iterator_tag, typename It::value_type::elem_t::x_t>
    , private It
{
    typedef typename It::value_type field_t;
    typedef typename field_t::elem_t elem_t;
    typedef typename elem_t::x_t x_t;
 
public:
    field_center_iterator(It it = It())
        : It(it)
    {
    }
 
    It const &base() const
    {
        return *static_cast<It const *>(this);
    }
 
    It &base()
    {
        return *static_cast<It *>(this);
    }
 
    ptrdiff_t operator-(field_center_iterator const &other) const
    {
        return base() - other;
    }
 
    field_center_iterator operator-(ptrdiff_t diff) const
    {
        return base() - diff;
    }
 
    field_center_iterator operator+(ptrdiff_t diff) const
    {
        return base() + diff;
    }
 
    field_center_iterator &operator++()
    {
        return static_cast<field_center_iterator &>(++base());
    }
 
    field_center_iterator operator++(int)
    {
        return base()++;
    }
 
    x_t const &operator*() const
    {
        return (*base()).get_elem().get_center();
    }
 
    x_t const &operator[](size_t idx) const
    {
        return base()[idx].get_elem().get_center();
    }
 
    bool operator != (field_center_iterator const& other) const
    {
        return base() != other;
    }
 
    bool operator == (field_center_iterator const& other) const
    {
        return base() == other;
    }
};
 
template <class It>
auto create_field_center_iterator(It it)
{
    return field_center_iterator<It>(it);
}
 
 
} // end of namespace fmm
} // end of namespace NiHu
 
 
#endif /* NIHU_ELEM_CENTER_ITERATOR_HPP_INCLUDED */