chemistry_balance.h

Go to the documentation of this file.

 
#ifndef CHEMISTRY_BALANCE_H
#define CHEMISTRY_BALANCE_H
 
#include <cstdint>
#include <functional>
#include <vector>
 
#include <Kokkos_Core.hpp>
 
#include "cell_chemistry.h"
#include "comm.h"
 
class ChemistryBalance {
    public:
        using Integrate = std::function<uint32_t(const ChemistryCells &, const Kokkos::View<uint32_t *> &,
                                                 const Kokkos::View<float *> &, uint32_t)>;
 
        ChemistryBalance(uint32_t n_species, double threshold);
 
        uint32_t send(const ChemistryCells & cells, const Kokkos::View<uint32_t *> & queue, uint32_t n_queued);
 
        uint32_t receive(const ChemistryCells & cells, const Kokkos::View<uint32_t *> & queue, const Integrate & integrate);
 
    private:
        struct Transfer {
            int rank;
            uint32_t first, count;  // cells: queue slots (sent) or guest slots (received)
        };
 
        void reserve_guests(uint32_t n);
 
        uint32_t n_species;
        double threshold;
        uint32_t state_stride = 0, result_stride;  // rtype values per cell in the messages
        uint32_t kept = 0, n_sent = 0, n_guests = 0;
        std::vector<Transfer> sends, receives;
        Kokkos::View<rtype *> send_buffer, guest_buffer;  // states, then results in place
        Kokkos::View<rtype *>::host_mirror_type h_send_buffer, h_guest_buffer;
        ChemistryCells guests;
        Kokkos::View<uint32_t *> guest_queue;
        Kokkos::View<float *> guest_cost;
        bool scaled = false;  // the cells carry a multiplier of dt
        bool forced = false;  // and a forcing
#ifdef Mallard_HAS_MPI
        std::vector<MPI_Request> requests;
#endif
};
 
#endif // CHEMISTRY_BALANCE_H