distribution.h

Go to the documentation of this file.

 
#ifndef DISTRIBUTION_H
#define DISTRIBUTION_H
 
#include <cstdint>
#include <vector>
 
inline constexpr const char * PARTITION_ZONE = "__partition__";
 
struct Distribution {
    uint32_t n_owned = 0;
    uint8_t halo_layers = 0;
    std::vector<uint64_t> global_cell;  // local cell -> global cell
    std::vector<uint8_t> layer;         // 0 for owned cells, k for halo layer k
    std::vector<int> neighbors;         // ranks this rank exchanges with
    std::vector<std::vector<uint32_t>> send_cells;  // per neighbor: owned cells it needs
    std::vector<std::vector<uint32_t>> recv_cells;  // per neighbor: halo cells it owns
};
 
void plan_halo_exchange(Distribution & dist, const std::vector<int> & halo_owner);
 
void trim_halo_exchange(Distribution & dist, const std::vector<uint8_t> & needed);
 
#endif // DISTRIBUTION_H