[Bridges] pass variable names through ParameterToEqualToBridge

Author: odow

src/Bridges/bridge_optimizer.jl

@@ -1757,6 +1757,20 @@ function MOI.get(
     end
 end
 
+function _get_variable_if_equivalent(b, x)
+    return _get_variable_if_equivalent(bridged_variable_function(b, x))
+end
+
+function _get_variable_if_equivalent(f::MOI.ScalarAffineFunction)
+    if length(f.terms) == 1
+        term = only(f.terms)
+        if isone(term.coefficient)
+            return term.variable
+        end
+    end
+    return nothing
+end
+
 function MOI.set(
     b::AbstractBridgeOptimizer,
     attr::MOI.VariableName,
@@ -1766,6 +1780,11 @@ function MOI.set(
     if is_bridged(b, vi)
         b.var_to_name[vi] = name
         b.name_to_var = nothing # Invalidate the name map.
+        # Set the internal variable name iff the bridged variable is equivalent
+        # to `y := 1 * x`.
+        if (x = _get_variable_if_equivalent(b, vi)) !== nothing
+            MOI.set(b.model, attr, x, name)
+        end
     else
         MOI.set(b.model, attr, vi, name)
     end

test/Bridges/bridge_optimizer.jl

@@ -1570,6 +1570,28 @@ function test_issue_2817c()
     return
 end
 
+function test_issue_2762_success()
+    inner = MOI.Utilities.Model{Float64}()
+    model = MOI.Bridges.Variable.ParameterToEqualTo{Float64}(inner)
+    p, _ = MOI.add_constrained_variable(model, MOI.Parameter(1.0))
+    MOI.set(model, MOI.VariableName(), p, "p")
+    @test MOI.get(model, MOI.VariableName(), p) == "p"
+    x = only(MOI.get(inner, MOI.ListOfVariableIndices()))
+    @test MOI.get(inner, MOI.VariableName(), x) == "p"
+    return
+end
+
+function test_issue_2762_fail()
+    inner = MOI.Utilities.Model{Float64}()
+    model = MOI.Bridges.Variable.NonposToNonneg{Float64}(inner)
+    y, _ = MOI.add_constrained_variables(model, MOI.Nonpositives(2))
+    MOI.set(model, MOI.VariableName(), y[1], "y1")
+    @test MOI.get(model, MOI.VariableName(), y[1]) == "y1"
+    x = MOI.get(inner, MOI.ListOfVariableIndices())
+    @test all(isempty, MOI.get(inner, MOI.VariableName(), x))
+    return
+end
+
 end  # module
 
 TestBridgeOptimizer.runtests()