PostProcessing.H

Go to the documentation of this file.

#ifndef POSTPROCESSING_H
#define POSTPROCESSING_H
 
#include <memory>
#include "amr-wind/core/Factory.H"
#include "AMReX_ParmParse.H"
#include "AMReX_REAL.H"
 
using namespace amrex::literals;

 
namespace amr_wind {
 
class CFDSim;

class PostProcessBase
    : public Factory<PostProcessBase, CFDSim&, const std::string&>
{
public:
    static std::string base_identifier() { return "PostProcessBase"; }
 
    ~PostProcessBase() override = default;
 
    virtual void pre_init_actions() = 0;

    virtual void initialize() = 0;

    virtual void post_advance_work() = 0;

    virtual void output_actions() = 0;

    virtual void post_regrid_actions() = 0;
 
    void populate_output_parameters(amrex::ParmParse& pp)
    {
        pp.query("output_interval", m_out_interval);
        pp.query("output_time_interval", m_out_time_interval);
        pp.query("output_delay", m_out_delay);
        pp.query("output_time_delay", m_out_time_delay);
        pp.query("enforce_output_time_dt", m_enforce_dt);
        pp.query("enforce_output_time_dt_reltol", m_enforce_dt_tol);
 
        // Check for presence of arguments
        const bool has_interval = pp.contains("output_interval");
        const bool has_time_int = pp.contains("output_time_interval");
        const bool has_frequency = pp.contains("output_frequency");
 
        if (has_frequency) {
            if (has_interval) {
                amrex::Print()
                    << "WARNING: output_interval and output_frequency "
                       "(legacy argument) are both specified; "
                       "output_frequency will be ignored.\n";
            } else {
                pp.get("output_frequency", m_out_interval);
                amrex::Print()
                    << "WARNING: legacy input argument of output_frequency has "
                       "been specified; the code will function as normal, but "
                       "the new input argument of output_interval is "
                       "preferred.\n";
            }
        }
 
        // Because of default, interval is valid unless turned off;
        // time_interval is invalid unless turned on
        const bool has_valid_interval =
            (has_interval || has_frequency) ? m_out_interval > 0 : true;
        const bool has_valid_interval_present =
            (has_interval || has_frequency) ? m_out_interval > 0 : false;
        const bool has_valid_time_int =
            has_time_int ? m_out_time_interval > 0. : false;
 
        if (!(has_valid_interval || has_valid_time_int)) {
            amrex::Abort(
                "post_processing output timing parameters are specified, but "
                "none are valid. Please specify an output_interval (or "
                "output_frequency) greater than 0 or an output_time_interval "
                "greater than 0.0_rt, or rely on the default output_interval.");
        }
 
        if (has_valid_interval_present && has_valid_time_int) {
            amrex::Abort(
                "output_interval (or output_frequency) and "
                "output_time_interval are both specified. timestep- and "
                "time-based output parameters should not be used together; "
                "please only specify one.");
        }
 
        if (!has_valid_time_int && m_enforce_dt) {
            amrex::Abort(
                "enforce_output_time_dt is true, but output_time_interval "
                "has not been specified.");
        }
 
        // Because of default, turn off step interval when time interval is
        // meant to replace it
        if (has_valid_time_int && !(has_interval || has_frequency)) {
            m_out_interval = -1;
        }
    }

    bool do_output_now(
        const int ntime,
        const amrex::Real time,
        const amrex::Real dt,
        amrex::Real tol) const
    {
        // Choose smallest tolerance between incoming and local
        tol = amrex::min<amrex::Real>(tol, m_out_time_tol * dt);
 
        const bool step_based =
            (m_out_interval > 0) &&
            ((ntime - m_out_delay >= 0) && (ntime % m_out_interval == 0));
        const bool time_based =
            (m_out_time_interval > 0.0_rt) &&
            ((time + tol - m_out_time_delay >= 0.0_rt) &&
             (time + tol -
                  std::floor((time + tol) / m_out_time_interval) *
                      m_out_time_interval <
              dt));
 
        return (step_based || time_based);
    }
 
    amrex::Real output_time_interval() const { return m_out_time_interval; }
    amrex::Real output_time_delay() const { return m_out_time_delay; }
    amrex::Real enforce_dt_tolerance() const { return m_enforce_dt_tol; }
    bool enforce_dt() const { return m_enforce_dt; }
 
protected:
    int m_out_interval{10};
    amrex::Real m_out_time_interval{-1.0_rt};
    int m_out_delay{0};
    amrex::Real m_out_time_delay{0.0_rt};
    amrex::Real m_out_time_tol{
        std::numeric_limits<amrex::Real>::epsilon() * 1.0e4_rt};

    amrex::Real m_enforce_dt_tol{1.0e-3_rt};
    bool m_enforce_dt{false};
};

class PostProcessManager
{
public:
    explicit PostProcessManager(CFDSim& sim);
 
    ~PostProcessManager() = default;
 
    void pre_init_actions();

    void post_init_actions();

    void post_advance_work();

    void final_output();
 
    void post_regrid_actions();
 
private:
    CFDSim& m_sim;
 
    amrex::Vector<std::unique_ptr<PostProcessBase>> m_post;
};
 
} // namespace amr_wind
 
#endif /* POSTPROCESSING_H */