Added stop_search method

CHANGELOG.md

@@ -1,3 +1,7 @@
+## 0.13.2 (unreleased)
+
+- Added `stop_search` method
+
 ## 0.13.1 (2024-10-05)
 
 - Added binary installation for Debian 12

ext/or-tools/constraint.cpp

@@ -1,9 +1,11 @@
 #include <google/protobuf/text_format.h>
 #include <ortools/sat/cp_model.h>
+#include <ortools/util/time_limit.h>
 
 #include "ext.h"
 
 using operations_research::Domain;
+using operations_research::TimeLimit;
 using operations_research::sat::BoolVar;
 using operations_research::sat::Constraint;
 using operations_research::sat::CpModelBuilder;
@@ -424,17 +426,21 @@ void init_constraint(Rice::Module& m) {
       [](Object self, CpModelBuilder& model, SatParameters& parameters, Object callback) {
         Model m;
 
+        std::atomic<bool> stopped(false);
+        m.GetOrCreate<TimeLimit>()->RegisterExternalBooleanAsLimit(&stopped);
+
         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) {
+            [&](const CpSolverResponse& r) {
               // ensure Ruby thread
               if (ruby_native_thread_p()) {
                 // TODO find a better way to do this
                 callback.call("response=", r);
                 callback.call("on_solution_callback");
+                stopped = callback.attr_get("@stopped");
               }
             })
           );

lib/or_tools/cp_solver_solution_callback.rb

@@ -16,5 +16,9 @@ def value(expr)
     def objective_value
       @response.objective_value
     end
+
+    def stop_search
+      @stopped = true
+    end
   end
 end