AnyItemArray2.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
//-----------------------------------------------------------------------------
/*---------------------------------------------------------------------------*/
/* AnyItemArray2.h                                             (C) 2000-2014 */
/*                                                                           */
/* Tableau 2D d'items de types quelconques.                                  */
/*---------------------------------------------------------------------------*/
 
#ifndef ARCANE_ANYITEM_ANYITEMARRAY2_H
#define ARCANE_ANYITEM_ANYITEMARRAY2_H
 
#include "arcane/IItemFamily.h"
#include "arcane/utils/Array2.h"
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
#include "arcane/anyitem/AnyItemGlobal.h"
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
ARCANE_BEGIN_NAMESPACE
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
ANYITEM_BEGIN_NAMESPACE
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/

template<typename DataType>
class Array2
{
public:
  
  Array2(const Group& group)
  : m_size(0)
  {
    for(Group::Enumerator e = group.enumerator(); e.hasNext(); ++e) {
      if(e.groupIndex() >= m_values.size())
        m_values.resize(e.groupIndex()+1);
    }
    for(Group::Enumerator e = group.enumerator(); e.hasNext(); ++e) {
      m_values[e.groupIndex()].resize(e.group().itemFamily()->maxLocalId(),m_size);
    }
  }
  
  inline void resize(Integer size)
  {
    m_size = size;
    for(Integer i = 0; i < m_values.size(); ++i) {
      m_values[i].resize(m_values[i].dim1Size(),m_size);
    }
  }

  void fill(const DataType& data) 
  {
    for(Integer i = 0; i < m_values.size(); ++i) {
      m_values[i].fill(data);
    }
  }
  
  template<typename T>
  inline ArrayView<DataType> operator[](const T& item) {
    return m_values[item.groupIndex()][item.varIndex()];
  }
  
  // Acceseur
  template<typename T>
  inline ConstArrayView<DataType> operator[](const T& item) const {
    return m_values[item.groupIndex()][item.varIndex()];
  }
  
  inline Integer size() const { return m_size; }
 
private:
  
  Integer m_size;

  Arcane::UniqueArray< Arcane::UniqueArray2<DataType> > m_values;
};
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
ANYITEM_END_NAMESPACE
ARCANE_END_NAMESPACE
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
#endif /* ARCANE_ANYITEM_ANYITEMARRAY2_H */