SpanViews.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
//-----------------------------------------------------------------------------
/*---------------------------------------------------------------------------*/
/* SpanViews.h                                                 (C) 2000-2024 */
/*                                                                           */
/* Gestion des vues pour les 'Span' pour les accélérateurs.                  */
/*---------------------------------------------------------------------------*/
#ifndef ARCANE_ACCELERATOR_SPANVIEW_H
#define ARCANE_ACCELERATOR_SPANVIEW_H
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
#include "arcane/utils/NumArray.h"
#include "arcane/accelerator/ViewsCommon.h"
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
namespace Arcane::Accelerator
{
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
template <typename DataType, typename Extents, typename LayoutPolicy>
class SpanViewSetter;
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
class SpanViewBase
{
 protected:
 
  // Pour l'instant n'utilise pas encore \a command
  // mais il ne faut pas le supprimer
  explicit SpanViewBase(const ViewBuildInfo&)
  {
  }
 
 private:
};
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
template <typename Accessor>
class SpanView
: public SpanViewBase
{
 public:
 
  using DataType = typename Accessor::ValueType;
  using AccessorReturnType = typename Accessor::AccessorReturnType;
  using SpanType = Span<DataType>;
  using size_type = typename SpanType::size_type;
 
 public:
 
  SpanView(const ViewBuildInfo& command, SpanType v)
  : SpanViewBase(command)
  , m_values(v)
  {}
 
  constexpr ARCCORE_HOST_DEVICE AccessorReturnType operator()(size_type i) const
  {
    return Accessor::build(m_values.ptrAt(i));
  }
 
  constexpr ARCCORE_HOST_DEVICE AccessorReturnType operator[](size_type i) const
  {
    return Accessor::build(m_values.ptrAt(i));
  }
  constexpr ARCCORE_HOST_DEVICE size_type size() const { return m_values.size(); }
 
 private:
 
  SpanType m_values;
};
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
template <typename Accessor>
class SmallSpanView
: public SpanViewBase
{
 public:
 
  using DataType = typename Accessor::ValueType;
  using AccessorReturnType = typename Accessor::AccessorReturnType;
  using SpanType = SmallSpan<DataType>;
  using size_type = typename SpanType::size_type;
 
 public:
 
  SmallSpanView(const ViewBuildInfo& command, SpanType v)
  : SpanViewBase(command)
  , m_values(v)
  {}
 
  constexpr ARCCORE_HOST_DEVICE AccessorReturnType operator()(size_type i) const
  {
    return Accessor::build(m_values.ptrAt(i));
  }
 
  constexpr ARCCORE_HOST_DEVICE AccessorReturnType operator[](size_type i) const
  {
    return Accessor::build(m_values.ptrAt(i));
  }
  constexpr ARCCORE_HOST_DEVICE size_type size() const { return m_values.size(); }
 
 private:
 
  SpanType m_values;
};
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
template <typename DataType> auto
viewOut(const ViewBuildInfo& command, Span<DataType> var)
{
  using Accessor = DataViewSetter<DataType>;
  return SpanView<Accessor>(command, var);
}
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
template <typename DataType> auto
viewOut(const ViewBuildInfo& command, Array<DataType>& var)
{
  return viewOut(command, var.span());
}
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
template <typename DataType> auto
viewOut(const ViewBuildInfo& command, SmallSpan<DataType> var)
{
  using Accessor = DataViewSetter<DataType>;
  return SmallSpanView<Accessor>(command, var);
}
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
template <typename DataType> auto
viewOut(const ViewBuildInfo& command, ArrayView<DataType> var)
{
  return viewOut(command, SmallSpan<DataType>(var));
}
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
template <typename DataType> auto
viewInOut(const ViewBuildInfo& command, Span<DataType> var)
{
  using Accessor = DataViewGetterSetter<DataType>;
  return SpanView<Accessor>(command, var);
}
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
template <typename DataType> auto
viewInOut(const ViewBuildInfo& command, Array<DataType>& var)
{
  return viewInOut(command, var.span());
}
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
template <typename DataType> auto
viewInOut(const ViewBuildInfo& command, SmallSpan<DataType> var)
{
  using Accessor = DataViewGetterSetter<DataType>;
  return SmallSpanView<Accessor>(command, var);
}
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
template <typename DataType> auto
viewInOut(const ViewBuildInfo& command, ArrayView<DataType> var)
{
  return viewInOut(command, SmallSpan<DataType>(var));
}
 
/*----------------------------------------------1-----------------------------*/
/*---------------------------------------------------------------------------*/
template <typename DataType> auto
viewIn(const ViewBuildInfo& command, Span<DataType> var)
{
  using Accessor = DataViewGetter<DataType>;
  return SpanView<Accessor>(command, var);
}
 
/*----------------------------------------------1-----------------------------*/
/*---------------------------------------------------------------------------*/
template <typename DataType> auto
viewIn(const ViewBuildInfo& command, const Array<DataType>& var)
{
  return viewIn(command, var.span());
}
 
/*----------------------------------------------1-----------------------------*/
/*---------------------------------------------------------------------------*/
template <typename DataType> auto
viewIn(const ViewBuildInfo& command, SmallSpan<DataType> var)
{
  using Accessor = DataViewGetter<DataType>;
  return SmallSpanView<Accessor>(command, var);
}
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
template <typename DataType> auto
viewIn(const ViewBuildInfo& command, ConstArrayView<DataType> var)
{
  return viewIn(command, SmallSpan<const DataType>(var));
}
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
} // End namespace Arcane::Accelerator
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
#endif