MultiArray2View.h

// -*- tab-width: 2; indent-tabs-mode: nil; coding: utf-8-with-signature -*-
//-----------------------------------------------------------------------------
// Copyright 2000-2022 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-2015 */
/*                                                                           */
/* 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"
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
ARCANE_BEGIN_NAMESPACE
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
template<typename DataType> class MultiArray2;
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
template<class DataType>
class MultiArray2View
{
 public:
  MultiArray2View(ArrayView<DataType> buf,IntegerConstArrayView indexes,IntegerConstArrayView sizes)
  : m_buffer(buf), m_indexes(indexes), m_sizes(sizes) { }
  MultiArray2View() { }
 public:
  Integer dim1Size() const { return m_sizes.size(); }
  ARCANE_DEPRECATED_122 Integer size() const { return dim1Size(); }
  IntegerConstArrayView dim2Sizes() const { return m_sizes; }
  Integer totalNbElement() const { return m_buffer.size(); }
 public:
  ArrayView<DataType> operator[](Integer i)
    {
      return ArrayView<DataType>(this->m_sizes[i],&this->m_buffer[this->m_indexes[i]]);
    }
  ConstArrayView<DataType> operator[](Integer i) const
  {
    return ConstArrayView<DataType>(this->m_sizes[i],this->m_buffer.data()+ (this->m_indexes[i]));
  }
 private:
  ArrayView<DataType> m_buffer;
  IntegerConstArrayView m_indexes;
  IntegerConstArrayView m_sizes;
};
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
template<class DataType>
class ConstMultiArray2View
{
 private:
  friend class MultiArray2<DataType>;
 public:
  ConstMultiArray2View(ConstArrayView<DataType> buf,IntegerConstArrayView indexes,
                       IntegerConstArrayView sizes)
  : m_buffer(buf), m_indexes(indexes), m_sizes(sizes) { }
  ConstMultiArray2View() { }
 public:
  Integer dim1Size() const { return m_sizes.size(); }
  ARCANE_DEPRECATED_122 Integer size() const { return dim1Size(); }
  IntegerConstArrayView dim2Sizes() const { return m_sizes; }
  Integer totalNbElement() const { return m_buffer.size(); }
 public:
  ConstArrayView<DataType> operator[](Integer i) const
  {
    return ConstArrayView<DataType>(this->m_sizes[i],this->m_buffer.data()+ (this->m_indexes[i]));
  }
 private:
  ConstArrayView<DataType> m_buffer;
  IntegerConstArrayView m_indexes;
  IntegerConstArrayView m_sizes;
};
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
ARCANE_END_NAMESPACE
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
#endif