IOManager.H

Go to the documentation of this file.

#ifndef IOMANAGER_H
#define IOMANAGER_H
 
#include <string>
#include <unordered_map>
#include <set>
 
#include "AMReX_Vector.H"
#include "AMReX_BoxArray.H"
#include "AMReX_DistributionMapping.H"
 
namespace amr_wind {
 
class CFDSim;
class Field;
class IntField;
class DerivedQtyMgr;
 
class IOManager
{
public:
    explicit IOManager(CFDSim& /*sim*/);
 
    ~IOManager();
 
    IOManager(const IOManager&) = delete;
    IOManager& operator=(const IOManager&) = delete;
 
    void initialize_io();
 
    void write_plot_file();
 
    void
    write_checkpoint_file(const int start_level = 0, const int end_level = -1);
 
    void read_checkpoint_fields(
        const std::string& restart_file,
        const amrex::Vector<amrex::BoxArray>& ba_chk,
        const amrex::Vector<amrex::DistributionMapping>& dm_chk,
        const amrex::IntVect& rep);
 
    void register_output_var(const std::string& fname)
    {
        m_pltvars_default.insert(fname);
    }
 
    void register_output_int_var(const std::string& fname)
    {
        m_int_pltvars_default.insert(fname);
    }
 
    void register_restart_var(const std::string& fname)
    {
        m_chkvars.insert(fname);
    }
 
    void register_io_var(const std::string& fname)
    {
        m_pltvars_default.insert(fname);
        m_chkvars.insert(fname);
    }
 
    const std::string& restart_file() { return m_restart_file; }
 
    bool is_restart() const { return !m_restart_file.empty(); }
 
    const amrex::Vector<Field*>& plot_fields() const { return m_plt_fields; }
 
    const amrex::Vector<Field*>& checkpoint_fields() const
    {
        return m_chk_fields;
    }
 
    const std::string& post_processing_directory()
    {
        if (!amrex::UtilCreateDirectory(m_post_dir, 0755)) {
            amrex::CreateDirectoryFailed(m_post_dir);
        }
        return m_post_dir;
    }
 
private:
    void write_header(
        const std::string& /*chkname*/,
        const int start_level,
        const int end_level);
 
    void write_info_file(const std::string& /*path*/);
 
    CFDSim& m_sim;
 
    std::unique_ptr<DerivedQtyMgr> m_derived_mgr;
 
    std::set<std::string> m_pltvars_default;
 
    std::set<std::string> m_int_pltvars_default;
 
    std::set<std::string> m_chkvars;
 
    amrex::Vector<Field*> m_plt_fields;
 
    amrex::Vector<IntField*> m_int_plt_fields;
 
    amrex::Vector<Field*> m_chk_fields;
 
    amrex::Vector<std::string> m_plt_var_names;
 
    std::string m_plt_prefix{"plt"};
 
    std::string m_chk_prefix{"chk"};
 
    std::string m_post_dir{"post_processing"};
 
    std::string m_restart_file;
 
    int m_plt_num_comp{0};
 
    bool m_output_default_vars{true};
 
    bool m_allow_missing_restart_fields{true};
 
#ifdef AMR_WIND_USE_HDF5
    bool m_output_hdf5_plotfile{false};
#ifdef AMR_WIND_USE_HDF5_ZFP
    std::string m_hdf5_compression{"None@0"};
#endif
#endif
 
    int m_nfiles{256};
};
 
} // namespace amr_wind
 
#endif /* IOMANAGER_H */