interp_method.c

Go to the documentation of this file.

// Copyright (c) 2024 The YAC Authors
//
// SPDX-License-Identifier: BSD-3-Clause
 
#ifdef HAVE_CONFIG_H
// Get the definition of the 'restrict' keyword.
#include "config.h"
#endif
 
#include "interpolation/methods/interp_method_internal.h"
#include "interpolation/methods/interp_method_check.h"
#include "interpolation/methods/interp_method_callback.h"
#include "instrument_internal.h"
#include "yac_mpi_common.h"
 
struct yac_interp_weights * yac_interp_method_do_search(
  struct interp_method ** method, struct yac_interp_grid * interp_grid) {
 
  YAC_INSTRUMENT_START(YAC_INSTR_INTERP_SEARCH);
 
  size_t num_src_fields = yac_interp_grid_get_num_src_fields(interp_grid);
  MPI_Comm interp_grid_comm = yac_interp_grid_get_MPI_Comm(interp_grid);
  enum yac_location src_field_locations[num_src_fields];
  for (size_t i = 0; i < num_src_fields; ++i)
    src_field_locations[i] =
      yac_interp_grid_get_src_field_location(interp_grid, i);
  struct yac_interp_weights * weights =
    yac_interp_weights_new(
      interp_grid_comm, yac_interp_grid_get_tgt_field_location(interp_grid),
      src_field_locations, num_src_fields);
 
  // (no early return here; the instrumented region has to be closed
  //  unconditionally below)
  if (*method != NULL) {
 
    size_t temp_count;
    size_t * tgt_points;
    yac_interp_grid_get_tgt_points(interp_grid, &tgt_points, &temp_count);
 
    size_t final_count = 0;
    int interpolation_complete = 0;
    while (*method != NULL) {
      final_count +=
        (*method)->vtable->do_search(
          *method, interp_grid, tgt_points + final_count,
          temp_count - final_count, weights, &interpolation_complete);
      yac_mpi_call(
        MPI_Allreduce(
          MPI_IN_PLACE, &interpolation_complete, 1, MPI_INT,
          MPI_MAX, interp_grid_comm), interp_grid_comm);
      ++method;
    }
 
    free(tgt_points);
  }
 
  YAC_INSTRUMENT_STOP(YAC_INSTR_INTERP_SEARCH);
 
  return weights;
}
 
void yac_interp_method_delete(struct interp_method ** method) {
 
  while (*method != NULL) {
    struct interp_method * curr_method = *method;
    curr_method->vtable->delete(curr_method);
    ++method;
  }
}
 
void yac_interp_method_cleanup() {
 
  yac_interp_method_callback_buf_free();
  yac_interp_method_check_buf_free();
}