1 files changed, 101 insertions, 0 deletions
diff --git a/libs/litehtml/src/iterators.cpp b/libs/litehtml/src/iterators.cpp
new file mode 100644
index 0000000000..f0a87555c2
--- /dev/null
+++ b/libs/litehtml/src/iterators.cpp
@@ -0,0 +1,101 @@
+#include "html.h"
+#include "iterators.h"
+#include "html_tag.h"
+#include "render_item.h"
+#include <iterator>
+
+litehtml::elements_iterator::elements_iterator(bool return_parents, iterator_selector* go_inside, iterator_selector* select) :
+	m_go_inside(go_inside),
+    m_select(select),
+	m_return_parent(return_parents)
+{
+}
+
+bool litehtml::elements_iterator::go_inside(const std::shared_ptr<render_item>& el)
+{
+    return 	/*!el->children().empty() &&*/ m_go_inside && m_go_inside->select(el);
+}
+
+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)
+{
+    for(auto& el : container->children())
+    {
+        if(go_inside(el))
+        {
+            if(m_return_parent)
+            {
+                // call function for parent
+                func(el, iterator_item_type_start_parent);
+            }
+            // go inside element and process its items
+            process(el, func);
+			if(m_return_parent)
+			{
+				// call function for parent
+				func(el, iterator_item_type_end_parent);
+			}
+        } else
+        {
+            // call function for element
+            if(!m_select || m_select->select(el))
+            {
+                func(el, iterator_item_type_child);
+            }
+        }
+    }
+}
+
+//////////////////////////////////////////////////////////////////////////
+//////////////////////////////////////////////////////////////////////////
+
+
+bool litehtml::go_inside_inline::select(const std::shared_ptr<render_item>& el)
+{
+	if(el->src_el()->css().get_display() == display_inline && el->src_el()->css().get_float() == float_none)
+	{
+		return true;
+	}
+	return false;
+}
+
+bool litehtml::inline_selector::select(const std::shared_ptr<render_item>& el)
+{
+    if(el->src_el()->css().get_display() == display_inline_text ||
+        el->src_el()->css().get_display() == display_inline_table ||
+        el->src_el()->css().get_display() == display_inline_block ||
+        el->src_el()->css().get_display() == display_inline_flex ||
+        el->src_el()->css().get_float() != float_none)
+    {
+        return true;
+    }
+    return false;
+}
+
+bool litehtml::go_inside_table::select(const std::shared_ptr<render_item>& el)
+{
+	if(	el->src_el()->css().get_display() == display_table_row_group ||
+		el->src_el()->css().get_display() == display_table_header_group ||
+		el->src_el()->css().get_display() == display_table_footer_group)
+	{
+		return true;
+	}
+	return false;
+}
+
+bool litehtml::table_rows_selector::select(const std::shared_ptr<render_item>& el)
+{
+	if(	el->src_el()->css().get_display() == display_table_row)
+	{
+		return true;
+	}
+	return false;
+}
+
+bool litehtml::table_cells_selector::select(const std::shared_ptr<render_item>& el)
+{
+	if(	el->src_el()->css().get_display() == display_table_cell)
+	{
+		return true;
+	}
+	return false;
+}