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 * const mf_src, const int lev,
                           const int ncomp, const int num_buffers,
                           amrex::Vector< std::string > varnames,
                           const amrex::IntVect crse_ratio= amrex::IntVect(1));

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

    void PrepareFunctorData ( int i_buffer, bool z_slice_in_domain,
                              amrex::Real current_z_boost,
                              amrex::Box buffer_box, const int k_index_zlab,
                              const int max_box_size, const int snapshot_full ) override;
    void InitData () override;
    void LorentzTransformZ (amrex::MultiFab& data, amrex::Real gamma_boost,
                           amrex::Real beta_boost) const;
private:
    amrex::MultiFab const * const m_mf_src = nullptr;
    int const m_lev;
    int const 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;

    int m_max_box_size = 256;
    amrex::Vector<int> m_map_varnames;
};

#endif