perf_toy_cube.c

Go to the documentation of this file.

// Copyright (c) 2024 The YAC Authors
//
// SPDX-License-Identifier: BSD-3-Clause
 
#include <mpi.h>
#include <stdio.h>
#include "yac.h"
#include "yac_utils.h"
#include "perf_toy_common.h"
#include "perf_toy_cube.h"
 
void init_cube(
  int comp_id, size_t n_cube, int * cell_point_id, int * corner_point_id) {
 
  // get cube component communicator
  MPI_Comm comp_comm;
  yac_cget_comp_comm(comp_id, &comp_comm);
  int comp_rank;
  MPI_Comm_rank(comp_comm,&comp_rank);
 
  // generate cubed sphere grid filename based on edge length
  char cube_grid_filename[256];
  sprintf(cube_grid_filename, "c_%zu.nc", n_cube);
 
  // check on root process if the grid file already exists
  if ((comp_rank == 0) && !yac_file_exists(cube_grid_filename)) {
 
    // generate cubed sphere grid and write it to file
    // (file format matches the one of the ICON grid, so that the same reading
    //  function can be used)
    void * write_timer =
      start_timer(MPI_COMM_SELF, "%s;%zu", "write_cube_grid", n_cube);
    yac_write_cubed_sphere_grid(n_cube, cube_grid_filename);
    stop_timer(write_timer);
  }
 
  // wait for rank 0 to finish writing the grid file (if not already existing)
  MPI_Barrier(comp_comm);
 
  // initialize grid data and register it with YAC
  init_grid(
    comp_id, "cube_grid", cube_grid_filename, cell_point_id, corner_point_id);
}