diff --git a/src/tactic/generic_model_converter.cpp b/src/tactic/generic_model_converter.cpp new file mode 100644 index 000000000..f8a701003 --- /dev/null +++ b/src/tactic/generic_model_converter.cpp @@ -0,0 +1,87 @@ +/*++ +Copyright (c) 2017 Microsoft Corporation + +Module Name: + + generic_model_converter.cpp + +Abstract: + + Generic model converter that hides and adds entries. + It subsumes filter_model_converter and extension_model_converter. + +Author: + + Nikolaj Bjorner (nbjorner) 2017-10-29 + +Notes: + +--*/ +#include "ast/ast_pp.h" +#include "tactic/generic_model_converter.h" +#include "model/model_v2_pp.h" +#include "model/model_evaluator.h" + + +void generic_model_converter::operator()(model_ref & md, unsigned goal_idx) { + std::cout << "model converter\n"; + TRACE("model_converter", tout << "before generic_model_converter\n"; model_v2_pp(tout, *md); display(tout);); + model_evaluator ev(*(md.get())); + ev.set_model_completion(false); + ev.set_expand_array_equalities(false); + expr_ref val(m); + unsigned arity; + for (unsigned i = m_entries.size(); i-- > 0; ) { + entry const& e = m_entries[i]; + switch (e.m_instruction) { + case HIDE: + std::cout << "hide " << e.m_f << "\n"; + md->unregister_decl(e.m_f); + break; + case ADD: + ev(e.m_def, val); + std::cout << e.m_f << " " << e.m_def << " " << val << "\n"; + TRACE("model_converter", tout << e.m_f->get_name() << " ->\n" << e.m_def << "\n==>\n" << val << "\n";); + arity = e.m_f->get_arity(); + if (arity == 0) { + md->register_decl(e.m_f, val); + } + else { + func_interp * new_fi = alloc(func_interp, m, arity); + new_fi->set_else(val); + md->register_decl(e.m_f, new_fi); + } + break; + } + } + TRACE("model_converter", tout << "after generic_model_converter\n"; model_v2_pp(tout, *md);); +} + +void generic_model_converter::display(std::ostream & out) { + for (entry const& e : m_entries) { + switch (e.m_instruction) { + case HIDE: + display_del(out, e.m_f); + break; + case ADD: + display_add(out, m, e.m_f, e.m_def); + break; + } + } +} + +model_converter * generic_model_converter::translate(ast_translation & translator) { + ast_manager& to = translator.to(); + generic_model_converter * res = alloc(generic_model_converter, to); + for (entry const& e : m_entries) + res->m_entries.push_back(entry(translator(e.m_f.get()), translator(e.m_def.get()), to, e.m_instruction)); + return res; +} + +void generic_model_converter::collect(ast_pp_util& visitor) { + m_env = &visitor.env(); + for (entry const& e : m_entries) { + visitor.coll.visit_func(e.m_f); + if (e.m_def) visitor.coll.visit(e.m_def); + } +} diff --git a/src/tactic/generic_model_converter.h b/src/tactic/generic_model_converter.h new file mode 100644 index 000000000..434514ce6 --- /dev/null +++ b/src/tactic/generic_model_converter.h @@ -0,0 +1,62 @@ +/*++ +Copyright (c) 2017 Microsoft Corporation + +Module Name: + + generic_model_converter.h + +Abstract: + + Generic model converter that hides and adds entries. + It subsumes filter_model_converter and extension_model_converter. + +Author: + + Nikolaj Bjorner (nbjorner) 2017-10-29 + +Notes: + +--*/ +#ifndef GENERIC_MODEL_CONVERTER_H_ +#define GENERIC_MODEL_CONVERTER_H_ + +#include "tactic/model_converter.h" + +class generic_model_converter : public model_converter { + enum instruction { HIDE, ADD }; + struct entry { + func_decl_ref m_f; + expr_ref m_def; + instruction m_instruction; + entry(func_decl* f, expr* d, ast_manager& m, instruction i): + m_f(f, m), m_def(d, m), m_instruction(i) {} + }; + ast_manager& m; + vector m_entries; +public: + generic_model_converter(ast_manager & m): m(m) {} + + virtual ~generic_model_converter() { } + + void hide(func_decl * f) { m_entries.push_back(entry(f, 0, m, HIDE)); } + + void add(func_decl * d, expr* e) { m_entries.push_back(entry(d, e, m, ADD)); } + + virtual void operator()(model_ref & md, unsigned goal_idx); + + virtual void operator()(svector & labels, unsigned goal_idx) {} + + virtual void operator()(model_ref & md) { operator()(md, 0); } + + virtual void cancel() {} + + virtual void display(std::ostream & out); + + virtual model_converter * translate(ast_translation & translator); + + virtual void collect(ast_pp_util& visitor); +}; + +typedef ref generic_model_converter_ref; + +#endif