Add identity_index_map and fix DoubleDict bug

src/Utilities/DoubleDicts.jl

@@ -417,7 +417,9 @@ end
 end
 
 function initialize_inner!(d::AbstractWithType{F,S,V,D}) where {F,S,V,D}
-    return d.inner = D()
+    d.inner = D()
+    d.dict.dict[(F, S)] = d.inner
+    return
 end
 
 inner_is_empty(d::AbstractWithType)::Bool = d.inner === nothing

src/Utilities/copy.jl

@@ -69,6 +69,25 @@ function IndexMap(n = 0)
         DoubleDicts.IndexDoubleDict(),
     )
 end
+function _identity_constraints_map(
+    model,
+    index_map::MOIU.DoubleDicts.IndexWithType{F,S},
+) where {F,S}
+    for ci in MOI.get(model, MOI.ListOfConstraintIndices{F,S}())
+        index_map[ci] = ci
+    end
+end
+function identity_index_map(model::MOI.ModelLike)
+    vis = MOI.get(model, MOI.ListOfVariableIndices())
+    index_map = IndexMap(length(vis))
+    for vi in vis
+        index_map[vi] = vi
+    end
+    for (F, S) in MOI.get(model, MOI.ListOfConstraints())
+        _identity_constraints_map(model, index_map.conmap[F, S])
+    end
+    return index_map
+end
 
 """
     index_map_for_variable_indices(variables)

test/Utilities/copy.jl

@@ -13,6 +13,22 @@ function compare_without_moi(x::String, y::String)
     @test remove_moi(x) == remove_moi(y)
 end
 
+function identity_index_map_test(::Type{T}) where {T}
+    model = MOI.Utilities.Model{T}()
+    x, y = MOI.add_variables(model, 2)
+    fx = MOI.SingleVariable(x)
+    fy = MOI.SingleVariable(y)
+    cx = MOI.add_constraint(model, fx, MOI.EqualTo(one(T)))
+    cy = MOI.add_constraint(model, fy, MOI.EqualTo(zero(T)))
+    c = MOI.add_constraint(model, one(T) * fx + fy, MOI.LessThan(zero(T)))
+    index_map = MOI.Utilities.identity_index_map(model)
+    @test x == index_map[x]
+    @test y == index_map[y]
+    @test cx == index_map[cx]
+    @test cy == index_map[cy]
+    @test c == index_map[c]
+end
+
 @testset "IndexMap" begin
     map = MOIU.IndexMap()
     @test length(map) == 0
@@ -695,3 +711,8 @@ end
     test_pass_copy(Int)
     test_pass_copy(Float64)
 end
+
+@testset "identity_index_map" begin
+    identity_index_map_test(Int)
+    identity_index_map_test(Float64)
+end