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
19
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/PatchDataLayerDDShared.hpp"
43#include "shamrock/solvergraph/PatchDataLayerEdge.hpp"
44#include "shamrock/solvergraph/RankGetter.hpp"
45#include "shamrock/solvergraph/ScalarEdge.hpp"
46#include "shamrock/solvergraph/ScalarsEdge.hpp"
47#include "shamsolvergraph/SolverGraph.hpp"
48#include "shamsolvergraph/node/OperationSequence.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 std::shared_ptr<shamrock::solvergraph::Field<Tscal>> rho_primitive;
96
97 std::shared_ptr<shamrock::solvergraph::Field<Tscal>> block_cell_sizes;
98 std::shared_ptr<shamrock::solvergraph::Field<Tvec>> cell0block_aabb_lower;
99 std::shared_ptr<shamrock::solvergraph::Field<Tvec>> coordinates;
100
101 std::shared_ptr<shamrock::solvergraph::Field<Tvec>> grad_rho;
102 std::shared_ptr<shamrock::solvergraph::Field<Tvec>> dx_v;
103 std::shared_ptr<shamrock::solvergraph::Field<Tvec>> dy_v;
104 std::shared_ptr<shamrock::solvergraph::Field<Tvec>> dz_v;
105 std::shared_ptr<shamrock::solvergraph::Field<Tvec>> grad_P;
107 std::shared_ptr<shamrock::solvergraph::Field<Tvec>> grad_rho_dust;
109 std::shared_ptr<shamrock::solvergraph::Field<Tvec>> dx_v_dust;
111 std::shared_ptr<shamrock::solvergraph::Field<Tvec>> dy_v_dust;
113 std::shared_ptr<shamrock::solvergraph::Field<Tvec>> dz_v_dust;
114
115 std::shared_ptr<shamrock::solvergraph::ScalarEdge<Tscal>> rho_mean;
116 std::shared_ptr<shamrock::solvergraph::ScalarEdge<Tscal>> simulation_volume;
117 std::shared_ptr<shamrock::solvergraph::Field<Tscal>> cell_mass;
118
119 Component<shambase::DistributedData<shammath::AABB<TgridVec>>> merge_patch_bounds;
120
121 std::shared_ptr<solvergraph::TreeEdge<Tmorton, TgridVec>> trees;
122
123 std::shared_ptr<shammodels::basegodunov::solvergraph::OrientedAMRGraphEdge<Tvec, TgridVec>>
124 block_graph_edge;
125
126 std::shared_ptr<shammodels::basegodunov::solvergraph::OrientedAMRGraphEdge<Tvec, TgridVec>>
127 cell_graph_edge;
128
129 std::shared_ptr<shamrock::solvergraph::OperationSequence> solver_sequence;
130
131 std::shared_ptr<shamrock::solvergraph::OperationSequence> node_cons_to_prim;
132 std::shared_ptr<modules::NodeBuildTrees<Tmorton, TgridVec>> build_trees;
133
134 Component<SerialPatchTree<TgridVec>> serial_patch_tree;
135
136 Component<GhostZonesData<Tvec, TgridVec>> ghost_zone_infos;
137
138 std::shared_ptr<shamrock::patch::PatchDataLayerLayout> ghost_layout;
139
140 std::shared_ptr<shamrock::solvergraph::PatchDataLayerRefs> source_patches;
141
142 std::shared_ptr<shamrock::solvergraph::PatchDataLayerEdge> merged_patchdata_ghost;
143
144 std::shared_ptr<shamrock::solvergraph::PatchDataLayerDDShared> exchange_gz_edge;
145
146 std::shared_ptr<shamrock::solvergraph::DDSharedScalar<modules::GhostLayerCandidateInfos>>
147 ghost_layers_candidates_edge;
148
149 std::shared_ptr<shamrock::solvergraph::ScalarEdge<shammath::AABB<TgridVec>>> sim_box_edge;
150
151 std::shared_ptr<shamrock::solvergraph::DDSharedBuffers<u32>> idx_in_ghost;
152
153 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<std::array<Tscal, 2>>> rho_face_xp;
154 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<std::array<Tscal, 2>>> rho_face_xm;
155 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<std::array<Tscal, 2>>> rho_face_yp;
156 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<std::array<Tscal, 2>>> rho_face_ym;
157 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<std::array<Tscal, 2>>> rho_face_zp;
158 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<std::array<Tscal, 2>>> rho_face_zm;
159
160 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<std::array<Tvec, 2>>> vel_face_xp;
161 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<std::array<Tvec, 2>>> vel_face_xm;
162 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<std::array<Tvec, 2>>> vel_face_yp;
163 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<std::array<Tvec, 2>>> vel_face_ym;
164 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<std::array<Tvec, 2>>> vel_face_zp;
165 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<std::array<Tvec, 2>>> vel_face_zm;
166
167 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<std::array<Tscal, 2>>> press_face_xp;
168 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<std::array<Tscal, 2>>> press_face_xm;
169 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<std::array<Tscal, 2>>> press_face_yp;
170 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<std::array<Tscal, 2>>> press_face_ym;
171 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<std::array<Tscal, 2>>> press_face_zp;
172 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<std::array<Tscal, 2>>> press_face_zm;
173
174 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<Tscal>> flux_rho_face_xp;
175 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<Tscal>> flux_rho_face_xm;
176 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<Tscal>> flux_rho_face_yp;
177 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<Tscal>> flux_rho_face_ym;
178 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<Tscal>> flux_rho_face_zp;
179 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<Tscal>> flux_rho_face_zm;
180
181 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<Tvec>> flux_rhov_face_xp;
182 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<Tvec>> flux_rhov_face_xm;
183 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<Tvec>> flux_rhov_face_yp;
184 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<Tvec>> flux_rhov_face_ym;
185 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<Tvec>> flux_rhov_face_zp;
186 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<Tvec>> flux_rhov_face_zm;
187
188 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<Tscal>> flux_rhoe_face_xp;
189 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<Tscal>> flux_rhoe_face_xm;
190 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<Tscal>> flux_rhoe_face_yp;
191 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<Tscal>> flux_rhoe_face_ym;
192 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<Tscal>> flux_rhoe_face_zp;
193 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<Tscal>> flux_rhoe_face_zm;
194
195 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<Tscal>> flux_rho_dust_face_xp;
196 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<Tscal>> flux_rho_dust_face_xm;
197 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<Tscal>> flux_rho_dust_face_yp;
198 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<Tscal>> flux_rho_dust_face_ym;
199 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<Tscal>> flux_rho_dust_face_zp;
200 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<Tscal>> flux_rho_dust_face_zm;
201
202 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<Tvec>> flux_rhov_dust_face_xp;
203 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<Tvec>> flux_rhov_dust_face_xm;
204 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<Tvec>> flux_rhov_dust_face_yp;
205 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<Tvec>> flux_rhov_dust_face_ym;
206 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<Tvec>> flux_rhov_dust_face_zp;
207 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<Tvec>> flux_rhov_dust_face_zm;
208
209 std::shared_ptr<shamrock::solvergraph::Field<Tscal>> dtrho;
210 std::shared_ptr<shamrock::solvergraph::Field<Tvec>> dtrhov;
211 std::shared_ptr<shamrock::solvergraph::Field<Tscal>> dtrhoe;
212
213 Component<shamrock::ComputeField<Tscal>> rho_next_no_drag;
214 Component<shamrock::ComputeField<Tvec>> rhov_next_no_drag;
215 Component<shamrock::ComputeField<Tscal>> rhoe_next_no_drag;
216
217 // next time step dust density before drag
218 Component<shamrock::ComputeField<Tscal>> rho_d_next_no_drag;
219 // next time step dust momentum before drag
220 Component<shamrock::ComputeField<Tvec>> rhov_d_next_no_drag;
221
222 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<std::array<Tscal, 2>>>
223 rho_dust_face_xp;
224 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<std::array<Tscal, 2>>>
225 rho_dust_face_xm;
226 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<std::array<Tscal, 2>>>
227 rho_dust_face_yp;
228 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<std::array<Tscal, 2>>>
229 rho_dust_face_ym;
230 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<std::array<Tscal, 2>>>
231 rho_dust_face_zp;
232 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<std::array<Tscal, 2>>>
233 rho_dust_face_zm;
234
235 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<std::array<Tvec, 2>>> vel_dust_face_xp;
236 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<std::array<Tvec, 2>>> vel_dust_face_xm;
237 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<std::array<Tvec, 2>>> vel_dust_face_yp;
238 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<std::array<Tvec, 2>>> vel_dust_face_ym;
239 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<std::array<Tvec, 2>>> vel_dust_face_zp;
240 std::shared_ptr<solvergraph::NeighGraphLinkFieldEdge<std::array<Tvec, 2>>> vel_dust_face_zm;
241
242 // time derivative dust density
243 std::shared_ptr<shamrock::solvergraph::Field<Tscal>> dtrho_dust;
244 // time derivative dust momemtum
245 std::shared_ptr<shamrock::solvergraph::Field<Tvec>> dtrhov_dust;
246
247 struct Timings {
248 f64 interface = 0;
249 f64 neighbors = 0;
250 f64 io = 0;
251
253 void reset() { *this = {}; }
254 } timings_details;
255 };
256
257} // 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
shammodels::basegodunov::SolverStorage< Tvec, TgridVec, u64 >::dy_v_dust
std::shared_ptr< shamrock::solvergraph::Field< Tvec > > dy_v_dust
Definition SolverStorage.hpp:111
shammodels::basegodunov::SolverStorage< Tvec, TgridVec, u64 >::dz_v_dust
std::shared_ptr< shamrock::solvergraph::Field< Tvec > > dz_v_dust
Definition SolverStorage.hpp:113
shammodels::basegodunov::SolverStorage< Tvec, TgridVec, u64 >::grad_rho_dust
std::shared_ptr< shamrock::solvergraph::Field< Tvec > > grad_rho_dust
Definition SolverStorage.hpp:107
shammodels::basegodunov::SolverStorage< Tvec, TgridVec, u64 >::dx_v_dust
std::shared_ptr< shamrock::solvergraph::Field< Tvec > > dx_v_dust
Definition SolverStorage.hpp:109
shamrock::solvergraph::SolverGraph
A graph container for managing solver nodes and edges with type-safe access.
Definition SolverGraph.hpp:110
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:253
Generated on for Shamrock by doxygen 1.15.0