kernel_iterator.h

#pragma once

namespace is_mesh
{
    template <class kernel_t_>
    class kernel_iterator
    {
    private:
        using element_type = typename kernel_t_::kernel_element;
    public:
        using kernel_type = kernel_t_;
        using iterator = kernel_iterator<kernel_type>;
        using kernel_element = typename kernel_type::kernel_element;
        using value_type = typename element_type::value_type;
        using key_type = typename kernel_t_::kernel_key_type;
        
    private:
        unsigned int     m_key;
        kernel_type*     m_kernel;
        
    public:
        kernel_iterator(kernel_type const * const kernel, unsigned int const & key) : m_key(key)
        {
            m_kernel = const_cast<kernel_type*>(kernel);
        }

        
    public:
        key_type     key()    const {
            return key_type{m_key};
        }
        kernel_type* kernel() const { return m_kernel; }
        
        friend bool operator==(const iterator& i, const iterator& j)
        {
            return (i.m_key == j.m_key) && (i.m_kernel == j.m_kernel);
        }
        friend bool operator!=(const iterator& i, const iterator& j)
        {
            return (i.m_key != j.m_key) || (i.m_kernel != j.m_kernel);
        }
        
        value_type* operator->()
        {
            assert(m_kernel->lookup(key_type{m_key}).state == element_type::VALID);
            return &m_kernel->m_data[m_key].value;
        }
        
        value_type& operator*()
        {
            assert(m_kernel->lookup(key_type{m_key}).state == element_type::VALID);
            return m_kernel->m_data[m_key].value;
        }
        
        iterator& operator++()
        {
            do {
                m_key++;
            } while (m_key < m_kernel->m_data.size() && m_kernel->m_data[m_key].state != element_type::VALID);
            return *this;
        }
        
        iterator operator++(int)
        {
            iterator tmp = *this;
            ++*this;
            return tmp;
        }
    };
}