Update Formulation docstring

docs/src/dev/formulation.md

@@ -7,10 +7,6 @@ end
 
 # Formulation
 
-```@docs
-Formulation
-```
-
 ## Duties of formulations
 
 ```@docs
@@ -86,4 +82,4 @@ getname
 ```@meta
 CurrentModule = nothing
 DocTestSetup = nothing
-```
+```

docs/src/dev/todo.md

@@ -54,6 +54,7 @@ Coluna.MathProg.get_ip_dual_bound
 Coluna.Algorithm.ReformData
 Coluna.Algorithm.ModelData
 Coluna.MathProg.get_optimization_target
+Coluna.MathProg.create_formulation!
 ```
 
 ```@meta

src/Coluna.jl

@@ -36,20 +36,21 @@ include("parameters.jl")
 include("ColunaBase/ColunaBase.jl")
 using .ColunaBase
 
-include("MathProg/MathProg.jl")
-using .MathProg
-
 mutable struct Env
     env_starting_time::DateTime
     optim_starting_time::Union{Nothing, DateTime}
     params::Params
     kpis::Kpis
+    form_counter::Int # 0 is for original form
 end
-Env(params::Params) = Env(now(), nothing, params, Kpis(nothing, nothing))
+Env(params::Params) = Env(now(), nothing, params, Kpis(nothing, nothing), 0)
 set_optim_start_time!(env::Env) = env.optim_starting_time = now()
 elapsed_optim_time(env::Env) = Dates.toms(now() - env.optim_starting_time) / Dates.toms(Second(1))
 Base.isinteger(x::Float64, tol::Float64) = abs(round(x) - x) < tol
 
+include("MathProg/MathProg.jl")
+using .MathProg
+
 include("Algorithm/Algorithm.jl")
 using .Algorithm
 

src/MOIwrapper.jl

