REMORA_NCPlotFile.cpp

Go to the documentation of this file.

#include <iomanip>
#include <iostream>
#include <map>
#include <string>
#include <vector>
#include <ctime>
 
#ifdef _OPENMP
#include <omp.h>
#endif
 
#include <AMReX_Utility.H>
#include <AMReX_buildInfo.H>
#include <AMReX_ParmParse.H>
 
#include "REMORA.H"
#include "REMORA_NCInterface.H"
#include "REMORA_NCPlotFile.H"
#include "REMORA_IndexDefines.H"
 
using namespace amrex;
 
namespace {
/**
 * \brief Was this 2D field named in remora.plot_vars_2d?
 *
 * Mirrors the name test the AMReX plotfile writer does when it walks plot_var_names_2d,
 * so both writers answer to the same input key.
 */
bool
plot_2d_var_requested (const amrex::Vector<std::string>& names, const std::string& nm)
{
    for (int i = 0; i < names.size(); ++i) {
        if (names[i] == nm) { return true; }
    }
    return false;
}
 
/**
 * \brief Accumulates each rank's hyperslabs and writes them with collective I/O.
 *
 * PnetCDF's collective APIs have to be called by every rank that opened the file,
 * the same number of times and in the same order. REMORA's write loops are MFIter
 * loops, so the number of hyperslabs is per-rank -- a plain put_all() inside those
 * loops would mismatch and hang. ncmpi_put_varn_*_all() takes all of a rank's
 * subarrays for one variable in a single call and tolerates counts that differ
 * between ranks, including zero, which is exactly the shape we need.
 *
 * So the loops hand their hyperslabs here instead of writing, and flush() issues
 * one collective call per variable at the end. Data is copied into m_entries on
 * add(), so callers may reuse or destroy their staging buffer immediately -- unlike
 * the nonblocking iput() path, where PnetCDF aliases buffers larger than 4 KB and
 * only reads them at wait_all().
 */
class VarnCollector
{
public:
    //! Stage one hyperslab of `varid`. `start`/`count` are in NetCDF dimension
    //! order and `dptr` holds product(count) values.
    void add (int varid,
              const std::vector<MPI_Offset>& start,
              const std::vector<MPI_Offset>& count,
              const amrex::Real* dptr)
    {
        AMREX_ASSERT(start.size() == count.size());
        MPI_Offset nelems = 1;
        for (auto c : count) { nelems *= c; }
 
        Entry& e = m_entries[varid];
        e.starts.push_back(start);
        e.counts.push_back(count);
        e.data.insert(e.data.end(), dptr, dptr + nelems);
    }
 
    /**
     * \brief Write everything staged so far. Collective: all ranks must call it.
     *
     * Ranks own different boxes, so they stage different variables -- a rank with
     * no boxes stages nothing at all. The set of variables to write therefore has
     * to be agreed on before any collective call, which is what the reduction over
     * the touched-variable mask below does. Iterating that agreed set in ascending
     * varid order gives every rank the same call sequence.
     */
    void flush (const ncutils::NCFile& ncf)
    {
        const int nvars = ncf.num_variables();
        if (nvars <= 0) { return; }
 
        std::vector<int> touched(nvars, 0);
        for (const auto& kv : m_entries) {
            if (kv.first >= 0 && kv.first < nvars) { touched[kv.first] = 1; }
        }
#ifdef AMREX_USE_MPI
        MPI_Allreduce(MPI_IN_PLACE, touched.data(), nvars, MPI_INT, MPI_MAX,
                      amrex::ParallelContext::CommunicatorSub());
#endif
 
        for (int varid = 0; varid < nvars; ++varid) {
            if (!touched[varid]) { continue; }
 
            auto it = m_entries.find(varid);
            if (it == m_entries.end()) {
                // This rank has nothing for this variable, but the call is
                // collective, so it still has to take part with zero subarrays.
                ncutils::NCVar { ncf.ncid, varid }.put_varn_all(
                    0, nullptr, nullptr, static_cast<const amrex::Real*>(nullptr));
                continue;
            }
 
            Entry& e = it->second;
            const int num = static_cast<int>(e.starts.size());
 
            // ncmpi_put_varn_* wants arrays of pointers into the start/count rows.
            std::vector<MPI_Offset*> start_ptrs(num);
            std::vector<MPI_Offset*> count_ptrs(num);
            for (int i = 0; i < num; ++i) {
                start_ptrs[i] = e.starts[i].data();
                count_ptrs[i] = e.counts[i].data();
            }
 
            ncutils::NCVar { ncf.ncid, varid }.put_varn_all(
                num, start_ptrs.data(), count_ptrs.data(), e.data.data());
        }
 
        m_entries.clear();
    }
 
    /**
     * \brief Handle with the same put() shape as ncutils::NCVar, staging instead
     *        of writing, so the write loops read the same as they always did.
     */
    class StagedVar
    {
    public:
        StagedVar (VarnCollector& coll, int varid) : m_coll(coll), m_varid(varid) {}
 
        void put (const amrex::Real* dptr,
                  const std::vector<MPI_Offset>& start,
                  const std::vector<MPI_Offset>& count) const
        {
            m_coll.add(m_varid, start, count, dptr);
        }
 
    private:
        VarnCollector& m_coll;
        int m_varid;
    };
 
    //! Handle for staging writes to the named variable.
    StagedVar var (const ncutils::NCFile& ncf, const std::string& name)
    {
        return StagedVar(*this, ncf.var(name).varid);
    }
 
private:
    struct Entry {
        std::vector<std::vector<MPI_Offset>> starts;
        std::vector<std::vector<MPI_Offset>> counts;
        std::vector<amrex::Real> data;
    };
 
    //! Keyed by varid and held in a std::map so iteration is in varid order.
    std::map<int, Entry> m_entries;
};
} // namespace
 
/**
 * @param which_step   current step for output
 */
void REMORA::WriteNCPlotFile(int which_step, MultiFab const* plotMF) {
    AMREX_ASSERT(max_level == 0);
    // For right now we assume single level -- we will generalize this later to multilevel
    int lev = 0;
    int which_subdomain = 0;
    int which_step_in_chunk = -1;
 
    // Create filename
    std::string plt_string;
    std::string plotfilename;
    if (REMORA::write_history_file) {
        plotfilename = plot_file_name + "_his";
    } else {
        plotfilename = Concatenate(plot_file_name, which_step, file_min_digits);
    }
    // If chunking, concatenate with which file we're in
    if (REMORA::write_history_file and REMORA::chunk_history_file) {
        int which_chunk = history_count / REMORA::steps_per_history_file;
        plotfilename = Concatenate(plotfilename, which_chunk, file_min_digits);
        which_step_in_chunk = history_count - which_chunk * REMORA::steps_per_history_file;
    }
 
    // Set the full IO path for NetCDF output
    std::string FullPath = plotfilename;
    if (lev == 0) {
        const std::string &extension = amrex::Concatenate("_d", lev + 1, 2);
        FullPath += extension + ".nc";
    } else {
        const std::string &extension = amrex::Concatenate("_d", lev + 1 + which_subdomain, 2);
        FullPath += extension + ".nc";
    }
 
    //
    // Check if this file/directory already exists and if so,
    //       have the IOProcessor move the existing
    //       file/directory to filename.old
    //
    if ((!REMORA::write_history_file) || (which_step == 0) || (which_step_in_chunk == 0)) {
        if (amrex::ParallelDescriptor::IOProcessor()) {
            if (amrex::FileExists(FullPath)) {
                std::string newoldname(FullPath + ".old." + amrex::UniqueString());
                amrex::Print() << "WriteNCPlotFile:  " << FullPath << " exists.  Renaming to:  " << newoldname << std::endl;
                if (std::rename(FullPath.c_str(), newoldname.c_str())) {
                    amrex::Abort("WriteNCPlotfile:: std::rename failed");
                }
            }
        }
        ParallelDescriptor::Barrier();
    }
 
    bool is_history;
 
    if (REMORA::write_history_file) {
        is_history = true;
        bool write_header = !(amrex::FileExists(FullPath));
 
        auto ncf =
                write_header ?
                        ncutils::NCFile::create(FullPath, NC_CLOBBER|NC_64BIT_DATA, amrex::ParallelContext::CommunicatorSub(), MPI_INFO_NULL) :
                        ncutils::NCFile::open(FullPath, NC_WRITE, amrex::ParallelContext::CommunicatorSub(), MPI_INFO_NULL);
 
        amrex::Print() << "Writing into level " << lev << " NetCDF history file " << FullPath << std::endl;
 
        WriteNCPlotFile_which(lev, which_subdomain, plotMF, write_header, ncf, is_history);
 
    } else {
 
        is_history = false;
        bool write_header = true;
 
        // Open new netcdf file to write data
        auto ncf = ncutils::NCFile::create(FullPath, NC_CLOBBER|NC_64BIT_DATA, amrex::ParallelContext::CommunicatorSub(), MPI_INFO_NULL);
        amrex::Print() << "Writing level " << lev << " NetCDF plot file " << FullPath << std::endl;
 
        WriteNCPlotFile_which(lev, which_subdomain, plotMF, write_header, ncf, is_history);
    }
}
 
/**
 * @param lev               level of data to output
 * @param which_subdomain   index of subdomain if lev != 0
 * @param write_header      whether to write a header
 * @param ncf               netcdf file object
 * @param is_history        whether the file being written is a history file
 */
