interp_operator_direct.c

Go to the documentation of this file.

// Copyright (c) 2024 The YAC Authors
//
// SPDX-License-Identifier: BSD-3-Clause
 
#include <string.h>
 
#include "interpolation/operators/interp_operator_direct.h"
#include "utils_core.h"
#include "interpolation/interpolation_utils.h"
#include "interpolation/interpolation_exchange.h"
#include "collection_selection_internal.h"
#include "interpolation/operators/interp_operator_internal.h"
 
static int yac_interp_operator_direct_is_source(
  struct yac_interp_operator * interp);
static int yac_interp_operator_direct_is_target(
  struct yac_interp_operator * interp);
static void yac_interp_operator_direct_execute(
  struct yac_interp_operator * interp,
  double *** src_fields, double *** src_frac_masks, double ** tgt_field,
  double frac_mask_fallback_value, double scale_factor, double scale_summand);
static void yac_interp_operator_direct_execute_put(
  struct yac_interp_operator * interp,
  double *** src_fields, double *** src_frac_masks,
  int is_target, double frac_mask_fallback_value,
  double scale_factor, double scale_summand);
static void yac_interp_operator_direct_execute_get(
  struct yac_interp_operator * interp, double ** tgt_field,
  double frac_mask_fallback_value, double scale_factor, double scale_summand);
static void yac_interp_operator_direct_execute_get_async(
  struct yac_interp_operator * interp, double ** tgt_field,
  double frac_mask_fallback_value, double scale_factor, double scale_summand);
static enum YAC_INTERP_TEST_STATUS yac_interp_operator_direct_execute_put_test(
  struct yac_interp_operator * interp);
static enum YAC_INTERP_TEST_STATUS yac_interp_operator_direct_execute_get_test(
  struct yac_interp_operator * interp);
static void yac_interp_operator_direct_execute_wait(
  struct yac_interp_operator * interp);
static struct yac_interp_operator * yac_interp_operator_direct_copy(
  struct yac_interp_operator * interp);
static void yac_interp_operator_direct_delete(
  struct yac_interp_operator * interp);
 
static struct yac_interp_operator_vtable const interpolation_direct_vtable = {
  .is_source         = yac_interp_operator_direct_is_source,
  .is_target         = yac_interp_operator_direct_is_target,
  .execute           = yac_interp_operator_direct_execute,
  .execute_put       = yac_interp_operator_direct_execute_put,
  .execute_get       = yac_interp_operator_direct_execute_get,
  .execute_get_async = yac_interp_operator_direct_execute_get_async,
  .execute_put_test  = yac_interp_operator_direct_execute_put_test,
  .execute_get_test  = yac_interp_operator_direct_execute_get_test,
  .execute_wait      = yac_interp_operator_direct_execute_wait,
  .copy              = yac_interp_operator_direct_copy,
  .delete            = yac_interp_operator_direct_delete,
};
 
struct yac_interpolation_direct {
 
  struct yac_interp_operator_vtable const * vtable;
 
  struct yac_collection_selection * collection_selection;
 
  struct yac_interpolation_buffer src_data;
  struct yac_interpolation_exchange * src2tgt;
 
  double ** src_field_buffer;
  int is_source;
  int is_target;
};
 
static struct yac_interp_operator * yac_interpolation_direct_new_(
  struct yac_collection_selection const * collection_selection,
  struct yac_interpolation_exchange * src2tgt,
  struct yac_interpolation_buffer src_data) {
 
  struct yac_interpolation_direct * direct = xmalloc(1 * sizeof(*direct));
 
  direct->vtable = &interpolation_direct_vtable;
  direct->collection_selection =
    yac_collection_selection_copy(collection_selection);
  direct->src_data = src_data;
  direct->src2tgt = src2tgt;
  direct->src_field_buffer =
    xcalloc(
      yac_collection_selection_get_collection_size(collection_selection),
      sizeof(*(direct->src_field_buffer)));
  direct->is_source = yac_interpolation_exchange_is_source(direct->src2tgt);
  direct->is_target = yac_interpolation_exchange_is_target(direct->src2tgt);
 
  return (struct yac_interp_operator *)direct;
}
 
struct yac_interp_operator * yac_interp_operator_direct_new(
  struct yac_collection_selection const * collection_selection,
  Xt_redist redist_) {
 
  size_t collection_size =
    yac_collection_selection_get_collection_size(collection_selection);
 
  return
    yac_interpolation_direct_new_(
      collection_selection,
      yac_interpolation_exchange_new(
        &redist_, 1, collection_size, 0, "source to target"),
      yac_interpolation_buffer_init(
        &redist_, 1, collection_size, SEND_BUFFER));
}
 
static int yac_interp_operator_direct_is_source(
  struct yac_interp_operator * interp) {
 
  struct yac_interpolation_direct * direct =
    (struct yac_interpolation_direct *)interp;
 
  return direct->is_source;
}
 
