2012-03-28 18:50:03 +00:00
|
|
|
// Copyright 2010-2012 Google
|
2010-09-15 12:42:33 +00:00
|
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
|
|
|
// you may not use this file except in compliance with the License.
|
|
|
|
|
// You may obtain a copy of the License at
|
|
|
|
|
//
|
|
|
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
|
//
|
|
|
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
|
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
|
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
|
|
|
// See the License for the specific language governing permissions and
|
|
|
|
|
// limitations under the License.
|
|
|
|
|
//
|
|
|
|
|
// Range constraints
|
|
|
|
|
|
2011-09-21 15:16:48 +00:00
|
|
|
#include <stddef.h>
|
|
|
|
|
#include <string>
|
|
|
|
|
|
2010-09-15 12:42:33 +00:00
|
|
|
#include "base/logging.h"
|
2011-09-21 15:16:48 +00:00
|
|
|
#include "constraint_solver/constraint_solver.h"
|
2012-07-04 22:05:58 +00:00
|
|
|
#include "constraint_solver/constraint_solveri.h"
|
2010-09-15 12:42:33 +00:00
|
|
|
|
|
|
|
|
namespace operations_research {
|
|
|
|
|
|
|
|
|
|
//-----------------------------------------------------------------------------
|
|
|
|
|
// RangeEquality
|
|
|
|
|
|
2011-08-11 05:15:18 +00:00
|
|
|
namespace {
|
2010-09-15 12:42:33 +00:00
|
|
|
class RangeEquality : public Constraint {
|
|
|
|
|
public:
|
2012-07-05 16:27:57 +00:00
|
|
|
RangeEquality(Solver* const s, IntExpr* const l, IntExpr* const r)
|
2012-07-05 14:36:19 +00:00
|
|
|
: Constraint(s), left_(l), right_(r) {}
|
|
|
|
|
|
2010-09-15 12:42:33 +00:00
|
|
|
virtual ~RangeEquality() {}
|
2012-07-05 14:36:19 +00:00
|
|
|
|
|
|
|
|
virtual void Post() {
|
|
|
|
|
Demon* const d = solver()->MakeConstraintInitialPropagateCallback(this);
|
|
|
|
|
left_->WhenRange(d);
|
|
|
|
|
right_->WhenRange(d);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
virtual void InitialPropagate() {
|
|
|
|
|
left_->SetRange(right_->Min(), right_->Max());
|
|
|
|
|
right_->SetRange(left_->Min(), left_->Max());
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
virtual string DebugString() const {
|
|
|
|
|
return left_->DebugString() + " == " + right_->DebugString();
|
|
|
|
|
}
|
|
|
|
|
|
2012-03-20 13:11:01 +00:00
|
|
|
virtual IntVar* Var() {
|
2012-03-19 17:14:57 +00:00
|
|
|
return solver()->MakeIsEqualVar(left_, right_);
|
|
|
|
|
}
|
2012-07-05 14:36:19 +00:00
|
|
|
|
2011-07-11 20:13:14 +00:00
|
|
|
virtual void Accept(ModelVisitor* const visitor) const {
|
2011-08-11 05:15:18 +00:00
|
|
|
visitor->BeginVisitConstraint(ModelVisitor::kEquality, this);
|
|
|
|
|
visitor->VisitIntegerExpressionArgument(ModelVisitor::kLeftArgument, left_);
|
|
|
|
|
visitor->VisitIntegerExpressionArgument(ModelVisitor::kRightArgument,
|
|
|
|
|
right_);
|
|
|
|
|
visitor->EndVisitConstraint(ModelVisitor::kEquality, this);
|
2011-07-11 20:13:14 +00:00
|
|
|
}
|
|
|
|
|
|
2010-09-15 12:42:33 +00:00
|
|
|
private:
|
2012-07-05 16:27:57 +00:00
|
|
|
IntExpr* const left_;
|
|
|
|
|
IntExpr* const right_;
|
2010-09-15 12:42:33 +00:00
|
|
|
};
|
|
|
|
|
|
|
|
|
|
//-----------------------------------------------------------------------------
|
|
|
|
|
// RangeLessOrEqual
|
|
|
|
|
|
|
|
|
|
class RangeLessOrEqual : public Constraint {
|
|
|
|
|
public:
|
2012-07-05 16:27:57 +00:00
|
|
|
RangeLessOrEqual(Solver* const s, IntExpr* const l, IntExpr* const r);
|
2010-09-15 12:42:33 +00:00
|
|
|
virtual ~RangeLessOrEqual() {}
|
|
|
|
|
virtual void Post();
|
|
|
|
|
virtual void InitialPropagate();
|
|
|
|
|
virtual string DebugString() const;
|
2012-03-20 13:11:01 +00:00
|
|
|
virtual IntVar* Var() {
|
2012-03-19 17:14:57 +00:00
|
|
|
return solver()->MakeIsLessOrEqualVar(left_, right_);
|
|
|
|
|
}
|
2011-07-11 20:13:14 +00:00
|
|
|
virtual void Accept(ModelVisitor* const visitor) const {
|
|
|
|
|
visitor->BeginVisitConstraint(ModelVisitor::kLessOrEqual, this);
|
2011-07-14 23:37:47 +00:00
|
|
|
visitor->VisitIntegerExpressionArgument(ModelVisitor::kLeftArgument, left_);
|
|
|
|
|
visitor->VisitIntegerExpressionArgument(ModelVisitor::kRightArgument,
|
2011-07-11 20:13:14 +00:00
|
|
|
right_);
|
|
|
|
|
visitor->EndVisitConstraint(ModelVisitor::kLessOrEqual, this);
|
|
|
|
|
}
|
|
|
|
|
|
2010-09-15 12:42:33 +00:00
|
|
|
private:
|
2012-07-05 16:27:57 +00:00
|
|
|
IntExpr* const left_;
|
|
|
|
|
IntExpr* const right_;
|
2010-09-15 12:42:33 +00:00
|
|
|
};
|
|
|
|
|
|
2012-07-05 16:27:57 +00:00
|
|
|
RangeLessOrEqual::RangeLessOrEqual(Solver* const s, IntExpr* const l,
|
|
|
|
|
IntExpr* const r)
|
2010-09-15 12:42:33 +00:00
|
|
|
: Constraint(s), left_(l), right_(r) {}
|
|
|
|
|
|
|
|
|
|
void RangeLessOrEqual::Post() {
|
|
|
|
|
Demon* d = solver()->MakeConstraintInitialPropagateCallback(this);
|
|
|
|
|
left_->WhenRange(d);
|
|
|
|
|
right_->WhenRange(d);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void RangeLessOrEqual::InitialPropagate() {
|
|
|
|
|
left_->SetMax(right_->Max());
|
|
|
|
|
right_->SetMin(left_->Min());
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
string RangeLessOrEqual::DebugString() const {
|
|
|
|
|
return left_->DebugString() + " <= " + right_->DebugString();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
//-----------------------------------------------------------------------------
|
|
|
|
|
// RangeGreaterOrEqual
|
|
|
|
|
|
|
|
|
|
class RangeGreaterOrEqual : public Constraint {
|
|
|
|
|
public:
|
2012-07-05 16:27:57 +00:00
|
|
|
RangeGreaterOrEqual(Solver* const s, IntExpr* const l, IntExpr* const r);
|
2010-09-15 12:42:33 +00:00
|
|
|
virtual ~RangeGreaterOrEqual() {}
|
|
|
|
|
virtual void Post();
|
|
|
|
|
virtual void InitialPropagate();
|
|
|
|
|
virtual string DebugString() const;
|
2012-03-20 13:11:01 +00:00
|
|
|
virtual IntVar* Var() {
|
2012-03-19 17:14:57 +00:00
|
|
|
return solver()->MakeIsGreaterOrEqualVar(left_, right_);
|
|
|
|
|
}
|
2011-07-11 20:13:14 +00:00
|
|
|
virtual void Accept(ModelVisitor* const visitor) const {
|
|
|
|
|
visitor->BeginVisitConstraint(ModelVisitor::kGreaterOrEqual, this);
|
2011-07-14 23:37:47 +00:00
|
|
|
visitor->VisitIntegerExpressionArgument(ModelVisitor::kLeftArgument, left_);
|
|
|
|
|
visitor->VisitIntegerExpressionArgument(ModelVisitor::kRightArgument,
|
2011-07-11 20:13:14 +00:00
|
|
|
right_);
|
|
|
|
|
visitor->EndVisitConstraint(ModelVisitor::kGreaterOrEqual, this);
|
|
|
|
|
}
|
|
|
|
|
|
2010-09-15 12:42:33 +00:00
|
|
|
private:
|
2012-07-05 16:27:57 +00:00
|
|
|
IntExpr* const left_;
|
|
|
|
|
IntExpr* const right_;
|
2010-09-15 12:42:33 +00:00
|
|
|
};
|
|
|
|
|
|
2012-07-05 16:27:57 +00:00
|
|
|
RangeGreaterOrEqual::RangeGreaterOrEqual(Solver* const s, IntExpr* const l,
|
|
|
|
|
IntExpr* const r)
|
2010-09-15 12:42:33 +00:00
|
|
|
: Constraint(s), left_(l), right_(r) {}
|
|
|
|
|
|
|
|
|
|
void RangeGreaterOrEqual::Post() {
|
|
|
|
|
Demon* d = solver()->MakeConstraintInitialPropagateCallback(this);
|
|
|
|
|
left_->WhenRange(d);
|
|
|
|
|
right_->WhenRange(d);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void RangeGreaterOrEqual::InitialPropagate() {
|
|
|
|
|
left_->SetMin(right_->Min());
|
|
|
|
|
right_->SetMax(left_->Max());
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
string RangeGreaterOrEqual::DebugString() const {
|
|
|
|
|
return left_->DebugString() + " >= " + right_->DebugString();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
//-----------------------------------------------------------------------------
|
|
|
|
|
// RangeLess
|
|
|
|
|
|
|
|
|
|
class RangeLess : public Constraint {
|
|
|
|
|
public:
|
2012-07-05 16:27:57 +00:00
|
|
|
RangeLess(Solver* const s, IntExpr* const l, IntExpr* const r);
|
2010-09-15 12:42:33 +00:00
|
|
|
virtual ~RangeLess() {}
|
|
|
|
|
virtual void Post();
|
|
|
|
|
virtual void InitialPropagate();
|
|
|
|
|
virtual string DebugString() const;
|
2012-03-20 13:11:01 +00:00
|
|
|
virtual IntVar* Var() {
|
2012-03-19 17:14:57 +00:00
|
|
|
return solver()->MakeIsLessVar(left_, right_);
|
|
|
|
|
}
|
2011-07-11 20:13:14 +00:00
|
|
|
virtual void Accept(ModelVisitor* const visitor) const {
|
|
|
|
|
visitor->BeginVisitConstraint(ModelVisitor::kLess, this);
|
2011-07-14 23:37:47 +00:00
|
|
|
visitor->VisitIntegerExpressionArgument(ModelVisitor::kLeftArgument, left_);
|
|
|
|
|
visitor->VisitIntegerExpressionArgument(ModelVisitor::kRightArgument,
|
2011-07-11 20:13:14 +00:00
|
|
|
right_);
|
|
|
|
|
visitor->EndVisitConstraint(ModelVisitor::kLess, this);
|
|
|
|
|
}
|
|
|
|
|
|
2010-09-15 12:42:33 +00:00
|
|
|
private:
|
2012-07-05 16:27:57 +00:00
|
|
|
IntExpr* const left_;
|
|
|
|
|
IntExpr* const right_;
|
2010-09-15 12:42:33 +00:00
|
|
|
};
|
|
|
|
|
|
2012-07-05 16:27:57 +00:00
|
|
|
RangeLess::RangeLess(Solver* const s, IntExpr* const l, IntExpr* const r)
|
2010-09-15 12:42:33 +00:00
|
|
|
: Constraint(s), left_(l), right_(r) {}
|
|
|
|
|
|
|
|
|
|
void RangeLess::Post() {
|
|
|
|
|
Demon* d = solver()->MakeConstraintInitialPropagateCallback(this);
|
|
|
|
|
left_->WhenRange(d);
|
|
|
|
|
right_->WhenRange(d);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void RangeLess::InitialPropagate() {
|
|
|
|
|
left_->SetMax(right_->Max() - 1);
|
|
|
|
|
right_->SetMin(left_->Min() + 1);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
string RangeLess::DebugString() const {
|
|
|
|
|
return left_->DebugString() + " < " + right_->DebugString();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
//-----------------------------------------------------------------------------
|
|
|
|
|
// RangeGreater
|
|
|
|
|
|
|
|
|
|
class RangeGreater : public Constraint {
|
|
|
|
|
public:
|
2012-07-05 16:27:57 +00:00
|
|
|
RangeGreater(Solver* const s, IntExpr* const l, IntExpr* const r);
|
2010-09-15 12:42:33 +00:00
|
|
|
virtual ~RangeGreater() {}
|
|
|
|
|
virtual void Post();
|
|
|
|
|
virtual void InitialPropagate();
|
|
|
|
|
virtual string DebugString() const;
|
2012-03-20 13:11:01 +00:00
|
|
|
virtual IntVar* Var() {
|
2012-03-19 17:14:57 +00:00
|
|
|
return solver()->MakeIsGreaterVar(left_, right_);
|
|
|
|
|
}
|
2011-07-11 20:13:14 +00:00
|
|
|
virtual void Accept(ModelVisitor* const visitor) const {
|
|
|
|
|
visitor->BeginVisitConstraint(ModelVisitor::kGreater, this);
|
2011-07-14 23:37:47 +00:00
|
|
|
visitor->VisitIntegerExpressionArgument(ModelVisitor::kLeftArgument, left_);
|
|
|
|
|
visitor->VisitIntegerExpressionArgument(ModelVisitor::kRightArgument,
|
2011-07-11 20:13:14 +00:00
|
|
|
right_);
|
|
|
|
|
visitor->EndVisitConstraint(ModelVisitor::kGreater, this);
|
|
|
|
|
}
|
|
|
|
|
|
2010-09-15 12:42:33 +00:00
|
|
|
private:
|
2012-07-05 16:27:57 +00:00
|
|
|
IntExpr* const left_;
|
|
|
|
|
IntExpr* const right_;
|
2010-09-15 12:42:33 +00:00
|
|
|
};
|
|
|
|
|
|
2012-07-05 16:27:57 +00:00
|
|
|
RangeGreater::RangeGreater(Solver* const s, IntExpr* const l, IntExpr* const r)
|
2010-09-15 12:42:33 +00:00
|
|
|
: Constraint(s), left_(l), right_(r) {}
|
|
|
|
|
|
|
|
|
|
void RangeGreater::Post() {
|
|
|
|
|
Demon* d = solver()->MakeConstraintInitialPropagateCallback(this);
|
|
|
|
|
left_->WhenRange(d);
|
|
|
|
|
right_->WhenRange(d);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void RangeGreater::InitialPropagate() {
|
|
|
|
|
left_->SetMin(right_->Min() + 1);
|
|
|
|
|
right_->SetMax(left_->Max() - 1);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
string RangeGreater::DebugString() const {
|
|
|
|
|
return left_->DebugString() + " > " + right_->DebugString();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
//-----------------------------------------------------------------------------
|
|
|
|
|
// DiffVar
|
|
|
|
|
|
|
|
|
|
class DiffVar : public Constraint {
|
|
|
|
|
public:
|
|
|
|
|
DiffVar(Solver* const s, IntVar* const l, IntVar* const r);
|
|
|
|
|
virtual ~DiffVar() {}
|
|
|
|
|
virtual void Post();
|
|
|
|
|
virtual void InitialPropagate();
|
|
|
|
|
virtual string DebugString() const;
|
2012-03-20 13:11:01 +00:00
|
|
|
virtual IntVar* Var() {
|
2012-03-19 17:14:57 +00:00
|
|
|
return solver()->MakeIsDifferentVar(left_, right_);
|
|
|
|
|
}
|
|
|
|
|
|
2011-07-11 20:13:14 +00:00
|
|
|
virtual void Accept(ModelVisitor* const visitor) const {
|
|
|
|
|
visitor->BeginVisitConstraint(ModelVisitor::kNonEqual, this);
|
2011-07-14 23:37:47 +00:00
|
|
|
visitor->VisitIntegerExpressionArgument(ModelVisitor::kLeftArgument, left_);
|
|
|
|
|
visitor->VisitIntegerExpressionArgument(ModelVisitor::kRightArgument,
|
2011-07-11 20:13:14 +00:00
|
|
|
right_);
|
|
|
|
|
visitor->EndVisitConstraint(ModelVisitor::kNonEqual, this);
|
|
|
|
|
}
|
|
|
|
|
|
2010-09-15 12:42:33 +00:00
|
|
|
private:
|
|
|
|
|
IntVar* const left_;
|
|
|
|
|
IntVar* const right_;
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
DiffVar::DiffVar(Solver* const s, IntVar* const l, IntVar* const r)
|
|
|
|
|
: Constraint(s), left_(l), right_(r) {}
|
|
|
|
|
|
|
|
|
|
void DiffVar::Post() {
|
|
|
|
|
Demon* d = solver()->MakeConstraintInitialPropagateCallback(this);
|
|
|
|
|
left_->WhenBound(d);
|
|
|
|
|
right_->WhenBound(d);
|
|
|
|
|
// TODO(user) : improve me, separated demons, actually to test
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void DiffVar::InitialPropagate() {
|
|
|
|
|
if (left_->Bound()) {
|
2012-06-16 07:30:12 +00:00
|
|
|
if (right_->Size() < 0xFFFFFF) {
|
|
|
|
|
right_->RemoveValue(left_->Min()); // we use min instead of value
|
|
|
|
|
} else {
|
|
|
|
|
solver()->AddConstraint(solver()->MakeNonEquality(right_, left_->Min()));
|
|
|
|
|
}
|
2010-09-15 12:42:33 +00:00
|
|
|
}
|
|
|
|
|
if (right_->Bound()) {
|
2012-06-16 07:30:12 +00:00
|
|
|
if (left_->Size() < 0xFFFFFF) {
|
|
|
|
|
left_->RemoveValue(right_->Min()); // see above
|
|
|
|
|
} else {
|
|
|
|
|
solver()->AddConstraint(solver()->MakeNonEquality(left_, right_->Min()));
|
|
|
|
|
}
|
2010-09-15 12:42:33 +00:00
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
string DiffVar::DebugString() const {
|
|
|
|
|
return left_->DebugString() + " != " + right_->DebugString();
|
|
|
|
|
}
|
2012-07-04 22:05:58 +00:00
|
|
|
|
|
|
|
|
// --------------------- Reified API -------------------
|
|
|
|
|
|
|
|
|
|
// IsEqualCt
|
|
|
|
|
|
|
|
|
|
class IsEqualCt : public CastConstraint {
|
|
|
|
|
public:
|
|
|
|
|
IsEqualCt(Solver* const s,
|
|
|
|
|
IntVar* const l,
|
|
|
|
|
IntVar* const r,
|
|
|
|
|
IntVar* const b)
|
|
|
|
|
: CastConstraint(s, b),
|
|
|
|
|
left_(l),
|
|
|
|
|
right_(r),
|
|
|
|
|
left_iterator_(l->MakeDomainIterator(true)),
|
|
|
|
|
right_iterator_(r->MakeDomainIterator(true)),
|
|
|
|
|
range_demon_(NULL),
|
|
|
|
|
domain_demon_(NULL),
|
|
|
|
|
support_(l->Min() - 1) {}
|
|
|
|
|
|
|
|
|
|
virtual ~IsEqualCt() {}
|
|
|
|
|
|
|
|
|
|
virtual void Post() {
|
|
|
|
|
range_demon_ =
|
|
|
|
|
MakeConstraintDemon0(solver(),
|
|
|
|
|
this,
|
|
|
|
|
&IsEqualCt::PropagateRange,
|
|
|
|
|
"PropagateRange");
|
|
|
|
|
domain_demon_ =
|
|
|
|
|
MakeConstraintDemon0(solver(),
|
|
|
|
|
this,
|
|
|
|
|
&IsEqualCt::PropagateDomain,
|
|
|
|
|
"PropagateDomain");
|
|
|
|
|
if (left_->Size() > 32 && right_->Size() > 32) {
|
|
|
|
|
domain_demon_->inhibit(solver());
|
|
|
|
|
} else {
|
|
|
|
|
range_demon_->inhibit(solver());
|
|
|
|
|
}
|
|
|
|
|
left_->WhenDomain(domain_demon_);
|
|
|
|
|
right_->WhenDomain(domain_demon_);
|
|
|
|
|
left_->WhenRange(range_demon_);
|
|
|
|
|
right_->WhenRange(range_demon_);
|
|
|
|
|
Demon* const target_demon =
|
|
|
|
|
MakeConstraintDemon0(solver(),
|
|
|
|
|
this,
|
|
|
|
|
&IsEqualCt::PropagateTarget,
|
|
|
|
|
"PropagateTarget");
|
|
|
|
|
target_var_->WhenBound(target_demon);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
virtual void InitialPropagate() {
|
|
|
|
|
if (left_->Size() > 32 && right_->Size() > 32) {
|
|
|
|
|
PropagateRange();
|
|
|
|
|
} else {
|
|
|
|
|
PropagateDomain();
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void PropagateDomain() {
|
|
|
|
|
if (target_var_->Bound()) {
|
|
|
|
|
PropagateTarget();
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
const int64 support = support_.Value();
|
|
|
|
|
if (!left_->Contains(support) || !right_->Contains(support)) {
|
|
|
|
|
if (!FindSupport()) {
|
|
|
|
|
domain_demon_->inhibit(solver());
|
|
|
|
|
target_var_->SetValue(0);
|
|
|
|
|
} else if (left_->Bound() && right_->Bound()) {
|
|
|
|
|
target_var_->SetValue(1);
|
|
|
|
|
}
|
|
|
|
|
} else if (left_->Bound() && right_->Bound()) {
|
|
|
|
|
target_var_->SetValue(1);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void PropagateRange() {
|
|
|
|
|
if (target_var_->Bound()) {
|
|
|
|
|
PropagateTarget();
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
if (left_->Size() <= 32 || right_->Size() <= 32) {
|
|
|
|
|
range_demon_->inhibit(solver());
|
|
|
|
|
domain_demon_->desinhibit(solver());
|
|
|
|
|
PropagateDomain();
|
|
|
|
|
} else if (left_->Min() > right_->Max() || left_->Max() < right_->Min()) {
|
|
|
|
|
target_var_->SetValue(0);
|
|
|
|
|
range_demon_->inhibit(solver());
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void PropagateTarget() {
|
|
|
|
|
if (target_var_->Min() == 0) {
|
|
|
|
|
if (left_->Bound()) {
|
|
|
|
|
range_demon_->inhibit(solver());
|
|
|
|
|
domain_demon_->inhibit(solver());
|
|
|
|
|
right_->RemoveValue(left_->Min());
|
|
|
|
|
} else if (right_->Bound()) {
|
|
|
|
|
range_demon_->inhibit(solver());
|
|
|
|
|
domain_demon_->inhibit(solver());
|
|
|
|
|
left_->RemoveValue(right_->Min());
|
|
|
|
|
}
|
|
|
|
|
} else { // Var is true.
|
|
|
|
|
left_->SetRange(right_->Min(), right_->Max());
|
|
|
|
|
right_->SetRange(left_->Min(), left_->Max());
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
string DebugString() const {
|
|
|
|
|
return StringPrintf("IsEqualCt(%s, %s, %s)",
|
|
|
|
|
left_->DebugString().c_str(),
|
|
|
|
|
right_->DebugString().c_str(),
|
|
|
|
|
target_var_->DebugString().c_str());
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void Accept(ModelVisitor* const visitor) const {
|
|
|
|
|
visitor->BeginVisitConstraint(ModelVisitor::kIsEqual, this);
|
|
|
|
|
visitor->VisitIntegerExpressionArgument(ModelVisitor::kLeftArgument, left_);
|
|
|
|
|
visitor->VisitIntegerExpressionArgument(ModelVisitor::kRightArgument,
|
|
|
|
|
right_);
|
|
|
|
|
visitor->VisitIntegerExpressionArgument(ModelVisitor::kTargetArgument,
|
|
|
|
|
target_var_);
|
|
|
|
|
visitor->EndVisitConstraint(ModelVisitor::kIsEqual, this);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private:
|
|
|
|
|
bool FindSupport() {
|
|
|
|
|
bool found = false;
|
|
|
|
|
if (left_->Max() < right_->Min() || left_->Min() > right_->Max()) {
|
|
|
|
|
return false;
|
|
|
|
|
}
|
|
|
|
|
// Find new support.
|
|
|
|
|
if (left_->Size() <= right_->Size()) {
|
|
|
|
|
for (left_iterator_->Init();
|
|
|
|
|
left_iterator_->Ok();
|
|
|
|
|
left_iterator_->Next()) {
|
|
|
|
|
const int64 value = left_iterator_->Value();
|
|
|
|
|
if (right_->Contains(value)) {
|
|
|
|
|
found = true;
|
|
|
|
|
support_.SetValue(solver(), value);
|
|
|
|
|
break;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
} else {
|
|
|
|
|
for (right_iterator_->Init();
|
|
|
|
|
right_iterator_->Ok();
|
|
|
|
|
right_iterator_->Next()) {
|
|
|
|
|
const int64 value = right_iterator_->Value();
|
|
|
|
|
if (left_->Contains(value)) {
|
|
|
|
|
found = true;
|
|
|
|
|
support_.SetValue(solver(), value);
|
|
|
|
|
break;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
return found;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
IntVar* const left_;
|
|
|
|
|
IntVar* const right_;
|
|
|
|
|
IntVarIterator* const left_iterator_;
|
|
|
|
|
IntVarIterator* const right_iterator_;
|
|
|
|
|
Demon* domain_demon_;
|
|
|
|
|
Demon* range_demon_;
|
|
|
|
|
Rev<int64> support_;
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
// IsDifferentCt
|
|
|
|
|
|
|
|
|
|
class IsDifferentCt : public CastConstraint {
|
|
|
|
|
public:
|
|
|
|
|
IsDifferentCt(Solver* const s,
|
|
|
|
|
IntVar* const l,
|
|
|
|
|
IntVar* const r,
|
|
|
|
|
IntVar* const b)
|
|
|
|
|
: CastConstraint(s, b),
|
|
|
|
|
left_(l),
|
|
|
|
|
right_(r),
|
|
|
|
|
left_iterator_(l->MakeDomainIterator(true)),
|
|
|
|
|
right_iterator_(r->MakeDomainIterator(true)),
|
|
|
|
|
range_demon_(NULL),
|
|
|
|
|
domain_demon_(NULL),
|
|
|
|
|
support_(l->Min() - 1) {}
|
|
|
|
|
|
|
|
|
|
virtual ~IsDifferentCt() {}
|
|
|
|
|
|
|
|
|
|
virtual void Post() {
|
|
|
|
|
range_demon_ =
|
|
|
|
|
MakeConstraintDemon0(solver(),
|
|
|
|
|
this,
|
|
|
|
|
&IsDifferentCt::PropagateRange,
|
|
|
|
|
"PropagateRange");
|
|
|
|
|
domain_demon_ =
|
|
|
|
|
MakeConstraintDemon0(solver(),
|
|
|
|
|
this,
|
|
|
|
|
&IsDifferentCt::PropagateDomain,
|
|
|
|
|
"PropagateDomain");
|
|
|
|
|
if (left_->Size() > 32 && right_->Size() > 32) {
|
|
|
|
|
domain_demon_->inhibit(solver());
|
|
|
|
|
} else {
|
|
|
|
|
range_demon_->inhibit(solver());
|
|
|
|
|
}
|
|
|
|
|
left_->WhenDomain(domain_demon_);
|
|
|
|
|
right_->WhenDomain(domain_demon_);
|
|
|
|
|
left_->WhenRange(range_demon_);
|
|
|
|
|
right_->WhenRange(range_demon_);
|
|
|
|
|
Demon* const target_demon =
|
|
|
|
|
MakeConstraintDemon0(solver(),
|
|
|
|
|
this,
|
|
|
|
|
&IsDifferentCt::PropagateTarget,
|
|
|
|
|
"PropagateTarget");
|
|
|
|
|
target_var_->WhenBound(target_demon);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
virtual void InitialPropagate() {
|
|
|
|
|
if (left_->Size() > 32 && right_->Size() > 32) {
|
|
|
|
|
PropagateRange();
|
|
|
|
|
} else {
|
|
|
|
|
PropagateDomain();
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void PropagateDomain() {
|
|
|
|
|
if (target_var_->Bound()) {
|
|
|
|
|
PropagateTarget();
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
const int64 support = support_.Value();
|
|
|
|
|
if (!left_->Contains(support) || !right_->Contains(support)) {
|
|
|
|
|
if (!FindSupport()) {
|
|
|
|
|
domain_demon_->inhibit(solver());
|
|
|
|
|
target_var_->SetValue(1);
|
|
|
|
|
} else if (left_->Bound() && right_->Bound()) {
|
|
|
|
|
target_var_->SetValue(0);
|
|
|
|
|
}
|
|
|
|
|
} else if (left_->Bound() && right_->Bound()) {
|
|
|
|
|
target_var_->SetValue(0);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void PropagateRange() {
|
|
|
|
|
if (target_var_->Bound()) {
|
|
|
|
|
PropagateTarget();
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
if (left_->Size() <= 32 || right_->Size() <= 32) {
|
|
|
|
|
range_demon_->inhibit(solver());
|
|
|
|
|
domain_demon_->desinhibit(solver());
|
|
|
|
|
PropagateDomain();
|
|
|
|
|
} else if (left_->Min() > right_->Max() || left_->Max() < right_->Min()) {
|
|
|
|
|
target_var_->SetValue(1);
|
|
|
|
|
range_demon_->inhibit(solver());
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void PropagateTarget() {
|
|
|
|
|
if (target_var_->Min() == 0) {
|
|
|
|
|
left_->SetRange(right_->Min(), right_->Max());
|
|
|
|
|
right_->SetRange(left_->Min(), left_->Max());
|
|
|
|
|
} else { // Var is true.
|
|
|
|
|
if (left_->Bound()) {
|
|
|
|
|
range_demon_->inhibit(solver());
|
|
|
|
|
domain_demon_->inhibit(solver());
|
|
|
|
|
right_->RemoveValue(left_->Min());
|
|
|
|
|
} else if (right_->Bound()) {
|
|
|
|
|
range_demon_->inhibit(solver());
|
|
|
|
|
domain_demon_->inhibit(solver());
|
|
|
|
|
left_->RemoveValue(right_->Min());
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
string DebugString() const {
|
|
|
|
|
return StringPrintf("IsDifferentCt(%s, %s, %s)",
|
|
|
|
|
left_->DebugString().c_str(),
|
|
|
|
|
right_->DebugString().c_str(),
|
|
|
|
|
target_var_->DebugString().c_str());
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void Accept(ModelVisitor* const visitor) const {
|
|
|
|
|
visitor->BeginVisitConstraint(ModelVisitor::kIsDifferent, this);
|
|
|
|
|
visitor->VisitIntegerExpressionArgument(ModelVisitor::kLeftArgument, left_);
|
|
|
|
|
visitor->VisitIntegerExpressionArgument(ModelVisitor::kRightArgument,
|
|
|
|
|
right_);
|
|
|
|
|
visitor->VisitIntegerExpressionArgument(ModelVisitor::kTargetArgument,
|
|
|
|
|
target_var_);
|
|
|
|
|
visitor->EndVisitConstraint(ModelVisitor::kIsDifferent, this);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private:
|
|
|
|
|
bool FindSupport() {
|
|
|
|
|
bool found = false;
|
|
|
|
|
if (left_->Max() < right_->Min() || left_->Min() > right_->Max()) {
|
|
|
|
|
return false;
|
|
|
|
|
}
|
|
|
|
|
// Find new support.
|
|
|
|
|
if (left_->Size() <= right_->Size()) {
|
|
|
|
|
for (left_iterator_->Init();
|
|
|
|
|
left_iterator_->Ok();
|
|
|
|
|
left_iterator_->Next()) {
|
|
|
|
|
const int64 value = left_iterator_->Value();
|
|
|
|
|
if (right_->Contains(value)) {
|
|
|
|
|
found = true;
|
|
|
|
|
support_.SetValue(solver(), value);
|
|
|
|
|
break;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
} else {
|
|
|
|
|
for (right_iterator_->Init();
|
|
|
|
|
right_iterator_->Ok();
|
|
|
|
|
right_iterator_->Next()) {
|
|
|
|
|
const int64 value = right_iterator_->Value();
|
|
|
|
|
if (left_->Contains(value)) {
|
|
|
|
|
found = true;
|
|
|
|
|
support_.SetValue(solver(), value);
|
|
|
|
|
break;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
return found;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
IntVar* const left_;
|
|
|
|
|
IntVar* const right_;
|
|
|
|
|
IntVarIterator* const left_iterator_;
|
|
|
|
|
IntVarIterator* const right_iterator_;
|
|
|
|
|
Demon* domain_demon_;
|
|
|
|
|
Demon* range_demon_;
|
|
|
|
|
Rev<int64> support_;
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
class IsLessOrEqualCt : public CastConstraint {
|
|
|
|
|
public:
|
|
|
|
|
IsLessOrEqualCt(Solver* const s,
|
2012-07-05 16:27:57 +00:00
|
|
|
IntExpr* const l,
|
|
|
|
|
IntExpr* const r,
|
2012-07-04 22:05:58 +00:00
|
|
|
IntVar* const b)
|
|
|
|
|
: CastConstraint(s, b),
|
|
|
|
|
left_(l),
|
|
|
|
|
right_(r),
|
|
|
|
|
demon_(NULL) {}
|
|
|
|
|
|
|
|
|
|
virtual ~IsLessOrEqualCt() {}
|
|
|
|
|
|
|
|
|
|
virtual void Post() {
|
|
|
|
|
demon_ = solver()->MakeConstraintInitialPropagateCallback(this);
|
|
|
|
|
left_->WhenRange(demon_);
|
|
|
|
|
right_->WhenRange(demon_);
|
|
|
|
|
target_var_->WhenBound(demon_);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
virtual void InitialPropagate() {
|
|
|
|
|
if (target_var_->Bound()) {
|
|
|
|
|
if (target_var_->Min() == 0) {
|
|
|
|
|
right_->SetMin(left_->Min() + 1);
|
|
|
|
|
left_->SetMax(right_->Max() - 1);
|
|
|
|
|
} else { // Var is true.
|
|
|
|
|
right_->SetMin(left_->Min());
|
|
|
|
|
left_->SetMax(right_->Max());
|
|
|
|
|
}
|
|
|
|
|
} else if (right_->Min() >= left_->Max()) {
|
|
|
|
|
demon_->inhibit(solver());
|
|
|
|
|
target_var_->SetValue(1);
|
|
|
|
|
} else if (right_->Max() < left_->Min()) {
|
|
|
|
|
demon_->inhibit(solver());
|
|
|
|
|
target_var_->SetValue(0);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
string DebugString() const {
|
|
|
|
|
return StringPrintf("IsLessOrEqualCt(%s, %s, %s)",
|
|
|
|
|
left_->DebugString().c_str(),
|
|
|
|
|
right_->DebugString().c_str(),
|
|
|
|
|
target_var_->DebugString().c_str());
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void Accept(ModelVisitor* const visitor) const {
|
|
|
|
|
visitor->BeginVisitConstraint(ModelVisitor::kIsLessOrEqual, this);
|
|
|
|
|
visitor->VisitIntegerExpressionArgument(ModelVisitor::kLeftArgument, left_);
|
|
|
|
|
visitor->VisitIntegerExpressionArgument(ModelVisitor::kRightArgument,
|
|
|
|
|
right_);
|
|
|
|
|
visitor->VisitIntegerExpressionArgument(ModelVisitor::kTargetArgument,
|
|
|
|
|
target_var_);
|
|
|
|
|
visitor->EndVisitConstraint(ModelVisitor::kIsLessOrEqual, this);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private:
|
2012-07-05 16:27:57 +00:00
|
|
|
IntExpr* const left_;
|
|
|
|
|
IntExpr* const right_;
|
2012-07-04 22:05:58 +00:00
|
|
|
Demon* demon_;
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
class IsLessCt : public CastConstraint {
|
|
|
|
|
public:
|
|
|
|
|
IsLessCt(Solver* const s,
|
2012-07-05 16:27:57 +00:00
|
|
|
IntExpr* const l,
|
|
|
|
|
IntExpr* const r,
|
2012-07-04 22:05:58 +00:00
|
|
|
IntVar* const b)
|
|
|
|
|
: CastConstraint(s, b),
|
|
|
|
|
left_(l),
|
|
|
|
|
right_(r),
|
|
|
|
|
demon_(NULL) {}
|
|
|
|
|
|
|
|
|
|
virtual ~IsLessCt() {}
|
|
|
|
|
|
|
|
|
|
virtual void Post() {
|
|
|
|
|
demon_ = solver()->MakeConstraintInitialPropagateCallback(this);
|
|
|
|
|
left_->WhenRange(demon_);
|
|
|
|
|
right_->WhenRange(demon_);
|
|
|
|
|
target_var_->WhenBound(demon_);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
virtual void InitialPropagate() {
|
|
|
|
|
if (target_var_->Bound()) {
|
|
|
|
|
if (target_var_->Min() == 0) {
|
|
|
|
|
right_->SetMin(left_->Min());
|
|
|
|
|
left_->SetMax(right_->Max());
|
|
|
|
|
} else { // Var is true.
|
|
|
|
|
right_->SetMin(left_->Min() + 1);
|
|
|
|
|
left_->SetMax(right_->Max() - 1);
|
|
|
|
|
}
|
|
|
|
|
} else if (right_->Min() > left_->Max()) {
|
|
|
|
|
demon_->inhibit(solver());
|
|
|
|
|
target_var_->SetValue(1);
|
|
|
|
|
} else if (right_->Max() <= left_->Min()) {
|
|
|
|
|
demon_->inhibit(solver());
|
|
|
|
|
target_var_->SetValue(0);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
string DebugString() const {
|
|
|
|
|
return StringPrintf("IsLessCt(%s, %s, %s)",
|
|
|
|
|
left_->DebugString().c_str(),
|
|
|
|
|
right_->DebugString().c_str(),
|
|
|
|
|
target_var_->DebugString().c_str());
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void Accept(ModelVisitor* const visitor) const {
|
|
|
|
|
visitor->BeginVisitConstraint(ModelVisitor::kIsLess, this);
|
|
|
|
|
visitor->VisitIntegerExpressionArgument(ModelVisitor::kLeftArgument, left_);
|
|
|
|
|
visitor->VisitIntegerExpressionArgument(ModelVisitor::kRightArgument,
|
|
|
|
|
right_);
|
|
|
|
|
visitor->VisitIntegerExpressionArgument(ModelVisitor::kTargetArgument,
|
|
|
|
|
target_var_);
|
|
|
|
|
visitor->EndVisitConstraint(ModelVisitor::kIsLess, this);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private:
|
2012-07-05 16:27:57 +00:00
|
|
|
IntExpr* const left_;
|
|
|
|
|
IntExpr* const right_;
|
2012-07-04 22:05:58 +00:00
|
|
|
Demon* demon_;
|
|
|
|
|
};
|
2011-08-11 05:15:18 +00:00
|
|
|
} // namespace
|
2010-09-15 12:42:33 +00:00
|
|
|
|
2012-07-05 16:27:57 +00:00
|
|
|
Constraint* Solver::MakeEquality(IntExpr* const l, IntExpr* const r) {
|
2012-07-04 22:05:58 +00:00
|
|
|
CHECK(l != NULL) << "left expression NULL, maybe a bad cast";
|
|
|
|
|
CHECK(r != NULL) << "left expression NULL, maybe a bad cast";
|
|
|
|
|
CHECK_EQ(this, l->solver());
|
|
|
|
|
CHECK_EQ(this, r->solver());
|
2012-07-05 13:27:16 +00:00
|
|
|
if (l->Bound()) {
|
|
|
|
|
return MakeEquality(r, l->Min());
|
|
|
|
|
} else if (r->Bound()) {
|
|
|
|
|
return MakeEquality(l, r->Min());
|
|
|
|
|
} else {
|
|
|
|
|
return RevAlloc(new RangeEquality(this, l, r));
|
|
|
|
|
}
|
2012-07-04 22:05:58 +00:00
|
|
|
}
|
|
|
|
|
|
2012-07-05 16:27:57 +00:00
|
|
|
Constraint* Solver::MakeLessOrEqual(IntExpr* const l, IntExpr* const r) {
|
2012-07-04 22:05:58 +00:00
|
|
|
CHECK(l != NULL) << "left expression NULL, maybe a bad cast";
|
|
|
|
|
CHECK(r != NULL) << "left expression NULL, maybe a bad cast";
|
|
|
|
|
CHECK_EQ(this, l->solver());
|
|
|
|
|
CHECK_EQ(this, r->solver());
|
2012-07-05 13:43:02 +00:00
|
|
|
if (l->Bound()) {
|
|
|
|
|
return MakeGreaterOrEqual(r, l->Min());
|
|
|
|
|
} else if (r->Bound()) {
|
|
|
|
|
return MakeLessOrEqual(l, r->Min());
|
|
|
|
|
} else {
|
|
|
|
|
return RevAlloc(new RangeLessOrEqual(this, l, r));
|
|
|
|
|
}
|
2012-07-04 22:05:58 +00:00
|
|
|
}
|
|
|
|
|
|
2012-07-05 16:27:57 +00:00
|
|
|
Constraint* Solver::MakeGreaterOrEqual(IntExpr* const l, IntExpr* const r) {
|
2012-07-04 22:05:58 +00:00
|
|
|
CHECK(l != NULL) << "left expression NULL, maybe a bad cast";
|
|
|
|
|
CHECK(r != NULL) << "left expression NULL, maybe a bad cast";
|
|
|
|
|
CHECK_EQ(this, l->solver());
|
|
|
|
|
CHECK_EQ(this, r->solver());
|
2012-07-05 13:43:02 +00:00
|
|
|
if (l->Bound()) {
|
|
|
|
|
return MakeLessOrEqual(r, l->Min());
|
|
|
|
|
} else if (r->Bound()) {
|
|
|
|
|
return MakeGreaterOrEqual(l, r->Min());
|
|
|
|
|
} else {
|
|
|
|
|
return RevAlloc(new RangeGreaterOrEqual(this, l, r));
|
|
|
|
|
}
|
2012-07-04 22:05:58 +00:00
|
|
|
}
|
|
|
|
|
|
2012-07-05 16:27:57 +00:00
|
|
|
Constraint* Solver::MakeLess(IntExpr* const l, IntExpr* const r) {
|
2012-07-04 22:05:58 +00:00
|
|
|
CHECK(l != NULL) << "left expression NULL, maybe a bad cast";
|
|
|
|
|
CHECK(r != NULL) << "left expression NULL, maybe a bad cast";
|
|
|
|
|
CHECK_EQ(this, l->solver());
|
|
|
|
|
CHECK_EQ(this, r->solver());
|
2012-07-05 13:43:02 +00:00
|
|
|
if (l->Bound()) {
|
|
|
|
|
return MakeGreater(r, l->Min());
|
|
|
|
|
} else if (r->Bound()) {
|
|
|
|
|
return MakeLess(l, r->Min());
|
|
|
|
|
} else {
|
|
|
|
|
return RevAlloc(new RangeLess(this, l, r));
|
|
|
|
|
}
|
2012-07-04 22:05:58 +00:00
|
|
|
}
|
|
|
|
|
|
2012-07-05 16:27:57 +00:00
|
|
|
Constraint* Solver::MakeGreater(IntExpr* const l, IntExpr* const r) {
|
2012-07-04 22:05:58 +00:00
|
|
|
CHECK(l != NULL) << "left expression NULL, maybe a bad cast";
|
|
|
|
|
CHECK(r != NULL) << "left expression NULL, maybe a bad cast";
|
|
|
|
|
CHECK_EQ(this, l->solver());
|
|
|
|
|
CHECK_EQ(this, r->solver());
|
2012-07-05 13:43:02 +00:00
|
|
|
if (l->Bound()) {
|
|
|
|
|
return MakeLess(r, l->Min());
|
|
|
|
|
} else if (r->Bound()) {
|
|
|
|
|
return MakeGreater(l, r->Min());
|
|
|
|
|
} else {
|
|
|
|
|
return RevAlloc(new RangeGreater(this, l, r));
|
|
|
|
|
}
|
2012-07-04 22:05:58 +00:00
|
|
|
}
|
|
|
|
|
|
2012-07-05 16:27:57 +00:00
|
|
|
Constraint* Solver::MakeNonEquality(IntExpr* const l, IntExpr* const r) {
|
2010-09-15 12:42:33 +00:00
|
|
|
CHECK(l != NULL) << "left expression NULL, maybe a bad cast";
|
|
|
|
|
CHECK(r != NULL) << "left expression NULL, maybe a bad cast";
|
|
|
|
|
CHECK_EQ(this, l->solver());
|
|
|
|
|
CHECK_EQ(this, r->solver());
|
2012-06-29 09:41:24 +00:00
|
|
|
if (l->Bound()) {
|
|
|
|
|
return MakeNonEquality(r, l->Min());
|
|
|
|
|
} else if (r->Bound()) {
|
|
|
|
|
return MakeNonEquality(l, r->Min());
|
|
|
|
|
}
|
2012-07-05 16:27:57 +00:00
|
|
|
return RevAlloc(new DiffVar(this, l->Var(), r->Var()));
|
2010-09-15 12:42:33 +00:00
|
|
|
}
|
|
|
|
|
|
2012-07-04 22:05:58 +00:00
|
|
|
IntVar* Solver::MakeIsEqualVar(IntExpr* const v1, IntExpr* const v2) {
|
|
|
|
|
CHECK_EQ(this, v1->solver());
|
|
|
|
|
CHECK_EQ(this, v2->solver());
|
|
|
|
|
if (v1->Bound()) {
|
|
|
|
|
return MakeIsEqualCstVar(v2->Var(), v1->Min());
|
|
|
|
|
} else if (v2->Bound()) {
|
|
|
|
|
return MakeIsEqualCstVar(v1->Var(), v2->Min());
|
|
|
|
|
}
|
|
|
|
|
IntVar* const var1 = v1->Var();
|
|
|
|
|
IntVar* const var2 = v2->Var();
|
2012-07-05 16:27:57 +00:00
|
|
|
IntExpr* const cache = model_cache_->FindExprExprExpression(
|
2012-07-04 22:05:58 +00:00
|
|
|
var1,
|
|
|
|
|
var2,
|
2012-07-05 16:27:57 +00:00
|
|
|
ModelCache::EXPR_EXPR_IS_EQUAL);
|
2012-07-04 22:05:58 +00:00
|
|
|
if (cache != NULL) {
|
|
|
|
|
return cache->Var();
|
|
|
|
|
} else {
|
|
|
|
|
string name1 = var1->name();
|
|
|
|
|
if (name1.empty()) {
|
|
|
|
|
name1 = var1->DebugString();
|
|
|
|
|
}
|
|
|
|
|
string name2 = var2->name();
|
|
|
|
|
if (name2.empty()) {
|
|
|
|
|
name2 = var2->DebugString();
|
|
|
|
|
}
|
|
|
|
|
IntVar* const boolvar = MakeBoolVar(
|
|
|
|
|
StringPrintf("IsEqualVar(%s, %s)", name1.c_str(), name2.c_str()));
|
|
|
|
|
AddConstraint(MakeIsEqualCt(v1, v2, boolvar));
|
2012-07-05 16:27:57 +00:00
|
|
|
model_cache_->InsertExprExprExpression(
|
2012-07-04 22:05:58 +00:00
|
|
|
boolvar,
|
|
|
|
|
var1,
|
|
|
|
|
var2,
|
2012-07-05 16:27:57 +00:00
|
|
|
ModelCache::EXPR_EXPR_IS_EQUAL);
|
2012-07-04 22:05:58 +00:00
|
|
|
return boolvar;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
Constraint* Solver::MakeIsEqualCt(IntExpr* const v1,
|
|
|
|
|
IntExpr* const v2,
|
|
|
|
|
IntVar* b) {
|
|
|
|
|
CHECK_EQ(this, v1->solver());
|
|
|
|
|
CHECK_EQ(this, v2->solver());
|
|
|
|
|
if (v1->Bound()) {
|
|
|
|
|
return MakeIsEqualCstCt(v2->Var(), v1->Min(), b);
|
|
|
|
|
} else if (v2->Bound()) {
|
|
|
|
|
return MakeIsEqualCstCt(v1->Var(), v2->Min(), b);
|
|
|
|
|
}
|
|
|
|
|
if (b->Bound()) {
|
|
|
|
|
if (b->Min() == 0) {
|
|
|
|
|
return MakeNonEquality(v1->Var(), v2->Var());
|
|
|
|
|
} else {
|
|
|
|
|
return MakeEquality(v1->Var(), v2->Var());
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
return RevAlloc(new IsEqualCt(this, v1->Var(), v2->Var(), b));
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
IntVar* Solver::MakeIsDifferentVar(IntExpr* const v1, IntExpr* const v2) {
|
|
|
|
|
CHECK_EQ(this, v1->solver());
|
|
|
|
|
CHECK_EQ(this, v2->solver());
|
|
|
|
|
if (v1->Bound()) {
|
|
|
|
|
return MakeIsDifferentCstVar(v2->Var(), v1->Min());
|
|
|
|
|
} else if (v2->Bound()) {
|
|
|
|
|
return MakeIsDifferentCstVar(v1->Var(), v2->Min());
|
|
|
|
|
}
|
|
|
|
|
IntVar* const var1 = v1->Var();
|
|
|
|
|
IntVar* const var2 = v2->Var();
|
2012-07-05 16:27:57 +00:00
|
|
|
IntExpr* const cache = model_cache_->FindExprExprExpression(
|
2012-07-04 22:05:58 +00:00
|
|
|
var1,
|
|
|
|
|
var2,
|
2012-07-05 16:27:57 +00:00
|
|
|
ModelCache::EXPR_EXPR_IS_NOT_EQUAL);
|
2012-07-04 22:05:58 +00:00
|
|
|
if (cache != NULL) {
|
|
|
|
|
return cache->Var();
|
|
|
|
|
} else {
|
|
|
|
|
string name1 = var1->name();
|
|
|
|
|
if (name1.empty()) {
|
|
|
|
|
name1 = var1->DebugString();
|
|
|
|
|
}
|
|
|
|
|
string name2 = var2->name();
|
|
|
|
|
if (name2.empty()) {
|
|
|
|
|
name2 = var2->DebugString();
|
|
|
|
|
}
|
|
|
|
|
IntVar* const boolvar = MakeBoolVar(
|
|
|
|
|
StringPrintf("IsDifferentVar(%s, %s)", name1.c_str(), name2.c_str()));
|
|
|
|
|
AddConstraint(MakeIsDifferentCt(v1, v2, boolvar));
|
2012-07-05 16:27:57 +00:00
|
|
|
model_cache_->InsertExprExprExpression(
|
2012-07-04 22:05:58 +00:00
|
|
|
boolvar,
|
|
|
|
|
var1,
|
|
|
|
|
var2,
|
2012-07-05 16:27:57 +00:00
|
|
|
ModelCache::EXPR_EXPR_IS_NOT_EQUAL);
|
2012-07-04 22:05:58 +00:00
|
|
|
return boolvar;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
Constraint* Solver::MakeIsDifferentCt(IntExpr* const v1,
|
|
|
|
|
IntExpr* const v2,
|
|
|
|
|
IntVar* b) {
|
|
|
|
|
CHECK_EQ(this, v1->solver());
|
|
|
|
|
CHECK_EQ(this, v2->solver());
|
|
|
|
|
if (v1->Bound()) {
|
|
|
|
|
return MakeIsDifferentCstCt(v2->Var(), v1->Min(), b);
|
|
|
|
|
} else if (v2->Bound()) {
|
|
|
|
|
return MakeIsDifferentCstCt(v1->Var(), v2->Min(), b);
|
|
|
|
|
}
|
|
|
|
|
return RevAlloc(new IsDifferentCt(this, v1->Var(), v2->Var(), b));
|
|
|
|
|
}
|
|
|
|
|
|
2012-07-05 16:27:57 +00:00
|
|
|
IntVar* Solver::MakeIsLessOrEqualVar(IntExpr* const left,
|
|
|
|
|
IntExpr* const right) {
|
2012-07-04 22:05:58 +00:00
|
|
|
CHECK_EQ(this, left->solver());
|
|
|
|
|
CHECK_EQ(this, right->solver());
|
|
|
|
|
if (left->Bound()) {
|
|
|
|
|
return MakeIsGreaterOrEqualCstVar(right->Var(), left->Min());
|
|
|
|
|
} else if (right->Bound()) {
|
|
|
|
|
return MakeIsLessOrEqualCstVar(left->Var(), right->Min());
|
|
|
|
|
}
|
2012-07-05 16:27:57 +00:00
|
|
|
IntExpr* const cache = model_cache_->FindExprExprExpression(
|
|
|
|
|
left,
|
|
|
|
|
right,
|
|
|
|
|
ModelCache::EXPR_EXPR_IS_LESS_OR_EQUAL);
|
2012-07-04 22:05:58 +00:00
|
|
|
if (cache != NULL) {
|
|
|
|
|
return cache->Var();
|
|
|
|
|
} else {
|
2012-07-05 16:27:57 +00:00
|
|
|
string name1 = left->name();
|
2012-07-04 22:05:58 +00:00
|
|
|
if (name1.empty()) {
|
2012-07-05 16:27:57 +00:00
|
|
|
name1 = left->DebugString();
|
2012-07-04 22:05:58 +00:00
|
|
|
}
|
2012-07-05 16:27:57 +00:00
|
|
|
string name2 = right->name();
|
2012-07-04 22:05:58 +00:00
|
|
|
if (name2.empty()) {
|
2012-07-05 16:27:57 +00:00
|
|
|
name2 = right->DebugString();
|
2012-07-04 22:05:58 +00:00
|
|
|
}
|
|
|
|
|
IntVar* const boolvar = MakeBoolVar(
|
|
|
|
|
StringPrintf("IsLessOrEqual(%s, %s)", name1.c_str(), name2.c_str()));
|
|
|
|
|
|
|
|
|
|
AddConstraint(RevAlloc(
|
|
|
|
|
new IsLessOrEqualCt(this, left->Var(), right->Var(), boolvar)));
|
2012-07-05 16:27:57 +00:00
|
|
|
model_cache_->InsertExprExprExpression(
|
2012-07-04 22:05:58 +00:00
|
|
|
boolvar,
|
2012-07-05 16:27:57 +00:00
|
|
|
left,
|
|
|
|
|
right,
|
|
|
|
|
ModelCache::EXPR_EXPR_IS_LESS_OR_EQUAL);
|
2012-07-04 22:05:58 +00:00
|
|
|
return boolvar;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
Constraint* Solver::MakeIsLessOrEqualCt(
|
|
|
|
|
IntExpr* const left, IntExpr* const right, IntVar* const b) {
|
|
|
|
|
CHECK_EQ(this, left->solver());
|
|
|
|
|
CHECK_EQ(this, right->solver());
|
|
|
|
|
if (left->Bound()) {
|
|
|
|
|
return MakeIsGreaterOrEqualCstCt(right->Var(), left->Min(), b);
|
|
|
|
|
} else if (right->Bound()) {
|
|
|
|
|
return MakeIsLessOrEqualCstCt(left->Var(), right->Min(), b);
|
|
|
|
|
}
|
2012-07-05 12:54:25 +00:00
|
|
|
return RevAlloc(new IsLessOrEqualCt(this, left->Var(), right->Var(), b));
|
2012-07-04 22:05:58 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
IntVar* Solver::MakeIsLessVar(
|
|
|
|
|
IntExpr* const left, IntExpr* const right) {
|
|
|
|
|
CHECK_EQ(this, left->solver());
|
|
|
|
|
CHECK_EQ(this, right->solver());
|
|
|
|
|
if (left->Bound()) {
|
|
|
|
|
return MakeIsGreaterCstVar(right->Var(), left->Min());
|
|
|
|
|
} else if (right->Bound()) {
|
|
|
|
|
return MakeIsLessCstVar(left->Var(), right->Min());
|
|
|
|
|
}
|
2012-07-05 16:27:57 +00:00
|
|
|
IntExpr* const cache = model_cache_->FindExprExprExpression(
|
|
|
|
|
left,
|
|
|
|
|
right,
|
|
|
|
|
ModelCache::EXPR_EXPR_IS_LESS);
|
2012-07-04 22:05:58 +00:00
|
|
|
if (cache != NULL) {
|
|
|
|
|
return cache->Var();
|
|
|
|
|
} else {
|
2012-07-05 16:27:57 +00:00
|
|
|
string name1 = left->name();
|
2012-07-04 22:05:58 +00:00
|
|
|
if (name1.empty()) {
|
2012-07-05 16:27:57 +00:00
|
|
|
name1 = left->DebugString();
|
2012-07-04 22:05:58 +00:00
|
|
|
}
|
2012-07-05 16:27:57 +00:00
|
|
|
string name2 = right->name();
|
2012-07-04 22:05:58 +00:00
|
|
|
if (name2.empty()) {
|
2012-07-05 16:27:57 +00:00
|
|
|
name2 = right->DebugString();
|
2012-07-04 22:05:58 +00:00
|
|
|
}
|
|
|
|
|
IntVar* const boolvar = MakeBoolVar(
|
|
|
|
|
StringPrintf("IsLessOrEqual(%s, %s)", name1.c_str(), name2.c_str()));
|
|
|
|
|
|
|
|
|
|
AddConstraint(RevAlloc(
|
|
|
|
|
new IsLessCt(this, left->Var(), right->Var(), boolvar)));
|
2012-07-05 16:27:57 +00:00
|
|
|
model_cache_->InsertExprExprExpression(
|
2012-07-04 22:05:58 +00:00
|
|
|
boolvar,
|
2012-07-05 16:27:57 +00:00
|
|
|
left,
|
|
|
|
|
right,
|
|
|
|
|
ModelCache::EXPR_EXPR_IS_LESS);
|
2012-07-04 22:05:58 +00:00
|
|
|
return boolvar;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
Constraint* Solver::MakeIsLessCt(
|
|
|
|
|
IntExpr* const left, IntExpr* const right, IntVar* const b) {
|
|
|
|
|
CHECK_EQ(this, left->solver());
|
|
|
|
|
CHECK_EQ(this, right->solver());
|
|
|
|
|
if (left->Bound()) {
|
|
|
|
|
return MakeIsGreaterCstCt(right->Var(), left->Min(), b);
|
|
|
|
|
} else if (right->Bound()) {
|
|
|
|
|
return MakeIsLessCstCt(left->Var(), right->Min(), b);
|
|
|
|
|
}
|
2012-07-05 12:54:25 +00:00
|
|
|
return RevAlloc(new IsLessCt(this, left->Var(), right->Var(), b));
|
2012-07-04 22:05:58 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
IntVar* Solver::MakeIsGreaterOrEqualVar(
|
|
|
|
|
IntExpr* const left, IntExpr* const right) {
|
|
|
|
|
return MakeIsLessOrEqualVar(right, left);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
Constraint* Solver::MakeIsGreaterOrEqualCt(
|
|
|
|
|
IntExpr* const left, IntExpr* const right, IntVar* const b) {
|
|
|
|
|
|
|
|
|
|
return MakeIsLessOrEqualCt(right, left, b);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
IntVar* Solver::MakeIsGreaterVar(
|
|
|
|
|
IntExpr* const left, IntExpr* const right) {
|
|
|
|
|
return MakeIsLessVar(right, left);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
Constraint* Solver::MakeIsGreaterCt(
|
|
|
|
|
IntExpr* const left, IntExpr* const right, IntVar* const b) {
|
|
|
|
|
return MakeIsLessCt(right, left, b);
|
|
|
|
|
}
|
|
|
|
|
|
2010-09-15 12:42:33 +00:00
|
|
|
} // namespace operations_research
|