YetAnotherCoupler 3.2.0_a
Loading...
Searching...
No Matches
test_dummy_coupling_c.c

This example simulates a whole model setup with three components (ocean, atmosphere, io). It uses one process for each component.

// Copyright (c) 2024 The YAC Authors
//
// SPDX-License-Identifier: BSD-3-Clause
#include <mpi.h>
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include "tests.h"
#include "test_common.h"
#include "yac.h"
#include "basic_grid.h"
#define NBR_CELLS 2
#define NBR_VERTICES 4
struct {
char const * name;
int collection_size;
} field[] =
{{.name = "TAUX", .collection_size = 2},
{.name = "TAUY", .collection_size = 2},
{.name = "SFWFLX", .collection_size = 3},
{.name = "SFTEMP", .collection_size = 1},
{.name = "THFLX", .collection_size = 4},
{.name = "ICEATM", .collection_size = 4},
{.name = "SST", .collection_size = 1},
{.name = "OCEANU", .collection_size = 1},
{.name = "OCEANV", .collection_size = 1},
{.name = "ICEOCE", .collection_size = 5}};
char * yaml_filename;
enum {no_of_fields = sizeof(field) / sizeof(field[0])};
const int nbr_cells = NBR_CELLS;
const int nbr_vertices = NBR_VERTICES;
int nbr_vertices_per_cell[NBR_CELLS];
int info, ierror;
int comp_id;
int comp_ids[1];
int cell_point_ids[1];
int cell_mask_ids[1];
int grid_id;
int global_index[NBR_CELLS];
int cell_core_mask[NBR_CELLS];
double * buffer;
double * buffer_lon;
double * buffer_lat;
int * cell_to_vertex;
int * field_id;
int * cell_mask;
int flag;
static void dummy_atmosphere ();
static void dummy_ocean ();
static void dummy_io ();
int main(int argc, char** argv) {
yac_cinit();
int size, rank;
MPI_Comm_rank ( MPI_COMM_WORLD, &rank );
MPI_Comm_size ( MPI_COMM_WORLD, &size );
if (argc != 2) {
PUT_ERR("ERROR: missing config file directory");
xt_finalize();
MPI_Finalize();
return TEST_EXIT_CODE;
}
yaml_filename =
strcat(
strcpy(
malloc(strlen(argv[1]) + 32), argv[1]), "coupling_test.yaml");
switch ( rank ) {
case 0:
dummy_atmosphere ( );
break;
case 1:
dummy_ocean ( );
break;
case 2:
dummy_io ( );
break;
default:
PUT_ERR("Too many processes have been launched\n");
return TEST_EXIT_CODE;
}
free(yaml_filename);
yac_cfinalize ();
return TEST_EXIT_CODE;
}
/* -------------------------------------------------------------------- */
static void dummy_atmosphere () {
int size, rank;
MPI_Comm local_comm;
char * comp_name = "dummy_atmosphere";
char * grid_name = "dummy_atmosphere_grid";
// Inform the coupler about what we are
yac_cdef_comp ( comp_name, &comp_id );
comp_ids[0] = comp_id;
yac_cget_comp_comm ( comp_id, &local_comm );
MPI_Comm_rank ( local_comm, &rank );
MPI_Comm_size ( local_comm, &size );
// printf (" %s rank %d : local size %d \n", comp_name, rank, size );
buffer_lon = xmalloc ( nbr_vertices * sizeof(*buffer_lon) );
buffer_lat = xmalloc ( nbr_vertices * sizeof(*buffer_lat) );
cell_to_vertex = xmalloc ( 3 * nbr_cells * sizeof(*cell_to_vertex) );
/* Define vertices
0
/ \
/ o \
/ \
1-------2 Eq.
\ /
\ o /
\ /
3
*/
buffer_lon[0] = 0.0; buffer_lat[0] = 1.0;
buffer_lon[1] = -1.0; buffer_lat[1] = 0.0;
buffer_lon[2] = 1.0; buffer_lat[2] = 0.0;
buffer_lon[3] = 0.0; buffer_lat[3] = -1.0;
// Connectivity
cell_to_vertex[0] = 0; cell_to_vertex[1] = 1; cell_to_vertex[2] = 2; // cell 1
cell_to_vertex[3] = 1; cell_to_vertex[4] = 3; cell_to_vertex[5] = 2; // cell 2
for (unsigned i = 0; i < nbr_cells; ++i ) nbr_vertices_per_cell[i] = 3;
// Define unstructured grid
yac_cdef_grid_unstruct(
grid_name, nbr_vertices, nbr_cells, nbr_vertices_per_cell,
buffer_lon, buffer_lat, cell_to_vertex, &grid_id);
// Decomposition information
for (int i = 0; i < nbr_cells; ++i ) {
global_index[i] = i;
cell_core_mask[i] = 1;
}
yac_cset_global_index(global_index, YAC_LOC_CELL, grid_id);
yac_cset_core_mask(cell_core_mask, YAC_LOC_CELL, grid_id);
// Center points in cells (needed e.g. for nearest neighbour)
buffer_lon[0] = 0.0; buffer_lat[0] = 0.5;
buffer_lon[1] = 0.0; buffer_lat[1] = -0.5;
yac_cdef_points_unstruct(
grid_id, nbr_cells, YAC_LOC_CELL, buffer_lon, buffer_lat, &cell_point_ids[0]);
free ( buffer_lon );
free ( buffer_lat );
// Mask generation
cell_mask = xmalloc (nbr_cells * sizeof(*cell_mask));
for (unsigned i = 0; i < nbr_cells; ++i) cell_mask[i] = 1;
yac_cdef_mask ( grid_id, nbr_cells, YAC_LOC_CELL, cell_mask, cell_mask_ids );
yac_cset_mask(cell_mask, cell_point_ids[0]);
free (cell_mask);
yac_cdef_calendar(YAC_PROLEPTIC_GREGORIAN);
field_id = xmalloc ( no_of_fields * sizeof(*field_id));
for (unsigned i = 0; i < no_of_fields/2; ++i )
yac_cdef_field ( field[i].name,
comp_id,
cell_point_ids,
1, field[i].collection_size,
"600", YAC_TIME_UNIT_SECOND,
&field_id[i] );
for (unsigned i = no_of_fields/2; i < no_of_fields; ++i )
yac_cdef_field_mask ( field[i].name,
comp_id,
cell_point_ids,
cell_mask_ids,
1, field[i].collection_size,
"600", YAC_TIME_UNIT_SECOND,
&field_id[i] );
// read couplings from YAML configuration file
yac_cread_config_yaml(yaml_filename);
yac_cenddef ( );
// these queries can only be fulfilled once the search has been completed.
for (unsigned i = 0; i < no_of_fields; ++i ) {
const char* timestep_string = yac_cget_timestep_from_field_id ( field_id[i] );
if (strcmp(timestep_string,"PT10M")) PUT_ERR ( "wrong model time step\n");
int collection_size;
int const ref_collection_size[no_of_fields] = {2,2,3,1,4,4,1,1,1,5};
collection_size = yac_cget_collection_size_from_field_id( field_id[i] );
if (collection_size != ref_collection_size[i])
PUT_ERR ( "wrong collection size\n");
int role = yac_cget_role_from_field_id(field_id[i] );
if ((i < 6) && (role != 1)) PUT_ERR( "Wrong requested role\n" );
if ((i > 5) && (role != 2)) PUT_ERR( "Wrong requested role\n" );
}
// Data exchange
buffer = xmalloc (5 * nbr_cells * sizeof (*buffer));
/* field_id[0] represents "TAUX" wind stress component
field_id[1] represents "TAUY" wind stress component
field_id[2] represents "SFWFLX" surface fresh water flux
field_id[3] represents "SFTEMP" surface temperature
field_id[4] represents "THFLX" total heat flux
field_id[5] represents "ICEATM" ice temperatures and melt potential
field_id[6] represents "SST" sea surface temperature
field_id[7] represents "OCEANU" u component of ocean surface current
field_id[8] represents "OCEANV" v component of ocean surface current
field_id[9] represents "ICEOCE" ice thickness, concentration and temperatures
*/
// Send fields to ocean
// --------------------
{
double *point_set_data[5];
double **collection_data[5];
for (int i = 0; i < 5; ++i) {
point_set_data[i] = buffer + i * nbr_cells;
collection_data[i] = &(point_set_data[i]);
}
// meridional wind stress
for (unsigned i = 0; i < nbr_cells; ++i ) buffer[0*nbr_cells+i] = 10.1;
for (unsigned i = 0; i < nbr_cells; ++i ) buffer[1*nbr_cells+i] = 10.2;
yac_cput ( field_id[0], 2, collection_data, &info, &ierror );
// zonal wind stress
for (unsigned i = 0; i < nbr_cells; ++i ) buffer[0*nbr_cells+i] = 20.1;
for (unsigned i = 0; i < nbr_cells; ++i ) buffer[1*nbr_cells+i] = 20.2;
yac_cput ( field_id[1], 2, collection_data, &info, &ierror );
// surface fresh water flux
for (unsigned i = 0; i < nbr_cells; ++i ) buffer[0*nbr_cells+i] = 30.1;
for (unsigned i = 0; i < nbr_cells; ++i ) buffer[1*nbr_cells+i] = 30.2;
for (unsigned i = 0; i < nbr_cells; ++i ) buffer[2*nbr_cells+i] = 30.3;
yac_cput ( field_id[2], 3, collection_data, &info, &ierror );
// surface temperature
for (unsigned i = 0; i < nbr_cells; ++i ) buffer[0*nbr_cells+i] = 40.1;
yac_cput ( field_id[3], 1, collection_data, &info, &ierror );
// total heat flux
for (unsigned i = 0; i < nbr_cells; ++i ) buffer[0*nbr_cells+i] = 50.1;
for (unsigned i = 0; i < nbr_cells; ++i ) buffer[1*nbr_cells+i] = 50.2;
for (unsigned i = 0; i < nbr_cells; ++i ) buffer[2*nbr_cells+i] = 50.3;
for (unsigned i = 0; i < nbr_cells; ++i ) buffer[3*nbr_cells+i] = 50.4;
yac_cput ( field_id[4], 4, collection_data, &info, &ierror );
// ice temperatures and melt potential
for (unsigned i = 0; i < nbr_cells; ++i ) buffer[0*nbr_cells+i] = 60.1;
for (unsigned i = 0; i < nbr_cells; ++i ) buffer[1*nbr_cells+i] = 60.2;
for (unsigned i = 0; i < nbr_cells; ++i ) buffer[2*nbr_cells+i] = 60.3;
for (unsigned i = 0; i < nbr_cells; ++i ) buffer[3*nbr_cells+i] = 60.4;
yac_cput ( field_id[5], 4, collection_data, &info, &ierror );
// check whether yac_ctest does not crash here
yac_ctest ( field_id[5], &flag );
}
//
// Receive fields from ocean
// -------------------------
{
double *collection_data[5];
for (int i = 0; i < 5; ++i) {
collection_data[i] = buffer + i * nbr_cells;
}
// SST
yac_cget ( field_id[6], 1, collection_data, &info, &ierror );
if (info > 0 && double_are_unequal(buffer[0], 110.1 ))
PUT_ERR("wrong atmosphere CPL SST\n");
// zonal velocity
yac_cget ( field_id[7], 1, collection_data, &info, &ierror );
if (info > 0 && double_are_unequal(buffer[0], 120.1 ))
PUT_ERR("wrong atmosphere CPL OCEANU\n");
// meridional velocity
yac_cget ( field_id[8], 1, collection_data, &info, &ierror );
if (info > 0 && double_are_unequal(buffer[0], 130.1 ))
PUT_ERR("wrong atmosphere CPL OCEANV\n");
// Ice thickness, concentration, T1 and T2
yac_cget_async ( field_id[9], 5, collection_data, &info, &ierror );
yac_cwait ( field_id[9] );
for (int i = 0; i < nbr_cells; ++i) {
if ( info > 0 ) {
if (double_are_unequal(buffer[0*nbr_cells + i], 140.1 ))
PUT_ERR ( "wrong atmosphere CPL ice 1\n");
if (double_are_unequal(buffer[1*nbr_cells + i], 140.2 ))
PUT_ERR ( "wrong atmosphere CPL ice 2\n");
if (double_are_unequal(buffer[2*nbr_cells + i], 140.3 ))
PUT_ERR ( "wrong atmosphere CPL ice 3\n");
if (double_are_unequal(buffer[3*nbr_cells + i], 140.4 ))
PUT_ERR ( "wrong atmosphere CPL ice 4\n");
if (double_are_unequal(buffer[4*nbr_cells + i], 140.5 ))
PUT_ERR ( "wrong atmosphere CPL ice 5\n");
}
}
// for target fields, yac_ctest should always return 1
yac_ctest ( field_id[9], &flag );
if (!flag) PUT_ERR("error in yac_ctest");
}
free (buffer);
free (field_id);
MPI_Comm_free ( &local_comm );
}
/* -------------------------------------------------------------------- */
static void dummy_ocean () {
int size, rank;
MPI_Comm local_comm;
char * comp_name = "dummy_ocean";
char * grid_name = "dummy_ocean_grid";
// Inform the coupler about what we are
yac_cdef_comp ( comp_name, &comp_id );
comp_ids[0] = comp_id;
yac_cget_comp_comm ( comp_id, &local_comm );
MPI_Comm_rank ( local_comm, &rank );
MPI_Comm_size ( local_comm, &size );
// printf (" %s rank %d : local size %d \n", comp_name, rank, size );
buffer_lon = xmalloc ( nbr_vertices * sizeof(*buffer_lon) );
buffer_lat = xmalloc ( nbr_vertices * sizeof(*buffer_lat) );
cell_to_vertex = xmalloc ( 3 * nbr_cells * sizeof(*cell_to_vertex) );
/* Define vertices
0
/ \
/ o \
/ \
1-------2 Eq.
\ /
\ o /
\ /
3
*/
buffer_lon[0] = 0.0; buffer_lat[0] = 1.0;
buffer_lon[1] = -1.0; buffer_lat[1] = 0.0;
buffer_lon[2] = 1.0; buffer_lat[2] = 0.0;
buffer_lon[3] = 0.0; buffer_lat[3] = -1.0;
// Connectivity
cell_to_vertex[0] = 0; cell_to_vertex[1] = 1; cell_to_vertex[2] = 2; // cell 1
cell_to_vertex[3] = 1; cell_to_vertex[4] = 3; cell_to_vertex[5] = 2; // cell 2
for (unsigned i = 0; i < nbr_cells; ++i ) nbr_vertices_per_cell[i] = 3;
yac_cdef_grid_unstruct(
grid_name, nbr_vertices, nbr_cells, nbr_vertices_per_cell,
buffer_lon, buffer_lat, cell_to_vertex, &grid_id);
// Decomposition information
for (int i = 0; i < nbr_cells; ++i ) {
global_index[i] = i;
cell_core_mask[i] = 1;
}
yac_cset_global_index(global_index, YAC_LOC_CELL, grid_id);
yac_cset_core_mask(cell_core_mask, YAC_LOC_CELL, grid_id);
// Center points in cells (needed e.g. for nearest neighbour)
buffer_lon[0] = 0.0; buffer_lat[0] = 0.5;
buffer_lon[1] = 0.0; buffer_lat[1] = -0.5;
yac_cdef_points_unstruct(
grid_id, nbr_cells, YAC_LOC_CELL, buffer_lon, buffer_lat, &cell_point_ids[0]);
free ( buffer_lon );
free ( buffer_lat );
// Mask generation
cell_mask = xmalloc (nbr_cells * sizeof(*cell_mask));
for (unsigned i = 0; i < nbr_cells; ++i) cell_mask[i] = 1;
yac_cdef_mask ( grid_id, nbr_cells, YAC_LOC_CELL, cell_mask, cell_mask_ids );
yac_cset_mask(cell_mask, cell_point_ids[0]);
free (cell_mask);
yac_cdef_calendar(YAC_PROLEPTIC_GREGORIAN);
field_id = xmalloc ( no_of_fields * sizeof(*field_id));
for (unsigned i = 0; i < no_of_fields/2; ++i )
yac_cdef_field ( field[i].name,
comp_id,
cell_point_ids,
1, field[i].collection_size,
"3600", YAC_TIME_UNIT_SECOND,
&field_id[i] );
for (unsigned i = no_of_fields/2; i < no_of_fields; ++i )
yac_cdef_field_mask ( field[i].name,
comp_id,
cell_point_ids,
cell_mask_ids,
1, field[i].collection_size,
"3600", YAC_TIME_UNIT_SECOND,
&field_id[i] );
// read couplings from YAML configuration file
yac_cread_config_yaml(yaml_filename);
yac_cenddef ( );
// Data exchange
buffer = xmalloc (5 * nbr_cells * sizeof (*buffer));
/* field_id[0] represents "TAUX" wind stress component
field_id[1] represents "TAUY" wind stress component
field_id[2] represents "SFWFLX" surface fresh water flux
field_id[3] represents "SFTEMP" surface temperature
field_id[4] represents "THFLX" total heat flux
field_id[5] represents "ICEATM" ice temperatures and melt potential
field_id[6] represents "SST" sea surface temperature
field_id[7] represents "OCEANU" u component of ocean surface current
field_id[8] represents "OCEANV" v component of ocean surface current
field_id[9]represents "ICEOCE" ice thickness, concentration and temperatures
*/
// Send fields from ocean to atmosphere
// ------------------------------------
{
double *point_set_data[5];
double **collection_data[5];
for (unsigned i = 0; i < 5; ++i) {
point_set_data[i] = buffer + i * nbr_cells;
collection_data[i] = &(point_set_data[i]);
}
// SST
for (unsigned i = 0; i < nbr_cells; ++i ) buffer[0*nbr_cells+i] = 110.1;
yac_cput ( field_id[6], 1, collection_data, &info, &ierror );
// zonal velocity
for (unsigned i = 0; i < nbr_cells; ++i ) buffer[0*nbr_cells+i] = 120.1;
yac_cput ( field_id[7], 1, collection_data, &info, &ierror );
// meridional velocity
for (unsigned i = 0; i < nbr_cells; ++i ) buffer[0*nbr_cells+i] = 130.1;
yac_cput ( field_id[8], 1, collection_data, &info, &ierror );
// Ice thickness, concentration, T1 and T2
for (unsigned i = 0; i < nbr_cells; ++i ) buffer[0*nbr_cells+i] = 140.1;
for (unsigned i = 0; i < nbr_cells; ++i ) buffer[1*nbr_cells+i] = 140.2;
for (unsigned i = 0; i < nbr_cells; ++i ) buffer[2*nbr_cells+i] = 140.3;
for (unsigned i = 0; i < nbr_cells; ++i ) buffer[3*nbr_cells+i] = 140.4;
for (unsigned i = 0; i < nbr_cells; ++i ) buffer[4*nbr_cells+i] = 140.5;
yac_cput ( field_id[9], 5, collection_data, &info, &ierror );
}
// Receive fields from atmosphere
// ------------------------------
{
double *collection_data[5];
for (unsigned i = 0; i < 5; ++i) {
collection_data[i] = buffer + i * nbr_cells;
}
// zonal wind stress
yac_cget ( field_id[0], 2, collection_data, &info, &ierror );
// meridional wind stress
yac_cget ( field_id[1], 2, collection_data, &info, &ierror );
// freshwater flux
yac_cget ( field_id[2], 3, collection_data, &info, &ierror );
// surface air temperature
yac_cget ( field_id[3], 1, collection_data, &info, &ierror );
// total heat flux - 4 parts - record 5
yac_cget ( field_id[4], 4, collection_data, &info, &ierror );
// ice parameter
yac_cget ( field_id[5], 4, collection_data, &info, &ierror );
}
free (buffer);
free (field_id);
MPI_Comm_free ( &local_comm );
}
/* -------------------------------------------------------------------- */
static void dummy_io () {
int size, rank;
MPI_Comm local_comm;
field_id = NULL;
char * comp_name = "dummy_io";
// Inform the coupler about what we are
yac_cdef_comp ( comp_name, &comp_id );
yac_cget_comp_comm ( comp_id, &local_comm );
MPI_Comm_rank ( local_comm, &rank );
MPI_Comm_size ( local_comm, &size );
// printf (" %s rank %d : local size %d \n", comp_name, rank, size );
// An empty search call to mark the end of the definition phase
yac_cenddef ( );
MPI_Comm_free ( &local_comm );
}
basic_grid.h
YAC_LOC_CELL
@ YAC_LOC_CELL
Definition location.h:14
yac_replay.collection_size
collection_size
Definition yac_replay.py:121
xmalloc
#define xmalloc(size)
Definition ppm_xfuncs.h:66
main
int main(int argc, char **argv)
Definition weights2vtk.c:147
yac_cenddef
void yac_cenddef(void)
Definition yac.c:2840
yac_cdef_field_mask
void yac_cdef_field_mask(char const *name, int const comp_id, int const *point_ids, int const *mask_ids, int const num_pointsets, int collection_size, const char *timestep, int time_unit, int *field_id)
Definition yac.c:1012
yac_cset_global_index
void yac_cset_global_index(int const *global_index, int location, int grid_id)
Definition yac.c:3244
yac_cget_role_from_field_id
int yac_cget_role_from_field_id(int field_id)
Definition yac.c:3088
yac_cget_timestep_from_field_id
const char * yac_cget_timestep_from_field_id(int field_id)
Definition yac.c:3074
yac_cinit
void yac_cinit(void)
Definition yac.c:402
yac_cfinalize
void yac_cfinalize()
Definition yac.c:533
yac_cget_comp_comm
void yac_cget_comp_comm(int comp_id, MPI_Comm *comp_comm)
Definition yac.c:638
yac_cget
void yac_cget(int const field_id, int collection_size, double **recv_field, int *info, int *ierr)
Definition yac.c:1850
yac_cget_collection_size_from_field_id
int yac_cget_collection_size_from_field_id(int field_id)
Definition yac.c:3081
YAC_TIME_UNIT_SECOND
int const YAC_TIME_UNIT_SECOND
Definition yac.c:52
yac_cdef_grid_unstruct
void yac_cdef_grid_unstruct(const char *grid_name, int nbr_vertices, int nbr_cells, int *num_vertices_per_cell, double *x_vertices, double *y_vertices, int *cell_to_vertex, int *grid_id)
Definition yac.c:3208
yac_cdef_points_unstruct
void yac_cdef_points_unstruct(int const grid_id, int const nbr_points, int const located, double const *x_points, double const *y_points, int *point_id)
Definition yac.c:919
yac_cput
void yac_cput(int const field_id, int const collection_size, double ***const send_field, int *info, int *ierr)
Definition yac.c:2514
yac_cget_async
void yac_cget_async(int const field_id, int collection_size, double **recv_field, int *info, int *ierr)
Definition yac.c:1859
yac_cread_config_yaml
void yac_cread_config_yaml(const char *yaml_filename)
Definition yac.c:457
yac_cdef_calendar
void yac_cdef_calendar(int calendar)
Definition yac.c:562
yac_cset_core_mask
void yac_cset_core_mask(int const *is_core, int location, int grid_id)
Definition yac.c:3285
yac_cwait
void yac_cwait(int field_id)
Definition yac.c:2090
YAC_PROLEPTIC_GREGORIAN
int const YAC_PROLEPTIC_GREGORIAN
Definition yac.c:61
yac_cset_mask
void yac_cset_mask(int const *is_valid, int points_id)
Definition yac.c:993
yac_cdef_mask
void yac_cdef_mask(int const grid_id, int const nbr_points, int const located, int const *is_valid, int *mask_id)
Definition yac.c:984
yac_ctest
void yac_ctest(int field_id, int *flag)
Definition yac.c:2066
yac_cdef_comp
void yac_cdef_comp(char const *comp_name, int *comp_id)
Definition yac.c:795
yac_cdef_field
void yac_cdef_field(char const *name, int const comp_id, int const *point_ids, int const num_pointsets, int collection_size, const char *timestep, int time_unit, int *field_id)
Definition yac.c:1085
Generated on Mon Apr 8 2024 08:41:00 for YetAnotherCoupler by doxygen 1.9.8