dynamic_config.c

Go to the documentation of this file.

// Copyright (c) 2024 The YAC Authors
//
// SPDX-License-Identifier: BSD-3-Clause
 
#define VERBOSE
 
#include <mpi.h>
#include <stdlib.h>
#include <stdio.h>
#include "yac.h"
 
#define YAC_RAD (0.01745329251994329576923690768489) // M_PI / 180
 
int main () {
  // Initialisation of MPI
 
  MPI_Init (0, NULL);
 
  /* The initialisation phase */
#ifdef VERBOSE
  printf (". main: calling yac_cinit\n");
#endif
 
  // start with an empty config
  yac_cinit ();
  yac_cdef_calendar(YAC_PROLEPTIC_GREGORIAN);
  yac_cdef_datetime("2022-11-11T13:18:00", "2022-11-12T13:18:00");
 
  int global_rank, global_size;
  MPI_Comm_rank(MPI_COMM_WORLD,&global_rank);
  MPI_Comm_size(MPI_COMM_WORLD,&global_size);
 
  /* Dynamic component definition. */
 
#ifdef VERBOSE
  printf (". main: calling yac_cdef_comp\n");
#endif
 
  int comp_id;
  char * comp_name;
  if(global_rank == 0)
    comp_name = "ICON-atmosphere";
  else if(global_rank == 1)
    comp_name = "ICON-ocean";
  else if(global_rank == 2)
    comp_name = "VIZ";
  else
    comp_name = "OUTPUT";
 
  yac_cdef_comp ( comp_name, &comp_id );
#ifdef VERBOSE
  printf ( ". main: defined %s with local comp ID %i \n", comp_name, comp_id );
#endif
 
  MPI_Comm localcomm;
  yac_cget_comp_comm(comp_id, &localcomm);
  int comp_size;
  MPI_Comm_size(localcomm, &comp_size);
#ifdef VERBOSE
  printf ( ". main: component %s has %i processes\n", comp_name, comp_size );
#endif
 
  MPI_Comm_free(&localcomm);
 
  // register grids (for simplicity one grid per comp)
  int num_vert[2] = {3,3};
  int cyclic[2] = {0,0};
  double vertices[3] = {0*YAC_RAD, 1.*YAC_RAD, 2.*YAC_RAD};
  int grid_id;
  yac_cdef_grid_reg2d(comp_name, num_vert, cyclic,
    vertices, vertices, &grid_id);
 
  int point_id;
  yac_cdef_points_reg2d(grid_id,
    num_vert,YAC_LOCATION_CORNER, vertices, vertices, &point_id);
 
  int t_id, t2_id, s_id;
  yac_cdef_field(
    "t", comp_id, &point_id, 1, 1, "1", YAC_TIME_UNIT_SECOND, &t_id);
  yac_cdef_field(
    "t2", comp_id, &point_id, 1, 1, "1", YAC_TIME_UNIT_SECOND, &t2_id);
  if(global_rank < 2)
    yac_cdef_field(
      "s", comp_id, &point_id, 1, 1, "1", YAC_TIME_UNIT_SECOND, &s_id);
 
  if(global_rank != 1)
    yac_csync_def();
  printf("synced!\n");
 
  if(global_rank == 0){
    int interp_stack_config_id;
    yac_cget_interp_stack_config(&interp_stack_config_id);
    yac_cadd_interp_stack_config_nnn(
      interp_stack_config_id, YAC_NNN_AVG, 1, 0.0, 1.);
    if(global_size > 1){
      yac_cdef_couple("ICON-atmosphere", "ICON-atmosphere", "t", // source (comp, grid, field)
        "ICON-ocean", "ICON-ocean", "t", // target (comp, grid, field)
        "1", YAC_TIME_UNIT_SECOND, YAC_REDUCTION_TIME_AVERAGE,
        interp_stack_config_id, 0, 0);
      yac_cdef_couple("ICON-atmosphere", "ICON-atmosphere", "t2", // source (comp, grid, field)
        "ICON-ocean", "ICON-ocean", "t2", // target (comp, grid, field)
        "1", YAC_TIME_UNIT_SECOND, YAC_REDUCTION_TIME_AVERAGE,
        interp_stack_config_id,  0, 0);
      yac_cdef_couple("ICON-ocean", "ICON-ocean", "s", // source (comp, grid, field)
        "ICON-atmosphere", "ICON-atmosphere", "s", // target (comp, grid, field)
        "1", YAC_TIME_UNIT_SECOND, YAC_REDUCTION_TIME_AVERAGE,
        interp_stack_config_id,  0, 0);
      yac_cfree_interp_stack_config(interp_stack_config_id);
    }
  }else if (global_rank == 2){
    int interp_stack_config_id;
    yac_cget_interp_stack_config(&interp_stack_config_id);
    yac_cadd_interp_stack_config_nnn(
      interp_stack_config_id, YAC_NNN_AVG, 1, 0.0, 1.);
    yac_cdef_couple("ICON-atmosphere", "ICON-atmosphere", "t", // source (comp, grid, field)
      "VIZ", "VIZ", "t", // target (comp, grid, field)
      "1", YAC_TIME_UNIT_SECOND, YAC_REDUCTION_TIME_AVERAGE,
      interp_stack_config_id,  0, 0);
    if(global_size > 3) // add couple for two other components
      yac_cdef_couple("ICON-atmosphere", "ICON-atmosphere", "t", // source (comp, grid, field)
        "OUTPUT", "OUTPUT", "t", // target (comp, grid, field)
        "1", YAC_TIME_UNIT_SECOND, YAC_REDUCTION_TIME_AVERAGE,
        interp_stack_config_id,  0, 0);
    yac_cfree_interp_stack_config(interp_stack_config_id);
  }
 
  int ierr;
  yac_cenddef( );
 
  double * data = malloc(sizeof(double) * num_vert[0]*num_vert[1]);
  int info;
  if(global_rank == 0){
    for(int i = 0; i < num_vert[0]*num_vert[1]; ++i)
      data[i] = 42.+i;
    double ** data_ptr = &data;
    yac_cput(t_id, 1, &data_ptr, &info, &ierr);
    for(int i = 0; i < num_vert[0]*num_vert[1]; ++i)
      data[i] = -42.+i;
    yac_cput(t2_id, 1, &data_ptr, &info, &ierr);
  }else{
    yac_cget(t2_id, 1, &data, &info, &ierr);
    printf("info: %d\nerr: %d\n%f\n", info, ierr, data[2]);
    yac_cget(t_id, 1, &data, &info, &ierr);
    printf("info: %d\nerr: %d\n%f\n", info, ierr, data[2]);
  }
 
  yac_cfinalize();
 
  MPI_Finalize();
 
  return EXIT_SUCCESS;
}