ComponentItemInternalData.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
//-----------------------------------------------------------------------------
/*---------------------------------------------------------------------------*/
/* ComponentItemInternalData.h                                 (C) 2000-2024 */
/*                                                                           */
/* Gestion des listes de 'ComponentItemInternal'.                            */
/*---------------------------------------------------------------------------*/
#ifndef ARCANE_MATERIALS_INTERNAL_COMPONENTITEMINTERNALDATA_H
#define ARCANE_MATERIALS_INTERNAL_COMPONENTITEMINTERNALDATA_H
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
#include "arcane/utils/TraceAccessor.h"
#include "arcane/utils/Array.h"
#include "arcane/utils/NumArray.h"
 
#include "arcane/materials/MaterialsGlobal.h"
#include "arcane/core/materials/ComponentItemInternal.h"
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
namespace Arcane::Materials
{
class MeshMaterialMng;
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
class ComponentItemInternalRange
{
 public:
 
  class Sentinel
  {};
  class Iterator
  {
    friend ComponentItemInternalRange;
 
   private:
 
    Iterator(Int32 current_value, Int32 last_value)
    : m_current_value(current_value)
    , m_last_value(last_value)
    {}
 
   public:
 
    ConstituentItemIndex operator*() const { return ConstituentItemIndex(m_current_value); }
    void operator++() { ++m_current_value; }
    bool operator==(const Sentinel&) const
    {
      return m_current_value == m_last_value;
    }
    bool operator!=(const Sentinel&) const
    {
      return m_current_value != m_last_value;
    }
 
   private:
 
    Int32 m_current_value = 0;
    Int32 m_last_value = 0;
  };
 
 public:
 
  ARCCORE_HOST_DEVICE ConstituentItemIndex operator[](Int32 index) const
  {
    ARCANE_CHECK_AT(index, m_nb_value);
    return ConstituentItemIndex(m_first_index + index);
  }
 
 public:
 
  void setRange(Int32 first_index, Int32 nb_value)
  {
    m_first_index = first_index;
    m_nb_value = nb_value;
  }
  Iterator begin() const
  {
    return Iterator(m_first_index, m_first_index + m_nb_value);
  }
  Sentinel end() const
  {
    return {};
  }
  Int32 size() const { return m_nb_value; }
  ConstituentItemIndex firstValue() const { return ConstituentItemIndex(m_first_index); }
 
 private:
 
  Int32 m_first_index = -1;
  Int32 m_nb_value = 0;
};
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
class ComponentItemInternalData
: public TraceAccessor
{
 public:
 
  class Storage
  {
   public:
 
    explicit Storage(const MemoryAllocationOptions& alloc_info, const String& base_name);
 
   public:
 
    void resize(Int32 new_size, ComponentItemSharedInfo* shared_info, RunQueue& queue);
    Int32 size() const { return m_size; }
 
   private:
 
    Int32 m_size = 0;
    UniqueArray<ConstituentItemIndex> m_first_sub_constituent_item_id_list;
    UniqueArray<ConstituentItemIndex> m_super_component_item_local_id_list;
    UniqueArray<Int16> m_component_id_list;
    UniqueArray<Int16> m_nb_sub_constituent_item_list;
    UniqueArray<Int32> m_global_item_local_id_list;
    UniqueArray<MatVarIndex> m_var_index_list;
 
   private:
 
    static MemoryAllocationOptions _allocInfo(const MemoryAllocationOptions& alloc_info,
                                              const String& base_name, const String& name);
  };
 
 public:
 
  explicit ComponentItemInternalData(MeshMaterialMng* mm);
 
 public:
 
  void endCreate();
 
 public:
 
  void resizeComponentItemInternals(Int32 max_local_id, Int32 total_env_cell);
 
  ComponentItemSharedInfo* sharedInfo(Int16 level) { return &m_shared_infos[level]; }
  ComponentItemSharedInfo* allEnvSharedInfo() { return sharedInfo(LEVEL_ALLENVIRONMENT); }
  ComponentItemSharedInfo* envSharedInfo() { return sharedInfo(LEVEL_ENVIRONMENT); }
  ComponentItemSharedInfo* matSharedInfo() { return sharedInfo(LEVEL_MATERIAL); }
 
 private:
 
  MeshMaterialMng* m_material_mng = nullptr;
 
  UniqueArray<ComponentItemSharedInfo> m_shared_infos;
 
  ComponentItemInternalRange m_all_env_items_internal_range;
  ComponentItemInternalRange m_env_items_internal_range;
 
  Storage m_all_env_storage;
  Storage m_env_storage;
  Storage m_mat_storage;
 
 private:
 
  void _initSharedInfos();
  static MemoryAllocationOptions _allocOptions();
 
 public:
 
  void _resetItemsInternal();
};
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
} // namespace Arcane::Materials
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
#endif