diff options
Diffstat (limited to 'src/proof/proof_utils.cpp')
-rw-r--r-- | src/proof/proof_utils.cpp | 12 |
1 files changed, 7 insertions, 5 deletions
diff --git a/src/proof/proof_utils.cpp b/src/proof/proof_utils.cpp index 3342d421a..84eefc17e 100644 --- a/src/proof/proof_utils.cpp +++ b/src/proof/proof_utils.cpp @@ -21,16 +21,18 @@ namespace CVC4 { namespace utils { -void collectAtoms(TNode node, std::set<Node>& seen) { +void collectAtoms(Environment* env, TNode node, std::set<Node>& seen) +{ if (seen.find(node) != seen.end()) return; - if (theory::Theory::theoryOf(node) != theory::THEORY_BOOL || node.isVar()) { - seen.insert(node); - return; + if (env->theoryOf(node) != theory::THEORY_BOOL || node.isVar()) + { + seen.insert(node); + return; } for (unsigned i = 0; i < node.getNumChildren(); ++i) { - collectAtoms(node[i], seen); + collectAtoms(env, node[i], seen); } } |