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

diff --git a/src/scene.cc b/src/scene.cc
index 35b562d..adc90fb 100644
--- a/src/scene.cc
+++ b/src/scene.cc
@@ -182,6 +182,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 +224,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;