src/plugins/litehtml_viewer/litehtml/iterators.cpp

   1 #include "html.h"
   2 #include "iterators.h"
   3 #include "html_tag.h"
   4 #include "render_item.h"
   5 #include <iterator>
   6
   7 litehtml::elements_iterator::elements_iterator(bool return_parents, iterator_selector* go_inside, iterator_selector* select) :
   8     m_return_parent(return_parents),
   9     m_go_inside(go_inside),
  10     m_select(select)
  11 {
  12 }
  13
  14 bool litehtml::elements_iterator::go_inside(const std::shared_ptr<render_item>& el)
  15 {
  16     return      /*!el->children().empty() &&*/ m_go_inside && m_go_inside->select(el);
  17 }
  18
  19 void litehtml::elements_iterator::process(const std::shared_ptr<render_item>& container, const std::function<void (std::shared_ptr<render_item>&, iterator_item_type)>& func)
  20 {
  21     for(auto& el : container->children())
  22     {
  23         if(go_inside(el))
  24         {
  25             if(m_return_parent)
  26             {
  27                 // call function for parent
  28                 func(el, iterator_item_type_start_parent);
  29             }
  30             // go inside element and process its items
  31             process(el, func);
  32                         if(m_return_parent)
  33                         {
  34                                 // call function for parent
  35                                 func(el, iterator_item_type_end_parent);
  36                         }
  37         } else
  38         {
  39             // call function for element
  40             if(!m_select || m_select->select(el))
  41             {
  42                 func(el, iterator_item_type_child);
  43             }
  44         }
  45     }
  46 }
  47
  48 //////////////////////////////////////////////////////////////////////////
  49 //////////////////////////////////////////////////////////////////////////
  50
  51
  52 bool litehtml::go_inside_inline::select(const std::shared_ptr<render_item>& el)
  53 {
  54         if(el->src_el()->css().get_display() == display_inline && el->src_el()->css().get_float() == float_none)
  55         {
  56                 return true;
  57         }
  58         return false;
  59 }
  60
  61 bool litehtml::inline_selector::select(const std::shared_ptr<render_item>& el)
  62 {
  63     if(el->src_el()->css().get_display() == display_inline_text ||
  64         el->src_el()->css().get_display() == display_inline_table ||
  65         el->src_el()->css().get_display() == display_inline_block ||
  66         el->src_el()->css().get_display() == display_inline_flex ||
  67         el->src_el()->css().get_float() != float_none)
  68     {
  69         return true;
  70     }
  71     return false;
  72 }
  73
  74 bool litehtml::go_inside_table::select(const std::shared_ptr<render_item>& el)
  75 {
  76         if(     el->src_el()->css().get_display() == display_table_row_group ||
  77                 el->src_el()->css().get_display() == display_table_header_group ||
  78                 el->src_el()->css().get_display() == display_table_footer_group)
  79         {
  80                 return true;
  81         }
  82         return false;
  83 }
  84
  85 bool litehtml::table_rows_selector::select(const std::shared_ptr<render_item>& el)
  86 {
  87         if(     el->src_el()->css().get_display() == display_table_row)
  88         {
  89                 return true;
  90         }
  91         return false;
  92 }
  93
  94 bool litehtml::table_cells_selector::select(const std::shared_ptr<render_item>& el)
  95 {
  96         if(     el->src_el()->css().get_display() == display_table_cell)
  97         {
  98                 return true;
  99         }
 100         return false;
 101 }