particles.hpp

Go to the documentation of this file.

#ifndef PARTICLES_HPP
#define PARTICLES_HPP
#include <Cabana_AoSoA.hpp>
#include <Kokkos_Core.hpp>
#include "globals.hpp"

// Set the type for the particle AoSoA.
using ParticleDataTypes = Cabana::MemberTypes<double[6],double[3],long long int>;

// Phase
struct SimdPhase{
    Simd<double> r;
    Simd<double> z;
    Simd<double> phi;
    Simd<double> rho;
    Simd<double> w1;
    Simd<double> w2;
};

// Constants
struct SimdConstants{
    Simd<double> mu;
    Simd<double> w0;
    Simd<double> f0;
};

// SoA particle structure, with inner array of length SIMD_SIZE (generally, VEC_LEN on CPU, 1 on GPU)
struct SimdParticles{
    SimdPhase ph;
    SimdConstants ct;
    Simd<long long int> gid;
};

// Phase
struct VecPhase{
    double r[VEC_LEN];
    double z[VEC_LEN];
    double phi[VEC_LEN];
    double rho[VEC_LEN];
    double w1[VEC_LEN];
    double w2[VEC_LEN];
};

// Constants
struct VecConstants{
    double mu[VEC_LEN];
    double w0[VEC_LEN];
    double f0[VEC_LEN];
};

// SoA particle structure, with inner array of length VEC_LEN (just like the AoSoA)
struct VecParticles{
    VecPhase ph;
    VecConstants ct;
    long long int gid[VEC_LEN];
};

KOKKOS_INLINE_FUNCTION void remove_particles(Simd<long long int>& gid, const Simd<bool>& deactivate);

KOKKOS_INLINE_FUNCTION void update_phases(SimdPhase &ph_new, const SimdPhase &ph, double local_dt, const SimdPhase &dph);

KOKKOS_INLINE_FUNCTION void simd_2_AoSoA(VecParticles *part, const SimdParticles &part_one, int a_vec, int s_vec);

KOKKOS_INLINE_FUNCTION void AoSoA_2_simd(SimdParticles &part_one, const VecParticles *part, int a_vec, int s_vec);

KOKKOS_INLINE_FUNCTION void AoSoA_indices(int i_item,int &s_vec,int &a_vec);

KOKKOS_INLINE_FUNCTION void update_dphm(SimdPhase &dphm,const SimdPhase &dpht);

KOKKOS_INLINE_FUNCTION void update_dpht(SimdPhase &dpht,const SimdPhase &dph,const SimdPhase &dphm);

int p_range(int num_particle);

#include "particles.tpp"
#endif