or-tools 0.3.4 → 0.5.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: dba74899e2a57b086643657e5093f12e67fd4fcdda7e22c2b533911f9991e43a
-  data.tar.gz: 963c0b75f55b4b772ead3a8402ec8bb4b90570c23de1c7dc3a437758e8949af2
+  metadata.gz: 405e26b51140e6605acd8904d7e90e0e7f48b7cfd83641fef6e2e737e7b8cda4
+  data.tar.gz: e6e0506979a829c260f8bae70e7769524d59d83b7a52e1cadfb6e553d8116587
 SHA512:
-  metadata.gz: a5dc8662ec580c97a905817867346615e2fbb187d5b75d6b2b839a86e7385a20b2ed4f9f4c13ed536079c6141cdd0bf95bfad7ea6408d4eb81d8f0ed09e32a68
-  data.tar.gz: 195967d1e662ba05751ff31fbec3c96ebac519cd58a343ca551f9ba6c12304542b7f4127a37f88e99cda6b214d42089df7ed9b5a3134c0ec1eb978ea72a95be4
+  metadata.gz: 237e13e7340682d974602743de724031c28c99add58c4cf49e670bb05f4fa0c691dcf00dd75797a9ab3efcc9b5505a96f3a2a3d542db8c8746652ec114b5a258
+  data.tar.gz: 5c782dd281122001095663da47fce8279f5a2c29b6eed65e2288df6727c55724ef424d4c3bfda3ed3b4ce53493964b88d64f67fc843e97dafd98f3e21a4c5183

data/CHANGELOG.md

@@ -1,3 +1,30 @@
+## 0.5.0 (2021-04-30)
+
+- Updated OR-Tools to 9.0
+- Added binary installation for CentOS 7
+- Added `sufficient_assumptions_for_infeasibility` to `CpSolver`
+
+## 0.4.3 (2021-03-26)
+
+- Added `add_assumption`, `add_assumptions`, and `clear_assumptions` to `CpModel`
+- Added `add_hint` and `clear_hints` to `CpModel`
+- Added `only_enforce_if` to `SatConstraint`
+- Fixed installation for Debian
+
+## 0.4.2 (2021-03-03)
+
+- Updated OR-Tools to 8.2
+
+## 0.4.1 (2021-02-23)
+
+- Added solution printers
+- Improved `inspect` and `to_s` for `CpModel`
+- Improved constraint construction
+
+## 0.4.0 (2021-01-14)
+
+- Updated OR-Tools to 8.1
+
 ## 0.3.4 (2021-01-14)
 
 - Added support for time limit for `CpSolver`

data/README.md

