org.chocosolver.solver.constraints.nary.automata.PropCostRegular 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) 2023, 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.nary.automata;
import gnu.trove.stack.TIntStack;
import gnu.trove.stack.array.TIntArrayStack;
import org.chocosolver.memory.IEnvironment;
import org.chocosolver.memory.IStateBool;
import org.chocosolver.solver.constraints.Propagator;
import org.chocosolver.solver.constraints.PropagatorPriority;
import org.chocosolver.solver.constraints.nary.automata.FA.ICostAutomaton;
import org.chocosolver.solver.constraints.nary.automata.FA.utils.Bounds;
import org.chocosolver.solver.constraints.nary.automata.structure.costregular.StoredValuedDirectedMultiGraph;
import org.chocosolver.solver.exception.ContradictionException;
import org.chocosolver.solver.variables.IntVar;
import org.chocosolver.solver.variables.delta.IIntDeltaMonitor;
import org.chocosolver.solver.variables.events.IntEventType;
import org.chocosolver.solver.variables.events.PropagatorEventType;
import org.chocosolver.util.ESat;
import org.chocosolver.util.iterators.DisposableIntIterator;
import org.chocosolver.util.objects.StoredIndexedBipartiteSet;
import org.chocosolver.util.procedure.UnaryIntProcedure;
import static java.lang.Math.ceil;
import static java.lang.Math.floor;
/**
*
*
* @author Charles Prud'homme
* @since 06/06/11
*/
public class PropCostRegular extends Propagator {
private final int zIdx;
private final StoredValuedDirectedMultiGraph graph;
private final ICostAutomaton cautomaton;
private final TIntStack toRemove;
private final IStateBool boundChange;
private int lastWorld = -1;
private long lastNbOfBacktracks = -1, lastNbOfRestarts = -1;
private final RemProc rem_proc;
private final IIntDeltaMonitor[] idms;
public PropCostRegular(IntVar[] variables, ICostAutomaton cautomaton, StoredValuedDirectedMultiGraph graph) {
super(variables, PropagatorPriority.CUBIC, true);
this.idms = new IIntDeltaMonitor[this.vars.length - 1];
for (int i = 0; i < this.vars.length - 1; i++) {
idms[i] = this.vars[i].monitorDelta(this);
idms[i].startMonitoring();
}
this.zIdx = vars.length - 1;
this.rem_proc = new RemProc(this);
IEnvironment environment = model.getEnvironment();
this.toRemove = new TIntArrayStack();
this.boundChange = environment.makeBool(false);
this.graph = graph;
this.cautomaton = cautomaton;
}
@Override
public int getPropagationConditions(int vIdx) {
return (vIdx != vars.length - 1 ? IntEventType.all() : IntEventType.boundAndInst());
}
/**
* Build internal structure of the propagator, if necessary
*
* @throws org.chocosolver.solver.exception.ContradictionException if initialisation encounters a contradiction
*/
protected void initialize() throws ContradictionException {
Bounds bounds = this.cautomaton.getCounters().get(0).bounds();
vars[zIdx].updateBounds(bounds.min.value, bounds.max.value, this);
this.prefilter();
}
@Override
public void propagate(int evtmask) throws ContradictionException {
if (PropagatorEventType.isFullPropagation(evtmask)) {
// as the graph was build on initial domain, this is allowed (specific case)
for (int i = 0; i < idms.length; i++) {
idms[i].forEachRemVal(rem_proc.set(i));
}
initialize();
}
filter();
}
@Override
public void propagate(int varIdx, int mask) throws ContradictionException {
checkWorld();
if (varIdx == zIdx) { // z only deals with bound events
boundChange.set(true);
} else { // other variables only deals with removal events
idms[varIdx].forEachRemVal(rem_proc.set(varIdx));
}
forcePropagate(PropagatorEventType.CUSTOM_PROPAGATION);
}
@Override
public ESat isEntailed() {
if (isCompletelyInstantiated()) {
int first = this.graph.sourceIndex;
boolean found;
double cost = 0.0;
int[] str = new int[vars.length - 1];
for (int i = 0; i < vars.length - 1; i++) {
found = false;
str[i] = vars[i].getValue();
StoredIndexedBipartiteSet bs = this.graph.GNodes.outArcs[first];
DisposableIntIterator it = bs.getIterator();
while (!found && it.hasNext()) {
int idx = it.next();
if (this.graph.GArcs.values[idx] == vars[i].getValue()) {
found = true;
first = this.graph.GArcs.dests[idx];
cost += this.graph.GArcs.costs[idx];
}
}
if (!found)
return ESat.FALSE;
}
int intCost = vars[zIdx].getValue();
return ESat.eval(cost == intCost && cautomaton.run(str));
}
return ESat.UNDEFINED;
}
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
private void prefilter() throws ContradictionException {
double zinf = this.graph.GNodes.spft.get(this.graph.sourceIndex);
double zsup = this.graph.GNodes.lpfs.get(this.graph.tinkIndex);
vars[zIdx].updateBounds((int) ceil(zinf), (int) floor(zsup), this);
DisposableIntIterator it = this.graph.inGraph.getIterator();
//for (int id = this.graph.inGraph.nextSetBit(0) ; id >=0 ; id = this.graph.inGraph.nextSetBit(id+1)) {
while (it.hasNext()) {
int id = it.next();
int orig = this.graph.GArcs.origs[id];
int dest = this.graph.GArcs.dests[id];
double acost = this.graph.GArcs.costs[id];
double spfs = this.graph.GNodes.spfs.get(orig);
double lpfs = this.graph.GNodes.lpfs.get(orig);
double spft = this.graph.GNodes.spft.get(dest);
double lpft = this.graph.GNodes.lpft.get(dest);
if ((spfs + spft + acost > vars[zIdx].getUB() || lpfs + lpft + acost < vars[zIdx].getLB()) && this.graph.isNotInStack(id)) {
this.graph.setInStack(id);
this.toRemove.push(id);
}
}
it.dispose();
try {
do {
while (toRemove.size() > 0) {
int id = toRemove.pop();
// toRemove.synchronize();
this.graph.removeArc(id, toRemove, this, this);
}
while (this.graph.toUpdateLeft.size() > 0) {
this.graph.updateLeft(this.graph.toUpdateLeft.pop(), toRemove, this);
}
while (this.graph.toUpdateRight.size() > 0) {
this.graph.updateRight(this.graph.toUpdateRight.pop(), toRemove, this);
}
} while (toRemove.size() > 0);
} catch (ContradictionException e) {
toRemove.clear();
this.graph.inStack.clear();
this.graph.toUpdateLeft.clear();
this.graph.toUpdateRight.clear();
throw e;
}
}
private void checkWorld() {
int currentworld = model.getEnvironment().getWorldIndex();
long currentbt = model.getSolver().getBackTrackCount();
long currentrestart = model.getSolver().getRestartCount();
if (currentworld < lastWorld || currentbt != lastNbOfBacktracks || currentrestart > lastNbOfRestarts) {
this.toRemove.clear();
this.graph.inStack.clear();
this.graph.toUpdateLeft.clear();
this.graph.toUpdateRight.clear();
}
lastWorld = currentworld;
lastNbOfBacktracks = currentbt;
lastNbOfRestarts = currentrestart;
}
private void filter() throws ContradictionException {
if (boundChange.get()) {
boundChange.set(false);
DisposableIntIterator it = this.graph.inGraph.getIterator();
//for (int id = this.graph.inGraph.nextSetBit(0) ; id >=0 ; id = this.graph.inGraph.nextSetBit(id+1)) {
while (it.hasNext()) {
int id = it.next();
int orig = this.graph.GArcs.origs[id];
int dest = this.graph.GArcs.dests[id];
double acost = this.graph.GArcs.costs[id];
double lpfs = this.graph.GNodes.lpfs.get(orig);
double lpft = this.graph.GNodes.lpft.get(dest);
double spfs = this.graph.GNodes.spfs.get(orig);
double spft = this.graph.GNodes.spft.get(dest);
if ((lpfs + lpft + acost < vars[zIdx].getLB() || spfs + spft + acost > vars[zIdx].getUB()) && this.graph.isNotInStack(id)) {
this.graph.setInStack(id);
this.toRemove.push(id);
}
}
it.dispose();
}
do {
while (toRemove.size() > 0) {
int id = toRemove.pop();
// toRemove.synchronize();
this.graph.removeArc(id, toRemove, this, this);
}
while (this.graph.toUpdateLeft.size() > 0) {
this.graph.updateLeft(this.graph.toUpdateLeft.pop(), toRemove, this);
}
while (this.graph.toUpdateRight.size() > 0) {
this.graph.updateRight(this.graph.toUpdateRight.pop(), toRemove, this
);
}
} while (toRemove.size() > 0);
double zinf = this.graph.GNodes.spft.get(this.graph.sourceIndex);
double zsup = this.graph.GNodes.lpfs.get(this.graph.tinkIndex);
vars[zIdx].updateBounds((int) ceil(zinf), (int) floor(zsup), this);
}
private static class RemProc implements UnaryIntProcedure {
private final PropCostRegular p;
private int idxVar;
public RemProc(PropCostRegular p) {
this.p = p;
}
@Override
public UnaryIntProcedure set(Integer idxVar) {
this.idxVar = idxVar;
return this;
}
@Override
public void execute(int i) throws ContradictionException {
StoredIndexedBipartiteSet sup = p.graph.getSupport(idxVar, i);
if (sup != null) {
DisposableIntIterator it = sup.getIterator();
while (it.hasNext()) {
int arcId = it.next();
if (p.graph.isNotInStack(arcId)) {
p.graph.setInStack(arcId);
p.toRemove.push(arcId);
// mod = true;
}
}
it.dispose();
}
}
}
}