SpectralSolver.H

Go to the documentation of this file.

/* Copyright 2019-2020 Maxence Thevenet, Remi Lehe, Edoardo Zoni
 *
 * This file is part of WarpX.
 *
 * License: BSD-3-Clause-LBNL
 */
#ifndef WARPX_SPECTRAL_SOLVER_H_
#define WARPX_SPECTRAL_SOLVER_H_
 
#include "SpectralSolver_fwd.H"
 
#include "SpectralAlgorithms/SpectralBaseAlgorithm.H"
#include "SpectralFieldData.H"
#include "Utils/WarpXAlgorithmSelection.H"
 
#include <ablastr/fields/MultiFabRegister.H>
#include <ablastr/utils/Enums.H>
 
#include <AMReX_Array.H>
#include <AMReX_REAL.H>
#include <AMReX_RealVect.H>
 
#include <AMReX_BaseFwd.H>
 
#include <array>
#include <memory>
 
#ifdef WARPX_USE_FFT
class SpectralSolver
{
    public:

        SpectralSolver (int lev,
                        const amrex::BoxArray& realspace_ba,
                        const amrex::DistributionMapping& dm,
                        int norder_x,
                        int norder_y,
                        int norder_z,
                        ablastr::utils::enums::GridType grid_type,
                        const amrex::Vector<amrex::Real>& v_galilean,
                        const amrex::Vector<amrex::Real>& v_comoving,
                        amrex::RealVect dx,
                        amrex::Real dt,
                        bool pml,
                        bool periodic_single_box,
                        bool update_with_rho,
                        bool fft_do_time_averaging,
                        PSATDSolutionType psatd_solution_type,
                        TimeDependencyJ time_dependency_J,
                        TimeDependencyRho time_dependency_rho,
                        bool dive_cleaning,
                        bool divb_cleaning);

        void ForwardTransform (int lev,
                               const amrex::MultiFab& mf,
                               int field_index,
                               int i_comp = 0);

        void BackwardTransform( int lev,
                                amrex::MultiFab& mf,
                                int field_index,
                                const amrex::IntVect& fill_guards,
                                int i_comp=0 );

        void pushSpectralFields();

        void ComputeSpectralDivE (
            int lev,
            ablastr::fields::VectorField const & Efield,
            amrex::MultiFab& divE);

        void CurrentCorrection ();

        void VayDeposition ();

        void CopySpectralDataComp (int src_comp, int dest_comp);

        void ZeroOutDataComp (int icomp);

        void ScaleDataComp (int icomp, amrex::Real scale_factor);
 
        SpectralFieldIndex m_spectral_index;
 
        // Solve time step size
        amrex::Real m_dt;
 
    protected:
 
        amrex::IntVect m_fill_guards;
 
    private:
 
        void ReadParameters ();
 
        // Store field in spectral space and perform the Fourier transforms
        SpectralFieldData field_data;
 
        // Defines field update equation in spectral space and the associated coefficients.
        // SpectralBaseAlgorithm is a base class; this pointer is meant to point
        // to an instance of a sub-class defining a specific algorithm
        std::unique_ptr<SpectralBaseAlgorithm> algorithm;
};
#endif // WARPX_USE_FFT
#endif // WARPX_SPECTRAL_SOLVER_H_