@@ -267,7 +267,7 @@ Constraint Optimization
 - [Solving an Optimization Problem](#solving-an-optimization-problem)
 - [Cryptarithmetic](#cryptarithmetic)
 - [The N-queens Problem](#the-n-queens-problem)
-- [Setting Solver Limits](#setting-solver-limits) [master]
+- [Setting Solver Limits](#setting-solver-limits)
 
 Assignment
 

data/ext/or-tools/assignment.cpp

@@ -0,0 +1,39 @@
+#include <ortools/graph/assignment.h>
+
+#include <rice/Array.hpp>
+#include <rice/Constructor.hpp>
+#include <rice/Module.hpp>
+
+using operations_research::SimpleLinearSumAssignment;
+
+using Rice::Array;
+using Rice::Symbol;
+
+void init_assignment(Rice::Module& m) {
+  Rice::define_class_under<SimpleLinearSumAssignment>(m, "LinearSumAssignment")
+    .define_constructor(Rice::Constructor<SimpleLinearSumAssignment>())
+    .define_method("add_arc_with_cost", &SimpleLinearSumAssignment::AddArcWithCost)
+    .define_method("num_nodes", &SimpleLinearSumAssignment::NumNodes)
+    .define_method("num_arcs", &SimpleLinearSumAssignment::NumArcs)
+    .define_method("left_node", &SimpleLinearSumAssignment::LeftNode)
+    .define_method("right_node", &SimpleLinearSumAssignment::RightNode)
+    .define_method("cost", &SimpleLinearSumAssignment::Cost)
+    .define_method("optimal_cost", &SimpleLinearSumAssignment::OptimalCost)
+    .define_method("right_mate", &SimpleLinearSumAssignment::RightMate)
+    .define_method("assignment_cost", &SimpleLinearSumAssignment::AssignmentCost)
+    .define_method(
+      "solve",
+      *[](SimpleLinearSumAssignment& self) {
+        auto status = self.Solve();
+
+        if (status == SimpleLinearSumAssignment::Status::OPTIMAL) {
+          return Symbol("optimal");
+        } else if (status == SimpleLinearSumAssignment::Status::INFEASIBLE) {
+          return Symbol("infeasible");
+        } else if (status == SimpleLinearSumAssignment::Status::POSSIBLE_OVERFLOW) {
+          return Symbol("possible_overflow");
+        } else {
+          throw std::runtime_error("Unknown status");
+        }
+      });
+}

data/ext/or-tools/bin_packing.cpp

@@ -0,0 +1,55 @@
+#include <ortools/algorithms/knapsack_solver.h>
+
+#include <rice/Array.hpp>
+#include <rice/Constructor.hpp>
+#include <rice/Module.hpp>
+
+using operations_research::KnapsackSolver;
+
+using Rice::Array;
+using Rice::Object;
+using Rice::Symbol;
+
+template<>
+inline
+KnapsackSolver::SolverType from_ruby<KnapsackSolver::SolverType>(Object x)
+{
+  std::string s = Symbol(x).str();
+  if (s == "branch_and_bound") {
+    return KnapsackSolver::KNAPSACK_MULTIDIMENSION_BRANCH_AND_BOUND_SOLVER;
+  } else {
+    throw std::runtime_error("Unknown solver type: " + s);
+  }
+}
+
+void init_bin_packing(Rice::Module& m) {
+  Rice::define_class_under<KnapsackSolver>(m, "KnapsackSolver")
+    .define_constructor(Rice::Constructor<KnapsackSolver, KnapsackSolver::SolverType, std::string>())
+    .define_method("_solve", &KnapsackSolver::Solve)
+    .define_method("best_solution_contains?", &KnapsackSolver::BestSolutionContains)
+    .define_method(
+      "init",
+      *[](KnapsackSolver& self, Array rb_values, Array rb_weights, Array rb_capacities) {
+        std::vector<int64_t> values;
+        for (std::size_t i = 0; i < rb_values.size(); ++i) {
+          values.push_back(from_ruby<int64_t>(rb_values[i]));
+        }
+
+        std::vector<std::vector<int64_t>> weights;
+        for (std::size_t i = 0; i < rb_weights.size(); ++i) {
+          Array rb_w = Array(rb_weights[i]);
+          std::vector<int64_t> w;
+          for (std::size_t j = 0; j < rb_w.size(); ++j) {
+            w.push_back(from_ruby<int64_t>(rb_w[j]));
+          }
+          weights.push_back(w);
+        }
+
+        std::vector<int64_t> capacities;
+        for (std::size_t i = 0; i < rb_capacities.size(); ++i) {
+          capacities.push_back(from_ruby<int64_t>(rb_capacities[i]));
+        }
+
+        self.Init(values, weights, capacities);
+      });
+}

data/ext/or-tools/constraint.cpp

@@ -0,0 +1,473 @@
+#include <google/protobuf/text_format.h>
+#include <ortools/sat/cp_model.h>
+
+#include <rice/Array.hpp>
+#include <rice/Constructor.hpp>
+#include <rice/Module.hpp>
+
+using operations_research::sat::BoolVar;
+using operations_research::sat::Constraint;
+using operations_research::sat::CpModelBuilder;
+using operations_research::sat::CpSolverResponse;
+using operations_research::sat::CpSolverStatus;
+using operations_research::sat::LinearExpr;
+using operations_research::sat::IntVar;
+using operations_research::sat::IntervalVar;
+using operations_research::sat::Model;
+using operations_research::sat::NewFeasibleSolutionObserver;
+using operations_research::sat::SatParameters;
+using operations_research::sat::SolutionBooleanValue;
+using operations_research::sat::SolutionIntegerValue;
+
+using Rice::Array;
+using Rice::Class;
+using Rice::Object;
+using Rice::String;
+using Rice::Symbol;
+
+Class rb_cBoolVar;
+Class rb_cSatIntVar;
+
+template<>
+inline
+LinearExpr from_ruby<LinearExpr>(Object x)
+{
+  LinearExpr expr;
+
+  if (x.respond_to("to_i")) {
+    expr = from_ruby<int64_t>(x.call("to_i"));
+  } else if (x.respond_to("vars")) {
+    Array vars = x.call("vars");
+    for (auto const& var: vars) {
+      auto cvar = (Array) var;
+      Object o = cvar[0];
+      if (o.is_a(rb_cBoolVar)) {
+        expr.AddTerm(from_ruby<BoolVar>(cvar[0]), from_ruby<int64_t>(cvar[1]));
+      } else if (o.is_a(rb_cInteger)) {
+        expr.AddConstant(from_ruby<int64_t>(cvar[0]) * from_ruby<int64_t>(cvar[1]));
+      } else {
+        expr.AddTerm(from_ruby<IntVar>(cvar[0]), from_ruby<int64_t>(cvar[1]));
+      }
+    }
+  } else if (x.is_a(rb_cBoolVar)) {
+    expr = from_ruby<BoolVar>(x);
+  } else {
+    expr = from_ruby<IntVar>(x);
+  }
+
+  return expr;
+}
+
+// need a wrapper class since absl::Span doesn't own
+class IntVarSpan {
+  std::vector<IntVar> vec;
+  public:
+    IntVarSpan(Object x) {
+      Array a = Array(x);
+      vec.reserve(a.size());
+      for (std::size_t i = 0; i < a.size(); ++i) {
+        vec.push_back(from_ruby<IntVar>(a[i]));
+      }
+    }
+    operator absl::Span<const IntVar>() {
+      return absl::Span<const IntVar>(vec);
+    }
+};
+
+template<>
+inline
+IntVarSpan from_ruby<IntVarSpan>(Object x)
+{
+  return IntVarSpan(x);
+}
+
+// need a wrapper class since absl::Span doesn't own
+class IntervalVarSpan {
+  std::vector<IntervalVar> vec;
+  public:
+    IntervalVarSpan(Object x) {
+      Array a = Array(x);
+      vec.reserve(a.size());
+      for (std::size_t i = 0; i < a.size(); ++i) {
+        vec.push_back(from_ruby<IntervalVar>(a[i]));
+      }
+    }
+    operator absl::Span<const IntervalVar>() {
+      return absl::Span<const IntervalVar>(vec);
+    }
+};
+
+template<>
+inline
+IntervalVarSpan from_ruby<IntervalVarSpan>(Object x)
+{
+  return IntervalVarSpan(x);
+}
+
+// need a wrapper class since absl::Span doesn't own
+class LinearExprSpan {
+  std::vector<LinearExpr> vec;
+  public:
+    LinearExprSpan(Object x) {
+      Array a = Array(x);
+      vec.reserve(a.size());
+      for (std::size_t i = 0; i < a.size(); ++i) {
+        vec.push_back(from_ruby<LinearExpr>(a[i]));
+      }
+    }
+    operator absl::Span<const LinearExpr>() {
+      return absl::Span<const LinearExpr>(vec);
+    }
+};
+
+template<>
+inline
+LinearExprSpan from_ruby<LinearExprSpan>(Object x)
+{
+  return LinearExprSpan(x);
+}
+
+// need a wrapper class since absl::Span doesn't own
+class BoolVarSpan {
+  std::vector<BoolVar> vec;
+  public:
+    BoolVarSpan(Object x) {
+      Array a = Array(x);
+      vec.reserve(a.size());
+      for (std::size_t i = 0; i < a.size(); ++i) {
+        if (((Object) a[i]).is_a(rb_cSatIntVar)) {
+          vec.push_back(from_ruby<IntVar>(a[i]).ToBoolVar());
+        } else {
+          vec.push_back(from_ruby<BoolVar>(a[i]));
+        }
+      }
+    }
+    operator absl::Span<const BoolVar>() {
+      return absl::Span<const BoolVar>(vec);
+    }
+};
+
+template<>
+inline
+BoolVarSpan from_ruby<BoolVarSpan>(Object x)
+{
+  return BoolVarSpan(x);
+}
+
+void init_constraint(Rice::Module& m) {
+  rb_cSatIntVar = Rice::define_class_under<IntVar>(m, "SatIntVar")
+    .define_method("name", &IntVar::Name);
+
+  Rice::define_class_under<IntervalVar>(m, "SatIntervalVar")
+    .define_method("name", &IntervalVar::Name);
+
+  Rice::define_class_under<Constraint>(m, "SatConstraint")
+    .define_method(
+      "only_enforce_if",
+      *[](Constraint& self, Object literal) {
+        if (literal.is_a(rb_cSatIntVar)) {
+          return self.OnlyEnforceIf(from_ruby<IntVar>(literal).ToBoolVar());
+        } else if (literal.is_a(rb_cArray)) {
+          // TODO support IntVarSpan
+          return self.OnlyEnforceIf(from_ruby<BoolVarSpan>(literal));
+        } else {
+          return self.OnlyEnforceIf(from_ruby<BoolVar>(literal));
+        }
+      });
+
+  rb_cBoolVar = Rice::define_class_under<BoolVar>(m, "BoolVar")
+    .define_method("name", &BoolVar::Name)
+    .define_method("index", &BoolVar::index)
+    .define_method("not", &BoolVar::Not)
+    .define_method(
+      "inspect",
+      *[](BoolVar& self) {
+        String name(self.Name());
+        return "#<ORTools::BoolVar @name=" + name.inspect().str() + ">";
+      });
+
+  Rice::define_class_under<SatParameters>(m, "SatParameters")
+    .define_constructor(Rice::Constructor<SatParameters>())
+    .define_method("max_time_in_seconds=",
+    *[](SatParameters& self, double value) {
+      self.set_max_time_in_seconds(value);
+    });
+
+  Rice::define_class_under<CpModelBuilder>(m, "CpModel")
+    .define_constructor(Rice::Constructor<CpModelBuilder>())
+    .define_method(
+      "new_int_var",
+      *[](CpModelBuilder& self, int64_t start, int64_t end, const std::string& name) {
+        const operations_research::Domain domain(start, end);
+        return self.NewIntVar(domain).WithName(name);
+      })
+    .define_method(
+      "new_bool_var",
+      *[](CpModelBuilder& self, const std::string& name) {
+        return self.NewBoolVar().WithName(name);
+      })
+    .define_method(
+      "new_constant",
+      *[](CpModelBuilder& self, int64_t value) {
+        return self.NewConstant(value);
+      })
+    .define_method(
+      "true_var",
+      *[](CpModelBuilder& self) {
+        return self.TrueVar();
+      })
+    .define_method(
+      "false_var",
+      *[](CpModelBuilder& self) {
+        return self.FalseVar();
+      })
+    .define_method(
+      "new_interval_var",
+      *[](CpModelBuilder& self, IntVar start, IntVar size, IntVar end, const std::string& name) {
+        return self.NewIntervalVar(start, size, end).WithName(name);
+      })
+    .define_method(
+      "new_optional_interval_var",
+      *[](CpModelBuilder& self, IntVar start, IntVar size, IntVar end, BoolVar presence, const std::string& name) {
+        return self.NewOptionalIntervalVar(start, size, end, presence).WithName(name);
+      })
+    .define_method(
+      "add_bool_or",
+      *[](CpModelBuilder& self, BoolVarSpan literals) {
+        return self.AddBoolOr(literals);
+      })
+    .define_method(
+      "add_bool_and",
+      *[](CpModelBuilder& self, BoolVarSpan literals) {
+        return self.AddBoolAnd(literals);
+      })
+    .define_method(
+      "add_bool_xor",
+      *[](CpModelBuilder& self, BoolVarSpan literals) {
+        return self.AddBoolXor(literals);
+      })
+    .define_method(
+      "add_implication",
+      *[](CpModelBuilder& self, BoolVar a, BoolVar b) {
+        return self.AddImplication(a, b);
+      })
+    .define_method(
+      "add_equality",
+      *[](CpModelBuilder& self, LinearExpr x, LinearExpr y) {
+        return self.AddEquality(x, y);
+      })
+    .define_method(
+      "add_greater_or_equal",
+      *[](CpModelBuilder& self, LinearExpr x, LinearExpr y) {
+        return self.AddGreaterOrEqual(x, y);
+      })
+    .define_method(
+      "add_greater_than",
+      *[](CpModelBuilder& self, LinearExpr x, LinearExpr y) {
+        return self.AddGreaterThan(x, y);
+      })
+    .define_method(
+      "add_less_or_equal",
+      *[](CpModelBuilder& self, LinearExpr x, LinearExpr y) {
+        return self.AddLessOrEqual(x, y);
+      })
+    .define_method(
+      "add_less_than",
+      *[](CpModelBuilder& self, LinearExpr x, LinearExpr y) {
+        return self.AddLessThan(x, y);
+      })
+    // TODO add domain
+    // .define_method(
+    //   "add_linear_constraint",
+    //   *[](CpModelBuilder& self, LinearExpr expr, Domain domain) {
+    //     return self.AddLinearConstraint(expr, domain);
+    //   })
+    .define_method(
+      "add_not_equal",
+      *[](CpModelBuilder& self, LinearExpr x, LinearExpr y) {
+        return self.AddNotEqual(x, y);
+      })
+    .define_method(
+      "add_all_different",
+      *[](CpModelBuilder& self, IntVarSpan vars) {
+        return self.AddAllDifferent(vars);
+      })
+    .define_method(
+      "add_inverse_constraint",
+      *[](CpModelBuilder& self, IntVarSpan variables, IntVarSpan inverse_variables) {
+        return self.AddInverseConstraint(variables, inverse_variables);
+      })
+    .define_method(
+      "add_min_equality",
+      *[](CpModelBuilder& self, IntVar target, IntVarSpan vars) {
+        return self.AddMinEquality(target, vars);
+      })
+    .define_method(
+      "add_lin_min_equality",
+      *[](CpModelBuilder& self, LinearExpr target, LinearExprSpan exprs) {
+        return self.AddLinMinEquality(target, exprs);
+      })
+    .define_method(
+      "add_max_equality",
+      *[](CpModelBuilder& self, IntVar target, IntVarSpan vars) {
+        return self.AddMaxEquality(target, vars);
+      })
+    .define_method(
+      "add_lin_max_equality",
+      *[](CpModelBuilder& self, LinearExpr target, LinearExprSpan exprs) {
+        return self.AddLinMaxEquality(target, exprs);
+      })
+    .define_method(
+      "add_division_equality",
+      *[](CpModelBuilder& self, IntVar target, IntVar numerator, IntVar denominator) {
+        return self.AddDivisionEquality(target, numerator, denominator);
+      })
+    .define_method(
+      "add_abs_equality",
+      *[](CpModelBuilder& self, IntVar target, IntVar var) {
+        return self.AddAbsEquality(target, var);
+      })
+    .define_method(
+      "add_modulo_equality",
+      *[](CpModelBuilder& self, IntVar target, IntVar var, IntVar mod) {
+        return self.AddModuloEquality(target, var, mod);
+      })
+    .define_method(
+      "add_product_equality",
+      *[](CpModelBuilder& self, IntVar target, IntVarSpan vars) {
+        return self.AddProductEquality(target, vars);
+      })
+    .define_method(
+      "add_no_overlap",
+      *[](CpModelBuilder& self, IntervalVarSpan vars) {
+        return self.AddNoOverlap(vars);
+      })
+    .define_method(
+      "maximize",
+      *[](CpModelBuilder& self, LinearExpr expr) {
+        self.Maximize(expr);
+      })
+    .define_method(
+      "minimize",
+      *[](CpModelBuilder& self, LinearExpr expr) {
+        self.Minimize(expr);
+      })
+    .define_method(
+      "scale_objective_by",
+      *[](CpModelBuilder& self, double scaling) {
+        self.ScaleObjectiveBy(scaling);
+      })
+    .define_method(
+      "add_hint",
+      *[](CpModelBuilder& self, IntVar var, int64_t value) {
+        self.AddHint(var, value);
+      })
+    .define_method(
+      "clear_hints",
+      *[](CpModelBuilder& self) {
+        self.ClearHints();
+      })
+    .define_method(
+      "add_assumption",
+      *[](CpModelBuilder& self, BoolVar lit) {
+        self.AddAssumption(lit);
+      })
+    .define_method(
+      "add_assumptions",
+      *[](CpModelBuilder& self, BoolVarSpan literals) {
+        self.AddAssumptions(literals);
+      })
+    .define_method(
+      "clear_assumptions",
+      *[](CpModelBuilder& self) {
+        self.ClearAssumptions();
+      })
+    .define_method(
+      "to_s",
+      *[](CpModelBuilder& self) {
+        std::string proto_string;
+        google::protobuf::TextFormat::PrintToString(self.Proto(), &proto_string);
+        return proto_string;
+      });
+
+  Rice::define_class_under(m, "CpSolver")
+    .define_method(
+      "_solve_with_observer",
+      *[](Object self, CpModelBuilder& model, SatParameters& parameters, Object callback, bool all_solutions) {
+        Model m;
+
+        if (all_solutions) {
+          // set parameters for SearchForAllSolutions
+          parameters.set_enumerate_all_solutions(true);
+        }
+        m.Add(NewSatParameters(parameters));
+
+        m.Add(NewFeasibleSolutionObserver(
+          [callback](const CpSolverResponse& r) {
+            // TODO find a better way to do this
+            callback.call("response=", r);
+            callback.call("on_solution_callback");
+          })
+        );
+        return SolveCpModel(model.Build(), &m);
+      })
+    .define_method(
+      "_solve",
+      *[](Object self, CpModelBuilder& model, SatParameters& parameters) {
+        Model m;
+        m.Add(NewSatParameters(parameters));
+        return SolveCpModel(model.Build(), &m);
+      })
+    .define_method(
+      "_solution_integer_value",
+      *[](Object self, CpSolverResponse& response, IntVar& x) {
+        return SolutionIntegerValue(response, x);
+      })
+    .define_method(
+      "_solution_boolean_value",
+      *[](Object self, CpSolverResponse& response, BoolVar& x) {
+        return SolutionBooleanValue(response, x);
+      });
+
+  Rice::define_class_under<CpSolverResponse>(m, "CpSolverResponse")
+    .define_method("objective_value", &CpSolverResponse::objective_value)
+    .define_method("num_conflicts", &CpSolverResponse::num_conflicts)
+    .define_method("num_branches", &CpSolverResponse::num_branches)
+    .define_method("wall_time", &CpSolverResponse::wall_time)
+    .define_method(
+      "solution_integer_value",
+      *[](CpSolverResponse& self, IntVar& x) {
+        LinearExpr expr(x);
+        return SolutionIntegerValue(self, expr);
+      })
+    .define_method("solution_boolean_value", &SolutionBooleanValue)
+    .define_method(
+      "status",
+      *[](CpSolverResponse& self) {
+        auto status = self.status();
+
+        if (status == CpSolverStatus::OPTIMAL) {
+          return Symbol("optimal");
+        } else if (status == CpSolverStatus::FEASIBLE) {
+          return Symbol("feasible");
+        } else if (status == CpSolverStatus::INFEASIBLE) {
+          return Symbol("infeasible");
+        } else if (status == CpSolverStatus::MODEL_INVALID) {
+          return Symbol("model_invalid");
+        } else if (status == CpSolverStatus::UNKNOWN) {
+          return Symbol("unknown");
+        } else {
+          throw std::runtime_error("Unknown solver status");
+        }
+      })
+    .define_method(
+      "sufficient_assumptions_for_infeasibility",
+      *[](CpSolverResponse& self) {
+        auto a = Array();
+        auto assumptions = self.sufficient_assumptions_for_infeasibility();
+        for (auto const& v : assumptions) {
+          a.push(v);
+        }
+        return a;
+      });
+}