core/materials/MeshEnvironmentVariableRef.h

Aller à la documentation de ce fichier.

// -*- 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
//-----------------------------------------------------------------------------
/*---------------------------------------------------------------------------*/
/* MeshEnvironmentVariableRef.h                                (C) 2000-2024 */
/*                                                                           */
/* Référence à une variable sur un milieu du maillage.                       */
/*---------------------------------------------------------------------------*/
#ifndef ARCANE_MATERIALS_MESHENVIRONMENTVARIABLEREF_H
#define ARCANE_MATERIALS_MESHENVIRONMENTVARIABLEREF_H
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
#include "arcane/core/materials/MeshMaterialVariableRef.h"
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
namespace Arcane::Materials
{
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
template <typename DataType_>
class CellEnvironmentVariableScalarRef
: public MeshMaterialVariableRef
{
 public:
 
  using DataType = DataType_;
  using PrivatePartType = IScalarMeshMaterialVariable<Cell, DataType>;
  using ThatClass = CellEnvironmentVariableScalarRef<DataType>;
  using ItemType = Cell;
  using GlobalVariableRefType = MeshVariableScalarRefT<ItemType, DataType>;
 
 public:
 
  explicit ARCANE_CORE_EXPORT CellEnvironmentVariableScalarRef(const VariableBuildInfo& vb);
  explicit ARCANE_CORE_EXPORT CellEnvironmentVariableScalarRef(const MaterialVariableBuildInfo& vb);
  ARCANE_CORE_EXPORT CellEnvironmentVariableScalarRef(const ThatClass& rhs);
 
 public:
 
  ARCANE_CORE_EXPORT ThatClass& operator=(const ThatClass& rhs) = delete;
  CellEnvironmentVariableScalarRef() = delete;
 
 public:
 
  ARCANE_CORE_EXPORT virtual void refersTo(const ThatClass& rhs);
 
  ARCANE_CORE_EXPORT void updateFromInternal() override;
 
 protected:
 
  DataType operator[](MatVarIndex mvi) const
  {
    return m_value[mvi.arrayIndex()][mvi.valueIndex()];
  }
  DataType& operator[](MatVarIndex mvi)
  {
    return m_value[mvi.arrayIndex()][mvi.valueIndex()];
  }
 
 public:
 
  DataType operator[](ComponentItemLocalId mc) const
  {
    return this->operator[](mc.localId());
  }
 
  DataType& operator[](ComponentItemLocalId mc)
  {
    return this->operator[](mc.localId());
  }
 
  DataType operator[](CellLocalId c) const
  {
    return m_value[0][c.localId()];
  }
 
  DataType& operator[](CellLocalId c)
  {
    return m_value[0][c.localId()];
  }
 
  ARCANE_CORE_EXPORT DataType envValue(AllEnvCell c,Int32 env_id) const;
 
 public:
  
  ARCANE_CORE_EXPORT void fill(const DataType& value);
  ARCANE_CORE_EXPORT void fillPartialValues(const DataType& value);
 
 public:
 
  ARCANE_CORE_EXPORT GlobalVariableRefType& globalVariable();
  ARCANE_CORE_EXPORT const GlobalVariableRefType& globalVariable() const;
 
 private:
 
  PrivatePartType* m_private_part = nullptr;
  ArrayView<DataType>* m_value = nullptr;
  ArrayView<ArrayView<DataType>> m_container_value;
 
 public:
 
  // TODO: Temporaire. a supprimer.
  ArrayView<DataType>* _internalValue() const { return m_value; }
 
 private:
 
  void _init();
  void _setContainerView();
};
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
template<typename DataType_>
class CellEnvironmentVariableArrayRef
: public MeshMaterialVariableRef
{
 public:
 
  using DataType = DataType_;
  using PrivatePartType = IArrayMeshMaterialVariable<Cell,DataType>;
  using ItemType = Cell;
  using GlobalVariableRefType = MeshVariableArrayRefT<ItemType, DataType>;
  using ThatClass = CellEnvironmentVariableArrayRef<DataType>;
 
 public:
 
  explicit ARCANE_CORE_EXPORT CellEnvironmentVariableArrayRef(const VariableBuildInfo& vb);
  explicit ARCANE_CORE_EXPORT CellEnvironmentVariableArrayRef(const MaterialVariableBuildInfo& vb);
  ARCANE_CORE_EXPORT CellEnvironmentVariableArrayRef(const ThatClass& rhs);
 
 public:
 
  ThatClass& operator=(const ThatClass& rhs) = delete;
  CellEnvironmentVariableArrayRef() = delete;
 
 public:
 
  ARCANE_CORE_EXPORT virtual void refersTo(const ThatClass& rhs);
 
  ARCANE_CORE_EXPORT void updateFromInternal() override;
 
 public:
 
  ARCANE_CORE_EXPORT GlobalVariableRefType& globalVariable();
  ARCANE_CORE_EXPORT const GlobalVariableRefType& globalVariable() const;
 
 public:
 
  ARCANE_CORE_EXPORT void resize(Integer dim2_size);
 
 
 protected:
 
  ConstArrayView<DataType> operator[](MatVarIndex mvi) const
  {
    return m_value[mvi.arrayIndex()][mvi.valueIndex()];
  }
  ArrayView<DataType> operator[](MatVarIndex mvi)
  {
    return m_value[mvi.arrayIndex()][mvi.valueIndex()];
  }
 
 public:
 
  ConstArrayView<DataType> operator[](ComponentItemLocalId mc) const
  {
    return this->operator[](mc.localId());
  }
 
  ArrayView<DataType> operator[](ComponentItemLocalId mc)
  {
    return this->operator[](mc.localId());
  }
 
  ConstArrayView<DataType> operator[](CellLocalId c) const
  {
    return m_value[0][c.localId()];
  }
 
  ArrayView<DataType> operator[](CellLocalId c)
  {
    return m_value[0][c.localId()];
  }
 
 private:
 
  PrivatePartType* m_private_part = nullptr;
  Array2View<DataType>* m_value = nullptr;
  ArrayView<Array2View<DataType>> m_container_value;
 
 private:
 
  void _init();
  void _setContainerView();
};
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
typedef CellEnvironmentVariableScalarRef<Byte> EnvironmentVariableCellByte;
typedef CellEnvironmentVariableScalarRef<Real> EnvironmentVariableCellReal;
typedef CellEnvironmentVariableScalarRef<Int16> EnvironmentVariableCellInt16;
typedef CellEnvironmentVariableScalarRef<Int32> EnvironmentVariableCellInt32;
typedef CellEnvironmentVariableScalarRef<Int64> EnvironmentVariableCellInt64;
typedef CellEnvironmentVariableScalarRef<Real2> EnvironmentVariableCellReal2;
typedef CellEnvironmentVariableScalarRef<Real3> EnvironmentVariableCellReal3;
typedef CellEnvironmentVariableScalarRef<Real2x2> EnvironmentVariableCellReal2x2;
typedef CellEnvironmentVariableScalarRef<Real3x3> EnvironmentVariableCellReal3x3;
 
#ifdef ARCANE_64BIT
typedef EnvironmentVariableCellInt64 EnvironmentVariableCellInteger;
#else
typedef EnvironmentVariableCellInt32 EnvironmentVariableCellInteger;
#endif
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
typedef CellEnvironmentVariableArrayRef<Byte> EnvironmentVariableCellArrayByte;
typedef CellEnvironmentVariableArrayRef<Real> EnvironmentVariableCellArrayReal;
typedef CellEnvironmentVariableArrayRef<Int16> EnvironmentVariableCellArrayInt16;
typedef CellEnvironmentVariableArrayRef<Int32> EnvironmentVariableCellArrayInt32;
typedef CellEnvironmentVariableArrayRef<Int64> EnvironmentVariableCellArrayInt64;
typedef CellEnvironmentVariableArrayRef<Real2> EnvironmentVariableCellArrayReal2;
typedef CellEnvironmentVariableArrayRef<Real3> EnvironmentVariableCellArrayReal3;
typedef CellEnvironmentVariableArrayRef<Real2x2> EnvironmentVariableCellArrayReal2x2;
typedef CellEnvironmentVariableArrayRef<Real3x3> EnvironmentVariableCellArrayReal3x3;
 
#ifdef ARCANE_64BIT
typedef EnvironmentVariableCellInt64 EnvironmentVariableCellArrayInteger;
#else
typedef EnvironmentVariableCellInt32 EnvironmentVariableCellArrayInteger;
#endif
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
} // End namespace Arcane::Materials
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
#endif