initial serialization support

Author: rayegun

Project.toml

@@ -12,17 +12,18 @@ MacroTools = "1914dd2f-81c6-5fcd-8719-6d5c9610ff09"
 Preferences = "21216c6a-2e73-6563-6e65-726566657250"
 Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
 SSGraphBLAS_jll = "7ed9a814-9cab-54e9-8e9e-d9e95b4d61b1"
+Serialization = "9e88b42a-f829-5b0c-bbe9-9e923198166b"
 SparseArrays = "2f01184e-e22b-5df5-ae63-d93ebab69eaf"
 SpecialFunctions = "276daf66-3868-5448-9aa4-cd146d93841b"
 StorageOrders = "e9177fbf-8fde-426c-9425-4eed0f22262a"
 SuiteSparse = "4607b0f0-06f3-5cda-b6b1-a6196a1729e9"
 
 [compat]
 ChainRulesCore = "1"
+HyperSparseMatrices = "0.2"
 MacroTools = "0.5"
 Preferences = "1"
 SSGraphBLAS_jll = "6.2.1"
 SpecialFunctions = "2"
-julia = "1.6"
-HyperSparseMatrices = "0.2"
 StorageOrders = "0.2"
+julia = "1.6"

src/SuiteSparseGraphBLAS.jl

@@ -21,6 +21,7 @@ using LinearAlgebra
 using Random: randsubseq, default_rng, AbstractRNG, GLOBAL_RNG
 using SpecialFunctions: lgamma, gamma, erf, erfc
 using Base.Broadcast
+using Serialization
 include("abstracts.jl")
 include("libutils.jl")
 
@@ -91,6 +92,7 @@ include("asjulia.jl")
 # include("spmgb/sparsemat.jl")
 include("mmread.jl")
 include("iterator.jl")
+include("serialization.jl")
 
 export SparseArrayCompat
 export LibGraphBLAS

src/serialization.jl

@@ -0,0 +1,27 @@
+function Serialization.serialize(s::AbstractSerializer, A::GBVecOrMat)
+    Serialization.writetag(s.io, Serialization.OBJECT_TAG)
+    Serialization.serialize(s, typeof(A))
+    Serialization.serialize(s, A.fill)
+    v = Vector{UInt8}(undef, serialize_sizehint(A))
+    sz = Ref{LibGraphBLAS.GrB_Index}()
+    @wraperror LibGraphBLAS.GrB_Matrix_serialize(v, sz, gbpointer(A))
+    resize!(v, sz[])
+    serialize(s, v)
+    return nothing
+end
+
+function Serialization.deserialize(s::AbstractSerializer, ::Type{GBMatrix{T, Tf}}) where {T, Tf}
+    fill = deserialize(s)
+    refA = Ref{LibGraphBLAS.GrB_Matrix}() # GrB will take care of size and such.
+    v = deserialize(s)
+    @wraperror LibGraphBLAS.GrB_Matrix_deserialize(refA, gbtype(T), v, LibGraphBLAS.GrB_Index(length(v)))
+    A = GBMatrix{T, Tf}(refA[], fill)
+    return A
+end
+
+function serialize_sizehint(A::GBVecOrMat)
+    sz = Ref{LibGraphBLAS.GrB_Index}()
+    @wraperror LibGraphBLAS.GrB_Matrix_serializeSize(sz, gbpointer(A))
+    return sz[]
+end
+