GCC Code Coverage Report
Directory: . Exec Total Coverage
File: src/prop/cnf_stream.cpp Lines: 386 412 93.7 %
Date: 2021-09-10 Branches: 756 1991 38.0 %

Line Exec Source
1
/******************************************************************************
2
 * Top contributors (to current version):
3
 *   Dejan Jovanovic, Haniel Barbosa, Liana Hadarean
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
 * A CNF converter that takes in asserts and has the side effect of given an
14
 * equisatisfiable stream of assertions to PropEngine.
15
 */
16
#include "prop/cnf_stream.h"
17
18
#include <queue>
19
20
#include "base/check.h"
21
#include "base/output.h"
22
#include "expr/node.h"
23
#include "options/bv_options.h"
24
#include "printer/printer.h"
25
#include "proof/clause_id.h"
26
#include "prop/minisat/minisat.h"
27
#include "prop/prop_engine.h"
28
#include "prop/theory_proxy.h"
29
#include "smt/dump.h"
30
#include "smt/env.h"
31
#include "smt/smt_engine_scope.h"
32
#include "smt/smt_statistics_registry.h"
33
#include "theory/theory.h"
34
#include "theory/theory_engine.h"
35
36
namespace cvc5 {
37
namespace prop {
38
39
16153
CnfStream::CnfStream(SatSolver* satSolver,
40
                     Registrar* registrar,
41
                     context::Context* context,
42
                     Env* env,
43
                     ResourceManager* rm,
44
                     FormulaLitPolicy flpol,
45
16153
                     std::string name)
46
    : d_satSolver(satSolver),
47
      d_env(env),
48
      d_booleanVariables(context),
49
      d_notifyFormulas(context),
50
      d_nodeToLiteralMap(context),
51
      d_literalToNodeMap(context),
52
      d_flitPolicy(flpol),
53
      d_registrar(registrar),
54
      d_name(name),
55
      d_removable(false),
56
      d_resourceManager(rm),
57
16153
      d_stats(name)
58
{
59
16153
}
60
61
8902644
bool CnfStream::assertClause(TNode node, SatClause& c)
62
{
63
8902644
  Trace("cnf") << "Inserting into stream " << c << " node = " << node << "\n";
64
8902644
  if (Dump.isOn("clauses") && d_env != nullptr)
65
  {
66
    const Printer& printer = d_env->getPrinter();
67
    std::ostream& out = d_env->getDumpOut();
68
    if (c.size() == 1)
69
    {
70
      printer.toStreamCmdAssert(out, getNode(c[0]));
71
    }
72
    else
73
    {
74
      Assert(c.size() > 1);
75
      NodeBuilder b(kind::OR);
76
      for (unsigned i = 0; i < c.size(); ++i)
77
      {
78
        b << getNode(c[i]);
79
      }
80
      Node n = b;
81
      printer.toStreamCmdAssert(out, n);
82
    }
83
  }
84
85
8902644
  ClauseId clauseId = d_satSolver->addClause(c, d_removable);
86
87
8902644
  return clauseId != ClauseIdUndef;
88
}
89
90
211155
bool CnfStream::assertClause(TNode node, SatLiteral a)
91
{
92
422310
  SatClause clause(1);
93
211155
  clause[0] = a;
94
422310
  return assertClause(node, clause);
95
}
96
97
4046232
bool CnfStream::assertClause(TNode node, SatLiteral a, SatLiteral b)
98
{
99
8092464
  SatClause clause(2);
100
4046232
  clause[0] = a;
101
4046232
  clause[1] = b;
102
8092464
  return assertClause(node, clause);
103
}
104
105
2759746
bool CnfStream::assertClause(TNode node,
106
                             SatLiteral a,
107
                             SatLiteral b,
108
                             SatLiteral c)
109
{
110
5519492
  SatClause clause(3);
111
2759746
  clause[0] = a;
112
2759746
  clause[1] = b;
113
2759746
  clause[2] = c;
114
5519492
  return assertClause(node, clause);
115
}
116
117
61036830
bool CnfStream::hasLiteral(TNode n) const {
118
61036830
  NodeToLiteralMap::const_iterator find = d_nodeToLiteralMap.find(n);
119
61036830
  return find != d_nodeToLiteralMap.end();
120
}
121
122
2690949
void CnfStream::ensureMappingForLiteral(TNode n)
123
{
124
2690949
  SatLiteral lit = getLiteral(n);
125
2690949
  if (!d_literalToNodeMap.contains(lit))
126
  {
127
    // Store backward-mappings
128
197
    d_literalToNodeMap.insert(lit, n);
129
197
    d_literalToNodeMap.insert(~lit, n.notNode());
130
  }
131
2690949
}
132
133
2728182
void CnfStream::ensureLiteral(TNode n)
134
{
135
2728182
  AlwaysAssertArgument(
136
      hasLiteral(n) || n.getType().isBoolean(),
137
      n,
138
      "ProofCnfStream::ensureLiteral() requires a node of Boolean type.\n"
139
      "got node: %s\n"
140
      "its type: %s\n",
141
      n.toString().c_str(),
142
      n.getType().toString().c_str());
143
2728182
  Trace("cnf") << "ensureLiteral(" << n << ")\n";
144
2791061
  TimerStat::CodeTimer codeTimer(d_stats.d_cnfConversionTime, true);
145
2728182
  if (hasLiteral(n))
146
  {
147
2665303
    ensureMappingForLiteral(n);
148
2665303
    return;
149
  }
150
  // remove top level negation
151
62879
  n = n.getKind() == kind::NOT ? n[0] : n;
152
62879
  if (theory::Theory::theoryOf(n) == theory::THEORY_BOOL && !n.isVar())
153
  {
154
    // If we were called with something other than a theory atom (or
155
    // Boolean variable), we get a SatLiteral that is definitionally
156
    // equal to it.
157
    // These are not removable and have no proof ID
158
43849
    d_removable = false;
159
160
43849
    SatLiteral lit = toCNF(n, false);
161
162
    // Store backward-mappings
163
    // These may already exist
164
43849
    d_literalToNodeMap.insert_safe(lit, n);
165
43849
    d_literalToNodeMap.insert_safe(~lit, n.notNode());
166
  }
167
  else
168
  {
169
    // We have a theory atom or variable.
170
19030
    convertAtom(n);
171
  }
172
}
173
174
2614191
SatLiteral CnfStream::newLiteral(TNode node, bool isTheoryAtom, bool preRegister, bool canEliminate) {
175
5228382
  Trace("cnf") << d_name << "::newLiteral(" << node << ", " << isTheoryAtom
176
2614191
               << ")\n"
177
2614191
               << push;
178
2614191
  Assert(node.getKind() != kind::NOT);
179
180
  // if we are tracking formulas, everything is a theory atom
181
2614191
  if (!isTheoryAtom && d_flitPolicy == FormulaLitPolicy::TRACK_AND_NOTIFY)
182
  {
183
    isTheoryAtom = true;
184
    d_notifyFormulas.insert(node);
185
  }
186
187
  // Get the literal for this node
188
2614191
  SatLiteral lit;
189
2614191
  if (!hasLiteral(node)) {
190
2614189
    Trace("cnf") << d_name << "::newLiteral: node already registered\n";
191
    // If no literal, we'll make one
192
2614189
    if (node.getKind() == kind::CONST_BOOLEAN) {
193
20162
      Trace("cnf") << d_name << "::newLiteral: boolean const\n";
194
20162
      if (node.getConst<bool>()) {
195
10025
        lit = SatLiteral(d_satSolver->trueVar());
196
      } else {
197
10137
        lit = SatLiteral(d_satSolver->falseVar());
198
      }
199
    } else {
200
2594027
      Trace("cnf") << d_name << "::newLiteral: new var\n";
201
2594027
      lit = SatLiteral(d_satSolver->newVar(isTheoryAtom, preRegister, canEliminate));
202
    }
203
2614189
    d_nodeToLiteralMap.insert(node, lit);
204
2614189
    d_nodeToLiteralMap.insert(node.notNode(), ~lit);
205
  } else {
206
2
    lit = getLiteral(node);
207
  }
208
209
  // If it's a theory literal, need to store it for back queries
210
4457346
  if (isTheoryAtom || d_flitPolicy == FormulaLitPolicy::TRACK
211
3788052
      || (Dump.isOn("clauses")))
212
  {
213
1440330
    d_literalToNodeMap.insert_safe(lit, node);
214
1440330
    d_literalToNodeMap.insert_safe(~lit, node.notNode());
215
  }
216
217
  // If a theory literal, we pre-register it
218
2614191
  if (preRegister) {
219
    // In case we are re-entered due to lemmas, save our state
220
771036
    bool backupRemovable = d_removable;
221
771040
    d_registrar->preRegister(node);
222
771032
    d_removable = backupRemovable;
223
  }
224
  // Here, you can have it
225
2614187
  Trace("cnf") << "newLiteral(" << node << ") => " << lit << "\n" << pop;
226
2614187
  return lit;
227
}
228
229
17951113
TNode CnfStream::getNode(const SatLiteral& literal) {
230
17951113
  Trace("cnf") << "getNode(" << literal << ")\n";
231
35902226
  Trace("cnf") << "getNode(" << literal << ") => "
232
17951113
               << d_literalToNodeMap[literal] << "\n";
233
17951113
  return d_literalToNodeMap[literal];
234
}
235
236
2903719
const CnfStream::NodeToLiteralMap& CnfStream::getTranslationCache() const
237
{
238
2903719
  return d_nodeToLiteralMap;
239
}
240
241
8111888
const CnfStream::LiteralToNodeMap& CnfStream::getNodeCache() const
242
{
243
8111888
  return d_literalToNodeMap;
244
}
245
246
16406
void CnfStream::getBooleanVariables(std::vector<TNode>& outputVariables) const {
247
16406
  context::CDList<TNode>::const_iterator it, it_end;
248
86222
  for (it = d_booleanVariables.begin(); it != d_booleanVariables.end(); ++ it) {
249
69816
    outputVariables.push_back(*it);
250
  }
251
16406
}
252
253
bool CnfStream::isNotifyFormula(TNode node) const
254
{
255
  return d_notifyFormulas.find(node) != d_notifyFormulas.end();
256
}
257
258
850094
SatLiteral CnfStream::convertAtom(TNode node)
259
{
260
850094
  Trace("cnf") << "convertAtom(" << node << ")\n";
261
262
850094
  Assert(!hasLiteral(node)) << "atom already mapped!";
263
264
850094
  bool theoryLiteral = false;
265
850094
  bool canEliminate = true;
266
850094
  bool preRegister = false;
267
268
  // Is this a variable add it to the list
269
850094
  if (node.isVar() && node.getKind() != kind::BOOLEAN_TERM_VARIABLE)
270
  {
271
79058
    d_booleanVariables.push_back(node);
272
  }
273
  else
274
  {
275
771036
    theoryLiteral = true;
276
771036
    canEliminate = false;
277
771036
    preRegister = true;
278
  }
279
280
  // Make a new literal (variables are not considered theory literals)
281
850098
  SatLiteral lit = newLiteral(node, theoryLiteral, preRegister, canEliminate);
282
  // Return the resulting literal
283
850090
  return lit;
284
}
285
286
45125707
SatLiteral CnfStream::getLiteral(TNode node) {
287
45125707
  Assert(!node.isNull()) << "CnfStream: can't getLiteral() of null node";
288
289
90251414
  Assert(d_nodeToLiteralMap.contains(node))
290
45125707
      << "Literal not in the CNF Cache: " << node << "\n";
291
292
45125707
  SatLiteral literal = d_nodeToLiteralMap[node];
293
90251414
  Trace("cnf") << "CnfStream::getLiteral(" << node << ") => " << literal
294
45125707
               << "\n";
295
45125707
  return literal;
296
}
297
298
249496
void CnfStream::handleXor(TNode xorNode)
299
{
300
249496
  Assert(!hasLiteral(xorNode)) << "Atom already mapped!";
301
249496
  Assert(xorNode.getKind() == kind::XOR) << "Expecting an XOR expression!";
302
249496
  Assert(xorNode.getNumChildren() == 2) << "Expecting exactly 2 children!";
303
249496
  Assert(!d_removable) << "Removable clauses can not contain Boolean structure";
304
249496
  Trace("cnf") << "CnfStream::handleXor(" << xorNode << ")\n";
305
306
249496
  SatLiteral a = getLiteral(xorNode[0]);
307
249496
  SatLiteral b = getLiteral(xorNode[1]);
308
309
249496
  SatLiteral xorLit = newLiteral(xorNode);
310
311
249496
  assertClause(xorNode.negate(), a, b, ~xorLit);
312
249496
  assertClause(xorNode.negate(), ~a, ~b, ~xorLit);
313
249496
  assertClause(xorNode, a, ~b, xorLit);
314
249496
  assertClause(xorNode, ~a, b, xorLit);
315
249496
}
316
317
390963
void CnfStream::handleOr(TNode orNode)
318
{
319
390963
  Assert(!hasLiteral(orNode)) << "Atom already mapped!";
320
390963
  Assert(orNode.getKind() == kind::OR) << "Expecting an OR expression!";
321
390963
  Assert(orNode.getNumChildren() > 1) << "Expecting more then 1 child!";
322
390963
  Assert(!d_removable) << "Removable clauses can not contain Boolean structure";
323
390963
  Trace("cnf") << "CnfStream::handleOr(" << orNode << ")\n";
324
325
  // Number of children
326
390963
  size_t numChildren = orNode.getNumChildren();
327
328
  // Get the literal for this node
329
390963
  SatLiteral orLit = newLiteral(orNode);
330
331
  // Transform all the children first
332
781926
  SatClause clause(numChildren + 1);
333
1347892
  for (size_t i = 0; i < numChildren; ++i)
334
  {
335
956929
    clause[i] = getLiteral(orNode[i]);
336
337
    // lit <- (a_1 | a_2 | a_3 | ... | a_n)
338
    // lit | ~(a_1 | a_2 | a_3 | ... | a_n)
339
    // (lit | ~a_1) & (lit | ~a_2) & (lit & ~a_3) & ... & (lit & ~a_n)
340
956929
    assertClause(orNode, orLit, ~clause[i]);
341
  }
342
343
  // lit -> (a_1 | a_2 | a_3 | ... | a_n)
344
  // ~lit | a_1 | a_2 | a_3 | ... | a_n
345
390963
  clause[numChildren] = ~orLit;
346
  // This needs to go last, as the clause might get modified by the SAT solver
347
390963
  assertClause(orNode.negate(), clause);
348
390963
}
349
350
644804
void CnfStream::handleAnd(TNode andNode)
351
{
352
644804
  Assert(!hasLiteral(andNode)) << "Atom already mapped!";
353
644804
  Assert(andNode.getKind() == kind::AND) << "Expecting an AND expression!";
354
644804
  Assert(andNode.getNumChildren() > 1) << "Expecting more than 1 child!";
355
644804
  Assert(!d_removable) << "Removable clauses can not contain Boolean structure";
356
644804
  Trace("cnf") << "handleAnd(" << andNode << ")\n";
357
358
  // Number of children
359
644804
  size_t numChildren = andNode.getNumChildren();
360
361
  // Get the literal for this node
362
644804
  SatLiteral andLit = newLiteral(andNode);
363
364
  // Transform all the children first (remembering the negation)
365
1289608
  SatClause clause(numChildren + 1);
366
3446946
  for (size_t i = 0; i < numChildren; ++i)
367
  {
368
2802142
    clause[i] = ~getLiteral(andNode[i]);
369
370
    // lit -> (a_1 & a_2 & a_3 & ... & a_n)
371
    // ~lit | (a_1 & a_2 & a_3 & ... & a_n)
372
    // (~lit | a_1) & (~lit | a_2) & ... & (~lit | a_n)
373
2802142
    assertClause(andNode.negate(), ~andLit, ~clause[i]);
374
  }
375
376
  // lit <- (a_1 & a_2 & a_3 & ... a_n)
377
  // lit | ~(a_1 & a_2 & a_3 & ... & a_n)
378
  // lit | ~a_1 | ~a_2 | ~a_3 | ... | ~a_n
379
644804
  clause[numChildren] = andLit;
380
  // This needs to go last, as the clause might get modified by the SAT solver
381
644804
  assertClause(andNode, clause);
382
644804
}
383
384
8500
void CnfStream::handleImplies(TNode impliesNode)
385
{
386
8500
  Assert(!hasLiteral(impliesNode)) << "Atom already mapped!";
387
8500
  Assert(impliesNode.getKind() == kind::IMPLIES)
388
      << "Expecting an IMPLIES expression!";
389
8500
  Assert(impliesNode.getNumChildren() == 2) << "Expecting exactly 2 children!";
390
8500
  Assert(!d_removable) << "Removable clauses can not contain Boolean structure";
391
8500
  Trace("cnf") << "handleImplies(" << impliesNode << ")\n";
392
393
  // Convert the children to cnf
394
8500
  SatLiteral a = getLiteral(impliesNode[0]);
395
8500
  SatLiteral b = getLiteral(impliesNode[1]);
396
397
8500
  SatLiteral impliesLit = newLiteral(impliesNode);
398
399
  // lit -> (a->b)
400
  // ~lit | ~ a | b
401
8500
  assertClause(impliesNode.negate(), ~impliesLit, ~a, b);
402
403
  // (a->b) -> lit
404
  // ~(~a | b) | lit
405
  // (a | l) & (~b | l)
406
8500
  assertClause(impliesNode, a, impliesLit);
407
8500
  assertClause(impliesNode, ~b, impliesLit);
408
8500
}
409
410
269591
void CnfStream::handleIff(TNode iffNode)
411
{
412
269591
  Assert(!hasLiteral(iffNode)) << "Atom already mapped!";
413
269591
  Assert(iffNode.getKind() == kind::EQUAL) << "Expecting an EQUAL expression!";
414
269591
  Assert(iffNode.getNumChildren() == 2) << "Expecting exactly 2 children!";
415
269591
  Assert(!d_removable) << "Removable clauses can not contain Boolean structure";
416
269591
  Trace("cnf") << "handleIff(" << iffNode << ")\n";
417
418
  // Convert the children to CNF
419
269591
  SatLiteral a = getLiteral(iffNode[0]);
420
269591
  SatLiteral b = getLiteral(iffNode[1]);
421
422
  // Get the now literal
423
269591
  SatLiteral iffLit = newLiteral(iffNode);
424
425
  // lit -> ((a-> b) & (b->a))
426
  // ~lit | ((~a | b) & (~b | a))
427
  // (~a | b | ~lit) & (~b | a | ~lit)
428
269591
  assertClause(iffNode.negate(), ~a, b, ~iffLit);
429
269591
  assertClause(iffNode.negate(), a, ~b, ~iffLit);
430
431
  // (a<->b) -> lit
432
  // ~((a & b) | (~a & ~b)) | lit
433
  // (~(a & b)) & (~(~a & ~b)) | lit
434
  // ((~a | ~b) & (a | b)) | lit
435
  // (~a | ~b | lit) & (a | b | lit)
436
269591
  assertClause(iffNode, ~a, ~b, iffLit);
437
269591
  assertClause(iffNode, a, b, iffLit);
438
269591
}
439
440
83610
void CnfStream::handleIte(TNode iteNode)
441
{
442
83610
  Assert(!hasLiteral(iteNode)) << "Atom already mapped!";
443
83610
  Assert(iteNode.getKind() == kind::ITE);
444
83610
  Assert(iteNode.getNumChildren() == 3);
445
83610
  Assert(!d_removable) << "Removable clauses can not contain Boolean structure";
446
167220
  Trace("cnf") << "handleIte(" << iteNode[0] << " " << iteNode[1] << " "
447
83610
               << iteNode[2] << ")\n";
448
449
83610
  SatLiteral condLit = getLiteral(iteNode[0]);
450
83610
  SatLiteral thenLit = getLiteral(iteNode[1]);
451
83610
  SatLiteral elseLit = getLiteral(iteNode[2]);
452
453
83610
  SatLiteral iteLit = newLiteral(iteNode);
454
455
  // If ITE is true then one of the branches is true and the condition
456
  // implies which one
457
  // lit -> (ite b t e)
458
  // lit -> (t | e) & (b -> t) & (!b -> e)
459
  // lit -> (t | e) & (!b | t) & (b | e)
460
  // (!lit | t | e) & (!lit | !b | t) & (!lit | b | e)
461
83610
  assertClause(iteNode.negate(), ~iteLit, thenLit, elseLit);
462
83610
  assertClause(iteNode.negate(), ~iteLit, ~condLit, thenLit);
463
83610
  assertClause(iteNode.negate(), ~iteLit, condLit, elseLit);
464
465
  // If ITE is false then one of the branches is false and the condition
466
  // implies which one
467
  // !lit -> !(ite b t e)
468
  // !lit -> (!t | !e) & (b -> !t) & (!b -> !e)
469
  // !lit -> (!t | !e) & (!b | !t) & (b | !e)
470
  // (lit | !t | !e) & (lit | !b | !t) & (lit | b | !e)
471
83610
  assertClause(iteNode, iteLit, ~thenLit, ~elseLit);
472
83610
  assertClause(iteNode, iteLit, ~condLit, ~thenLit);
473
83610
  assertClause(iteNode, iteLit, condLit, ~elseLit);
474
83610
}
475
476
3126832
SatLiteral CnfStream::toCNF(TNode node, bool negated)
477
{
478
6253664
  Trace("cnf") << "toCNF(" << node
479
3126832
               << ", negated = " << (negated ? "true" : "false") << ")\n";
480
481
6253664
  TNode cur;
482
3126832
  SatLiteral nodeLit;
483
6253664
  std::vector<TNode> visit;
484
6253664
  std::unordered_map<TNode, bool> cache;
485
486
3126832
  visit.push_back(node);
487
25078750
  while (!visit.empty())
488
  {
489
10975963
    cur = visit.back();
490
10975963
    Assert(cur.getType().isBoolean());
491
492
17011143
    if (hasLiteral(cur))
493
    {
494
6035180
      visit.pop_back();
495
6035180
      continue;
496
    }
497
498
4940783
    const auto& it = cache.find(cur);
499
4940783
    if (it == cache.end())
500
    {
501
2574944
      cache.emplace(cur, false);
502
2574944
      Kind k = cur.getKind();
503
      // Only traverse Boolean nodes
504
4940790
      if (k == kind::NOT || k == kind::XOR || k == kind::ITE
505
2032740
          || k == kind::IMPLIES || k == kind::OR || k == kind::AND
506
3563410
          || (k == kind::EQUAL && cur[0].getType().isBoolean()))
507
      {
508
        // Preserve the order of the recursive version
509
7130256
        for (size_t i = 0, size = cur.getNumChildren(); i < size; ++i)
510
        {
511
5274187
          visit.push_back(cur[size - 1 - i]);
512
        }
513
      }
514
2574944
      continue;
515
    }
516
2365839
    else if (!it->second)
517
    {
518
2365839
      it->second = true;
519
2365839
      Kind k = cur.getKind();
520
2365839
      switch (k)
521
      {
522
        case kind::NOT: Assert(hasLiteral(cur[0])); break;
523
249496
        case kind::XOR: handleXor(cur); break;
524
83610
        case kind::ITE: handleIte(cur); break;
525
8500
        case kind::IMPLIES: handleImplies(cur); break;
526
390963
        case kind::OR: handleOr(cur); break;
527
644804
        case kind::AND: handleAnd(cur); break;
528
988466
        default:
529
988466
          if (k == kind::EQUAL && cur[0].getType().isBoolean())
530
          {
531
269591
            handleIff(cur);
532
          }
533
          else
534
          {
535
718879
            convertAtom(cur);
536
          }
537
988462
          break;
538
      }
539
    }
540
2365835
    visit.pop_back();
541
  }
542
543
3126828
  nodeLit = getLiteral(node);
544
6253656
  Trace("cnf") << "toCNF(): resulting literal: "
545
3126828
               << (!negated ? nodeLit : ~nodeLit) << "\n";
546
6253656
  return negated ? ~nodeLit : nodeLit;
547
}
548
549
129671
void CnfStream::convertAndAssertAnd(TNode node, bool negated)
550
{
551
129671
  Assert(node.getKind() == kind::AND);
552
259342
  Trace("cnf") << "CnfStream::convertAndAssertAnd(" << node
553
129671
               << ", negated = " << (negated ? "true" : "false") << ")\n";
554
129671
  if (!negated) {
555
    // If the node is a conjunction, we handle each conjunct separately
556
374409
    for(TNode::const_iterator conjunct = node.begin(), node_end = node.end();
557
374409
        conjunct != node_end; ++conjunct ) {
558
357924
      convertAndAssert(*conjunct, false);
559
    }
560
  } else {
561
    // If the node is a disjunction, we construct a clause and assert it
562
113184
    int nChildren = node.getNumChildren();
563
226368
    SatClause clause(nChildren);
564
113184
    TNode::const_iterator disjunct = node.begin();
565
1478888
    for(int i = 0; i < nChildren; ++ disjunct, ++ i) {
566
1365704
      Assert(disjunct != node.end());
567
1365704
      clause[i] = toCNF(*disjunct, true);
568
    }
569
113184
    Assert(disjunct == node.end());
570
113184
    assertClause(node.negate(), clause);
571
  }
572
129670
}
573
574
440038
void CnfStream::convertAndAssertOr(TNode node, bool negated)
575
{
576
440038
  Assert(node.getKind() == kind::OR);
577
880076
  Trace("cnf") << "CnfStream::convertAndAssertOr(" << node
578
440038
               << ", negated = " << (negated ? "true" : "false") << ")\n";
579
440038
  if (!negated) {
580
    // If the node is a disjunction, we construct a clause and assert it
581
439857
    int nChildren = node.getNumChildren();
582
879714
    SatClause clause(nChildren);
583
439857
    TNode::const_iterator disjunct = node.begin();
584
1775670
    for(int i = 0; i < nChildren; ++ disjunct, ++ i) {
585
1335813
      Assert(disjunct != node.end());
586
1335813
      clause[i] = toCNF(*disjunct, false);
587
    }
588
439857
    Assert(disjunct == node.end());
589
439857
    assertClause(node, clause);
590
  } else {
591
    // If the node is a conjunction, we handle each conjunct separately
592
1183
    for(TNode::const_iterator conjunct = node.begin(), node_end = node.end();
593
1183
        conjunct != node_end; ++conjunct ) {
594
1002
      convertAndAssert(*conjunct, true);
595
    }
596
  }
597
440038
}
598
599
49
void CnfStream::convertAndAssertXor(TNode node, bool negated)
600
{
601
49
  Assert(node.getKind() == kind::XOR);
602
98
  Trace("cnf") << "CnfStream::convertAndAssertXor(" << node
603
49
               << ", negated = " << (negated ? "true" : "false") << ")\n";
604
49
  if (!negated) {
605
    // p XOR q
606
49
    SatLiteral p = toCNF(node[0], false);
607
49
    SatLiteral q = toCNF(node[1], false);
608
    // Construct the clauses (p => !q) and (!q => p)
609
98
    SatClause clause1(2);
610
49
    clause1[0] = ~p;
611
49
    clause1[1] = ~q;
612
49
    assertClause(node, clause1);
613
98
    SatClause clause2(2);
614
49
    clause2[0] = p;
615
49
    clause2[1] = q;
616
49
    assertClause(node, clause2);
617
  } else {
618
    // !(p XOR q) is the same as p <=> q
619
    SatLiteral p = toCNF(node[0], false);
620
    SatLiteral q = toCNF(node[1], false);
621
    // Construct the clauses (p => q) and (q => p)
622
    SatClause clause1(2);
623
    clause1[0] = ~p;
624
    clause1[1] = q;
625
    assertClause(node.negate(), clause1);
626
    SatClause clause2(2);
627
    clause2[0] = p;
628
    clause2[1] = ~q;
629
    assertClause(node.negate(), clause2);
630
  }
631
49
}
632
633
16103
void CnfStream::convertAndAssertIff(TNode node, bool negated)
634
{
635
16103
  Assert(node.getKind() == kind::EQUAL);
636
32206
  Trace("cnf") << "CnfStream::convertAndAssertIff(" << node
637
16103
               << ", negated = " << (negated ? "true" : "false") << ")\n";
638
16103
  if (!negated) {
639
    // p <=> q
640
15581
    SatLiteral p = toCNF(node[0], false);
641
15581
    SatLiteral q = toCNF(node[1], false);
642
    // Construct the clauses (p => q) and (q => p)
643
31162
    SatClause clause1(2);
644
15581
    clause1[0] = ~p;
645
15581
    clause1[1] = q;
646
15581
    assertClause(node, clause1);
647
31162
    SatClause clause2(2);
648
15581
    clause2[0] = p;
649
15581
    clause2[1] = ~q;
650
15581
    assertClause(node, clause2);
651
  } else {
652
    // !(p <=> q) is the same as p XOR q
653
522
    SatLiteral p = toCNF(node[0], false);
654
522
    SatLiteral q = toCNF(node[1], false);
655
    // Construct the clauses (p => !q) and (!q => p)
656
1044
    SatClause clause1(2);
657
522
    clause1[0] = ~p;
658
522
    clause1[1] = ~q;
659
522
    assertClause(node.negate(), clause1);
660
1044
    SatClause clause2(2);
661
522
    clause2[0] = p;
662
522
    clause2[1] = q;
663
522
    assertClause(node.negate(), clause2);
664
  }
665
16103
}
666
667
66492
void CnfStream::convertAndAssertImplies(TNode node, bool negated)
668
{
669
66492
  Assert(node.getKind() == kind::IMPLIES);
670
132984
  Trace("cnf") << "CnfStream::convertAndAssertImplies(" << node
671
66492
               << ", negated = " << (negated ? "true" : "false") << ")\n";
672
66492
  if (!negated) {
673
    // p => q
674
66253
    SatLiteral p = toCNF(node[0], false);
675
66253
    SatLiteral q = toCNF(node[1], false);
676
    // Construct the clause ~p || q
677
132506
    SatClause clause(2);
678
66253
    clause[0] = ~p;
679
66253
    clause[1] = q;
680
66253
    assertClause(node, clause);
681
  } else {// Construct the
682
    // !(p => q) is the same as (p && ~q)
683
239
    convertAndAssert(node[0], false);
684
239
    convertAndAssert(node[1], true);
685
  }
686
66492
}
687
688
17861
void CnfStream::convertAndAssertIte(TNode node, bool negated)
689
{
690
17861
  Assert(node.getKind() == kind::ITE);
691
35722
  Trace("cnf") << "CnfStream::convertAndAssertIte(" << node
692
17861
               << ", negated = " << (negated ? "true" : "false") << ")\n";
693
  // ITE(p, q, r)
694
17861
  SatLiteral p = toCNF(node[0], false);
695
17861
  SatLiteral q = toCNF(node[1], negated);
696
17861
  SatLiteral r = toCNF(node[2], negated);
697
  // Construct the clauses:
698
  // (p => q) and (!p => r)
699
  //
700
  // Note that below q and r can be used directly because whether they are
701
  // negated has been push to the literal definitions above
702
35722
  Node nnode = node;
703
17861
  if( negated ){
704
71
    nnode = node.negate();
705
  }
706
35722
  SatClause clause1(2);
707
17861
  clause1[0] = ~p;
708
17861
  clause1[1] = q;
709
17861
  assertClause(nnode, clause1);
710
35722
  SatClause clause2(2);
711
17861
  clause2[0] = p;
712
17861
  clause2[1] = r;
713
17861
  assertClause(nnode, clause2);
714
17861
}
715
716
// At the top level we must ensure that all clauses that are asserted are
717
// not unit, except for the direct assertions. This allows us to remove the
718
// clauses later when they are not needed anymore (lemmas for example).
719
473884
void CnfStream::convertAndAssert(TNode node,
720
                                 bool removable,
721
                                 bool negated,
722
                                 bool input)
723
{
724
947768
  Trace("cnf") << "convertAndAssert(" << node
725
947768
               << ", negated = " << (negated ? "true" : "false")
726
473884
               << ", removable = " << (removable ? "true" : "false") << ")\n";
727
473884
  d_removable = removable;
728
947768
  TimerStat::CodeTimer codeTimer(d_stats.d_cnfConversionTime, true);
729
473888
  convertAndAssert(node, negated);
730
473880
}
731
732
885457
void CnfStream::convertAndAssert(TNode node, bool negated)
733
{
734
1770914
  Trace("cnf") << "convertAndAssert(" << node
735
885457
               << ", negated = " << (negated ? "true" : "false") << ")\n";
736
737
885457
  d_resourceManager->spendResource(Resource::CnfStep);
738
739
885457
  switch(node.getKind()) {
740
129672
    case kind::AND: convertAndAssertAnd(node, negated); break;
741
440038
    case kind::OR: convertAndAssertOr(node, negated); break;
742
49
    case kind::XOR: convertAndAssertXor(node, negated); break;
743
66492
    case kind::IMPLIES: convertAndAssertImplies(node, negated); break;
744
17861
    case kind::ITE: convertAndAssertIte(node, negated); break;
745
52172
    case kind::NOT: convertAndAssert(node[0], !negated); break;
746
87788
    case kind::EQUAL:
747
87788
      if (node[0].getType().isBoolean())
748
      {
749
16103
        convertAndAssertIff(node, negated);
750
16103
        break;
751
      }
752
      CVC5_FALLTHROUGH;
753
    default:
754
    {
755
326146
      Node nnode = node;
756
163073
      if (negated)
757
      {
758
48581
        nnode = node.negate();
759
      }
760
      // Atoms
761
326146
      assertClause(nnode, toCNF(node, negated));
762
  }
763
163069
    break;
764
  }
765
885450
}
766
767
16153
CnfStream::Statistics::Statistics(const std::string& name)
768
16153
    : d_cnfConversionTime(smtStatisticsRegistry().registerTimer(
769
16153
        name + "::CnfStream::cnfConversionTime"))
770
{
771
16153
}
772
773
}  // namespace prop
774
29502
}  // namespace cvc5