Jip J. Dekker
35a3110598
git-subtree-dir: software/chuffed git-subtree-split: 2ed0c01558d2a5c49c1ce57e048d32c17adf92d3
90 lines
1.8 KiB
C++
90 lines
1.8 KiB
C++
#include <chuffed/core/propagator.h>
|
|
|
|
// sum x_i <= y
|
|
|
|
template <int U = 0>
|
|
class BoolLinearLE : public Propagator {
|
|
public:
|
|
vec<BoolView> x;
|
|
IntView<U> y;
|
|
|
|
// Persistent state
|
|
Tint ones;
|
|
|
|
vec<Lit> ps;
|
|
|
|
BoolLinearLE(vec<BoolView>& _x, IntView<U> _y) : x(_x), y(_y), ones(0) {
|
|
for (int i = 0; i < x.size(); i++) x[i].attach(this, i, EVENT_L);
|
|
y.attach(this, x.size(), EVENT_U);
|
|
}
|
|
|
|
void wakeup(int i, int c) {
|
|
if (i < x.size()) ones++;
|
|
pushInQueue();
|
|
}
|
|
|
|
bool propagate() {
|
|
int y_max = y.getMax();
|
|
|
|
if (ones > y_max) ones = y_max+1;
|
|
|
|
setDom2(y, setMin, ones, 1);
|
|
|
|
if (ones == y_max) {
|
|
for (int i = 0; i < x.size(); i++) {
|
|
if (!x[i].isFixed()) x[i].setVal2(0, Reason(prop_id, 0));
|
|
}
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
Clause* explain(Lit p, int inf_id) {
|
|
ps.clear(); ps.growTo(ones+1);
|
|
for (int i = 0, j = 1; j <= ones; i++) {
|
|
if (x[i].isTrue()) ps[j++] = ~x[i];
|
|
}
|
|
if (inf_id == 0) ps.push(y.getMaxLit());
|
|
return Reason_new(ps);
|
|
}
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
//-----
|
|
|
|
// sum x_i (=, <=, <, >=, >) y
|
|
|
|
void bool_linear(vec<BoolView>& x, IntRelType t, IntVar* y) {
|
|
vec<BoolView> x2;
|
|
for (int i = 0; i < x.size(); i++) x2.push(~x[i]);
|
|
switch (t) {
|
|
case IRT_EQ:
|
|
// sum x_i = y <=> sum x_i <= y /\ sum (1-x_i) <= (-y+x.size())
|
|
new BoolLinearLE<0>(x, IntView<0>(y));
|
|
new BoolLinearLE<5>(x2, IntView<5>(y,1,x.size()));
|
|
break;
|
|
case IRT_LE:
|
|
// sum x_i <= y
|
|
new BoolLinearLE<0>(x, IntView<0>(y));
|
|
break;
|
|
case IRT_LT:
|
|
// sum x_i < y <=> sum x_i <= (y-1)
|
|
new BoolLinearLE<4>(x, IntView<4>(y,1,-1));
|
|
break;
|
|
case IRT_GE:
|
|
// sum x_i >= y <=> sum (1-x_i) <= (-y+x.size())
|
|
new BoolLinearLE<5>(x2, IntView<5>(y,1,x.size()));
|
|
break;
|
|
case IRT_GT:
|
|
// sum x_i > y <=> sum (1-x_i) <= (-y+x.size()-1)
|
|
new BoolLinearLE<5>(x2, IntView<5>(y,1,x.size()-1));
|
|
break;
|
|
default:
|
|
CHUFFED_ERROR("Unknown IntRelType %d\n", t);
|
|
}
|
|
}
|
|
|