ThetaImplicitEM.H

Go to the documentation of this file.

/* Copyright 2024 Justin Angus
 *
 * This file is part of WarpX.
 *
 * License: BSD-3-Clause-LBNL
 */
#ifndef THETA_IMPLICIT_EM_H_
#define THETA_IMPLICIT_EM_H_
 
#include "FieldSolver/ImplicitSolvers/WarpXSolverVec.H"
#include "ImplicitSolver.H"
 
#include <AMReX_Array.H>
#include <AMReX_MultiFab.H>
#include <AMReX_REAL.H>

 
class ThetaImplicitEM : public ImplicitSolver
{
public:
 
    ThetaImplicitEM() = default;
 
    ~ThetaImplicitEM() override = default;
 
    // Prohibit Move and Copy operations
    ThetaImplicitEM(const ThetaImplicitEM&) = delete;
    ThetaImplicitEM& operator=(const ThetaImplicitEM&) = delete;
    ThetaImplicitEM(ThetaImplicitEM&&) = delete;
    ThetaImplicitEM& operator=(ThetaImplicitEM&&) = delete;
 
    void Define (WarpX*  a_WarpX, bool  from_restart) override;
 
    void PrintParameters () const override;

    void OneStep ( amrex::Real  start_time,
                   amrex::Real  a_dt,
                   int          a_step ) override;
 
    void ComputeRHS ( WarpXSolverVec&  a_RHS,
                const WarpXSolverVec&  a_E,
                      amrex::Real      start_time,
                      int              a_nl_iter,
                      bool             a_from_jacobian ) override;
 
    // This parameter is used for the time-step fraction in the PC for implicit
    // treatment of light waves in the curl-curl MLMG solver.
    // This function should return zero if light waves are not treated implicitly
    [[nodiscard]] amrex::Real  GetThetaForPC () const override { return m_theta; }

    [[nodiscard]] const amrex::MultiFab* GetCurl2BCmask (int lev, int field_dir) const override;

    void InitializeCurlCurlBCMasks ();
 
private:

    WarpXSolverVec m_E, m_Eold;

    void UpdateWarpXFields ( const WarpXSolverVec&  a_E,
                             amrex::Real            start_time );

    void FinishFieldUpdate ( amrex::Real end_time );
 
};
 
#endif