summaryrefslogtreecommitdiff
path: root/src/theory/quantifiers/term_enumeration.h
blob: f23640e07de8498eaa87b10739c297f133303e23 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
/******************************************************************************
 * Top contributors (to current version):
 *   Andrew Reynolds, Mathias Preiner
 *
 * This file is part of the cvc5 project.
 *
 * Copyright (c) 2009-2021 by the authors listed in the file AUTHORS
 * in the top-level source directory and their institutional affiliations.
 * All rights reserved.  See the file COPYING in the top-level source
 * directory for licensing information.
 * ****************************************************************************
 *
 * Utilities for term enumeration.
 */

#include "cvc5_private.h"

#ifndef CVC5__THEORY__QUANTIFIERS__TERM_ENUMERATION_H
#define CVC5__THEORY__QUANTIFIERS__TERM_ENUMERATION_H

#include <unordered_map>
#include <vector>

#include "expr/node.h"
#include "expr/type_node.h"
#include "theory/type_enumerator.h"

namespace cvc5 {
namespace theory {
namespace quantifiers {

class QuantifiersBoundInference;

/** Term enumeration
 *
 * This class has utilities for enumerating terms. It stores
 * a cache of terms enumerated per each type.
 * It also has various utility functions regarding type
 * enumeration.
 */
class TermEnumeration
{
 public:
  TermEnumeration(QuantifiersBoundInference* qbi = nullptr);
  ~TermEnumeration() {}
  /** get i^th term for type tn */
  Node getEnumerateTerm(TypeNode tn, unsigned i);

  /** get domain
   *
   * If tn is a type such that d_qbi.mayComplete(tn) returns true, this method
   * adds all domain elements of tn to dom and returns true. Otherwise, this
   * method returns false.
   */
  bool getDomain(TypeNode tn, std::vector<Node>& dom);

 private:
  /**
   * Reference to quantifiers bound inference, which determines when it is
   * possible to enumerate the entire domain of a type. If this is not provided,
   * getDomain above always returns false.
   */
  QuantifiersBoundInference* d_qbi;
  /** ground terms enumerated for types */
  std::unordered_map<TypeNode, std::vector<Node>> d_enum_terms;
  /** map from type to the index of its type enumerator in d_typ_enum. */
  std::unordered_map<TypeNode, size_t> d_typ_enum_map;
  /** type enumerators */
  std::vector<TypeEnumerator> d_typ_enum;
};

}  // namespace quantifiers
}  // namespace theory
}  // namespace cvc5

#endif /* CVC5__THEORY__QUANTIFIERS__TERM_ENUMERATION_H */
generated by cgit on debian on lair
contact matthew@masot.net with questions or feedback