WIP: begin support for vector-valued objective functions

Author: odow

Project.toml

@@ -7,6 +7,7 @@ version = "0.11.5"
 LazyArtifacts = "4af54fe1-eca0-43a8-85a7-787d91b784e3"
 Libdl = "8f399da3-3557-5675-b5ff-fb832c97cbdb"
 MathOptInterface = "b8f27783-ece8-5eb3-8dc8-9495eed66fee"
+Pkg = "44cfe95a-1eb2-52ea-b672-e2afdf69b78f"
 
 [compat]
 MathOptInterface = "1.7"

src/MOI_wrapper/MOI_multi_objective.jl

@@ -126,3 +126,21 @@ function MOI.get(model::Gurobi.Optimizer, attr::MultiObjectiveValue)
     _check_ret(model, ret)
     return val[]
 end
+
+function MOI.set(
+    model::Optimizer,
+    ::MOI.ObjectiveFunction{F},
+    f::F,
+) where {F<:MOI.VectorAffineFunction{Float64}}
+    for (i, fi) in enumerate(MOI.Utilities.eachscalar(f))
+        MOI.set(model, MultiObjectiveFunction(i), fi)
+    end
+    return
+end
+
+function MOI.supports(
+    model::Optimizer,
+    ::MOI.ObjectiveFunction{MOI.VectorAffineFunction{Float64}},
+)
+    return true
+end

src/MOI_wrapper/MOI_wrapper.jl

@@ -3091,6 +3091,10 @@ function MOI.get(model::Optimizer, attr::MOI.ObjectiveValue)
             attr.result_index - 1,
         )
     end
+    N = MOI.get(model, NumberOfObjectives())
+    if N > 1
+        return [MOI.get(model, MultiObjectiveValue(i)) for i in 1:N]
+    end
     valueP = Ref{Cdouble}()
     key = attr.result_index == 1 ? "ObjVal" : "PoolObjVal"
     ret = GRBgetdblattr(model, key, valueP)

test/MOI/MOI_multiobjective.jl

@@ -55,7 +55,7 @@ c4: y >= 0.25
     @test MOI.get(model, Gurobi.MultiObjectivePriority(2)) == 0
 
     MOI.optimize!(model)
-    @test MOI.get(model, MOI.ObjectiveValue()) ≈ 1.5
+    @test MOI.get(model, MOI.ObjectiveValue()) ≈ [1.5, 2.0]
     @test MOI.get(model, MOI.VariablePrimal(), x) ≈ 0.5
     @test MOI.get(model, MOI.VariablePrimal(), y) ≈ 0.5
 
@@ -101,6 +101,37 @@ c4: y >= 0.25
     @test MOI.get(model, Gurobi.MultiObjectiveValue(2)) ≈ BFS[3].f2
 end
 
+function test_example_biobjective_knapsack()
+    p1 = [77.0, 94, 71, 63, 96, 82, 85, 75, 72, 91, 99, 63, 84, 87, 79, 94, 90]
+    p2 = [65.0, 90, 90, 77, 95, 84, 70, 94, 66, 92, 74, 97, 60, 60, 65, 97, 93]
+    w = [80.0, 87, 68, 72, 66, 77, 99, 85, 70, 93, 98, 72, 100, 89, 67, 86, 91]
+    model = Gurobi.Optimizer()
+    x = MOI.add_variables(model, length(w))
+    MOI.add_constraint.(model, x, MOI.ZeroOne())
+    MOI.add_constraint(model, w' * x, MOI.LessThan(900.0))
+    obj_f = MOI.Utilities.operate(vcat, Float64, p1' * x, p2' * x)
+    MOI.set(model, MOI.ObjectiveSense(), MOI.MAX_SENSE)
+    MOI.set(model, MOI.ObjectiveFunction{typeof(obj_f)}(), obj_f)
+    MOI.optimize!(model)
+    results = Dict(
+        [955.0, 906.0] => [2, 3, 5, 6, 9, 10, 11, 14, 15, 16, 17],
+        [948.0, 939.0] => [1, 2, 3, 5, 6, 8, 10, 11, 15, 16, 17],
+        [934.0, 971.0] => [2, 3, 5, 6, 8, 10, 11, 12, 15, 16, 17],
+        [918.0, 983.0] => [2, 3, 4, 5, 6, 8, 10, 11, 12, 16, 17],
+    )
+    found_non_dominated_point = false
+    for i in 1:MOI.get(model, MOI.ResultCount())
+        X = findall(elt -> elt > 0.9, MOI.get.(model, MOI.VariablePrimal(i), x))
+        Y = MOI.get(model, MOI.ObjectiveValue(i))
+        if haskey(results, Y)
+            @test results[Y] == X
+            found_non_dominated_point = true
+        end
+    end
+    @test found_non_dominated_point
+    return
+end
+
 end
 
 TestMultiobjective.runtests()

test/runtests.jl

@@ -4,6 +4,9 @@
 # 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.
 
+import Pkg
+Pkg.pkg"add MathOptInterface#od/vector-optimization"
+
 using Gurobi
 using Test