opt-rb 0.1.0 → 0.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 300747a1f97ca67851cda0264f649d1e7e0a4f126609c658efb4f3fc509f14f7
-  data.tar.gz: 87fdcba2ec49277f3491f9a61b04ae74107b1df77d13c81392f63cf58989194f
+  metadata.gz: 8fbd52b19c465db0b32b7ef39a80c581904de6cfd257ca20f28c04fb0ac50b31
+  data.tar.gz: 1dea622c2248537c91c9d66ec80b749a60485195acbc25383a6377b53f4c7086
 SHA512:
-  metadata.gz: 1c850b8f467bb5b0b9f1fc7a960db0232a2cddd26a129fa8db072b7701e4ccd6ddb0fa828042f16bb4d6263e05c2ea856796648ad7f2619971bca559afc97749
-  data.tar.gz: 99e1b9798bb609fe56b844f21df02578e44984885076a631590569c47fc979c51113098c57f667ffccfbc4f86d91784bdd98eec7fc23ee9e5c38a9cc65ccabf3
+  metadata.gz: 405e7e4a8109885cb1c97e0fc9380e346ae9876439376635fa1d3cd0f1cdf293ca005e6bc31b7af49913c2a52ade0972690b6b155e13db07f4919eda12125625
+  data.tar.gz: 6d3b768fefe6388cf3b21dcb499e1c80e7d2eb4c5f61febd2ce72ec1f4ea7cae790267620a452ca805684d4434fa35bcada98d9ad89865cac2f099d063fe8255

data/CHANGELOG.md

@@ -1,3 +1,7 @@
+## 0.1.1 (2023-04-16)
+
+- Added support for semi-continuous and semi-integer variables
+
 ## 0.1.0 (2023-01-25)
 
 - First release

data/README.md

@@ -2,7 +2,7 @@
 
 :fire: Convex optimization for Ruby
 
