Mercurial > repos > AlgTools / file diff

--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/src/StructTools.jl	Mon Nov 18 11:00:54 2019 -0500
@@ -0,0 +1,41 @@
+#################################
+# Tools for working with structs
+#################################
+
+module StructTools
+
+##############
+# Our exports
+##############
+
+export replace,
+       IterableStruct
+
+######################################################
+# Replace entries by those given as keyword arguments
+######################################################
+
+function replace(base::T; kw...) where T
+    k = keys(kw)
+    T([n ∈ k ? kw[n] : getfield(base, n) for n ∈ fieldnames(T)]...)
+end
+
+#########################################################
+# Iteration of structs.
+# One only needs to make them instance of IterableStruct
+#########################################################
+
+abstract type IterableStruct end
+
+function Base.iterate(s::T) where T <: IterableStruct
+    return Base.iterate(s, (0, fieldnames(T)))
+end
+
+function Base.iterate(
+    s::T, st::Tuple{Integer,NTuple{N,Symbol}}
+) where T <: IterableStruct where N
+    (i, k)=st
+    return (i<N ? (getfield(s, i+1), (i+1, k)) : nothing)
+end
+
+end