Skip to content

Commit 58dc1ed

Browse files
ax3lax3l
authored
Particle Container: SoA Components (#226)
* Particle Container: SoA Components Expose properties for compile-time and runtime SoA components directly in the `ParticleContainer` class, too. * Update Stub Files --------- Co-authored-by: ax3l <[email protected]>
1 parent c85c7ba commit 58dc1ed

File tree

4 files changed

+729
-0
lines changed

4 files changed

+729
-0
lines changed

‎src/Particle/ParticleContainer.H‎

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -192,6 +192,15 @@ void make_ParticleContainer_and_Iterators (py::module &m, std::string allocstr)
192192
.def_property_readonly_static("NArrayReal", [](const py::object&){return ParticleContainerType::NArrayReal; })
193193
.def_property_readonly_static("NArrayInt", [](const py::object&){return ParticleContainerType::NArrayInt; })
194194

195+
.def_property_readonly("NumRealComps", &ParticleContainerType::NumRealComps,
196+
"The number of compile-time and runtime Real components in SoA")
197+
.def_property_readonly("NumIntComps", &ParticleContainerType::NumIntComps,
198+
"The number of compile-time and runtime int components in SoA")
199+
.def_property_readonly("NumRuntimeRealComps", &ParticleContainerType::NumRuntimeRealComps,
200+
"The number of runtime Real components in SoA")
201+
.def_property_readonly("NumRuntimeIntComps", &ParticleContainerType::NumRuntimeIntComps,
202+
"The number of runtime Int components in SoA")
203+
195204
.def_property_readonly("finest_level", &ParticleContainerBase::finestLevel)
196205

197206
// ParticleContainer ( const ParticleContainer &) = delete;

‎src/amrex/space1d/amrex_1d_pybind/__init__.pyi‎

Lines changed: 240 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5999,6 +5999,26 @@ class ParticleContainer_0_0_4_0_arena:
59995999
60006000
"""
60016001
@property
6002+
def NumIntComps(self) -> int:
6003+
"""
6004+
The number of compile-time and runtime int components in SoA
6005+
"""
6006+
@property
6007+
def NumRealComps(self) -> int:
6008+
"""
6009+
The number of compile-time and runtime Real components in SoA
6010+
"""
6011+
@property
6012+
def NumRuntimeIntComps(self) -> int:
6013+
"""
6014+
The number of runtime Int components in SoA
6015+
"""
6016+
@property
6017+
def NumRuntimeRealComps(self) -> int:
6018+
"""
6019+
The number of runtime Real components in SoA
6020+
"""
6021+
@property
60026022
def finest_level(self) -> int: ...
60036023

60046024
class ParticleContainer_0_0_4_0_default:
@@ -6126,6 +6146,26 @@ class ParticleContainer_0_0_4_0_default:
61266146
61276147
"""
61286148
@property
6149+
def NumIntComps(self) -> int:
6150+
"""
6151+
The number of compile-time and runtime int components in SoA
6152+
"""
6153+
@property
6154+
def NumRealComps(self) -> int:
6155+
"""
6156+
The number of compile-time and runtime Real components in SoA
6157+
"""
6158+
@property
6159+
def NumRuntimeIntComps(self) -> int:
6160+
"""
6161+
The number of runtime Int components in SoA
6162+
"""
6163+
@property
6164+
def NumRuntimeRealComps(self) -> int:
6165+
"""
6166+
The number of runtime Real components in SoA
6167+
"""
6168+
@property
61296169
def finest_level(self) -> int: ...
61306170

61316171
class ParticleContainer_0_0_4_0_pinned:
@@ -6253,6 +6293,26 @@ class ParticleContainer_0_0_4_0_pinned:
62536293
62546294
"""
62556295
@property
6296+
def NumIntComps(self) -> int:
6297+
"""
6298+
The number of compile-time and runtime int components in SoA
6299+
"""
6300+
@property
6301+
def NumRealComps(self) -> int:
6302+
"""
6303+
The number of compile-time and runtime Real components in SoA
6304+
"""
6305+
@property
6306+
def NumRuntimeIntComps(self) -> int:
6307+
"""
6308+
The number of runtime Int components in SoA
6309+
"""
6310+
@property
6311+
def NumRuntimeRealComps(self) -> int:
6312+
"""
6313+
The number of runtime Real components in SoA
6314+
"""
6315+
@property
62566316
def finest_level(self) -> int: ...
62576317

62586318
class ParticleContainer_0_0_5_0_arena:
@@ -6380,6 +6440,26 @@ class ParticleContainer_0_0_5_0_arena:
63806440
63816441
"""
63826442
@property
6443+
def NumIntComps(self) -> int:
6444+
"""
6445+
The number of compile-time and runtime int components in SoA
6446+
"""
6447+
@property
6448+
def NumRealComps(self) -> int:
6449+
"""
6450+
The number of compile-time and runtime Real components in SoA
6451+
"""
6452+
@property
6453+
def NumRuntimeIntComps(self) -> int:
6454+
"""
6455+
The number of runtime Int components in SoA
6456+
"""
6457+
@property
6458+
def NumRuntimeRealComps(self) -> int:
6459+
"""
6460+
The number of runtime Real components in SoA
6461+
"""
6462+
@property
63836463
def finest_level(self) -> int: ...
63846464

63856465
class ParticleContainer_0_0_5_0_default:
@@ -6507,6 +6587,26 @@ class ParticleContainer_0_0_5_0_default:
65076587
65086588
"""
65096589
@property
6590+
def NumIntComps(self) -> int:
6591+
"""
6592+
The number of compile-time and runtime int components in SoA
6593+
"""
6594+
@property
6595+
def NumRealComps(self) -> int:
6596+
"""
6597+
The number of compile-time and runtime Real components in SoA
6598+
"""
6599+
@property
6600+
def NumRuntimeIntComps(self) -> int:
6601+
"""
6602+
The number of runtime Int components in SoA
6603+
"""
6604+
@property
6605+
def NumRuntimeRealComps(self) -> int:
6606+
"""
6607+
The number of runtime Real components in SoA
6608+
"""
6609+
@property
65106610
def finest_level(self) -> int: ...
65116611

65126612
class ParticleContainer_0_0_5_0_pinned:
@@ -6634,6 +6734,26 @@ class ParticleContainer_0_0_5_0_pinned:
66346734
66356735
"""
66366736
@property
6737+
def NumIntComps(self) -> int:
6738+
"""
6739+
The number of compile-time and runtime int components in SoA
6740+
"""
6741+
@property
6742+
def NumRealComps(self) -> int:
6743+
"""
6744+
The number of compile-time and runtime Real components in SoA
6745+
"""
6746+
@property
6747+
def NumRuntimeIntComps(self) -> int:
6748+
"""
6749+
The number of runtime Int components in SoA
6750+
"""
6751+
@property
6752+
def NumRuntimeRealComps(self) -> int:
6753+
"""
6754+
The number of runtime Real components in SoA
6755+
"""
6756+
@property
66376757
def finest_level(self) -> int: ...
66386758

66396759
class ParticleContainer_1_1_2_1_arena:
@@ -6761,6 +6881,26 @@ class ParticleContainer_1_1_2_1_arena:
67616881
67626882
"""
67636883
@property
6884+
def NumIntComps(self) -> int:
6885+
"""
6886+
The number of compile-time and runtime int components in SoA
6887+
"""
6888+
@property
6889+
def NumRealComps(self) -> int:
6890+
"""
6891+
The number of compile-time and runtime Real components in SoA
6892+
"""
6893+
@property
6894+
def NumRuntimeIntComps(self) -> int:
6895+
"""
6896+
The number of runtime Int components in SoA
6897+
"""
6898+
@property
6899+
def NumRuntimeRealComps(self) -> int:
6900+
"""
6901+
The number of runtime Real components in SoA
6902+
"""
6903+
@property
67646904
def finest_level(self) -> int: ...
67656905

67666906
class ParticleContainer_1_1_2_1_default:
@@ -6888,6 +7028,26 @@ class ParticleContainer_1_1_2_1_default:
68887028
68897029
"""
68907030
@property
7031+
def NumIntComps(self) -> int:
7032+
"""
7033+
The number of compile-time and runtime int components in SoA
7034+
"""
7035+
@property
7036+
def NumRealComps(self) -> int:
7037+
"""
7038+
The number of compile-time and runtime Real components in SoA
7039+
"""
7040+
@property
7041+
def NumRuntimeIntComps(self) -> int:
7042+
"""
7043+
The number of runtime Int components in SoA
7044+
"""
7045+
@property
7046+
def NumRuntimeRealComps(self) -> int:
7047+
"""
7048+
The number of runtime Real components in SoA
7049+
"""
7050+
@property
68917051
def finest_level(self) -> int: ...
68927052

68937053
class ParticleContainer_1_1_2_1_pinned:
@@ -7015,6 +7175,26 @@ class ParticleContainer_1_1_2_1_pinned:
70157175
70167176
"""
70177177
@property
7178+
def NumIntComps(self) -> int:
7179+
"""
7180+
The number of compile-time and runtime int components in SoA
7181+
"""
7182+
@property
7183+
def NumRealComps(self) -> int:
7184+
"""
7185+
The number of compile-time and runtime Real components in SoA
7186+
"""
7187+
@property
7188+
def NumRuntimeIntComps(self) -> int:
7189+
"""
7190+
The number of runtime Int components in SoA
7191+
"""
7192+
@property
7193+
def NumRuntimeRealComps(self) -> int:
7194+
"""
7195+
The number of runtime Real components in SoA
7196+
"""
7197+
@property
70187198
def finest_level(self) -> int: ...
70197199

70207200
class ParticleContainer_pureSoA_8_2_arena:
@@ -7128,6 +7308,26 @@ class ParticleContainer_pureSoA_8_2_arena:
71287308
71297309
"""
71307310
@property
7311+
def NumIntComps(self) -> int:
7312+
"""
7313+
The number of compile-time and runtime int components in SoA
7314+
"""
7315+
@property
7316+
def NumRealComps(self) -> int:
7317+
"""
7318+
The number of compile-time and runtime Real components in SoA
7319+
"""
7320+
@property
7321+
def NumRuntimeIntComps(self) -> int:
7322+
"""
7323+
The number of runtime Int components in SoA
7324+
"""
7325+
@property
7326+
def NumRuntimeRealComps(self) -> int:
7327+
"""
7328+
The number of runtime Real components in SoA
7329+
"""
7330+
@property
71317331
def finest_level(self) -> int: ...
71327332

71337333
class ParticleContainer_pureSoA_8_2_default:
@@ -7241,6 +7441,26 @@ class ParticleContainer_pureSoA_8_2_default:
72417441
72427442
"""
72437443
@property
7444+
def NumIntComps(self) -> int:
7445+
"""
7446+
The number of compile-time and runtime int components in SoA
7447+
"""
7448+
@property
7449+
def NumRealComps(self) -> int:
7450+
"""
7451+
The number of compile-time and runtime Real components in SoA
7452+
"""
7453+
@property
7454+
def NumRuntimeIntComps(self) -> int:
7455+
"""
7456+
The number of runtime Int components in SoA
7457+
"""
7458+
@property
7459+
def NumRuntimeRealComps(self) -> int:
7460+
"""
7461+
The number of runtime Real components in SoA
7462+
"""
7463+
@property
72447464
def finest_level(self) -> int: ...
72457465

72467466
class ParticleContainer_pureSoA_8_2_pinned:
@@ -7354,6 +7574,26 @@ class ParticleContainer_pureSoA_8_2_pinned:
73547574
73557575
"""
73567576
@property
7577+
def NumIntComps(self) -> int:
7578+
"""
7579+
The number of compile-time and runtime int components in SoA
7580+
"""
7581+
@property
7582+
def NumRealComps(self) -> int:
7583+
"""
7584+
The number of compile-time and runtime Real components in SoA
7585+
"""
7586+
@property
7587+
def NumRuntimeIntComps(self) -> int:
7588+
"""
7589+
The number of runtime Int components in SoA
7590+
"""
7591+
@property
7592+
def NumRuntimeRealComps(self) -> int:
7593+
"""
7594+
The number of runtime Real components in SoA
7595+
"""
7596+
@property
73577597
def finest_level(self) -> int: ...
73587598

73597599
class ParticleInitType_0_0_4_0:

0 commit comments

Comments
 (0)