MultiArray2View.h

// -*- tab-width: 2; indent-tabs-mode: nil; coding: utf-8-with-signature -*-
//-----------------------------------------------------------------------------
// Copyright 2000-2025 CEA (www.cea.fr) IFPEN (www.ifpenergiesnouvelles.com)
// See the top-level COPYRIGHT file for details.
// SPDX-License-Identifier: Apache-2.0
//-----------------------------------------------------------------------------
/*---------------------------------------------------------------------------*/
/* MultiArray2View.h                                           (C) 2000-2025 */
/*                                                                           */
/* Vue d'un tableau 2D à taille multiple.                                    */
/*---------------------------------------------------------------------------*/
#ifndef ARCANE_UTILS_MULTIARRAY2VIEW_H
#define ARCANE_UTILS_MULTIARRAY2VIEW_H
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
#include "arcane/utils/UtilsTypes.h"
#include "arcane/utils/ArrayView.h"
 
#include <type_traits>
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
namespace Arcane
{
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
template <class DataType>
class MultiArray2View
{
 public:

  MultiArray2View(ArrayView<DataType> buf, ConstArrayView<Int32> indexes, ConstArrayView<Int32> sizes)
  : m_buffer(buf)
  , m_indexes(indexes)
  , m_sizes(sizes)
  {}

  MultiArray2View() = default;
 
 public:

  Int32 dim1Size() const { return m_sizes.size(); }
  ARCANE_DEPRECATED_122 Int32 size() const { return dim1Size(); }
  ConstArrayView<Int32> dim2Sizes() const { return m_sizes; }
  Int32 totalNbElement() const { return m_buffer.size(); }
 
 public:

  ArrayView<DataType> operator[](Int32 i)
  {
    return ArrayView<DataType>(this->m_sizes[i], &this->m_buffer[this->m_indexes[i]]);
  }

  ConstArrayView<DataType> operator[](Int32 i) const
  {
    return ConstArrayView<DataType>(this->m_sizes[i], this->m_buffer.data() + (this->m_indexes[i]));
  }
 
 private:
 
  ArrayView<DataType> m_buffer;
  ConstArrayView<Int32> m_indexes;
  ConstArrayView<Int32> m_sizes;
};
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
template <class DataType>
class ConstMultiArray2View
{
 private:
 
  friend class MultiArray2<DataType>;
 
 public:

  ConstMultiArray2View(ConstArrayView<DataType> buf, ConstArrayView<Int32> indexes,
                       ConstArrayView<Int32> sizes)
  : m_buffer(buf)
  , m_indexes(indexes)
  , m_sizes(sizes)
  {}

  ConstMultiArray2View() = default;
 
 public:

  Int32 dim1Size() const { return m_sizes.size(); }
  ARCANE_DEPRECATED_122 Int32 size() const { return dim1Size(); }
  ConstArrayView<Int32> dim2Sizes() const { return m_sizes; }
  Int32 totalNbElement() const { return m_buffer.size(); }
 
 public:

  ConstArrayView<DataType> operator[](Int32 i) const
  {
    return ConstArrayView<DataType>(this->m_sizes[i], this->m_buffer.data() + (this->m_indexes[i]));
  }
 
 private:
 
  ConstArrayView<DataType> m_buffer;
  ConstArrayView<Int32> m_indexes;
  ConstArrayView<Int32> m_sizes;
};
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
template <class DataType>
class JaggedSmallSpan
{
 private:
 
  friend class MultiArray2<std::remove_cv_t<DataType>>;
 
 public:

  JaggedSmallSpan() = default;
 
 private:

  JaggedSmallSpan(SmallSpan<DataType> buf, SmallSpan<const Int32> indexes,
                       SmallSpan<const Int32> sizes)
  : m_buffer(buf)
  , m_indexes(indexes)
  , m_sizes(sizes)
  {}
 
 public:

  constexpr ARCCORE_HOST_DEVICE Int32 dim1Size() const { return m_sizes.size(); }
  constexpr ARCCORE_HOST_DEVICE SmallSpan<const Int32> dim2Sizes() const { return m_sizes; }
  constexpr ARCCORE_HOST_DEVICE Int32 totalNbElement() const { return m_buffer.size(); }
 
 public:

  constexpr ARCCORE_HOST_DEVICE SmallSpan<DataType> operator[](Int32 i) const
  {
    return m_buffer.subSpan(m_indexes[i], m_sizes[i]);
  }
 
 private:
 
  SmallSpan<DataType> m_buffer;
  SmallSpan<const Int32> m_indexes;
  SmallSpan<const Int32> m_sizes;
};
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
} // namespace Arcane
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
#endif