-Supports [Cbc](https://github.com/ankane/cbc-ruby), [Clp](https://github.com/ankane/clp-ruby), [GLOP](https://github.com/ankane/or-tools-ruby), [GLPK](https://github.com/ankane/glpk-ruby), [HiGHS](https://github.com/ankane/highs-ruby), [OSQP](https://github.com/ankane/osqp-ruby), and [SCS](https://github.com/ankane/scs-ruby)
+Supports Cbc, Clp, GLOP, GLPK, HiGHS, OSQP, and SCS
 
 [![Build Status](https://github.com/ankane/opt/workflows/build/badge.svg?branch=master)](https://github.com/ankane/opt/actions)
 
@@ -14,15 +14,19 @@ Add this line to your application’s Gemfile:
 gem "opt-rb"
 ```
 
-And install a solver:
+And install one or more solvers based on your problem types:
 
-- [Cbc](https://github.com/ankane/cbc-ruby#installation)
-- [Clp](https://github.com/ankane/clp-ruby#installation)
-- [GLOP](https://github.com/ankane/or-tools-ruby#installation)
-- [GLPK](https://github.com/ankane/glpk-ruby#installation)
-- [HiGHS](https://github.com/ankane/highs-ruby#installation)
-- [OSQP](https://github.com/ankane/osqp-ruby#installation)
-- [SCS](https://github.com/ankane/scs-ruby#installation)
+Solver | LP | QP | MIP | License
+--- | --- | --- | --- | ---
+[Cbc](https://github.com/ankane/cbc-ruby) | ✓ | | ✓ | EPL-2.0
+[Clp](https://github.com/ankane/clp-ruby) | ✓ | | | EPL-2.0
+[GLOP](https://github.com/ankane/or-tools-ruby) | ✓ | | | Apache-2.0
+[GLPK](https://github.com/ankane/glpk-ruby) | ✓ | | ✓ | GPL-3.0-or-later
+[HiGHS](https://github.com/ankane/highs-ruby) | ✓ | ✓ | ✓ | MIT
+[OSQP](https://github.com/ankane/osqp-ruby) | ✓ | ✓ | | Apache-2.0
+[SCS](https://github.com/ankane/scs-ruby) | ✓ | * | | MIT
+
+\* supports, but not implemented yet
 
 ## Getting Started
 
@@ -46,29 +50,34 @@ Get the value of a variable
 x1.value
 ```
 
+QP
+
+```ruby
+prob.minimize(x1 * x1)
+```
+
 MIP
 
 ```ruby
 x1 = Opt::Integer.new(0.., "x1")
-x1 = Opt::Binary.new("x1")
+x2 = Opt::Binary.new("x2")
 ```
 
-## Problem Types
-
-Solver | LP | QP | MIP | License
---- | --- | --- | --- | ---
-Cbc | ✓ | | ✓ | EPL-2.0
-Clp | ✓ | | | EPL-2.0
-GLOP | ✓ | | | Apache-2.0
-GLPK | ✓ | | ✓ | GPL-3.0-or-later
-HiGHS | ✓ | ✓ | ✓ | MIT
-OSQP | ✓ | ✓ | | Apache-2.0
-SCS | ✓ | * | | MIT
+MIP with semi-continuous variables - *HiGHS only at the moment* [unreleased]
 
-\* supports, but not implemented yet
+```ruby
+x1 = Opt::SemiContinuous.new(2.., "x1")
+x2 = Opt::SemiInteger.new(2.., "x2")
+```
 
 ## Reference
 
+Specify the solver
+
+```ruby
+prob.solve(solver: :cbc)
+```
+
 Enable verbose logging
 
 ```ruby

data/lib/opt/problem.rb

@@ -27,7 +27,10 @@ module Opt
 
       vars = self.vars
       raise Error, "No variables" if vars.empty?
-      type = vars.any? { |v| v.is_a?(Integer) } ? :mip : :lp
+      has_semi_continuous_var = vars.any? { |v| v.is_a?(SemiContinuous) }
+      has_semi_integer_var = vars.any? { |v| v.is_a?(SemiInteger) }
+      has_integer_var = vars.any? { |v| v.is_a?(Integer) }
+      type = has_semi_continuous_var || has_semi_integer_var || has_integer_var ? :mip : :lp
       quadratic = @indexed_objective.any? { |k, _| k.is_a?(Array) }
 
       if quadratic
@@ -44,6 +47,9 @@ module Opt
       solver_cls = Opt.solvers.fetch(solver)
       raise Error, "Solver does not support #{type}" unless solver_cls.supports_type?(type)
 
+      raise Error, "Solver does not support semi-continuous variables" if has_semi_continuous_var && !solver_cls.supports_semi_continuous_variables?
+      raise Error, "Solver does not support semi-integer variables" if has_semi_integer_var && !solver_cls.supports_semi_integer_variables?
+
       col_lower = []
       col_upper = []
       obj = []
@@ -54,7 +60,8 @@ module Opt
         col_lower << (var.bounds.begin || -Float::INFINITY)
         upper = var.bounds.end
         if upper && var.bounds.exclude_end?
-          if var.is_a?(Integer)
+          case var
+          when Integer, SemiInteger
             upper -= 1
           else
             upper -= Float::EPSILON
@@ -117,7 +124,7 @@ module Opt
             case a
             when Binary
               b.round != 0
-            when Integer
+            when Integer, SemiInteger
               b.round
             else
               b
@@ -149,6 +156,9 @@ module Opt
           else
             "#{var.name} #{end_op} #{bounds.end}"
           end
+        if var.is_a?(SemiContinuous) || var.is_a?(SemiInteger)
+          var_str = "#{var_str} or #{var.name} = 0"
+        end
         str << "  #{var_str}\n"
       end
       str

data/lib/opt/semi_continuous.rb

@@ -0,0 +1,4 @@
+module Opt
+  class SemiContinuous < Variable
+  end
+end

data/lib/opt/semi_integer.rb

@@ -0,0 +1,4 @@
+module Opt
+  class SemiInteger < Variable
+  end
+end

data/lib/opt/solvers/abstract_solver.rb

@@ -4,6 +4,14 @@ module Opt
       def self.supports_type?(type)
         supported_types.include?(type)
       end
+
+      def self.supports_semi_continuous_variables?
+        false
+      end
+
+      def self.supports_semi_integer_variables?
+        false
+      end
     end
   end
 end

data/lib/opt/solvers/highs_solver.rb

@@ -19,7 +19,7 @@ module Opt
               a_index: index,
               a_value: value,
               offset: offset,
-              integrality: vars.map { |v| v.is_a?(Integer) ? 1 : 0 }
+              integrality: vars.map { |var| integrality(var) }
             )
           when :qp
             q_start = []
@@ -80,6 +80,17 @@ module Opt
         }
       end
 
+      private
+
+      def integrality(var)
+        case var
+        when SemiInteger then 3
+        when SemiContinuous then 2
+        when Integer then 1
+        else 0
+        end
+      end
+
       def self.available?
         defined?(Highs)
       end
@@ -87,6 +98,14 @@ module Opt
       def self.supported_types
         [:lp, :qp, :mip]
       end
+
+      def self.supports_semi_continuous_variables?
+        true
+      end
+
+      def self.supports_semi_integer_variables?
+        true
+      end
     end
   end
 end

data/lib/opt/version.rb

@@ -1,3 +1,3 @@
 module Opt
-  VERSION = "0.1.0"
+  VERSION = "0.1.1"
 end

data/lib/opt.rb

@@ -5,6 +5,8 @@ require_relative "opt/constant"
 require_relative "opt/variable"
 require_relative "opt/integer"
 require_relative "opt/binary"
+require_relative "opt/semi_continuous"
+require_relative "opt/semi_integer"
 require_relative "opt/problem"
 require_relative "opt/product"
 require_relative "opt/version"

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: opt-rb
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.1
 platform: ruby
 authors:
 - Andrew Kane
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-01-26 00:00:00.000000000 Z
+date: 2023-04-17 00:00:00.000000000 Z
 dependencies: []
 description:
 email: andrew@ankane.org
@@ -28,6 +28,8 @@ files:
 - lib/opt/integer.rb
 - lib/opt/problem.rb
 - lib/opt/product.rb
+- lib/opt/semi_continuous.rb
+- lib/opt/semi_integer.rb
 - lib/opt/solvers/abstract_solver.rb
 - lib/opt/solvers/cbc_solver.rb
 - lib/opt/solvers/clp_solver.rb
@@ -57,7 +59,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.4.1
+rubygems_version: 3.4.10
 signing_key:
 specification_version: 4
 summary: Convex optimization for Ruby