MeshVariableArrayRef.h

// -*- tab-width: 2; indent-tabs-mode: nil; coding: utf-8-with-signature -*-
//-----------------------------------------------------------------------------
// Copyright 2000-2026 CEA (www.cea.fr) IFPEN (www.ifpenergiesnouvelles.com)
// See the top-level COPYRIGHT file for details.
// SPDX-License-Identifier: Apache-2.0
//-----------------------------------------------------------------------------
/*---------------------------------------------------------------------------*/
/* MeshVariableArrayRef.h                                      (C) 2000-2024 */
/*                                                                           */
/* Class managing a vector variable on a mesh entity.                        */
/*---------------------------------------------------------------------------*/
#ifndef ARCANE_CORE_MESHVARIABLEARRAYREF_H
#define ARCANE_CORE_MESHVARIABLEARRAYREF_H
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
#include "arcane/core/PrivateVariableArray.h"
#include "arcane/core/ItemEnumerator.h"
#include "arcane/core/ItemGroupRangeIterator.h"
#include "arcane/core/ItemPairEnumerator.h"
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
namespace Arcane
{
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/

template <class DataTypeT>
class ItemVariableArrayRefT
: 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 ItemVariableArrayRefT(const VariableBuildInfo& b, eItemKind ik);

  explicit ARCANE_CORE_EXPORT ItemVariableArrayRefT(IVariable* var);

  ARCANE_CORE_EXPORT ItemVariableArrayRefT(const ItemVariableArrayRefT<DataType>& rhs);
 
 protected:

  ARCANE_CORE_EXPORT ItemVariableArrayRefT<DataType>& operator=(const ItemVariableArrayRefT<DataType>& rhs);
 
 public:

  ConstArrayType operator[](ItemLocalId item) const { return this->m_view[item.localId()]; }

  ArrayType operator[](ItemLocalId item) { return this->m_view[item.localId()]; }

  const DataType operator()(ItemLocalId item, Int32 i) const { return this->m_view.item(item.localId(), i); }

  DataType& operator()(ItemLocalId item, Int32 i) { return this->m_view[item.localId()][i]; }
 
 public:

  ARCANE_CORE_EXPORT void copy(const ItemVariableArrayRefT<DataType>& v);
  ARCANE_CORE_EXPORT void copy(const ItemVariableArrayRefT<DataType>& v, const ItemGroup& group);
  ARCANE_CORE_EXPORT void copy(const ItemVariableArrayRefT<DataType>& v, RunQueue* queue);
  ARCANE_CORE_EXPORT void fill(const DataType& value);
  ARCANE_CORE_EXPORT void fill(const DataType& value, const ItemGroup& group);
  ARCANE_CORE_EXPORT void fill(const DataType& value, RunQueue* queue);
 
 public:
 
  static ARCANE_CORE_EXPORT VariableInfo _internalVariableInfo(const VariableBuildInfo& vbi, eItemKind ik);
  static ARCANE_CORE_EXPORT VariableTypeInfo _internalVariableTypeInfo(eItemKind ik);
};
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/

template <class ItemTypeT, class DataTypeT>
class MeshVariableArrayRefT
: public ItemVariableArrayRefT<DataTypeT>
{
 public:
 
  typedef DataTypeT DataType;
  typedef ItemTypeT ItemType;
  typedef UniqueArray2<DataType> ValueType;
  typedef ConstArrayView<DataType> ConstReturnReferenceType;
  typedef ArrayView<DataType> ReturnReferenceType;
 
 protected:
 
  typedef MeshVariableRef BaseClass;
 
  typedef Array2View<DataType> ArrayBase;
 
  typedef ArrayView<DataType> ArrayType;
  typedef ConstArrayView<DataType> ConstArrayType;
 
  typedef Array2VariableT<DataType> PrivatePartType;
 
  typedef typename ItemTraitsT<ItemType>::ItemGroupType GroupType;
  typedef typename ItemTypeT::LocalIdType ItemLocalIdType;
 
  typedef MeshVariableArrayRefT<ItemType, DataType> ThatClass;
 
 public:

  ARCANE_CORE_EXPORT MeshVariableArrayRefT(const VariableBuildInfo& b);
  explicit ARCANE_CORE_EXPORT MeshVariableArrayRefT(IVariable* var);
  ARCANE_CORE_EXPORT MeshVariableArrayRefT(const MeshVariableArrayRefT<ItemType, DataType>& rhs);
  ARCANE_CORE_EXPORT void refersTo(const MeshVariableArrayRefT<ItemType, DataType>& rhs);
 
  ThatClass& operator=(const ThatClass& rhs) = delete; // Forbidden.
 
 public:

  ConstArrayType operator[](ItemLocalIdType item) const { return this->m_view[item.localId()]; }

  ArrayType operator[](ItemLocalIdType item) { return this->m_view[item.localId()]; }

  const DataType operator()(ItemLocalIdType item, Int32 i) const { return this->m_view[item.localId()][i]; }

  DataType& operator()(ItemLocalIdType item, Int32 i) { return this->m_view[item.localId()][i]; }
 
 public:

  ARCANE_CORE_EXPORT GroupType itemGroup() const;
  ARCANE_CORE_EXPORT void swapValues(MeshVariableArrayRefT<ItemType, DataType>& rhs);
 
 public:
 
  static ARCANE_CORE_EXPORT VariableTypeInfo _internalVariableTypeInfo();
  static ARCANE_CORE_EXPORT VariableInfo _internalVariableInfo(const VariableBuildInfo& vbi);
 
 private:
 
  static VariableFactoryRegisterer m_auto_registerer;
  static VariableRef* _autoCreate(const VariableBuildInfo& vb);
};
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
} // End namespace Arcane
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
#endif