dist_grid_internal.h

Go to the documentation of this file.

// Copyright (c) 2024 The YAC Authors
//
// SPDX-License-Identifier: BSD-3-Clause
 
#ifndef DIST_GRID_INTERNAL_H
#define DIST_GRID_INTERNAL_H
 
#include "dist_grid.h"
#include "geometry.h"
 
struct yac_dist_grid;
typedef size_t const (* const const_size_t_2_pointer)[2];
typedef int const * const const_int_pointer;
typedef size_t const * const const_size_t_pointer;
typedef yac_int const * const const_yac_int_pointer;
typedef enum yac_edge_type const * const const_yac_edge_type_pointer;
typedef struct bounding_circle const * const const_bounding_circle_pointer;
typedef struct remote_point_infos const * const const_remote_point_infos_pointer;
 
struct yac_const_basic_grid_data {
  yac_const_coordinate_pointer vertex_coordinates;
  const_yac_int_pointer ids[3];
  const_int_pointer num_vertices_per_cell;
  const_size_t_pointer cell_to_vertex;
  const_size_t_pointer cell_to_vertex_offsets;
  const_size_t_pointer cell_to_edge;
  const_size_t_pointer cell_to_edge_offsets;
  const_size_t_2_pointer edge_to_vertex;
  const_bounding_circle_pointer cell_bnd_circles;
  const_yac_edge_type_pointer edge_type;
  const_remote_point_infos_pointer cell_owners;
  const_remote_point_infos_pointer vertex_owners;
  const_remote_point_infos_pointer edge_owners;
  const size_t count[3];
};
 
// information about a point on a remote process
struct remote_point_info {
  int rank;          // MPI rank
  uint64_t orig_pos; // local id on the remote process
};
 
struct remote_point_infos {
  int count;
  union {
    struct remote_point_info single; // valid if count == 1;
    struct remote_point_info * multi; // valid if count != -1;
  } data;
};
 
struct remote_point {
  yac_int global_id;
  struct remote_point_infos data;
};
 
struct remote_points {
  struct remote_point * data;
  size_t count;
  struct remote_point_info buffer[];
};
 
MPI_Comm yac_dist_grid_pair_get_MPI_Comm(
  struct yac_dist_grid_pair * grid_pair);
 
struct yac_dist_grid * yac_dist_grid_pair_get_dist_grid(
  struct yac_dist_grid_pair * grid_pair, char const * grid_name);
 
void yac_dist_grid_pair_do_point_search(
  struct yac_dist_grid_pair * grid_pair, char const * grid_name,
  yac_coordinate_pointer search_coords, size_t count, size_t * cells);
 
void yac_dist_grid_pair_do_point_search_gc(
  struct yac_dist_grid_pair * grid_pair, char const * grid_name,
  yac_coordinate_pointer search_coords, size_t count, size_t * cells);
 
void yac_dist_grid_pair_do_nnn_search(
  struct yac_dist_grid_pair * grid_pair, char const * grid_name,
  yac_coordinate_pointer search_coords, size_t count, size_t * local_ids,
  size_t n, struct yac_interp_field field);
 
void yac_dist_grid_pair_do_bnd_circle_search(
  struct yac_dist_grid_pair * grid_pair, char const * grid_name,
  const_bounding_circle_pointer bnd_circles, size_t count, size_t ** cells,
  size_t * num_results_per_bnd_circle, struct yac_interp_field field);
 
void yac_dist_grid_pair_do_cell_search(
  struct yac_dist_grid_pair * grid_pair,
  char const * search_grid_name, char const * result_grid_name,
  size_t * search_cells, size_t count, size_t ** result_cells,
  size_t * num_results_per_search_cell, struct yac_interp_field result_field);
 
void yac_dist_grid_pair_get_cell_neighbours(
  struct yac_dist_grid_pair * grid_pair, char const * grid_name,
  size_t * cells, size_t count, size_t * neighbours);
 
struct yac_const_basic_grid_data *
yac_dist_grid_get_basic_grid_data(struct yac_dist_grid * dist_grid);
 
size_t yac_dist_grid_get_local_count(
  struct yac_dist_grid * dist_grid, enum yac_location location);
 
void yac_dist_grid_get_local_unmasked_points(
  struct yac_dist_grid * grid, struct yac_interp_field field,
  size_t ** indices, size_t * count);
 
MPI_Datatype yac_get_remote_point_info_mpi_datatype(MPI_Comm comm);
 
int yac_remote_point_get_pack_size(
  struct remote_point * point, MPI_Datatype point_info_dt, MPI_Comm comm);
 
void yac_remote_point_pack(
  struct remote_point * point, void * buffer, int buffer_size, int * position,
  MPI_Datatype point_info_dt, MPI_Comm comm);
 
void yac_remote_point_unpack(
  void * buffer, int buffer_size, int * position, struct remote_point * point,
  MPI_Datatype point_info_dt, MPI_Comm comm);
 
int yac_remote_points_get_pack_size(
  struct remote_points * points, MPI_Datatype point_info_dt, MPI_Comm comm);
 
void yac_remote_points_pack(
  struct remote_points * points, void * buffer, int buffer_size, int * position,
  MPI_Datatype point_info_dt, MPI_Comm comm);
 
void yac_remote_points_unpack(
  void * buffer, int buffer_size, int * position,
  struct remote_points ** points, MPI_Datatype point_info_dt, MPI_Comm comm);
 
struct remote_point * yac_dist_grid_get_remote_points(
  struct yac_dist_grid * dist_grid, enum yac_location location,
  size_t * points, size_t count);
 
void yac_dist_grid_global_to_local(
  struct yac_dist_grid * dist_grid, enum yac_location location,
  yac_int * global_ids, size_t count, size_t * local_ids);
 
void yac_dist_grid_pair_get_aux_grid(
  struct yac_dist_grid_pair * grid_pair, char const * grid_name,
  size_t * cells, size_t count,
  size_t ** vertex_to_cell, size_t ** vertex_to_cell_offsets,
  int ** num_cells_per_vertex, struct yac_interp_field field);
 
void yac_dist_grid_pair_relocate_point_pairs(
  struct yac_dist_grid_pair * grid_pair, int a_is_ref, int to_dist_owner,
  char const * grid_name_a, size_t ** points_a, enum yac_location location_a,
  char const * grid_name_b, size_t ** points_b, enum yac_location location_b,
  double ** weights, size_t * count);
 
int const * yac_dist_grid_get_field_mask(
  struct yac_dist_grid * dist_grid, struct yac_interp_field field);
 
yac_const_coordinate_pointer yac_dist_grid_get_field_coords(
  struct yac_dist_grid * dist_grid, struct yac_interp_field field);
 
void yac_const_basic_grid_data_get_grid_cell(
  struct yac_const_basic_grid_data * grid_data, size_t cell_idx,
  struct yac_grid_cell * cell);
 
void yac_dist_grid_pair_determine_dist_owner(
  struct yac_dist_grid_pair * grid_pair, char const * grid_name,
  size_t * points, size_t count, enum yac_location location, int * ranks);
 
void yac_dist_grid_pair_get_vertex_neighbours(
  struct yac_dist_grid_pair * grid_pair, char const * grid_name,
  size_t * vertices, size_t count, size_t ** neigh_vertices,
  int * num_neighs_per_vertex, struct yac_interp_field field);
 
void yac_dist_grid_pair_get_corner_cells(
  struct yac_dist_grid_pair * grid_pair, char const * grid_name,
  size_t * vertices, size_t count, size_t ** vertex_to_cell,
  size_t * num_cells_per_vertex);
 
#endif // DIST_GRID_INTERNAL_H