static int yac_interp_operator_direct_is_target(
  struct yac_interp_operator * interp) {
 
  struct yac_interpolation_direct * direct =
    (struct yac_interpolation_direct *)interp;
 
  return direct->is_target;
}
 
static void yac_interp_operator_direct_execute(
  struct yac_interp_operator * interp,
  double *** src_fields, double *** src_frac_masks, double ** tgt_field,
  double frac_mask_fallback_value, double scale_factor, double scale_summand) {
 
  struct yac_interpolation_direct * direct =
    (struct yac_interpolation_direct *)interp;
 
  double ** src_send_buffer = NULL;
 
  yac_interpolation_exchange_wait(
    direct->src2tgt, "yac_interp_operator_direct_execute");
 
  if (direct->is_source) {
 
    size_t collection_size =
      yac_collection_selection_get_collection_size(
        direct->collection_selection);
    size_t const * collection_indices =
      yac_collection_selection_get_indices(direct->collection_selection);
 
    if (collection_indices != NULL) {
 
      double *** temp_src_fields =
        xmalloc(collection_size * sizeof(*temp_src_fields));
      temp_src_fields[0] =
        xmalloc(collection_size * sizeof(**temp_src_fields));
      for (size_t i = 0; i < collection_size; ++i) {
        temp_src_fields[i] = temp_src_fields[0] + i;
        temp_src_fields[i][0] = src_fields[collection_indices[i]][0];
      }
      src_fields = temp_src_fields;
 
      if (src_frac_masks != NULL) {
        double *** temp_src_frac_masks =
          xmalloc(collection_size * sizeof(*temp_src_frac_masks));
        temp_src_frac_masks[0] =
          xmalloc(collection_size * sizeof(**temp_src_frac_masks));
        for (size_t i = 0; i < collection_size; ++i) {
          temp_src_frac_masks[i] = temp_src_frac_masks[0] + i;
          temp_src_frac_masks[i][0] = src_frac_masks[collection_indices[i]][0];
        }
        src_frac_masks = temp_src_frac_masks;
      }
    }
 
    if ((YAC_FRAC_MASK_VALUE_IS_VALID(frac_mask_fallback_value)) ||
        (scale_factor != 1.0) || (scale_summand != 0.0)) {
 
      src_send_buffer = direct->src_data.buffer;
 
      compute_tgt_field(
        (double const * restrict **)src_fields,
        (double const * restrict **)src_frac_masks, src_send_buffer,
        direct->src_data.buffer_sizes, 1, collection_size,
        frac_mask_fallback_value, scale_factor, scale_summand);
 
    } else {
 
      src_send_buffer = direct->src_field_buffer;
      for (size_t i = 0; i < collection_size; ++i)
        src_send_buffer[i] = src_fields[i][0];
    }
 
    if (collection_indices != NULL) {
      free(src_fields[0]);
      free(src_fields);
      if (src_frac_masks != NULL) {
        free(src_frac_masks[0]);
        free(src_frac_masks);
      }
    }
  }
 
  yac_interpolation_exchange_execute(
    direct->src2tgt, (double const **)src_send_buffer, tgt_field,
    "yac_interp_operator_direct_execute");
}
 
static void yac_interp_operator_direct_execute_put(
  struct yac_interp_operator * interp,
  double *** src_fields, double *** src_frac_masks, int is_target,
  double frac_mask_fallback_value, double scale_factor, double scale_summand) {
 
  UNUSED(is_target);
 
  struct yac_interpolation_direct * direct =
    (struct yac_interpolation_direct *)interp;
 
  double ** src_send_buffer = NULL;
 
  if (direct->is_source) {
 
    size_t collection_size =
      yac_collection_selection_get_collection_size(
        direct->collection_selection);
    size_t const * collection_indices =
      yac_collection_selection_get_indices(direct->collection_selection);
 
    if (collection_indices != NULL) {
 
      double *** temp_src_fields =
        xmalloc(collection_size * sizeof(*temp_src_fields));
      temp_src_fields[0] =
        xmalloc(collection_size * sizeof(**temp_src_fields));
      for (size_t i = 0; i < collection_size; ++i) {
        temp_src_fields[i] = temp_src_fields[0] + i;
        temp_src_fields[i][0] = src_fields[collection_indices[i]][0];
      }
      src_fields = temp_src_fields;
 
      if (src_frac_masks != NULL) {
        double *** temp_src_frac_masks =
          xmalloc(collection_size * sizeof(*temp_src_frac_masks));
        temp_src_frac_masks[0] =
          xmalloc(collection_size * sizeof(**temp_src_frac_masks));
        for (size_t i = 0; i < collection_size; ++i) {
          temp_src_frac_masks[i] = temp_src_frac_masks[0] + i;
          temp_src_frac_masks[i][0] = src_frac_masks[collection_indices[i]][0];
        }
        src_frac_masks = temp_src_frac_masks;
      }
    }
 
    // wait until previous exchange is completed
    if (yac_interpolation_exchange_status(
          direct->src2tgt, "yac_interp_operator_direct_execute_put") ==
        YAC_INTERP_EXCH_ACTIVE)
      yac_interpolation_exchange_wait(
        direct->src2tgt, "yac_interp_operator_direct_execute_put");
 
    src_send_buffer = direct->src_data.buffer;
 
    compute_tgt_field(
      (double const * restrict **)src_fields,
      (double const * restrict **)src_frac_masks, src_send_buffer,
      direct->src_data.buffer_sizes, 1,
      collection_size, frac_mask_fallback_value,
      scale_factor, scale_summand);
 
    if (collection_indices != NULL) {
      free(src_fields[0]);
      free(src_fields);
      if (src_frac_masks != NULL) {
        free(src_frac_masks[0]);
        free(src_frac_masks);
      }
    }
  }
 
  yac_interpolation_exchange_execute_put(
    direct->src2tgt, (double const **)src_send_buffer,
    "yac_interp_operator_direct_execute_put");
}
 
