linear_programming.cc 4.48 KB
Newer Older
Valentin Platzgummer's avatar
Valentin Platzgummer committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
// Copyright 2010-2018 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.

// Linear programming example that shows how to use the API.

Valentin Platzgummer's avatar
Valentin Platzgummer committed
16
#include "ortools/base/commandlineflags.h"
Valentin Platzgummer's avatar
Valentin Platzgummer committed
17 18 19 20 21
#include "ortools/base/logging.h"
#include "ortools/linear_solver/linear_solver.h"
#include "ortools/linear_solver/linear_solver.pb.h"

namespace operations_research {
Valentin Platzgummer's avatar
Valentin Platzgummer committed
22 23 24 25 26 27 28 29 30 31 32 33
void RunLinearProgrammingExample(const std::string& optimization_problem_type) {
  LOG(INFO) << "---- Linear programming example with "
            << optimization_problem_type << " ----";
  if (!MPSolver::ParseAndCheckSupportForProblemType(
          optimization_problem_type)) {
    LOG(INFO) << "  support for solver not linked in.";
    return;
  }

  MPSolver solver("IntegerProgrammingExample",
                  MPSolver::ParseSolverTypeOrDie(optimization_problem_type));

Valentin Platzgummer's avatar
Valentin Platzgummer committed
34
  const double infinity = solver.infinity();
Valentin Platzgummer's avatar
Valentin Platzgummer committed
35 36 37 38
  // x1, x2 and x3 are continuous non-negative variables.
  MPVariable* const x1 = solver.MakeNumVar(0.0, infinity, "x1");
  MPVariable* const x2 = solver.MakeNumVar(0.0, infinity, "x2");
  MPVariable* const x3 = solver.MakeNumVar(0.0, infinity, "x3");
Valentin Platzgummer's avatar
Valentin Platzgummer committed
39

Valentin Platzgummer's avatar
Valentin Platzgummer committed
40
  // Maximize 10 * x1 + 6 * x2 + 4 * x3.
Valentin Platzgummer's avatar
Valentin Platzgummer committed
41
  MPObjective* const objective = solver.MutableObjective();
Valentin Platzgummer's avatar
Valentin Platzgummer committed
42 43 44
  objective->SetCoefficient(x1, 10);
  objective->SetCoefficient(x2, 6);
  objective->SetCoefficient(x3, 4);
Valentin Platzgummer's avatar
Valentin Platzgummer committed
45 46
  objective->SetMaximization();

Valentin Platzgummer's avatar
Valentin Platzgummer committed
47 48 49 50 51 52 53 54 55 56 57
  // x1 + x2 + x3 <= 100.
  MPConstraint* const c0 = solver.MakeRowConstraint(-infinity, 100.0);
  c0->SetCoefficient(x1, 1);
  c0->SetCoefficient(x2, 1);
  c0->SetCoefficient(x3, 1);

  // 10 * x1 + 4 * x2 + 5 * x3 <= 600.
  MPConstraint* const c1 = solver.MakeRowConstraint(-infinity, 600.0);
  c1->SetCoefficient(x1, 10);
  c1->SetCoefficient(x2, 4);
  c1->SetCoefficient(x3, 5);
Valentin Platzgummer's avatar
Valentin Platzgummer committed
58

Valentin Platzgummer's avatar
Valentin Platzgummer committed
59 60 61 62 63
  // 2 * x1 + 2 * x2 + 6 * x3 <= 300.
  MPConstraint* const c2 = solver.MakeRowConstraint(-infinity, 300.0);
  c2->SetCoefficient(x1, 2);
  c2->SetCoefficient(x2, 2);
  c2->SetCoefficient(x3, 6);
Valentin Platzgummer's avatar
Valentin Platzgummer committed
64

Valentin Platzgummer's avatar
Valentin Platzgummer committed
65
  // TODO(user): Change example to show = and >= constraints.
Valentin Platzgummer's avatar
Valentin Platzgummer committed
66 67 68 69 70

  LOG(INFO) << "Number of variables = " << solver.NumVariables();
  LOG(INFO) << "Number of constraints = " << solver.NumConstraints();

  const MPSolver::ResultStatus result_status = solver.Solve();
Valentin Platzgummer's avatar
Valentin Platzgummer committed
71

Valentin Platzgummer's avatar
Valentin Platzgummer committed
72 73 74 75
  // Check that the problem has an optimal solution.
  if (result_status != MPSolver::OPTIMAL) {
    LOG(FATAL) << "The problem does not have an optimal solution!";
  }
Valentin Platzgummer's avatar
Valentin Platzgummer committed
76 77 78 79

  LOG(INFO) << "Problem solved in " << solver.wall_time() << " milliseconds";

  // The objective value of the solution.
Valentin Platzgummer's avatar
Valentin Platzgummer committed
80
  LOG(INFO) << "Optimal objective value = " << objective->Value();
Valentin Platzgummer's avatar
Valentin Platzgummer committed
81 82 83 84 85 86

  // The value of each variable in the solution.
  LOG(INFO) << "x1 = " << x1->solution_value();
  LOG(INFO) << "x2 = " << x2->solution_value();
  LOG(INFO) << "x3 = " << x3->solution_value();

Valentin Platzgummer's avatar
Valentin Platzgummer committed
87 88
  LOG(INFO) << "Advanced usage:";
  LOG(INFO) << "Problem solved in " << solver.iterations() << " iterations";
Valentin Platzgummer's avatar
Valentin Platzgummer committed
89 90 91
  LOG(INFO) << "x1: reduced cost = " << x1->reduced_cost();
  LOG(INFO) << "x2: reduced cost = " << x2->reduced_cost();
  LOG(INFO) << "x3: reduced cost = " << x3->reduced_cost();
Valentin Platzgummer's avatar
Valentin Platzgummer committed
92 93 94 95 96 97 98 99
  const std::vector<double> activities = solver.ComputeConstraintActivities();
  LOG(INFO) << "c0: dual value = " << c0->dual_value()
            << " activity = " << activities[c0->index()];
  LOG(INFO) << "c1: dual value = " << c1->dual_value()
            << " activity = " << activities[c1->index()];
  LOG(INFO) << "c2: dual value = " << c2->dual_value()
            << " activity = " << activities[c2->index()];
}
Valentin Platzgummer's avatar
Valentin Platzgummer committed
100 101 102 103 104 105 106 107 108

void RunAllExamples() {
  RunLinearProgrammingExample("GLOP");
  RunLinearProgrammingExample("CLP");
  RunLinearProgrammingExample("GUROBI_LP");
  RunLinearProgrammingExample("CPLEX_LP");
  RunLinearProgrammingExample("GLPK_LP");
  RunLinearProgrammingExample("XPRESS_LP");
}
Valentin Platzgummer's avatar
Valentin Platzgummer committed
109 110 111 112
}  // namespace operations_research

int main(int argc, char** argv) {
  google::InitGoogleLogging(argv[0]);
Valentin Platzgummer's avatar
Valentin Platzgummer committed
113 114 115 116
  absl::SetFlag(&FLAGS_logtostderr, true);
  absl::SetFlag(&FLAGS_log_prefix, false);
  gflags::ParseCommandLineFlags(&argc, &argv, true);
  operations_research::RunAllExamples();
Valentin Platzgummer's avatar
Valentin Platzgummer committed
117 118
  return EXIT_SUCCESS;
}