@@ -15,9 +15,9 @@ const SupportedConstrSets = Union{
 
 @enum(ObjectiveType, SINGLE_VARIABLE, SCALAR_AFFINE)
 mutable struct Optimizer <: MOI.AbstractOptimizer
+    env::Env
     inner::Problem
     objective_type::ObjectiveType
-    params::Params
     annotations::Annotations
     #varmap::Dict{MOI.VariableIndex,VarId} # For the user to get VariablePrimal
     vars::CleverDicts.CleverDict{MOI.VariableIndex, Variable}
@@ -33,13 +33,11 @@ mutable struct Optimizer <: MOI.AbstractOptimizer
 
     feasibility_sense::Bool # Coluna supports only Max or Min.
 
-    kpis::Union{Nothing, Kpis}
-
 
     function Optimizer()
         model = new()
-        model.inner = Problem()
-        model.params = Params()
+        model.env = Env(Params())
+        model.inner = Problem(model.env)
         model.annotations = Annotations()
         model.vars = CleverDicts.CleverDict{MOI.VariableIndex, Variable}()
         model.varids = CleverDicts.CleverDict{MOI.VariableIndex, VarId}() # TODO : check if necessary to have two dicts for variables
@@ -51,7 +49,6 @@ mutable struct Optimizer <: MOI.AbstractOptimizer
         model.names_to_constrs = Dict{String, MOI.ConstraintIndex}()
         model.default_optimizer_builder = nothing
         model.feasibility_sense = false
-        model.kpis = nothing
         return model
     end
 end
@@ -69,7 +66,7 @@ MOI.supports(::Optimizer, ::MOI.ConstraintDualStart) = false
 # Parameters
 function MOI.set(model::Optimizer, param::MOI.RawParameter, val)
     if param.name == "params"
-        model.params = val
+        model.env.params = val
     elseif param.name == "default_optimizer"
         optimizer_builder = () -> MoiOptimizer(MOI._instantiate_and_check(val))
         model.default_optimizer_builder = optimizer_builder
@@ -98,8 +95,8 @@ end
 MOI.get(optimizer::Coluna.Optimizer, ::MOI.SolverName) = "Coluna"
 
 function MOI.optimize!(optimizer::Optimizer)
-    optimizer.result, optimizer.kpis = optimize!(
-        optimizer.inner, optimizer.annotations, optimizer.params
+    optimizer.result = optimize!(
+        optimizer.env, optimizer.inner, optimizer.annotations
     )
     return
 end
@@ -552,7 +549,7 @@ function MOI.set(model::Coluna.Optimizer, ::BD.ObjectivePrimalBound, pb)
 end
 
 function MOI.empty!(model::Coluna.Optimizer)
-    model.inner = Problem()
+    model.inner = Problem(model.env)
     model.annotations = Annotations()
     model.vars = CleverDicts.CleverDict{MOI.VariableIndex, Variable}()
     model.varids = CleverDicts.CleverDict{MOI.VariableIndex, VarId}()
@@ -674,8 +671,8 @@ function MOI.get(optimizer::Optimizer, ::MOI.ConstraintPrimal, index::MOI.Constr
     return constraint_primal(best_primal_sol, getid(constrid))
 end
 
-MOI.get(optimizer::Optimizer, ::MOI.NodeCount) = optimizer.kpis.node_count
-MOI.get(optimizer::Optimizer, ::MOI.SolveTime) = optimizer.kpis.elapsed_optimization_time
+MOI.get(optimizer::Optimizer, ::MOI.NodeCount) = optimizer.env.kpis.node_count
+MOI.get(optimizer::Optimizer, ::MOI.SolveTime) = optimizer.env.kpis.elapsed_optimization_time
 
 # function MOI.get(optimizer::Optimizer, ::MOI.ConstraintDual, index::MOI.ConstraintIndex)
 #     return 0.0

src/MathProg/MathProg.jl

@@ -78,8 +78,8 @@ export Reformulation, getmaster, add_dw_pricing_sp!, add_benders_sep_sp!, get_dw
     get_dw_pricing_sp_lb_constrid, setmaster!
 
 # Methods related to formulations
-export AbstractFormulation, Formulation, getreformulation, getvar, getvars, getconstr,
-    getconstrs, getelem, getcoefmatrix, getprimalsolmatrix, getprimalsolcosts,
+export AbstractFormulation, Formulation, create_formulation!, getreformulation, getvar, getvars,
+    getconstr, getconstrs, getelem, getcoefmatrix, getprimalsolmatrix, getprimalsolcosts,
     getdualsolmatrix, getdualsolrhss, setvar!, setconstr!, setprimalsol!, setdualsol!,
     set_robust_constr_generator!, get_robust_constr_generators,
     setcol_from_sp_primalsol!, setcut_from_sp_dualsol!, # TODO : merge with setvar! & setconstr

src/MathProg/formulation.jl

@@ -12,25 +12,27 @@ end
 """
 `Formulation` stores a mixed-integer linear program.
 
-    Formulation{Duty}(
-        form_counter::Counter;
+    create_formulation!(
+        env::Coluna.Env,
+        duty::Type{<:AbstractFormDuty};
         parent_formulation = nothing,
         obj_sense::Type{<:Coluna.AbstractSense} = MinSense
-    ) where {Duty<:AbstractFormDuty}
+    )
 
-Construct a `Formulation` of duty `Duty` with objective sense `obj_sense` and parent formulation
-`parent_formulation`.
+    Create a new formulation in the Coluna's environment `env` with duty `duty`,
+    parent formulation `parent_formulation`, and objective sense `obj_sense`.
 """
-function Formulation{D}(
-    form_counter::Counter;
+function create_formulation!(
+    env::Coluna.Env,
+    duty::Type{<:AbstractFormDuty};
     parent_formulation = nothing,
     obj_sense::Type{<:Coluna.AbstractSense} = MinSense
-) where {D<:AbstractFormDuty}
-    if form_counter.value >= MAX_NB_FORMULATIONS
+)
+    if env.form_counter >= MAX_NB_FORMULATIONS
         error("Maximum number of formulations reached.")
     end
-    return Formulation{D}(
-        getnewuid(form_counter), Counter(), Counter(), parent_formulation, NoOptimizer(), 
+    return Formulation{duty}(
+        env.form_counter += 1, Counter(), Counter(), parent_formulation, NoOptimizer(), 
         FormulationManager(), obj_sense, FormulationBuffer()
     )
 end
@@ -85,7 +87,6 @@ getprimalsolcosts(form::Formulation) = form.manager.primal_sol_costs
 getdualsolmatrix(form::Formulation) = form.manager.dual_sols
 getdualsolrhss(form::Formulation) = form.manager.dual_sol_rhss
 
-
 "Returns the `uid` of `Formulation` `form`."
 getuid(form::Formulation) = form.uid
 

src/MathProg/problem.jl

@@ -3,7 +3,6 @@ mutable struct Problem <: AbstractProblem
     initial_dual_bound::Union{Nothing, Float64}
     original_formulation::Formulation
     re_formulation::Union{Nothing, Reformulation}
-    form_counter::Counter # 0 is for original form
     default_optimizer_builder::Function
 end
 
@@ -12,11 +11,10 @@ end
 
 Constructs an empty `Problem`.
 """
-function Problem()
-    counter = Counter(-1)
-    original_formulation = Formulation{Original}(counter)
+function Problem(env::Coluna.Env)
+    original_formulation = create_formulation!(env, Original)
     return Problem(
-        nothing, nothing, original_formulation, nothing, counter,
+        nothing, nothing, original_formulation, nothing,
         no_optimizer_builder
     )
 end

src/decomposition.jl

@@ -27,11 +27,12 @@ function create_global_art_vars!(masterform::Formulation, env::Env)
 end
 
 function instantiatemaster!(
-    prob::Problem, reform::Reformulation, ::Type{BD.Master},
+    env::Env, prob::Problem, reform::Reformulation, ::Type{BD.Master},
     ::Type{BD.DantzigWolfe}
 )
-    form = Formulation{DwMaster}(
-        prob.form_counter;
+    form = create_formulation!(
+        env,
+        MathProg.DwMaster;
         parent_formulation = reform,
         obj_sense = getobjsense(get_original_formulation(prob))
     )
@@ -41,10 +42,11 @@ function instantiatemaster!(
 end
 
 function instantiatemaster!(
-    prob::Problem, reform::Reformulation, ::Type{BD.Master}, ::Type{BD.Benders}
+    env::Env, prob::Problem, reform::Reformulation, ::Type{BD.Master}, ::Type{BD.Benders}
 )
-    masterform = Formulation{BendersMaster}(
-        prob.form_counter;
+    masterform = create_formulation!(
+        env,
+        MathProg.BendersMaster;
         parent_formulation = reform,
         obj_sense = getobjsense(get_original_formulation(prob))
     )
@@ -53,11 +55,12 @@ function instantiatemaster!(
 end
 
 function instantiatesp!(
-    prob::Problem, reform::Reformulation, masterform::Formulation{DwMaster},
+    env::Env, reform::Reformulation, masterform::Formulation{DwMaster},
     ::Type{BD.DwPricingSp}, ::Type{BD.DantzigWolfe}
 )
-    spform = Formulation{DwSp}(
-        prob.form_counter;
+    spform = create_formulation!(
+        env,
+        MathProg.DwSp;
         parent_formulation = masterform,
         obj_sense = getobjsense(masterform)
     )
@@ -66,11 +69,12 @@ function instantiatesp!(
 end
 
 function instantiatesp!(
-    prob::Problem, reform::Reformulation, masterform::Formulation{BendersMaster},
+    env::Env, reform::Reformulation, masterform::Formulation{BendersMaster},
     ::Type{BD.BendersSepSp}, ::Type{BD.Benders}
 )
-    spform = Formulation{BendersSp}(
-        prob.form_counter;
+    spform = create_formulation!(
+        env,
+        MathProg.BendersSp;
         parent_formulation = masterform,
         obj_sense = getobjsense(masterform)
     )
@@ -438,7 +442,7 @@ function buildformulations!(
     ann = BD.annotation(node)
     form_type = BD.getformulation(ann)
     dec_type = BD.getdecomposition(ann)
-    masterform = instantiatemaster!(prob, reform, form_type, dec_type)
+    masterform = instantiatemaster!(env, prob, reform, form_type, dec_type)
     store!(annotations, masterform, ann)
     origform = get_original_formulation(prob)
      for (id, child) in BD.subproblems(node)
@@ -461,7 +465,7 @@ function buildformulations!(
     form_type = BD.getformulation(ann)
     dec_type = BD.getdecomposition(ann)
     masterform = getmaster(reform)
-    spform = instantiatesp!(prob, reform, masterform, form_type, dec_type)
+    spform = instantiatesp!(env, reform, masterform, form_type, dec_type)
     store!(annotations, spform, ann)
     origform = get_original_formulation(prob)
     assign_orig_vars_constrs!(spform, origform, env, annotations, ann)

src/optimize.jl

@@ -39,7 +39,7 @@ end
 """
 Starting point of the solver.
 """
-function optimize!(prob::MathProg.Problem, annotations::Annotations, params::Params)
+function optimize!(env::Env, prob::MathProg.Problem, annotations::Annotations)
     _welcome_message()
 
     buffer_reset = prob.original_formulation.buffer
@@ -49,9 +49,7 @@ function optimize!(prob::MathProg.Problem, annotations::Annotations, params::Par
     ## Retrieve initial bounds on the objective given by the user
     init_pb = get_initial_primal_bound(prob)
     init_db = get_initial_dual_bound(prob)
-    _adjust_params(params, init_pb)
-
-    env = Env(params)
+    _adjust_params(env.params, init_pb)
 
     # Apply decomposition
     reformulate!(prob, annotations, env)
@@ -75,7 +73,7 @@ function optimize!(prob::MathProg.Problem, annotations::Annotations, params::Par
     @logmsg LogLevel(0) "Terminated"
     @logmsg LogLevel(0) string("Primal bound: ", get_ip_primal_bound(optstate))
     @logmsg LogLevel(0) string("Dual bound: ", get_ip_dual_bound(optstate))
-    return optstate, env.kpis
+    return optstate
 end
 
 function optimize!(

test/unit/MathProg/variables.jl

@@ -3,13 +3,8 @@ function variables_unit_tests()
     return
 end
 
-function createformulation()
-    counter = ClF.Counter()
-    return ClF.Formulation{ClF.Original}(counter)
-end
-
 function getset_variables()
-    form = createformulation()
+    form = create_formulation!(Env(Coluna.Params()), Original)
     var = ClF.setvar!(
         form, "var1", ClF.OriginalVar, cost = 2.0, lb = -1.0, ub = 1.0, 
         kind = ClF.Integ, inc_val = 4.0

test/unit/constraint.jl

@@ -15,7 +15,7 @@ end
 
 function constraint_getters_and_setters_tests()
 
-    form = createformulation()
+    form = create_formulation!(Env(Coluna.Params()), Original)
 
     c = ClF.setconstr!(form, "fake_constr", ClF.MasterBranchOnOrigVarConstr,
     rhs = -13.0, kind = ClF.Facultative, sense = ClF.Equal,

test/unit/variable.jl

@@ -25,7 +25,7 @@ function moi_var_record_getters_and_setters_tests()
 end
 
 function variable_getters_and_setters_tests()
-    form = createformulation()
+    form = create_formulation!(Env(Coluna.Params()), Original)
 
     v_data = ClF.VarData(
         ; cost = 13.0, lb = -10.0, ub = 100.0, kind = ClF.Continuous,