NumArrayViews.h

Aller à la documentation de ce fichier.

// -*- tab-width: 2; indent-tabs-mode: nil; coding: utf-8-with-signature -*-
//-----------------------------------------------------------------------------
// Copyright 2000-2024 CEA (www.cea.fr) IFPEN (www.ifpenergiesnouvelles.com)
// See the top-level COPYRIGHT file for details.
// SPDX-License-Identifier: Apache-2.0
//-----------------------------------------------------------------------------
/*---------------------------------------------------------------------------*/
/* NumArrayViews.h                                             (C) 2000-2024 */
/*                                                                           */
/* Gestion des vues pour les 'NumArray' pour les accélérateurs.              */
/*---------------------------------------------------------------------------*/
#ifndef ARCANE_ACCELERATOR_NUMARRAYVIEWS_H
#define ARCANE_ACCELERATOR_NUMARRAYVIEWS_H
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
#include "arcane/utils/NumArray.h"
#include "arcane/accelerator/ViewsCommon.h"
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
namespace Arcane::Accelerator
{
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
template <typename DataType, typename Extents, typename LayoutPolicy>
class NumArrayViewSetter;
template <typename Accessor, typename Extents, typename LayoutPolicy>
class NumArrayView;
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
class ARCANE_ACCELERATOR_EXPORT NumArrayViewBase
{
 protected:
 
  // Pour l'instant n'utilise pas encore \a command
  // mais il ne faut pas le supprimer
  explicit NumArrayViewBase(const ViewBuildInfo&,Span<const std::byte> bytes);
};
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
template <typename Accessor, typename Extents, typename LayoutType>
class NumArrayView
: public NumArrayViewBase
{
 public:
 
  using DataType = typename Accessor::ValueType;
  using SpanType = MDSpan<DataType, Extents, LayoutType>;
  using AccessorReturnType = typename Accessor::AccessorReturnType;
 
 public:
 
  NumArrayView(const ViewBuildInfo& command, SpanType v)
  : NumArrayViewBase(command, Arccore::asBytes(v.to1DSpan()))
  , m_values(v)
  {}
 
  template <typename X = Extents, typename = std::enable_if_t<X::rank() == 1, void>>
  constexpr ARCCORE_HOST_DEVICE AccessorReturnType operator()(Int32 i) const
  {
    return Accessor::build(m_values.ptrAt(i));
  }
  template <typename X = Extents, typename = std::enable_if_t<X::rank() == 1, void>>
  constexpr ARCCORE_HOST_DEVICE AccessorReturnType operator()(ArrayIndex<1> idx) const
  {
    return Accessor::build(m_values.ptrAt(idx));
  }
  template <typename X = Extents, typename = std::enable_if_t<X::rank() == 1, void>>
  constexpr ARCCORE_HOST_DEVICE AccessorReturnType operator[](Int32 i) const
  {
    return Accessor::build(m_values.ptrAt(i));
  }
  template <typename X = Extents, typename = std::enable_if_t<X::rank() == 1, void>>
  constexpr ARCCORE_HOST_DEVICE AccessorReturnType operator[](ArrayIndex<1> idx) const
  {
    return Accessor::build(m_values.ptrAt(idx));
  }
 
  template <typename X = Extents, typename = std::enable_if_t<X::rank() == 2, void>>
  constexpr ARCCORE_HOST_DEVICE AccessorReturnType operator()(Int32 i, Int32 j) const
  {
    return Accessor::build(m_values.ptrAt(i, j));
  }
  template <typename X = Extents, typename = std::enable_if_t<X::rank() == 2, void>>
  constexpr ARCCORE_HOST_DEVICE AccessorReturnType operator()(ArrayIndex<2> idx) const
  {
    return Accessor::build(m_values.ptrAt(idx));
  }
 
  template <typename X = Extents, typename = std::enable_if_t<X::rank() == 3, void>>
  constexpr ARCCORE_HOST_DEVICE AccessorReturnType operator()(Int32 i, Int32 j, Int32 k) const
  {
    return Accessor::build(m_values.ptrAt(i, j, k));
  }
  template <typename X = Extents, typename = std::enable_if_t<X::rank() == 3, void>>
  constexpr ARCCORE_HOST_DEVICE AccessorReturnType operator()(ArrayIndex<3> idx) const
  {
    return Accessor::build(m_values.ptrAt(idx));
  }
 
  template <typename X = Extents, typename = std::enable_if_t<X::rank() == 4, void>>
  constexpr ARCCORE_HOST_DEVICE AccessorReturnType operator()(Int32 i, Int32 j, Int32 k, Int32 l) const
  {
    return Accessor::build(m_values.ptrAt(i, j, k, l));
  }
  template <typename X = Extents, typename = std::enable_if_t<X::rank() == 4, void>>
  constexpr ARCCORE_HOST_DEVICE AccessorReturnType operator()(ArrayIndex<4> idx) const
  {
    return Accessor::build(m_values.ptrAt(idx));
  }
 
  constexpr ARCCORE_HOST_DEVICE Span<DataType> to1DSpan() const
  {
    return m_values.to1DSpan();
  }
 
 private:
 
  SpanType m_values;
};
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
template <typename DataType, typename Extents, typename LayoutPolicy> auto
viewOut(const ViewBuildInfo& command, NumArray<DataType, Extents, LayoutPolicy>& var)
{
  using Accessor = DataViewSetter<DataType>;
  return NumArrayView<Accessor, Extents, LayoutPolicy>(command, var.mdspan());
}
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
template <typename DataType, typename Extents, typename LayoutPolicy> auto
viewInOut(const ViewBuildInfo& command, NumArray<DataType, Extents, LayoutPolicy>& v)
{
  using Accessor = DataViewGetterSetter<DataType>;
  return NumArrayView<Accessor, Extents, LayoutPolicy>(command, v.mdspan());
}
 
/*----------------------------------------------1-----------------------------*/
/*---------------------------------------------------------------------------*/
template <typename DataType, typename Extents, typename LayoutType> auto
viewIn(const ViewBuildInfo& command, const NumArray<DataType, Extents, LayoutType>& v)
{
  using Accessor = DataViewGetter<DataType>;
  return NumArrayView<Accessor, Extents, LayoutType>(command, v.constMDSpan());
}
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
template <typename DataType, typename Extents, typename LayoutType = DefaultLayout>
using NumArrayInView = NumArrayView<DataViewGetter<DataType>, Extents, LayoutType>;
 
template <typename DataType, typename Extents, typename LayoutType = DefaultLayout>
using NumArrayOutView = NumArrayView<DataViewSetter<DataType>, Extents, LayoutType>;
 
template <typename DataType, typename Extents, typename LayoutType = DefaultLayout>
using NumArrayInOutView = NumArrayView<DataViewGetterSetter<DataType>, Extents, LayoutType>;
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
} // End namespace Arcane::Accelerator
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
#endif