test_read_mpiom.c

Go to the documentation of this file.

// Copyright (c) 2024 The YAC Authors
//
// SPDX-License-Identifier: BSD-3-Clause
 
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
 
#include "tests.h"
#include "test_common.h"
#include "grid_file_common.h"
#include "read_mpiom_grid.h"
#include "grid2vtk.h"
#include "io_utils.h"
 
int main(int argc, char** argv) {
 
  if (argc != 2) {
    PUT_ERR("ERROR: missing grid file directory");
    return TEST_EXIT_CODE;
  }
 
  char * grid_filename =
    strcat(
      strcpy(
        malloc(strlen(argv[1]) + 32), argv[1]), "GR30_lsm.nc");
 
  if (!yac_file_exists(grid_filename)) return EXIT_SKIP_TEST;
 
  {
    struct yac_basic_grid_data mpiom_grid =
      yac_read_mpiom_basic_grid_data(grid_filename);
 
    if (mpiom_grid.num_cells != 12120)
      PUT_ERR("wrong number of grid cells");
 
// #define WRITE_VTK_GRID_FILE
#ifdef WRITE_VTK_GRID_FILE
    yac_write_basic_grid_data_to_file(&mpiom_grid, "mpiom");
#endif // WRITE_VTK_GRID_FILE
 
    yac_basic_grid_data_free(mpiom_grid);
  }
 
  {
    struct yac_basic_grid * mpiom_grid =
      yac_read_mpiom_basic_grid(grid_filename, "mpiom");
 
 
    if (yac_basic_grid_get_data_size(mpiom_grid, YAC_LOC_CELL) != 12120)
      PUT_ERR("wrong number of grid cells");
 
    yac_basic_grid_delete(mpiom_grid);
  }
 
  free(grid_filename);
 
  { // grids whose longitudes are outside of [0;360], which have to be
    // shifted into that range before duplicated coordinates can be
    // detected; the first one additionally uses a four dimensional mask
    // variable
 
    char * dummy_grid_filename = "test_read_mpiom_dummy.nc";
    enum {NUM_X = 20, NUM_Y = 10};
 
    struct {
      double lon_range[2];
      int mask_ndims;
    } configs[] = {{.lon_range = {-360.0, 0.0}, .mask_ndims = 4},
                   {.lon_range = {360.0, 720.0}, .mask_ndims = 3}};
    enum {NUM_CONFIGS = sizeof(configs) / sizeof(configs[0])};
 
    for (int i = 0; i < NUM_CONFIGS; ++i) {
 
      write_dummy_mpiom_grid_file(
        dummy_grid_filename, NUM_X, NUM_Y,
        configs[i].lon_range, (double[]){-80.0, 80.0}, configs[i].mask_ndims);
 
      struct yac_basic_grid_data mpiom_grid =
        yac_read_mpiom_basic_grid_data(dummy_grid_filename);
 
      if (mpiom_grid.num_cells != NUM_X * NUM_Y)
        PUT_ERR("wrong number of grid cells");
 
      yac_basic_grid_data_free(mpiom_grid);
 
      unlink(dummy_grid_filename);
    }
  }
 
  return TEST_EXIT_CODE;
}