3
0
Fork 0
mirror of https://github.com/Z3Prover/z3 synced 2026-07-15 03:25:43 +00:00
z3/src/ast/decl_collector.h
copilot-swe-agent[bot] 1e07ca76af
fix: add dedicated lim_svector<sort*> m_type_vars to decl_collector and emit declare-type-var in display_decls
- decl_collector.h: add m_type_vars lim_svector, get_type_vars() getter, update reset()
- decl_collector.cpp: push poly_family_id sorts to m_type_vars instead of m_sorts;
  push_scope/pop_scope m_type_vars in push()/pop()
- ast_pp_util.h: add stacked_value<unsigned> m_type_vars counter
- ast_pp_util.cpp: emit (declare-type-var <name>) before other sort declarations
  in display_decls(); reset/push/pop the new m_type_vars counter
2026-07-13 03:27:16 +00:00

76 lines
2 KiB
C++

/*++
Copyright (c) 2011 Microsoft Corporation
Module Name:
decl_collector.h
Abstract:
Collect uninterpreted func_delcs and sorts.
This class was originally in ast_smt_pp.h
Author:
Leonardo (leonardo) 2011-10-04
Revision History:
--*/
#pragma once
#include "util/top_sort.h"
#include "util/lim_vector.h"
#include "ast/ast.h"
#include "ast/datatype_decl_plugin.h"
#include "ast/array_decl_plugin.h"
class decl_collector {
ast_manager & m;
lim_svector<sort*> m_sorts;
lim_svector<sort*> m_type_vars;
lim_svector<func_decl*> m_decls;
lim_svector<func_decl*> m_rec_decls;
ast_mark m_visited;
ast_ref_vector m_trail;
unsigned_vector m_trail_lim;
family_id m_basic_fid;
family_id m_dt_fid;
datatype_util m_dt_util;
array_util m_ar_util;
family_id m_rec_fid;
ptr_vector<ast> m_todo;
void visit_sort(sort* n);
bool is_bool(sort* s);
typedef obj_hashtable<sort> sort_set;
sort_set* collect_deps(sort* s);
void collect_deps(top_sort<sort>& st);
void collect_deps(sort* s, sort_set& set);
public:
decl_collector(ast_manager & m);
bool should_declare(func_decl* f) const;
void reset() { m_sorts.reset(); m_type_vars.reset(); m_decls.reset(); m_visited.reset(); m_trail.reset(); }
void visit_func(func_decl* n);
void visit(ast * n);
void visit(unsigned n, expr* const* es);
void visit(expr_ref_vector const& es);
void push();
void pop(unsigned n);
void order_deps(unsigned n);
unsigned get_num_sorts() const { return m_sorts.size(); }
unsigned get_num_decls() const { return m_decls.size(); }
lim_svector<sort*> const& get_sorts() const { return m_sorts; }
lim_svector<sort*> const& get_type_vars() const { return m_type_vars; }
lim_svector<func_decl*> const& get_func_decls() const { return m_decls; }
lim_svector<func_decl*> const& get_rec_decls() const { return m_rec_decls; }
};