SparseMatrix with abstract Tv (#18)

Author: blegat

src/conic_form.jl

@@ -29,7 +29,7 @@ function get_conic_form(::Type{T}, model::M, con_idx) where {T, M <: MOI.Abstrac
     cones = _set_type.(con_idx)
     cones = unique(cones)
 
-    conic = ConicForm{T, SparseMatrixCSRtoCSC{Int}, Array{T, 1}}(Tuple(cones))
+    conic = ConicForm{T, SparseMatrixCSRtoCSC{T, Int}, Vector{T}}(Tuple(cones))
 
     idxmap = MOI.copy_to(conic, model)
 

src/sparse_matrix.jl

@@ -1,22 +1,22 @@
-mutable struct SparseMatrixCSRtoCSC{T}
+mutable struct SparseMatrixCSRtoCSC{Tv,Ti<:Integer}
     m::Int # Number of rows
     n::Int # Number of columns
-    colptr::Vector{T}
-    rowval::Vector{T}
-    nzval::Vector{Float64}
-    function SparseMatrixCSRtoCSC{T}(n) where T
+    colptr::Vector{Ti}
+    rowval::Vector{Ti}
+    nzval::Vector{Tv}
+    function SparseMatrixCSRtoCSC{Tv, Ti}(n) where {Tv, Ti<:Integer}
         A = new()
         A.n = n
-        A.colptr = zeros(T, n + 1)
+        A.colptr = zeros(Ti, n + 1)
         return A
     end
 end
-function allocate_nonzeros(A::SparseMatrixCSRtoCSC{T}) where T
+function allocate_nonzeros(A::SparseMatrixCSRtoCSC{Tv, Ti}) where {Tv, Ti}
     for i in 3:length(A.colptr)
         A.colptr[i] += A.colptr[i - 1]
     end
-    A.rowval = Vector{T}(undef, A.colptr[end])
-    A.nzval = Vector{Float64}(undef, A.colptr[end])
+    A.rowval = Vector{Ti}(undef, A.colptr[end])
+    A.nzval = Vector{Tv}(undef, A.colptr[end])
 end
 function final_touch(A::SparseMatrixCSRtoCSC)
     for i in length(A.colptr):-1:2