BackTransformFunctor.H

Go to the documentation of this file.

/* Copyright 2021 Revathi Jambunathan
 *
 * This file is part of WarpX.
 *
 * License: BSD-3-Clause-LBNL
 */
#ifndef WARPX_BACKTRANSFORMFUNCTOR_H_
#define WARPX_BACKTRANSFORMFUNCTOR_H_
 
#include "ComputeDiagFunctor.H"
 
#include <AMReX_Box.H>
#include <AMReX_IntVect.H>
#include <AMReX_REAL.H>
#include <AMReX_Vector.H>
 
#include <AMReX_BaseFwd.H>
 
#include <string>

 
class BackTransformFunctor final : public ComputeDiagFunctor
{
public:
    BackTransformFunctor ( const amrex::MultiFab * mf_src, int lev,
                           int ncomp, int num_buffers,
                           amrex::Vector< std::string > varnames,
                           amrex::Vector< std::string > varnames_fields,
                           amrex::IntVect crse_ratio= amrex::IntVect(1));

    void operator ()(amrex::MultiFab& mf_dst, int dcomp, int i_buffer) const override;

    void PrepareFunctorData ( int i_buffer, bool z_slice_in_domain,
                              amrex::Real current_z_boost,
                              amrex::Box buffer_box, int k_index_zlab,
                              int snapshot_full ) override;
    void InitData () override;
    void LorentzTransformZ (amrex::MultiFab& data, amrex::Real gamma_boost,
                           amrex::Real beta_boost) const;
private:
    const amrex::MultiFab* m_mf_src = nullptr;
    int m_lev;
    int m_num_buffers;
    amrex::Vector<amrex::Box> m_buffer_box;
    amrex::Vector<amrex::Real> m_current_z_boost;
    amrex::Vector<int> m_perform_backtransform;
    amrex::Vector<int> m_k_index_zlab;
    amrex::Vector< std::string > m_varnames;
    amrex::Vector< std::string > m_varnames_fields;

    amrex::Vector<int> m_map_varnames;
};
 
#endif