osqp 0.2.1 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 1b2906722de1daf0e3f2441edee63d8cec2452012cd9fa91634f37418e66f6fc
-  data.tar.gz: ee0cf11cc8ea7d1676843699265ad27623d237940c756fdb9c49c8e404137b37
+  metadata.gz: b6ed92bcdae0ff5291f26741fc77be7da17b2dc1e4bb6680df7c20516f6b894e
+  data.tar.gz: 7f66c0c4d72a6bb77abb7d3cf4119d79117c92079d7b4dde460d2d952277aed1
 SHA512:
-  metadata.gz: 835c7ac35d791e794b7da902a1a248696860ed09d05e8831d0d4b6c54615758f6d930443e1f4b4f21d0b9d10453882653db60e2bda46b107d342a3cda304e889
-  data.tar.gz: 027ec047ca635e1b9f36377921110444a90226dfa85ebeac44ccab5b1945aac2d37f487f7725ba5604f264551f852392855e087c84afd1c39cd7790278fa3c68
+  metadata.gz: e16b34d395a64988f04c2779d67dfd164162940e92cbd857ae324aa694ee04e05162fd8cb9532a2a7b828d598814bff5fd4cce8601a8b29e3f69913d530085b9
+  data.tar.gz: f8b0dfb4a6657d7398bd3f366c8368bc82233e611890f19e603e21b45a566cd9a7b12e07cb894a669872eab28948d79df919aa973f686d980a94889f4cbdef1e

data/CHANGELOG.md

@@ -1,3 +1,11 @@
+## 0.3.0 (2024-10-23)
+
+- Dropped support for Ruby < 3.1
+
+## 0.2.2 (2023-01-29)
+
+- Fixed mutating `p` argument when not upper triangle
+
 ## 0.2.1 (2022-07-05)
 
 - Added `Matrix` class

data/README.md

