hdf5 halo: store in kji order only in 3d

igfuw · Feb 10, 2023 · 7e8f38e · 7e8f38e
1 parent 60ce20a
commit 7e8f38e
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/libmpdata++/output/hdf5.hpp b/libmpdata++/output/hdf5.hpp
@@ -411,11 +411,15 @@ namespace libmpdataxx
         space.selectHyperslab(H5S_SELECT_SET, shape_h.data(), offst_h.data());
         sspace_mem_h.selectHyperslab(H5S_SELECT_SET, shape_h.data(), offst_mem_h.data());
 
-        // convert from kij to kji storage order
-        typename solver_t::arr_t kji_arr(shape_h);
-        kji_arr = arr;
-
-        aux.write(kji_arr.data(), flttype_solver, sspace_mem_h, space, dxpl_id);
+        // in 3D convert from kij to kji storage order
+        if(parent_t::n_dims == 3)
+        {
+          typename solver_t::arr_t kji_arr(shape_h);
+          kji_arr = arr;
+          aux.write(kji_arr.data(), flttype_solver, sspace_mem_h, space, dxpl_id);
+        }
+        else
+          aux.write(arr.data(), flttype_solver, sspace_mem_h, space, dxpl_id);
       }
 
       void record_scalar_hlpr(const std::string &name, const std::string &group_name, typename solver_t::real_t data, H5::H5File hdf)