summaryrefslogtreecommitdiff
path: root/src/preprocessing/passes/pseudo_boolean_processor.h
blob: b5bb051382ecc2d728209a63b73a79124b2a74d5 (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
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
/******************************************************************************
 * Top contributors (to current version):
 *   Tim King, Andres Noetzli, 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.
 * ****************************************************************************
 *
 * [[ Add one-line brief description here ]]
 *
 * [[ Add lengthier description here ]]
 * \todo document this file
 */

#include "cvc5_private.h"

#ifndef CVC5__PREPROCESSING__PASSES__PSEUDO_BOOLEAN_PROCESSOR_H
#define CVC5__PREPROCESSING__PASSES__PSEUDO_BOOLEAN_PROCESSOR_H

#include <optional>
#include <unordered_set>
#include <vector>

#include "context/cdhashmap.h"
#include "context/cdo.h"
#include "expr/node.h"
#include "preprocessing/preprocessing_pass.h"
#include "theory/substitutions.h"
#include "util/rational.h"

namespace cvc5 {
namespace preprocessing {
namespace passes {

class PseudoBooleanProcessor : public PreprocessingPass
{
 public:
  PseudoBooleanProcessor(PreprocessingPassContext* preprocContext);

 protected:
  PreprocessingPassResult applyInternal(
      AssertionPipeline* assertionsToPreprocess) override;

 private:
  /** Assumes that the assertions have been rewritten. */
  void learn(const std::vector<Node>& assertions);

  /** Assumes that the assertions have been rewritten. */
  void applyReplacements(AssertionPipeline* assertionsToPreprocess);

  bool likelyToHelp() const;

  bool isPseudoBoolean(Node v) const;

  // Adds the fact the that integert typed variable v
  //   must be >= 0 to the context.
  // This is explained by the explanation exp.
  // exp cannot be null.
  void addGeqZero(Node v, Node exp);

  // Adds the fact the that integert typed variable v
  //   must be <= 1 to the context.
  // This is explained by the explanation exp.
  // exp cannot be null.
  void addLeqOne(Node v, Node exp);

  static inline bool isIntVar(Node v)
  {
    return v.isVar() && v.getType().isInteger();
  }

  void clear();

  /** Assumes that the assertion has been rewritten. */
  void learn(Node assertion);

  /** Rewrites a node  */
  Node applyReplacements(Node pre);

  void learnInternal(Node assertion, bool negated, Node orig);
  void learnRewrittenGeq(Node assertion, bool negated, Node orig);

  void addSub(Node from, Node to);
  void learnGeqSub(Node geq);

  static Node mkGeqOne(Node v);

  // x ->  <geqZero, leqOne>
  typedef context::CDHashMap<Node, std::pair<Node, Node>> CDNode2PairMap;
  CDNode2PairMap d_pbBounds;
  theory::SubstitutionMap d_subCache;

  typedef std::unordered_set<Node> NodeSet;
  NodeSet d_learningCache;

  context::CDO<unsigned> d_pbs;

  // decompose into \sum pos >= neg + off
  std::optional<Rational> d_off;
  std::vector<Node> d_pos;
  std::vector<Node> d_neg;

  /** Returns true if successful. */
  bool decomposeAssertion(Node assertion, bool negated);
};

}  // namespace passes
}  // namespace preprocessing
}  // namespace cvc5

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