Add hash functions

Project.toml

@@ -1,7 +1,7 @@
 name = "PeriodicGraphs"
 uuid = "18c5b727-b240-4874-878a-f2e242435bab"
 authors = ["Lionel Zoubritzky [email protected]"]
-version = "0.4.0"
+version = "0.5.0"
 
 [deps]
 Graphs = "86223c79-3864-5bf0-83f7-82e725a168b6"

src/PeriodicGraphs.jl

@@ -13,7 +13,7 @@ export PeriodicVertex, PeriodicEdge, PeriodicGraph,
        dimensionality, change_dimension
 
 import Base: (==), isless, convert, show, showerror, eltype, iterate, zero,
-             length, in, ndims, print, cmp
+             length, in, ndims, print, cmp, hash
 import Base.Order: Forward, Lt
 
 """
@@ -54,6 +54,7 @@ function cmp(x::PeriodicVertex{N}, y::PeriodicVertex{N}) where N
 end
 isless(x::PeriodicVertex{N}, y::PeriodicVertex{N}) where {N} = cmp(x, y) < 0
 ==(x::PeriodicVertex{N}, y::PeriodicVertex{M}) where {N,M} = N == M && iszero(cmp(x, y))
+hash(x::PeriodicVertex, h::UInt) = hash(x.ofs, hash(x.v, h))
 
 const PeriodicVertex1D = PeriodicVertex{1}
 const PeriodicVertex2D = PeriodicVertex{2}
@@ -203,6 +204,7 @@ function cmp(x::PeriodicEdge{N}, y::PeriodicEdge{N}) where N
 end
 isless(x::PeriodicEdge{N}, y::PeriodicEdge{N}) where {N} = cmp(x, y) < 0
 ==(x::PeriodicEdge{N}, y::PeriodicEdge{M}) where {N,M} = N == M && iszero(cmp(x, y))
+hash(x::PeriodicEdge, h::UInt) = hash(x.dst, hash(x.src, h))
 
 
 """
@@ -402,6 +404,14 @@ function ==(g1::PeriodicGraph{N}, g2::PeriodicGraph{M}) where {N,M}
     return N == M && nv(g1) == nv(g2) && edges(g1) == edges(g2)
 end
 
+function hash(g::PeriodicGraph, h::UInt)
+    ret = h
+    for e in edges(g)
+        ret = hash(e, ret)
+    end
+    return ret
+end
+
 ndims(::PeriodicVertex{N}) where {N} = N
 ndims(::PeriodicEdge{N}) where {N} = N
 ndims(::PeriodicGraph{N}) where {N} = N

src/precompile.jl

@@ -26,9 +26,12 @@ function _precompile_()
         @enforce Base.precompile(Tuple{typeof(==),PeriodicVertex{i},PeriodicVertex{i}})
         @enforce Base.precompile(Tuple{typeof(<),PeriodicVertex{i},PeriodicVertex{i}})
         @enforce Base.precompile(Tuple{typeof(isless),PeriodicVertex{i},PeriodicVertex{i}})
+        @enforce Base.precompile(Tuple{typeof(hash),PeriodicVertex{i},UInt})
         @enforce Base.precompile(Tuple{typeof(==),PeriodicEdge{i},PeriodicEdge{i}})
         @enforce Base.precompile(Tuple{typeof(<),PeriodicEdge{i},PeriodicEdge{i}})
+        @enforce Base.precompile(Tuple{typeof(hash),PeriodicEdge{i},UInt})
         @enforce Base.precompile(Tuple{typeof(==),PeriodicGraph{i},PeriodicGraph{i}})
+        @enforce Base.precompile(Tuple{typeof(hash),PeriodicGraph{i},UInt})
 
         @enforce Base.precompile(Tuple{typeof(Graphs._neighborhood),PeriodicGraph{i},Int,Int,Graphs.DefaultDistance,Function})
         @enforce Base.precompile(Tuple{typeof(Graphs.add_edge!),PeriodicGraph{i},PeriodicEdge{i}})