diff options
Diffstat (limited to 'tests/cgaltest.cc')
| -rw-r--r-- | tests/cgaltest.cc | 15 |
1 files changed, 13 insertions, 2 deletions
diff --git a/tests/cgaltest.cc b/tests/cgaltest.cc index 44e87f9..f2fc169 100644 --- a/tests/cgaltest.cc +++ b/tests/cgaltest.cc @@ -66,6 +66,15 @@ void cgalTree(Tree &tree) evaluate.execute(); } +AbstractNode *find_root_tag(AbstractNode *n) +{ + foreach(AbstractNode *v, n->children) { + if (v->modinst->tag_root) return v; + if (AbstractNode *vroot = find_root_tag(v)) return vroot; + } + return NULL; +} + int main(int argc, char **argv) { if (argc != 2) { @@ -121,7 +130,6 @@ int main(int argc, char **argv) AbstractModule *root_module; ModuleInstantiation root_inst; - AbstractNode *root_node; QFileInfo fileInfo(filename); handle_dep(filename); @@ -148,7 +156,10 @@ int main(int argc, char **argv) QDir::setCurrent(fileInfo.absolutePath()); AbstractNode::resetIndexCounter(); - root_node = root_module->evaluate(&root_ctx, &root_inst); + AbstractNode *absolute_root_node = root_module->evaluate(&root_ctx, &root_inst); + AbstractNode *root_node; + // Do we have an explicit root node (! modifier)? + if (!(root_node = find_root_tag(absolute_root_node))) root_node = absolute_root_node; Tree tree(root_node); |