Add preserve_buffer

Author: chriselrod

Project.toml

@@ -1,7 +1,7 @@
 name = "ManualMemory"
 uuid = "d125e4d3-2237-4719-b19c-fa641b8a4667"
 authors = ["chriselrod <[email protected]> and contributors"]
-version = "0.1.0"
+version = "0.1.1"
 
 [compat]
 julia = "1.4"

src/ManualMemory.jl

@@ -17,13 +17,20 @@ end
     Expr(:block, Expr(:meta,:inline), :(ccall(:jl_value_ptr, Ref{$T}, (Ptr{Cvoid},), unsafe_load(Base.unsafe_convert(Ptr{Ptr{Cvoid}}, p)))))
   end
 end
-@inline load(p::Ptr{UInt}, ::Type{T}) where {T} = load(reinterpret(Ptr{T}, p))
 @generated function store!(p::Ptr{T}, v::T) where {T}
   if Base.allocatedinline(T)
     Expr(:block, Expr(:meta,:inline), :(unsafe_store!(p, v); return nothing))
   else
     Expr(:block, Expr(:meta,:inline), :(unsafe_store!(Base.unsafe_convert(Ptr{Ptr{Cvoid}}, p), Base.pointer_from_objref(v)); return nothing))
   end
 end
+@generated offsetsize(::Type{T}) where {T} = Base.allocatedinline(T) ? sizeof(T) : sizeof(Int)
+
+@inline store!(p::Ptr{T}, v) where {T} = store!(p, convert(T, v))
+@inline preserve_buffer(x) = x
+@inline preserve_buffer(A::AbstractArray) = _preserve_buffer(A, parent(A))
+@inline _preserve_buffer(a::A, p::P) where {A,P<:AbstractArray} = _preserve_buffer(p, parent(p))
+@inline _preserve_buffer(a::A, p::A) where {A<:AbstractArray} = a
+@inline _preserve_buffer(a::A, p::P) where {A,P} = p
 
 end