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 <AMReX_Array.H>
#include <AMReX_MultiFab.H>
#include <AMReX_REAL.H>
 
#include "ImplicitSolver.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 ) override;
 
    void PrintParameters () const override;
 
    void OneStep ( amrex::Real  a_time,
                   amrex::Real  a_dt,
                   int          a_step ) override;
 
    void ComputeRHS ( WarpXSolverVec&  a_RHS,
                const WarpXSolverVec&  a_E,
                      amrex::Real      a_time,
                      amrex::Real      a_dt,
                      int              a_nl_iter,
                      bool             a_from_jacobian ) override;
 
    [[nodiscard]] amrex::Real theta () const { return m_theta; }
 
private:
 
    amrex::Real m_theta = 0.5;
 
    WarpXSolverVec m_E, m_Eold;
 
    amrex::Vector<std::array< std::unique_ptr<amrex::MultiFab>, 3 > > m_Bold;
 
    void UpdateWarpXFields ( const WarpXSolverVec&  a_E,
                             amrex::Real            a_time,
                             amrex::Real            a_dt );
 
    void FinishFieldUpdate ( amrex::Real  a_new_time );
 
};
 
#endif