[Utilities] fix support for complex values in Indicator distance_to_set

Author: odow

src/Utilities/distance_to_set.jl

@@ -613,27 +613,28 @@ function distance_to_set(
     distance::ProjectionUpperBoundDistance,
     x::AbstractVector{T},
     set::MOI.Indicator{MOI.ACTIVATE_ON_ONE},
-) where {T<:Real}
+) where {T<:Number}
     _check_dimension(x, set)
+
     return min(
         # Distance of x[1] from 0
         abs(x[1]),
         # Distance of x[1] from 1 + distance to set
-        sqrt((1 - x[1])^2 + distance_to_set(distance, x[2], set.set)^2),
+        sqrt(abs(1 - x[1])^2 + distance_to_set(distance, x[2], set.set)^2),
     )
 end
 
 function distance_to_set(
     distance::ProjectionUpperBoundDistance,
     x::AbstractVector{T},
     set::MOI.Indicator{MOI.ACTIVATE_ON_ZERO},
-) where {T}
+) where {T<:Number}
     _check_dimension(x, set)
     return min(
         # Distance of x[1] from 1
         abs(one(T) - x[1]),
         # Distance of x[1] from 0 + distance to set
-        sqrt(x[1]^2 + distance_to_set(distance, x[2], set.set)^2),
+        sqrt(abs(x[1])^2 + distance_to_set(distance, x[2], set.set)^2),
     )
 end
 
@@ -647,7 +648,7 @@ function distance_to_set(
     ::ProjectionUpperBoundDistance,
     x::AbstractVector{T},
     set::MOI.NormNuclearCone,
-) where {T}
+) where {T<:Real}
     _check_dimension(x, set)
     X = reshape(x[2:end], set.row_dim, set.column_dim)
     return max(sum(LinearAlgebra.svdvals(X)) - x[1], zero(T))
@@ -663,7 +664,7 @@ function distance_to_set(
     ::ProjectionUpperBoundDistance,
     x::AbstractVector{T},
     set::MOI.NormSpectralCone,
-) where {T}
+) where {T<:Real}
     _check_dimension(x, set)
     X = reshape(x[2:end], set.row_dim, set.column_dim)
     return max(maximum(LinearAlgebra.svdvals(X)) - x[1], zero(T))

test/Utilities/distance_to_set.jl

@@ -515,6 +515,24 @@ function test_HermitianPositiveSemidefiniteConeTriangle()
     return
 end
 
+function test_indicator_complex()
+    _test_set(
+        MOI.Indicator{MOI.ACTIVATE_ON_ZERO}(MOI.EqualTo(1.0+2.0im)),
+        [0.0, 1.0+2.0im] => 0.0,
+        [0.01, 1.0+1.0im] => 0.99,
+        [0.0, 1.0+1.0im] => 1.0,
+        [0.0, 1.0+1.5im] => 0.5,
+    )
+    _test_set(
+        MOI.Indicator{MOI.ACTIVATE_ON_ONE}(MOI.EqualTo(1.0+2.0im)),
+        [1.0, 1.0+2.0im] => 0.0,
+        [0.99, 1.0+1.0im] => 0.99,
+        [1.0, 1.0+1.0im] => 1.0,
+        [1.0, 1.0+1.5im] => 0.5,
+    )
+    return
+end
+
 end
 
 TestFeasibilityChecker.runtests()