Add specific dispatch for >= constraints.

src/attributes/compiler.jl

@@ -161,11 +161,7 @@
     return get(model.compiler_settings, :quadratization_method, PBO.DEFAULT())
 end
 
-function MOI.set(
-    model::Optimizer,
-    ::QuadratizationMethod,
-    method::PBO.QuadratizationMethod,
-)
+function MOI.set(model::Optimizer, ::QuadratizationMethod, method::PBO.QuadratizationMethod)
     model.compiler_settings[:quadratization_method] = method
 
     return nothing
@@ -248,7 +244,11 @@
     model::Optimizer,
     ::DefaultVariableEncodingMethod,
 )::Encoding.VariableEncodingMethod
-    return get(model.compiler_settings, :default_variable_encoding_method, Encoding.Binary())
+    return get(
+        model.compiler_settings,
+        :default_variable_encoding_method,
+        Encoding.Binary(),
+    )
 end
 
 function MOI.set(
@@ -503,18 +503,13 @@
     end
 end
 
-function MOI.set(
-    model::Optimizer{T},
-    ::VariableEncodingPenaltyHint,
-    vi::VI,
-    ρ,
-) where {T}
+function MOI.set(model::Optimizer{T}, ::VariableEncodingPenaltyHint, vi::VI, ρ) where {T}
     attr = :variable_encoding_penalty_hint
 
     if !haskey(model.variable_settings, attr)
         model.variable_settings[attr] = Dict{VI,Any}()
     end
-    
+
     model.variable_settings[attr][vi] = convert(T, ρ)
 
     return nothing
@@ -588,7 +583,8 @@
 function MOI.get(model::Optimizer{T}, ::ConstraintEncodingPenaltyHint, ci::CI) where {T}
     attr = :constraint_encoding_penalty_hint
 
-    if !haskey(model.constraint_settings, attr) || !haskey(model.constraint_settings[attr], ci)
+    if !haskey(model.constraint_settings, attr) ||
+       !haskey(model.constraint_settings[attr], ci)
         return nothing
     else
         return model.constraint_settings[attr][ci]::T
@@ -606,7 +602,7 @@
     if !haskey(model.constraint_settings, attr)
         model.constraint_settings[attr] = Dict{CI,Any}()
     end
-    
+
     model.constraint_settings[attr][ci] = convert(T, ρ)
 
     return nothing
@@ -661,4 +657,279 @@
     return nothing
 end
 
+@doc raw"""
+    SlackVariableEncodingMethod()
+
+Sets the encoding method for slack variables generated by constraints.
+"""
+struct SlackVariableEncodingMethod <: CompilerConstraintAttribute end
+
+function slack_variable_encoding_method(model::Optimizer, ci::CI)
+    return MOI.get(model, SlackVariableEncodingMethod(), ci)
+end
+
+function MOI.get(
+    model::Optimizer,
+    ::SlackVariableEncodingMethod,
+    ci::CI,
+)::Union{Encoding.VariableEncodingMethod,Nothing}
+    attr = :slack_variable_encoding_method
+
+    if !haskey(model.constraint_settings, attr) ||
+       !haskey(model.constraint_settings[attr], ci)
+        return nothing
+    else
+        return model.constraint_settings[attr][ci]
+    end
+end
+
+function MOI.set(
+    model::Optimizer,
+    ::SlackVariableEncodingMethod,
+    ci::CI,
+    e::Encoding.VariableEncodingMethod,
+)
+    attr = :slack_variable_encoding_method
+
+    if !haskey(model.constraint_settings, attr)
+        model.constraint_settings[attr] = Dict{CI,Any}()
+    end
+
+    model.constraint_settings[attr][ci] = e
+
+    return nothing
+end
+
+function MOI.set(
+    model::Optimizer,
+    ::SlackVariableEncodingMethod,
+    ci::CI,
+    ::Nothing,
+)
+    attr = :slack_variable_encoding_method
+
+    if haskey(model.constraint_settings, attr)
+        delete!(model.constraint_settings[attr], ci)
+    end
+
+    return nothing
+end
+
+@doc raw"""
+    SlackVariableEncodingATol()
+
+Sets the tolerance for slack variables generated by constraints.
+"""
+struct SlackVariableEncodingATol <: CompilerConstraintAttribute end
+
+function slack_variable_encoding_atol(model::Optimizer, ci::CI)
+    return MOI.get(model, SlackVariableEncodingATol(), ci)
+end
+
+function MOI.get(
+    model::Optimizer{T},
+    ::SlackVariableEncodingATol,
+    ci::CI,
+)::Union{T,Nothing} where {T}
+    attr = :slack_variable_encoding_atol
+
+    if !haskey(model.constraint_settings, attr) ||
+       !haskey(model.constraint_settings[attr], ci)
+        return nothing
+    else
+        return model.constraint_settings[attr][ci]::T
+    end
+end
+
+function MOI.set(
+    model::Optimizer{T},
+    ::SlackVariableEncodingATol,
+    ci::CI,
+    τ::T,
+)::Nothing where {T}
+    attr = :slack_variable_encoding_atol
+
+    if !haskey(model.constraint_settings, attr)
+        model.constraint_settings[attr] = Dict{CI,Any}()
+    end
+
+    model.constraint_settings[attr][ci] = τ
+
+    return nothing
+end
+
+function MOI.set(
+    model::Optimizer,
+    ::SlackVariableEncodingATol,
+    ci::CI,
+    ::Nothing,
+)
+    attr = :slack_variable_encoding_atol
+
+    if haskey(model.constraint_settings, attr)
+        delete!(model.constraint_settings[attr], ci)
+    end
+
+    return nothing
+end
+
+@doc raw"""
+    SlackVariableEncodingBits()
+
+Sets the number of bits for slack variables generated by constraints.
+"""
+struct SlackVariableEncodingBits <: CompilerConstraintAttribute end
+
+function slack_variable_encoding_bits(model::Optimizer, ci::CI)
+    return MOI.get(model, SlackVariableEncodingBits(), ci)
+end
+
+function MOI.get(
+    model::Optimizer,
+    ::SlackVariableEncodingBits,
+    ci::CI,
+)::Union{Integer,Nothing}
+    attr = :slack_variable_encoding_bits
+
+    if !haskey(model.constraint_settings, attr) ||
+       !haskey(model.constraint_settings[attr], ci)
+        return nothing
+    else
+        return model.constraint_settings[attr][ci]
+    end
+end
+
+function MOI.set(
+    model::Optimizer,
+    ::SlackVariableEncodingBits,
+    ci::CI,
+    n::Integer,
+)::Nothing
+    attr = :slack_variable_encoding_bits
+
+    if !haskey(model.constraint_settings, attr)
+        model.constraint_settings[attr] = Dict{CI,Any}()
+    end
+
+    model.constraint_settings[attr][ci] = n
+
+    return nothing
+end
+
+function MOI.set(
+    model::Optimizer,
+    ::SlackVariableEncodingBits,
+    ci::CI,
+    ::Nothing,
+)
+    attr = :slack_variable_encoding_bits
+
+    if haskey(model.constraint_settings, attr)
+        delete!(model.constraint_settings[attr], ci)
+    end
+
+    return nothing
+end
+
+@doc raw"""
+    SlackVariableEncodingPenaltyHint()
+
+Allows the user to hint the penalty factor used for encoding slack variables.
+"""
+struct SlackVariableEncodingPenaltyHint <: CompilerConstraintAttribute end
+
+function slack_variable_encoding_penalty_hint(model::Optimizer, ci::CI)
+    return MOI.get(model, SlackVariableEncodingPenaltyHint(), ci)
+end
+
+function MOI.get(
+    model::Optimizer{T},
+    ::SlackVariableEncodingPenaltyHint,
+    ci::CI,
+)::Union{T,Nothing} where {T}
+    attr = :slack_variable_encoding_penalty_hint
+
+    if !haskey(model.constraint_settings, attr) ||
+       !haskey(model.constraint_settings[attr], ci)
+        return nothing
+    else
+        return model.constraint_settings[attr][ci]::T
+    end
+end
+
+function MOI.set(
+    model::Optimizer{T},
+    ::SlackVariableEncodingPenaltyHint,
+    ci::CI,
+    ρ::T,
+)::Nothing where {T}
+    attr = :slack_variable_encoding_penalty_hint
+
+    if !haskey(model.constraint_settings, attr)
+        model.constraint_settings[attr] = Dict{CI,Any}()
+    end
+
+    model.constraint_settings[attr][ci] = ρ
+
+    return nothing
+end
+
+function MOI.set(
+    model::Optimizer,
+    ::SlackVariableEncodingPenaltyHint,
+    ci::CI,
+    ::Nothing,
+)
+    attr = :slack_variable_encoding_penalty_hint
+
+    if haskey(model.constraint_settings, attr)
+        delete!(model.constraint_settings[attr], ci)
+    end
+
+    return nothing
+end
+
+@doc raw"""
+    SlackVariableEncodingPenalty()
+
+Allows the user to retrieve the penalty factor used for encoding slack variables.
+"""
+struct SlackVariableEncodingPenalty <: CompilerConstraintAttribute end
+
+MOI.is_set_by_optimize(::SlackVariableEncodingPenalty) = true
+
+function slack_variable_encoding_penalty(model::Optimizer, ci::CI)
+    return MOI.get(model, SlackVariableEncodingPenalty(), ci)
+end
+
+function MOI.get(
+    model::Optimizer{T},
+    ::SlackVariableEncodingPenalty,
+    ci::CI,
+)::Union{T,Nothing} where {T}
+    return get(model.ρ, ci, nothing)
+end
+
+function MOI.set(
+    model::Optimizer{T},
+    ::SlackVariableEncodingPenalty,
+    ci::CI,
+    ρ::T,
+)::Nothing where {T}
+    model.ρ[ci] = ρ
+
+    return nothing
+end
+
+function MOI.set(
+    model::Optimizer,
+    ::SlackVariableEncodingPenalty,
+    ci::CI,
+    ::Nothing,
+)
+    delete!(model.ρ, ci)
+
+    return nothing
+end
+
 end # module Attributes