X-Git-Url: http://git.mutantstargoat.com/user/nuclear/?p=laserbrain_demo;a=blobdiff_plain;f=src%2Fscene.cc;h=484310f855b2a840112be2acff5534ee89bf7899;hp=35b562d1d18c2626e8ef9dcbd6cd8caa3fb5deac;hb=35c329e4b66fc60622080be1b9ff30148a7e74e4;hpb=14fd44f4dfec922dae20ccf1225bd1fdb5864ab9

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