Add set for low-rank constrained SDP

src/Bridges/Variable/set_map.jl

@@ -129,7 +129,7 @@
     bridge::SetMapBridge,
 )
     set = MOI.get(model, attr, bridge.constraint)
-    return MOI.Bridges.map_set(typeof(bridge), set)
+    return MOI.Bridges.map_set(bridge, set)
 end
 
 function MOI.set(
@@ -149,7 +149,7 @@
     bridge::SetMapBridge,
 )
     value = MOI.get(model, attr, bridge.constraint)
-    return MOI.Bridges.map_function(typeof(bridge), value)
+    return MOI.Bridges.map_function(bridge, value)
 end
 
 function MOI.get(
@@ -171,7 +171,7 @@
     if any(isnothing, value)
         return nothing
     end
-    return MOI.Bridges.map_function(typeof(bridge), value, i)
+    return MOI.Bridges.map_function(bridge, value, i)
 end
 
 function MOI.supports(
@@ -203,7 +203,7 @@
     i::MOI.Bridges.IndexInVector,
 ) where {T}
     func = MOI.Bridges.map_function(
-        typeof(bridge),
+        bridge,
         MOI.VectorOfVariables(bridge.variables),
         i,
     )
@@ -212,7 +212,7 @@
 
 function unbridged_map(bridge::SetMapBridge{T}, vi::MOI.VariableIndex) where {T}
     F = MOI.ScalarAffineFunction{T}
-    mapped = MOI.Bridges.inverse_map_function(typeof(bridge), vi)
+    mapped = MOI.Bridges.inverse_map_function(bridge, vi)
     return Pair{MOI.VariableIndex,F}[bridge.variable=>mapped]
 end
 
@@ -222,9 +222,10 @@
 ) where {T}
     F = MOI.ScalarAffineFunction{T}
     func = MOI.VectorOfVariables(vis)
-    funcs = MOI.Bridges.inverse_map_function(typeof(bridge), func)
+    funcs = MOI.Bridges.inverse_map_function(bridge, func)
     scalars = MOI.Utilities.eachscalar(funcs)
+    # FIXME not correct for SetDotProducts, it won't recover the dot product variables
     return Pair{MOI.VariableIndex,F}[
-        bridge.variables[i] => scalars[i] for i in eachindex(vis)
+        bridge.variables[i] => scalars[i] for i in eachindex(bridge.variables)
     ]
 end

src/Bridges/set_map.jl

@@ -80,6 +80,10 @@
     return MOI.Utilities.eachscalar(map_function(BT, func))[i.value]
 end
 
+function map_function(bridge::AbstractBridge, func, i::IndexInVector)
+    return map_function(typeof(bridge), func, i)
+end
+
 """
     inverse_map_function(bridge::MOI.Bridges.AbstractBridge, func)
     inverse_map_function(::Type{BT}, func) where {BT}

src/Utilities/functions.jl

@@ -638,18 +638,24 @@ end
 A type that allows iterating over the scalar-functions that comprise an
 `AbstractVectorFunction`.
 """
-struct ScalarFunctionIterator{F<:MOI.AbstractVectorFunction,C}
+struct ScalarFunctionIterator{F<:MOI.AbstractVectorFunction,C,S} <:
+       AbstractVector{S}
     f::F
     # Cache that can be used to store a precomputed datastructure that allows
     # an efficient implementation of `getindex`.
     cache::C
+    function ScalarFunctionIterator(f::MOI.AbstractVectorFunction, cache)
+        return new{typeof(f),typeof(cache),scalar_type(typeof(f))}(f, cache)
+    end
 end
 
 function ScalarFunctionIterator(func::MOI.AbstractVectorFunction)
     return ScalarFunctionIterator(func, scalar_iterator_cache(func))
 end
 
-scalar_iterator_cache(func::MOI.AbstractVectorFunction) = nothing
+Base.size(s::ScalarFunctionIterator) = (MOI.output_dimension(s.f),)
+
+scalar_iterator_cache(::MOI.AbstractVectorFunction) = nothing
 
 function output_index_iterator(terms::AbstractVector, output_dimension)
     start = zeros(Int, output_dimension)