|
24 | 24 | #include "shammodels/ramses/SolverConfig.hpp" |
25 | 25 | #include "shammodels/ramses/modules/AMRGridRefinementHandler.hpp" |
26 | 26 | #include "shammodels/ramses/modules/BlockNeighToCellNeigh.hpp" |
| 27 | +#include "shammodels/ramses/modules/ComputeAMRLevel.hpp" |
27 | 28 | #include "shammodels/ramses/modules/ComputeCFL.hpp" |
28 | 29 | #include "shammodels/ramses/modules/ComputeCellAABB.hpp" |
29 | 30 | #include "shammodels/ramses/modules/ComputeLevel0CellSize.hpp" |
@@ -412,6 +413,9 @@ void shammodels::basegodunov::Solver<Tvec, TgridVec>::init_solver_graph() { |
412 | 413 | // get blocks at level0 sizes for all patches |
413 | 414 | storage.level0_size = std::make_shared<shamrock::solvergraph::ScalarsEdge<TgridVec>>( |
414 | 415 | "level0_amr", "level0_amr"); |
| 416 | + // get blocks amr leves |
| 417 | + using TgridUint = typename std::make_unsigned<shambase::VecComponent<TgridVec>>::type; |
| 418 | + storage.amr_block_levels = std::make_shared<shamrock::solvergraph::Field<TgridUint>>(1, "", ""); |
415 | 419 |
|
416 | 420 | storage.grad_rho = std::make_shared<shamrock::solvergraph::Field<Tvec>>( |
417 | 421 | AMRBlock::block_size, "grad_rho", "\\nabla \\rho"); |
@@ -921,6 +925,18 @@ void shammodels::basegodunov::Solver<Tvec, TgridVec>::init_solver_graph() { |
921 | 925 | std::make_shared<decltype(node_level0_sizes)>(std::move(node_level0_sizes))); |
922 | 926 | } |
923 | 927 |
|
| 928 | + { // compute block amr level in patch |
| 929 | + modules::ComputeAMRLevel<TgridVec> node_amr_level{}; |
| 930 | + node_amr_level.set_edges( |
| 931 | + storage.block_counts, |
| 932 | + storage.level0_size, |
| 933 | + storage.refs_block_min, |
| 934 | + storage.refs_block_max, |
| 935 | + storage.amr_block_levels); |
| 936 | + solver_sequence.push_back( |
| 937 | + std::make_shared<decltype(node_amr_level)>(std::move(node_amr_level))); |
| 938 | + } |
| 939 | + |
924 | 940 | if (solver_config.should_compute_rho_mean()) { |
925 | 941 | modules::NodeComputeMass<Tvec, TgridVec> node{AMRBlock::block_size}; |
926 | 942 | node.set_edges( |
|
0 commit comments