Skip to content

Get ParticleList of Vector/Matrix #798

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
kwitaechong wants to merge 1 commit into from
Closed

Get ParticleList of Vector/Matrix #798

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 8 additions & 0 deletions core/src/Cabana_Fields.hpp
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -16,6 +16,8 @@
#ifndef CABANA_FIELDS_HPP
#define CABANA_FIELDS_HPP

#include <Cabana_BatchedLinearAlgebra.hpp>

namespace Cabana
{
namespace Field
Expand Down Expand Up @@ -47,6 +49,8 @@ struct Scalar
static constexpr int size = 1;
//! Scalar type.
using data_type = value_type;
//! Linear algebra type.
using linear_algebra_type = value_type;
};

//! Vector (1D) particle field type.
Expand All @@ -63,6 +67,8 @@ struct Vector
static constexpr int dim0 = D;
//! Scalar type.
using data_type = value_type[D];
//! Linear algebra type.
using linear_algebra_type = Cabana::LinearAlgebra::VectorView<T, D>;
};

//! Matrix (2D) particle field type.
Expand All @@ -81,6 +87,8 @@ struct Matrix
static constexpr int dim1 = D1;
//! Scalar type.
using data_type = value_type[D0][D1];
//! Linear algebra type.
using linear_algebra_type = LinearAlgebra::MatrixView<T, D0, D1>;
};

//---------------------------------------------------------------------------//
Expand Down
53 changes: 53 additions & 0 deletions core/src/Cabana_ParticleList.hpp
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -172,6 +172,59 @@ get( ParticleView<VectorLength, FieldTags...>& particle, FieldTag,
particle.soa(), particle.vectorIndex(), indices... );
}

//---------------------------------------------------------------------------//
// Get a view of a particle member as a vector. (Works for both Particle
// and ParticleView)
template <class ParticleType, class FieldTag>
KOKKOS_FORCEINLINE_FUNCTION typename std::enable_if<
LinearAlgebra::is_vector<typename FieldTag::linear_algebra_type>::value,
typename FieldTag::linear_algebra_type>::type
get( ParticleType& particle, FieldTag tag )
{
return typename FieldTag::linear_algebra_type(
&( Cabana::get( particle, tag, 0 ) ), ParticleType::vector_length );
}

template <class ParticleType, class FieldTag>
KOKKOS_FORCEINLINE_FUNCTION typename std::enable_if<
LinearAlgebra::is_vector<typename FieldTag::linear_algebra_type>::value,
const typename FieldTag::linear_algebra_type>::type
get( const ParticleType& particle, FieldTag tag )
{
return typename FieldTag::linear_algebra_type(
const_cast<typename FieldTag::value_type*>(
&( Cabana::get( particle, tag, 0 ) ) ),
ParticleType::vector_length );
}

//---------------------------------------------------------------------------//
// Get a view of a particle member as a matrix. (Works for both Particle
// and ParticleView)
template <class ParticleType, class FieldTag>
KOKKOS_FORCEINLINE_FUNCTION typename std::enable_if<
LinearAlgebra::is_matrix<typename FieldTag::linear_algebra_type>::value,
typename FieldTag::linear_algebra_type>::type
get( ParticleType& particle, FieldTag tag )
{
return typename FieldTag::linear_algebra_type(
&( Cabana::get( particle, tag, 0, 0 ) ),
ParticleType::vector_length * FieldTag::dim1,
ParticleType::vector_length );
}

template <class ParticleType, class FieldTag>
KOKKOS_FORCEINLINE_FUNCTION typename std::enable_if<
LinearAlgebra::is_matrix<typename FieldTag::linear_algebra_type>::value,
const typename FieldTag::linear_algebra_type>::type
get( const ParticleType& particle, FieldTag tag )
{
return typename FieldTag::linear_algebra_type(
const_cast<typename FieldTag::value_type*>(
&( Cabana::get( particle, tag, 0, 0 ) ) ),
ParticleType::vector_length * FieldTag::dim1,
ParticleType::vector_length );
}

//---------------------------------------------------------------------------//
//! List of particle fields stored in AoSoA.
template <class MemorySpace, int VectorLength, class... FieldTags>
Expand Down