make JuMP a weakdep

Author: joaquimg

Project.toml

@@ -4,10 +4,15 @@ authors = ["guilhermebodin <guilherme.b.moraes@gmail.com>"]
 version = "0.5.9"
 
 [deps]
-JuMP = "4076af6c-e467-56ae-b986-b466b2749572"
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
 MathOptInterface = "b8f27783-ece8-5eb3-8dc8-9495eed66fee"
 
+[weakdeps]
+JuMP = "4076af6c-e467-56ae-b986-b466b2749572"
+
+[extensions]
+DualizationJuMPExt = "JuMP"
+
 [compat]
 JuMP = "0.23, 1"
 MathOptInterface = "1"

src/JuMP.jl renamed to ext/DualizationJuMPExt/DualizationJuMPExt.jl

@@ -1,11 +1,26 @@
-function dualize(model::JuMP.Model, optimizer_constructor = nothing; kwargs...)
+# Copyright (c) 2017: Guilherme Bodin, and contributors
+#
+# Use of this source code is governed by an MIT-style license that can be found
+# in the LICENSE.md file or at https://opensource.org/licenses/MIT.
+
+module DualizationJuMPExt
+
+import Dualization
+import JuMP
+import MathOptInterface as MOI
+
+function Dualization.dualize(
+    model::JuMP.Model,
+    optimizer_constructor = nothing;
+    kwargs...,
+)
     mode = JuMP.mode(model)
     if mode != JuMP.AUTOMATIC
         error("Dualization does not support solvers in $(mode) mode")
     end
     dual_model = JuMP.Model()
-    dual_problem = DualProblem(JuMP.backend(dual_model))
-    dualize(JuMP.backend(model), dual_problem; kwargs...)
+    dual_problem = Dualization.DualProblem(JuMP.backend(dual_model))
+    Dualization.dualize(JuMP.backend(model), dual_problem; kwargs...)
     _fill_obj_dict_with_variables!(dual_model)
     _fill_obj_dict_with_constraints!(dual_model)
     if optimizer_constructor !== nothing
@@ -60,10 +75,13 @@ function _get_primal_dual_map(model::JuMP.Model)
     return model.ext[:_Dualization_jl_PrimalDualMap]
 end
 
-function _get_dual_constraint(dual_model, primal_ref::JuMP.VariableRef)
+function Dualization._get_dual_constraint(
+    dual_model,
+    primal_ref::JuMP.VariableRef,
+)
     map = _get_primal_dual_map(dual_model)
     moi_primal_vi = JuMP.index(primal_ref)
-    moi_dual_ci, idx = _get_dual_constraint(map, moi_primal_vi)
+    moi_dual_ci, idx = Dualization._get_dual_constraint(map, moi_primal_vi)
     # dual_model = nothing # TODO
     if idx === nothing
         # variables fixed at zero
@@ -72,27 +90,27 @@ function _get_dual_constraint(dual_model, primal_ref::JuMP.VariableRef)
     return JuMP.constraint_ref_with_index(dual_model, moi_dual_ci), idx
 end
 
-function _get_primal_constraint(
+function Dualization._get_primal_constraint(
     dual_model::JuMP.Model,
     primal_vi::JuMP.VariableRef,
 )
     primal_model = JuMP.owner_model(primal_vi)
     map = _get_primal_dual_map(dual_model)
     moi_primal_vi = JuMP.index(primal_vi)
-    primal_ci, idx = _get_primal_constraint(map, moi_primal_vi)
+    primal_ci, idx = Dualization._get_primal_constraint(map, moi_primal_vi)
     if primal_ci === nothing
         return nothing, idx
     end
     return JuMP.constraint_ref_with_index(primal_model, primal_ci), idx
 end
 
-function _get_dual_variables(
+function Dualization._get_dual_variables(
     dual_model::JuMP.Model,
     primal_ref::JuMP.ConstraintRef,
 )
     map = _get_primal_dual_map(dual_model)
     moi_primal_ci = JuMP.index(primal_ref)
-    moi_dual_vis = _get_dual_variables(map, moi_primal_ci)
+    moi_dual_vis = Dualization._get_dual_variables(map, moi_primal_ci)
     if moi_dual_vis === nothing
         # main constraint of a constrained variable
         return nothing
@@ -101,13 +119,13 @@ function _get_dual_variables(
 end
 
 # this is a constrained variable constraint
-function _get_dual_constraint(
+function Dualization._get_dual_constraint(
     dual_model::JuMP.Model,
     primal_ref::JuMP.ConstraintRef,
 )
     map = _get_primal_dual_map(dual_model)
     moi_primal_ci = JuMP.index(primal_ref)
-    moi_dual_ci = _get_dual_constraint(map, moi_primal_ci)
+    moi_dual_ci = Dualization._get_dual_constraint(map, moi_primal_ci)
     if moi_dual_ci === nothing
         # main constraint of a constrained variable
         # or
@@ -117,13 +135,15 @@ function _get_dual_constraint(
     return JuMP.constraint_ref_with_index(dual_model, moi_dual_ci)
 end
 
-function _get_dual_parameter(
+function Dualization._get_dual_parameter(
     dual_model::JuMP.Model,
     primal_ref::JuMP.VariableRef,
 )
     map = _get_primal_dual_map(dual_model)
     moi_primal_vi = JuMP.index(primal_ref)
-    moi_dual_vi = _get_dual_parameter(map, moi_primal_vi)
+    moi_dual_vi = Dualization._get_dual_parameter(map, moi_primal_vi)
     # the above line might error
     return JuMP.VariableRef(dual_model, moi_dual_vi)
 end
+
+end # module DualizationJuMPExt

src/Dualization.jl

@@ -5,11 +5,8 @@
 
 module Dualization
 
-import JuMP
 import LinearAlgebra
-import MathOptInterface
-
-const MOI = MathOptInterface
+import MathOptInterface as MOI
 
 include("structures.jl")
 include("utils.jl")
@@ -23,6 +20,5 @@ include("dual_model_variables.jl")
 include("dual_equality_constraints.jl")
 include("dualize.jl")
 include("MOI_wrapper.jl")
-include("JuMP.jl")
 
 end # module