iterators.cpp

#include "html.h"
#include "iterators.h"
#include "html_tag.h"
 
litehtml::element::ptr litehtml::elements_iterator::next(bool ret_parent)
{
        next_idx();
 
        while(m_idx < (int) m_el->get_children_count())
        {
                element::ptr el = m_el->get_child(m_idx);
                if(     el->get_children_count() && m_go_inside && m_go_inside->select(el) )
                {
                        stack_item si;
                        si.idx          = m_idx;
                        si.el           = m_el;
                        m_stack.push_back(si);
                        m_el            = el;
                        m_idx           = -1;
                        if(ret_parent)
                        {
                                return el;
                        }
                        next_idx();
                } else
                {
                        if( !m_select || (m_select && m_select->select(m_el->get_child(m_idx))) )
                        {
                                return m_el->get_child(m_idx);
                        } else
                        {
                                next_idx();
                        }
                }
        }
 
        return 0;
}
 
void litehtml::elements_iterator::next_idx()
{
        m_idx++;
        while(m_idx >= (int) m_el->get_children_count() && m_stack.size())
        {
                stack_item si = m_stack.back();
                m_stack.pop_back();
                m_idx   = si.idx;
                m_el    = si.el;
                m_idx++;
                continue;
        }
}
 
 
 
bool litehtml::go_inside_inline::select(const element::ptr& el)
{
        if(el->get_display() == display_inline || el->get_display() == display_inline_text)
        {
                return true;
        }
        return false;
}
 
bool litehtml::go_inside_table::select(const element::ptr& el)
{
        if(     el->get_display() == display_table_row_group ||
                el->get_display() == display_table_header_group ||
                el->get_display() == display_table_footer_group)
        {
                return true;
        }
        return false;
}
 
bool litehtml::table_rows_selector::select(const element::ptr& el)
{
        if(     el->get_display() == display_table_row)
        {
                return true;
        }
        return false;
}
 
bool litehtml::table_cells_selector::select(const element::ptr& el)
{
        if(     el->get_display() == display_table_cell)
        {
                return true;
        }
        return false;
}