static void yac_interp_operator_direct_execute_get(
  struct yac_interp_operator * interp, double ** tgt_field,
  double frac_mask_fallback_value, double scale_factor, double scale_summand) {
 
  UNUSED(frac_mask_fallback_value);
  UNUSED(scale_factor);
  UNUSED(scale_summand);
 
  struct yac_interpolation_direct * direct =
    (struct yac_interpolation_direct*)interp;
 
  yac_interpolation_exchange_execute_get(
    direct->src2tgt, tgt_field, "yac_interp_operator_direct_execute_get");
}
 
static void yac_interp_operator_direct_execute_get_async(
  struct yac_interp_operator * interp, double ** tgt_field,
  double frac_mask_fallback_value, double scale_factor, double scale_summand) {
 
  UNUSED(frac_mask_fallback_value);
  UNUSED(scale_factor);
  UNUSED(scale_summand);
 
  struct yac_interpolation_direct * direct =
    (struct yac_interpolation_direct*)interp;
 
  // wait until previous exchange is completed
  if (yac_interpolation_exchange_status(
        direct->src2tgt, "yac_interp_operator_direct_execute_get_async") ==
      YAC_INTERP_EXCH_ACTIVE)
    yac_interpolation_exchange_wait(
      direct->src2tgt, "yac_interp_operator_direct_execute_get_async");
 
  yac_interpolation_exchange_execute_get_async(
    direct->src2tgt, tgt_field, "yac_interp_operator_direct_execute_get_async");
}
 
static struct yac_interp_operator * yac_interp_operator_direct_copy(
  struct yac_interp_operator * interp) {
 
  struct yac_interpolation_direct * direct =
    (struct yac_interpolation_direct*)interp;
 
  return
    yac_interpolation_direct_new_(
      direct->collection_selection,
      yac_interpolation_exchange_copy(direct->src2tgt),
      yac_interpolation_buffer_copy(
        direct->src_data, 1,
        yac_collection_selection_get_collection_size(
          direct->collection_selection)));
}
 
static enum YAC_INTERP_TEST_STATUS yac_interp_operator_direct_execute_put_test(
  struct yac_interp_operator * interp) {
 
  struct yac_interpolation_direct * direct =
    (struct yac_interpolation_direct*)interp;
 
  return
    (yac_interpolation_exchange_put_test(
       direct->src2tgt,
       "yac_interp_operator_direct_execute_put_test") ==
     YAC_INTERP_EXCH_COMPLETE)?YAC_INTERP_COMPLETE:YAC_INTERP_INCOMPLETE;
}
 
static enum YAC_INTERP_TEST_STATUS yac_interp_operator_direct_execute_get_test(
  struct yac_interp_operator * interp) {
 
  struct yac_interpolation_direct * direct =
    (struct yac_interpolation_direct*)interp;
 
  return
    (yac_interpolation_exchange_get_test(
       direct->src2tgt,
       "yac_interp_operator_direct_execute_get_test") ==
     YAC_INTERP_EXCH_COMPLETE)?YAC_INTERP_COMPLETE:YAC_INTERP_INCOMPLETE;
}
 
static void yac_interp_operator_direct_execute_wait(
  struct yac_interp_operator * interp) {
 
  struct yac_interpolation_direct * direct =
    (struct yac_interpolation_direct*)interp;
 
  yac_interpolation_exchange_wait(
    direct->src2tgt, "yac_interp_operator_direct_execute_wait");
}
 
static void yac_interp_operator_direct_delete(
  struct yac_interp_operator * interp) {
 
  if (interp == NULL) return;
 
  struct yac_interpolation_direct * direct =
    (struct yac_interpolation_direct*)interp;
 
  yac_interpolation_exchange_delete(
    direct->src2tgt, "yac_interp_operator_direct_delete");
  yac_interpolation_buffer_free(&(direct->src_data));
  free(direct->src_field_buffer);
  yac_collection_selection_delete(direct->collection_selection);
  free(direct);
}