or-tools 0.5.3 → 0.6.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: c16581924f813458dc535d2a8804bf02a7f0a605fac108d14894f13ed7af984a
-  data.tar.gz: 28bd124da203c4dd0ac50ee86aca3deb7ba283a1cbd711521985972a02bdcaa1
+  metadata.gz: c0e0f455741da72e3a215287df4ef1858893fb31d0b4be40db6ef9ab025e339c
+  data.tar.gz: 9291933eee08f800f48f3d83ca4f2d0fc71efb0a328134ba9667ca2ac8efb185
 SHA512:
-  metadata.gz: 0612e5bfcf96ba9ec2591dab7bbfe175625f874ff89e87a168d771a0f81948cb4d8d29e31de0487470eb3b993ef53b793ac25a0b078c5cc54385d4cd3bb159f9
-  data.tar.gz: dd548021defe7c1249ecee70790cb29f4c85a5ec04cace94cc9bfa0192b24d68e20040a7dd355b148f75f443bf882cc41ec3ea6613785a24aeb4a37674ca8017
+  metadata.gz: 61ccc7c5f3a6c3cdf1e3dffb1ee6572d0b8cde09d4684d50353efd6f2efcb93ca4ba940343f994a3bf65992e518331bd62a02ef74ecc52f2427e12350ea7a430
+  data.tar.gz: 589423f5b52c872b453569400ff1fd60bb1fcd3b0035e70aadbb47eedefec9b7f4187a5442893e50608529f836d2009bd77333aae3f64ce1b9d700df6fe020e7

data/CHANGELOG.md

@@ -1,3 +1,24 @@
+## 0.6.2 (2022-02-09)
+
+- Fixed segfaults with `Solver`
+
+## 0.6.1 (2022-01-22)
+
+- Added installation instructions for Mac ARM
+- Removed dependency on `lsb_release` for binary installation on Linux
+
+## 0.6.0 (2021-12-16)
+
+- Updated OR-Tools to 9.2
+- Renamed `add_product_equality` to `add_multiplication_equality`
+- Removed `scale_objective_by`
+
+## 0.5.4 (2021-10-01)
+
+- Updated OR-Tools to 9.1
+- Added binary installation for Debian 11
+- Deprecated `solve_with_solution_callback` and `search_for_all_solutions`
+
 ## 0.5.3 (2021-08-02)
 
 - Added more methods to `IntVar`, `IntervalVar`, and `Constraint`

data/NOTICE.txt

@@ -1,5 +1,5 @@
-Copyright 2010-2018 Google LLC
-Copyright 2020-2021 Andrew Kane
+Copyright 2010-2021 Google LLC
+Copyright 2020-2022 Andrew Kane
 
 Licensed under the Apache License, Version 2.0 (the "License");
 you may not use this file except in compliance with the License.

data/README.md