@@ -2,7 +2,9 @@
 
 The [OSQP](https://osqp.org/) (Operator Splitting Quadratic Program) solver for Ruby
 
-[![Build Status](https://github.com/ankane/osqp-ruby/workflows/build/badge.svg?branch=master)](https://github.com/ankane/osqp-ruby/actions)
+Check out [Opt](https://github.com/ankane/opt) for a high-level interface
+
+[![Build Status](https://github.com/ankane/osqp-ruby/actions/workflows/build.yml/badge.svg)](https://github.com/ankane/osqp-ruby/actions)
 
 ## Installation
 

data/lib/osqp/matrix.rb

@@ -11,6 +11,8 @@ module OSQP
     def []=(row_index, column_index, value)
       raise IndexError, "row index out of bounds" if row_index < 0 || row_index >= @m
       raise IndexError, "column index out of bounds" if column_index < 0 || column_index >= @n
+      # dictionary of keys, optimized for converting to CSC
+      # TODO try COO for performance
       if value == 0
         (@data[column_index] ||= {}).delete(row_index)
       else
@@ -18,7 +20,7 @@ module OSQP
       end
     end
 
-    def to_ptr
+    def to_csc
       cx = []
       ci = []
       cp = []
@@ -35,15 +37,21 @@ module OSQP
         cp << cx.size
       end
 
-      nnz = cx.size
-      cx = Utils.float_array(cx)
-      ci = Utils.int_array(ci)
-      cp = Utils.int_array(cp)
+      {
+        start: cp,
+        index: ci,
+        value: cx
+      }
+    end
+
+    # private, for tests
+    def nnz
+      @data.sum { |_, v| v.count }
+    end
 
-      ptr = FFI.csc_matrix(m, n, nnz, cx, ci, cp)
-      # save refs
-      ptr.instance_variable_set(:@osqp_refs, [cx, ci, cp])
-      ptr
+    def initialize_copy(other)
+      super
+      @data = @data.transform_values(&:dup)
     end
 
     def self.from_dense(data)

data/lib/osqp/solver.rb

@@ -11,17 +11,17 @@ module OSQP
       # data
       # do not assign directly to struct to keep refs
       p = csc_matrix(p, upper: true)
-      q = Utils.float_array(q)
+      q = float_array(q)
       a = csc_matrix(a)
-      l = Utils.float_array(l)
-      u = Utils.float_array(u)
+      l = float_array(l)
+      u = float_array(u)
 
       data = FFI::Data.malloc
       data.n = a.n
       data.m = a.m
-      data.p = p
+      data.p = matrix_ptr(p)
       data.q = q
-      data.a = a
+      data.a = matrix_ptr(a)
       data.l = l
       data.u = u
 
@@ -73,11 +73,11 @@ module OSQP
       raise Error, "Expected y to be size #{m}, got #{y.size}" if y && y.size != m
 
       if x && y
-        check_result FFI.osqp_warm_start(@work, Utils.float_array(x), Utils.float_array(y))
+        check_result FFI.osqp_warm_start(@work, float_array(x), float_array(y))
       elsif x
-        check_result FFI.osqp_warm_start_x(@work, Utils.float_array(x))
+        check_result FFI.osqp_warm_start_x(@work, float_array(x))
       elsif y
-        check_result FFI.osqp_warm_start_y(@work, Utils.float_array(y))
+        check_result FFI.osqp_warm_start_y(@work, float_array(y))
       else
         raise Error, "Must set x or y"
       end
@@ -113,6 +113,16 @@ module OSQP
       end
     end
 
+    def float_array(arr)
+      # OSQP float = double
+      Fiddle::Pointer[arr.to_a.pack("d*")]
+    end
+
+    def int_array(arr)
+      # OSQP int = long long
+      Fiddle::Pointer[arr.to_a.pack("q*")]
+    end
+
     def read_float_array(ptr, size)
       # OSQP float = double
       ptr[0, size * Fiddle::SIZEOF_DOUBLE].unpack("d*")
@@ -127,6 +137,8 @@ module OSQP
       mtx = Matrix.from_dense(mtx) unless mtx.is_a?(Matrix)
 
       if upper
+        # TODO improve performance
+        mtx = mtx.dup
         mtx.m.times do |i|
           mtx.n.times do |j|
             mtx[i, j] = 0 if i > j
@@ -137,6 +149,19 @@ module OSQP
       mtx
     end
 
+    def matrix_ptr(mtx)
+      csc = mtx.to_csc
+      nnz = csc[:value].size
+      cx = float_array(csc[:value])
+      ci = int_array(csc[:index])
+      cp = int_array(csc[:start])
+
+      ptr = FFI.csc_matrix(mtx.m, mtx.n, nnz, cx, ci, cp)
+      # save refs
+      ptr.instance_variable_set(:@osqp_refs, [cx, ci, cp])
+      ptr
+    end
+
     def dimensions
       data = FFI::Data.new(@work.data)
       [data.m, data.n]

data/lib/osqp/version.rb

@@ -1,3 +1,3 @@
 module OSQP
-  VERSION = "0.2.1"
+  VERSION = "0.3.0"
 end

data/lib/osqp.rb

@@ -4,7 +4,6 @@ require "fiddle/import"
 # modules
 require "osqp/matrix"
 require "osqp/solver"
-require "osqp/utils"
 require "osqp/version"
 
 module OSQP

metadata

@@ -1,15 +1,29 @@
 --- !ruby/object:Gem::Specification
 name: osqp
 version: !ruby/object:Gem::Version
-  version: 0.2.1
+  version: 0.3.0
 platform: ruby
 authors:
 - Andrew Kane
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2022-07-05 00:00:00.000000000 Z
-dependencies: []
+date: 2024-10-23 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: fiddle
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 description:
 email: andrew@ankane.org
 executables: []
@@ -63,14 +77,14 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: '2.7'
+      version: '3.1'
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.3.7
+rubygems_version: 3.5.16
 signing_key:
 specification_version: 4
 summary: OSQP (Operator Splitting Quadratic Program) solver for Ruby