Particle Data Structure

Creating a general AoSoA data structure with general types.

packages/Core/src/Cabana_AoSoA.hpp

@@ -223,6 +223,15 @@ class AoSoA
     // -------------------------------
     // Raw array data access
 
+    // Get the size of the data array at a given struct member index.
+    template<std::size_t I>
+    CABANA_INLINE_FUNCTION
+    std::size_t arraySize( const std::size_t s )
+    {
+        return
+            ( s < _num_soa - 1 ) ? soa_array_size : ( _size % soa_array_size );
+    }
+
     // Access the data array at a given struct member index.
     template<std::size_t I>
     CABANA_INLINE_FUNCTION