31#include "absl/container/flat_hash_map.h"
32#include "absl/strings/str_cat.h"
33#include "absl/strings/str_format.h"
34#include "absl/strings/str_join.h"
56bool StartMinLessThan(Task*
const w1, Task*
const w2) {
57 return (w1->interval->StartMin() < w2->interval->StartMin());
64bool ShortestDurationStartMinLessThan(Task*
const w1, Task*
const w2) {
65 return w1->interval->EndMin() - w1->interval->DurationMin() <
66 w2->interval->EndMin() - w2->interval->DurationMin();
70bool StartMaxLessThan(Task*
const w1, Task*
const w2) {
71 return (w1->interval->StartMax() < w2->interval->StartMax());
75bool EndMinLessThan(Task*
const w1, Task*
const w2) {
76 return (w1->interval->EndMin() < w2->interval->EndMin());
80bool EndMaxLessThan(Task*
const w1, Task*
const w2) {
81 return (w1->interval->EndMax() < w2->interval->EndMax());
84bool IntervalStartMinLessThan(IntervalVar* i1, IntervalVar* i2) {
85 return i1->StartMin() < i2->StartMin();
94struct DisjunctiveTask {
95 explicit DisjunctiveTask(IntervalVar*
const interval_)
98 std::string DebugString()
const {
return interval->DebugString(); }
109struct CumulativeTask {
110 CumulativeTask(IntervalVar*
const interval_, int64_t demand_)
113 int64_t EnergyMin()
const {
return interval->DurationMin() *
demand; }
115 int64_t DemandMin()
const {
return demand; }
117 void WhenAnything(Demon*
const demon) {
interval->WhenAnything(demon); }
119 std::string DebugString()
const {
120 return absl::StrFormat(
"Task{ %s, demand: %d }",
interval->DebugString(),
135struct VariableCumulativeTask {
136 VariableCumulativeTask(IntervalVar*
const interval_, IntVar* demand_)
139 int64_t EnergyMin()
const {
return interval->DurationMin() *
demand->Min(); }
141 int64_t DemandMin()
const {
return demand->Min(); }
143 void WhenAnything(Demon*
const demon) {
148 std::string DebugString()
const {
149 return absl::StrFormat(
"Task{ %s, demand: %s }",
interval->DebugString(),
170 explicit ThetaNode(
const IntervalVar*
const interval)
183 void Compute(
const ThetaNode& left,
const ThetaNode& right) {
189 bool IsIdentity()
const {
194 std::string DebugString()
const {
210class ThetaTree :
public MonoidOperationTree<ThetaNode> {
212 explicit ThetaTree(
int size) : MonoidOperationTree<ThetaNode>(size) {}
214 int64_t Ect()
const {
return result().total_ect; }
216 void Insert(
const DisjunctiveTask*
const task) {
217 Set(task->index, ThetaNode(task->interval));
220 void Remove(
const DisjunctiveTask*
const task) { Reset(task->index); }
222 bool IsInserted(
const DisjunctiveTask*
const task)
const {
223 return !GetOperand(task->index).IsIdentity();
233struct LambdaThetaNode {
247 LambdaThetaNode(int64_t
capacity,
const CumulativeTask& task)
248 :
energy(task.EnergyMin()),
256 LambdaThetaNode(int64_t
capacity,
const CumulativeTask& task,
int index)
268 LambdaThetaNode(int64_t
capacity,
const VariableCumulativeTask& task)
269 :
energy(task.EnergyMin()),
277 LambdaThetaNode(int64_t
capacity,
const VariableCumulativeTask& task,
290 explicit LambdaThetaNode(
const IntervalVar*
const interval)
300 LambdaThetaNode(
const IntervalVar*
const interval,
int index)
317 void Compute(
const LambdaThetaNode& left,
const LambdaThetaNode& right) {
320 CapAdd(left.energetic_end_min, right.energy));
321 const int64_t energy_left_opt =
CapAdd(left.energy_opt, right.energy);
322 const int64_t energy_right_opt =
CapAdd(left.energy, right.energy_opt);
323 if (energy_left_opt > energy_right_opt) {
330 const int64_t ect1 = right.energetic_end_min_opt;
331 const int64_t ect2 =
CapAdd(left.energetic_end_min, right.energy_opt);
332 const int64_t ect3 =
CapAdd(left.energetic_end_min_opt, right.energy);
333 if (ect1 >= ect2 && ect1 >= ect3) {
336 }
else if (ect2 >= ect1 && ect2 >= ect3) {
378class DisjunctiveLambdaThetaTree :
public MonoidOperationTree<LambdaThetaNode> {
380 explicit DisjunctiveLambdaThetaTree(
int size)
381 : MonoidOperationTree<LambdaThetaNode>(size) {}
383 void Insert(
const DisjunctiveTask& task) {
384 Set(task.index, LambdaThetaNode(task.interval));
387 void Grey(
const DisjunctiveTask& task) {
388 const int index = task.index;
389 Set(
index, LambdaThetaNode(task.interval,
index));
392 int64_t Ect()
const {
return result().energetic_end_min; }
393 int64_t EctOpt()
const {
return result().energetic_end_min_opt; }
394 int ResponsibleOpt()
const {
return result().argmax_energetic_end_min_opt; }
398class CumulativeLambdaThetaTree :
public MonoidOperationTree<LambdaThetaNode> {
400 CumulativeLambdaThetaTree(
int size, int64_t capacity_max)
401 : MonoidOperationTree<LambdaThetaNode>(size),
402 capacity_max_(capacity_max) {}
404 void Init(int64_t capacity_max) {
406 capacity_max_ = capacity_max;
409 void Insert(
const CumulativeTask& task) {
410 Set(task.index, LambdaThetaNode(capacity_max_, task));
413 void Grey(
const CumulativeTask& task) {
414 const int index = task.index;
415 Set(
index, LambdaThetaNode(capacity_max_, task,
index));
418 void Insert(
const VariableCumulativeTask& task) {
419 Set(task.index, LambdaThetaNode(capacity_max_, task));
422 void Grey(
const VariableCumulativeTask& task) {
423 const int index = task.index;
424 Set(
index, LambdaThetaNode(capacity_max_, task,
index));
429 return result().energetic_end_min_opt;
431 int64_t Ect()
const {
434 int64_t EctOpt()
const {
438 return result().argmax_energetic_end_min_opt;
442 int64_t capacity_max_;
451 NotLast(Solver*
const solver,
const std::vector<IntervalVar*>& intervals,
452 bool mirror,
bool strict);
459 ThetaTree theta_tree_;
460 std::vector<DisjunctiveTask*> by_start_min_;
461 std::vector<DisjunctiveTask*> by_end_max_;
462 std::vector<DisjunctiveTask*> by_start_max_;
463 std::vector<int64_t> new_lct_;
467NotLast::NotLast(Solver*
const solver,
468 const std::vector<IntervalVar*>& intervals,
bool mirror,
470 : theta_tree_(intervals.size()),
471 by_start_min_(intervals.size()),
472 by_end_max_(intervals.size()),
473 by_start_max_(intervals.size()),
474 new_lct_(intervals.size(), -1LL),
477 for (
int i = 0; i < intervals.size(); ++i) {
478 IntervalVar*
const underlying =
479 mirror ? solver->MakeMirrorInterval(intervals[i]) : intervals[i];
480 IntervalVar*
const relaxed = solver->MakeIntervalRelaxedMin(underlying);
481 by_start_min_[i] =
new DisjunctiveTask(relaxed);
482 by_end_max_[i] = by_start_min_[i];
483 by_start_max_[i] = by_start_min_[i];
487bool NotLast::Propagate() {
489 std::sort(by_start_max_.begin(), by_start_max_.end(),
490 StartMaxLessThan<DisjunctiveTask>);
491 std::sort(by_end_max_.begin(), by_end_max_.end(),
492 EndMaxLessThan<DisjunctiveTask>);
494 std::sort(by_start_min_.begin(), by_start_min_.end(),
495 StartMinLessThan<DisjunctiveTask>);
496 for (
int i = 0; i < by_start_min_.size(); ++i) {
497 by_start_min_[i]->index = i;
500 for (
int i = 0; i < by_start_min_.size(); ++i) {
501 new_lct_[i] = by_start_min_[i]->interval->EndMax();
506 for (DisjunctiveTask*
const twi : by_end_max_) {
507 while (j < by_start_max_.size() &&
508 twi->interval->EndMax() > by_start_max_[j]->interval->StartMax()) {
509 if (j > 0 && theta_tree_.Ect() > by_start_max_[j]->interval->StartMax()) {
510 const int64_t new_end_max = by_start_max_[j - 1]->interval->StartMax();
511 new_lct_[by_start_max_[j]->index] =
514 theta_tree_.Insert(by_start_max_[j]);
517 const bool inserted = theta_tree_.IsInserted(twi);
519 theta_tree_.Remove(twi);
521 const int64_t ect_theta_less_i = theta_tree_.Ect();
523 theta_tree_.Insert(twi);
526 if (ect_theta_less_i > twi->interval->StartMax() && j > 0) {
527 const int64_t new_end_max = by_start_max_[j - 1]->interval->StartMax();
528 if (new_end_max < new_lct_[twi->index]) {
529 new_lct_[twi->index] = new_end_max;
535 bool modified =
false;
536 for (
int i = 0; i < by_start_min_.size(); ++i) {
537 IntervalVar*
const var = by_start_min_[i]->interval;
538 if ((strict_ ||
var->DurationMin() > 0) &&
var->EndMax() > new_lct_[i]) {
540 var->SetEndMax(new_lct_[i]);
551class EdgeFinderAndDetectablePrecedences {
553 EdgeFinderAndDetectablePrecedences(Solver*
const solver,
554 const std::vector<IntervalVar*>& intervals,
555 bool mirror,
bool strict);
556 ~EdgeFinderAndDetectablePrecedences() {
559 int64_t size()
const {
return by_start_min_.size(); }
562 void OverloadChecking();
563 bool DetectablePrecedences();
567 Solver*
const solver_;
578 ThetaTree theta_tree_;
579 std::vector<DisjunctiveTask*> by_end_min_;
580 std::vector<DisjunctiveTask*> by_start_min_;
581 std::vector<DisjunctiveTask*> by_end_max_;
582 std::vector<DisjunctiveTask*> by_start_max_;
584 std::vector<int64_t> new_est_;
586 std::vector<int64_t> new_lct_;
587 DisjunctiveLambdaThetaTree lt_tree_;
591EdgeFinderAndDetectablePrecedences::EdgeFinderAndDetectablePrecedences(
592 Solver*
const solver,
const std::vector<IntervalVar*>& intervals,
593 bool mirror,
bool strict)
595 theta_tree_(intervals.size()),
596 lt_tree_(intervals.size()),
599 for (IntervalVar*
const interval : intervals) {
600 IntervalVar*
const underlying =
602 IntervalVar*
const relaxed = solver->MakeIntervalRelaxedMax(underlying);
603 DisjunctiveTask*
const task =
new DisjunctiveTask(relaxed);
604 by_end_min_.push_back(task);
605 by_start_min_.push_back(task);
606 by_end_max_.push_back(task);
607 by_start_max_.push_back(task);
612void EdgeFinderAndDetectablePrecedences::UpdateEst() {
613 std::sort(by_start_min_.begin(), by_start_min_.end(),
614 ShortestDurationStartMinLessThan<DisjunctiveTask>);
615 for (
int i = 0; i < size(); ++i) {
616 by_start_min_[i]->index = i;
620void EdgeFinderAndDetectablePrecedences::OverloadChecking() {
623 std::sort(by_end_max_.begin(), by_end_max_.end(),
624 EndMaxLessThan<DisjunctiveTask>);
627 for (DisjunctiveTask*
const task : by_end_max_) {
628 theta_tree_.Insert(task);
629 if (theta_tree_.Ect() > task->interval->EndMax()) {
635bool EdgeFinderAndDetectablePrecedences::DetectablePrecedences() {
641 std::sort(by_end_min_.begin(), by_end_min_.end(),
642 EndMinLessThan<DisjunctiveTask>);
643 std::sort(by_start_max_.begin(), by_start_max_.end(),
644 StartMaxLessThan<DisjunctiveTask>);
647 for (DisjunctiveTask*
const task_i : by_end_min_) {
649 DisjunctiveTask* task_j = by_start_max_[j];
650 while (task_i->interval->EndMin() > task_j->interval->StartMax()) {
651 theta_tree_.Insert(task_j);
653 if (j == size())
break;
654 task_j = by_start_max_[j];
657 const int64_t esti = task_i->interval->StartMin();
658 bool inserted = theta_tree_.IsInserted(task_i);
660 theta_tree_.Remove(task_i);
662 const int64_t oesti = theta_tree_.Ect();
664 theta_tree_.Insert(task_i);
667 new_est_[task_i->index] = oesti;
674 bool modified =
false;
675 for (
int i = 0; i < size(); ++i) {
676 IntervalVar*
const var = by_start_min_[i]->interval;
678 (strict_ ||
var->DurationMin() > 0)) {
680 by_start_min_[i]->interval->SetStartMin(new_est_[i]);
686bool EdgeFinderAndDetectablePrecedences::EdgeFinder() {
689 for (
int i = 0; i < size(); ++i) {
690 new_est_[i] = by_start_min_[i]->interval->StartMin();
694 std::sort(by_end_max_.begin(), by_end_max_.end(),
695 EndMaxLessThan<DisjunctiveTask>);
697 for (
int i = 0; i < size(); ++i) {
698 lt_tree_.Insert(*by_start_min_[i]);
701 for (
int j = size() - 2; j >= 0; --j) {
702 lt_tree_.Grey(*by_end_max_[j + 1]);
703 DisjunctiveTask*
const twj = by_end_max_[j];
705 DCHECK_LE(lt_tree_.Ect(), twj->interval->EndMax());
706 while (lt_tree_.EctOpt() > twj->interval->EndMax()) {
707 const int i = lt_tree_.ResponsibleOpt();
709 if (lt_tree_.Ect() > new_est_[i]) {
710 new_est_[i] = lt_tree_.Ect();
717 bool modified =
false;
718 for (
int i = 0; i < size(); ++i) {
719 IntervalVar*
const var = by_start_min_[i]->interval;
720 if (
var->StartMin() < new_est_[i] && (strict_ ||
var->DurationMin() > 0)) {
722 var->SetStartMin(new_est_[i]);
732class RankedPropagator :
public Constraint {
734 RankedPropagator(Solver*
const solver,
const std::vector<IntVar*>& nexts,
735 const std::vector<IntervalVar*>& intervals,
736 const std::vector<IntVar*>& slacks,
737 DisjunctiveConstraint*
const disjunctive)
738 : Constraint(solver),
740 intervals_(intervals),
742 disjunctive_(disjunctive),
743 partial_sequence_(intervals.size()),
744 previous_(intervals.size() + 2, 0) {}
746 ~RankedPropagator()
override {}
748 void Post()
override {
749 Demon*
const delayed =
750 solver()->MakeDelayedConstraintInitialPropagateCallback(
this);
751 for (
int i = 0; i < intervals_.size(); ++i) {
752 nexts_[i]->WhenBound(delayed);
753 intervals_[i]->WhenAnything(delayed);
754 slacks_[i]->WhenRange(delayed);
756 nexts_.back()->WhenBound(delayed);
759 void InitialPropagate()
override {
764 void PropagateNexts() {
765 Solver*
const s = solver();
766 const int ranked_first = partial_sequence_.NumFirstRanked();
767 const int ranked_last = partial_sequence_.NumLastRanked();
771 : partial_sequence_[intervals_.size() - ranked_last] + 1;
774 while (nexts_[first]->Bound()) {
776 first = nexts_[first]->Min();
777 if (first == sentinel) {
780 if (++counter > ranked_first) {
781 DCHECK(intervals_[first - 1]->MayBePerformed());
782 partial_sequence_.RankFirst(s, first - 1);
783 VLOG(2) <<
"RankFirst " << first - 1 <<
" -> "
784 << partial_sequence_.DebugString();
787 previous_.assign(previous_.size(), -1);
788 for (
int i = 0; i < nexts_.size(); ++i) {
789 if (nexts_[i]->Bound()) {
790 previous_[nexts_[i]->Min()] = i;
793 int last = previous_.size() - 1;
795 while (previous_[last] != -1) {
796 last = previous_[last];
797 if (++counter > ranked_last) {
798 partial_sequence_.RankLast(s, last - 1);
799 VLOG(2) <<
"RankLast " << last - 1 <<
" -> "
800 << partial_sequence_.DebugString();
805 void PropagateSequence() {
806 const int last_position = intervals_.size() - 1;
807 const int first_sentinel = partial_sequence_.NumFirstRanked();
808 const int last_sentinel = last_position - partial_sequence_.NumLastRanked();
810 for (
int i = 0; i < first_sentinel - 1; ++i) {
811 IntervalVar*
const interval = RankedInterval(i);
812 IntervalVar*
const next_interval = RankedInterval(i + 1);
813 IntVar*
const slack = RankedSlack(i);
814 const int64_t transition_time = RankedTransitionTime(i, i + 1);
815 next_interval->SetStartRange(
820 for (
int i = last_position; i > last_sentinel + 1; --i) {
821 IntervalVar*
const interval = RankedInterval(i - 1);
822 IntervalVar*
const next_interval = RankedInterval(i);
823 IntVar*
const slack = RankedSlack(i - 1);
824 const int64_t transition_time = RankedTransitionTime(i - 1, i);
826 CapAdd(slack->Max(), transition_time)),
827 CapSub(next_interval->StartMax(),
828 CapAdd(slack->Min(), transition_time)));
831 IntervalVar*
const first_interval =
832 first_sentinel > 0 ? RankedInterval(first_sentinel - 1) : nullptr;
833 IntVar*
const first_slack =
834 first_sentinel > 0 ? RankedSlack(first_sentinel - 1) : nullptr;
835 IntervalVar*
const last_interval = last_sentinel < last_position
836 ? RankedInterval(last_sentinel + 1)
840 if (first_interval ==
nullptr && last_interval ==
nullptr) {
845 for (
int i = first_sentinel; i <= last_sentinel; ++i) {
846 IntervalVar*
const interval = RankedInterval(i);
847 IntVar*
const slack = RankedSlack(i);
850 if (first_interval !=
nullptr) {
851 const int64_t transition_time =
852 RankedTransitionTime(first_sentinel - 1, i);
854 CapAdd(first_interval->StartMin(),
855 CapAdd(first_slack->Min(), transition_time)),
856 CapAdd(first_interval->StartMax(),
857 CapAdd(first_slack->Max(), transition_time)));
859 first_interval->SetStartRange(
861 CapAdd(first_slack->Max(), transition_time)),
863 CapAdd(first_slack->Min(), transition_time)));
866 if (last_interval !=
nullptr) {
867 const int64_t transition_time =
868 RankedTransitionTime(i, last_sentinel + 1);
870 CapSub(last_interval->StartMin(),
871 CapAdd(slack->Max(), transition_time)),
872 CapSub(last_interval->StartMax(),
873 CapAdd(slack->Min(), transition_time)));
875 last_interval->SetStartRange(
877 CapAdd(slack->Min(), transition_time)),
879 CapAdd(slack->Max(), transition_time)));
886 for (
int i =
std::min(first_sentinel - 2, last_position - 1); i >= 0; --i) {
887 IntervalVar*
const interval = RankedInterval(i);
888 IntervalVar*
const next_interval = RankedInterval(i + 1);
889 IntVar*
const slack = RankedSlack(i);
890 const int64_t transition_time = RankedTransitionTime(i, i + 1);
892 CapAdd(slack->Max(), transition_time)),
893 CapSub(next_interval->StartMax(),
894 CapAdd(slack->Min(), transition_time)));
897 for (
int i = last_sentinel + 1; i < last_position - 1; ++i) {
898 IntervalVar*
const interval = RankedInterval(i);
899 IntervalVar*
const next_interval = RankedInterval(i + 1);
900 IntVar*
const slack = RankedSlack(i);
901 const int64_t transition_time = RankedTransitionTime(i, i + 1);
902 next_interval->SetStartRange(
909 IntervalVar* RankedInterval(
int i)
const {
910 const int index = partial_sequence_[i];
911 return intervals_[
index];
914 IntVar* RankedSlack(
int i)
const {
915 const int index = partial_sequence_[i];
916 return slacks_[
index];
919 int64_t RankedTransitionTime(
int before,
int after)
const {
920 const int before_index = partial_sequence_[before];
921 const int after_index = partial_sequence_[after];
923 return disjunctive_->TransitionTime(before_index, after_index);
926 std::string DebugString()
const override {
927 return absl::StrFormat(
928 "RankedPropagator([%s], nexts = [%s], intervals = [%s])",
933 void Accept(ModelVisitor*
const visitor)
const override {
934 LOG(
FATAL) <<
"Not yet implemented";
939 std::vector<IntVar*> nexts_;
940 std::vector<IntervalVar*> intervals_;
941 std::vector<IntVar*> slacks_;
942 DisjunctiveConstraint*
const disjunctive_;
943 RevPartialSequence partial_sequence_;
944 std::vector<int> previous_;
950class FullDisjunctiveConstraint :
public DisjunctiveConstraint {
952 FullDisjunctiveConstraint(Solver*
const s,
953 const std::vector<IntervalVar*>& intervals,
954 const std::string&
name,
bool strict)
955 : DisjunctiveConstraint(s, intervals,
name),
956 sequence_var_(nullptr),
957 straight_(s, intervals, false, strict),
958 mirror_(s, intervals, true, strict),
959 straight_not_last_(s, intervals, false, strict),
960 mirror_not_last_(s, intervals, true, strict),
963 ~FullDisjunctiveConstraint()
override {}
965 void Post()
override {
967 solver(),
this, &FullDisjunctiveConstraint::InitialPropagate,
969 for (int32_t i = 0; i < straight_.size(); ++i) {
970 straight_.interval(i)->WhenAnything(d);
974 void InitialPropagate()
override {
975 bool all_optional_or_unperformed =
true;
976 for (
const IntervalVar*
const interval : intervals_) {
978 all_optional_or_unperformed =
false;
982 if (all_optional_or_unperformed) {
986 bool all_times_fixed =
true;
987 for (
const IntervalVar*
const interval : intervals_) {
992 all_times_fixed =
false;
997 if (all_times_fixed) {
998 PropagatePerformed();
1005 straight_.OverloadChecking();
1006 }
while (straight_.DetectablePrecedences() ||
1007 mirror_.DetectablePrecedences());
1008 }
while (straight_not_last_.Propagate() ||
1009 mirror_not_last_.Propagate());
1010 }
while (straight_.EdgeFinder() || mirror_.EdgeFinder());
1014 bool Intersect(IntervalVar*
const i1, IntervalVar*
const i2)
const {
1015 return i1->StartMin() < i2->EndMax() && i2->StartMin() < i1->EndMax();
1018 void PropagatePerformed() {
1021 for (IntervalVar*
const interval : intervals_) {
1024 }
else if (
interval->MayBePerformed()) {
1029 if (performed_.empty())
return;
1030 std::sort(performed_.begin(), performed_.end(), IntervalStartMinLessThan);
1031 for (
int i = 0; i < performed_.size() - 1; ++i) {
1032 if (performed_[i]->EndMax() > performed_[i + 1]->StartMin()) {
1038 if (optional_.empty())
return;
1040 const int num_performed = performed_.size();
1041 std::sort(optional_.begin(), optional_.end(), IntervalStartMinLessThan);
1042 for (IntervalVar*
const candidate : optional_) {
1043 const int64_t
start = candidate->StartMin();
1044 while (index < num_performed && start >= performed_[
index]->EndMax()) {
1047 if (
index == num_performed)
return;
1048 if (Intersect(candidate, performed_[
index]) ||
1049 (
index < num_performed - 1 &&
1050 Intersect(candidate, performed_[
index + 1]))) {
1051 candidate->SetPerformed(
false);
1056 void Accept(ModelVisitor*
const visitor)
const override {
1057 visitor->BeginVisitConstraint(ModelVisitor::kDisjunctive,
this);
1058 visitor->VisitIntervalArrayArgument(ModelVisitor::kIntervalsArgument,
1060 if (sequence_var_ !=
nullptr) {
1061 visitor->VisitSequenceArgument(ModelVisitor::kSequenceArgument,
1064 visitor->EndVisitConstraint(ModelVisitor::kDisjunctive,
this);
1067 SequenceVar* MakeSequenceVar()
override {
1068 BuildNextModelIfNeeded();
1069 if (sequence_var_ ==
nullptr) {
1070 solver()->SaveValue(
reinterpret_cast<void**
>(&sequence_var_));
1071 sequence_var_ = solver()->RevAlloc(
1072 new SequenceVar(solver(), intervals_, nexts_,
name()));
1074 return sequence_var_;
1077 std::string DebugString()
const override {
1078 return absl::StrFormat(
"FullDisjunctiveConstraint([%s], %i)",
1082 const std::vector<IntVar*>& nexts()
const override {
return nexts_; }
1084 const std::vector<IntVar*>& actives()
const override {
return actives_; }
1086 const std::vector<IntVar*>& time_cumuls()
const override {
1087 return time_cumuls_;
1090 const std::vector<IntVar*>& time_slacks()
const override {
1091 return time_slacks_;
1095 int64_t
Distance(int64_t activity_plus_one, int64_t next_activity_plus_one) {
1096 return (activity_plus_one == 0 ||
1097 next_activity_plus_one > intervals_.size())
1099 : transition_time_(activity_plus_one - 1,
1100 next_activity_plus_one - 1);
1103 void BuildNextModelIfNeeded() {
1104 if (!nexts_.empty()) {
1107 Solver*
const s = solver();
1108 const std::string& ct_name =
name();
1109 const int num_intervals = intervals_.size();
1110 const int num_nodes = intervals_.size() + 1;
1111 int64_t horizon = 0;
1112 for (
int i = 0; i < intervals_.size(); ++i) {
1113 if (intervals_[i]->MayBePerformed()) {
1114 horizon =
std::max(horizon, intervals_[i]->EndMax());
1119 s->MakeIntVarArray(num_nodes, 1, num_nodes, ct_name +
"_nexts", &nexts_);
1121 s->AddConstraint(s->MakeAllDifferent(nexts_));
1123 actives_.resize(num_nodes);
1124 for (
int i = 0; i < num_intervals; ++i) {
1125 actives_[i + 1] = intervals_[i]->PerformedExpr()->Var();
1127 s->MakeIsDifferentCstCt(nexts_[i + 1], i + 1, actives_[i + 1]));
1129 std::vector<IntVar*> short_actives(actives_.begin() + 1, actives_.end());
1130 actives_[0] = s->MakeMax(short_actives)->Var();
1133 s->AddConstraint(s->MakeNoCycle(nexts_, actives_));
1136 time_cumuls_.resize(num_nodes + 1);
1138 time_slacks_.resize(num_nodes);
1140 time_slacks_[0] = s->MakeIntVar(0, horizon,
"initial_slack");
1142 time_cumuls_[0] = s->MakeIntConst(0);
1144 for (int64_t i = 0; i < num_intervals; ++i) {
1145 IntervalVar*
const var = intervals_[i];
1146 if (
var->MayBePerformed()) {
1147 const int64_t duration_min =
var->DurationMin();
1148 time_slacks_[i + 1] = s->MakeIntVar(
1149 duration_min, horizon, absl::StrFormat(
"time_slacks(%d)", i + 1));
1151 time_cumuls_[i + 1] =
var->SafeStartExpr(
var->StartMin())->Var();
1152 if (
var->DurationMax() != duration_min) {
1153 s->AddConstraint(s->MakeGreaterOrEqual(
1154 time_slacks_[i + 1],
var->SafeDurationExpr(duration_min)));
1157 time_slacks_[i + 1] = s->MakeIntVar(
1158 0, horizon, absl::StrFormat(
"time_slacks(%d)", i + 1));
1159 time_cumuls_[i + 1] = s->MakeIntConst(horizon);
1163 time_cumuls_[num_nodes] = s->MakeIntVar(0, 2 * horizon, ct_name +
"_ect");
1164 s->AddConstraint(s->MakePathCumul(
1165 nexts_, actives_, time_cumuls_, time_slacks_,
1166 [
this](int64_t x, int64_t y) { return Distance(x, y); }));
1168 std::vector<IntVar*> short_slacks(time_slacks_.begin() + 1,
1169 time_slacks_.end());
1170 s->AddConstraint(s->RevAlloc(
1171 new RankedPropagator(s, nexts_, intervals_, short_slacks,
this)));
1174 SequenceVar* sequence_var_;
1175 EdgeFinderAndDetectablePrecedences straight_;
1176 EdgeFinderAndDetectablePrecedences mirror_;
1177 NotLast straight_not_last_;
1178 NotLast mirror_not_last_;
1179 std::vector<IntVar*> nexts_;
1180 std::vector<IntVar*> actives_;
1181 std::vector<IntVar*> time_cumuls_;
1182 std::vector<IntVar*> time_slacks_;
1183 std::vector<IntervalVar*> performed_;
1184 std::vector<IntervalVar*> optional_;
1195struct DualCapacityThetaNode {
1197 static const int kNone;
1200 DualCapacityThetaNode()
1206 DualCapacityThetaNode(int64_t
capacity, int64_t residual_capacity,
1207 const CumulativeTask& task)
1208 :
energy(task.EnergyMin()),
1214 DualCapacityThetaNode(int64_t
capacity, int64_t residual_capacity,
1215 const VariableCumulativeTask& task)
1216 :
energy(task.EnergyMin()),
1227 void Compute(
const DualCapacityThetaNode& left,
1228 const DualCapacityThetaNode& right) {
1231 right.energetic_end_min);
1234 right.residual_energetic_end_min);
1251class DualCapacityThetaTree
1252 :
public MonoidOperationTree<DualCapacityThetaNode> {
1256 explicit DualCapacityThetaTree(
int size)
1257 : MonoidOperationTree<DualCapacityThetaNode>(size),
1259 residual_capacity_(-1) {}
1261 virtual ~DualCapacityThetaTree() {}
1263 void Init(int64_t capacity_max, int64_t residual_capacity) {
1265 DCHECK_LE(residual_capacity, capacity_max);
1267 capacity_max_ = capacity_max;
1268 residual_capacity_ = residual_capacity;
1271 void Insert(
const CumulativeTask* task) {
1273 DualCapacityThetaNode(capacity_max_, residual_capacity_, *task));
1276 void Insert(
const VariableCumulativeTask* task) {
1278 DualCapacityThetaNode(capacity_max_, residual_capacity_, *task));
1282 int64_t capacity_max_;
1283 int64_t residual_capacity_;
1299class EnvJCComputeDiver {
1302 explicit EnvJCComputeDiver(
int energy_threshold)
1303 : energy_threshold_(energy_threshold),
1306 void OnArgumentReached(
int index,
const DualCapacityThetaNode& argument) {
1307 energy_alpha_ = argument.energy;
1308 energetic_end_min_alpha_ = argument.energetic_end_min;
1313 bool ChooseGoLeft(
const DualCapacityThetaNode& current,
1314 const DualCapacityThetaNode& left_child,
1316 if (
right_child.residual_energetic_end_min > energy_threshold_) {
1323 void OnComeBackFromLeft(
const DualCapacityThetaNode& current,
1324 const DualCapacityThetaNode& left_child,
1331 void OnComeBackFromRight(
const DualCapacityThetaNode& current,
1332 const DualCapacityThetaNode& left_child,
1336 energetic_end_min_alpha_ =
1338 CapAdd(left_child.energetic_end_min, energy_alpha_));
1339 energy_alpha_ += left_child.energy;
1341 int64_t GetEnvJC(
const DualCapacityThetaNode& root)
const {
1342 const int64_t
energy = root.energy;
1343 const int64_t energy_beta =
CapSub(
energy, energy_alpha_);
1344 return CapAdd(energetic_end_min_alpha_, energy_beta);
1353 int64_t energy_threshold_;
1359 int64_t energy_alpha_;
1364 int64_t energetic_end_min_alpha_;
1376class UpdatesForADemand {
1378 explicit UpdatesForADemand(
int size)
1379 : updates_(size, 0), up_to_date_(false) {}
1381 const int64_t Update(
int index) {
return updates_[
index]; }
1382 void Reset() { up_to_date_ =
false; }
1383 void SetUpdate(
int index, int64_t update) {
1386 updates_[
index] = update;
1388 bool up_to_date()
const {
return up_to_date_; }
1389 void set_up_to_date() { up_to_date_ =
true; }
1392 std::vector<int64_t> updates_;
1398template <
class Task>
1399class EdgeFinder :
public Constraint {
1401 EdgeFinder(Solver*
const solver,
const std::vector<Task*>& tasks,
1403 : Constraint(solver),
1406 by_start_min_(tasks.size()),
1407 by_end_max_(tasks.size()),
1408 by_end_min_(tasks.size()),
1409 lt_tree_(tasks.size(), capacity_->Max()),
1410 dual_capacity_tree_(tasks.size()),
1411 has_zero_demand_tasks_(true) {}
1413 ~EdgeFinder()
override {
1418 void Post()
override {
1421 solver(),
this, &EdgeFinder::InitialPropagate,
"RangeChanged");
1422 for (Task*
const task : tasks_) {
1425 task->WhenAnything(demon);
1427 capacity_->WhenRange(demon);
1432 void InitialPropagate()
override {
1434 PropagateBasedOnEndMinGreaterThanEndMax();
1436 PropagateBasedOnEnergy();
1440 void Accept(ModelVisitor*
const visitor)
const override {
1441 LOG(
FATAL) <<
"Should Not Be Visited";
1444 std::string DebugString()
const override {
return "EdgeFinder"; }
1447 UpdatesForADemand* GetOrMakeUpdate(int64_t demand_min) {
1449 if (update ==
nullptr) {
1450 update =
new UpdatesForADemand(tasks_.size());
1451 update_map_[demand_min] = update;
1457 void InitPropagation() {
1459 start_min_update_.clear();
1461 if (has_zero_demand_tasks_.Value()) {
1462 by_start_min_.clear();
1463 by_end_min_.clear();
1464 by_end_max_.clear();
1466 bool zero_demand =
false;
1467 for (Task*
const task : tasks_) {
1468 if (task->DemandMin() > 0) {
1469 by_start_min_.push_back(task);
1470 by_end_min_.push_back(task);
1471 by_end_max_.push_back(task);
1477 has_zero_demand_tasks_.SetValue(solver(),
false);
1482 std::sort(by_start_min_.begin(), by_start_min_.end(),
1483 StartMinLessThan<Task>);
1484 for (
int i = 0; i < by_start_min_.size(); ++i) {
1485 by_start_min_[i]->index = i;
1488 std::sort(by_end_max_.begin(), by_end_max_.end(), EndMaxLessThan<Task>);
1490 std::sort(by_end_min_.begin(), by_end_min_.end(), EndMinLessThan<Task>);
1492 lt_tree_.Init(capacity_->Max());
1494 for (
const auto& entry : update_map_) {
1495 entry.second->Reset();
1503 void ComputeConditionalStartMins(UpdatesForADemand* updates,
1504 int64_t demand_min) {
1506 DCHECK(updates !=
nullptr);
1507 const int64_t capacity_max = capacity_->Max();
1508 const int64_t residual_capacity =
CapSub(capacity_max, demand_min);
1509 dual_capacity_tree_.Init(capacity_max, residual_capacity);
1514 int64_t update = IntervalVar::kMinValidValue;
1515 for (
int i = 0; i < by_end_max_.size(); ++i) {
1516 Task*
const task = by_end_max_[i];
1517 if (task->EnergyMin() == 0)
continue;
1518 const int64_t current_end_max = task->interval->EndMax();
1519 dual_capacity_tree_.Insert(task);
1520 const int64_t energy_threshold = residual_capacity * current_end_max;
1521 const DualCapacityThetaNode& root = dual_capacity_tree_.result();
1522 const int64_t res_energetic_end_min = root.residual_energetic_end_min;
1523 if (res_energetic_end_min > energy_threshold) {
1524 EnvJCComputeDiver diver(energy_threshold);
1525 dual_capacity_tree_.DiveInTree(&diver);
1526 const int64_t enjv = diver.GetEnvJC(dual_capacity_tree_.result());
1527 const int64_t numerator =
CapSub(enjv, energy_threshold);
1528 const int64_t diff = MathUtil::CeilOfRatio(numerator, demand_min);
1531 updates->SetUpdate(i, update);
1533 updates->set_up_to_date();
1538 int64_t ConditionalStartMin(
const Task& task_to_push,
int end_max_index) {
1539 if (task_to_push.EnergyMin() == 0) {
1540 return task_to_push.interval->StartMin();
1542 const int64_t demand_min = task_to_push.DemandMin();
1543 UpdatesForADemand*
const updates = GetOrMakeUpdate(demand_min);
1544 if (!updates->up_to_date()) {
1545 ComputeConditionalStartMins(updates, demand_min);
1547 DCHECK(updates->up_to_date());
1548 return updates->Update(end_max_index);
1556 void PropagateBasedOnEndMinGreaterThanEndMax() {
1557 int end_max_index = 0;
1559 for (Task*
const task : by_end_min_) {
1560 const int64_t
end_min = task->interval->EndMin();
1561 while (end_max_index < by_start_min_.size() &&
1562 by_end_max_[end_max_index]->interval->EndMax() <=
end_min) {
1564 max_start_min, by_end_max_[end_max_index]->
interval->StartMin());
1567 if (end_max_index > 0 && task->interval->StartMin() <= max_start_min &&
1568 task->interval->EndMax() > task->interval->EndMin()) {
1582 const int64_t update = ConditionalStartMin(*task, end_max_index - 1);
1583 start_min_update_.push_back(std::make_pair(task->interval, update));
1590 for (Task*
const task : by_end_max_) {
1591 lt_tree_.Insert(*task);
1593 const int64_t max_feasible =
1594 CapProd(capacity_->Max(), task->interval->EndMax());
1595 if (lt_tree_.energetic_end_min() > max_feasible) {
1603 void PropagateBasedOnEnergy() {
1604 for (
int j = by_start_min_.size() - 2; j >= 0; --j) {
1605 lt_tree_.Grey(*by_end_max_[j + 1]);
1606 Task*
const twj = by_end_max_[j];
1608 const int64_t max_feasible =
1609 CapProd(capacity_->Max(), twj->interval->EndMax());
1610 DCHECK_LE(lt_tree_.energetic_end_min(), max_feasible);
1611 while (lt_tree_.energetic_end_min_opt() > max_feasible) {
1612 const int i = lt_tree_.argmax_energetic_end_min_opt();
1614 PropagateTaskCannotEndBefore(i, j);
1622 void PropagateTaskCannotEndBefore(
int index,
int end_max_index) {
1623 Task*
const task_to_push = by_start_min_[
index];
1624 const int64_t update = ConditionalStartMin(*task_to_push, end_max_index);
1625 start_min_update_.push_back(std::make_pair(task_to_push->interval, update));
1629 void ApplyNewBounds() {
1630 for (
const std::pair<IntervalVar*, int64_t>& update : start_min_update_) {
1631 update.first->SetStartMin(update.second);
1636 IntVar*
const capacity_;
1639 std::vector<Task*> tasks_;
1642 std::vector<Task*> by_start_min_;
1645 std::vector<Task*> by_end_max_;
1648 std::vector<Task*> by_end_min_;
1651 CumulativeLambdaThetaTree lt_tree_;
1654 DualCapacityThetaTree dual_capacity_tree_;
1657 std::vector<std::pair<IntervalVar*, int64_t>> start_min_update_;
1662 absl::flat_hash_map<int64_t, UpdatesForADemand*> update_map_;
1665 Rev<bool> has_zero_demand_tasks_;
1691struct ProfileDelta {
1692 ProfileDelta(int64_t _time, int64_t _delta) :
time(_time),
delta(_delta) {}
1697bool TimeLessThan(
const ProfileDelta& delta1,
const ProfileDelta& delta2) {
1698 return delta1.time < delta2.time;
1713template <
class Task>
1714class CumulativeTimeTable :
public Constraint {
1716 CumulativeTimeTable(Solver*
const solver,
const std::vector<Task*>& tasks,
1718 : Constraint(solver), by_start_min_(tasks), capacity_(
capacity) {
1721 const int profile_max_size = 2 * by_start_min_.size() + 2;
1722 profile_non_unique_time_.reserve(profile_max_size);
1723 profile_unique_time_.reserve(profile_max_size);
1728 void InitialPropagate()
override {
1735 void Post()
override {
1737 solver(),
this, &CumulativeTimeTable::InitialPropagate,
1738 "InitialPropagate");
1739 for (Task*
const task : by_start_min_) {
1740 task->WhenAnything(demon);
1742 capacity_->WhenRange(demon);
1745 void Accept(ModelVisitor*
const visitor)
const override {
1746 LOG(
FATAL) <<
"Should not be visited";
1749 std::string DebugString()
const override {
return "CumulativeTimeTable"; }
1753 void BuildProfile() {
1755 profile_non_unique_time_.clear();
1756 for (
const Task*
const task : by_start_min_) {
1757 const IntervalVar*
const interval = task->interval;
1761 const int64_t demand_min = task->DemandMin();
1762 if (demand_min > 0) {
1763 profile_non_unique_time_.emplace_back(
start_max, +demand_min);
1764 profile_non_unique_time_.emplace_back(
end_min, -demand_min);
1769 std::sort(profile_non_unique_time_.begin(), profile_non_unique_time_.end(),
1772 profile_unique_time_.clear();
1775 for (
const ProfileDelta& step : profile_non_unique_time_) {
1776 if (step.time == profile_unique_time_.back().time) {
1777 profile_unique_time_.back().delta += step.delta;
1779 profile_unique_time_.push_back(step);
1782 usage += step.delta;
1787 int64_t max_usage = 0;
1788 for (
const ProfileDelta& step : profile_unique_time_) {
1789 usage += step.delta;
1790 if (usage > max_usage) {
1795 capacity_->SetMin(max_usage);
1802 std::sort(by_start_min_.begin(), by_start_min_.end(),
1803 StartMinLessThan<Task>);
1805 int profile_index = 0;
1806 for (
const Task*
const task : by_start_min_) {
1807 const IntervalVar*
const interval = task->interval;
1812 while (
interval->StartMin() > profile_unique_time_[profile_index].time) {
1813 DCHECK(profile_index < profile_unique_time_.size());
1815 usage += profile_unique_time_[profile_index].delta;
1817 PushTask(task, profile_index, usage);
1825 void PushTask(
const Task*
const task,
int profile_index, int64_t usage) {
1827 const IntervalVar*
const interval = task->interval;
1828 const int64_t demand_min = task->DemandMin();
1829 if (demand_min == 0) {
1832 const int64_t residual_capacity =
CapSub(capacity_->Max(), demand_min);
1833 const int64_t duration = task->interval->DurationMin();
1834 const ProfileDelta& first_prof_delta = profile_unique_time_[profile_index];
1836 int64_t new_start_min =
interval->StartMin();
1840 if (first_prof_delta.time >
interval->StartMin()) {
1849 const int64_t usage_at_start_min =
CapSub(usage, first_prof_delta.delta);
1850 if (usage_at_start_min > residual_capacity) {
1851 new_start_min = profile_unique_time_[profile_index].time;
1859 ProfileDelta delta_end(
end_min, 0);
1861 delta_start.delta = +demand_min;
1862 delta_end.delta = -demand_min;
1864 while (profile_unique_time_[profile_index].
time <
1865 CapAdd(duration, new_start_min)) {
1866 const ProfileDelta& profile_delta = profile_unique_time_[profile_index];
1867 DCHECK(profile_index < profile_unique_time_.size());
1869 if (profile_delta.time == delta_start.time) {
1870 usage -= delta_start.delta;
1872 if (profile_delta.time == delta_end.time) {
1873 usage -= delta_end.delta;
1877 DCHECK(profile_index < profile_unique_time_.size());
1879 if (usage > residual_capacity) {
1880 new_start_min = profile_unique_time_[profile_index].time;
1882 usage += profile_unique_time_[profile_index].delta;
1884 task->interval->SetStartMin(new_start_min);
1887 typedef std::vector<ProfileDelta> Profile;
1889 Profile profile_unique_time_;
1890 Profile profile_non_unique_time_;
1891 std::vector<Task*> by_start_min_;
1892 IntVar*
const capacity_;
1907template <
class Task>
1908class TimeTableSync :
public Constraint {
1910 TimeTableSync(Solver*
const solver,
const std::vector<Task*>& tasks,
1912 : Constraint(solver), tasks_(tasks), capacity_(
capacity) {
1913 num_tasks_ = tasks_.size();
1919 start_min_.reserve(num_tasks_);
1920 start_max_.reserve(num_tasks_);
1921 end_min_.reserve(num_tasks_);
1922 durations_.reserve(num_tasks_);
1923 demands_.reserve(num_tasks_);
1928 void InitialPropagate()
override {
1931 while (!events_scp_.empty() && !events_ecp_.empty()) {
1933 pos_ = NextEventTime();
1938 capacity_->SetMin(capacity_->Max() - gap_);
1940 next_pos_ = NextScpTime();
1948 void Post()
override {
1950 solver(),
this, &TimeTableSync::InitialPropagate,
"InitialPropagate");
1951 for (Task*
const task : tasks_) {
1952 task->WhenAnything(demon);
1954 capacity_->WhenRange(demon);
1957 void Accept(ModelVisitor*
const visitor)
const override {
1958 LOG(
FATAL) <<
"Should not be visited";
1961 std::string DebugString()
const override {
return "TimeTableSync"; }
1965 enum State { NONE, READY,
CHECK, CONFLICT };
1967 inline int64_t NextScpTime() {
1968 return !events_scp_.empty() ? events_scp_.top().first
1972 inline int64_t NextEventTime() {
1974 if (!events_pr_.empty()) {
1975 time = events_pr_.top().first;
1977 if (!events_scp_.empty()) {
1978 int64_t t = events_scp_.top().first;
1981 if (!events_ecp_.empty()) {
1982 int64_t t = events_ecp_.top().first;
1988 void ProcessEventsScp() {
1989 while (!events_scp_.empty() && events_scp_.top().first == pos_) {
1990 const int64_t task_id = events_scp_.top().second;
1992 const int64_t old_end_min = end_min_[task_id];
1993 if (states_[task_id] == State::CONFLICT) {
1995 const int64_t new_end_min = pos_ + durations_[task_id];
1996 start_min_[task_id] = pos_;
1997 end_min_[task_id] = new_end_min;
1999 tasks_[task_id]->interval->SetStartMin(pos_);
2002 states_[task_id] = State::READY;
2004 if (pos_ < end_min_[task_id]) {
2005 gap_ -= demands_[task_id];
2006 if (old_end_min <= pos_) {
2007 events_ecp_.push(kv(end_min_[task_id], task_id));
2013 void ProcessEventsEcp() {
2014 while (!events_ecp_.empty() && events_ecp_.top().first == pos_) {
2015 const int64_t task_id = events_ecp_.top().second;
2018 if (pos_ < end_min_[task_id]) {
2019 events_ecp_.push(kv(end_min_[task_id], task_id));
2021 gap_ += demands_[task_id];
2026 void ProcessEventsPr() {
2027 while (!events_pr_.empty() && events_pr_.top().first == pos_) {
2028 const int64_t task_id = events_pr_.top().second;
2031 if (demands_[task_id] > gap_) {
2032 states_[task_id] = State::CONFLICT;
2033 conflict_.push(kv(demands_[task_id], task_id));
2037 if (next_pos_ < end_min_[task_id]) {
2039 check_.push(kv(demands_[task_id], task_id));
2043 states_[task_id] = State::READY;
2049 capacity_->SetMin(capacity_->Max() - gap_);
2051 if (gap_ < prev_gap_) {
2053 while (!check_.empty() && demands_[check_.top().second] > gap_) {
2054 const int64_t task_id = check_.top().second;
2056 if (states_[task_id] ==
State::CHECK && pos_ < end_min_[task_id]) {
2057 states_[task_id] = State::CONFLICT;
2058 conflict_.push(kv(demands_[task_id], task_id));
2061 states_[task_id] = State::READY;
2066 if (gap_ > prev_gap_) {
2068 while (!conflict_.empty() && demands_[conflict_.top().second] <= gap_) {
2069 const int64_t task_id = conflict_.top().second;
2071 if (states_[task_id] != State::CONFLICT) {
2074 const int64_t old_end_min = end_min_[task_id];
2076 start_min_[task_id] = pos_;
2077 end_min_[task_id] = pos_ + durations_[task_id];
2079 tasks_[task_id]->interval->SetStartMin(pos_);
2081 if (next_pos_ < end_min_[task_id]) {
2083 check_.push(kv(demands_[task_id], task_id));
2085 states_[task_id] = State::READY;
2088 const int64_t
start_max = start_max_[task_id];
2090 events_ecp_.push(kv(end_min_[task_id], task_id));
2097 void BuildEvents() {
2101 gap_ = capacity_->Max();
2102 prev_gap_ = capacity_->Max();
2104 conflict_ = min_heap();
2105 check_ = max_heap();
2107 events_pr_ = min_heap();
2108 events_scp_ = min_heap();
2109 events_ecp_ = min_heap();
2118 for (
int i = 0; i < num_tasks_; i++) {
2119 const int64_t s_min = tasks_[i]->interval->StartMin();
2120 const int64_t s_max = tasks_[i]->interval->StartMax();
2121 const int64_t e_min = tasks_[i]->interval->EndMin();
2123 start_min_.push_back(s_min);
2124 start_max_.push_back(s_max);
2125 end_min_.push_back(e_min);
2126 durations_.push_back(tasks_[i]->
interval->DurationMin());
2127 demands_.push_back(tasks_[i]->DemandMin());
2129 states_.push_back(State::NONE);
2131 events_scp_.push(kv(s_max, i));
2133 if (s_min != s_max) {
2134 events_pr_.push(kv(s_min, i));
2137 if (s_max < e_min) {
2138 events_ecp_.push(kv(e_min, i));
2144 std::vector<Task*> tasks_;
2145 IntVar*
const capacity_;
2147 std::vector<int64_t> start_min_;
2148 std::vector<int64_t> start_max_;
2149 std::vector<int64_t> end_min_;
2150 std::vector<int64_t> end_max_;
2151 std::vector<int64_t> durations_;
2152 std::vector<int64_t> demands_;
2155 typedef std::pair<int64_t, int64_t> kv;
2156 typedef std::priority_queue<kv, std::vector<kv>, std::greater<kv>> min_heap;
2157 typedef std::priority_queue<kv, std::vector<kv>, std::less<kv>> max_heap;
2160 min_heap events_pr_;
2161 min_heap events_scp_;
2162 min_heap events_ecp_;
2165 std::vector<State> states_;
2176class CumulativeConstraint :
public Constraint {
2178 CumulativeConstraint(Solver*
const s,
2179 const std::vector<IntervalVar*>& intervals,
2180 const std::vector<int64_t>& demands,
2184 intervals_(intervals),
2186 tasks_.reserve(intervals.size());
2187 for (
int i = 0; i < intervals.size(); ++i) {
2188 tasks_.push_back(CumulativeTask(intervals[i], demands[i]));
2192 void Post()
override {
2196 const ConstraintSolverParameters& params = solver()->parameters();
2197 if (params.use_cumulative_time_table()) {
2198 if (params.use_cumulative_time_table_sync()) {
2199 PostOneSidedConstraint(
false,
false,
true);
2200 PostOneSidedConstraint(
true,
false,
true);
2202 PostOneSidedConstraint(
false,
false,
false);
2203 PostOneSidedConstraint(
true,
false,
false);
2206 if (params.use_cumulative_edge_finder()) {
2207 PostOneSidedConstraint(
false,
true,
false);
2208 PostOneSidedConstraint(
true,
true,
false);
2210 if (params.use_sequence_high_demand_tasks()) {
2211 PostHighDemandSequenceConstraint();
2213 if (params.use_all_possible_disjunctions()) {
2214 PostAllDisjunctions();
2218 void InitialPropagate()
override {
2222 void Accept(ModelVisitor*
const visitor)
const override {
2224 visitor->BeginVisitConstraint(ModelVisitor::kCumulative,
this);
2225 visitor->VisitIntervalArrayArgument(ModelVisitor::kIntervalsArgument,
2227 visitor->VisitIntegerArrayArgument(ModelVisitor::kDemandsArgument,
2229 visitor->VisitIntegerExpressionArgument(ModelVisitor::kCapacityArgument,
2231 visitor->EndVisitConstraint(ModelVisitor::kCumulative,
this);
2234 std::string DebugString()
const override {
2235 return absl::StrFormat(
"CumulativeConstraint([%s], %s)",
2237 capacity_->DebugString());
2242 void PostAllDisjunctions() {
2243 for (
int i = 0; i < intervals_.size(); ++i) {
2244 IntervalVar*
const interval_i = intervals_[i];
2245 if (interval_i->MayBePerformed()) {
2246 for (
int j = i + 1; j < intervals_.size(); ++j) {
2247 IntervalVar*
const interval_j = intervals_[j];
2248 if (interval_j->MayBePerformed()) {
2250 Constraint*
const constraint =
2251 solver()->MakeTemporalDisjunction(interval_i, interval_j);
2252 solver()->AddConstraint(constraint);
2262 void PostHighDemandSequenceConstraint() {
2263 Constraint* constraint =
nullptr;
2265 std::vector<IntervalVar*> high_demand_intervals;
2266 high_demand_intervals.reserve(intervals_.size());
2267 for (
int i = 0; i < demands_.size(); ++i) {
2268 const int64_t
demand = tasks_[i].demand;
2275 if (
demand * 2 > capacity_->Max() &&
2276 tasks_[i].interval->MayBePerformed()) {
2277 high_demand_intervals.push_back(tasks_[i].
interval);
2280 if (high_demand_intervals.size() >= 2) {
2283 std::string seq_name = absl::StrCat(
name(),
"-HighDemandSequence");
2284 constraint = solver()->MakeDisjunctiveConstraint(high_demand_intervals,
2288 if (constraint !=
nullptr) {
2289 solver()->AddConstraint(constraint);
2295 void PopulateVectorUsefulTasks(
2296 bool mirror, std::vector<CumulativeTask*>*
const useful_tasks) {
2297 DCHECK(useful_tasks->empty());
2298 for (
int i = 0; i < tasks_.size(); ++i) {
2299 const CumulativeTask& original_task = tasks_[i];
2300 IntervalVar*
const interval = original_task.interval;
2302 if (original_task.demand > capacity_->Max()) {
2307 if (
interval->MayBePerformed() && original_task.demand > 0) {
2308 Solver*
const s = solver();
2309 IntervalVar*
const original_interval = original_task.interval;
2311 mirror ? s->MakeMirrorInterval(original_interval)
2312 : original_interval;
2313 IntervalVar*
const relaxed_max = s->MakeIntervalRelaxedMax(
interval);
2314 useful_tasks->push_back(
2315 new CumulativeTask(relaxed_max, original_task.demand));
2322 Constraint* MakeOneSidedConstraint(
bool mirror,
bool edge_finder,
2324 std::vector<CumulativeTask*> useful_tasks;
2325 PopulateVectorUsefulTasks(mirror, &useful_tasks);
2326 if (useful_tasks.empty()) {
2329 Solver*
const s = solver();
2331 const ConstraintSolverParameters& params = solver()->parameters();
2332 return useful_tasks.size() < params.max_edge_finder_size()
2333 ? s->RevAlloc(
new EdgeFinder<CumulativeTask>(s, useful_tasks,
2339 new TimeTableSync<CumulativeTask>(s, useful_tasks, capacity_));
2342 new CumulativeTimeTable<CumulativeTask>(s, useful_tasks, capacity_));
2347 void PostOneSidedConstraint(
bool mirror,
bool edge_finder,
bool tt_sync) {
2348 Constraint*
const constraint =
2349 MakeOneSidedConstraint(mirror, edge_finder, tt_sync);
2350 if (constraint !=
nullptr) {
2351 solver()->AddConstraint(constraint);
2356 IntVar*
const capacity_;
2359 std::vector<CumulativeTask> tasks_;
2362 const std::vector<IntervalVar*> intervals_;
2364 const std::vector<int64_t> demands_;
2369class VariableDemandCumulativeConstraint :
public Constraint {
2371 VariableDemandCumulativeConstraint(Solver*
const s,
2372 const std::vector<IntervalVar*>& intervals,
2373 const std::vector<IntVar*>& demands,
2375 const std::string&
name)
2378 intervals_(intervals),
2380 tasks_.reserve(intervals.size());
2381 for (
int i = 0; i < intervals.size(); ++i) {
2382 tasks_.push_back(VariableCumulativeTask(intervals[i], demands[i]));
2386 void Post()
override {
2390 const ConstraintSolverParameters& params = solver()->parameters();
2391 if (params.use_cumulative_time_table()) {
2392 PostOneSidedConstraint(
false,
false,
false);
2393 PostOneSidedConstraint(
true,
false,
false);
2395 if (params.use_cumulative_edge_finder()) {
2396 PostOneSidedConstraint(
false,
true,
false);
2397 PostOneSidedConstraint(
true,
true,
false);
2399 if (params.use_sequence_high_demand_tasks()) {
2400 PostHighDemandSequenceConstraint();
2402 if (params.use_all_possible_disjunctions()) {
2403 PostAllDisjunctions();
2407 void InitialPropagate()
override {
2411 void Accept(ModelVisitor*
const visitor)
const override {
2413 visitor->BeginVisitConstraint(ModelVisitor::kCumulative,
this);
2414 visitor->VisitIntervalArrayArgument(ModelVisitor::kIntervalsArgument,
2416 visitor->VisitIntegerVariableArrayArgument(ModelVisitor::kDemandsArgument,
2418 visitor->VisitIntegerExpressionArgument(ModelVisitor::kCapacityArgument,
2420 visitor->EndVisitConstraint(ModelVisitor::kCumulative,
this);
2423 std::string DebugString()
const override {
2424 return absl::StrFormat(
"VariableDemandCumulativeConstraint([%s], %s)",
2426 capacity_->DebugString());
2431 void PostAllDisjunctions() {
2432 for (
int i = 0; i < intervals_.size(); ++i) {
2433 IntervalVar*
const interval_i = intervals_[i];
2434 if (interval_i->MayBePerformed()) {
2435 for (
int j = i + 1; j < intervals_.size(); ++j) {
2436 IntervalVar*
const interval_j = intervals_[j];
2437 if (interval_j->MayBePerformed()) {
2438 if (
CapAdd(tasks_[i].
demand->Min(), tasks_[j].demand->Min()) >
2440 Constraint*
const constraint =
2441 solver()->MakeTemporalDisjunction(interval_i, interval_j);
2442 solver()->AddConstraint(constraint);
2452 void PostHighDemandSequenceConstraint() {
2453 Constraint* constraint =
nullptr;
2455 std::vector<IntervalVar*> high_demand_intervals;
2456 high_demand_intervals.reserve(intervals_.size());
2457 for (
int i = 0; i < demands_.size(); ++i) {
2458 const int64_t
demand = tasks_[i].demand->Min();
2465 if (
demand * 2 > capacity_->Max() &&
2466 tasks_[i].interval->MayBePerformed()) {
2467 high_demand_intervals.push_back(tasks_[i].
interval);
2470 if (high_demand_intervals.size() >= 2) {
2473 const std::string seq_name =
2474 absl::StrCat(
name(),
"-HighDemandSequence");
2475 constraint = solver()->MakeStrictDisjunctiveConstraint(
2476 high_demand_intervals, seq_name);
2479 if (constraint !=
nullptr) {
2480 solver()->AddConstraint(constraint);
2486 void PopulateVectorUsefulTasks(
2487 bool mirror, std::vector<VariableCumulativeTask*>*
const useful_tasks) {
2488 DCHECK(useful_tasks->empty());
2489 for (
int i = 0; i < tasks_.size(); ++i) {
2490 const VariableCumulativeTask& original_task = tasks_[i];
2491 IntervalVar*
const interval = original_task.interval;
2493 if (original_task.demand->Min() > capacity_->Max()) {
2498 if (
interval->MayBePerformed() && original_task.demand->Max() > 0) {
2499 Solver*
const s = solver();
2500 IntervalVar*
const original_interval = original_task.interval;
2502 mirror ? s->MakeMirrorInterval(original_interval)
2503 : original_interval;
2504 IntervalVar*
const relaxed_max = s->MakeIntervalRelaxedMax(
interval);
2505 useful_tasks->push_back(
2506 new VariableCumulativeTask(relaxed_max, original_task.demand));
2513 Constraint* MakeOneSidedConstraint(
bool mirror,
bool edge_finder,
2515 std::vector<VariableCumulativeTask*> useful_tasks;
2516 PopulateVectorUsefulTasks(mirror, &useful_tasks);
2517 if (useful_tasks.empty()) {
2520 Solver*
const s = solver();
2523 new EdgeFinder<VariableCumulativeTask>(s, useful_tasks, capacity_));
2526 return s->RevAlloc(
new TimeTableSync<VariableCumulativeTask>(
2527 s, useful_tasks, capacity_));
2529 return s->RevAlloc(
new CumulativeTimeTable<VariableCumulativeTask>(
2530 s, useful_tasks, capacity_));
2535 void PostOneSidedConstraint(
bool mirror,
bool edge_finder,
bool tt_sync) {
2536 Constraint*
const constraint =
2537 MakeOneSidedConstraint(mirror, edge_finder, tt_sync);
2538 if (constraint !=
nullptr) {
2539 solver()->AddConstraint(constraint);
2544 IntVar*
const capacity_;
2547 std::vector<VariableCumulativeTask> tasks_;
2550 const std::vector<IntervalVar*> intervals_;
2552 const std::vector<IntVar*> demands_;
2562DisjunctiveConstraint::DisjunctiveConstraint(
2563 Solver*
const s,
const std::vector<IntervalVar*>& intervals,
2564 const std::string&
name)
2566 if (!
name.empty()) {
2575 std::function<int64_t(int64_t, int64_t)> transition_time) {
2576 if (transition_time !=
nullptr) {
2586 const std::vector<IntervalVar*>& intervals,
const std::string&
name) {
2587 return RevAlloc(
new FullDisjunctiveConstraint(
this, intervals,
name,
false));
2591 const std::vector<IntervalVar*>& intervals,
const std::string&
name) {
2592 return RevAlloc(
new FullDisjunctiveConstraint(
this, intervals,
name,
true));
2598 const std::vector<int64_t>& demands,
2600 CHECK_EQ(intervals.size(), demands.size());
2601 for (
int i = 0; i < intervals.size(); ++i) {
2607 return RevAlloc(
new CumulativeConstraint(
this, intervals, demands,
2612 const std::vector<int>& demands,
2618 const std::vector<int64_t>& demands,
2620 const std::string&
name) {
2621 CHECK_EQ(intervals.size(), demands.size());
2622 for (
int i = 0; i < intervals.size(); ++i) {
2626 new CumulativeConstraint(
this, intervals, demands,
capacity,
name));
2630 const std::vector<int>& demands,
2632 const std::string&
name) {
2639 const std::vector<IntVar*>& demands,
2641 CHECK_EQ(intervals.size(), demands.size());
2642 for (
int i = 0; i < intervals.size(); ++i) {
2646 std::vector<int64_t> fixed_demands(demands.size());
2647 for (
int i = 0; i < demands.size(); ++i) {
2648 fixed_demands[i] = demands[i]->Value();
2652 return RevAlloc(
new VariableDemandCumulativeConstraint(
2657 const std::vector<IntVar*>& demands,
2659 const std::string&
name) {
2660 CHECK_EQ(intervals.size(), demands.size());
2661 for (
int i = 0; i < intervals.size(); ++i) {
2665 std::vector<int64_t> fixed_demands(demands.size());
2666 for (
int i = 0; i < demands.size(); ++i) {
2667 fixed_demands[i] = demands[i]->Value();
2671 return RevAlloc(
new VariableDemandCumulativeConstraint(
#define DCHECK_LE(val1, val2)
#define DCHECK_NE(val1, val2)
#define CHECK_EQ(val1, val2)
#define CHECK_GE(val1, val2)
#define DCHECK_GE(val1, val2)
#define DCHECK_GT(val1, val2)
#define DCHECK_LT(val1, val2)
#define DCHECK(condition)
#define DCHECK_EQ(val1, val2)
#define VLOG(verboselevel)
A constraint is the main modeling object.
~DisjunctiveConstraint() override
void SetTransitionTime(Solver::IndexEvaluator2 transition_time)
Add a transition time between intervals.
Solver::IndexEvaluator2 transition_time_
The class IntVar is a subset of IntExpr.
static IntegralType CeilOfRatio(IntegralType numerator, IntegralType denominator)
virtual std::string name() const
Object naming.
void set_name(const std::string &name)
DisjunctiveConstraint * MakeStrictDisjunctiveConstraint(const std::vector< IntervalVar * > &intervals, const std::string &name)
This constraint forces all interval vars into an non-overlapping sequence.
DisjunctiveConstraint * MakeDisjunctiveConstraint(const std::vector< IntervalVar * > &intervals, const std::string &name)
This constraint forces all interval vars into an non-overlapping sequence.
Constraint * MakeCumulative(const std::vector< IntervalVar * > &intervals, const std::vector< int64_t > &demands, int64_t capacity, const std::string &name)
This constraint forces that, for any integer t, the sum of the demands corresponding to an interval c...
IntVar * MakeIntConst(int64_t val, const std::string &name)
IntConst will create a constant expression.
T * RevAlloc(T *object)
Registers the given object as being reversible.
#define DISALLOW_COPY_AND_ASSIGN(TypeName)
const Collection::value_type::second_type FindPtrOrNull(const Collection &collection, const typename Collection::value_type::first_type &key)
void STLDeleteValues(T *v)
void STLDeleteElements(T *container)
double Distance(const VectorXd &vector1, const VectorXd &vector2, const Sharder &sharder)
Collection of objects used to extend the Constraint Solver library.
int64_t CapAdd(int64_t x, int64_t y)
int64_t CapSub(int64_t x, int64_t y)
Demon * MakeDelayedConstraintDemon0(Solver *const s, T *const ct, void(T::*method)(), const std::string &name)
std::string JoinDebugStringPtr(const std::vector< T > &v, const std::string &separator)
int64_t CapProd(int64_t x, int64_t y)
std::vector< int64_t > ToInt64Vector(const std::vector< int > &input)
bool AreAllOnes(const std::vector< T > &values)
bool AreAllBound(const std::vector< IntVar * > &vars)
std::string JoinDebugString(const std::vector< T > &v, const std::string &separator)
int64_t residual_energetic_end_min
int64_t energetic_end_min
int64_t energetic_end_min_opt
static const int64_t kNotInitialized
int argmax_energetic_end_min_opt
static const int64_t kNotAvailable