Files
ortools-clone/ortools/sat/samples/assumptions_sample_sat.cc
2025-05-16 18:02:19 +02:00

82 lines
2.5 KiB
C++

// Copyright 2010-2025 Google LLC
// 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.
// [START program]
// [START import]
#include <stdlib.h>
#include "absl/base/log_severity.h"
#include "absl/log/globals.h"
#include "absl/types/span.h"
#include "ortools/base/init_google.h"
#include "ortools/base/logging.h"
#include "ortools/sat/cp_model.h"
#include "ortools/sat/cp_model.pb.h"
#include "ortools/sat/cp_model_solver.h"
#include "ortools/util/sorted_interval_list.h"
// [END import]
namespace operations_research {
namespace sat {
void AssumptionsSampleSat() {
// [START model]
CpModelBuilder cp_model;
// [END model]
// [START variables]
const Domain domain(0, 10);
const IntVar x = cp_model.NewIntVar(domain).WithName("x");
const IntVar y = cp_model.NewIntVar(domain).WithName("y");
const IntVar z = cp_model.NewIntVar(domain).WithName("z");
const BoolVar a = cp_model.NewBoolVar().WithName("a");
const BoolVar b = cp_model.NewBoolVar().WithName("b");
const BoolVar c = cp_model.NewBoolVar().WithName("c");
// [END variables]
// [START constraints]
cp_model.AddGreaterThan(x, y).OnlyEnforceIf(a);
cp_model.AddGreaterThan(y, z).OnlyEnforceIf(b);
cp_model.AddGreaterThan(z, x).OnlyEnforceIf(c);
// [END constraints]
// Add assumptions
cp_model.AddAssumptions({a, b, c});
// Solving part.
// [START solve]
const CpSolverResponse response = Solve(cp_model.Build());
// [END solve]
// Print solution.
// [START print_solution]
LOG(INFO) << CpSolverResponseStats(response);
if (response.status() == CpSolverStatus::INFEASIBLE) {
for (const int index :
response.sufficient_assumptions_for_infeasibility()) {
LOG(INFO) << index;
}
}
// [END print_solution]
}
} // namespace sat
} // namespace operations_research
int main(int argc, char* argv[]) {
InitGoogle(argv[0], &argc, &argv, true);
absl::SetStderrThreshold(absl::LogSeverityAtLeast::kInfo);
operations_research::sat::AssumptionsSampleSat();
return EXIT_SUCCESS;
}
// [END program]