.

Author: dnguyen227

src/mbm.jl

@@ -431,6 +431,42 @@ function _maximize_M(
           "has not been implemented for MBM subproblems\nF: $(F)")
 end
 
+"""
+    copy_model_with_constraints(model, constraints, method)
+
+Create a new model with only the variables (and their bounds)
+from `model` and the selected `constraints`. Returns a
+`GDPSubmodel` wrapping the new model.
+"""
+function copy_model_with_constraints(
+    model::JuMP.AbstractModel,
+    constraints::Vector{<:DisjunctConstraintRef},
+    method::_MBM
+    )
+    var_type = JuMP.variable_ref_type(model)
+    sub_model = _copy_model(model)
+    decision_vars = collect_all_vars(model)
+    fwd_map = Dict{var_type, Vector{var_type}}()
+
+    for var in decision_vars
+        copy_var = variable_copy(sub_model, var)
+        fwd_map[var] = [copy_var]
+    end
+
+    for cref in constraints
+        con = JuMP.constraint_object(cref)
+        flat_map = Dict(v => only(ws) for (v, ws) in fwd_map)
+        expr = _replace_variables_in_constraint(con.func, flat_map)
+        T = one(JuMP.value_type(typeof(sub_model)))
+        JuMP.@constraint(sub_model, expr * T in con.set)
+    end
+
+    JuMP.set_optimizer(sub_model, method.optimizer)
+    JuMP.set_silent(sub_model)
+
+    return GDPSubmodel(sub_model, decision_vars, fwd_map)
+end
+
 ################################################################################
 #                    REPLACE VARIABLES IN CONSTRAINT
 ################################################################################

src/utilities.jl

@@ -8,42 +8,6 @@ function _copy_model(
     return M()
 end
 
-"""
-    copy_model_with_constraints(model, constraints, method)
-
-Create a new model with only the variables (and their bounds)
-from `model` and the selected `constraints`. Returns a
-`GDPSubmodel` wrapping the new model.
-"""
-function copy_model_with_constraints(
-    model::JuMP.AbstractModel,
-    constraints::Vector{<:DisjunctConstraintRef},
-    method::_MBM
-    )
-    var_type = JuMP.variable_ref_type(model)
-    sub_model = _copy_model(model)
-    decision_vars = collect_all_vars(model)
-    fwd_map = Dict{var_type, Vector{var_type}}()
-
-    for var in decision_vars
-        copy_var = variable_copy(sub_model, var)
-        fwd_map[var] = [copy_var]
-    end
-
-    for cref in constraints
-        con = JuMP.constraint_object(cref)
-        flat_map = Dict(v => only(ws) for (v, ws) in fwd_map)
-        expr = _replace_variables_in_constraint(con.func, flat_map)
-        T = one(JuMP.value_type(typeof(sub_model)))
-        JuMP.@constraint(sub_model, expr * T in con.set)
-    end
-
-    JuMP.set_optimizer(sub_model, method.optimizer)
-    JuMP.set_silent(sub_model)
-
-    return GDPSubmodel(sub_model, decision_vars, fwd_map)
-end
-
 """
     copy_and_reformulate(model, decision_vars, reform_method, method)