YetAnotherCoupler 3.5.2
Loading...
Searching...
No Matches
test_dummy_coupling4_c.c

This test checks the various aggregation options.

// 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 <math.h>
#include <float.h>
#include "tests.h"
#include "test_common.h"
#include "yac.h"
enum {
NUM_FIELDS = 5,
COLLECTION_SIZE = 3,
NUM_POINTSETS = 1,
NUM_POINTS = 9
};
static void init_ref_recv_field(
double ref_recv_field[][COLLECTION_SIZE][NUM_POINTS]) {
for (int i = 0; i < 3; ++i)
for (int j = 0; j < COLLECTION_SIZE; ++j)
for (int k = 0; k < NUM_POINTS; ++k)
ref_recv_field[i][j][k] = 0.0;
for (int j = 0; j < COLLECTION_SIZE; ++j) {
for (int k = 0; k < NUM_POINTS; ++k) {
ref_recv_field[3][j][k] = DBL_MAX;
ref_recv_field[4][j][k] = -DBL_MAX;
}
}
}
int main(int argc, char** argv) {
int size, rank;
MPI_Comm_rank ( MPI_COMM_WORLD, &rank );
MPI_Comm_size ( MPI_COMM_WORLD, &size );
if (size != 2) {
fputs("wrong number of processes (has to be 2)\n", stderr);
exit(EXIT_FAILURE);
}
if (argc != 2) {
PUT_ERR("ERROR: missing config file directory");
xt_finalize();
MPI_Finalize();
return TEST_EXIT_CODE;
}
char * yaml_filename =
strcat(
strcpy(
malloc(strlen(argv[1]) + 32), argv[1]), "coupling_test4.yaml");
yac_cread_config_yaml(yaml_filename);
free(yaml_filename);
int is_target = rank == 1;
// define local component
int comp_id;
yac_cdef_comp((is_target)?"target_comp":"source_comp", &comp_id);
// define grid (both components use an identical grid
int grid_id;
(is_target)?"target_grid":"source_grid", (int[2]){3,3}, (int[2]){0,0},
(double[]){0,1,2}, (double[]){0,1,2}, &grid_id);
// define points at the vertices of the grid
int point_id;
grid_id, (int[2]){3,3}, YAC_LOCATION_CORNER,
(double[]){0,1,2}, (double[]){0,1,2}, &point_id);
// define fields
int field_ids[NUM_FIELDS];
const char * fieldName[NUM_FIELDS] =
{"time_op_accu_field",
"time_op_avg_field",
"time_op_none_field",
"time_op_min_field",
"time_op_max_field"};
for (int i = 0; i < NUM_FIELDS; ++i )
fieldName[i], comp_id, &point_id, 1, 3,
(is_target)?"2":"1", YAC_TIME_UNIT_SECOND,
&field_ids[i]);
double send_field[COLLECTION_SIZE][NUM_POINTSETS][NUM_POINTS] =
{{{ 1, 2, 3, 4, 5, 6, 7, 8, 9}},
{{10,11,12,13,14,15,16,17,18}},
{{19,20,21,22,23,24,25,26,27}}};
double ref_recv_field[NUM_FIELDS][COLLECTION_SIZE][NUM_POINTS];
init_ref_recv_field(ref_recv_field);
// do time steps
for (int t = 0; t < 32; ++t) {
if (!is_target) {
for (int i = 0; i < NUM_FIELDS; ++i) {
int info, ierror;
field_ids[i], COLLECTION_SIZE, &send_field[0][0][0], &info, &ierror);
int ref_send_info =
(t%4)?
if (info != ref_send_info) PUT_ERR("error in yac_cput_: wrong info");
if (ierror) PUT_ERR("error in yac_cput_: wrong ierror");
}
}
double scale = ((t == 0)?1.0:0.25);
for (int i = 0; i < COLLECTION_SIZE; ++i)
for (int j = 0; j < NUM_POINTS; ++j) {
// update ref_recv_field (accu)
ref_recv_field[0][i][j] += send_field[i][0][j];
// update ref_recv_field (avg)
ref_recv_field[1][i][j] += send_field[i][0][j] * scale;
// update ref_recv_field (none)
ref_recv_field[2][i][j] = send_field[i][0][j];
// update ref_recv_field (minimum)
ref_recv_field[3][i][j] =
MIN(ref_recv_field[3][i][j], send_field[i][0][j]);
// update ref_recv_field (maximum)
ref_recv_field[4][i][j] =
MAX(ref_recv_field[4][i][j], send_field[i][0][j]);
}
if (is_target) {
// target calls get every second timestep
if ((t % 2) == 0) {
for (int i = 0; i < NUM_FIELDS; ++i) {
// initialise recv_field
double recv_field[COLLECTION_SIZE][NUM_POINTS];
for (int j = 0; j < COLLECTION_SIZE; ++j)
for (int k = 0; k < NUM_POINTS; ++k)
recv_field[j][k] = -1;
int info, ierror;
field_ids[i], COLLECTION_SIZE, &recv_field[0][0], &info, &ierror);
int ref_recv_info = (t%4)?YAC_ACTION_NONE:YAC_ACTION_COUPLING;
if (info != ref_recv_info) PUT_ERR("error in yac_cget_: wrong info");
if (ierror) PUT_ERR("error in yac_cget_: wrong ierror");
if (info == YAC_ACTION_COUPLING) {
for (int j = 0; j < COLLECTION_SIZE; ++j)
for (int k = 0; k < NUM_POINTS; ++k)
if (fabs(recv_field[j][k] - ref_recv_field[i][j][k]) > 1e-6)
PUT_ERR("error in yac_cget_: wrong recv_field");
} else {
for (int j = 0; j < COLLECTION_SIZE; ++j)
for (int k = 0; k < NUM_POINTS; ++k)
if (recv_field[j][k] != -1)
PUT_ERR("error in yac_cget_: wrong recv_field");
}
}
}
}
// update send_field
for (int i = 0; i < 3; ++i)
for (int j = 0; j < NUM_POINTS; ++j)
send_field[i][0][j] += i - 1;
// clean ref_recv_field at every coupling timestep
if ((t % 4) == 0) init_ref_recv_field(ref_recv_field);
}
return TEST_EXIT_CODE;
}
int main(int argc, char **argv)
#define MIN(a, b)
#define MAX(a, b)
void yac_cenddef(void)
Definition yac.c:3000
void yac_cget_(int const field_id, int const collection_size, double *recv_field, int *info, int *ierr)
Definition yac.c:1989
void yac_cinit(void)
Definition yac.c:419
void yac_cfinalize()
Definition yac.c:619
int const YAC_LOCATION_CORNER
Definition yac.c:28
void yac_cdef_grid_reg2d(const char *grid_name, int nbr_vertices[2], int cyclic[2], double *x_vertices, double *y_vertices, int *grid_id)
Definition yac.c:3416
void yac_cdef_points_reg2d(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:976
int const YAC_TIME_UNIT_SECOND
Definition yac.c:52
void yac_cread_config_yaml(const char *yaml_filename)
Definition yac.c:474
int const YAC_ACTION_NONE
no data exchanges
Definition yac.c:35
void yac_cdef_calendar(int calendar)
Definition yac.c:648
int const YAC_ACTION_REDUCTION
data reduction, but data exchange
Definition yac.c:36
void yac_cput_(int const field_id, int const collection_size, double *send_field, int *info, int *ierr)
Definition yac.c:2074
int const YAC_PROLEPTIC_GREGORIAN
Definition yac.c:61
int const YAC_ACTION_COUPLING
data exchange
Definition yac.c:37
void yac_cdef_comp(char const *comp_name, int *comp_id)
Definition yac.c:886
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:1216