org.chocosolver.solver.constraints.binary.PropDistanceXYC Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of choco-solver Show documentation
Show all versions of choco-solver Show documentation
Open-source constraint solver.
/*
* This file is part of choco-solver, http://choco-solver.org/
*
* Copyright (c) 2019, IMT Atlantique. All rights reserved.
*
* Licensed under the BSD 4-clause license.
*
* See LICENSE file in the project root for full license information.
*/
package org.chocosolver.solver.constraints.binary;
import org.chocosolver.solver.constraints.Operator;
import org.chocosolver.solver.constraints.Propagator;
import org.chocosolver.solver.constraints.PropagatorPriority;
import org.chocosolver.solver.exception.ContradictionException;
import org.chocosolver.solver.exception.SolverException;
import org.chocosolver.solver.variables.IntVar;
import org.chocosolver.solver.variables.delta.IIntDeltaMonitor;
import org.chocosolver.solver.variables.events.IntEventType;
import org.chocosolver.util.ESat;
import org.chocosolver.util.iterators.DisposableRangeIterator;
import org.chocosolver.util.procedure.UnaryIntProcedure;
/**
* | X - Y | op C
op = {"==", "<", ">", "=/="}
*
* @author Charles Prud'homme
* @since 21/03/12
*/
public class PropDistanceXYC extends Propagator {
private final Operator operator;
private final int cste;
private final RemProc remproc;
private final IIntDeltaMonitor[] idms;
public PropDistanceXYC(IntVar[] vars, Operator operator, int cste) {
super(vars, PropagatorPriority.BINARY, true);
if (operator == Operator.EQ) {
this.idms = new IIntDeltaMonitor[this.vars.length];
for (int i = 0; i < this.vars.length; i++) {
idms[i] = vars[i].hasEnumeratedDomain() ? this.vars[i].monitorDelta(this) : IIntDeltaMonitor.Default.NONE;
}
} else {
this.idms = new IIntDeltaMonitor[0];
}
this.operator = operator;
this.cste = cste;
this.remproc = new RemProc(this);
}
@Override
public int getPropagationConditions(int idx) {
if (vars[idx].hasEnumeratedDomain()) {
return IntEventType.all();
} else {
return IntEventType.boundAndInst();
}
}
@Override
public void propagate(int evtmask) throws ContradictionException {
//cste < 0, and |vars[0]-vars[1]| always >= 0
if (cste < 0) {
switch (operator) {
case NQ:
case GT:
this.setPassive();
break;
default: throw new SolverException("Invalid PropDistanceXYC operator "+operator);
}
}
if (operator == Operator.EQ) {
if (vars[0].hasEnumeratedDomain()) {
filterFromVarToVar(vars[0], vars[1]);
} else {
vars[0].updateBounds(vars[1].getLB() - cste, vars[1].getUB() + cste, this);
}
if (vars[1].hasEnumeratedDomain()) {
filterFromVarToVar(vars[1], vars[0]);
} else {
vars[1].updateBounds(vars[0].getLB() - cste, vars[0].getUB() + cste, this);
}
} else if (operator == Operator.GT) {
filterGT();
} else if (operator == Operator.LT) {
filterLT();
} else {
filterNeq();
}
for (int i = 0; i < idms.length; i++) {
idms[i].unfreeze();
}
}
@Override
public void propagate(int varIdx, int mask) throws ContradictionException {
int idx2 = varIdx == 0 ? 1 : 0;
switch (operator) {
case EQ:
if (IntEventType.isInstantiate(mask)) {
filterOnInst(vars[idx2], vars[varIdx].getValue());
} else {
if (IntEventType.isRemove(mask) && vars[varIdx].hasEnumeratedDomain()) {
idms[varIdx].freeze();
idms[varIdx].forEachRemVal(remproc.set(varIdx));
idms[varIdx].unfreeze();
}
if (IntEventType.isInclow(mask)) {
filterOnInf(vars[varIdx], vars[idx2]);
}
if (IntEventType.isDecupp(mask)) {
filterOnSup(vars[varIdx], vars[idx2]);
}
}
break;
case NQ:
filterNeq();
break;
case GT:
if (IntEventType.isInstantiate(mask)) {
filterGTonVar(vars[varIdx], vars[idx2]);
} else if (IntEventType.isBound(mask)) {
filterGTonVar(vars[varIdx], vars[idx2]);
}
break;
case LT:
if (IntEventType.isInstantiate(mask)) {
filterLTonVar(vars[varIdx], vars[idx2]);
} else if (IntEventType.isBound(mask)) {
filterLTonVar(vars[varIdx], vars[idx2]);
}
break;
default: throw new SolverException("Invalid PropDistanceXYC operator "+operator);
}
}
@Override
public ESat isEntailed() {
if (isCompletelyInstantiated()) {
switch (operator) {
case EQ:
return ESat.eval(Math.abs(vars[0].getValue() - vars[1].getValue()) == cste);
case GT:
return ESat.eval(Math.abs(vars[0].getValue() - vars[1].getValue()) > cste);
case LT:
return ESat.eval(Math.abs(vars[0].getValue() - vars[1].getValue()) < cste);
case NQ:
return ESat.eval(Math.abs(vars[0].getValue() - vars[1].getValue()) != cste);
default: throw new SolverException("Invalid PropDistanceXYC operator "+operator);
}
}
return ESat.UNDEFINED;
}
@Override
public String toString() {
StringBuilder st = new StringBuilder();
st.append("|").append(vars[0].getName()).append(" - ").append(vars[1].getName()).append("|");
switch (operator) {
case EQ:
st.append("=");
break;
case GT:
st.append(">");
break;
case LT:
st.append("<");
break;
case NQ:
st.append("=/=");
break;
default: throw new SolverException("Invalid PropDistanceXYC operator "+operator);
}
st.append(cste);
return st.toString();
}
//*************************************************************//
// Methods for filtering //
//*************************************************************//
/**
* Initial propagation in case of EQ and enumerated domains
*
* @throws ContradictionException
*/
public void filterFromVarToVar(IntVar var1, IntVar var2) throws ContradictionException {
DisposableRangeIterator it = var1.getRangeIterator(true);
try {
while (it.hasNext()) {
int from = it.min();
int to = it.max();
for (int value = from; value <= to; value++)
if (!var2.contains(value - cste) && !var2.contains(value + cste)) {
var1.removeValue(value, this);
}
it.next();
}
} finally {
it.dispose();
}
}
/**
* In case of a GT
*/
public void filterGT() throws ContradictionException {
if (cste >= 0) {
int lbv0 = vars[1].getUB() - cste;
int ubv0 = vars[1].getLB() + cste;
// remove interval [lbv0, ubv0] from domain of vars[0]
vars[0].removeInterval(lbv0, ubv0, this);
int lbv1 = vars[0].getUB() - cste;
int ubv1 = vars[0].getLB() + cste;
// remove interval [lbv1, ubv1] from domain of vars[1]
vars[1].removeInterval(lbv1, ubv1, this);
} else {
this.setPassive();
}
}
/**
* In case of a GT, due to a modification on v0 domain
*/
public void filterGTonVar(IntVar v0, IntVar v1) throws ContradictionException {
if (cste >= 0) {
int lbv0 = v0.getUB() - cste;
int ubv0 = v0.getLB() + cste;
// remove interval [lbv0, ubv0] from domain of vars[0]
v1.removeInterval(lbv0, ubv0, this);
} else {
this.setPassive();
}
}
/**
* In case of a LT
*/
public void filterLT() throws ContradictionException {
vars[0].updateBounds(vars[1].getLB() - cste + 1, vars[1].getUB() + cste - 1, this);
vars[1].updateBounds(vars[0].getLB() - cste + 1, vars[0].getUB() + cste - 1, this);
}
/**
* In case of a LT, due to a modification on v0 domain
*/
public void filterLTonVar(IntVar v0, IntVar v1) throws ContradictionException {
v1.updateBounds(v0.getLB() - cste + 1, v0.getUB() + cste - 1, this);
}
/**
* In case of a EQ, due to a modification of the lower bound of v0
*/
public void filterOnInf(IntVar v0, IntVar v1) throws ContradictionException {
if (v1.hasEnumeratedDomain()) {
int end = v0.getLB() + cste;
for (int val = v0.getLB(); val <= end; val = v1.nextValue(val)) {
if (!v0.contains(val - cste) && !v0.contains(val + cste)) {
v1.removeValue(val, this);
}
}
} else {
v1.updateLowerBound(v0.getLB() - cste, this);
}
}
/**
* In case of a EQ, due to a modification of the upper bound of v0
*/
public void filterOnSup(IntVar v0, IntVar v1) throws ContradictionException {
if (v1.hasEnumeratedDomain()) {
int initval;
if (v0.getUB() - cste > v1.getLB()) {
initval = v1.nextValue(v0.getUB() - cste - 1);
} else {
initval = v1.getLB();
}
int val = initval;
do {
if (!v0.contains(val - cste) && !v0.contains(val + cste)) {
v1.removeValue(val, this);
}
val = v1.nextValue(val);
} while (val <= v1.getUB() && val > initval); //todo : pourquoi besoin du deuxieme currentElement ?
} else {
v1.updateUpperBound(v0.getUB() + cste, this);
}
}
/**
* In case of a EQ, due to the instantion to one variable to val
*/
public void filterOnInst(IntVar v, int val) throws ContradictionException {
if (!v.contains(val + cste)) {
v.instantiateTo(val - cste, this);
} else if (!v.contains(val - cste)) {
v.instantiateTo(val + cste, this);
} else {
if (v.hasEnumeratedDomain()) {
DisposableRangeIterator rit = v.getRangeIterator(true);
try {
while (rit.hasNext()) {
int from = rit.min();
int to = rit.max();
for (int value = from; value <= to; value++) {
if (value != (val - cste) && value != (val + cste)) {
v.removeValue(value, this);
}
}
rit.next();
}
} finally {
rit.dispose();
}
} else {
v.updateBounds(val - cste, val + cste, this);
}
}
}
public void filterNeq() throws ContradictionException {
if (cste >= 0) {
if (vars[0].isInstantiated()) {
vars[1].removeValue(vars[0].getValue() + cste, this);
vars[1].removeValue(vars[0].getValue() - cste, this);
}
if (vars[1].isInstantiated()) {
vars[0].removeValue(vars[1].getValue() + cste, this);
vars[0].removeValue(vars[1].getValue() - cste, this);
}
} else {
this.setPassive();
}
}
private static class RemProc implements UnaryIntProcedure {
private int idx;
private final PropDistanceXYC p;
public RemProc(PropDistanceXYC p) {
this.p = p;
}
@Override
public UnaryIntProcedure set(Integer integer) {
this.idx = integer;
return this;
}
@Override
public void execute(int i) throws ContradictionException {
if (idx == 0) {
if (!p.vars[0].contains(i + 2 * p.cste)) {
p.vars[1].removeValue(i + p.cste, this.p);
}
if (!p.vars[0].contains(i - 2 * p.cste)) {
p.vars[1].removeValue(i - p.cste, this.p);
}
} else {
if (!p.vars[1].contains(i + 2 * p.cste)) {
p.vars[0].removeValue(i + p.cste, this.p);
}
if (!p.vars[1].contains(i - 2 * p.cste)) {
p.vars[0].removeValue(i - p.cste, this.p);
}
}
}
}
}