void REMORA::WriteNCPlotFile_which(int lev, int which_subdomain, MultiFab const* plotMF,
                                   bool write_header, ncutils::NCFile &ncf, bool is_history)
{
    // Number of cells in this "domain" at this level
    std::vector<int> n_cells;
 
    // We only do single-level writes when using NetCDF format
    int flev = 1; //max_level;
 
    Box subdomain;
    if (lev == 0) {
        subdomain = geom[lev].Domain();
    } else {
        subdomain = boxes_at_level[lev][which_subdomain];
    }
 
    int nx = subdomain.length(0);
    int ny = subdomain.length(1);
    int nz = subdomain.length(2);
 
    n_cells.push_back(nx);
    n_cells.push_back(ny);
    n_cells.push_back(nz);
 
    const std::string nt_name = "ocean_time";
    const std::string ndim_name = "num_geo_dimensions";
 
    const std::string flev_name = "FINEST_LEVEL";
 
    const std::string nx_name = "NX";
    const std::string ny_name = "NY";
    const std::string nz_name = "NZ";
 
    const std::string nx_r_name = "xi_rho";
    const std::string ny_r_name = "eta_rho";
    const std::string nz_r_name = "s_rho";
 
    const std::string nx_u_name = "xi_u";
    const std::string ny_u_name = "eta_u";
 
    const std::string nx_v_name = "xi_v";
    const std::string ny_v_name = "eta_v";
 
    const std::string nx_p_name = "xi_psi";
    const std::string ny_p_name = "eta_psi";
    const std::string nz_w_name = "s_w";
 
    if (write_header) {
        ncf.enter_def_mode();
        ncf.put_attr("title", "REMORA data ");
        // The time dimension is unlimited so the record count reflects what was
        // actually written rather than an up-front estimate from max_step/plot_int.
        // Note PnetCDF does not prefill record variables, so every element of a
        // time-varying variable has to be written explicitly (see the loops below).
        ncf.def_dim(nt_name, NC_UNLIMITED);
        ncf.def_dim(ndim_name, AMREX_SPACEDIM);
 
        ncf.def_dim(nx_r_name, nx + 2);
        ncf.def_dim(ny_r_name, ny + 2);
        ncf.def_dim(nz_r_name, nz);
 
        ncf.def_dim(nx_u_name, nx + 1);
        ncf.def_dim(ny_u_name, ny + 2);
 
        ncf.def_dim(nx_v_name, nx + 2);
        ncf.def_dim(ny_v_name, ny + 1);
 
        ncf.def_dim(nx_p_name, nx + 1);
        ncf.def_dim(ny_p_name, ny + 1);
 
        ncf.def_dim(nz_w_name, nz + 1);
 
        ncf.def_dim(flev_name, flev);
 
        ncf.def_dim(nx_name, n_cells[0]);
        ncf.def_dim(ny_name, n_cells[1]);
        ncf.def_dim(nz_name, n_cells[2]);
 
        ncf.def_var("probLo", ncutils::NCDType::Real, { ndim_name });
        ncf.var("probLo").put_attr("long_name","Low side of problem domain in internal AMReX grid");
        ncf.var("probLo").put_attr("units","meter");
        ncf.def_var("probHi", ncutils::NCDType::Real, { ndim_name });
        ncf.var("probHi").put_attr("long_name","High side of problem domain in internal AMReX grid");
        ncf.var("probHi").put_attr("units","meter");
 
        ncf.def_var("Geom.smallend", NC_INT, { flev_name, ndim_name });
        ncf.var("Geom.smallend").put_attr("long_name","Low side of problem domain in index space");
        ncf.def_var("Geom.bigend", NC_INT, { flev_name, ndim_name });
        ncf.var("Geom.bigend").put_attr("long_name","High side of problem domain in index space");
        ncf.def_var("CellSize", ncutils::NCDType::Real, { flev_name, ndim_name });
        ncf.var("CellSize").put_attr("long_name","Cell size on internal AMReX grid");
        ncf.var("CellSize").put_attr("units","meter");
 
        ncf.def_var("theta_s",ncutils::NCDType::Real,{});
        ncf.var("theta_s").put_attr("long_name","S-coordinate surface control parameter");
        ncf.def_var("theta_b",ncutils::NCDType::Real,{});
        ncf.var("theta_b").put_attr("long_name","S-coordinate bottom control parameter");
        ncf.def_var("hc",ncutils::NCDType::Real,{});
        ncf.var("hc").put_attr("long_name","S-coordinate parameter, critical depth");
        ncf.var("hc").put_attr("units","meter");
 
        ncf.def_var("grid",NC_INT, {});
        ncf.var("grid").put_attr("cf_role","grid_topology");
        ncf.var("grid").put_attr("topology_dimension",std::vector({2}));
        ncf.var("grid").put_attr("node_dimensions", "xi_psi eta_psi");
        ncf.var("grid").put_attr("face_dimensions", "xi_rho: xi_psi (padding: both) eta_rho: eta_psi (padding: both)");
        ncf.var("grid").put_attr("edge1_dimensions", "xi_u: xi_psi eta_u: eta_psi (padding: both)");
        ncf.var("grid").put_attr("edge2_dimensions", "xi_v: xi_psi (padding: both) eta_v: eta_psi");
        ncf.var("grid").put_attr("node_coordinates", "x_psi y_psi");
        ncf.var("grid").put_attr("face_coordinates", "x_rho y_rho");
        ncf.var("grid").put_attr("edge1_coordinates", "x_u y_u");
        ncf.var("grid").put_attr("edge2_coordinates", "x_v y_v");
        ncf.var("grid").put_attr("vertical_dimensions", "s_rho: s_w (padding: none)");
 
        ncf.def_var("s_rho",ncutils::NCDType::Real, {nz_r_name});
        ncf.var("s_rho").put_attr("long_name","S-coordinate at RHO-points");
        ncf.var("s_rho").put_attr("field","s_rho, scalar");
 
        ncf.def_var("s_w",ncutils::NCDType::Real, {nz_w_name});
        ncf.var("s_w").put_attr("long_name","S-coordinate at W-points");
        ncf.var("s_w").put_attr("field","s_w, scalar");
 
        ncf.def_var("pm",ncutils::NCDType::Real, {ny_r_name, nx_r_name});
        ncf.var("pm").put_attr("long_name","curvilinear coordinate metric in XI");
        ncf.var("pm").put_attr("units","meter-1");
        ncf.var("pm").put_attr("grid","grid");
        ncf.var("pm").put_attr("location","face");
        ncf.var("pm").put_attr("coordinates","x_rho y_rho");
        ncf.var("pm").put_attr("field","pm, scalar");
 
        ncf.def_var("pn",ncutils::NCDType::Real, {ny_r_name, nx_r_name});
        ncf.var("pn").put_attr("long_name","curvilinear coordinate metric in ETA");
        ncf.var("pn").put_attr("units","meter-1");
        ncf.var("pn").put_attr("grid","grid");
        ncf.var("pn").put_attr("location","face");
        ncf.var("pn").put_attr("coordinates","x_rho y_rho");
        ncf.var("pn").put_attr("field","pn, scalar");
 
        ncf.def_var("f",ncutils::NCDType::Real, {ny_r_name, nx_r_name});
        ncf.var("f").put_attr("long_name","Coriolis parameter at RHO-points");
        ncf.var("f").put_attr("units","second-1");
        ncf.var("f").put_attr("grid","grid");
        ncf.var("f").put_attr("location","face");
        ncf.var("f").put_attr("coordinates","x_rho y_rho");
        ncf.var("f").put_attr("field","coriolis, scalar");
 
        ncf.def_var("x_rho",ncutils::NCDType::Real, {ny_r_name, nx_r_name});
        ncf.var("x_rho").put_attr("long_name","x-locations of RHO-points");
        ncf.var("x_rho").put_attr("units","meter");
        ncf.var("x_rho").put_attr("field","x_rho, scalar");
 
        ncf.def_var("y_rho",ncutils::NCDType::Real, {ny_r_name, nx_r_name});
        ncf.var("y_rho").put_attr("long_name","y-locations of RHO-points");
        ncf.var("y_rho").put_attr("units","meter");
        ncf.var("y_rho").put_attr("field","y_rho, scalar");
 
        ncf.def_var("x_u",ncutils::NCDType::Real, {ny_u_name, nx_u_name});
        ncf.var("x_u").put_attr("long_name","x-locations of U-points");
        ncf.var("x_u").put_attr("units","meter");
        ncf.var("x_u").put_attr("field","x_u, scalar");
 
        ncf.def_var("y_u",ncutils::NCDType::Real, {ny_u_name, nx_u_name});
        ncf.var("y_u").put_attr("long_name","y-locations of U-points");
        ncf.var("y_u").put_attr("units","meter");
        ncf.var("y_u").put_attr("field","y_u, scalar");
 
        ncf.def_var("x_v",ncutils::NCDType::Real, {ny_v_name, nx_v_name});
        ncf.var("x_v").put_attr("long_name","x-locations of V-points");
        ncf.var("x_v").put_attr("units","meter");
        ncf.var("x_v").put_attr("field","x_v, scalar");
 
        ncf.def_var("y_v",ncutils::NCDType::Real, {ny_v_name, nx_v_name});
        ncf.var("y_v").put_attr("long_name","y-locations of V-points");
        ncf.var("y_v").put_attr("units","meter");
        ncf.var("y_v").put_attr("field","y_v, scalar");
 
        ncf.def_var("x_psi",ncutils::NCDType::Real, {ny_p_name, nx_p_name});
        ncf.var("x_psi").put_attr("long_name","x-locations of PSI-points");
        ncf.var("x_psi").put_attr("units","meter");
        ncf.var("x_psi").put_attr("field","x_psi, scalar");
 
        ncf.def_var("y_psi",ncutils::NCDType::Real, {ny_p_name, nx_p_name});
        ncf.var("y_psi").put_attr("long_name","y-locations of PSI-points");
        ncf.var("y_psi").put_attr("units","meter");
        ncf.var("y_psi").put_attr("field","y_psi, scalar");
 
        ncf.def_var("ocean_time", ncutils::NCDType::Real, { nt_name });
        ncf.var("ocean_time").put_attr("long_name","time since initialization");
        ncf.var("ocean_time").put_attr("units","seconds since 0001-01-01 00:00:00");
        ncf.var("ocean_time").put_attr("field","time, scalar, series");
 
        ncf.def_var("Cs_r", ncutils::NCDType::Real, {nz_r_name});
        ncf.var("Cs_r").put_attr("long_name", "S-coordinate stretching curves at RHO points");
        ncf.var("Cs_r").put_attr("valid_min",std::vector({-one}));
        ncf.var("Cs_r").put_attr("valid_max",std::vector({zero}));
        ncf.var("Cs_r").put_attr("field","Cs_r, scalar");
 
        ncf.def_var("Cs_w", ncutils::NCDType::Real, {nz_w_name});
        ncf.var("Cs_w").put_attr("long_name", "S-coordinate stretching curves at W points");
        ncf.var("Cs_w").put_attr("valid_min",std::vector({-one}));
        ncf.var("Cs_w").put_attr("valid_max",std::vector({zero}));
        ncf.var("Cs_w").put_attr("field","Cs_w, scalar");
 
        ncf.def_var("h", ncutils::NCDType::Real, { ny_r_name, nx_r_name });
        ncf.var("h").put_attr("long_name","bathymetry at RHO-points");
        ncf.var("h").put_attr("units","meter");
        ncf.var("h").put_attr("grid","grid");
        ncf.var("h").put_attr("location","face");
        ncf.var("h").put_attr("coordinates","x_rho y_rho");
        ncf.var("h").put_attr("field","bath, scalar");
 
        ncf.def_var_fill("zeta", ncutils::NCDType::Real, { nt_name, ny_r_name, nx_r_name }, &netcdf_fill_value);
        ncf.var("zeta").put_attr("long_name","free-surface");
        ncf.var("zeta").put_attr("units","meter");
        ncf.var("zeta").put_attr("time","ocean_time");
        ncf.var("zeta").put_attr("grid","grid");
        ncf.var("zeta").put_attr("location","face");
        ncf.var("zeta").put_attr("coordinates","x_rho y_rho ocean_time");
        ncf.var("zeta").put_attr("field","free-surface, scalar, series");
 
        for (int n = 0; n < ncons; ++n) {
            int comp = -1;
            for (int i = 0; i < plot_var_names_3d.size(); i++) {
                if (plot_var_names_3d[i] == cons_names[n]) comp = i;
            }
            if (comp >= 0) {
                const std::string& nm = cons_names[n];
                ncf.def_var_fill(nm, ncutils::NCDType::Real, { nt_name, nz_r_name, ny_r_name, nx_r_name }, &netcdf_fill_value);
                if (n == Temp_comp) {
                    ncf.var(nm).put_attr("long_name", "potential temperature");
                    ncf.var(nm).put_attr("units", "Celsius");
                    ncf.var(nm).put_attr("field", "temperature, scalar, series");
                } else if (n == Salt_comp) {
                    ncf.var(nm).put_attr("long_name", "salinity");
                    ncf.var(nm).put_attr("field", "salinity, scalar, series");
                } else {
                    ncf.var(nm).put_attr("long_name", nm);
                    ncf.var(nm).put_attr("field", nm + ", scalar, series");
                }
                ncf.var(nm).put_attr("time", "ocean_time");
                ncf.var(nm).put_attr("grid", "grid");
                ncf.var(nm).put_attr("location", "face");
                ncf.var(nm).put_attr("coordinates", "x_rho y_rho s_rho ocean_time");
            }
        }
 
        {
            int comp = -1;
            for (int i = 0; i < plot_var_names_3d.size(); i++) {
                if (plot_var_names_3d[i] == "vorticity") comp = i;
            }
            if (comp >= 0) {
               ncf.def_var_fill("vorticity", ncutils::NCDType::Real, { nt_name, nz_r_name, ny_r_name, nx_r_name }, &netcdf_fill_value);
               ncf.var("vorticity").put_attr("long_name","vorticity");
               ncf.var("vorticity").put_attr("time","ocean_time");
               ncf.var("vorticity").put_attr("grid","grid");
               ncf.var("vorticity").put_attr("location","face");
               ncf.var("vorticity").put_attr("coordinates","x_rho y_rho s_rho ocean_time");
               ncf.var("vorticity").put_attr("field","vorticity, scalar, series");
            }
        } // end vorticity
 
        // Output 2D horizontal mixing coefficients if using scaled_to_grid option
        if (solverChoice.horiz_mixing_type == HorizMixingType::scaled_to_grid) {
            ncf.def_var_fill("visc2", ncutils::NCDType::Real, { ny_r_name, nx_r_name }, &netcdf_fill_value);
            ncf.var("visc2").put_attr("long_name","horizontal harmonic viscosity coefficient at RHO-points");
            ncf.var("visc2").put_attr("units","meter2 second-1");
            ncf.var("visc2").put_attr("grid","grid");
            ncf.var("visc2").put_attr("location","face");
            ncf.var("visc2").put_attr("coordinates","x_rho y_rho");
            ncf.var("visc2").put_attr("field","visc2, scalar");
 
            for (int n = 0; n < ncons; ++n) {
                const std::string nm = std::string("diff2_") + cons_names[n];
                ncf.def_var_fill(nm, ncutils::NCDType::Real, { ny_r_name, nx_r_name }, &netcdf_fill_value);
                ncf.var(nm).put_attr("long_name", std::string("horizontal harmonic diffusivity coefficient for ") + cons_names[n] + " at RHO-points");
                ncf.var(nm).put_attr("units","meter2 second-1");
                ncf.var(nm).put_attr("grid","grid");
                ncf.var(nm).put_attr("location","face");
                ncf.var(nm).put_attr("coordinates","x_rho y_rho");
                ncf.var(nm).put_attr("field", nm + ", scalar");
            }
        }
 
        ncf.def_var_fill("u", ncutils::NCDType::Real, { nt_name, nz_r_name, ny_u_name, nx_u_name }, &netcdf_fill_value);
        ncf.var("u").put_attr("long_name","u-momentum component");
        ncf.var("u").put_attr("units","meter second-1");
        ncf.var("u").put_attr("time","ocean_time");
        ncf.var("u").put_attr("grid","grid");
        ncf.var("u").put_attr("location","edge1");
        ncf.var("u").put_attr("coordinates","x_u y_u s_rho ocean_time");
        ncf.var("u").put_attr("field","u-velocity, scalar, series");
 
        ncf.def_var_fill("v", ncutils::NCDType::Real, { nt_name, nz_r_name, ny_v_name, nx_v_name }, &netcdf_fill_value);
        ncf.var("v").put_attr("long_name","v-momentum component");
        ncf.var("v").put_attr("units","meter second-1");
        ncf.var("v").put_attr("time","ocean_time");
        ncf.var("v").put_attr("grid","grid");
        ncf.var("v").put_attr("location","edge2");
        ncf.var("v").put_attr("coordinates","x_v y_v s_rho ocean_time");
        ncf.var("v").put_attr("field","v-velocity, scalar, series");
 
        ncf.def_var_fill("ubar", ncutils::NCDType::Real, { nt_name, ny_u_name, nx_u_name }, &netcdf_fill_value);
        ncf.var("ubar").put_attr("long_name","vertically integrated u-momentum component");
        ncf.var("ubar").put_attr("units","meter second-1");
        ncf.var("ubar").put_attr("time","ocean_time");
        ncf.var("ubar").put_attr("grid","grid");
        ncf.var("ubar").put_attr("location","edge1");
        ncf.var("ubar").put_attr("coordinates","x_u y_u ocean_time");
        ncf.var("ubar").put_attr("field","ubar-velocity, scalar, series");
 
        ncf.def_var_fill("vbar", ncutils::NCDType::Real, { nt_name, ny_v_name, nx_v_name }, &netcdf_fill_value);
        ncf.var("vbar").put_attr("long_name","vertically integrated v-momentum component");
        ncf.var("vbar").put_attr("units","meter second-1");
        ncf.var("vbar").put_attr("time","ocean_time");
        ncf.var("vbar").put_attr("grid","grid");
        ncf.var("vbar").put_attr("location","edge2");
        ncf.var("vbar").put_attr("coordinates","x_v y_v ocean_time");
        ncf.var("vbar").put_attr("field","vbar-velocity, scalar, series");
 
        ncf.def_var("sustr", ncutils::NCDType::Real, { nt_name, ny_u_name, nx_u_name });
        ncf.var("sustr").put_attr("long_name","surface u-momentum stress");
        ncf.var("sustr").put_attr("units","newton meter-2");
        ncf.var("sustr").put_attr("time","ocean_time");
        ncf.var("sustr").put_attr("grid","grid");
        ncf.var("sustr").put_attr("location","edge1");
        ncf.var("sustr").put_attr("coordinates","x_u y_u ocean_time");
        ncf.var("sustr").put_attr("field","surface u-momentum stress, scalar, series");
 
        ncf.def_var("svstr", ncutils::NCDType::Real, { nt_name, ny_v_name, nx_v_name });
        ncf.var("svstr").put_attr("long_name","surface v-momentum stress");
        ncf.var("svstr").put_attr("units","newton meter-2");
        ncf.var("svstr").put_attr("time","ocean_time");
        ncf.var("svstr").put_attr("grid","grid");
        ncf.var("svstr").put_attr("location","edge2");
        ncf.var("svstr").put_attr("coordinates","x_v y_v ocean_time");
        ncf.var("svstr").put_attr("field","surface v-momentum stress, scalar, series");
 
        ncf.def_var("mask_rho", ncutils::NCDType::Real, { nt_name, ny_r_name, nx_r_name });
        ncf.var("mask_rho").put_attr("long_name","mask on RHO-points");
        ncf.var("mask_rho").put_attr("time","ocean_time");
        ncf.var("mask_rho").put_attr("flag_values",std::vector({Real(0.0),Real(1.0)}));
        ncf.var("mask_rho").put_attr("flag_meanings","land water");
 
        ncf.def_var("mask_u", ncutils::NCDType::Real, { nt_name, ny_u_name, nx_u_name });
        ncf.var("mask_u").put_attr("long_name","mask on U-points");
        ncf.var("mask_u").put_attr("time","ocean_time");
        ncf.var("mask_u").put_attr("flag_values",std::vector({Real(0.0),Real(1.0)}));
        ncf.var("mask_u").put_attr("flag_meanings","land water");
 
        ncf.def_var("mask_v", ncutils::NCDType::Real, { nt_name, ny_v_name, nx_v_name });
        ncf.var("mask_v").put_attr("long_name","mask on V-points");
        ncf.var("mask_v").put_attr("time","ocean_time");
        ncf.var("mask_v").put_attr("flag_values",std::vector({Real(0.0),Real(1.0)}));
        ncf.var("mask_v").put_attr("flag_meanings","land water");
 
        // Surface tracer fluxes, one per cell-centered tracer, matching what the AMReX
        // plotfile writer already offers and requested through the same remora.plot_vars_2d
        // key. vec_stflux is ncons wide and unconditionally allocated, so any tracer can be
        // named. Components past temp and salt read zero unless something fills them --
        // Fennel's air-sea gas exchange acts on the tracer directly rather than through this
        // array -- so they show what the solver actually applied, which may be nothing.
        for (int n = 0; n < ncons; ++n) {
            const std::string nm = std::string("stflux_") + cons_names[n];
            if (!plot_2d_var_requested(plot_var_names_2d, nm)) { continue; }
            ncf.def_var_fill(nm, ncutils::NCDType::Real, { nt_name, ny_r_name, nx_r_name }, &netcdf_fill_value);
            ncf.var(nm).put_attr("long_name", std::string("surface flux of ") + cons_names[n]);
            // Kinematic, as ssflux is: the tracer's own units times meter second-1.
            ncf.var(nm).put_attr("units", (n == Temp_comp) ? "Celsius meter second-1"
                                                           : "meter second-1");
            ncf.var(nm).put_attr("time","ocean_time");
            ncf.var(nm).put_attr("grid","grid");
            ncf.var(nm).put_attr("location","face");
            ncf.var(nm).put_attr("coordinates","x_rho y_rho ocean_time");
            ncf.var(nm).put_attr("field", nm + ", scalar, series");
        }
 
        if (solverChoice.output_forcing) {
            // Surface air temperature (Celsius)
            ncf.def_var("Tair", ncutils::NCDType::Real, {nt_name, ny_r_name, nx_r_name });
            ncf.var("Tair").put_attr("long_name","surface air temperature");
            ncf.var("Tair").put_attr("units","Celsius");
            ncf.var("Tair").put_attr("time","ocean_time");
            ncf.var("Tair").put_attr("grid","grid");
            ncf.var("Tair").put_attr("location","face");
            ncf.var("Tair").put_attr("coordinates","x_rho y_rho ocean_time");
            ncf.var("Tair").put_attr("field","Tair, scalar, series");
 
            // Surface air pressure (Pascal)
            ncf.def_var("Pair", ncutils::NCDType::Real,{ nt_name, ny_r_name, nx_r_name });
            ncf.var("Pair").put_attr("long_name","surface air pressure");
            ncf.var("Pair").put_attr("units","Pascal");
            ncf.var("Pair").put_attr("time","ocean_time");
            ncf.var("Pair").put_attr("grid","grid");
            ncf.var("Pair").put_attr("location","face");
            ncf.var("Pair").put_attr("coordinates","x_rho y_rho ocean_time");
            ncf.var("Pair").put_attr("field","Pair, scalar, series");
 
            // Surface net heat flux (W/m2)
            ncf.def_var("qnet", ncutils::NCDType::Real, {nt_name, ny_r_name, nx_r_name });
            ncf.var("qnet").put_attr("long_name","surface net heat flux");
            ncf.var("qnet").put_attr("units","watt meter-2");
            ncf.var("qnet").put_attr("time","ocean_time");
            ncf.var("qnet").put_attr("grid","grid");
            ncf.var("qnet").put_attr("location","face");
            ncf.var("qnet").put_attr("coordinates","x_rho y_rho ocean_time");
            ncf.var("qnet").put_attr("field","surface heat flux, scalar, series");
 
            // Surface net salt flux (kinematic)
            ncf.def_var("ssflux", ncutils::NCDType::Real, {nt_name, ny_r_name, nx_r_name });
            ncf.var("ssflux").put_attr("long_name","kinematic surface net salt flux, SALT*(E-P)/rhow");
            ncf.var("ssflux").put_attr("units","meter second-1");
            ncf.var("ssflux").put_attr("time","ocean_time");
            ncf.var("ssflux").put_attr("grid","grid");
            ncf.var("ssflux").put_attr("location","face");
            ncf.var("ssflux").put_attr("coordinates","x_rho y_rho ocean_time");
            ncf.var("ssflux").put_attr("field","surface net salt flux, scalar, series");
 
            // Latent heat flux (W/m2)
            ncf.def_var("latent", ncutils::NCDType::Real, {nt_name, ny_r_name, nx_r_name });
            ncf.var("latent").put_attr("long_name","net latent heat flux");
            ncf.var("latent").put_attr("units","watt meter-2");
            ncf.var("latent").put_attr("time","ocean_time");
            ncf.var("latent").put_attr("grid","grid");
            ncf.var("latent").put_attr("location","face");
            ncf.var("latent").put_attr("coordinates","x_rho y_rho ocean_time");
            ncf.var("latent").put_attr("field","latent heat flux, scalar, series");
 
            // Sensible heat flux (W/m2)
            ncf.def_var("sensible", ncutils::NCDType::Real, {nt_name, ny_r_name, nx_r_name });
            ncf.var("sensible").put_attr("long_name","net sensible heat flux");
            ncf.var("sensible").put_attr("units","watt meter-2");
            ncf.var("sensible").put_attr("time","ocean_time");
            ncf.var("sensible").put_attr("grid","grid");
            ncf.var("sensible").put_attr("location","face");
            ncf.var("sensible").put_attr("coordinates","x_rho y_rho ocean_time");
            ncf.var("sensible").put_attr("field","sensible heat flux, scalar, series");
 
            // Longwave radiation (W/m2)
            ncf.def_var("lwrad", ncutils::NCDType::Real, {nt_name, ny_r_name, nx_r_name });
            ncf.var("lwrad").put_attr("long_name","net longwave radiation flux");
            ncf.var("lwrad").put_attr("units","watt meter-2");
            ncf.var("lwrad").put_attr("time","ocean_time");
            ncf.var("lwrad").put_attr("grid","grid");
            ncf.var("lwrad").put_attr("location","face");
            ncf.var("lwrad").put_attr("coordinates","x_rho y_rho ocean_time");
            ncf.var("lwrad").put_attr("field","longwave radiation, scalar, series");
 
            // Shortwave radiation (W/m2)
            ncf.def_var("swrad", ncutils::NCDType::Real, {nt_name, ny_r_name, nx_r_name });
            ncf.var("swrad").put_attr("long_name","solar shortwave radiation flux");
            ncf.var("swrad").put_attr("units","watt meter-2");
            ncf.var("swrad").put_attr("time","ocean_time");
            ncf.var("swrad").put_attr("grid","grid");
            ncf.var("swrad").put_attr("location","face");
            ncf.var("swrad").put_attr("coordinates","x_rho y_rho ocean_time");
            ncf.var("swrad").put_attr("field","shortwave radiation, scalar, series");
 
            // Evaporation rate (kg m-2 s-1)
            ncf.def_var("evaporation", ncutils::NCDType::Real, {nt_name, ny_r_name, nx_r_name });
            ncf.var("evaporation").put_attr("long_name","evaporation rate");
            ncf.var("evaporation").put_attr("units","kilogram meter-2 second-1");
            ncf.var("evaporation").put_attr("time","ocean_time");
            ncf.var("evaporation").put_attr("grid","grid");
            ncf.var("evaporation").put_attr("location","face");
            ncf.var("evaporation").put_attr("coordinates","x_rho y_rho ocean_time");
            ncf.var("evaporation").put_attr("field","evaporation, scalar, series");
 
            // Rain rate (kg m-2 s-1)
            ncf.def_var("rain", ncutils::NCDType::Real, {nt_name, ny_r_name, nx_r_name });
            ncf.var("rain").put_attr("long_name","rain fall rate");
            ncf.var("rain").put_attr("units","kilogram meter-2 second-1");
            ncf.var("rain").put_attr("time","ocean_time");
            ncf.var("rain").put_attr("grid","grid");
            ncf.var("rain").put_attr("location","face");
            ncf.var("rain").put_attr("coordinates","x_rho y_rho ocean_time");
            ncf.var("rain").put_attr("field","rain, scalar, series");
        }
        // Right now this is hard-wired to {temp, salt, tracer, u, v}
        ncf.put_attr("space_dimension", std::vector<int> { AMREX_SPACEDIM });
//        ncf.put_attr("current_time", std::vector<double> { time });
        ncf.put_attr("start_time", std::vector<double> { start_bdy_time });
        ncf.put_attr("CurrentLevel", std::vector<int> { flev });
        ncf.put_attr("DefaultGeometry", std::vector<int> { amrex::DefaultGeometry().Coord() });
 
        ncf.exit_def_mode();
 
        // We are doing single-level writes but it doesn't have to be level 0
        //
        // Write out the header information.
        //
 
        Real dx[AMREX_SPACEDIM];
        for (int i = 0; i < AMREX_SPACEDIM; i++) {
            dx[i] = geom[lev].CellSize()[i];
        }
        const auto *base = geom[lev].ProbLo();
        RealBox rb(subdomain, dx, base);
 
        amrex::Vector<Real> probLo;
        amrex::Vector<Real> probHi;
        for (int i = 0; i < AMREX_SPACEDIM; i++) {
            probLo.push_back(rb.lo(i));
            probHi.push_back(rb.hi(i));
        }
 
        //nc_probLo.par_access(NC_COLLECTIVE);
        // small variable data written by just the master proc
        ncmpi_begin_indep_data(ncf.ncid);
        if (amrex::ParallelDescriptor::IOProcessor()) // only master proc
        {
            auto nc_probLo = ncf.var("probLo");
 
            nc_probLo.put(probLo.data(), { 0 }, { AMREX_SPACEDIM });
 
            auto nc_probHi = ncf.var("probHi");
            //nc_probHi.par_access(NC_COLLECTIVE);
            nc_probHi.put(probHi.data(), { 0 }, { AMREX_SPACEDIM });
 
            amrex::Vector<int> smallend;
            amrex::Vector<int> bigend;
            for (int i = lev; i < flev; i++) {
                smallend.clear();
                bigend.clear();
                for (int j = 0; j < AMREX_SPACEDIM; j++) {
                    smallend.push_back(subdomain.smallEnd(j));
                    bigend.push_back(subdomain.bigEnd(j));
                }
                auto nc_Geom_smallend = ncf.var("Geom.smallend");
                //nc_Geom_smallend.par_access(NC_COLLECTIVE);
                nc_Geom_smallend.put(smallend.data(), { static_cast<long long int>(i - lev), 0 }, { 1,
                AMREX_SPACEDIM });
 
                auto nc_Geom_bigend = ncf.var("Geom.bigend");
                //nc_Geom_bigend.par_access(NC_COLLECTIVE);
                nc_Geom_bigend.put(bigend.data(), { static_cast<long long int>(i - lev), 0 }, { 1,
                AMREX_SPACEDIM });
            }
 
            amrex::Vector<Real> CellSize;
            for (int i = lev; i < flev; i++) {
                CellSize.clear();
                for (Real &j : dx) {
                    CellSize.push_back(amrex::Real(j));
                }
                auto nc_CellSize = ncf.var("CellSize");
                //nc_CellSize.par_access(NC_COLLECTIVE);
                nc_CellSize.put(CellSize.data(), { static_cast<long long int>(i - lev), 0 }, { 1,
                AMREX_SPACEDIM });
            }
            Real hc = solverChoice.tcline;
            Real theta_s = solverChoice.theta_s;
            Real theta_b = solverChoice.theta_b;
            ncf.var("hc").put(&hc);
            ncf.var("theta_s").put(&theta_s);
            ncf.var("theta_b").put(&theta_b);
 
        }
        ncmpi_end_indep_data(ncf.ncid);
 
    } // end if write_header
 
    // Past this point every write is collective. The loops below stage their
    // hyperslabs into the collector rather than writing them, because the number
    // of hyperslabs a rank contributes depends on how many boxes it owns; the
    // single flush() at the end turns each variable into one ncmpi_put_varn_*_all.
    VarnCollector collector;
 
    //
    // We compute the offsets based on location of the box within the domain
    //
    long long adjusted_history_count = chunk_history_file ? history_count % steps_per_history_file : history_count;
    long long local_start_nt = (is_history ? static_cast<long long>(adjusted_history_count) : static_cast<long long>(0));
    long long local_nt = 1; // We write data for only one time
 
    if (amrex::ParallelDescriptor::IOProcessor()) // only master proc
    {
        auto nc_plot_var = collector.var(ncf, "ocean_time");
        //nc_plot_var.par_access(NC_COLLECTIVE);
        nc_plot_var.put(&t_new[lev], { local_start_nt }, { local_nt });
    }
 
    // Check whether there are any nans or infs in variables that we will write out
    if (vec_Zt_avg1[lev]->contains_nan() || vec_Zt_avg1[lev]->contains_inf()) {
        amrex::Abort("Found while writing output: zeta contains nan or inf");
    }
    // Check every cell-centered tracer that is actually being written: temperature,
    // salinity, the passive scalars and the biology tracers. plotMF is indexed by
    // position in plot_var_names_3d, not by cons component, so look the name up the
    // same way the write loops below do.
    for (int n = 0; n < ncons; ++n) {
        int comp = -1;
        for (int i = 0; i < plot_var_names_3d.size(); i++) {
            if (plot_var_names_3d[i] == cons_names[n]) comp = i;
        }
        if (comp < 0) { continue; }
        if (plotMF->contains_nan(comp,1) || plotMF->contains_inf(comp,1)) {
            amrex::Abort("Found while writing output: " + cons_names[n] +
                         " contains nan or inf");
        }
    }
    if (xvel_new[lev]->contains_nan() || xvel_new[lev]->contains_inf()) {
        amrex::Abort("Found while writing output: velocity u contains nan or inf");
    }
    if (vec_ubar[lev]->contains_nan(0,1) || vec_ubar[lev]->contains_inf(0,1)) {
        amrex::Abort("Found while writing output: velocity ubar contains nan or inf");
    }
    if (yvel_new[lev]->contains_nan() || yvel_new[lev]->contains_inf()) {
        amrex::Abort("Found while writing output: velocity v contains nan or inf");
    }
    if (vec_vbar[lev]->contains_nan(0,1) || vec_vbar[lev]->contains_inf(0,1)) {
        amrex::Abort("Found while writing output: velocity vbar contains nan or inf");
    }
 
    for (MFIter mfi(*plotMF, false); mfi.isValid(); ++mfi) {
        auto bx = mfi.validbox();
        if (subdomain.contains(bx)) {
            //
            // We only include one grow cell at subdomain boundaries, not internal grid boundaries
            //
            Box tmp_bx(bx);
            if (tmp_bx.smallEnd()[0] == subdomain.smallEnd()[0])
                tmp_bx.growLo(0, 1);
            if (tmp_bx.smallEnd()[1] == subdomain.smallEnd()[1])
                tmp_bx.growLo(1, 1);
            if (tmp_bx.bigEnd()[0] == subdomain.bigEnd()[0])
                tmp_bx.growHi(0, 1);
            if (tmp_bx.bigEnd()[1] == subdomain.bigEnd()[1])
                tmp_bx.growHi(1, 1);
            // amrex::Print() << "    BX " << bx << std::endl;
            // amrex::Print() << "TMP_BX " << tmp_bx << std::endl;
 
            Box tmp_bx_2d(tmp_bx);
            tmp_bx_2d.makeSlab(2, 0);
 
            Box tmp_bx_1d(tmp_bx);
            tmp_bx_1d.makeSlab(0, 0);
            tmp_bx_1d.makeSlab(1, 0);
 
            //
            // These are the dimensions of the data we write for only this box
            //
            long long local_nx = tmp_bx.length()[0];
            long long local_ny = tmp_bx.length()[1];
            long long local_nz = tmp_bx.length()[2];
 
            // We do the "+1" because the offset needs to start at 0
            long long local_start_x = static_cast<long long>(tmp_bx.smallEnd()[0] + 1);
            long long local_start_y = static_cast<long long>(tmp_bx.smallEnd()[1] + 1);
            long long local_start_z = static_cast<long long>(tmp_bx.smallEnd()[2]);
 
            if (write_header) {
                // Only write out s_rho and s_w at x=0,y=0 to avoid NaNs
                if (bx.contains(IntVect(0,0,0)))
                {
                    {
                        amrex::Vector<amrex::Real> tmp_srho(local_nz);
 
#ifdef AMREX_USE_GPU
                        Gpu::dtoh_memcpy(tmp_srho.data(), s_r.data(), sizeof(amrex::Real)*local_nz);
#else
                        std::memcpy(tmp_srho.data(), s_r.data(), sizeof(amrex::Real)*local_nz);
#endif
                        Gpu::streamSynchronize();
 
                        auto nc_plot_var = collector.var(ncf, "s_rho");
                        //nc_plot_var.par_access(NC_INDEPENDENT);
                        nc_plot_var.put(tmp_srho.data(), { local_start_z }, { local_nz });
                    }
                    {
                        amrex::Vector<amrex::Real> tmp_sw(local_nz+1);
 
#ifdef AMREX_USE_GPU
                        Gpu::dtoh_memcpy(tmp_sw.data(), s_w.data(), sizeof(amrex::Real)*(local_nz+1));
#else
                        std::memcpy(tmp_sw.data(), s_w.data(), sizeof(amrex::Real)*(local_nz+1));
#endif
                        Gpu::streamSynchronize();
 
                        auto nc_plot_var = collector.var(ncf, "s_w");
                        //nc_plot_var.par_access(NC_INDEPENDENT);
                        nc_plot_var.put(tmp_sw.data(), { local_start_z }, { local_nz + 1});
                    }
                    {
                        amrex::Vector<amrex::Real> tmp_Csrho(local_nz);
 
#ifdef AMREX_USE_GPU
                        Gpu::dtoh_memcpy(tmp_Csrho.data(), Cs_r.data(), sizeof(amrex::Real)*(local_nz));
#else
                        std::memcpy(tmp_Csrho.data(), Cs_r.data(), sizeof(amrex::Real)*(local_nz));
#endif
                        Gpu::streamSynchronize();
 
                        auto nc_plot_var = collector.var(ncf, "Cs_r");
                        //nc_plot_var.par_access(NC_INDEPENDENT);
                        nc_plot_var.put(tmp_Csrho.data(), { local_start_z }, { local_nz });
                    }
                    {
                        amrex::Vector<amrex::Real> tmp_Csw(local_nz+1);
 
#ifdef AMREX_USE_GPU
                        Gpu::dtoh_memcpy(tmp_Csw.data(), Cs_w.data(), sizeof(amrex::Real)*(local_nz+1));
#else
                        std::memcpy(tmp_Csw.data(), Cs_w.data(), sizeof(amrex::Real)*(local_nz+1));
#endif
 
                        Gpu::streamSynchronize();
 
                        auto nc_plot_var = collector.var(ncf, "Cs_w");
                        //nc_plot_var.par_access(NC_INDEPENDENT);
                        nc_plot_var.put(tmp_Csw.data(), { local_start_z }, { local_nz + 1});
                    }
                }
 
                {
                    FArrayBox tmp_bathy;
                    tmp_bathy.resize(tmp_bx_2d, 1, amrex::The_Pinned_Arena());
 
                    tmp_bathy.template copy<RunOn::Device>((*vec_h[lev])[mfi.index()], 0, 0, 1);
                    Gpu::streamSynchronize();
 
                    auto nc_plot_var = collector.var(ncf, "h");
                    //nc_plot_var.par_access(NC_INDEPENDENT);
                    nc_plot_var.put(tmp_bathy.dataPtr(), { local_start_y, local_start_x }, { local_ny, local_nx });
                }
 
                {
                    FArrayBox tmp_pm;
                    tmp_pm.resize(tmp_bx_2d, 1, amrex::The_Pinned_Arena());
 
                    tmp_pm.template copy<RunOn::Device>((*vec_pm[lev])[mfi.index()], 0, 0, 1);
                    Gpu::streamSynchronize();
 
                    auto nc_plot_var = collector.var(ncf, "pm");
                    //nc_plot_var.par_access(NC_INDEPENDENT);
 
                    nc_plot_var.put(tmp_pm.dataPtr(), { local_start_y, local_start_x }, { local_ny, local_nx });
                }
 
                {
                    FArrayBox tmp_pn;
                    tmp_pn.resize(tmp_bx_2d, 1, amrex::The_Pinned_Arena());
 
                    tmp_pn.template copy<RunOn::Device>((*vec_pn[lev])[mfi.index()], 0, 0, 1);
                    Gpu::streamSynchronize();
 
                    auto nc_plot_var = collector.var(ncf, "pn");
                    //nc_plot_var.par_access(NC_INDEPENDENT);
                    nc_plot_var.put(tmp_pn.dataPtr(), { local_start_y, local_start_x }, { local_ny, local_nx });
                }
 
                {
                    FArrayBox tmp_f;
                    tmp_f.resize(tmp_bx_2d, 1, amrex::The_Pinned_Arena());
 
                    tmp_f.template copy<RunOn::Device>((*vec_fcor[lev])[mfi.index()], 0, 0, 1);
                    Gpu::streamSynchronize();
 
                    auto nc_plot_var = collector.var(ncf, "f");
                    //nc_plot_var.par_access(NC_INDEPENDENT);
                    nc_plot_var.put(tmp_f.dataPtr(), { local_start_y, local_start_x }, { local_ny, local_nx });
                }
 
                {
                    FArrayBox tmp_xr;
                    tmp_xr.resize(tmp_bx_2d, 1, amrex::The_Pinned_Arena());
 
                    tmp_xr.template copy<RunOn::Device>((*vec_xr[lev])[mfi.index()], 0, 0, 1);
                    Gpu::streamSynchronize();
 
                    auto nc_plot_var = collector.var(ncf, "x_rho");
                    //nc_plot_var.par_access(NC_INDEPENDENT);
 
                    nc_plot_var.put(tmp_xr.dataPtr(), { local_start_y, local_start_x }, { local_ny, local_nx });
                }
 
                {
                    FArrayBox tmp_yr;
                    tmp_yr.resize(tmp_bx_2d, 1, amrex::The_Pinned_Arena());
 
                    tmp_yr.template copy<RunOn::Device>((*vec_yr[lev])[mfi.index()], 0, 0, 1);
                    Gpu::streamSynchronize();
 
                    auto nc_plot_var = collector.var(ncf, "y_rho");
                    //nc_plot_var.par_access(NC_INDEPENDENT);
                    nc_plot_var.put(tmp_yr.dataPtr(), { local_start_y, local_start_x }, { local_ny, local_nx });
                }
            }
 
            {
                FArrayBox tmp_zeta;
                tmp_zeta.resize(tmp_bx_2d, 1, amrex::The_Pinned_Arena());
                tmp_zeta.template copy<RunOn::Device>((*vec_Zt_avg1[lev])[mfi.index()], 0, 0, 1);
                Gpu::streamSynchronize();
 
                auto nc_plot_var = collector.var(ncf, "zeta");
                nc_plot_var.put(tmp_zeta.dataPtr(), { local_start_nt, local_start_y, local_start_x }, { local_nt, local_ny,
                        local_nx });
            }
 
            {
                FArrayBox tmp_mask_rho;
                tmp_mask_rho.resize(tmp_bx_2d, 1, amrex::The_Pinned_Arena());
                tmp_mask_rho.template copy<RunOn::Device>((*vec_mskr[lev])[mfi.index()], 0, 0, 1);
                Gpu::streamSynchronize();
 
                auto nc_plot_var = collector.var(ncf, "mask_rho");
                nc_plot_var.put(tmp_mask_rho.dataPtr(), { local_start_nt, local_start_y, local_start_x }, { local_nt, local_ny,
                        local_nx });
            }
 
            // stflux_*, one per tracer. Defined above under the same condition.
            for (int n = 0; n < ncons; ++n) {
                const std::string nm = std::string("stflux_") + cons_names[n];
                if (!plot_2d_var_requested(plot_var_names_2d, nm)) { continue; }
                FArrayBox tmp;
                tmp.resize(tmp_bx_2d, 1, amrex::The_Pinned_Arena());
                tmp.template copy<RunOn::Device>((*vec_stflux[lev])[mfi.index()], n, 0, 1);
                Gpu::streamSynchronize();
 
                auto nc_var = collector.var(ncf, nm);
                nc_var.put(tmp.dataPtr(), { local_start_nt, local_start_y, local_start_x },
                                          { local_nt,       local_ny,       local_nx });
            }
 
            if (solverChoice.output_forcing)
            {
                const Real Hscale = solverChoice.rho0 * Cp;
                // The copy and the mult below are both async on the same stream, so the
                // sync has to follow the last of them and precede the host-side put().
                // Tair
                {
                    FArrayBox tmp_Tair;
                    tmp_Tair.resize(tmp_bx_2d, 1, amrex::The_Pinned_Arena());
                    tmp_Tair.template copy<RunOn::Device>((*vec_Tair[lev])[mfi.index()], 0, 0, 1);
                    Gpu::streamSynchronize();
 
                    auto nc_plot_var = collector.var(ncf, "Tair");
                    nc_plot_var.put(tmp_Tair.dataPtr(), { local_start_nt, local_start_y, local_start_x }, { local_nt, local_ny, local_nx });
                }
                // Pair
                {
                    FArrayBox tmp_Pair;
                    tmp_Pair.resize(tmp_bx_2d, 1, amrex::The_Pinned_Arena());
                    tmp_Pair.template copy<RunOn::Device>((*vec_Pair[lev])[mfi.index()], 0, 0, 1);
                    Gpu::streamSynchronize();
 
                    auto nc_plot_var = collector.var(ncf, "Pair");
                    nc_plot_var.put(tmp_Pair.dataPtr(), { local_start_nt, local_start_y, local_start_x }, { local_nt, local_ny, local_nx });
                }
                // qnet  (stored °C m/s → write W/m²)
                {
                    FArrayBox tmp;
                    tmp.resize(tmp_bx_2d, 1, amrex::The_Pinned_Arena());
 
                    // Copy stflux Temp component
                    tmp.template copy<RunOn::Device>(
                        (*vec_stflux[lev])[mfi.index()],
                        Temp_comp,  // source component
                        0,          // dest component
                        1           // number of comps
                    );
 
                    // Convert °C·m/s → W/m²
                    tmp.mult<RunOn::Device>(Hscale);
 
                    Gpu::streamSynchronize();
 
                    auto nc_var = collector.var(ncf, "qnet");
                    nc_var.put(tmp.dataPtr(),
                            { local_start_nt, local_start_y, local_start_x },
                            { local_nt,       local_ny,       local_nx });
                }
                // ssflux = surface net freshwater flux (kg/m²/s converted to m/s)
                {
                    FArrayBox tmp;
                    tmp.resize(tmp_bx_2d, 1, amrex::The_Pinned_Arena());
 
                    // Copy stflux Salt component
                    tmp.template copy<RunOn::Device>(
                        (*vec_stflux[lev])[mfi.index()],
                        Salt_comp, // source component
                        0,         // destination component
                        1          // number of components
                    );
 
                    Gpu::streamSynchronize();
 
                    auto nc_var = collector.var(ncf, "ssflux");
                    nc_var.put(tmp.dataPtr(),
                            { local_start_nt, local_start_y, local_start_x },
                            { local_nt,       local_ny,       local_nx });
                }
                // latent  (stored °C m/s → write W/m²)
                {
                    FArrayBox tmp;
                    tmp.resize(tmp_bx_2d, 1, amrex::The_Pinned_Arena());
                    tmp.template copy<RunOn::Device>((*vec_lhflx[lev])[mfi.index()], 0, 0, 1);
 
                    // Convert °C·m/s → W/m²
                    tmp.mult<RunOn::Device>(Hscale);
 
                    Gpu::streamSynchronize();
 
                    auto nc_var = collector.var(ncf, "latent");
                    nc_var.put(tmp.dataPtr(),
                            { local_start_nt, local_start_y, local_start_x },
                            { local_nt,       local_ny,       local_nx });
                }
                // sensible  (stored °C m/s → write W/m²)
                {
                    FArrayBox tmp;
                    tmp.resize(tmp_bx_2d, 1, amrex::The_Pinned_Arena());
                    tmp.template copy<RunOn::Device>((*vec_shflx[lev])[mfi.index()], 0, 0, 1);
 
                    // Convert °C·m/s → W/m²
                    tmp.mult<RunOn::Device>(Hscale);
 
                    Gpu::streamSynchronize();
 
                    auto nc_var = collector.var(ncf, "sensible");
                    nc_var.put(tmp.dataPtr(),
                            { local_start_nt, local_start_y, local_start_x },
                            { local_nt,       local_ny,       local_nx });
                }
                // lwrad  (stored °C m/s → write W/m²)
                {
                    FArrayBox tmp;
                    tmp.resize(tmp_bx_2d, 1, amrex::The_Pinned_Arena());
                    tmp.template copy<RunOn::Device>((*vec_lrflx[lev])[mfi.index()], 0, 0, 1);
 
                    // Convert °C·m/s → W/m²
                    tmp.mult<RunOn::Device>(Hscale);
 
                    Gpu::streamSynchronize();
 
                    auto nc_var = collector.var(ncf, "lwrad");
                    nc_var.put(tmp.dataPtr(),
                            { local_start_nt, local_start_y, local_start_x },
                            { local_nt,       local_ny,       local_nx });
                }
                // swrad, note this is stored explicitly as W/m², not degC m/s in REMORA.bulk_flux.cpp
                {
                    FArrayBox tmp;
                    tmp.resize(tmp_bx_2d, 1, amrex::The_Pinned_Arena());
                    tmp.template copy<RunOn::Device>((*vec_srflx[lev])[mfi.index()], 0, 0, 1);
                    Gpu::streamSynchronize();
 
                    auto nc_var = collector.var(ncf, "swrad");
                    nc_var.put(tmp.dataPtr(),
                            { local_start_nt, local_start_y, local_start_x },
                            { local_nt,       local_ny,       local_nx });
                }
                // evaporation
                {
                    FArrayBox tmp;
                    tmp.resize(tmp_bx_2d, 1, amrex::The_Pinned_Arena());
                    tmp.template copy<RunOn::Device>((*vec_evap[lev])[mfi.index()], 0, 0, 1);
                    Gpu::streamSynchronize();
 
                    auto nc_var = collector.var(ncf, "evaporation");
                    nc_var.put(tmp.dataPtr(),
                            { local_start_nt, local_start_y, local_start_x },
                            { local_nt,       local_ny,       local_nx });
                }
                // rain
                {
                    FArrayBox tmp;
                    tmp.resize(tmp_bx_2d, 1, amrex::The_Pinned_Arena());
                    tmp.template copy<RunOn::Device>((*vec_rain[lev])[mfi.index()], 0, 0, 1);
                    Gpu::streamSynchronize();
 
                    auto nc_var = collector.var(ncf, "rain");
                    nc_var.put(tmp.dataPtr(),
                            { local_start_nt, local_start_y, local_start_x },
                            { local_nt,       local_ny,       local_nx });
                }
            } // end output forcing
 
            // **************************************************************************
            for (int n = 0; n < ncons; ++n) {
                int comp = -1;
                for (int i = 0; i < plot_var_names_3d.size(); i++) {
                    if (plot_var_names_3d[i] == cons_names[n]) comp = i;
                }
                if (comp >= 0) {
                    FArrayBox tmp;
                    tmp.resize(tmp_bx, 1, amrex::The_Pinned_Arena());
                    tmp.template copy<RunOn::Device>((*plotMF)[mfi.index()], comp, 0, 1);
                    Gpu::streamSynchronize();
 
                    auto nc_plot_var = collector.var(ncf, cons_names[n]);
                    nc_plot_var.put(tmp.dataPtr(), { local_start_nt, local_start_z, local_start_y, local_start_x }, { local_nt,
                            local_nz, local_ny, local_nx });
                }
            }
            // **************************************************************************
 
            // **************************************************************************
            { // Vorticity
                int comp = -1;
                for (int i = 0; i < plot_var_names_3d.size(); i++) {
                    if (plot_var_names_3d[i] == "vorticity") comp = i;
                }
                if (comp >= 0) {
                    FArrayBox tmp;
                    tmp.resize(tmp_bx, 1, amrex::The_Pinned_Arena());
                    tmp.template copy<RunOn::Device>((*plotMF)[mfi.index()], comp, 0, 1);
                    Gpu::streamSynchronize();
 
                    auto nc_plot_var = collector.var(ncf, plot_var_names_3d[comp]);
                    nc_plot_var.put(tmp.dataPtr(), { local_start_nt, local_start_z, local_start_y, local_start_x }, { local_nt,
                            local_nz, local_ny, local_nx });
                } // if vorticity exists in plotMF
            } // end vorticity
            // **************************************************************************
 
            // **************************************************************************
            // Horizontal mixing coefficients (scaled_to_grid):
            // vertically homogeneous and time-invariant -> write static 2D fields once.
            // **************************************************************************
            if ((solverChoice.horiz_mixing_type == HorizMixingType::scaled_to_grid) && write_header) {
                { // visc2
                    FArrayBox tmp;
                    tmp.resize(tmp_bx_2d, 1, amrex::The_Pinned_Arena());
                    tmp.template copy<RunOn::Device>((*vec_visc2_r[lev])[mfi.index()], 0, 0, 1);
                    Gpu::streamSynchronize();
 
                    auto nc_var = collector.var(ncf, "visc2");
                    nc_var.put(tmp.dataPtr(), { local_start_y, local_start_x }, { local_ny, local_nx });
                }
 
                // diff2_*
                for (int n = 0; n < ncons; ++n) {
                    const std::string nm = std::string("diff2_") + cons_names[n];
                    FArrayBox tmp;
                    tmp.resize(tmp_bx_2d, 1, amrex::The_Pinned_Arena());
                    tmp.template copy<RunOn::Device>((*vec_diff2[lev])[mfi.index()], n, 0, 1);
                    Gpu::streamSynchronize();
 
                    auto nc_var = collector.var(ncf, nm);
                    nc_var.put(tmp.dataPtr(), { local_start_y, local_start_x }, { local_ny, local_nx });
                }
            }
 
        } // subdomain
    } // mfi
 
    // Flush at each loop boundary rather than once at the end, so the collector
    // only ever holds one loop's worth of staged data. All ranks reach these
    // points, which is what the collective call requires.
    collector.flush(ncf);
 
    // Writing u (we loop over cons to get cell-centered box)
    for (MFIter mfi(*plotMF, false); mfi.isValid(); ++mfi) {
        Box bx = mfi.validbox();
 
        if (subdomain.contains(bx)) {
            //
            // We only include one grow cell at subdomain boundaries, not internal grid boundaries
            //
            Box tmp_bx(bx);
            tmp_bx.surroundingNodes(0);
            if (tmp_bx.smallEnd()[1] == subdomain.smallEnd()[1])
                tmp_bx.growLo(1, 1);
            if (tmp_bx.bigEnd()[1] == subdomain.bigEnd()[1])
                tmp_bx.growHi(1, 1);
            Box tmp_bx_2d(tmp_bx);
            tmp_bx_2d.makeSlab(2, 0);
 
            //
            // These are the dimensions of the data we write for only this box
            //
            long long local_nx = tmp_bx.length()[0];
            long long local_ny = tmp_bx.length()[1];
            long long local_nz = tmp_bx.length()[2];
 
            // We do the "+1" because the offset needs to start at 0
            long long local_start_x = static_cast<long long>(tmp_bx.smallEnd()[0]);
            long long local_start_y = static_cast<long long>(tmp_bx.smallEnd()[1] + 1);
            long long local_start_z = static_cast<long long>(tmp_bx.smallEnd()[2]);
 
            if (write_header) {
                {
                    FArrayBox tmp;
                    tmp.resize(tmp_bx_2d, 1, amrex::The_Pinned_Arena());
                    tmp.template copy<RunOn::Device>((*vec_xu[lev])[mfi.index()], 0, 0, 1);
                    Gpu::streamSynchronize();
 
                    auto nc_plot_var = collector.var(ncf, "x_u");
                    //nc_plot_var.par_access(NC_INDEPENDENT);
                    nc_plot_var.put(tmp.dataPtr(), { local_start_y, local_start_x }, { local_ny, local_nx });
                }
                {
                    FArrayBox tmp;
                    tmp.resize(tmp_bx_2d, 1, amrex::The_Pinned_Arena());
                    tmp.template copy<RunOn::Device>((*vec_yu[lev])[mfi.index()], 0, 0, 1);
                    Gpu::streamSynchronize();
 
                    auto nc_plot_var = collector.var(ncf, "y_u");
                    //nc_plot_var.par_access(NC_INDEPENDENT);
                    nc_plot_var.put(tmp.dataPtr(), { local_start_y, local_start_x }, { local_ny, local_nx });
                }
            }
 
            {
                FArrayBox tmp;
                tmp.resize(tmp_bx, 1, amrex::The_Pinned_Arena());
                tmp.template copy<RunOn::Device>((*xvel_new[lev])[mfi.index()], 0, 0, 1);
                Gpu::streamSynchronize();
 
                auto nc_plot_var = collector.var(ncf, "u");
                nc_plot_var.put(tmp.dataPtr(), { local_start_nt, local_start_z, local_start_y, local_start_x }, { local_nt,
                        local_nz, local_ny, local_nx });
            }
 
            {
                FArrayBox tmp;
                tmp.resize(tmp_bx_2d, 1, amrex::The_Pinned_Arena());
                tmp.template copy<RunOn::Device>((*vec_ubar[lev])[mfi.index()], 0, 0, 1);
                Gpu::streamSynchronize();
 
                auto nc_plot_var = collector.var(ncf, "ubar");
                nc_plot_var.put(tmp.dataPtr(), { local_start_nt, local_start_y, local_start_x }, { local_nt, local_ny, local_nx });
            }
            {
                FArrayBox tmp;
                tmp.resize(tmp_bx_2d, 1, amrex::The_Pinned_Arena());
                tmp.template copy<RunOn::Device>((*vec_sustr[lev])[mfi.index()], 0, 0, 1);
                Gpu::streamSynchronize();
 
                auto nc_plot_var = collector.var(ncf, "sustr");
                nc_plot_var.put(tmp.dataPtr(), { local_start_nt, local_start_y, local_start_x }, { local_nt, local_ny, local_nx });
            }
            {
                FArrayBox tmp;
                tmp.resize(tmp_bx_2d, 1, amrex::The_Pinned_Arena());
                tmp.template copy<RunOn::Device>((*vec_msku[lev])[mfi.index()], 0, 0, 1);
                Gpu::streamSynchronize();
 
                auto nc_plot_var = collector.var(ncf, "mask_u");
                nc_plot_var.put(tmp.dataPtr(), { local_start_nt, local_start_y, local_start_x }, { local_nt, local_ny, local_nx });
            }
        } // in subdomain
    } // mfi
 
    collector.flush(ncf);
 
    // Writing v (we loop over cons to get cell-centered box)
    for (MFIter mfi(*plotMF, false); mfi.isValid(); ++mfi) {
        Box bx = mfi.validbox();
 
        if (subdomain.contains(bx)) {
            //
            // We only include one grow cell at subdomain boundaries, not internal grid boundaries
            //
            Box tmp_bx(bx);
            tmp_bx.surroundingNodes(1);
            if (tmp_bx.smallEnd()[0] == subdomain.smallEnd()[0])
                tmp_bx.growLo(0, 1);
            if (tmp_bx.bigEnd()[0] == subdomain.bigEnd()[0])
                tmp_bx.growHi(0, 1);
            // amrex::Print() << "    BX " << bx << std::endl;
            // amrex::Print() << "TMP_BX " << tmp_bx << std::endl;
 
            Box tmp_bx_2d(tmp_bx);
            tmp_bx_2d.makeSlab(2, 0);
 
            //
            // These are the dimensions of the data we write for only this box
            //
            long long local_nx = tmp_bx.length()[0];
            long long local_ny = tmp_bx.length()[1];
            long long local_nz = tmp_bx.length()[2];
 
            // We do the "+1" because the offset needs to start at 0
            long long local_start_x = static_cast<long long>(tmp_bx.smallEnd()[0] + 1);
            long long local_start_y = static_cast<long long>(tmp_bx.smallEnd()[1]);
            long long local_start_z = static_cast<long long>(tmp_bx.smallEnd()[2]);
 
            if (write_header) {
                {
                FArrayBox tmp;
                tmp.resize(tmp_bx_2d, 1, amrex::The_Pinned_Arena());
                tmp.template copy<RunOn::Device>((*vec_xv[lev])[mfi.index()], 0, 0, 1);
                Gpu::streamSynchronize();
 
                auto nc_plot_var = collector.var(ncf, "x_v");
                //nc_plot_var.par_access(NC_INDEPENDENT);
                nc_plot_var.put(tmp.dataPtr(), { local_start_y, local_start_x }, { local_ny, local_nx });
                }
                {
                FArrayBox tmp;
                tmp.resize(tmp_bx_2d, 1, amrex::The_Pinned_Arena());
                tmp.template copy<RunOn::Device>((*vec_yv[lev])[mfi.index()], 0, 0, 1);
                Gpu::streamSynchronize();
 
                auto nc_plot_var = collector.var(ncf, "y_v");
                //nc_plot_var.par_access(NC_INDEPENDENT);
                nc_plot_var.put(tmp.dataPtr(), { local_start_y, local_start_x }, { local_ny, local_nx });
                }
            }
 
            {
                FArrayBox tmp;
                tmp.resize(tmp_bx, 1, amrex::The_Pinned_Arena());
                tmp.template copy<RunOn::Device>((*yvel_new[lev])[mfi.index()], 0, 0, 1);
                Gpu::streamSynchronize();
 
                auto nc_plot_var = collector.var(ncf, "v");
                nc_plot_var.put(tmp.dataPtr(), { local_start_nt, local_start_z, local_start_y, local_start_x }, { local_nt,
                        local_nz, local_ny, local_nx });
            }
 
            {
                FArrayBox tmp;
                tmp.resize(tmp_bx_2d, 1, amrex::The_Pinned_Arena());
                tmp.template copy<RunOn::Device>((*vec_vbar[lev])[mfi.index()], 0, 0, 1);
                Gpu::streamSynchronize();
 
                auto nc_plot_var = collector.var(ncf, "vbar");
                nc_plot_var.put(tmp.dataPtr(), { local_start_nt, local_start_y, local_start_x }, { local_nt, local_ny, local_nx });
            }
 
            {
                FArrayBox tmp;
                tmp.resize(tmp_bx_2d, 1, amrex::The_Pinned_Arena());
                tmp.template copy<RunOn::Device>((*vec_svstr[lev])[mfi.index()], 0, 0, 1);
                Gpu::streamSynchronize();
 
                auto nc_plot_var = collector.var(ncf, "svstr");
                nc_plot_var.put(tmp.dataPtr(), { local_start_nt, local_start_y, local_start_x }, { local_nt, local_ny, local_nx });
            }
            {
                FArrayBox tmp;
                tmp.resize(tmp_bx_2d, 1, amrex::The_Pinned_Arena());
                tmp.template copy<RunOn::Device>((*vec_mskv[lev])[mfi.index()], 0, 0, 1);
                Gpu::streamSynchronize();
 
                auto nc_plot_var = collector.var(ncf, "mask_v");
                nc_plot_var.put(tmp.dataPtr(), { local_start_nt, local_start_y, local_start_x }, { local_nt, local_ny, local_nx });
            }
 
        } // in subdomain
    } // mfi
 
    collector.flush(ncf);
 
    for (MFIter mfi(*plotMF, false); mfi.isValid(); ++mfi) {
        Box bx = mfi.validbox();
 
        if (subdomain.contains(bx)) {
            //
            // We only include one grow cell at subdomain boundaries, not internal grid boundaries
            //
            Box tmp_bx(bx);
            tmp_bx.surroundingNodes(0);
            tmp_bx.surroundingNodes(1);
 
            Box tmp_bx_2d(tmp_bx);
            tmp_bx_2d.makeSlab(2, 0);
 
            //
            // These are the dimensions of the data we write for only this box
            //
            long long local_nx = tmp_bx.length()[0];
            long long local_ny = tmp_bx.length()[1];
 
            // We do the "+1" because the offset needs to start at 0
            long long local_start_x = static_cast<long long>(tmp_bx.smallEnd()[0]);
            long long local_start_y = static_cast<long long>(tmp_bx.smallEnd()[1]);
 
            if (write_header) {
            {
                FArrayBox tmp;
                tmp.resize(tmp_bx_2d, 1, amrex::The_Pinned_Arena());
                tmp.template copy<RunOn::Device>((*vec_xp[lev])[mfi.index()], 0, 0, 1);
                Gpu::streamSynchronize();
 
                auto nc_plot_var = collector.var(ncf, "x_psi");
                //nc_plot_var.par_access(NC_INDEPENDENT);
                nc_plot_var.put(tmp.dataPtr(), { local_start_y, local_start_x }, { local_ny, local_nx });
            }
            {
                FArrayBox tmp;
                tmp.resize(tmp_bx_2d, 1, amrex::The_Pinned_Arena());
                tmp.template copy<RunOn::Device>((*vec_yp[lev])[mfi.index()], 0, 0, 1);
                Gpu::streamSynchronize();
 
                auto nc_plot_var = collector.var(ncf, "y_psi");
                //nc_plot_var.par_access(NC_INDEPENDENT);
                nc_plot_var.put(tmp.dataPtr(), { local_start_y, local_start_x }, { local_ny, local_nx });
            }
 
            } // header
        } // in subdomain
    } // mfi
 
    // One collective ncmpi_put_varn_*_all per variable. Collective writes keep
    // numrecs synced in the file header as they go, so the separate
    // ncmpi_end_indep_data() sync the independent path needed is no longer
    // required here.
    collector.flush(ncf);
 
    ncf.close();
 
    REMORA::total_nc_plot_file_step += 1;
}