Merge pull request #13 from brenhinkeller/main

Add equality comparisons on `MemoryBuffer`s

Project.toml

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

src/ManualMemory.jl

@@ -13,6 +13,17 @@ mutable struct MemoryBuffer{N,T}
 end
 @inline Base.unsafe_convert(::Type{Ptr{T}}, m::MemoryBuffer) where {T} = Ptr{T}(pointer_from_objref(m))
 @inline Base.pointer(m::MemoryBuffer{N,T}) where {N,T} = Ptr{T}(pointer_from_objref(m))
+@inline Base.:(==)(::MemoryBuffer, ::MemoryBuffer) = false
+@inline function Base.:(==)(a::MemoryBuffer{N,A}, b::MemoryBuffer{N,B}) where {N,A,B}
+    GC.@preserve a b begin
+        pa = pointer(a)
+        pb = pointer(b)
+        for n in 0:N-1
+            load(pa + n*offsetsize(A)) == load(pb + n*offsetsize(B)) || return false
+        end
+        return true
+    end
+end
 
 """
     PseudoPtr(data, position=firstindex(data))

test/runtests.jl

@@ -35,6 +35,13 @@ using Test
   # Test construction with existing data
   s = (1,2,3,4,5)
   @test MemoryBuffer(s).data === s
+
+  # Test equality and inequality
+  @test MemoryBuffer((1,2,3,4,5)) == MemoryBuffer((1,2,3,4,5))
+  @test MemoryBuffer((1,2,3,4,6)) != MemoryBuffer((1,2,3,4,5))
+  @test MemoryBuffer((0x01,0x02)) == MemoryBuffer((1,2))
+  @test MemoryBuffer((0x01,0x02)) != MemoryBuffer((0x01,0x02,0x03))
+
 end
 
 using ThreadingUtilities