Add add_cons_local

Author: Joao-Dionisio

src/model.rs

@@ -862,6 +862,26 @@ pub trait ProblemOrSolving {
         rhs: f64,
         name: &str,
     ) -> Constraint;
+
+    /// Adds the given constraint to the model at the given node and its children.
+    ///
+    /// # Arguments
+    ///
+    /// * `cons`      - The constraint to add.
+    /// * `validnode` - The node at which the constraint is valid.
+    ///
+    /// # Returns
+    ///
+    /// A reference-counted pointer to the new constraint.
+    ///
+    /// # Panics
+    ///
+    /// This method panics if the constraint cannot be created in the current state.
+    fn add_cons_local(
+        &mut self,
+        cons: Constraint,
+        validnode: Node,
+    ) -> Constraint;
 }
 
 /// A trait for model stages that have a problem or are during solving.
@@ -1182,6 +1202,34 @@ impl<S: ModelStageProblemOrSolving> ProblemOrSolving for Model<S> {
             scip: self.scip.clone(),
         }
     }
+
+    /// Adds the given constraint to the model at the given node and its children.
+    ///
+    /// # Arguments
+    ///
+    /// * `cons`      - The constraint to add.
+    /// * `validnode` - The node at which the constraint is valid.
+    ///
+    /// # Returns
+    ///
+    /// A reference-counted pointer to the new constraint.
+    ///
+    /// # Panics
+    ///
+    /// This method panics if the constraint cannot be created in the current state.
+    fn add_cons_local(
+        &mut self,
+        cons: Constraint,
+        validnode: Node,
+    ) -> Constraint {
+        let result = self
+            .scip
+            .add_cons_local(&cons, &validnode)
+            .expect("Failed to create constraint in state ProblemCreated");
+        
+        cons
+    }
+    
 }
 
 /// A trait for optimization models with any state that might have solutions.

src/scip.rs

@@ -1,4 +1,5 @@
 use crate::branchrule::{BranchRule, BranchingCandidate};
+use crate::node::Node;
 use crate::pricer::{Pricer, PricerResultState};
 use crate::{
     ffi, scip_call_panic, BranchingResult, Conshdlr, Constraint, Event, Eventhdlr, HeurResult,
@@ -544,6 +545,7 @@ impl ScipPtr {
             Some(var)
         }
     }
+
     pub(crate) fn create_cons_indicator(
         &self,
         bin_var: &Variable,
@@ -574,6 +576,18 @@ impl ScipPtr {
         Ok(scip_cons)
     }
 
+    pub(crate) fn add_cons_local(
+        &self,
+        cons: &Constraint,
+        validnode: &Node,
+    ) -> Result<*mut SCIP_Cons, Retcode> {
+        let cons_ptr = MaybeUninit::uninit();
+        scip_call! { ffi::SCIPaddConsLocal(self.raw, cons.raw, validnode.raw) };        
+        let cons_ptr = unsafe { cons_ptr.assume_init() };
+        Ok(cons_ptr)
+    }
+
+
     /// Create solution
     pub(crate) fn create_sol(&self, original: bool) -> Result<*mut SCIP_SOL, Retcode> {
         let mut sol = MaybeUninit::uninit();