Add LazyScalarSet

Author: odow

src/Bridges/Constraint/Constraint.jl

@@ -72,6 +72,7 @@ function add_all_bridges(model, ::Type{T}) where {T}
     #       It is also really useful only to PATHSolver.jl, which could add this
     #       to MOI.ListOfRequiredBridges.
     MOI.Bridges.add_bridge(model, IntegerToZeroOneBridge{T})
+    MOI.Bridges.add_bridge(model, LazyScalarSetBridge{T})
     MOI.Bridges.add_bridge(model, LessToGreaterBridge{T})
     if T <: AbstractFloat  # See note in docstring of AbstractToIntervalBridge
         MOI.Bridges.add_bridge(model, LessToIntervalBridge{T})

src/Bridges/Constraint/bridges/LazyScalarSetBridge.jl

@@ -0,0 +1,51 @@
+# Copyright (c) 2017: Miles Lubin and contributors
+# Copyright (c) 2017: Google Inc.
+#
+# 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.
+
+struct LazyScalarSetBridge{
+    T,
+    F<:MOI.AbstractScalarFunction,
+    S<:MOI.AbstractScalarSet,
+} <: MOI.Bridges.Constraint.SetMapBridge{T,S,MOI.LazyScalarSet{S},F,F}
+    constraint::MOI.ConstraintIndex{F,S}
+end
+
+MOI.Bridges.map_function(::Type{<:LazyScalarSetBridge}, f) = f
+
+MOI.Bridges.inverse_map_function(::Type{<:LazyScalarSetBridge}, f) = f
+
+MOI.Bridges.adjoint_map_function(::Type{<:LazyScalarSetBridge}, f) = f
+
+MOI.Bridges.inverse_adjoint_map_function(::Type{<:LazyScalarSetBridge}, f) = f
+
+function MOI.Bridges.map_set(
+    ::Type{LazyScalarSetBridge{T,F,S}},
+    set::MOI.LazyScalarSet{S},
+) where {T,F,S}
+    return set.set
+end
+
+function MOI.Bridges.inverse_map_set(
+    ::Type{LazyScalarSetBridge{T,F,S}},
+    set::S,
+) where {T,F,S}
+    return MOI.LazyScalarSet(set)
+end
+
+function MOI.supports_constraint(
+    ::Type{<:LazyScalarSetBridge},
+    ::Type{<:MOI.AbstractScalarFunction},
+    ::Type{<:MOI.LazyScalarSet},
+)
+    return true
+end
+
+function MOI.Bridges.Constraint.concrete_bridge_type(
+    ::Type{<:LazyScalarSetBridge{T}},
+    ::Type{F},
+    ::Type{MOI.LazyScalarSet{S}},
+) where {T,F<:MOI.AbstractScalarFunction,S<:MOI.AbstractScalarSet}
+    return LazyScalarSetBridge{T,F,S}
+end

src/Utilities/sets.jl

@@ -90,6 +90,16 @@ end
 
 supports_shift_constant(::Type{<:MOI.Parameter}) = true
 
+function supports_shift_constant(
+    ::Type{MOI.LazyScalarSet{S}},
+) where {S<:MOI.AbstractScalarSet}
+    return supports_shift_constant(S)
+end
+
+function shift_constant(set::MOI.LazyScalarSet, constant)
+    return MOI.LazyScalarSet(shift_constant(set.set, constant))
+end
+
 """
     ScalarLinearSet{T}
 

src/sets.jl

@@ -2866,6 +2866,13 @@ function Base.show(io::IO, s::VectorNonlinearOracle{T}) where {T}
     return
 end
 
+"""
+    LazyScalarSet(set::MOI.AbstractScalarSet) <: MOI.AbstractScalarSet
+"""
+struct LazyScalarSet{S<:MOI.AbstractScalarSet} <: MOI.AbstractScalarSet
+    set::S
+end
+
 # TODO(odow): these are not necessarily isbits. They may not be safe to return
 # without copying if the number is BigFloat, for example.
 function Base.copy(