ConstituentModifierWorkInfo.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
//-----------------------------------------------------------------------------
/*---------------------------------------------------------------------------*/
/* ConstituentModifierWorkInfo.h                               (C) 2000-2024 */
/*                                                                           */
/* Working structure used during the modification of constituents.           */
/*---------------------------------------------------------------------------*/
#ifndef ARCANE_MATERIALS_INTERNAL_CONSTITUENTMODIFIERWORKINFO_H
#define ARCANE_MATERIALS_INTERNAL_CONSTITUENTMODIFIERWORKINFO_H
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
#include "arcane/utils/TraceAccessor.h"
#include "arcane/utils/DualUniqueArray.h"
 
#include "arcane/core/materials/internal/IMeshMaterialVariableInternal.h"
#include "arcane/materials/MaterialsGlobal.h"
#include "arcane/materials/internal/MeshMaterialVariableIndexer.h"
#include "arcane/materials/internal/ComponentItemListBuilder.h"
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
namespace Arcane::Materials
{
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/

class ARCANE_MATERIALS_EXPORT ConstituentModifierWorkInfo
{
 public:
 
  ConstituentModifierWorkInfo(const MemoryAllocationOptions& opts, eMemoryRessource mem);
 
 public:

  DualUniqueArray<Int32> pure_local_ids;
  DualUniqueArray<Int32> partial_indexes;
  DualUniqueArray<bool> m_is_materials_modified;
  DualUniqueArray<bool> m_is_environments_modified;
  bool is_verbose = false;

  UniqueArray<Int32> cells_changed_in_env;
  UniqueArray<Int32> cells_unchanged_in_env;

  DualUniqueArray<MatVarIndex> m_saved_matvar_indexes;
  DualUniqueArray<Int32> m_saved_local_ids;

  UniqueArray<Int16> m_cells_current_nb_material;
 
  // Filter indicating if a cell will be partial after addition.
  // This array is dimensioned by the number of cells added during the current transformation.
  NumArray<bool, MDDim1> m_cells_is_partial;
 
  ComponentItemListBuilder list_builder;

  UniqueArray<CopyBetweenDataInfo> m_host_variables_copy_data;

  NumArray<CopyBetweenDataInfo, MDDim1> m_variables_copy_data;
 
 public:

  void initialize(Int32 max_local_id, Int32 nb_material, Int32 nb_environment, RunQueue& queue);
 
 public:

  bool isTransformedCell(CellLocalId local_id) const
  {
    return m_cells_to_transform[local_id.localId()];
  }

  void setTransformedCell(CellLocalId local_id, bool v)
  {
    m_cells_to_transform[local_id.localId()] = v;
  }

  void resetTransformedCells(ConstArrayView<Int32> local_ids)
  {
    for (Int32 x : local_ids)
      m_cells_to_transform[x] = false;
  }

  bool isRemovedCell(Int32 local_id) const { return m_removed_local_ids_filter[local_id]; }

  void setRemovedCells(ConstArrayView<Int32> local_ids, bool value);

  void setCurrentOperation(MaterialModifierOperation* operation);

  bool isAdd() const { return m_is_add; }
 
  SmallSpan<const bool> transformedCells() const { return m_cells_to_transform.to1DSmallSpan(); }
  SmallSpan<bool> transformedCells() { return m_cells_to_transform.to1DSmallSpan(); }
  SmallSpan<const bool> removedCells() const { return m_removed_local_ids_filter.to1DSmallSpan(); }
  SmallSpan<bool> removedCells() { return m_removed_local_ids_filter.to1DSmallSpan(); }
 
 private:

  // This array is dimensioned by the number of cells.
  NumArray<bool, MDDim1> m_removed_local_ids_filter;

  // This array is dimensioned by the number of cells.
  NumArray<bool, MDDim1> m_cells_to_transform;
 
  bool m_is_add = false;
};
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
} // namespace Arcane::Materials
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
#endif