@@ -1,18 +1,18 @@
-# OR-Tools
+# OR-Tools Ruby
 
 [OR-Tools](https://github.com/google/or-tools) - operations research tools - for Ruby
 
-[![Build Status](https://github.com/ankane/or-tools/workflows/build/badge.svg?branch=master)](https://github.com/ankane/or-tools/actions)
+[![Build Status](https://github.com/ankane/or-tools-ruby/workflows/build/badge.svg?branch=master)](https://github.com/ankane/or-tools-ruby/actions)
 
 ## Installation
 
 Add this line to your application’s Gemfile:
 
 ```ruby
-gem 'or-tools'
+gem "or-tools"
 ```
 
-Installation can take a few minutes as OR-Tools downloads and builds.
+Installation can take a few minutes as OR-Tools downloads and builds. For Mac ARM, also follow [these instructions](#additional-instructions).
 
 ## Higher Level Interfaces
 
@@ -2139,7 +2139,7 @@ model.add(seats[[0, 0]] == 1)
 # Solve model
 solver = ORTools::CpSolver.new
 solution_printer = WeddingChartPrinter.new(seats, names, num_tables, num_guests)
-solver.solve_with_solution_callback(model, solution_printer)
+solver.solve(model, solution_printer)
 
 puts "Statistics"
 puts "  - conflicts    : %i" % solver.num_conflicts
@@ -2201,24 +2201,35 @@ possible_tables.each do |table|
 end
 ```
 
+## Additional Instructions
+
+### Mac ARM
+
+ARM binaries are not available for Mac yet, so use Homebrew to install OR-Tools before installing the gem.
+
+```sh
+brew install or-tools
+bundle config build.or-tools --with-or-tools-dir=/opt/homebrew
+```
+
 ## History
 
-View the [changelog](https://github.com/ankane/or-tools/blob/master/CHANGELOG.md)
+View the [changelog](https://github.com/ankane/or-tools-ruby/blob/master/CHANGELOG.md)
 
 ## Contributing
 
 Everyone is encouraged to help improve this project. Here are a few ways you can help:
 
-- [Report bugs](https://github.com/ankane/or-tools/issues)
-- Fix bugs and [submit pull requests](https://github.com/ankane/or-tools/pulls)
+- [Report bugs](https://github.com/ankane/or-tools-ruby/issues)
+- Fix bugs and [submit pull requests](https://github.com/ankane/or-tools-ruby/pulls)
 - Write, clarify, or fix documentation
 - Suggest or add new features
 
 To get started with development:
 
 ```sh
-git clone https://github.com/ankane/or-tools.git
-cd or-tools
+git clone https://github.com/ankane/or-tools-ruby.git
+cd or-tools-ruby
 bundle install
 bundle exec rake compile
 bundle exec rake test

data/ext/or-tools/constraint.cpp

@@ -131,10 +131,21 @@ void init_constraint(Rice::Module& m) {
 
   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);
-    });
+    .define_method(
+      "max_time_in_seconds=",
+      [](SatParameters& self, double value) {
+        self.set_max_time_in_seconds(value);
+      })
+    .define_method(
+      "enumerate_all_solutions=",
+      [](SatParameters& self, bool value) {
+        self.set_enumerate_all_solutions(value);
+      })
+    .define_method(
+      "enumerate_all_solutions",
+      [](SatParameters& self) {
+        return self.enumerate_all_solutions();
+      });
 
   Rice::define_class_under<CpModelBuilder>(m, "CpModel")
     .define_constructor(Rice::Constructor<CpModelBuilder>())
@@ -276,9 +287,9 @@ void init_constraint(Rice::Module& m) {
         return self.AddModuloEquality(target, var, mod);
       })
     .define_method(
-      "add_product_equality",
+      "add_multiplication_equality",
       [](CpModelBuilder& self, IntVar target, std::vector<IntVar> vars) {
-        return self.AddProductEquality(target, vars);
+        return self.AddMultiplicationEquality(target, vars);
       })
     .define_method(
       "add_no_overlap",
@@ -295,11 +306,6 @@ void init_constraint(Rice::Module& m) {
       [](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) {
@@ -377,29 +383,23 @@ void init_constraint(Rice::Module& m) {
 
   Rice::define_class_under(m, "CpSolver")
     .define_method(
-      "_solve_with_observer",
-      [](Object self, CpModelBuilder& model, SatParameters& parameters, Object callback, bool all_solutions) {
+      "_solve",
+      [](Object self, CpModelBuilder& model, SatParameters& parameters, Object callback) {
         Model m;
 
-        if (all_solutions) {
-          // set parameters for SearchForAllSolutions
-          parameters.set_enumerate_all_solutions(true);
+        if (!callback.is_nil()) {
+          // TODO figure out how to use callback with multiple cores
+          parameters.set_num_search_workers(1);
+
+          m.Add(NewFeasibleSolutionObserver(
+            [callback](const CpSolverResponse& r) {
+              // TODO find a better way to do this
+              callback.call("response=", r);
+              callback.call("on_solution_callback");
+            })
+          );
         }
-        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);
       })

data/ext/or-tools/linear.cpp

@@ -46,86 +46,14 @@ namespace Rice::detail
 
 void init_linear(Rice::Module& m) {
   Rice::define_class_under<LinearRange>(m, "LinearRange");
+
+  // TODO remove in 0.7.0
   auto rb_cLinearExpr = Rice::define_class_under<LinearExpr>(m, "LinearExpr");
+  rb_cLinearExpr.define_constructor(Rice::Constructor<LinearExpr>());
 
-  Rice::define_class_under<MPVariable>(m, "MPVariable")
+  Rice::define_class_under<MPVariable, LinearExpr>(m, "MPVariable")
     .define_method("name", &MPVariable::name)
-    .define_method("solution_value", &MPVariable::solution_value)
-    .define_method(
-      "+",
-      [](MPVariable& self, LinearExpr& other) {
-        LinearExpr s(&self);
-        return s + other;
-      })
-    .define_method(
-      "-",
-      [](MPVariable& self, LinearExpr& other) {
-        LinearExpr s(&self);
-        return s - other;
-      })
-    .define_method(
-      "*",
-      [](MPVariable& self, double other) {
-        LinearExpr s(&self);
-        return s * other;
-      })
-    .define_method(
-      "inspect",
-      [](MPVariable& self) {
-        return "#<ORTools::MPVariable @name=\"" + self.name() + "\">";
-      });
-
-  rb_cLinearExpr
-    .define_constructor(Rice::Constructor<LinearExpr>())
-    .define_method(
-      "_add_linear_expr",
-      [](LinearExpr& self, LinearExpr& other) {
-        return self + other;
-      })
-    .define_method(
-      "_add_mp_variable",
-      [](LinearExpr& self, MPVariable &other) {
-        LinearExpr o(&other);
-        return self + o;
-      })
-    .define_method(
-      "_gte_double",
-      [](LinearExpr& self, double other) {
-        LinearExpr o(other);
-        return self >= o;
-      })
-    .define_method(
-      "_gte_linear_expr",
-      [](LinearExpr& self, LinearExpr& other) {
-        return self >= other;
-      })
-    .define_method(
-      "_lte_double",
-      [](LinearExpr& self, double other) {
-        LinearExpr o(other);
-        return self <= o;
-      })
-    .define_method(
-      "_lte_linear_expr",
-      [](LinearExpr& self, LinearExpr& other) {
-        return self <= other;
-      })
-    .define_method(
-      "==",
-      [](LinearExpr& self, double other) {
-        LinearExpr o(other);
-        return self == o;
-      })
-    .define_method(
-      "to_s",
-      [](LinearExpr& self) {
-        return self.ToString();
-      })
-    .define_method(
-      "inspect",
-      [](LinearExpr& self) {
-        return "#<ORTools::LinearExpr \"" + self.ToString() + "\">";
-      });
+    .define_method("solution_value", &MPVariable::solution_value);
 
   Rice::define_class_under<MPConstraint>(m, "MPConstraint")
     .define_method("set_coefficient", &MPConstraint::SetCoefficient);
@@ -133,7 +61,8 @@ void init_linear(Rice::Module& m) {
   Rice::define_class_under<MPObjective>(m, "MPObjective")
     .define_method("value", &MPObjective::Value)
     .define_method("set_coefficient", &MPObjective::SetCoefficient)
-    .define_method("set_maximization", &MPObjective::SetMaximization);
+    .define_method("set_maximization", &MPObjective::SetMaximization)
+    .define_method("set_minimization", &MPObjective::SetMinimization);
 
   Rice::define_class_under<MPSolver>(m, "Solver")
     .define_constructor(Rice::Constructor<MPSolver, std::string, MPSolver::OptimizationProblemType>())
@@ -155,21 +84,6 @@ void init_linear(Rice::Module& m) {
     .define_method("iterations", &MPSolver::iterations)
     .define_method("nodes", &MPSolver::nodes)
     .define_method("objective", &MPSolver::MutableObjective)
-    .define_method(
-      "maximize",
-      [](MPSolver& self, LinearExpr& expr) {
-        return self.MutableObjective()->MaximizeLinearExpr(expr);
-      })
-    .define_method(
-      "minimize",
-      [](MPSolver& self, LinearExpr& expr) {
-        return self.MutableObjective()->MinimizeLinearExpr(expr);
-      })
-    .define_method(
-      "add",
-      [](MPSolver& self, const LinearRange& range) {
-        return self.MakeRowConstraint(range);
-      })
     .define_method(
       "constraint",
       [](MPSolver& self, double lb, double ub) {

data/ext/or-tools/vendor.rb

@@ -1,31 +1,54 @@
+require "csv"
 require "digest"
 require "fileutils"
 require "net/http"
 require "tmpdir"
 
-version = "9.0.9048"
+version = "9.2.9972"
 
 if RbConfig::CONFIG["host_os"] =~ /darwin/i
-  filename = "or-tools_MacOsX-11.2.3_v#{version}.tar.gz"
-  checksum = "adf73a00d4ec49558b67be5ce3cfc8f30268da2253b35feb11d0d40700550bf6"
+  if RbConfig::CONFIG["host_cpu"] =~ /arm|aarch64/i
+    raise <<~MSG
+      Binary installation not available for this platform: Mac ARM
+
+      Run:
+      brew install or-tools
+      bundle config build.or-tools --with-or-tools-dir=/opt/homebrew
+
+    MSG
+  else
+    filename = "or-tools_MacOsX-12.0.1_v#{version}.tar.gz"
+    checksum = "796791a8ef84507d62e193e647cccb1c7725dae4f1474476e1777fe4a44ee3e0"
+  end
 else
-  os = %x[lsb_release -is].chomp rescue nil
-  os_version = %x[lsb_release -rs].chomp rescue nil
-  if os == "Ubuntu" && os_version == "20.04"
-    filename = "or-tools_ubuntu-20.04_v#{version}.tar.gz"
-    checksum = "5565343c1c310d2885a40ce850ae7e3468299b3fee97ae8eed8425ce06bd4960"
-  elsif os == "Ubuntu" && os_version == "18.04"
-    filename = "or-tools_ubuntu-18.04_v#{version}.tar.gz"
-    checksum = "08cf548d0179f7fa814bb7458be94cd1b8a3de14985e6a9faf6118a1d8571539"
-  elsif os == "Debian" && os_version == "10"
-    filename = "or-tools_debian-10_v#{version}.tar.gz"
-    checksum = "063fb1d8765ae23b0bb25b9c561e904532713416fe0458f7db45a0f72190eb50"
-  elsif os == "CentOS" && os_version == "8"
-    filename = "or-tools_centos-8_v#{version}.tar.gz"
-    checksum = "c98212ed4fc699d8ae70c1f53cd1d8dacd28e52970336fab5b86dedf7406f215"
-  elsif os == "CentOS" && os_version == "7"
-    filename = "or-tools_centos-7_v#{version}.tar.gz"
-    checksum = "b992bda4614bbc703583b0e9edcd2ade54bacfb9909399b20c8aa95ff7197d68"
+  # try /etc/os-release with fallback to /usr/lib/os-release
+  # https://www.freedesktop.org/software/systemd/man/os-release.html
+  os_filename = File.exist?("/etc/os-release") ? "/etc/os-release" : "/usr/lib/os-release"
+
+  # for safety, parse rather than source
+  os_info = CSV.read(os_filename, col_sep: "=").to_h rescue {}
+
+  os = os_info["ID"]
+  os_version = os_info["VERSION_ID"]
+
+  if os == "ubuntu" && os_version == "20.04"
+    filename = "or-tools_amd64_ubuntu-20.04_v#{version}.tar.gz"
+    checksum = "985e3036eaecacfc8a0258ec2ebef429240491577d4e0896d68fc076e65451ec"
+  elsif os == "ubuntu" && os_version == "18.04"
+    filename = "or-tools_amd64_ubuntu-18.04_v#{version}.tar.gz"
+    checksum = "e36406c4fe8c111e1ace0ede9d0787ff0e98f11afd7db9cc074adfd0f55628a6"
+  elsif os == "debian" && os_version == "11"
+    filename = "or-tools_amd64_debian-11_v#{version}.tar.gz"
+    checksum = "bd49ee916213b2140ab255414d35a28f19dff7caf87632309753d3fc553f85dd"
+  elsif os == "debian" && os_version == "10"
+    filename = "or-tools_amd64_debian-10_v#{version}.tar.gz"
+    checksum = "b152fee584f0c8228fe2ff21b74c789870ff9b7064e42ca26305c6b5653f0064"
+  elsif os == "centos" && os_version == "8"
+    filename = "or-tools_amd64_centos-8_v#{version}.tar.gz"
+    checksum = "66ed4bb800acf92c672f7e68acdf4ea27bbfdb17bbddc02f8326cd55a97305f6"
+  elsif os == "centos" && os_version == "7"
+    filename = "or-tools_amd64_centos-7_v#{version}.tar.gz"
+    checksum = "4a5c1b1639a2828cd7e1ba82a574ef37876557b59e8aab8b81811bb750d53035"
   else
     platform =
       if Gem.win_platform?

data/lib/or-tools.rb

@@ -5,17 +5,22 @@ require "or_tools/ext"
 require "or_tools/comparison"
 require "or_tools/comparison_operators"
 require "or_tools/bool_var"
+require "or_tools/constant"
 require "or_tools/cp_model"
 require "or_tools/cp_solver"
 require "or_tools/cp_solver_solution_callback"
 require "or_tools/int_var"
 require "or_tools/knapsack_solver"
+require "or_tools/linear_constraint"
 require "or_tools/linear_expr"
+require "or_tools/mp_variable"
+require "or_tools/product_cst"
 require "or_tools/routing_index_manager"
 require "or_tools/routing_model"
 require "or_tools/sat_linear_expr"
 require "or_tools/sat_int_var"
 require "or_tools/solver"
+require "or_tools/sum_array"
 require "or_tools/version"
 
 # solution printers

data/lib/or_tools/comparison.rb

@@ -8,8 +8,13 @@ module ORTools
       @right = right
     end
 
-    def inspect
+    def to_s
       "#{left.inspect} #{operator} #{right.inspect}"
     end
+
+    # TODO add class
+    def inspect
+      to_s
+    end
   end
 end

data/lib/or_tools/constant.rb

@@ -0,0 +1,23 @@
+module ORTools
+  class Constant < LinearExpr
+    def initialize(val)
+      @val = val
+    end
+
+    def to_s
+      @val.to_s
+    end
+
+    def add_self_to_coeff_map_or_stack(coeffs, multiplier, stack)
+      coeffs[OFFSET_KEY] += @val * multiplier
+    end
+  end
+
+  class FakeMPVariableRepresentingTheConstantOffset
+    def solution_value
+      1
+    end
+  end
+
+  OFFSET_KEY = FakeMPVariableRepresentingTheConstantOffset.new
+end

data/lib/or_tools/cp_solver.rb

@@ -6,8 +6,8 @@ module ORTools
 
     def_delegators :@response, :objective_value, :num_conflicts, :num_branches, :wall_time
 
-    def solve(model)
-      @response = _solve(model, parameters)
+    def solve(model, observer = nil)
+      @response = _solve(model, parameters, observer)
       @response.status
     end
 
@@ -20,13 +20,19 @@ module ORTools
     end
 
     def solve_with_solution_callback(model, observer)
-      @response = _solve_with_observer(model, parameters, observer, false)
-      @response.status
+      warn "[or-tools] solve_with_solution_callback is deprecated; use solve(model, callback)"
+      solve(model, observer)
     end
 
     def search_for_all_solutions(model, observer)
-      @response = _solve_with_observer(model, parameters, observer, true)
-      @response.status
+      warn "[or-tools] search_for_all_solutions is deprecated; use solve() with solver.parameters.enumerate_all_solutions = true"
+      previous_value = parameters.enumerate_all_solutions
+      begin
+        parameters.enumerate_all_solutions = true
+        solve(model, observer)
+      ensure
+        parameters.enumerate_all_solutions = previous_value
+      end
     end
 
     def sufficient_assumptions_for_infeasibility

data/lib/or_tools/linear_constraint.rb

@@ -0,0 +1,50 @@
+module ORTools
+  class LinearConstraint
+    attr_reader :expr, :lb, :ub
+
+    def initialize(expr, lb, ub)
+      @expr = expr
+      @lb = lb
+      @ub = ub
+    end
+
+    def to_s
+      if @lb > -Float::INFINITY && @ub < Float::INFINITY
+        if @lb == @ub
+          "#{@expr} == #{@lb}"
+        else
+          "#{@lb} <= #{@expr} <= #{@ub}"
+        end
+      elsif @lb > -Float::INFINITY
+        "#{@expr} >= #{@lb}"
+      elsif @ub < Float::INFINITY
+        "#{@expr} <= #{@ub}"
+      else
+        "Trivial inequality (always true)"
+      end
+    end
+
+    def inspect
+      "#<#{self.class.name} #{to_s}>"
+    end
+
+    def extract(solver)
+      coeffs = @expr.coeffs
+      constant = coeffs.delete(OFFSET_KEY) || 0.0
+      lb = -solver.infinity
+      ub = solver.infinity
+      if @lb > -Float::INFINITY
+        lb = @lb - constant
+      end
+      if @ub < Float::INFINITY
+        ub = @ub - constant
+      end
+
+      constraint = solver.constraint(lb, ub)
+      coeffs.each do |v, c|
+        constraint.set_coefficient(v, c.to_f)
+      end
+      constraint
+    end
+  end
+end

data/lib/or_tools/linear_expr.rb

@@ -1,27 +1,73 @@
 module ORTools
   class LinearExpr
-    def +(other)
-      if other.is_a?(LinearExpr)
-        _add_linear_expr(other)
+    def solution_value
+      coeffs.sum { |var, coeff| var.solution_value * coeff }
+    end
+
+    def coeffs
+      coeffs = Hash.new(0.0)
+      stack = [[1.0, self]]
+      while stack.any?
+        current_multiplier, current_expression = stack.pop
+
+        # skip empty LinearExpr for backwards compatibility
+        next if current_expression.instance_of?(LinearExpr)
+
+        current_expression.add_self_to_coeff_map_or_stack(coeffs, current_multiplier, stack)
+      end
+      coeffs
+    end
+
+    def +(expr)
+      SumArray.new([self, expr])
+    end
+
+    def -(expr)
+      SumArray.new([self, -expr])
+    end
+
+    def *(other)
+      ProductCst.new(self, other)
+    end
+
+    def /(cst)
+      ProductCst.new(self, 1.0 / other)
+    end
+
+    def -@
+      ProductCst.new(self, -1)
+    end
+
+    def ==(arg)
+      if arg.is_a?(Numeric)
+        LinearConstraint.new(self, arg, arg)
       else
-        _add_mp_variable(other)
+        LinearConstraint.new(self - arg, 0.0, 0.0)
       end
     end
 
-    def >=(other)
-      if other.is_a?(LinearExpr)
-        _gte_linear_expr(other)
+    def >=(arg)
+      if arg.is_a?(Numeric)
+        LinearConstraint.new(self, arg, Float::INFINITY)
       else
-        _gte_double(other)
+        LinearConstraint.new(self - arg, 0.0, Float::INFINITY)
       end
     end
 
-    def <=(other)
-      if other.is_a?(LinearExpr)
-        _lte_linear_expr(other)
+    def <=(arg)
+      if arg.is_a?(Numeric)
+        LinearConstraint.new(self, -Float::INFINITY, arg)
       else
-        _lte_double(other)
+        LinearConstraint.new(self - arg, -Float::INFINITY, 0.0)
       end
     end
+
+    def to_s
+      "(empty)"
+    end
+
+    def inspect
+      "#<#{self.class.name} #{to_s}>"
+    end
   end
 end

data/lib/or_tools/mp_variable.rb

@@ -0,0 +1,12 @@
+module ORTools
+  # TODO change to VariableExpr in 0.7.0
+  class MPVariable < LinearExpr
+    def add_self_to_coeff_map_or_stack(coeffs, multiplier, stack)
+      coeffs[self] += multiplier
+    end
+
+    def to_s
+      name
+    end
+  end
+end

data/lib/or_tools/product_cst.rb

@@ -0,0 +1,31 @@
+module ORTools
+  class ProductCst < LinearExpr
+    attr_reader :expr, :coef
+
+    def initialize(expr, coef)
+      @expr = cast_to_lin_exp(expr)
+      # TODO improve message
+      raise TypeError, "expected numeric" unless coef.is_a?(Numeric)
+      @coef = coef
+    end
+
+    def to_s
+      if @coef == -1
+        "-#{@expr}"
+      else
+        "(#{@coef} * #{@expr})"
+      end
+    end
+
+    def add_self_to_coeff_map_or_stack(coeffs, multiplier, stack)
+      current_multiplier = multiplier * @coef
+      if current_multiplier
+        stack << [current_multiplier, @expr]
+      end
+    end
+
+    def cast_to_lin_exp(v)
+      v.is_a?(Numeric) ? Constant.new(v) : v
+    end
+  end
+end

data/lib/or_tools/sat_int_var.rb

@@ -18,9 +18,13 @@ module ORTools
       SatLinearExpr.new([[self, -1]])
     end
 
-    # for now
-    def inspect
+    def to_s
       name
     end
+
+    # TODO add class
+    def inspect
+      to_s
+    end
   end
 end

data/lib/or_tools/sat_linear_expr.rb

@@ -24,7 +24,7 @@ module ORTools
       end
     end
 
-    def inspect
+    def to_s
       vars.map do |v|
         k = v[0]
         k = k.respond_to?(:name) ? k.name : k.inspect
@@ -36,6 +36,11 @@ module ORTools
       end.join(" + ").sub(" + -", " - ")
     end
 
+    # TODO add class
+    def inspect
+      to_s
+    end
+
     private
 
     def add(other, sign)

data/lib/or_tools/solver.rb

@@ -1,7 +1,25 @@
 module ORTools
   class Solver
     def sum(arr)
-      arr.sum(LinearExpr.new)
+      SumArray.new(arr)
+    end
+
+    def add(expr)
+      expr.extract(self)
+    end
+
+    def maximize(expr)
+      expr.coeffs.each do |v, c|
+        objective.set_coefficient(v, c)
+      end
+      objective.set_maximization
+    end
+
+    def minimize(expr)
+      expr.coeffs.each do |v, c|
+        objective.set_coefficient(v, c)
+      end
+      objective.set_minimization
     end
   end
 end

data/lib/or_tools/sum_array.rb

@@ -0,0 +1,23 @@
+module ORTools
+  class SumArray < LinearExpr
+    attr_reader :array
+
+    def initialize(array)
+      @array = array.map { |v| cast_to_lin_exp(v) }
+    end
+
+    def add_self_to_coeff_map_or_stack(coeffs, multiplier, stack)
+      @array.reverse.each do |arg|
+        stack << [multiplier, arg]
+      end
+    end
+
+    def cast_to_lin_exp(v)
+      v.is_a?(Numeric) ? Constant.new(v) : v
+    end
+
+    def to_s
+      "(#{@array.map(&:to_s).join(" + ")})"
+    end
+  end
+end

data/lib/or_tools/version.rb

@@ -1,3 +1,3 @@
 module ORTools
-  VERSION = "0.5.3"
+  VERSION = "0.6.2"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: or-tools
 version: !ruby/object:Gem::Version
-  version: 0.5.3
+  version: 0.6.2
 platform: ruby
 authors:
 - Andrew Kane
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2021-08-03 00:00:00.000000000 Z
+date: 2022-02-09 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rice
@@ -50,13 +50,17 @@ files:
 - lib/or_tools/bool_var.rb
 - lib/or_tools/comparison.rb
 - lib/or_tools/comparison_operators.rb
+- lib/or_tools/constant.rb
 - lib/or_tools/cp_model.rb
 - lib/or_tools/cp_solver.rb
 - lib/or_tools/cp_solver_solution_callback.rb
 - lib/or_tools/int_var.rb
 - lib/or_tools/knapsack_solver.rb
+- lib/or_tools/linear_constraint.rb
 - lib/or_tools/linear_expr.rb
+- lib/or_tools/mp_variable.rb
 - lib/or_tools/objective_solution_printer.rb
+- lib/or_tools/product_cst.rb
 - lib/or_tools/routing_index_manager.rb
 - lib/or_tools/routing_model.rb
 - lib/or_tools/sat_int_var.rb
@@ -64,11 +68,12 @@ files:
 - lib/or_tools/seating.rb
 - lib/or_tools/solver.rb
 - lib/or_tools/sudoku.rb
+- lib/or_tools/sum_array.rb
 - lib/or_tools/tsp.rb
 - lib/or_tools/var_array_and_objective_solution_printer.rb
 - lib/or_tools/var_array_solution_printer.rb
 - lib/or_tools/version.rb
-homepage: https://github.com/ankane/or-tools
+homepage: https://github.com/ankane/or-tools-ruby
 licenses:
 - MIT
 metadata: {}
@@ -87,7 +92,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.2.22
+rubygems_version: 3.3.3
 signing_key:
 specification_version: 4
 summary: Operations research tools for Ruby