Merge pull request #36 from JuliaDiffEq/aos

general function for AoS -> SoA

Author: ChrisRackauckas

Project.toml

@@ -1,6 +1,6 @@
 name = "ArrayInterface"
 uuid = "4fba245c-0d91-5ea0-9b3e-6abc04ee57a9"
-version = "2.3.1"
+version = "2.4.0"
 
 [deps]
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"

README.md

@@ -16,6 +16,10 @@ and exported in a future Base Julia there will be no issues with the upgrade.
 A trait function for whether `x` is a mutable or immutable array. Used for
 dispatching to in-place and out-of-place versions of functions.
 
+## aos_to_soa(x)
+
+Converts an array of structs formulation to a struct of arrays.
+
 ## isstructured(x)
 
 A trait function for whether a matrix `x` is a sparse structured matrix.
@@ -84,7 +88,7 @@ development.
 
 ## Breaking Release Notes
 
-2.0: Changed the default of `ismutable(array::AbstractArray) = true`. We previously defaulted to 
+2.0: Changed the default of `ismutable(array::AbstractArray) = true`. We previously defaulted to
 `Base.@pure ismutable(array::AbstractArray) = typeof(array).mutable`, but there are a lot of cases
 where this tends to not work out in a way one would expect. For example, if you put a normal array
 into an immutable struct that adds more information to it, this is considered immutable, even if

src/ArrayInterface.jl

@@ -44,6 +44,13 @@ can_setindex(x) = true
 can_setindex(x::AbstractArray) = can_setindex(typeof(x))
 can_setindex(::Type{<:AbstractRange}) = false
 
+"""
+    aos_to_soa(x)
+
+Converts an array of structs formulation to a struct of array
+"""
+aos_to_soa(x) = x
+
 """
     fast_scalar_indexing(x)
 
@@ -389,6 +396,7 @@ function __init__()
     ismutable(::Type{<:Tracker.TrackedArray}) = false
     can_setindex(::Type{<:Tracker.TrackedArray}) = false
     fast_scalar_indexing(::Type{<:Tracker.TrackedArray}) = false
+    aos_to_soa(x::AbstractArray{<:Tracker.TrackedArray,N}) where N = Tracker.collect(x)
   end
 
   @require CuArrays="3a865a2d-5b23-5a0f-bc46-62713ec82fae" begin