SecondMomentAveraging.H

Go to the documentation of this file.

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

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

    [[nodiscard]] amrex::Real
    line_average_interpolated(amrex::Real x, int comp1, int comp2) 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) const;
    [[nodiscard]] amrex::Real line_average_cell(int ind, int comp) const;
 
    const amrex::Vector<amrex::Real>& line_moment()
    {
        return m_second_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_second_moments_line; 
 
    int m_precision = 4;           
    int m_last_updated_index = -1; 
 
    FieldPlaneAveraging& m_plane_average1;
    FieldPlaneAveraging& m_plane_average2;
 
public: // public for GPU
    template <typename IndexSelector>
    void compute_average(const IndexSelector& idxOp);
};
 
} // namespace kynema_sgf
 
#endif /* SecondMomentAveraging_H */