1 /********************* */
4 ** Top contributors (to current version):
5 ** Andrew Reynolds, Tim King, Mathias Preiner
6 ** This file is part of the CVC4 project.
7 ** Copyright (c) 2009-2020 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.\endverbatim
12 ** \brief Base for theory interface.
14 ** Base for theory interface.
17 #include "theory/theory.h"
24 #include "base/check.h"
25 #include "expr/node_algorithm.h"
26 #include "options/smt_options.h"
27 #include "options/theory_options.h"
28 #include "smt/smt_statistics_registry.h"
29 #include "theory/ext_theory.h"
30 #include "theory/quantifiers_engine.h"
31 #include "theory/substitutions.h"
32 #include "theory/theory_rewriter.h"
39 /** Default value for the uninterpreted sorts is the UF theory */
40 TheoryId
Theory::s_uninterpretedSortOwner
= THEORY_UF
;
42 std::ostream
& operator<<(std::ostream
& os
, Theory::Effort level
){
44 case Theory::EFFORT_STANDARD
:
45 os
<< "EFFORT_STANDARD"; break;
46 case Theory::EFFORT_FULL
:
47 os
<< "EFFORT_FULL"; break;
48 case Theory::EFFORT_LAST_CALL
:
49 os
<< "EFFORT_LAST_CALL"; break;
54 }/* ostream& operator<<(ostream&, Theory::Effort) */
56 Theory::Theory(TheoryId id
,
57 context::Context
* satContext
,
58 context::UserContext
* userContext
,
61 const LogicInfo
& logicInfo
,
62 ProofNodeManager
* pnm
,
65 d_satContext(satContext
),
66 d_userContext(userContext
),
67 d_logicInfo(logicInfo
),
69 d_factsHead(satContext
, 0),
70 d_sharedTermsIndex(satContext
, 0),
73 d_decManager(nullptr),
75 d_checkTime(getStatsPrefix(id
) + name
+ "::checkTime"),
76 d_computeCareGraphTime(getStatsPrefix(id
) + name
77 + "::computeCareGraphTime"),
78 d_sharedTerms(satContext
),
80 d_valuation(valuation
),
81 d_equalityEngine(nullptr),
82 d_allocEqualityEngine(nullptr),
83 d_theoryState(nullptr),
84 d_inferManager(nullptr),
87 smtStatisticsRegistry()->registerStat(&d_checkTime
);
88 smtStatisticsRegistry()->registerStat(&d_computeCareGraphTime
);
92 smtStatisticsRegistry()->unregisterStat(&d_checkTime
);
93 smtStatisticsRegistry()->unregisterStat(&d_computeCareGraphTime
);
96 bool Theory::needsEqualityEngine(EeSetupInfo
& esi
)
98 // by default, this theory does not use an (official) equality engine
102 void Theory::setEqualityEngine(eq::EqualityEngine
* ee
)
104 // set the equality engine pointer
105 d_equalityEngine
= ee
;
106 if (d_theoryState
!= nullptr)
108 d_theoryState
->setEqualityEngine(ee
);
110 if (d_inferManager
!= nullptr)
112 d_inferManager
->setEqualityEngine(ee
);
116 void Theory::setQuantifiersEngine(QuantifiersEngine
* qe
)
118 Assert(d_quantEngine
== nullptr);
119 // quantifiers engine may be null if not in quantified logic
123 void Theory::setDecisionManager(DecisionManager
* dm
)
125 Assert(d_decManager
== nullptr);
126 Assert(dm
!= nullptr);
130 void Theory::finishInitStandalone()
133 if (needsEqualityEngine(esi
))
135 // always associated with the same SAT context as the theory (d_satContext)
136 d_allocEqualityEngine
.reset(new eq::EqualityEngine(
137 *esi
.d_notify
, d_satContext
, esi
.d_name
, esi
.d_constantsAreTriggers
));
138 // use it as the official equality engine
139 setEqualityEngine(d_allocEqualityEngine
.get());
144 TheoryId
Theory::theoryOf(options::TheoryOfMode mode
, TNode node
)
146 TheoryId tid
= THEORY_BUILTIN
;
148 case options::TheoryOfMode::THEORY_OF_TYPE_BASED
:
149 // Constants, variables, 0-ary constructors
152 if (node
.getKind() == kind::BOOLEAN_TERM_VARIABLE
)
158 tid
= Theory::theoryOf(node
.getType());
161 else if (node
.isConst())
163 tid
= Theory::theoryOf(node
.getType());
165 else if (node
.getKind() == kind::EQUAL
)
167 // Equality is owned by the theory that owns the domain
168 tid
= Theory::theoryOf(node
[0].getType());
172 // Regular nodes are owned by the kind
173 tid
= kindToTheoryId(node
.getKind());
176 case options::TheoryOfMode::THEORY_OF_TERM_BASED
:
180 if (Theory::theoryOf(node
.getType()) != theory::THEORY_BOOL
)
182 // We treat the variables as uninterpreted
183 tid
= s_uninterpretedSortOwner
;
187 if (node
.getKind() == kind::BOOLEAN_TERM_VARIABLE
)
189 // Boolean vars go to UF
194 // Except for the Boolean ones
199 else if (node
.isConst())
201 // Constants go to the theory of the type
202 tid
= Theory::theoryOf(node
.getType());
204 else if (node
.getKind() == kind::EQUAL
)
206 // If one of them is an ITE, it's irelevant, since they will get
207 // replaced out anyhow
208 if (node
[0].getKind() == kind::ITE
)
210 tid
= Theory::theoryOf(node
[0].getType());
212 else if (node
[1].getKind() == kind::ITE
)
214 tid
= Theory::theoryOf(node
[1].getType());
220 TypeNode ltype
= l
.getType();
221 TypeNode rtype
= r
.getType();
224 tid
= Theory::theoryOf(l
.getType());
228 // If both sides belong to the same theory the choice is easy
229 TheoryId T1
= Theory::theoryOf(l
);
230 TheoryId T2
= Theory::theoryOf(r
);
237 TheoryId T3
= Theory::theoryOf(ltype
);
239 // * x*y = f(z) -> UF
241 // * f(x) = read(a, y) -> either UF or ARRAY
242 // at least one of the theories has to be parametric, i.e. theory
243 // of the type is different from the theory of the term
254 // If both are parametric, we take the smaller one (arbitrary)
255 tid
= T1
< T2
? T1
: T2
;
263 // Regular nodes are owned by the kind
264 tid
= kindToTheoryId(node
.getKind());
270 Trace("theory::internal") << "theoryOf(" << mode
<< ", " << node
<< ") -> " << tid
<< std::endl
;
274 void Theory::notifySharedTerm(TNode n
)
279 void Theory::computeCareGraph() {
280 Debug("sharing") << "Theory::computeCareGraph<" << getId() << ">()" << endl
;
281 for (unsigned i
= 0; i
< d_sharedTerms
.size(); ++ i
) {
282 TNode a
= d_sharedTerms
[i
];
283 TypeNode aType
= a
.getType();
284 for (unsigned j
= i
+ 1; j
< d_sharedTerms
.size(); ++ j
) {
285 TNode b
= d_sharedTerms
[j
];
286 if (b
.getType() != aType
) {
287 // We don't care about the terms of different types
290 switch (d_valuation
.getEqualityStatus(a
, b
)) {
291 case EQUALITY_TRUE_AND_PROPAGATED
:
292 case EQUALITY_FALSE_AND_PROPAGATED
:
293 // If we know about it, we should have propagated it, so we can skip
304 void Theory::printFacts(std::ostream
& os
) const {
305 unsigned i
, n
= d_facts
.size();
306 for(i
= 0; i
< n
; i
++){
307 const Assertion
& a_i
= d_facts
[i
];
308 Node assertion
= a_i
;
309 os
<< d_id
<< '[' << i
<< ']' << " " << assertion
<< endl
;
313 void Theory::debugPrintFacts() const{
314 DebugChannel
.getStream() << "Theory::debugPrintFacts()" << endl
;
315 printFacts(DebugChannel
.getStream());
318 bool Theory::isLegalElimination(TNode x
, TNode val
)
321 if (x
.getKind() == kind::BOOLEAN_TERM_VARIABLE
322 || val
.getKind() == kind::BOOLEAN_TERM_VARIABLE
)
326 if (expr::hasSubterm(val
, x
))
330 if (!val
.getType().isSubtypeOf(x
.getType()))
334 if (!options::produceModels())
336 // don't care about the model, we are fine
339 // if there is a model object
340 TheoryModel
* tm
= d_valuation
.getModel();
341 Assert(tm
!= nullptr);
342 return tm
->isLegalElimination(x
, val
);
345 std::unordered_set
<TNode
, TNodeHashFunction
> Theory::currentlySharedTerms() const{
346 std::unordered_set
<TNode
, TNodeHashFunction
> currentlyShared
;
347 for (shared_terms_iterator i
= shared_terms_begin(),
348 i_end
= shared_terms_end(); i
!= i_end
; ++i
) {
349 currentlyShared
.insert (*i
);
351 return currentlyShared
;
354 bool Theory::collectModelInfo(TheoryModel
* m
, const std::set
<Node
>& termSet
)
356 // if we are using an equality engine, assert it to the model
357 if (d_equalityEngine
!= nullptr)
359 if (!m
->assertEqualityEngine(d_equalityEngine
, &termSet
))
364 // now, collect theory-specific value assigments
365 return collectModelValues(m
, termSet
);
368 void Theory::computeRelevantTerms(std::set
<Node
>& termSet
)
370 // by default, there are no additional relevant terms
373 bool Theory::collectModelValues(TheoryModel
* m
, const std::set
<Node
>& termSet
)
378 Theory::PPAssertStatus
Theory::ppAssert(TrustNode tin
,
379 TrustSubstitutionMap
& outSubstitutions
)
381 TNode in
= tin
.getNode();
382 if (in
.getKind() == kind::EQUAL
)
384 // (and (= x t) phi) can be replaced by phi[x/t] if
385 // 1) x is a variable
386 // 2) x is not in the term t
387 // 3) x : T and t : S, then S <: T
388 if (in
[0].isVar() && isLegalElimination(in
[0], in
[1])
389 && in
[0].getKind() != kind::BOOLEAN_TERM_VARIABLE
)
391 outSubstitutions
.addSubstitutionSolved(in
[0], in
[1], tin
);
392 return PP_ASSERT_STATUS_SOLVED
;
394 if (in
[1].isVar() && isLegalElimination(in
[1], in
[0])
395 && in
[1].getKind() != kind::BOOLEAN_TERM_VARIABLE
)
397 outSubstitutions
.addSubstitutionSolved(in
[1], in
[0], tin
);
398 return PP_ASSERT_STATUS_SOLVED
;
400 if (in
[0].isConst() && in
[1].isConst())
404 return PP_ASSERT_STATUS_CONFLICT
;
409 return PP_ASSERT_STATUS_UNSOLVED
;
412 std::pair
<bool, Node
> Theory::entailmentCheck(TNode lit
)
414 return make_pair(false, Node::null());
417 void Theory::addCarePair(TNode t1
, TNode t2
) {
419 d_careGraph
->insert(CarePair(t1
, t2
, d_id
));
423 void Theory::getCareGraph(CareGraph
* careGraph
) {
424 Assert(careGraph
!= NULL
);
426 Trace("sharing") << "Theory<" << getId() << ">::getCareGraph()" << std::endl
;
427 TimerStat::CodeTimer
computeCareGraphTime(d_computeCareGraphTime
);
428 d_careGraph
= careGraph
;
433 EqualityStatus
Theory::getEqualityStatus(TNode a
, TNode b
)
435 // if not using an equality engine, then by default we don't know the status
436 if (d_equalityEngine
== nullptr)
438 return EQUALITY_UNKNOWN
;
440 Trace("sharing") << "Theory<" << getId() << ">::getEqualityStatus(" << a
<< ", " << b
<< ")" << std::endl
;
441 Assert(d_equalityEngine
->hasTerm(a
) && d_equalityEngine
->hasTerm(b
));
443 // Check for equality (simplest)
444 if (d_equalityEngine
->areEqual(a
, b
))
446 // The terms are implied to be equal
447 return EQUALITY_TRUE
;
450 // Check for disequality
451 if (d_equalityEngine
->areDisequal(a
, b
, false))
453 // The terms are implied to be dis-equal
454 return EQUALITY_FALSE
;
457 // All other terms are unknown, which is conservative. A theory may override
458 // this method if it knows more information.
459 return EQUALITY_UNKNOWN
;
462 void Theory::check(Effort level
)
464 // see if we are already done (as an optimization)
465 if (done() && level
< EFFORT_FULL
)
469 Assert(d_theoryState
!=nullptr);
470 // standard calls for resource, stats
471 d_out
->spendResource(ResourceManager::Resource::TheoryCheckStep
);
472 TimerStat::CodeTimer
checkTimer(d_checkTime
);
473 Trace("theory-check") << "Theory::preCheck " << level
<< " " << d_id
475 // pre-check at level
478 // check aborted for a theory-specific reason
481 Assert(d_theoryState
!= nullptr);
482 Trace("theory-check") << "Theory::process fact queue " << d_id
<< std::endl
;
483 // process the pending fact queue
484 while (!done() && !d_theoryState
->isInConflict())
486 // Get the next assertion from the fact queue
487 Assertion assertion
= get();
488 TNode fact
= assertion
.d_assertion
;
489 Trace("theory-check") << "Theory::preNotifyFact " << fact
<< " " << d_id
491 bool polarity
= fact
.getKind() != kind::NOT
;
492 TNode atom
= polarity
? fact
: fact
[0];
493 // call the pre-notify method
494 if (preNotifyFact(atom
, polarity
, fact
, assertion
.d_isPreregistered
, false))
496 // handled in theory-specific way that doesn't involve equality engine
499 Trace("theory-check") << "Theory::assert " << fact
<< " " << d_id
501 // Theories that don't have an equality engine should always return true
503 Assert(d_equalityEngine
!= nullptr);
504 // assert to the equality engine
505 if (atom
.getKind() == kind::EQUAL
)
507 d_equalityEngine
->assertEquality(atom
, polarity
, fact
);
511 d_equalityEngine
->assertPredicate(atom
, polarity
, fact
);
513 Trace("theory-check") << "Theory::notifyFact " << fact
<< " " << d_id
515 // notify the theory of the new fact, which is not internal
516 notifyFact(atom
, polarity
, fact
, false);
518 Trace("theory-check") << "Theory::postCheck " << d_id
<< std::endl
;
519 // post-check at level
521 Trace("theory-check") << "Theory::finish check " << d_id
<< std::endl
;
524 bool Theory::preCheck(Effort level
) { return false; }
526 void Theory::postCheck(Effort level
) {}
528 bool Theory::preNotifyFact(
529 TNode atom
, bool polarity
, TNode fact
, bool isPrereg
, bool isInternal
)
534 void Theory::notifyFact(TNode atom
, bool polarity
, TNode fact
, bool isInternal
)
538 void Theory::preRegisterTerm(TNode node
) {}
540 void Theory::addSharedTerm(TNode n
)
542 Debug("sharing") << "Theory::addSharedTerm<" << getId() << ">(" << n
<< ")"
544 Debug("theory::assertions")
545 << "Theory::addSharedTerm<" << getId() << ">(" << n
<< ")" << std::endl
;
546 d_sharedTerms
.push_back(n
);
547 // now call theory-specific method notifySharedTerm
549 // if we have an equality engine, add the trigger term
550 if (d_equalityEngine
!= nullptr)
552 d_equalityEngine
->addTriggerTerm(n
, d_id
);
556 eq::EqualityEngine
* Theory::getEqualityEngine()
558 // get the assigned equality engine, which is a pointer stored in this class
559 return d_equalityEngine
;
562 }/* CVC4::theory namespace */
563 }/* CVC4 namespace */