diff options
Diffstat (limited to 'Source/Diagnostics/FlushFormats/FlushFormatCheckpoint.cpp')
| -rw-r--r-- | Source/Diagnostics/FlushFormats/FlushFormatCheckpoint.cpp | 17 |
1 files changed, 14 insertions, 3 deletions
diff --git a/Source/Diagnostics/FlushFormats/FlushFormatCheckpoint.cpp b/Source/Diagnostics/FlushFormats/FlushFormatCheckpoint.cpp index 890d4ceec..8e5122219 100644 --- a/Source/Diagnostics/FlushFormats/FlushFormatCheckpoint.cpp +++ b/Source/Diagnostics/FlushFormats/FlushFormatCheckpoint.cpp @@ -1,6 +1,9 @@ #include "FlushFormatCheckpoint.H" #include "BoundaryConditions/PML.H" +#if (defined WARPX_DIM_RZ) && (defined WARPX_USE_PSATD) +# include "BoundaryConditions/PML_RZ.H" +#endif #include "Diagnostics/ParticleDiag/ParticleDiag.H" #include "Particles/WarpXParticleContainer.H" #include "Utils/WarpXProfilerWrapper.H" @@ -138,9 +141,17 @@ FlushFormatCheckpoint::WriteToFile ( } } - if (warpx.DoPML() && warpx.GetPML(lev)) { - warpx.GetPML(lev)->CheckPoint( - amrex::MultiFabFileFullPrefix(lev, checkpointname, default_level_prefix, "pml")); + if (warpx.DoPML()) { + if (warpx.GetPML(lev)) { + warpx.GetPML(lev)->CheckPoint( + amrex::MultiFabFileFullPrefix(lev, checkpointname, default_level_prefix, "pml")); + } +#if (defined WARPX_DIM_RZ) && (defined WARPX_USE_PSATD) + if (warpx.GetPML_RZ(lev)) { + warpx.GetPML_RZ(lev)->CheckPoint( + amrex::MultiFabFileFullPrefix(lev, checkpointname, default_level_prefix, "pml_rz")); + } +#endif } } |