X-Git-Url: http://git.mutantstargoat.com/user/nuclear/?p=laserbrain_demo;a=blobdiff_plain;f=src%2Fscene.cc;h=012847c9fb2f807254705413beb4aa4a111091f8;hp=35b562d1d18c2626e8ef9dcbd6cd8caa3fb5deac;hb=e12a327cc0b4f1e59f4a66a80b170ec41ce97be6;hpb=68e71418a70dab4174981d5a579e96a9089f8682

diff --git a/src/scene.cc b/src/scene.cc
index 35b562d..012847c 100644
--- a/src/scene.cc
+++ b/src/scene.cc
@@ -2,6 +2,7 @@
 #include "scene.h"
 #include "objmesh.h"
 #include "app.h"
+#include "dbg_gui.h"
 
 static void destroy_node_tree(SceneNode *n);
 
@@ -182,6 +183,36 @@ bool Scene::have_node(SceneNode *n) const
 	return n->scene == this;
 }
 
+/* traverse scene graph and find node by name */
+static SceneNode *find_node_rec(SceneNode *tree, const char *name)
+{
+	if(strcmp(tree->get_name(), name) == 0) {
+		return tree;
+	}
+
+	int num = tree->get_num_children();
+	for(int i=0; i<num; i++) {
+		SceneNode *n = find_node_rec(tree->get_child(i), name);
+		if(n) return n;
+	}
+	return 0;
+}
+
+static SceneNode *find_node_rec(SceneNode *tree, const std::regex &re)
+{
+	if(std::regex_match(tree->get_name(), re)) {
+		return tree;
+	}
+	debug_log("no match: \"%s\"\n", tree->get_name());
+
+	int num = tree->get_num_children();
+	for(int i=0; i<num; i++) {
+		SceneNode *n = find_node_rec(tree, re);
+		if(n) return n;
+	}
+	return 0;
+}
+
 static void find_nodes_rec(std::list<SceneNode*> *res, SceneNode *tree, const std::regex &re)
 {
 	if(std::regex_match(tree->get_name(), re)) {
@@ -194,6 +225,30 @@ static void find_nodes_rec(std::list<SceneNode*> *res, SceneNode *tree, const st
 	}
 }
 
+SceneNode *Scene::find_node(const char *name) const
+{
+	if(!nodes) return 0;
+	return find_node_rec(nodes, name);
+}
+
+SceneNode *Scene::match_node(const char *qstr) const
+{
+	if(!nodes) return 0;
+
+	std::regex re{qstr};
+	return find_node_rec(nodes, re);
+}
+
+std::list<SceneNode*> Scene::match_nodes(const char *qstr) const
+{
+	std::list<SceneNode*> res;
+	if(nodes) {
+		std::regex re{qstr};
+		find_nodes_rec(&res, nodes, re);
+	}
+	return std::move(res);
+}
+
 Scene *Scene::extract_nodes(const char *qstr)
 {
 	if(!nodes) return 0;