getWeightsSOS1

Author: Joao-Dionisio

src/pyscipopt/scip.pxd

@@ -1594,8 +1594,6 @@ cdef extern from "scip/cons_sos1.h":
 
     SCIP_Real* SCIPgetWeightsSOS1(SCIP* scip, SCIP_CONS* cons)
 
-    int SCIPgetNSOS1Vars(SCIP_CONSHDLR* conshdlr)
-
     SCIP_RETCODE SCIPmakeSOS1Feasible(SCIP* scip,
                                       SCIP_CONSHDLR* conshdlr,
                                       SCIP_SOL* solution,

src/pyscipopt/scip.pxi

@@ -6508,6 +6508,40 @@ cdef class Model:
 
         return Constraint.create(scip_cons)
 
+    def getWeightsSOS1(self, Constraint cons):
+        """
+        Retrieve the coefficients of an SOS1 constraint
+
+        Parameters
+        ----------
+        cons : Constraint
+            SOS1 constraint to get the coefficients of
+
+        Returns
+        -------
+        dict of str to float
+
+        """
+        cdef SCIP_VAR** vars
+        cdef SCIP_Longint* vals
+        cdef int nvars
+        cdef int i
+
+        constype = bytes(SCIPconshdlrGetName(SCIPconsGetHdlr(cons.scip_cons))).decode('UTF-8')
+        if not constype == 'SOS1':
+            raise Warning("weights not available for constraints of type ", constype)
+
+        nvars = SCIPgetNVarsSOS1(self._scip, cons.scip_cons)
+        vals = SCIPgetWeightsSOS1(self._scip, cons.scip_cons)
+        vars = SCIPgetVarsSOS1(self._scip, cons.scip_cons)
+
+        valsdict = {}
+        for i in range(nvars):
+            var_name = bytes(SCIPvarGetName(vars[i])).decode('utf-8')
+            valsdict[var_name] = vals[i]
+
+        return valsdict
+    
     def addConsSOS2(self, vars, weights=None, name="",
                 initial=True, separate=True, enforce=True, check=True,
                 propagate=True, local=False, dynamic=False,