Introduce getunstoredindex_function

Author: mtfishman

Project.toml

@@ -1,7 +1,7 @@
 name = "SparseArraysBase"
 uuid = "0d5efcca-f356-4864-8770-e1ed8d78f208"
 authors = ["ITensor developers <[email protected]> and contributors"]
-version = "0.5.11"
+version = "0.6.1"
 
 [deps]
 Accessors = "7d9f7c33-5ae7-4f3b-8dc6-eff91059b697"

src/abstractsparsearrayinterface.jl

@@ -4,14 +4,19 @@ using DerivableInterfaces:
 
 function eachstoredindex end
 function getstoredindex end
-function getunstoredindex end
+function getunstoredindex_function end
 function isstored end
 function setstoredindex! end
 function setunstoredindex! end
 function storedlength end
 function storedpairs end
 function storedvalues end
 
+@inline function getunstoredindex(a::AbstractArray{<:Any,N}, I::Vararg{Int,N}) where {N}
+  @boundscheck checkbounds(a, I...)
+  return getunstoredindex_function(a)(a, I...)
+end
+
 # Replace the function for accessing
 # unstored values.
 function set_getunstoredindex end
@@ -90,7 +95,7 @@ end
   a::AbstractArray, T::Type, size::Tuple{Vararg{Int}}
 )
   # TODO: Define `default_similartype` or something like that?
-  return SparseArrayDOK{T}(undef, size)
+  return SparseArrayDOK{T}(undef, size; getunstored=getunstoredindex_function(a))
 end
 
 # map over a specified subset of indices of the inputs.

src/sparsearraydok.jl

@@ -87,9 +87,10 @@ end
   @boundscheck checkbounds(a, I...)
   return storage(a)[CartesianIndex(I)]
 end
+@inline getunstoredindex_function(a::SparseArrayDOK) = a.getunstored
 @inline function getunstoredindex(a::SparseArrayDOK{<:Any,N}, I::Vararg{Int,N}) where {N}
   @boundscheck checkbounds(a, I...)
-  return a.getunstored(a, I...)
+  return getunstoredindex_function(a)(a, I...)
 end
 @inline function setstoredindex!(
   a::SparseArrayDOK{<:Any,N}, value, I::Vararg{Int,N}