3
0
Fork 0
mirror of https://github.com/Z3Prover/z3 synced 2025-04-08 18:31:49 +00:00
z3/lib/recurse_expr.h
Leonardo de Moura e9eab22e5c Z3 sources
Signed-off-by: Leonardo de Moura <leonardo@microsoft.com>
2012-10-02 11:35:25 -07:00

48 lines
1.2 KiB
C++

/*++
Copyright (c) 2006 Microsoft Corporation
Module Name:
recurse_expr.h
Abstract:
Traverse an expression applying a visitor.
Author:
Leonardo de Moura (leonardo) 2008-01-11.
Revision History:
--*/
#ifndef _RECURSE_EXPR_H_
#define _RECURSE_EXPR_H_
#include"ast.h"
#include"obj_hashtable.h"
template<typename T, typename Visitor, bool IgnorePatterns=false, bool CallDestructors=true>
class recurse_expr : public Visitor {
obj_map<expr, T> m_cache;
ptr_vector<expr> m_todo;
vector<T, CallDestructors> m_results1;
vector<T, CallDestructors> m_results2;
bool is_cached(expr * n) const { T c; return m_cache.find(n, c); }
T get_cached(expr * n) const { T c; m_cache.find(n, c); return c; }
void cache_result(expr * n, T c) { m_cache.insert(n, c); }
void visit(expr * n, bool & visited);
bool visit_children(expr * n);
void process(expr * n);
public:
recurse_expr(Visitor const & v = Visitor()):Visitor(v) {}
T operator()(expr * n);
void reset() { m_cache.reset(); m_todo.reset(); }
void finalize() { m_cache.finalize(); m_todo.finalize(); }
};
#endif /* _RECURSE_EXPR_H_ */