Use the right array data structure for the mpi rank

trixi-framework · Oct 28, 2024 · d1f6d44 · d1f6d44
1 parent 9559793
commit d1f6d44
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/src/solvers/dg.jl b/src/solvers/dg.jl
@@ -15,7 +15,7 @@ end
 # Function to define "element variables" for the SaveSolutionCallback. It does
 # nothing by default, but can be specialized for certain mesh types. For instance,
 # parallel meshes output the mpi rank as an "element variable".
-function get_element_variables!(element_variables, mesh)
+function get_element_variables!(element_variables, mesh, dg, cache)
     nothing
 end
 
@@ -438,7 +438,7 @@ Base.summary(io::IO, dg::DG) = print(io, "DG(" * summary(dg.basis) * ")")
 function get_element_variables!(element_variables, u, mesh, equations, dg::DG, cache)
     get_element_variables!(element_variables, u, mesh, equations, dg.volume_integral,
                            dg, cache)
-    get_element_variables!(element_variables, mesh)
+    get_element_variables!(element_variables, mesh, dg, cache)
 end
 
 function get_node_variables!(node_variables, mesh, equations, dg::DG, cache)

diff --git a/src/solvers/dgsem_tree/dg_parallel.jl b/src/solvers/dgsem_tree/dg_parallel.jl
@@ -8,8 +8,10 @@
 # Function to output the mpi rank for visualization
 function get_element_variables!(element_variables,
                                 mesh::Union{ParallelTreeMesh, ParallelP4estMesh,
-                                            ParallelT8codeMesh})
-    element_variables[:mpi_rank] = mpi_rank()
+                                            ParallelT8codeMesh},
+                                dg, cache)
+    element_variables[:mpi_rank] = fill(mpi_rank(),
+                                        SVector{nelements(mesh, dg, cache), real(dg)})
     return nothing
 end