69 lines
2.3 KiB
C++
69 lines
2.3 KiB
C++
// Copyright 2010-2017 Google
|
|
// 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.
|
|
|
|
#include "ortools/sat/cp_model.pb.h"
|
|
#include "ortools/sat/cp_model_solver.h"
|
|
#include "ortools/sat/cp_model_utils.h"
|
|
#include "ortools/sat/model.h"
|
|
|
|
namespace operations_research {
|
|
namespace sat {
|
|
|
|
void OptionalIntervalSample() {
|
|
CpModelProto cp_model;
|
|
const int kHorizon = 100;
|
|
|
|
auto new_variable = [&cp_model](int64 lb, int64 ub) {
|
|
CHECK_LE(lb, ub);
|
|
const int index = cp_model.variables_size();
|
|
IntegerVariableProto* const var = cp_model.add_variables();
|
|
var->add_domain(lb);
|
|
var->add_domain(ub);
|
|
return index;
|
|
};
|
|
|
|
auto new_constant = [&new_variable](int64 v) { return new_variable(v, v); };
|
|
|
|
auto new_optional_interval = [&cp_model](int start, int duration, int end,
|
|
int presence) {
|
|
const int index = cp_model.constraints_size();
|
|
ConstraintProto* const ct = cp_model.add_constraints();
|
|
ct->add_enforcement_literal(presence);
|
|
IntervalConstraintProto* const interval = ct->mutable_interval();
|
|
interval->set_start(start);
|
|
interval->set_size(duration);
|
|
interval->set_end(end);
|
|
return index;
|
|
};
|
|
|
|
const int start_var = new_variable(0, kHorizon);
|
|
const int duration_var = new_constant(10);
|
|
const int end_var = new_variable(0, kHorizon);
|
|
const int presence_var = new_variable(0, 1);
|
|
const int interval_var =
|
|
new_optional_interval(start_var, duration_var, end_var, presence_var);
|
|
LOG(INFO) << "start_var = " << start_var
|
|
<< ", duration_var = " << duration_var << ", end_var = " << end_var
|
|
<< ", presence_var = " << presence_var
|
|
<< ", interval_var = " << interval_var;
|
|
}
|
|
|
|
} // namespace sat
|
|
} // namespace operations_research
|
|
|
|
int main() {
|
|
operations_research::sat::OptionalIntervalSample();
|
|
|
|
return EXIT_SUCCESS;
|
|
}
|