AllCellToAllEnvCellConverter.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
//-----------------------------------------------------------------------------
/*---------------------------------------------------------------------------*/
/* AllCellToAllEnvCellConverter.h                              (C) 2000-2024 */
/*                                                                           */
/* Conversion de 'Cell' en 'AllEnvCell'.                                     */
/*---------------------------------------------------------------------------*/
#ifndef ARCANE_MATERIALS_ALLCELLTOALLENVCELLCONVERTER_H
#define ARCANE_MATERIALS_ALLCELLTOALLENVCELLCONVERTER_H
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
#include "arcane/materials/MaterialsGlobal.h"
 
#include "arcane/utils/PlatformUtils.h"
#include "arcane/utils/NumArray.h"
 
#include "arcane/core/IMesh.h"
#include "arcane/core/materials/MatItem.h"
#include "arcane/core/materials/IMeshMaterialMng.h"
#include "arcane/core/materials/MatItemEnumerator.h"
#include "arcane/core/materials/CellToAllEnvCellConverter.h"
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
namespace ArcaneTest
{
class MeshMaterialAcceleratorUnitTest;
}
 
namespace Arcane::Materials
{
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
class ARCANE_MATERIALS_EXPORT AllCellToAllEnvCell
{
  friend class CellToAllEnvCellAccessor;
  friend class CellToAllComponentCellEnumerator;
  friend class MeshMaterialMng;
  friend class AllEnvData;
  friend ArcaneTest::MeshMaterialAcceleratorUnitTest;
 
 public:
 
  class Impl;
 
 private:
 
  explicit AllCellToAllEnvCell(IMeshMaterialMng* mm);
 
 public:
 
  AllCellToAllEnvCell& operator=(const AllCellToAllEnvCell&) = delete;
 
 private:
 
  void initialize();
 
  ARCCORE_HOST_DEVICE Span<ComponentItemLocalId>* internal() const
  {
    return m_allcell_allenvcell_ptr;
  }
 
  Int32 maxNbEnvPerCell() const;
 
  /*
   * On regarde si le nb max d'env par cell à l'instant t a changé,
   * et si c'est le cas, on force la reconstruction de la table.
   * Est appelé par le forceRecompute du ImeshMaterialMng
   */
  void bruteForceUpdate();
 
 private:
 
  void reset();
 
 private:
 
  IMeshMaterialMng* m_material_mng = nullptr;
  Integer m_size = 0;
  NumArray<Span<ComponentItemLocalId>, MDDim1> m_allcell_allenvcell;
  Span<ComponentItemLocalId>* m_allcell_allenvcell_ptr = nullptr;
  NumArray<ComponentItemLocalId, MDDim1> m_mem_pool;
  Int32 m_current_max_nb_env = 0;
};
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
class ARCANE_MATERIALS_EXPORT CellToAllEnvCellAccessor
{
 public:
 
  using size_type = Span<ComponentItemLocalId>::size_type;
 
 public:
 
  CellToAllEnvCellAccessor() = default;
  explicit CellToAllEnvCellAccessor(const IMeshMaterialMng* mm);
 
  ARCCORE_HOST_DEVICE const AllCellToAllEnvCell* getAllCellToAllEnvCell() const
  {
    return m_cell_allenvcell;
  }
 
  ARCCORE_HOST_DEVICE size_type nbEnvironment(Integer cid) const
  {
    return m_cell_allenvcell->internal()[cid].size();
  }
 
 private:
 
  AllCellToAllEnvCell* m_cell_allenvcell = nullptr;
};
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
class ARCANE_MATERIALS_EXPORT CellToAllComponentCellEnumerator
{
  friend class EnumeratorTracer;
 
 public:
 
  using index_type = Span<ComponentItemLocalId>::index_type;
  using size_type = Span<ComponentItemLocalId>::size_type;
 
 public:
 
  // La version CPU permet de vérifier qu'on a bien fait l'init avant l'ENUMERATE
  ARCCORE_HOST_DEVICE explicit CellToAllComponentCellEnumerator(Integer cell_id, const CellToAllEnvCellAccessor& acc)
  : m_cid(cell_id)
  {
#if defined(ARCCORE_DEVICE_CODE)
    m_ptr = &(acc.getAllCellToAllEnvCell()->internal()[cell_id]);
    m_size = m_ptr->size();
#else
    if (acc.getAllCellToAllEnvCell()) {
      m_ptr = &(acc.getAllCellToAllEnvCell()->internal()[cell_id]);
      m_size = m_ptr->size();
    }
    else
      ARCANE_FATAL("Must create AllCellToAllEnvCell before using ENUMERATE_ALLENVCELL");
#endif
  }
  ARCCORE_HOST_DEVICE void operator++()
  {
    ++m_index;
  }
 
  ARCCORE_HOST_DEVICE bool hasNext() const
  {
    return m_index < m_size;
  }
 
  ARCCORE_HOST_DEVICE ComponentItemLocalId operator*() const
  {
    return (*m_ptr)[m_index];
  }
 
 private:
 
  Integer m_cid = 0;
  index_type m_index = 0;
  Span<ComponentItemLocalId>* m_ptr = nullptr;
  size_type m_size = 0;
};
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
#define RUNCOMMAND_ENUMERATE_CELL_ALLENVCELL(cell_to_allenvcellaccessor, iter_name, cell_group) \
  A_FUNCINFO << cell_group << [=] ARCCORE_HOST_DEVICE(CellLocalId iter_name)
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
// TODO: Très certainement à déplacer ailleurs si on garde ce proto
#define A_ENUMERATE_CELL_ALLCOMPONENTCELL(_EnumeratorClassName, iname, cid, cell_to_allenvcellaccessor) \
  for (A_TRACE_COMPONENT(_EnumeratorClassName) iname(::Arcane::Materials::_EnumeratorClassName(cid, cell_to_allenvcellaccessor) A_TRACE_ENUMERATOR_WHERE); iname.hasNext(); ++iname)
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
// TODO: Très certainement à déplacer ailleurs si on garde ce proto
#define ENUMERATE_CELL_ALLENVCELL(iname, cid, cell_to_allenvcellaccessor) \
  A_ENUMERATE_CELL_ALLCOMPONENTCELL(CellToAllComponentCellEnumerator, iname, cid, cell_to_allenvcellaccessor)
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
} // End namespace Arcane::Materials
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
#endif