Fortran: Add amrex_write_hdf5plotfile

Src/F_Interfaces/Base/AMReX_plotfile_fi.cpp

@@ -1,6 +1,11 @@
+#include <AMReX_Config.H>
 #include <AMReX.H>
 #include <AMReX_PlotFileUtil.H>
 
+#ifdef AMREX_USE_HDF5
+#include <AMReX_PlotFileUtilHDF5.H>
+#endif
+
 using namespace amrex;
 
 extern "C"
@@ -23,4 +28,25 @@ extern "C"
         amrex::WriteMultiLevelPlotfile(name, nlevs, mfarr, varnamearr, geomarr,
                                        time, lsarr, rrarr);
     }
+
+#ifdef AMREX_USE_HDF5
+    void amrex_fi_write_hdf5plotfile (const char* name, int nlevs, const MultiFab* mf[],
+                                      const char* varname[], const Geometry* geom[],
+                                      Real time, const int level_steps[], const int ref_ratio[])
+    {
+        Vector<const MultiFab*> mfarr {mf, mf+nlevs};
+        Vector<std::string> varnamearr {varname, varname+mf[0]->nComp()};
+        Vector<Geometry> geomarr;
+        for (int lev = 0; lev < nlevs; ++lev) {
+            geomarr.emplace_back(*geom[lev]);
+        }
+        Vector<int> lsarr {level_steps, level_steps+nlevs};
+        Vector<IntVect> rrarr;
+        for (int lev = 0; lev < nlevs-1; ++lev) {
+            rrarr.emplace_back(AMREX_D_DECL(ref_ratio[lev],ref_ratio[lev],ref_ratio[lev]));
+        }
+        amrex::WriteMultiLevelPlotfileHDF5(name, nlevs, mfarr, varnamearr, geomarr,
+                                           time, lsarr, rrarr);
+    }
+#endif
 }

Src/F_Interfaces/Base/AMReX_plotfile_mod.F90

@@ -1,3 +1,4 @@
+#include <AMReX_Config.H>
 
 module amrex_plotfile_module
 
@@ -12,6 +13,9 @@ module amrex_plotfile_module
   private
 
   public :: amrex_write_plotfile
+#ifdef AMREX_USE_HDF5
+  public :: amrex_write_hdf5plotfile
+#endif
 
   interface
      subroutine amrex_fi_write_plotfile (name, nlevs, mf, varname, geom, t, steps, rr) bind(c)
@@ -23,6 +27,17 @@ subroutine amrex_fi_write_plotfile (name, nlevs, mf, varname, geom, t, steps, rr
        real(amrex_real), value :: t
        integer(c_int), intent(in) :: steps(*), rr(*)
      end subroutine amrex_fi_write_plotfile
+#ifdef AMREX_USE_HDF5
+     subroutine amrex_fi_write_hdf5plotfile (name, nlevs, mf, varname, geom, t, steps, rr) bind(c)
+       import
+       implicit none
+       character(kind=c_char), intent(in) :: name(*)
+       integer(c_int), value :: nlevs
+       type(c_ptr), intent(in) :: mf(*), varname(*), geom(*)
+       real(amrex_real), value :: t
+       integer(c_int), intent(in) :: steps(*), rr(*)
+     end subroutine amrex_fi_write_hdf5plotfile
+#endif
   end interface
 
 contains
@@ -52,4 +67,31 @@ subroutine amrex_write_plotfile (name, nlevs, mf, varname, geom, t, steps, rr)
          t, steps, rr)
   end subroutine amrex_write_plotfile
 
+#ifdef AMREX_USE_HDF5
+  subroutine amrex_write_hdf5plotfile (name, nlevs, mf, varname, geom, t, steps, rr)
+    character(len=*), intent(in) :: name
+    integer, intent(in) :: nlevs, steps(*), rr(*)
+    type(amrex_multifab), intent(in) :: mf(*)
+    type(amrex_string), intent(in), target :: varname(*)
+    type(amrex_geometry), intent(in) :: geom(*)
+    real(amrex_real), intent(in) :: t
+
+    integer :: lev, icomp
+    type(c_ptr) :: mf_c(nlevs), geom_c(nlevs)
+    type(c_ptr) :: varname_c(mf(1)%ncomp())
+
+    do lev = 1, nlevs
+       mf_c(lev) = mf(lev)%p
+       geom_c(lev) = geom(lev)%p
+    end do
+
+    do icomp = 1, size(varname_c)
+       varname_c(icomp) = c_loc(varname(icomp)%data(1))
+    end do
+
+    call amrex_fi_write_hdf5plotfile(amrex_string_f_to_c(name), nlevs, mf_c, varname_c, geom_c, &
+         t, steps, rr)
+  end subroutine amrex_write_hdf5plotfile
+#endif
+
 end module amrex_plotfile_module