ReverseDiff support. Closes #37 closes #78

Author: jonniedie

Project.toml

@@ -1,7 +1,7 @@
 name = "ComponentArrays"
 uuid = "b0b7db55-cfe3-40fc-9ded-d10e2dbeff66"
 authors = ["Jonnie Diegelman <47193959+jonniedie@users.noreply.github.com>"]
-version = "0.9.4"
+version = "0.9.5"
 
 [deps]
 ArrayInterface = "4fba245c-0d91-5ea0-9b3e-6abc04ee57a9"

src/ComponentArrays.jl

@@ -35,6 +35,7 @@ function __init__()
     @require StaticArrays="90137ffa-7385-5640-81b9-e52037218182" required("staticarrays.jl")
     # @require Zygote="e88e6eb3-aa80-5325-afca-941959d7151f" required("zygote.jl")
     @require ChainRulesCore="d360d2e6-b24c-11e9-a2a3-2a2ae2dbcce4" required("chainrulescore.jl")
+    @require ReverseDiff="37e2e3b7-166d-5795-8a7a-e32c996b4267" required("reversediff.jl")
 end
 
 

src/if_required/reversediff.jl

@@ -0,0 +1,22 @@
+const TrackedComponentArray{V, D, N, DA, N, A, Ax} = ReverseDiff.TrackedArray{V,D,N,ComponentArray{V,N,A,Ax},DA}
+
+maybe_tracked_array(val::AbstractArray, der, t) = ReverseDiff.TrackedArray(val, der, t)
+maybe_tracked_array(val, der, t) = ReverseDiff.TrackedReal(val, der, t)
+
+function Base.getindex(tca::TrackedComponentArray, inds::Union{Symbol, Val}...)
+        val = ReverseDiff.value(tca)[inds...]
+        der = ReverseDiff.deriv(tca)[inds...]
+        t = ReverseDiff.tape(tca)
+    return maybe_tracked_array(val, der, t)
+end
+
+function Base.getproperty(tca::TrackedComponentArray, s::Symbol)
+    if s in (:value, :deriv, :tape)
+        return getfield(tca, s)
+    else
+        val = getproperty(ReverseDiff.value(tca), s)
+        der = getproperty(ReverseDiff.deriv(tca), s)
+        t = ReverseDiff.tape(tca)
+        return maybe_tracked_array(val, der, t)
+    end
+end

src/plot_utils.jl

@@ -86,7 +86,7 @@ see also `labels`
 """
 label2index(x::ComponentVector, str) = label2index(labels(x), str)
 function label2index(labs, str)
-    idx = findall(startswith.(labs, Regex("\\Q$str\\E(?:(\\.|\\[))"))) #str * r"(\.|\[)"))
+    idx = findall(startswith.(labs, Regex("\\Q$str\\E(?:(\\.|\\[))")))
     if !isempty(idx)
         return idx
     else