mls_interface.cpp

Go to the documentation of this file.

//
// This file is subject to the terms and conditions defined in
// file 'LICENSE', which is part of this source code package.
//
 
#include "interface.hpp"
 
#include <ArborX_Point.hpp>
#include <Kokkos_Core.hpp>
#include <Kokkos_Profiling_ScopedRegion.hpp>
#include <stdexcept>
#include <string>
 
#include "common/concepts.hpp"
#include "common/transfer.hxx"
// Extern-template declarations in interpolator.hxx suppress re-instantiation
// in this TU. ETI is provided by MLS_interface_eti_*.cpp.
#include "mls/interpolator.hxx"
 
// ---------------------------------------------------------------------------
// Internal helpers (anonymous namespace)
// ---------------------------------------------------------------------------
 
namespace {
 
using namespace PACMAN;
 
template <typename ExecSpace, PACMAN::int_t Dim>
void
RunMLSInterpolate(PACMAN::coordinates_t *pSourcePoints,
                  PACMAN::int_t nSourcePoints, PACMAN::fp_t *pSourceValues,
                  PACMAN::coordinates_t *pTargetPoints,
                  PACMAN::int_t nTargetPoints,
                  PACMAN::int_t nComponents,
                  PACMAN::fp_t *pTargetValues) {
  PACMAN::Transfer<ExecSpace, Dim> transfer(PACMAN::TransferMethods::MLS);
  PACMAN::SetupTransferClass(transfer, nSourcePoints, 0, 0, nTargetPoints,
                             nComponents,
                             pSourcePoints, pSourceValues, nullptr, nullptr,
                             nullptr, pTargetPoints);
 
  PACMAN::MLS::MLSInterpolator<ExecSpace, Dim> interpolator(transfer);
 
  PACMAN::CopyToRowMajorHost(transfer.targetValues, pTargetValues);
}
 
template <PACMAN::int_t Dim>
requires PACMAN::IsValidDim<Dim> void
DispatchMLS(unsigned char execSpace,
            PACMAN::coordinates_t *pSourcePoints, PACMAN::int_t nSourcePoints,
            PACMAN::fp_t *pSourceValues, PACMAN::coordinates_t *pTargetPoints,
            PACMAN::int_t nTargetPoints, PACMAN::int_t nComponents,
            PACMAN::fp_t *pTargetValues) {
        std::visit(
      [&](auto execSpaceObj) {
          RunMLSInterpolate<decltype(execSpaceObj), Dim>(
            pSourcePoints, nSourcePoints, pSourceValues, pTargetPoints,
            nTargetPoints, nComponents, pTargetValues);
      },
      PACMAN::MakeExecSpace(execSpace));
}
 
} // anonymous namespace
 
// ---------------------------------------------------------------------------
// Public API implementation
// ---------------------------------------------------------------------------
 
namespace PACMAN {
 
void
MLS_interpolate(int_t spaceDimension, unsigned char execSpace,
                coordinates_t *sourcePoints, int_t nSourcePoints,
                fp_t *sourceValues, coordinates_t *targetPoints,
                int_t nTargetPoints, fp_t *targetValues, int_t nComponents) {
  const std::string _region_name = "PACMAN::MLS_interpolate";
  const Kokkos::Profiling::ScopedRegion region(_region_name);
 
  // LCOV_EXCL_START
  if (sourcePoints == nullptr && nSourcePoints > 0)
    throw std::invalid_argument("sourcePoints is null but nSourcePoints > 0");
  if (targetPoints == nullptr && nTargetPoints > 0)
    throw std::invalid_argument("targetPoints is null but nTargetPoints > 0");
  if (sourceValues == nullptr && nSourcePoints > 0)
    throw std::invalid_argument("sourceValues is null but nSourcePoints > 0");
  if (targetValues == nullptr && nTargetPoints > 0)
    throw std::invalid_argument("targetValues is null but nTargetPoints > 0");
  if (nComponents < 1)
    throw std::invalid_argument("nComponents must be greater than zero");
  // LCOV_EXCL_STOP
 
  switch (spaceDimension) {
  case 1:
    return DispatchMLS<1>(execSpace, sourcePoints, nSourcePoints, sourceValues,
                          targetPoints, nTargetPoints, nComponents, targetValues);
  case 2:
    return DispatchMLS<2>(execSpace, sourcePoints, nSourcePoints, sourceValues,
                          targetPoints, nTargetPoints, nComponents, targetValues);
  case 3:
    return DispatchMLS<3>(execSpace, sourcePoints, nSourcePoints, sourceValues,
                          targetPoints, nTargetPoints, nComponents, targetValues);
  // LCOV_EXCL_START
  default:
    throw std::runtime_error(
        "The dimension of the points can only be: 1, 2 or 3.\n");
  // LCOV_EXCL_STOP
  }
}
 
} // namespace PACMAN