feat: uiwidget::recursiveGetChildrenByStyleName

mehah · Apr 21, 2023 · f4d6bb6 · f4d6bb6
1 parent 518722f
commit f4d6bb6
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 0 deletions.
diff --git a/src/framework/luafunctions.cpp b/src/framework/luafunctions.cpp
@@ -450,6 +450,7 @@ void Application::registerLuaFunctions()
     g_lua.bindClassMemberFunction<UIWidget>("recursiveGetChildrenByPos", &UIWidget::recursiveGetChildrenByPos);
     g_lua.bindClassMemberFunction<UIWidget>("recursiveGetChildrenByMarginPos", &UIWidget::recursiveGetChildrenByMarginPos);
     g_lua.bindClassMemberFunction<UIWidget>("recursiveGetChildrenByState", &UIWidget::recursiveGetChildrenByState);
+    g_lua.bindClassMemberFunction<UIWidget>("recursiveGetChildrenByStyleName", &UIWidget::recursiveGetChildrenByStyleName);
     g_lua.bindClassMemberFunction<UIWidget>("backwardsGetWidgetById", &UIWidget::backwardsGetWidgetById);
     g_lua.bindClassMemberFunction<UIWidget>("resize", &UIWidget::resize);
     g_lua.bindClassMemberFunction<UIWidget>("move", &UIWidget::move);

diff --git a/src/framework/ui/uiwidget.cpp b/src/framework/ui/uiwidget.cpp
@@ -1385,6 +1385,16 @@ UIWidgetList UIWidget::recursiveGetChildrenByState(Fw::WidgetState state)
     return children;
 }
 
+UIWidgetList UIWidget::recursiveGetChildrenByStyleName(const std::string_view styleName)
+{
+    UIWidgetList children;
+    for (const auto& child : m_children) {
+        if (child->getStyleName() == styleName)
+            children.emplace_back(child);
+    }
+    return children;
+}
+
 UIWidgetPtr UIWidget::backwardsGetWidgetById(const std::string_view id)
 {
     UIWidgetPtr widget = getChildById(id);

diff --git a/src/framework/ui/uiwidget.h b/src/framework/ui/uiwidget.h
@@ -188,6 +188,7 @@ class UIWidget : public LuaObject
     UIWidgetList recursiveGetChildrenByPos(const Point& childPos);
     UIWidgetList recursiveGetChildrenByMarginPos(const Point& childPos);
     UIWidgetList recursiveGetChildrenByState(Fw::WidgetState state);
+    UIWidgetList recursiveGetChildrenByStyleName(const std::string_view styleName);
     UIWidgetPtr backwardsGetWidgetById(const std::string_view id);
 
     void setProp(FlagProp prop, bool v);