ThirdMomentAveraging.H

Go to the documentation of this file.

#ifndef THIRDMOMENTAVERAGING_H
#define THIRDMOMENTAVERAGING_H
 
#include "AMReX_AmrCore.H"
#include "src/utilities/DirectionSelector.H"
#include "src/core/Field.H"
#include "src/utilities/FieldPlaneAveraging.H"
 
namespace kynema_sgf {

class ThirdMomentAveraging
{
public:
    ThirdMomentAveraging(
        FieldPlaneAveraging& pa1,
        FieldPlaneAveraging& pa2,
        FieldPlaneAveraging& pa3);
 
    ~ThirdMomentAveraging() = default;
 
    void operator()();

    [[nodiscard]] amrex::Real line_average_interpolated(
        amrex::Real x, int comp1, int comp2, int comp3) const;
    [[nodiscard]] amrex::Real
    line_average_interpolated(amrex::Real x, int comp) const;
    [[nodiscard]] amrex::Real
    line_average_cell(int ind, int comp1, int comp2, int comp3) const;
    [[nodiscard]] amrex::Real line_average_cell(int ind, int comp) const;
 
    const amrex::Vector<amrex::Real>& line_moment()
    {
        return m_third_moments_line;
    };
    void line_moment(int comp, amrex::Vector<amrex::Real>& l_vec);
 
    void output_line_average_ascii(
        const std::string& filename, int step, amrex::Real time);
    void output_line_average_ascii(int step, amrex::Real time);

    void set_precision(int p) { m_precision = p; };
 
private:
    int m_num_moments; 
    amrex::Vector<amrex::Real>
        m_third_moments_line; 
 
    int m_precision = 4;           
    int m_last_updated_index = -1; 
 
    FieldPlaneAveraging& m_plane_average1;
    FieldPlaneAveraging& m_plane_average2;
    FieldPlaneAveraging& m_plane_average3;
 
public: // public for GPU
    template <typename IndexSelector>
    void compute_average(
        const IndexSelector& idxOp,
        const amrex::MultiFab& mfab1,
        const amrex::MultiFab& mfab2,
        const amrex::MultiFab& mfab3);
};
 
} // namespace kynema_sgf
 
#endif /* ThirdMomentAveraging_H */