Merge pull request #71 from oriolcg/pvd_functions

createpvd and savepvd functions

Author: amartinhuertas

NEWS.md

@@ -10,6 +10,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
  - Support for periodic boundary conditions for `CartesianDiscreteModel`. Since PR [#79](https://github.com/gridap/GridapDistributed.jl/pull/79)
  - Skeleton documentation and some content. Since PR [#77](https://github.com/gridap/GridapDistributed.jl/pull/77)
  - Added `interpolate_everywhere` and `interpolate_dirichlet` functions. Since PR [#74](https://github.com/gridap/GridapDistributed.jl/pull/74)
+ - Added `createpvd` and `savepvd` functions to save collections of VTK files. Since PR [#71](https://github.com/gridap/GridapDistributed.jl/pull/71)
 
 ### Fixed
 - Visualization of functions and numbers.  Since PR [#78](https://github.com/gridap/GridapDistributed.jl/pull/78)

src/Visualization.jl

@@ -156,3 +156,36 @@ function Visualization.create_vtk_file(
       celldata=c,nodaldata=n)
   end
 end
+
+struct DistributedPvd{T<:AbstractPData}
+  pvds::T
+end
+
+function Visualization.createpvd(parts::AbstractPData,args...;kwargs...)
+  pvds = map_main(parts) do part
+    paraview_collection(args...;kwargs...)
+  end
+  DistributedPvd(pvds)
+end
+
+function Visualization.createpvd(f,parts::AbstractPData,args...;kwargs...)
+  pvd = createpvd(parts,args...;kwargs...)
+  try
+    f(pvd)
+  finally
+    savepvd(pvd)
+  end
+end
+
+function Visualization.savepvd(pvd::DistributedPvd)
+  map_main(pvd.pvds) do pvd
+    vtk_save(pvd)
+  end
+end
+
+function Base.setindex!(pvd::DistributedPvd,pvtk::AbstractPData,time::Real)
+  map_parts(vtk_save,pvtk)
+  map_main(pvtk,pvd.pvds) do pvtk,pvd
+    pvd[time] = pvtk
+  end
+end

test/CellDataTests.jl

@@ -24,6 +24,12 @@ function main(parts)
   writevtk(Ω,joinpath(output,"Ω"),cellfields=["f"=>f,"u"=>u])
   writevtk(Γ,joinpath(output,"Γ"),cellfields=["f"=>f,"g"=>g,"u"=>u])
 
+  createpvd(parts,joinpath(output,"Ω_pvd")) do pvd
+    pvd[0.1] = createvtk(Ω,joinpath(output,"Ω_1"),cellfields=["f"=>f])
+    pvd[0.2] = createvtk(Ω,joinpath(output,"Ω_2"),cellfields=["f"=>f])
+  end
+  @test isfile(joinpath(output,"Ω_pvd")*".pvd")
+
   x_Γ = get_cell_points(Γ)
   @test isa(f(x_Γ),AbstractPData)