GCC Code Coverage Report
Directory: . Exec Total Coverage
File: src/theory/term_registration_visitor.cpp Lines: 109 136 80.1 %
Date: 2021-08-16 Branches: 232 562 41.3 %

Line Exec Source
1
/******************************************************************************
2
 * Top contributors (to current version):
3
 *   Andrew Reynolds, Dejan Jovanovic, Morgan Deters
4
 *
5
 * This file is part of the cvc5 project.
6
 *
7
 * Copyright (c) 2009-2021 by the authors listed in the file AUTHORS
8
 * in the top-level source directory and their institutional affiliations.
9
 * All rights reserved.  See the file COPYING in the top-level source
10
 * directory for licensing information.
11
 * ****************************************************************************
12
 *
13
 * [[ Add lengthier description here ]]
14
 * \todo document this file
15
 */
16
17
#include "theory/term_registration_visitor.h"
18
19
#include "base/configuration.h"
20
#include "options/quantifiers_options.h"
21
#include "smt/logic_exception.h"
22
#include "theory/theory_engine.h"
23
24
using namespace cvc5::theory;
25
26
namespace cvc5 {
27
28
std::string PreRegisterVisitor::toString() const {
29
  std::stringstream ss;
30
  TNodeToTheorySetMap::const_iterator it = d_visited.begin();
31
  for (; it != d_visited.end(); ++ it) {
32
    ss << (*it).first << ": " << TheoryIdSetUtil::setToString((*it).second)
33
       << std::endl;
34
  }
35
  return ss.str();
36
}
37
38
/**
39
 * Return true if we already visited the term current with the given parent,
40
 * assuming that the set of theories in visitedTheories has already processed
41
 * current. This method is used by PreRegisterVisitor and SharedTermsVisitor
42
 * below.
43
 */
44
6535730
bool isAlreadyVisited(TheoryEngine* te,
45
                      TheoryIdSet visitedTheories,
46
                      TNode current,
47
                      TNode parent)
48
{
49
6535730
  TheoryId currentTheoryId = Theory::theoryOf(current);
50
6535730
  if (!TheoryIdSetUtil::setContains(currentTheoryId, visitedTheories))
51
  {
52
    // current theory not visited, return false
53
    return false;
54
  }
55
56
6535730
  if (current == parent)
57
  {
58
    // top-level and current visited, return true
59
1051792
    return true;
60
  }
61
62
  // The current theory has already visited it, so now it depends on the parent
63
  // and the type
64
5483938
  TheoryId parentTheoryId = Theory::theoryOf(parent);
65
5483938
  if (!TheoryIdSetUtil::setContains(parentTheoryId, visitedTheories))
66
  {
67
    // parent theory not visited, return false
68
100313
    return false;
69
  }
70
71
  // do we need to consider the type?
72
10767250
  TypeNode type = current.getType();
73
5383625
  if (currentTheoryId == parentTheoryId && !te->isFiniteType(type))
74
  {
75
    // current and parent are the same theory, and we are infinite, return true
76
3222921
    return true;
77
  }
78
2160704
  TheoryId typeTheoryId = Theory::theoryOf(type);
79
2160704
  return TheoryIdSetUtil::setContains(typeTheoryId, visitedTheories);
80
}
81
82
1605717
bool PreRegisterVisitor::alreadyVisited(TNode current, TNode parent) {
83
84
1605717
  Debug("register::internal") << "PreRegisterVisitor::alreadyVisited(" << current << "," << parent << ")" << std::endl;
85
86
3211434
  if ((parent.isClosure()
87
1592304
       || parent.getKind() == kind::SEP_STAR
88
1592304
       || parent.getKind() == kind::SEP_WAND
89
3198021
       || (parent.getKind() == kind::SEP_LABEL && current.getType().isBoolean())
90
       // parent.getKind() == kind::CARDINALITY_CONSTRAINT
91
       )
92
3224847
      && current != parent)
93
  {
94
5385
    Debug("register::internal") << "quantifier:true" << std::endl;
95
5385
    return true;
96
  }
97
98
  // Get the theories that have already visited this node
99
1600332
  TNodeToTheorySetMap::iterator find = d_visited.find(current);
100
1600332
  if (find == d_visited.end()) {
101
    // not visited at all, return false
102
859725
    return false;
103
  }
104
105
740607
  TheoryIdSet visitedTheories = (*find).second;
106
740607
  return isAlreadyVisited(d_engine, visitedTheories, current, parent);
107
}
108
109
356036
void PreRegisterVisitor::visit(TNode current, TNode parent) {
110
111
356036
  Debug("register") << "PreRegisterVisitor::visit(" << current << "," << parent << ")" << std::endl;
112
356036
  if (Debug.isOn("register::internal")) {
113
    Debug("register::internal") << toString() << std::endl;
114
  }
115
116
  // get the theories we already preregistered with
117
356036
  TheoryIdSet visitedTheories = d_visited[current];
118
119
  // call the preregistration on current, parent or type theories and update
120
  // visitedTheories. The set of preregistering theories coincides with
121
  // visitedTheories here.
122
356037
  preRegister(d_engine, visitedTheories, current, parent, visitedTheories);
123
124
712070
  Debug("register::internal")
125
356035
      << "PreRegisterVisitor::visit(" << current << "," << parent
126
356035
      << "): now registered with "
127
356035
      << TheoryIdSetUtil::setToString(visitedTheories) << std::endl;
128
  // update the theories set for current
129
356035
  d_visited[current] = visitedTheories;
130
356035
  Assert(d_visited.find(current) != d_visited.end());
131
356035
  Assert(alreadyVisited(current, parent));
132
356035
}
133
134
5326910
void PreRegisterVisitor::preRegister(TheoryEngine* te,
135
                                     TheoryIdSet& visitedTheories,
136
                                     TNode current,
137
                                     TNode parent,
138
                                     TheoryIdSet preregTheories)
139
{
140
  // Preregister with the current theory, if necessary
141
5326910
  TheoryId currentTheoryId = Theory::theoryOf(current);
142
5326914
  preRegisterWithTheory(
143
      te, visitedTheories, currentTheoryId, current, parent, preregTheories);
144
145
5326906
  if (current != parent)
146
  {
147
    // preregister with parent theory, if necessary
148
4276596
    TheoryId parentTheoryId = Theory::theoryOf(parent);
149
4276596
    preRegisterWithTheory(
150
        te, visitedTheories, parentTheoryId, current, parent, preregTheories);
151
152
    // Note that if enclosed by different theories it's shared, for example,
153
    // in read(a, f(a)), f(a) should be shared with integers.
154
8553192
    TypeNode type = current.getType();
155
4276596
    if (currentTheoryId != parentTheoryId || te->isFiniteType(type))
156
    {
157
      // preregister with the type's theory, if necessary
158
1547392
      TheoryId typeTheoryId = Theory::theoryOf(type);
159
1547392
      preRegisterWithTheory(
160
          te, visitedTheories, typeTheoryId, current, parent, preregTheories);
161
    }
162
  }
163
5326906
}
164
11150898
void PreRegisterVisitor::preRegisterWithTheory(TheoryEngine* te,
165
                                               TheoryIdSet& visitedTheories,
166
                                               TheoryId id,
167
                                               TNode current,
168
                                               TNode parent,
169
                                               TheoryIdSet preregTheories)
170
{
171
11150898
  if (TheoryIdSetUtil::setContains(id, visitedTheories))
172
  {
173
    // already visited
174
4698494
    return;
175
  }
176
6452404
  visitedTheories = TheoryIdSetUtil::setInsert(id, visitedTheories);
177
6452404
  if (TheoryIdSetUtil::setContains(id, preregTheories))
178
  {
179
    // already pregregistered
180
4466563
    return;
181
  }
182
1985841
  if (Configuration::isAssertionBuild())
183
  {
184
3971682
    Debug("register::internal")
185
1985841
        << "PreRegisterVisitor::visit(" << current << "," << parent
186
1985841
        << "): adding " << id << std::endl;
187
    // This should never throw an exception, since theories should be
188
    // guaranteed to be initialized.
189
    // These checks don't work with finite model finding, because it
190
    // uses Rational constants to represent cardinality constraints,
191
    // even though arithmetic isn't actually involved.
192
1985841
    if (!options::finiteModelFind())
193
    {
194
1899822
      if (!te->isTheoryEnabled(id))
195
      {
196
        const LogicInfo& l = te->getLogicInfo();
197
        LogicInfo newLogicInfo = l.getUnlockedCopy();
198
        newLogicInfo.enableTheory(id);
199
        newLogicInfo.lock();
200
        std::stringstream ss;
201
        ss << "The logic was specified as " << l.getLogicString()
202
           << ", which doesn't include " << id
203
           << ", but found a term in that theory." << std::endl
204
           << "You might want to extend your logic to "
205
           << newLogicInfo.getLogicString() << std::endl;
206
        throw LogicException(ss.str());
207
      }
208
    }
209
  }
210
  // call the theory's preRegisterTerm method
211
1985841
  Theory* th = te->theoryOf(id);
212
1985845
  th->preRegisterTerm(current);
213
}
214
215
210515
void PreRegisterVisitor::start(TNode node) {}
216
217
std::string SharedTermsVisitor::toString() const {
218
  std::stringstream ss;
219
  TNodeVisitedMap::const_iterator it = d_visited.begin();
220
  for (; it != d_visited.end(); ++ it) {
221
    ss << (*it).first << ": " << TheoryIdSetUtil::setToString((*it).second)
222
       << std::endl;
223
  }
224
  return ss.str();
225
}
226
227
20093340
bool SharedTermsVisitor::alreadyVisited(TNode current, TNode parent) const {
228
229
20093340
  Debug("register::internal") << "SharedTermsVisitor::alreadyVisited(" << current << "," << parent << ")" << std::endl;
230
231
40186680
  if ((parent.isClosure()
232
19939749
       || parent.getKind() == kind::SEP_STAR
233
19939084
       || parent.getKind() == kind::SEP_WAND
234
40032295
       || (parent.getKind() == kind::SEP_LABEL && current.getType().isBoolean())
235
       // parent.getKind() == kind::CARDINALITY_CONSTRAINT
236
       )
237
40343861
      && current != parent)
238
  {
239
66808
    Debug("register::internal") << "quantifier:true" << std::endl;
240
66808
    return true;
241
  }
242
20026532
  TNodeVisitedMap::const_iterator find = d_visited.find(current);
243
  // If node is not visited at all, just return false
244
20026532
  if (find == d_visited.end()) {
245
14231409
    Debug("register::internal") << "1:false" << std::endl;
246
14231409
    return false;
247
  }
248
249
5795123
  TheoryIdSet visitedTheories = (*find).second;
250
5795123
  return isAlreadyVisited(d_engine, visitedTheories, current, parent);
251
}
252
253
4970874
void SharedTermsVisitor::visit(TNode current, TNode parent) {
254
255
4970874
  Debug("register") << "SharedTermsVisitor::visit(" << current << "," << parent << ")" << std::endl;
256
4970874
  if (Debug.isOn("register::internal")) {
257
    Debug("register::internal") << toString() << std::endl;
258
  }
259
4970874
  TheoryIdSet visitedTheories = d_visited[current];
260
4970874
  TheoryIdSet preregTheories = d_preregistered[current];
261
262
  // preregister the term with the current, parent or type theories, as needed
263
4970877
  PreRegisterVisitor::preRegister(
264
      d_engine, visitedTheories, current, parent, preregTheories);
265
266
  // Record the new theories that we visited
267
4970871
  d_visited[current] = visitedTheories;
268
269
  // add visited theories to those who have preregistered
270
4970871
  d_preregistered[current] =
271
9941742
      TheoryIdSetUtil::setUnion(preregTheories, visitedTheories);
272
273
  // If there is more than two theories and a new one has been added notify the shared terms database
274
4970871
  TheoryId currentTheoryId = Theory::theoryOf(current);
275
9941742
  if (TheoryIdSetUtil::setDifference(
276
4970871
          visitedTheories, TheoryIdSetUtil::setInsert(currentTheoryId)))
277
  {
278
1138130
    d_sharedTerms.addSharedTerm(d_atom, current, visitedTheories);
279
  }
280
281
4970871
  Assert(d_visited.find(current) != d_visited.end());
282
4970871
  Assert(alreadyVisited(current, parent));
283
4970871
}
284
285
841281
void SharedTermsVisitor::start(TNode node) {
286
841281
  d_visited.clear();
287
841281
  d_atom = node;
288
841281
}
289
290
841278
void SharedTermsVisitor::done(TNode node) {
291
841278
  clear();
292
841278
}
293
294
841278
void SharedTermsVisitor::clear() {
295
841278
  d_atom = TNode();
296
841278
  d_visited.clear();
297
841278
}
298
299
29340
}  // namespace cvc5