state.h

Go to the documentation of this file.

 
#ifndef STATE_H
#define STATE_H
 
#include <cstdint>
#include <string>
 
#include <Kokkos_Core.hpp>
 
#include "common_typedef.h"
 
using StateView = Kokkos::View<rtype *[N_CONSERVATIVE]>;
 
#ifdef Mallard_SPECIES_LAYOUT_LEFT
using SpeciesLayout = Kokkos::LayoutLeft;
#else
using SpeciesLayout = Kokkos::LayoutRight;
#endif
 
using SpeciesView = Kokkos::View<rtype **, SpeciesLayout>;
 
struct CellSpecies {
    const rtype * first;
    size_t stride;
    KOKKOS_INLINE_FUNCTION rtype operator[](const uint32_t k) const { return first[k * stride]; }
};
 
KOKKOS_INLINE_FUNCTION
CellSpecies cell_species(const SpeciesView & rhoY, const uint32_t c) {
    return CellSpecies{&rhoY(c, 0), rhoY.stride(1)};
}
 
struct State {
    StateView flow;
    SpeciesView species;
 
    State() = default;
    State(StateView flow_, SpeciesView species_) : flow(flow_), species(species_) {}
    State(const std::string & label, uint32_t n_cells, uint32_t n_species)
        : flow(label, n_cells), species(label + "_species", n_cells, n_species) {}
 
    uint32_t n_species() const { return static_cast<uint32_t>(species.extent(1)); }
};
 
inline void deep_copy(const State & dst, const State & src) {
    Kokkos::deep_copy(dst.flow, src.flow);
    if (src.species.span() > 0) Kokkos::deep_copy(dst.species, src.species);
}
 
template <typename ExecSpace>
inline void deep_copy(const ExecSpace & exec, const State & dst, const State & src) {
    Kokkos::deep_copy(exec, dst.flow, src.flow);
    if (src.species.span() > 0) Kokkos::deep_copy(exec, dst.species, src.species);
}
 
#endif // STATE_H