Shamrock 2025.10.0
Astrophysical Code
Loading...
Searching...
No Matches
SolverStorage.hpp
Go to the documentation of this file.
1// -------------------------------------------------------//
2//
3// SHAMROCK code for hydrodynamics
4// Copyright (c) 2021-2026 Timothée David--Cléris <tim.shamrock@proton.me>
5// SPDX-License-Identifier: CeCILL Free Software License Agreement v2.1
6// Shamrock is licensed under the CeCILL 2.1 License, see LICENSE for more information
7//
8// -------------------------------------------------------//
9
10#pragma once
11
20#include "shambase/StorageComponent.hpp"
21#include "shambase/stacktrace.hpp"
22#include "shambackends/vec.hpp"
23#include "shammodels/common/amr/AMRCellInfos.hpp"
24#include "shammodels/common/amr/AMRStencilCache.hpp"
25#include "shammodels/common/amr/NeighGraph.hpp"
26#include "shammodels/common/amr/NeighGraphLinkField.hpp"
27#include "shammodels/ramses/GhostZoneData.hpp"
28#include "shammodels/ramses/modules/FindGhostLayerCandidates.hpp"
29#include "shammodels/ramses/modules/NodeBuildTrees.hpp"
30#include "shammodels/ramses/solvegraph/NeighGraphLinkFieldEdge.hpp"
31#include "shammodels/ramses/solvegraph/OrientedAMRGraphEdge.hpp"
32#include "shammodels/ramses/solvegraph/TreeEdge.hpp"
33#include "shamrock/scheduler/ComputeField.hpp"
34#include "shamrock/scheduler/InterfacesUtility.hpp"
35#include "shamrock/scheduler/SerialPatchTree.hpp"
36#include "shamrock/scheduler/ShamrockCtx.hpp"
37#include "shamrock/solvergraph/DDSharedBuffers.hpp"
38#include "shamrock/solvergraph/DDSharedScalar.hpp"
39#include "shamrock/solvergraph/Field.hpp"
40#include "shamrock/solvergraph/FieldSpan.hpp"
41#include "shamrock/solvergraph/Indexes.hpp"
42#include "shamrock/solvergraph/OperationSequence.hpp"
43#include "shamrock/solvergraph/PatchDataLayerDDShared.hpp"
44#include "shamrock/solvergraph/PatchDataLayerEdge.hpp"
45#include "shamrock/solvergraph/RankGetter.hpp"
46#include "shamrock/solvergraph/ScalarEdge.hpp"
47#include "shamrock/solvergraph/ScalarsEdge.hpp"
48#include "shamrock/solvergraph/SolverGraph.hpp"
49#include "shamsys/legacy/log.hpp"
50#include "shamtree/RadixTree.hpp"
51#include "shamtree/TreeTraversalCache.hpp"
52#include <memory>
53
54namespace shammodels::basegodunov {
55
56 template<class T>
57 using Component = shambase::StorageComponent<T>;
58
59 template<class Tvec, class TgridVec, class Tmorton_>
60 class SolverStorage {
61 public:
62 using Tmorton = Tmorton_;
63 using Tscal = shambase::VecComponent<Tvec>;
64 using Tgridscal = shambase::VecComponent<TgridVec>;
65 using TgridUint = typename std::make_unsigned<shambase::VecComponent<TgridVec>>::type;
66 static constexpr u32 dim = shambase::VectorProperties<Tvec>::dimension;
67
68 using RTree = RadixTree<Tmorton, TgridVec>;
69
70 shamrock::solvergraph::SolverGraph solver_graph;
71
72 std::shared_ptr<shamrock::solvergraph::IDataEdge<std::vector<u64>>> local_patch_ids;
73
74 std::shared_ptr<shamrock::solvergraph::RankGetter> patch_rank_owner;
75
76 std::shared_ptr<shamrock::solvergraph::ScalarEdge<Tscal>> dt_over2;
77
78 std::shared_ptr<shamrock::solvergraph::FieldRefs<TgridVec>> refs_block_min;
79 std::shared_ptr<shamrock::solvergraph::FieldRefs<TgridVec>> refs_block_max;
80
81 std::shared_ptr<shamrock::solvergraph::ScalarsEdge<TgridVec>> level0_size;
82 std::shared_ptr<shamrock::solvergraph::Field<TgridUint>> amr_block_levels;
83
84 std::shared_ptr<shamrock::solvergraph::Indexes<u32>> block_counts;
85 std::shared_ptr<shamrock::solvergraph::Indexes<u32>> block_counts_with_ghost;
86 std::shared_ptr<shamrock::solvergraph::FieldRefs<Tscal>> refs_rho;
87 std::shared_ptr<shamrock::solvergraph::FieldRefs<Tvec>> refs_rhov;
88 std::shared_ptr<shamrock::solvergraph::FieldRefs<Tscal>> refs_rhoe;
89 std::shared_ptr<shamrock::solvergraph::FieldRefs<Tscal>> refs_rho_dust;
90 std::shared_ptr<shamrock::solvergraph::FieldRefs<Tvec>> refs_rhov_dust;
91
92 std::shared_ptr<shamrock::solvergraph::Field<Tvec>> vel;
93 std::shared_ptr<shamrock::solvergraph::Field<Tscal>> press;
94 std::shared_ptr<shamrock::solvergraph::Field<Tvec>> vel_dust;
95
96 std::shared_ptr<shamrock::solvergraph::Field<Tscal>> block_cell_sizes;
97 std::shared_ptr<shamrock::solvergraph::Field<Tvec>> cell0block_aabb_lower;
98 std::shared_ptr<shamrock::solvergraph::Field<Tvec>> coordinates;
99
100 std::shared_ptr<shamrock::solvergraph::Field<Tvec>> grad_rho;
101 std::shared_ptr<shamrock::solvergraph::Field<Tvec>> dx_v;
102 std::shared_ptr<shamrock::solvergraph::Field<Tvec>> dy_v;
103 std::shared_ptr<shamrock::solvergraph::Field<Tvec>> dz_v;
104 std::shared_ptr<shamrock::solvergraph::Field<Tvec>> grad_P;
106 std::shared_ptr<shamrock::solvergraph::Field<Tvec>> grad_rho_dust;
108 std::shared_ptr<shamrock::solvergraph::Field<Tvec>> dx_v_dust;
110 std::shared_ptr<shamrock::solvergraph::Field<Tvec>> dy_v_dust;
112 std::shared_ptr<shamrock::solvergraph::Field<Tvec>> dz_v_dust;
113
114 std::shared_ptr<shamrock::solvergraph::ScalarEdge<Tscal>> rho_mean;
115 std::shared_ptr<shamrock::solvergraph::ScalarEdge<Tscal>> simulation_volume;
116 std::shared_ptr<shamrock::solvergraph::Field<Tscal>> cell_mass;
117
118 Component<shambase::DistributedData<shammath::AABB<TgridVec>>> merge_patch_bounds;
119
120 std::shared_ptr<solvergraph::TreeEdge<Tmorton, TgridVec>> trees;
121
122 std::shared_ptr<shammodels::basegodunov::solvergraph::OrientedAMRGraphEdge<Tvec, TgridVec>>
123 block_graph_edge;
124
125 std::shared_ptr<shammodels::basegodunov::solvergraph::OrientedAMRGraphEdge<Tvec, TgridVec>>
126 cell_graph_edge;
127
128 std::shared_ptr<shamrock::solvergraph::OperationSequence> solver_sequence;
129
130 std::shared_ptr<shamrock::solvergraph::OperationSequence> node_cons_to_prim;
131 std::shared_ptr<modules::NodeBuildTrees<Tmorton, TgridVec>> build_trees;
132
133 Component<SerialPatchTree<TgridVec>> serial_patch_tree;
134
135 Component<GhostZonesData<Tvec, TgridVec>> ghost_zone_infos;
136
137 std::shared_ptr<shamrock::patch::PatchDataLayerLayout> ghost_layout;
138
139 std::shared_ptr<shamrock::solvergraph::PatchDataLayerRefs> source_patches;
140
141 std::shared_ptr<shamrock::solvergraph::PatchDataLayerEdge> merged_patchdata_ghost;
142
143 std::shared_ptr<shamrock::solvergraph::PatchDataLayerDDShared> exchange_gz_edge;
144
145 std::shared_ptr<shamrock::solvergraph::DDSharedScalar<modules::GhostLayerCandidateInfos>>
146 ghost_layers_candidates_edge;
147
148 std::shared_ptr<shamrock::solvergraph::ScalarEdge<shammath::AABB<TgridVec>>> sim_box_edge;
149
150 std::shared_ptr<shamrock::solvergraph::DDSharedBuffers<u32>> idx_in_ghost;
151
152 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<std::array<Tscal, 2>>> rho_face_xp;
153 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<std::array<Tscal, 2>>> rho_face_xm;
154 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<std::array<Tscal, 2>>> rho_face_yp;
155 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<std::array<Tscal, 2>>> rho_face_ym;
156 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<std::array<Tscal, 2>>> rho_face_zp;
157 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<std::array<Tscal, 2>>> rho_face_zm;
158
159 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<std::array<Tvec, 2>>> vel_face_xp;
160 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<std::array<Tvec, 2>>> vel_face_xm;
161 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<std::array<Tvec, 2>>> vel_face_yp;
162 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<std::array<Tvec, 2>>> vel_face_ym;
163 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<std::array<Tvec, 2>>> vel_face_zp;
164 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<std::array<Tvec, 2>>> vel_face_zm;
165
166 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<std::array<Tscal, 2>>> press_face_xp;
167 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<std::array<Tscal, 2>>> press_face_xm;
168 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<std::array<Tscal, 2>>> press_face_yp;
169 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<std::array<Tscal, 2>>> press_face_ym;
170 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<std::array<Tscal, 2>>> press_face_zp;
171 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<std::array<Tscal, 2>>> press_face_zm;
172
173 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<Tscal>> flux_rho_face_xp;
174 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<Tscal>> flux_rho_face_xm;
175 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<Tscal>> flux_rho_face_yp;
176 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<Tscal>> flux_rho_face_ym;
177 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<Tscal>> flux_rho_face_zp;
178 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<Tscal>> flux_rho_face_zm;
179
180 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<Tvec>> flux_rhov_face_xp;
181 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<Tvec>> flux_rhov_face_xm;
182 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<Tvec>> flux_rhov_face_yp;
183 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<Tvec>> flux_rhov_face_ym;
184 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<Tvec>> flux_rhov_face_zp;
185 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<Tvec>> flux_rhov_face_zm;
186
187 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<Tscal>> flux_rhoe_face_xp;
188 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<Tscal>> flux_rhoe_face_xm;
189 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<Tscal>> flux_rhoe_face_yp;
190 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<Tscal>> flux_rhoe_face_ym;
191 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<Tscal>> flux_rhoe_face_zp;
192 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<Tscal>> flux_rhoe_face_zm;
193
194 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<Tscal>> flux_rho_dust_face_xp;
195 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<Tscal>> flux_rho_dust_face_xm;
196 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<Tscal>> flux_rho_dust_face_yp;
197 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<Tscal>> flux_rho_dust_face_ym;
198 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<Tscal>> flux_rho_dust_face_zp;
199 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<Tscal>> flux_rho_dust_face_zm;
200
201 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<Tvec>> flux_rhov_dust_face_xp;
202 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<Tvec>> flux_rhov_dust_face_xm;
203 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<Tvec>> flux_rhov_dust_face_yp;
204 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<Tvec>> flux_rhov_dust_face_ym;
205 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<Tvec>> flux_rhov_dust_face_zp;
206 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<Tvec>> flux_rhov_dust_face_zm;
207
208 std::shared_ptr<shamrock::solvergraph::Field<Tscal>> dtrho;
209 std::shared_ptr<shamrock::solvergraph::Field<Tvec>> dtrhov;
210 std::shared_ptr<shamrock::solvergraph::Field<Tscal>> dtrhoe;
211
212 Component<shamrock::ComputeField<Tscal>> rho_next_no_drag;
213 Component<shamrock::ComputeField<Tvec>> rhov_next_no_drag;
214 Component<shamrock::ComputeField<Tscal>> rhoe_next_no_drag;
215
216 // next time step dust density before drag
217 Component<shamrock::ComputeField<Tscal>> rho_d_next_no_drag;
218 // next time step dust momentum before drag
219 Component<shamrock::ComputeField<Tvec>> rhov_d_next_no_drag;
220
221 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<std::array<Tscal, 2>>>
222 rho_dust_face_xp;
223 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<std::array<Tscal, 2>>>
224 rho_dust_face_xm;
225 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<std::array<Tscal, 2>>>
226 rho_dust_face_yp;
227 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<std::array<Tscal, 2>>>
228 rho_dust_face_ym;
229 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<std::array<Tscal, 2>>>
230 rho_dust_face_zp;
231 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<std::array<Tscal, 2>>>
232 rho_dust_face_zm;
233
234 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<std::array<Tvec, 2>>> vel_dust_face_xp;
235 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<std::array<Tvec, 2>>> vel_dust_face_xm;
236 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<std::array<Tvec, 2>>> vel_dust_face_yp;
237 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<std::array<Tvec, 2>>> vel_dust_face_ym;
238 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<std::array<Tvec, 2>>> vel_dust_face_zp;
239 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<std::array<Tvec, 2>>> vel_dust_face_zm;
240
241 // time derivative dust density
242 std::shared_ptr<shamrock::solvergraph::Field<Tscal>> dtrho_dust;
243 // time derivative dust momemtum
244 std::shared_ptr<shamrock::solvergraph::Field<Tvec>> dtrhov_dust;
245
246 struct Timings {
247 f64 interface = 0;
248 f64 neighbors = 0;
249 f64 io = 0;
250
252 void reset() { *this = {}; }
253 } timings_details;
254 };
255
256} // namespace shammodels::basegodunov
DDSharedBuffers.hpp
Defines the DDSharedBuffers class for managing buffers contained in a distributed data shared.
PatchDataLayerDDShared.hpp
Shared distributed data layer for patch data management in solver graphs.
PatchDataLayerEdge.hpp
Defines the PatchDataLayerEdge class for managing patch data layer edges.
SolverGraph.hpp
Declare a class to register and retrieve nodes and edges from a unique container.
f64
double f64
Alias for double.
Definition aliases_float.hpp:20
u32
std::uint32_t u32
32 bit unsigned integer
Definition aliases_int.hpp:27
RadixTree
The radix tree.
Definition RadixTree.hpp:50
shambase::StorageComponent
Helper class for Storage Module of any solver.
Definition StorageComponent.hpp:35
shammodels::basegodunov::SolverStorage::dy_v_dust
std::shared_ptr< shamrock::solvergraph::Field< Tvec > > dy_v_dust
dust fields gradients (d vdust / d y)
Definition SolverStorage.hpp:110
shammodels::basegodunov::SolverStorage::dz_v_dust
std::shared_ptr< shamrock::solvergraph::Field< Tvec > > dz_v_dust
dust fields gradients (d vdust / d z)
Definition SolverStorage.hpp:112
shammodels::basegodunov::SolverStorage::grad_rho_dust
std::shared_ptr< shamrock::solvergraph::Field< Tvec > > grad_rho_dust
dust fields gradients (grad rho_dust)
Definition SolverStorage.hpp:106
shammodels::basegodunov::SolverStorage::dx_v_dust
std::shared_ptr< shamrock::solvergraph::Field< Tvec > > dx_v_dust
dust fields gradients (d vdust / d x)
Definition SolverStorage.hpp:108
shamrock::solvergraph::SolverGraph
A graph container for managing solver nodes and edges with type-safe access.
Definition SolverGraph.hpp:56
shammodels::basegodunov
namespace for the basegodunov model
Definition AMRCellInfos.hpp:20
stacktrace.hpp
This file contains the definition for the stacktrace related functionality.
shammodels::basegodunov::SolverStorage::Timings::reset
void reset()
Reset the timings logged in the storage.
Definition SolverStorage.hpp:252
Generated on Wed May 13 2026 13:21:49 for Shamrock by doxygen 1.9.8