core/materials/MatItem.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
//-----------------------------------------------------------------------------
/*---------------------------------------------------------------------------*/
/* MatItem.h                                                   (C) 2000-2024 */
/*                                                                           */
/* Entités matériau et milieux.                                              */
/*---------------------------------------------------------------------------*/
#ifndef ARCANE_CORE_MATERIALS_MATITEM_H
#define ARCANE_CORE_MATERIALS_MATITEM_H
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
#include "arcane/core/Item.h"
 
#include "arcane/core/materials/ComponentItem.h"
#include "arcane/core/materials/ComponentItemInternal.h"
#include "arcane/core/materials/IMeshMaterial.h"
#include "arcane/core/materials/IMeshEnvironment.h"
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
namespace Arcane::Materials
{
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
class MatCell
: public ComponentCell
{
 public:
 
  ARCCORE_HOST_DEVICE MatCell(const matimpl::ConstituentItemBase& item_base)
  : ComponentCell(item_base)
  {
#ifdef ARCANE_CHECK
    _checkLevel(item_base,LEVEL_MATERIAL);
#endif
  }
 
  explicit ARCCORE_HOST_DEVICE MatCell(const ComponentCell& item)
  : MatCell(item.constituentItemBase())
  {
  }
 
  MatCell() = default;
 
 public:
 
  ARCCORE_HOST_DEVICE inline EnvCell envCell() const;
 
  IMeshMaterial* material() const { return _material(); }
 
  IUserMeshMaterial* userMaterial() const { return _material()->userMaterial(); }
 
  ARCCORE_HOST_DEVICE Int32 materialId() const { return componentId(); }
 
 private:
 
  IMeshMaterial* _material() const { return static_cast<IMeshMaterial*>(component()); }
};
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
class EnvCell
: public ComponentCell
{
 public:
 
  explicit ARCCORE_HOST_DEVICE EnvCell(const matimpl::ConstituentItemBase& item_base)
  : ComponentCell(item_base)
  {
#ifdef ARCANE_CHECK
    _checkLevel(item_base,LEVEL_ENVIRONMENT);
#endif
  }
  explicit ARCCORE_HOST_DEVICE EnvCell(const ComponentCell& item)
  : EnvCell(item.constituentItemBase())
  {
  }
  EnvCell() = default;
 
 public:
 
  // Nombre de matériaux du milieu présents dans la maille
  ARCCORE_HOST_DEVICE Int32 nbMaterial() const { return nbSubItem(); }
 
  ARCCORE_HOST_DEVICE inline AllEnvCell allEnvCell() const;
 
  ARCCORE_HOST_DEVICE inline MatCell cell(Integer i) const { return _subItemBase(i); }
 
  IMeshEnvironment* environment() const { return _environment(); }
 
  ARCCORE_HOST_DEVICE Int32 environmentId() const { return componentId(); }
 
  ARCCORE_HOST_DEVICE CellMatCellEnumerator subMatItems() const
  {
    return CellMatCellEnumerator(*this);
  }
 
 private:
 
  IMeshEnvironment* _environment() const { return static_cast<IMeshEnvironment*>(component()); }
};
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
class AllEnvCell
: public ComponentCell
{
 public:
 
  explicit ARCCORE_HOST_DEVICE AllEnvCell(const matimpl::ConstituentItemBase& item_base)
  : ComponentCell(item_base)
  {
#if defined(ARCANE_CHECK)
    _checkLevel(item_base,LEVEL_ALLENVIRONMENT);
#endif
  }
 
  explicit ARCCORE_HOST_DEVICE AllEnvCell(const ComponentCell& item)
  : AllEnvCell(item.constituentItemBase())
  {
  }
 
  AllEnvCell() = default;
 
 public:
 
  ARCCORE_HOST_DEVICE Int32 nbEnvironment() const { return nbSubItem(); }
 
  EnvCell cell(Int32 i) const { return EnvCell(_subItemBase(i)); }
 
  ARCCORE_HOST_DEVICE CellEnvCellEnumerator subEnvItems() const
  {
    return CellEnvCellEnumerator(*this);
  }
};
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
ARCCORE_HOST_DEVICE inline EnvCell MatCell::
envCell() const
{
  return EnvCell(_superItemBase());
}
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
ARCCORE_HOST_DEVICE inline AllEnvCell EnvCell::
allEnvCell() const
{
  return AllEnvCell(_superItemBase());
}
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
} // End namespace Arcane::Materials
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
#endif