Add a connection_index function for getting the index of a specific…

… connection matrix (#7)

* add a `connection_index` function for getting the index of a specific
connection matrix

* bump version

Project.toml

@@ -1,6 +1,6 @@
 name = "GraphDynamics"
 uuid = "bcd5d0fe-e6b7-4ef1-9848-780c183c7f4c"
-version = "0.1.4"
+version = "0.1.5"
 
 [deps]
 Accessors = "7d9f7c33-5ae7-4f3b-8dc6-eff91059b697"

src/GraphDynamics.jl

@@ -33,6 +33,7 @@ end
     isstochastic,
 
     event_times,
+    connection_index
 )
 
 export
@@ -244,6 +245,23 @@ Base.getindex(m::ConnectionMatrices, i) = m.matrices[i]
 Base.length(m::ConnectionMatrices) = length(m.matrices)
 Base.size(m::ConnectionMatrix{N}) where {N} = (N, N)
 
+"""
+    connection_index(ConnType, M::ConnectionMatrices)
+
+give the first index `n` such that `M[n]` is a `ConnectionMatrix{N, ConnType} where {N}`, or throw an error if no such index exists.
+"""
+connection_index(::Type{ConnType}, M::ConnectionMatrices) where {ConnType} = _conn_index(ConnType, M.matrices, 1)
+function _conn_index(::Type{ConnType}, tup::Tuple, i) where {ConnType}
+    if first(tup) isa ConnectionMatrix{N, ConnType} where {N}
+        return i
+    else
+        _conn_index(ConnType, Base.tail(tup), i+1)
+    end
+end
+@noinline _conn_index(::Type{ConnType}, ::Tuple{}, _) where {ConnType} =
+    error("ConnectionMatrices did not contain a ConnectionMatrix with connection type ", ConnType)
+
+
 abstract type GraphSystem end
 
 @kwdef struct ODEGraphSystem{CM <: ConnectionMatrices, S, P, EVT, CDEP, CCEP, Ns, SNM, PNM} <: GraphSystem