MeshPartialVariableArrayRef.h

// -*- 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
//-----------------------------------------------------------------------------
/*---------------------------------------------------------------------------*/
/* MeshPartialVariableArrayRef.h                               (C) 2000-2024 */
/*                                                                           */
/* Classe gérant une variable partielle array sur une entité du maillage.    */
/*---------------------------------------------------------------------------*/
#ifndef ARCANE_CORE_MESHPARTIALVARIABLEARRAYREF_H
#define ARCANE_CORE_MESHPARTIALVARIABLEARRAYREF_H
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
#include "arcane/utils/FatalErrorException.h"
 
#include "arcane/core/MeshVariableRef.h"
#include "arcane/core/PrivateVariableArray.h"
#include "arcane/core/ItemEnumerator.h"
#include "arcane/core/ItemGroupRangeIterator.h"
#include "arcane/core/ItemPairEnumerator.h"
#include "arcane/core/GroupIndexTable.h"
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
namespace Arcane
{
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
template<typename DataTypeT>
class ItemPartialVariableArrayRefT
: public PrivateVariableArrayT<DataTypeT>
{
 public:
  
  typedef DataTypeT DataType;
  typedef UniqueArray2<DataType> ValueType;
  typedef ConstArrayView<DataType> ConstReturnReferenceType;
  typedef ArrayView<DataType> ReturnReferenceType;
 
 protected:
 
  typedef PrivateVariableArrayT<DataType> BaseClass;
  typedef typename BaseClass::PrivatePartType PrivatePartType;
  
  typedef ArrayView<DataType> ArrayType;
  typedef ConstArrayView<DataType> ConstArrayType;
 
 public:
 
  ARCANE_CORE_EXPORT ItemPartialVariableArrayRefT(const VariableBuildInfo& vb,eItemKind ik);
  
  explicit ARCANE_CORE_EXPORT ItemPartialVariableArrayRefT(IVariable* var);
  
  ARCANE_CORE_EXPORT ItemPartialVariableArrayRefT(const ItemPartialVariableArrayRefT<DataType>& rhs);
  
 protected:
  
  ARCANE_CORE_EXPORT void operator=(const ItemPartialVariableArrayRefT<DataType>& rhs);
 
 public:
  
  ARCANE_CORE_EXPORT void fill(const DataType& value);
  ARCANE_CORE_EXPORT void copy(const ItemPartialVariableArrayRefT<DataType>& v);
  ARCANE_CORE_EXPORT void internalSetUsed(bool v);
 
 public:
 
  ConstArrayType operator[](const Item& i) const
  {
    ARCANE_CHECK_VALID_ITEM_AND_GROUP_KIND(i);
    ARCANE_ASSERT((m_table.isUsed()),("GroupIndexTable expired"));
    const GroupIndexTable& table = *m_table;
    return this->m_view[table[i.localId()]]; 
  }
  ArrayType operator[](const Item& i)
  { 
    ARCANE_CHECK_VALID_ITEM_AND_GROUP_KIND(i);
    ARCANE_ASSERT((m_table.isUsed()),("GroupIndexTable expired"));
    const GroupIndexTable& table = *m_table;
    return this->m_view[table[i.localId()]]; 
  }
  ConstArrayType operator[](const ItemGroupRangeIterator& i) const
  {
    ARCANE_CHECK_VALID_ITEM_AND_GROUP_KIND(i);
    return this->m_view[i.index()];
  }
  ArrayType operator[](const ItemGroupRangeIterator& i)
  {
    ARCANE_CHECK_VALID_ITEM_AND_GROUP_KIND(i);
    return this->m_view[i.index()];
  }
  ConstArrayType operator[](const ItemEnumerator& i) const
  {
    ARCANE_CHECK_ENUMERATOR(i,this->itemGroup());
    return this->m_view[i.index()];
  }
  ArrayType operator[](const ItemEnumerator& i)
  {
    ARCANE_CHECK_ENUMERATOR(i,this->itemGroup());
    return this->m_view[i.index()];
  }
  ConstArrayType operator[](ItemEnumeratorIndex i) const
  {
    return this->m_view[i.index()];
  }
  ArrayType operator[](ItemEnumeratorIndex i)
  {
    return this->m_view[i.index()];
  }
 
 public:
 
  GroupIndexTableView tableView() const { return m_table->view(); }
 
 protected:
  
  SharedPtrT<GroupIndexTable> m_table;
 
 protected:
 
  static VariableInfo _buildVariableInfo(const VariableBuildInfo& vbi,eItemKind ik);
  static VariableTypeInfo _buildVariableTypeInfo(eItemKind ik);
};
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
template<typename ItemTypeT,typename DataTypeT>
class MeshPartialVariableArrayRefT
: public ItemPartialVariableArrayRefT<DataTypeT>
{
 public:
  
  typedef DataTypeT DataType;
  typedef ItemTypeT ItemType;
  typedef UniqueArray2<DataType> ValueType;
  typedef ConstArrayView<DataType> ConstReturnReferenceType;
  typedef ArrayView<DataType> ReturnReferenceType;
 
 public:
 
  typedef ItemPartialVariableArrayRefT<DataType> BaseClass;
 
  typedef typename ItemTraitsT<ItemType>::ItemGroupType GroupType;
 
  typedef MeshPartialVariableArrayRefT<ItemType,DataType> ThatClass;
 
  typedef ArrayView<DataType> ArrayType;
  typedef ConstArrayView<DataType> ConstArrayType;
 
 public:
 
  ARCANE_CORE_EXPORT MeshPartialVariableArrayRefT(const VariableBuildInfo& vb);
  ARCANE_CORE_EXPORT MeshPartialVariableArrayRefT(const MeshPartialVariableArrayRefT<ItemType,DataType>& rhs);
  ARCANE_CORE_EXPORT void refersTo(const MeshPartialVariableArrayRefT<ItemType,DataType>& rhs);
  
 public:
   
  ConstArrayType operator[](const ItemType& i) const
  {
    ARCANE_ASSERT((this->m_table.isUsed()),("GroupIndexTable expired"));
    const GroupIndexTable& table = *this->m_table;
    return this->m_view[table[i.localId()]]; 
  }
  ArrayType operator[](const ItemType& i)
  {
    ARCANE_ASSERT((this->m_table.isUsed()),("GroupIndexTable expired"));
    const GroupIndexTable& table = *this->m_table;
    return this->m_view[table[i.localId()]]; 
  }
  ConstArrayType operator[](const ItemGroupRangeIteratorT<ItemType>& i) const
  {
    return this->m_view[i.index()]; 
  }
  ArrayType operator[](const ItemGroupRangeIteratorT<ItemType>& i)
  {
    return this->m_view[i.index()]; 
  }
  ConstArrayType operator[](const ItemPairEnumeratorSubT<ItemType>& i) const
  {
    return this->m_view[i.index()]; 
  }
  ArrayType operator[](const ItemPairEnumeratorSubT<ItemType>& i)
  {
    return this->m_view[i.index()]; 
  }
  ConstArrayType operator[](const ItemEnumeratorT<ItemType>& i) const
  {
    ARCANE_CHECK_ENUMERATOR(i,this->itemGroup());
    return this->m_view[i.index()]; 
  }
  ArrayType operator[](const ItemEnumeratorT<ItemType>& i)
  {
    ARCANE_CHECK_ENUMERATOR(i,this->itemGroup());
    return this->m_view[i.index()]; 
  }
  ConstArrayType operator[](ItemEnumeratorIndexT<ItemType> i) const
  {
    return this->m_view[i.index()];
  }
  ArrayType operator[](ItemEnumeratorIndexT<ItemType> i)
  {
    return this->m_view[i.index()];
  }
 
  ARCANE_CORE_EXPORT GroupType itemGroup() const;
 
 private:
 
  static VariableFactoryRegisterer m_auto_registerer;
  static VariableInfo _buildVariableInfo(const VariableBuildInfo& vbi);
  static VariableTypeInfo _buildVariableTypeInfo();
  static VariableRef* _autoCreate(const VariableBuildInfo& vb);
};
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
} // End namespace Arcane
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
#endif