Merge branch 'main' into simple-advection

Author: LasNikas

src/io/write_vtk.jl

@@ -269,7 +269,8 @@ end
 function write2vtk!(vtk, v, u, t, system::FluidSystem; write_meta_data=true)
     vtk["velocity"] = [current_velocity(v, system, particle)
                        for particle in active_particles(system)]
-    vtk["density"] = current_density(v, system)
+    vtk["density"] = [current_density(v, system, particle)
+                      for particle in active_particles(system)]
     # Indexing the pressure is a workaround for slicing issue (see https://github.com/JuliaSIMD/StrideArrays.jl/issues/88)
     vtk["pressure"] = [current_pressure(v, system, particle)
                        for particle in active_particles(system)]
@@ -373,13 +374,17 @@ end
 function write2vtk!(vtk, v, u, t, system::TotalLagrangianSPHSystem; write_meta_data=true)
     n_fixed_particles = nparticles(system) - n_moving_particles(system)
 
-    vtk["velocity"] = hcat(view(v, 1:ndims(system), :),
-                           zeros(ndims(system), n_fixed_particles))
+    vtk["velocity"] = [current_velocity(v, system, particle)
+                       for particle in active_particles(system)]
     vtk["jacobian"] = [det(deformation_gradient(system, particle))
                        for particle in eachparticle(system)]
 
     vtk["von_mises_stress"] = von_mises_stress(system)
 
+    vtk["displacement"] = [current_coords(system, particle) -
+                           initial_coords(system, particle)
+                           for particle in eachparticle(system)]
+
     sigma = cauchy_stress(system)
     vtk["sigma_11"] = sigma[1, 1, :]
     vtk["sigma_22"] = sigma[2, 2, :]