12#ifndef ARCANE_UTILS_NUMARRAY_H
13#define ARCANE_UTILS_NUMARRAY_H
17#include "arcane/utils/MDSpan.h"
18#include "arcane/utils/MDDim.h"
19#include "arcane/utils/NumArrayContainer.h"
20#include "arcane/utils/ArcaneCxx20.h"
21#include "arcane/utils/String.h"
55template <
typename DataType,
typename Extents,
typename LayoutPolicy>
61#if !defined(ARCANE_NO_CONCEPT_FOR_NUMARRAY)
67 using ExtentsType = Extents;
69 using DynamicDimsType =
typename ExtentsType::DynamicDimsType;
73 using ArrayBoundsIndexType =
typename MDSpanType::ArrayBoundsIndexType;
74 using value_type = DataType;
75 using LayoutPolicyType = LayoutPolicy;
77 using ConstSpanType ARCANE_DEPRECATED_REASON(
"Use 'ConstMDSpanType' instead") = ConstMDSpanType;
78 using SpanType ARCANE_DEPRECATED_REASON(
"Use 'MDSpanType' instead") = MDSpanType;
83 static constexpr int rank() {
return Extents::rank(); }
114 Int32 dim3_size,
Int32 dim4_size)
requires(Extents::nb_dynamic == 4)
115 : ThatClass(DynamicDimsType(dim1_size, dim2_size, dim3_size, dim4_size))
122 : ThatClass(DynamicDimsType(dim1_size, dim2_size, dim3_size, dim4_size), r)
128 : ThatClass(DynamicDimsType(dim1_size, dim2_size, dim3_size))
133 : ThatClass(DynamicDimsType(dim1_size, dim2_size, dim3_size), r)
139 : ThatClass(DynamicDimsType(dim1_size, dim2_size))
144 : ThatClass(DynamicDimsType(dim1_size, dim2_size), r)
150 : ThatClass(DynamicDimsType(dim1_size))
155 : ThatClass(DynamicDimsType(dim1_size), r)
166 requires(Extents::is_full_dynamic() && Extents::rank() == 2)
169 this->m_data.copyInitializerList(alist);
174 requires(Extents::isDynamic1D())
177 this->m_data.copyInitializerList(alist);
182 requires(Extents::isDynamic1D())
185 this->m_data.copy(v);
190 requires(Extents::isDynamic1D())
193 this->m_data.copy(v);
199 , m_total_nb_element(rhs.m_total_nb_element)
201 _updateSpanPointerFromData();
206 , m_data(std::move(rhs.m_data))
207 , m_total_nb_element(rhs.m_total_nb_element)
211 ThatClass& operator=(ThatClass&&) =
default;
227 m_data = ArrayWrapper(rhs_r);
241 m_data.swap(rhs.m_data);
242 std::swap(m_span, rhs.m_span);
243 std::swap(m_total_nb_element, rhs.m_total_nb_element);
256 return m_span.extentsWithOffset();
258 Int64 capacity()
const {
return m_data.
capacity(); }
259 eMemoryRessource memoryRessource()
const {
return m_data.memoryRessource(); }
290 constexpr Int32 extent0() const requires(Extents::
rank() >= 1) {
return m_span.extent0(); }
292 constexpr Int32 extent1() const requires(Extents::
rank() >= 2) {
return m_span.extent1(); }
294 constexpr Int32 extent2() const requires(Extents::
rank() >= 3) {
return m_span.extent2(); }
296 constexpr Int32 extent3() const requires(Extents::
rank() >= 4) {
return m_span.extent3(); }
303 m_span.m_extents = DynamicDimsType(dim1_size);
311 this->
resizeDestructive(DynamicDimsType(dim1_size, dim2_size, dim3_size, dim4_size));
337 this->
resizeDestructive(DynamicDimsType(dim1_size, dim2_size, dim3_size, dim4_size));
368 m_span.m_extents = dims;
395 m_data.fill(v, indexes, queue);
406 m_data.fill(v, indexes, &queue);
416 m_data.fill(v, queue);
426 m_data.fill(v, &queue);
437 _checkHost(memoryRessource());
460 void copy(ConstMDSpanType rhs) {
copy(rhs,
nullptr); }
468 void copy(
const ThatClass& rhs) {
copy(rhs,
nullptr); }
480 _resizeAndCopy(ConstMDSpanType(rhs), eMemoryRessource::Unknown, queue);
493 _resizeAndCopy(rhs, eMemoryRessource::Unknown, queue);
506 _resizeAndCopy(ConstMDSpanType(rhs), eMemoryRessource::Unknown, &queue);
519 _resizeAndCopy(rhs, eMemoryRessource::Unknown, &queue);
532 _resizeAndCopy(rhs.
constMDSpan(), rhs.memoryRessource(), queue);
545 _resizeAndCopy(rhs.
constMDSpan(), rhs.memoryRessource(), &queue);
553 DataType
operator[](
Int32 i)
const requires(Extents::rank() == 1) {
return m_span(i); }
560 return m_span(i, j, k, l);
565 return m_span(i, j, k, l);
571 return m_span(i, j, k);
576 return m_span(i, j, k);
590 DataType
operator()(
Int32 i)
const requires(Extents::rank() == 1) {
return m_span(i); }
611 ARCANE_DEPRECATED_REASON(
"Y2023: Use operator() instead")
614 return m_span(i, j, k, l);
617 ARCANE_DEPRECATED_REASON(
"Y2023: Use operator() instead")
620 return m_span(i, j, k);
623 ARCANE_DEPRECATED_REASON(
"Y2023: Use operator() instead")
629 ARCANE_DEPRECATED_REASON(
"Y2023: Use operator() instead")
630 DataType&
s(
Int32 i) requires(Extents::
rank() == 1) {
return m_span(i); }
633 ARCANE_DEPRECATED_REASON(
"Y2023: Use operator() instead")
634 DataType&
s(ArrayBoundsIndexType idx)
642 ARCANE_DEPRECATED_REASON(
"Y2024: Use mdspan() instead")
643 MDSpanType
span() {
return m_span; }
646 ARCANE_DEPRECATED_REASON(
"Y2024: Use mdspan() instead")
647 ConstMDSpanType
span()
const {
return m_span.constMDSpan(); }
650 ARCANE_DEPRECATED_REASON(
"Y2024: Use constMDSpan() instead")
651 ConstMDSpanType
constSpan()
const {
return m_span.constMDSpan(); }
657 ConstMDSpanType
mdspan()
const {
return m_span.constMDSpan(); }
660 ConstMDSpanType
constMDSpan()
const {
return m_span.constMDSpan(); }
669 constexpr operator MDSpanType() {
return this->
mdspan(); }
671 constexpr operator ConstMDSpanType()
const {
return this->
constMDSpan(); }
694 Int64 m_total_nb_element = 0;
698 void _updateSpanPointerFromData()
700 m_span.m_ptr = m_data.to1DSpan().data();
703 void _resizeAndCopy(ConstMDSpanType rhs,
eMemoryRessource input_ressource,
const RunQueue* queue)
705 this->
resize(rhs.extents().dynamicExtents());
706 m_data.copyOnly(rhs.to1DSpan(), input_ressource, queue);
707 _updateSpanPointerFromData();
713 m_total_nb_element = m_span.extents().totalNbElement();
714 m_data.resize(m_total_nb_element);
715 _updateSpanPointerFromData();
727 if constexpr (ExtentsType::nb_dynamic == 0) {
728 resize(DynamicDimsType());
Integer capacity() const
Capacité (nombre d'éléments alloués) du vecteur.
static constexpr int rank()
NumArray()
Construit un tableau vide.
File d'exécution pour un accélérateur.
Interface d'un allocateur pour la mémoire.
Classe de base des vues multi-dimensionnelles.
void resizeDestructive(Int32 dim1_size, Int32 dim2_size, Int32 dim3_size, Int32 dim4_size)
Modifie la taille du tableau.
DataType operator()(Int32 i) const
Valeur pour l'élément i.
void resizeDestructive(Int32 dim1_size, Int32 dim2_size)
Modifie la taille du tableau en ne gardant pas les valeurs actuelles.
void copy(const ThatClass &rhs, const RunQueue *queue)
Copie dans l'instance les valeurs de rhs via la file queue.
constexpr Int32 extent0() const
Valeur de la première dimension.
Span< const DataType > to1DSpan() const
Vue 1D constante sur l'instance.
Span< std::byte > bytes()
Vue sous forme d'octets.
DataType & s(Int32 i, Int32 j, Int32 k, Int32 l)
DataType & operator()(Int32 i, Int32 j)
Positionne la valeur pour l'élément i,j.
constexpr SmallSpan< const DataType > to1DConstSmallSpan() const
Vue constante 1D sur l'instance (uniquement si rank == 1)
constexpr Int32 dim4Size() const
Valeur de la quatrième dimension.
DataType & operator()(Int32 i)
Positionne la valeur pour l'élément i.
NumArray(Int32 dim1_size, Int32 dim2_size, Int32 dim3_size)
Construit un tableau avec 3 valeurs dynamiques.
void swap(ThatClass &rhs)
Échange les données avec rhs.
void copy(const ThatClass &rhs)
Copie dans l'instance les valeurs de rhs.
Span< DataType > to1DSpan()
Vue 1D sur l'instance.
static constexpr int rank()
Nombre de dimensions du tableau.
DataType * _internalData()
DataType operator()(Int32 i, Int32 j) const
Valeur pour l'élément i,j.
NumArray(Int32 dim1_size, Int32 dim2_size)
Construit un tableau avec 2 valeurs dynamiques.
void fill(const DataType &v, const RunQueue *queue)
Remplit les éléments de l'instance la valeur v en utilisant la file queue.
void copy(SmallSpan< const DataType > rhs)
Copie dans l'instance les valeurs de rhs.
constexpr Int32 dim3Size() const
Valeur de la troisième dimension.
NumArray(eMemoryRessource r)
Créé un tableau vide utilisant la ressource mémoire r.
MDSpanType mdspan()
Vue multi-dimension sur l'instance.
constexpr SmallSpan< DataType > to1DSmallSpan()
Vue 1D sur l'instance (uniquement si rank == 1)
ThatClass & operator=(const ThatClass &rhs)
Opérateur de recopie.
DataType operator()(Int32 i, Int32 j, Int32 k) const
Valeur pour l'élément i,j,k.
DataType & operator()(Int32 i, Int32 j, Int32 k)
Positionne la valeur pour l'élément i,j,k.
NumArray(Int32 dim1_size, std::initializer_list< DataType > alist)
Construit un tableau à partir de valeurs prédéfinies (uniquement tableaux 1D dynamiques)
DataType & operator[](Int32 i)
Récupère une référence pour l'élément i.
void copy(ConstMDSpanType rhs)
Copie dans l'instance les valeurs de rhs.
void _resize()
Redimensionne le tableau à partir des valeurs de m_span.extents()
void resizeDestructive(Int32 dim1_size, Int32 dim2_size, Int32 dim3_size)
Modifie la taille du tableau en ne gardant pas les valeurs actuelles.
void fill(const DataType &v, const RunQueue &queue)
Remplit les éléments de l'instance la valeur v en utilisant la file queue.
String debugName()
Nom de debug (nul si aucun nom spécifié)
Span< const std::byte > bytes() const
Vue constante forme d'octets.
IMemoryAllocator * memoryAllocator() const
Allocateur mémoire associé
DataType & operator()(ArrayBoundsIndexType idx)
Référence modifiable l'élément idx.
void copy(ConstMDSpanType rhs, const RunQueue &queue)
Copie dans l'instance les valeurs de rhs via la file queue.
ConstMDSpanType mdspan() const
Vue constante multi-dimension sur l'instance.
NumArray(Int32 dim1_size, eMemoryRessource r)
Construit un tableau avec 1 valeur dynamique.
constexpr SmallSpan< const DataType > to1DSmallSpan() const
Vue constante 1D sur l'instance (uniquement si rank == 1)
const DataType & operator()(ArrayBoundsIndexType idx) const
Référence constante pour l'élément idx.
NumArray(Int32 dim1_size, Int32 dim2_size, Int32 dim3_size, Int32 dim4_size)
Construit un tableau avec 4 valeurs dynamiques.
void fill(const DataType &v, SmallSpan< const Int32 > indexes, const RunQueue *queue)
Remplit via la file queue, les valeurs du tableau d'indices données par indexes par la valeur v .
void copy(SmallSpan< const DataType > rhs, const RunQueue &queue)
Copie dans l'instance les valeurs de rhs via la file queue.
void copy(ConstMDSpanType rhs, const RunQueue *queue)
Copie dans l'instance les valeurs de rhs via la file queue.
void resize(const DynamicDimsType &dims)
Modifie la taille du tableau en ne gardant pas les valeurs actuelles.
NumArray(Int32 dim1_size, Int32 dim2_size, eMemoryRessource r)
Construit un tableau avec 2 valeurs dynamiques.
void fill(const DataType &v)
Remplit les valeurs du tableau par v.
void fillHost(const DataType &v)
Remplit les valeurs du tableau par v.
constexpr Int32 extent1() const
Valeur de la deuxième dimension.
void resize(Int32 dim1_size, Int32 dim2_size)
Modifie la taille du tableau en ne gardant pas les valeurs actuelles.
static constexpr Int32 nbDimension()
Nombre de dimensions.
void _resizeInit()
Allocation éventuelle lors de l'initialisation.
void resizeDestructive(const DynamicDimsType &dims)
Modifie la taille du tableau en ne gardant pas les valeurs actuelles.
void resizeDestructive(Int32 dim1_size)
Modifie la taille du tableau en ne gardant pas les valeurs actuelles.
constexpr Int32 dim1Size() const
Valeur de la première dimension.
NumArray(const DynamicDimsType &extents, eMemoryRessource r)
Construit un tableau en spécifiant directement la liste des dimensions.
NumArray()
Construit un tableau vide.
DataType operator()(Int32 i, Int32 j, Int32 k, Int32 l) const
Valeur pour l'élément i,j,k,l.
void fill(const DataType &v, SmallSpan< const Int32 > indexes, const RunQueue &queue)
Remplit via la file queue, les valeurs du tableau d'indices données par indexes par la valeur v .
constexpr Int32 extent3() const
Valeur de la quatrième dimension.
NumArray(Int32 dim1_size, Int32 dim2_size, Int32 dim3_size, Int32 dim4_size, eMemoryRessource r)
Construit un tableau avec 4 valeurs dynamiques.
void copy(SmallSpan< const DataType > rhs, const RunQueue *queue)
Copie dans l'instance les valeurs de rhs via la file queue.
constexpr Int32 dim2Size() const
Valeur de la deuxième dimension.
ConstMDSpanType constSpan() const
NumArray(Int32 dim1_size)
Construit un tableau avec 1 valeur dynamique.
void setDebugName(const String &str)
Positionne le nom du tableau pour les informations de debug.
ConstMDSpanType constMDSpan() const
Vue constante multi-dimension sur l'instance.
void resize(Int32 dim1_size, Int32 dim2_size, Int32 dim3_size, Int32 dim4_size)
Modifie la taille du tableau en ne gardant pas les valeurs actuelles.
constexpr Int32 extent2() const
Valeur de la troisième dimension.
void resize(Int32 dim1_size, Int32 dim2_size, Int32 dim3_size)
Modifie la taille du tableau en ne gardant pas les valeurs actuelles.
NumArray(Int32 dim1_size, Int32 dim2_size, std::initializer_list< DataType > alist)
Construit un tableau à partir de valeurs prédéfinies (tableaux 2D dynamiques).
NumArray(Int32 dim1_size, Int32 dim2_size, Int32 dim3_size, eMemoryRessource r)
Construit un tableau avec 3 valeurs dynamiques.
void copy(const ThatClass &rhs, const RunQueue &queue)
Copie dans l'instance les valeurs de rhs via la file queue.
NumArray(DynamicDimsType extents)
Construit un tableau en spécifiant directement la liste des dimensions.
NumArray(SmallSpan< const DataType > v)
Construit une instance à partir d'une vue (uniquement tableaux 1D dynamiques)
DataType & operator()(Int32 i, Int32 j, Int32 k, Int32 l)
Positionne la valeur pour l'élément i,j,k,l.
ArrayExtents< MDDim1 > extents() const
DataType operator[](Int32 i) const
Valeur pour l'élément i.
NumArray(Span< const DataType > v)
Construit une instance à partir d'une vue (uniquement tableaux 1D dynamiques)
constexpr Int64 totalNbElement() const
Nombre total d'éléments du tableau.
void resize(Int32 dim1_size)
Modifie la taille du tableau en gardant pas les valeurs actuelles.
Vue d'un tableau d'éléments de type T.
Vue d'un tableau d'éléments de type T.
Chaîne de caractères unicode.
Implémentation commune à pour NumArray.
Container pour la classe NumArray.
constexpr view_type smallView()
Vue constante sur cette vue.
constexpr ConstArrayView< value_type > constSmallView() const
Vue constante sur cette vue.
-*- tab-width: 2; indent-tabs-mode: nil; coding: utf-8-with-signature -*-
Integer arcaneCheckArraySize(unsigned long long size)
Vérifie que size peut être converti dans un 'Integer' pour servir de taille à un tableau....
impl::SpanTypeFromSize< std::byte, SizeType >::SpanType asWritableBytes(const SpanImpl< DataType, SizeType, Extent > &s)
Converti la vue en un tableau d'octets modifiables.
std::int64_t Int64
Type entier signé sur 64 bits.
impl::SpanTypeFromSize< conststd::byte, SizeType >::SpanType asBytes(const SpanImpl< DataType, SizeType, Extent > &s)
Converti la vue en un tableau d'octets non modifiables.
Arcane::eMemoryResource eMemoryRessource
Typedef pour la version Arcane historique (avec 2's')
std::int32_t Int32
Type entier signé sur 32 bits.