nlopt 0.1.1 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: dc7b9fe5b30fd9f60c726dccd8edb8859e65eca488cd30e3e252c5ac83eacc4d
-  data.tar.gz: 260121b1a693646f681419086ababd4a569903289b2e2031610af9dd2a1357c7
+  metadata.gz: c779a5c494c4f27159c477063ddf7574c40b9813936d25ca24a0e9b72a6c897c
+  data.tar.gz: dc7dd6083dfc4696ad9795fcb3565d7a67ecd150ca14b56bc99afdc171908335
 SHA512:
-  metadata.gz: af0d54c03bfa3f38c080d65ef8f4ba5b3550085cb1acc8ad5990cb3a052e21e35bd7fa2ac75cfa417f21e4ef0e2dd2a0fb662aee7fe1cb40890463da3b6c1e6f
-  data.tar.gz: b9273f26c650eee060ce9b8228bcc2223ed7a28df4582d903cfff8ba598cd77215529a24078cd668197e23a0bb314e399d01410cd819f7eb892691a21e8c6c93
+  metadata.gz: f9dde169b1604d8a2c4a726aff6faecfde7d63180a936ec4f0d640d02bb6f36cee1833718f3486c926da8332b34e0a6cdcf34518b853c1cbcf7a386873ccb378
+  data.tar.gz: 37b4f469c22ff6e21d4e1012ad1fcd7f6c09cc478cc215917d01a73c9aef04003dbce4c95f44f836dbf6b71c82e604acf43dddc8f9b8bb6ff510a871e6d3bd69

data/CHANGELOG.md

@@ -1,3 +1,8 @@
+## 0.1.2 (2024-12-16)
+
+- Added `add_inequality_constraint` and `add_equality_constraint` methods
+- Added `initial_step` method
+
 ## 0.1.1 (2024-12-15)
 
 - Added more methods

data/lib/nlopt/gradient.rb

@@ -6,12 +6,12 @@ module NLopt
     end
 
     def [](index)
-      check_index(index)
+      index = check_index(index)
       @ptr[index * Fiddle::SIZEOF_DOUBLE, Fiddle::SIZEOF_DOUBLE].unpack1("d")
     end
 
     def []=(index, value)
-      check_index(index)
+      index = check_index(index)
       @ptr[index * Fiddle::SIZEOF_DOUBLE, Fiddle::SIZEOF_DOUBLE] = [value].pack("d")
     end
 
@@ -26,10 +26,13 @@ module NLopt
 
     private
 
-    def check_index(index)
-      if index >= @n
-        raise IndexError, "index #{index} outside of array bounds"
+    def check_index(original_index)
+      index = original_index
+      index += @n if index < 0
+      if index < 0 || index >= @n
+        raise IndexError, "index #{original_index} outside of array bounds"
       end
+      index
     end
   end
 end

data/lib/nlopt/opt.rb

@@ -14,6 +14,9 @@ module NLopt
 
       @opt = FFI.nlopt_create(algorithm, n)
       @opt.free = FFI["nlopt_destroy"]
+
+      @inequality_constraints = []
+      @equality_constraints = []
     end
 
     def dimension
@@ -64,12 +67,26 @@ module NLopt
       out_dptr { |ptr| FFI.nlopt_get_upper_bounds(@opt, ptr) }
     end
 
+    def add_inequality_constraint(fc, tol: 0)
+      cb = objective_callback(fc)
+      check_res FFI.nlopt_add_inequality_constraint(@opt, cb, nil, tol)
+      @inequality_constraints << cb # keep reference
+    end
+
+    def add_equality_constraint(h, tol: 0)
+      cb = objective_callback(h)
+      check_res FFI.nlopt_add_equality_constraint(@opt, cb, nil, tol)
+      @equality_constraints << cb # keep reference
+    end
+
     def remove_inequality_constraints
       check_res FFI.nlopt_remove_inequality_constraints(@opt)
+      @inequality_constraints.clear
     end
 
     def remove_equality_constraints
       check_res FFI.nlopt_remove_equality_constraints(@opt)
+      @equality_constraints.clear
     end
 
     def set_stopval(stopval)
@@ -207,6 +224,10 @@ module NLopt
       end
     end
 
+    def initial_step(x)
+      out_dptr { |ptr| FFI.nlopt_get_initial_step(@opt, alloc_dptr(x), ptr) }
+    end
+
     def set_population(pop)
       check_res FFI.nlopt_set_population(@opt, pop)
     end
@@ -223,6 +244,10 @@ module NLopt
       FFI.nlopt_get_vector_storage(@opt)
     end
 
+    def to_ptr
+      @opt
+    end
+
     private
 
     def check_res(res)

data/lib/nlopt/version.rb

@@ -1,3 +1,3 @@
 module NLopt
-  VERSION = "0.1.1"
+  VERSION = "0.1.2"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: nlopt
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.2
 platform: ruby
 authors:
 - Andrew Kane
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-12-15 00:00:00.000000000 Z
+date: 2024-12-16 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: fiddle