Tensor solver with shear/bulk viscosity inputs for high-Reynolds flows. More...

#include <AMReX_MLTensorOp.H>

Inheritance diagram for amrex::MLTensorOp:

Public Member Functions
	MLTensorOp ()
	Construct an empty tensor operator; call define() before solving.

	MLTensorOp (const Vector< Geometry > &a_geom, const Vector< BoxArray > &a_grids, const Vector< DistributionMapping > &a_dmap, const LPInfo &a_info=LPInfo(), const Vector< FabFactory< FArrayBox > const * > &a_factory={})
	Convenience constructor without overset masks (for cell-only domains).

	MLTensorOp (const Vector< Geometry > &a_geom, const Vector< BoxArray > &a_grids, const Vector< DistributionMapping > &a_dmap, const Vector< iMultiFab const * > &a_overset_mask, const LPInfo &a_info=LPInfo(), const Vector< FabFactory< FArrayBox > const * > &a_factory={})
	Convenience constructor that includes overset-mask support (1 = unknown, 0 = known).

	~MLTensorOp () override=default

	MLTensorOp (const MLTensorOp &)=delete

	MLTensorOp (MLTensorOp &&)=delete

MLTensorOp &	operator= (const MLTensorOp &)=delete

MLTensorOp &	operator= (MLTensorOp &&)=delete

void	define (const Vector< Geometry > &a_geom, const Vector< BoxArray > &a_grids, const Vector< DistributionMapping > &a_dmap, const LPInfo &a_info=LPInfo(), const Vector< FabFactory< FArrayBox > const * > &a_factory={})
	Define the hierarchy for standard tensor solves.

void	define (const Vector< Geometry > &a_geom, const Vector< BoxArray > &a_grids, const Vector< DistributionMapping > &a_dmap, const Vector< iMultiFab const * > &a_overset_mask, const LPInfo &a_info=LPInfo(), const Vector< FabFactory< FArrayBox > const * > &a_factory={})
	Define the hierarchy with overset masks (1 = unknown, 0 = known).

void	setShearViscosity (int amrlev, const Array< MultiFab const *, 3 > &eta)
	Provide shear viscosity `eta` as face-centered MultiFabs.

void	setShearViscosity (int amrlev, Real eta)
	Populate `eta` with a constant scalar.

void	setBulkViscosity (int amrlev, const Array< MultiFab const *, 3 > &kappa)
	Provide bulk viscosity `kappa` as face-centered MultiFabs.

void	setBulkViscosity (int amrlev, Real kappa)
	Populate `kappa` with a constant scalar.

int	getNComp () const final
	Return number of components.

bool	isCrossStencil () const final
	Whether the stencil is the cross shape.

bool	isTensorOp () const final
	Whether this operator is a tensor solve.

bool	needsUpdate () const final
	True if viscosity coefficients need to be rebuilt before apply().

void	update () final
	Rebuild coefficient averages (not yet implemented).

void	prepareForSolve () final
	Assemble viscosity data/BC caches before solving.

bool	isSingular (int) const final
	Query whether AMR level `amrlev` is singular (null space present).

bool	isBottomSingular () const final
	Convenience helper for the coarsest level singularity flag.

void	apply (int amrlev, int mglev, MultiFab &out, MultiFab &in, BCMode bc_mode, StateMode s_mode, const MLMGBndry *bndry=nullptr) const final
	Apply the tensor operator on AMR level `amrlev` / MG level `mglev`.

void	compFlux (int amrlev, const Array< MultiFab *, 3 > &fluxes, MultiFab &sol, Location loc) const override
	Assemble viscous fluxes on AMR level `amrlev`.

void	compVelGrad (int amrlev, const Array< MultiFab *, 3 > &fluxes, MultiFab &sol, Location loc) const
	Compute velocity gradients (symmetrized) at the requested location.

void	setBCoeffs (int amrlev, const Array< MultiFab const *, 3 > &beta)=delete

void	applyBCTensor (int amrlev, int mglev, MultiFab &vel, BCMode bc_mode, StateMode s_mode, const MLMGBndry *bndry) const
	Apply tensor-specific boundary conditions on (`amrlev`,`mglev`).

Public Member Functions inherited from amrex::MLABecLaplacianT< MF >
	MLABecLaplacianT ()=default
	Construct an empty operator; call define() before solving.

	MLABecLaplacianT (const Vector< Geometry > &a_geom, const Vector< BoxArray > &a_grids, const Vector< DistributionMapping > &a_dmap, const LPInfo &a_info=LPInfo(), const Vector< FabFactory< FAB > const * > &a_factory={}, int a_ncomp=1)
	Convenience constructor for cell-based solves without overset regions.

	MLABecLaplacianT (const Vector< Geometry > &a_geom, const Vector< BoxArray > &a_grids, const Vector< DistributionMapping > &a_dmap, const Vector< iMultiFab const * > &a_overset_mask, const LPInfo &a_info=LPInfo(), const Vector< FabFactory< FAB > const * > &a_factory={}, int a_ncomp=1)
	Convenience constructor that includes overset-mask handling (1 = unknown, 0 = known).

	~MLABecLaplacianT () override

	MLABecLaplacianT (const MLABecLaplacianT< MF > &)=delete

	MLABecLaplacianT (MLABecLaplacianT< MF > &&)=delete

MLABecLaplacianT< MF > &	operator= (const MLABecLaplacianT< MF > &)=delete

MLABecLaplacianT< MF > &	operator= (MLABecLaplacianT< MF > &&)=delete

void	define (const Vector< Geometry > &a_geom, const Vector< BoxArray > &a_grids, const Vector< DistributionMapping > &a_dmap, const LPInfo &a_info=LPInfo(), const Vector< FabFactory< FAB > const * > &a_factory={}, int a_ncomp=1)
	Define coefficients/layouts for a standard cell-centered solve.

void	define (const Vector< Geometry > &a_geom, const Vector< BoxArray > &a_grids, const Vector< DistributionMapping > &a_dmap, const Vector< iMultiFab const * > &a_overset_mask, const LPInfo &a_info=LPInfo(), const Vector< FabFactory< FAB > const * > &a_factory={}, int a_ncomp=1)
	Define coefficients/layouts plus overset-mask information.

template<typename T1 , typename T2 > requires (std::is_convertible_v<T1, typename MF::value_type> && std::is_convertible_v<T2, typename MF::value_type>)
void	setScalars (T1 a, T2 b) noexcept

template<FabArrayType AMF> requires (std::is_convertible_v<typename AMF::value_type, typename MF::value_type>)
void	setACoeffs (int amrlev, const AMF &alpha)

template<typename T > requires (std::is_convertible_v<T, typename MF::value_type>)
void	setACoeffs (int amrlev, T alpha)

template<FabArrayType AMF> requires (std::is_convertible_v<typename AMF::value_type, typename MF::value_type>)
void	setBCoeffs (int amrlev, const Array< AMF const *, 3 > &beta)

template<typename T > requires (std::is_convertible_v<T, typename MF::value_type>)
void	setBCoeffs (int amrlev, T beta)

template<typename T > requires (std::is_convertible_v<T, typename MF::value_type>)
void	setBCoeffs (int amrlev, Vector< T > const &beta)

void	Fapply (int amrlev, int mglev, MF &out, const MF &in) const override
	Apply the operator on (`amrlev`,`mglev`), storing L(`in`) in `out`.

void	Fsmooth (int amrlev, int mglev, MF &sol, const MF &rhs, int redblack) const override
	Perform one smoothing pass on (`amrlev`,`mglev`) updating `sol` against `rhs`. `redblack` selects the red (0) or black (1) half of the grid.

void	FFlux (int amrlev, const MFIter &mfi, const Array< FAB *, 3 > &flux, const FAB &sol, Location, int face_only=0) const override
	Compute fluxes on AMR level `amrlev` for the tilebox described by `mfi` using `sol`, writing to `flux` and honoring `face_only`.

void	normalize (int amrlev, int mglev, MF &mf) const override
	Divide `mf` by the diagonal of the operator (used by CG-family bottom solvers).

RT	getAScalar () const final
	Scalar alpha applied to the a-coefficient term.

RT	getBScalar () const final
	Scalar beta applied to the b-coefficient term.

MF const *	getACoeffs (int amrlev, int mglev) const final
	Access the stored `a` coefficient on AMR level `amrlev` and MG level `mglev`.

Array< MF const *, 3 >	getBCoeffs (int amrlev, int mglev) const final
	Access the stored `b` coefficients on AMR level `amrlev` and MG level `mglev`.

std::unique_ptr< MLLinOpT< MF > >	makeNLinOp (int) const final
	Build the NSolve counterpart of this operator.

bool	supportNSolve () const override
	Whether this operator supports NSolve.

void	copyNSolveSolution (MF &dst, MF const &src) const final
	Copy an NSolve solution from `src` to `dst`.

void	averageDownCoeffsSameAmrLevel (int amrlev, Vector< MF > &a, Vector< Array< MF, 3 > > &b)
	Average coefficients down within AMR level `amrlev` (fine-to-coarse multigrid) updating `a` and `b`.

void	averageDownCoeffs ()
	Average `a` and `b` coefficients down across all AMR and MG levels.

void	averageDownCoeffsToCoarseAmrLevel (int flev)
	Average coefficients from fine AMR level `flev` to coarse AMR level `flev-1`.

void	applyMetricTermsCoeffs ()
	Apply metric factors to the stored coefficients when solving in mapped space.

void	applyRobinBCTermsCoeffs ()
	Modify coefficients to honor Robin BC terms introduced at level boundaries.

Public Member Functions inherited from amrex::MLCellABecLapT< MF >
	MLCellABecLapT ()=default

	~MLCellABecLapT () override=default

	MLCellABecLapT (const MLCellABecLapT< MF > &)=delete

	MLCellABecLapT (MLCellABecLapT< MF > &&)=delete

MLCellABecLapT< MF > &	operator= (const MLCellABecLapT< MF > &)=delete

MLCellABecLapT< MF > &	operator= (MLCellABecLapT< MF > &&)=delete

void	define (const Vector< Geometry > &a_geom, const Vector< BoxArray > &a_grids, const Vector< DistributionMapping > &a_dmap, const LPInfo &a_info=LPInfo(), const Vector< FabFactory< FAB > const * > &a_factory={})
	Describe the AMR hierarchy when overset masks are not required.

void	define (const Vector< Geometry > &a_geom, const Vector< BoxArray > &a_grids, const Vector< DistributionMapping > &a_dmap, const Vector< iMultiFab const * > &a_overset_mask, const LPInfo &a_info=LPInfo(), const Vector< FabFactory< FAB > const * > &a_factory={})
	Describe the AMR hierarchy when overset masks participate.

iMultiFab const *	getOversetMask (int amrlev, int mglev) const
	Overset mask for (`amrlev`,`mglev`); returns `nullptr` when not defined.

void	setDirichletNodesToZero (int amrlev, int mglev, MF &mf) const override
	Zero out Dirichlet nodes on (`amrlev`,`mglev`) so that GMRES can treat them as known.

void	getFluxes (const Vector< Array< MF , 3 > > &a_flux, const Vector< MF > &a_sol, Location a_loc) const final
	Fill per-face fluxes using the supplied solution hierarchy.

void	getFluxes (const Vector< MF * > &a_flux, const Vector< MF * > &a_sol) const final
	Guard overload that aborts if called (cell-centered flux extraction requires per-direction face arrays).

void	applyInhomogNeumannTerm (int amrlev, MF &rhs) const final
	Apply stored Neumann data to the RHS `rhs` on AMR level `amrlev`.

void	addInhomogNeumannFlux (int amrlev, const Array< MF *, 3 > &grad, MF const &sol, bool mult_bcoef) const final
	Add inhomogeneous Neumann/Robin flux contributions into `grad` from `sol` (include `b` when `mult_bcoef` is true).

void	applyOverset (int amrlev, MF &rhs) const override
	Zero RHS entries in `rhs` that are covered by overset masks on level `amrlev`.

Public Member Functions inherited from amrex::MLCellLinOpT< MF >
	MLCellLinOpT ()

	~MLCellLinOpT () override=default

	MLCellLinOpT (const MLCellLinOpT< MF > &)=delete

	MLCellLinOpT (MLCellLinOpT< MF > &&)=delete

MLCellLinOpT< MF > &	operator= (const MLCellLinOpT< MF > &)=delete

MLCellLinOpT< MF > &	operator= (MLCellLinOpT< MF > &&)=delete

void	define (const Vector< Geometry > &a_geom, const Vector< BoxArray > &a_grids, const Vector< DistributionMapping > &a_dmap, const LPInfo &a_info=LPInfo(), const Vector< FabFactory< FAB > const * > &a_factory={})
	Bind the operator to an AMR hierarchy.

void	setLevelBC (int amrlev, const MF levelbcdata, const MF robinbc_a=nullptr, const MF robinbc_b=nullptr, const MF robinbc_f=nullptr) final
	Provide per-level inhomogeneous boundary data.

template<MultiFabLike AMF> requires (!std::same_as<MF,AMF>)
void	setLevelBC (int amrlev, const AMF levelbcdata, const AMF robinbc_a=nullptr, const AMF robinbc_b=nullptr, const AMF robinbc_f=nullptr)

void	setGaussSeidel (bool flag) noexcept
	Toggle Gauss–Seidel smoothing in place of Jacobi relaxation.

void	updateSolBC (int amrlev, const MF &crse_bcdata) const
	Refresh stored solution BC data from coarse inputs.

void	updateCorBC (int amrlev, const MF &crse_bcdata) const
	Refresh stored correction BC data from coarse inputs.

virtual void	applyBC (int amrlev, int mglev, MF &in, BCMode bc_mode, StateMode s_mode, const MLMGBndryT< MF > *bndry=nullptr, bool skip_fillboundary=false) const
	Apply physical BCs (optionally skipping FillBoundary).

BoxArray	makeNGrids (int grid_size) const
	Helper that builds a BoxArray for NSolve with boxes no larger than the requested `grid_size`.

void	restriction (int amrlev, int cmglev, MF &crse, MF &fine) const override
	Restrict a fine-grid field onto its coarse counterpart.

void	interpolation (int amrlev, int fmglev, MF &fine, const MF &crse) const override
	Add the prolongation of coarse data onto the fine grid (`fine` += prolong(`crse`)).

void	interpAssign (int amrlev, int fmglev, MF &fine, MF &crse) const override
	Overwrite fine data with the prolongation of coarse data (`fine` = prolong(`crse`)).

void	interpolationAmr (int famrlev, MF &fine, const MF &crse, IntVect const &nghost) const override
	Prolong AMR-level data during FMG initialization.

void	averageDownSolutionRHS (int camrlev, MF &crse_sol, MF &crse_rhs, const MF &fine_sol, const MF &fine_rhs) override
	Average fine solution/RHS onto the next coarser AMR level.

void	apply (int amrlev, int mglev, MF &out, MF &in, BCMode bc_mode, StateMode s_mode, const MLMGBndryT< MF > *bndry=nullptr) const override
	Apply the linear operator with boundary conditions.

void	smooth (int amrlev, int mglev, MF &sol, const MF &rhs, bool skip_fillboundary, int niter) const final
	Perform `niter` smoothing iterations on the supplied residual equation.

void	solutionResidual (int amrlev, MF &resid, MF &x, const MF &b, const MF *crse_bcdata=nullptr) override
	Compute the residual `resid = b - A(x)` using solution boundary data.

void	prepareForFluxes (int amrlev, const MF *crse_bcdata=nullptr) override
	Ensure BC caches are ready for flux computations (e.g., getFluxes).

void	correctionResidual (int amrlev, int mglev, MF &resid, MF &x, const MF &b, BCMode bc_mode, const MF *crse_bcdata=nullptr) final
	Compute the correction residual with optional coarse data.

void	reflux (int crse_amrlev, MF &res, const MF &crse_sol, const MF &, MF &, MF &fine_sol, const MF &) const final
	Reflux fine-level fluxes into the coarse residual.

void	compFlux (int amrlev, const Array< MF *, 3 > &fluxes, MF &sol, Location loc) const override
	Compute face-centered fluxes from the supplied solution.

void	compGrad (int amrlev, const Array< MF *, 3 > &grad, MF &sol, Location loc) const override
	Compute directional gradients of the solution.

void	applyMetricTerm (int amrlev, int mglev, MF &rhs) const final
	Multiply the RHS by metric terms appropriate for curvilinear coordinates.

void	unapplyMetricTerm (int amrlev, int mglev, MF &rhs) const final
	Remove metric scaling previously applied to the RHS.

Vector< RT >	getSolvabilityOffset (int amrlev, int mglev, MF const &rhs) const override
	Compute the average offset needed to enforce solvability constraints.

void	fixSolvabilityByOffset (int amrlev, int mglev, MF &rhs, Vector< RT > const &offset) const override
	Apply solvability offsets to the RHS (subtracting the average).

RT	xdoty (int amrlev, int mglev, const MF &x, const MF &y, bool local) const final
	Dot product helper.

RT	dotProductPrecond (Vector< MF const * > const &x, Vector< MF const * > const &y) const final
	Dot product over the composite AMR hierarchy, excluding cells covered by finer levels (used when the operator is a preconditioner).

RT	norm2Precond (Vector< MF const * > const &x) const final
	L2 norm over the composite AMR hierarchy, excluding cells covered by finer levels (used when the operator is a preconditioner).

virtual void	FFlux (int amrlev, const MFIter &mfi, const Array< FAB *, 3 > &flux, const FAB &sol, Location loc, int face_only=0) const =0

RT	normInf (int amrlev, MF const &mf, bool local) const override
	Infinity norm helper used by solvers and diagnostics.

void	averageDownAndSync (Vector< MF > &sol) const override
	Average the solution hierarchy down (fine-to-coarse) and sync.

void	avgDownResAmr (int clev, MF &cres, MF const &fres) const override
	Average a residual from a fine AMR level to its coarse parent.

void	beginPrecondBC () override
	Called when the operator starts being used as a preconditioner.

void	endPrecondBC () override
	Called when the operator stops being used as a preconditioner.

void	setInterpBndryHalfWidth (int w)
	Control how many cells the interpolation boundary stencil spans.

Public Member Functions inherited from amrex::MLLinOpT< MF >
	MLLinOpT ()=default

virtual	~MLLinOpT ()=default

	MLLinOpT (const MLLinOpT< MF > &)=delete

	MLLinOpT (MLLinOpT< MF > &&)=delete

MLLinOpT< MF > &	operator= (const MLLinOpT< MF > &)=delete

MLLinOpT< MF > &	operator= (MLLinOpT< MF > &&)=delete

void	define (const Vector< Geometry > &a_geom, const Vector< BoxArray > &a_grids, const Vector< DistributionMapping > &a_dmap, const LPInfo &a_info, const Vector< FabFactory< FAB > const * > &a_factory, bool eb_limit_coarsening=true)
	Initialize the operator hierarchy on a set of AMR levels.

virtual std::string	name () const

void	setDomainBC (const Array< BCType, 3 > &lobc, const Array< BCType, 3 > &hibc) noexcept
	Boundary of the whole domain.

void	setDomainBC (const Vector< Array< BCType, 3 > > &lobc, const Vector< Array< BCType, 3 > > &hibc)
	Boundary of the whole domain.

void	setDomainBCLoc (const Array< Real, 3 > &lo_bcloc, const Array< Real, 3 > &hi_bcloc) noexcept
	Set location offsets for the physical domain boundaries.

bool	needsCoarseDataForBC () const noexcept
	Needs coarse data for bc?

void	setCoarseFineBC (const MF *crse, int crse_ratio, LinOpBCType bc_type=LinOpBCType::Dirichlet) noexcept
	Set coarse/fine boundary conditions. For cell-centered solves only.

void	setCoarseFineBC (const MF *crse, IntVect const &crse_ratio, LinOpBCType bc_type=LinOpBCType::Dirichlet) noexcept

template<typename AMF > requires (!std::same_as<MF,AMF>)
void	setCoarseFineBC (const AMF *crse, int crse_ratio, LinOpBCType bc_type=LinOpBCType::Dirichlet) noexcept

template<typename AMF > requires (!std::same_as<MF,AMF>)
void	setCoarseFineBC (const AMF *crse, IntVect const &crse_ratio, LinOpBCType bc_type=LinOpBCType::Dirichlet) noexcept

template<MultiFabLike AMF> requires (!std::same_as<MF,AMF>)
void	setLevelBC (int amrlev, const AMF levelbcdata, const AMF robinbc_a=nullptr, const AMF robinbc_b=nullptr, const AMF robinbc_f=nullptr)

void	setVerbose (int v) noexcept
	Set verbosity.

void	setMaxOrder (int o) noexcept
	Set order of interpolation at coarse/fine boundary.

int	getMaxOrder () const noexcept
	Get order of interpolation at coarse/fine boundary.

void	setEnforceSingularSolvable (bool o) noexcept
	Control whether the solver should try to make singular problems solvable.

bool	getEnforceSingularSolvable () const noexcept

virtual BottomSolver	getDefaultBottomSolver () const

virtual int	getNGrow (int=0, int=0) const

virtual void	unimposeNeumannBC (int amrlev, MF &rhs) const
	Undo Neumann contributions stored on the RHS.

virtual bool	scaleRHS (int amrlev, MF *rhs) const
	Optionally scale the RHS to fix solvability.

virtual void	preparePrecond ()
	Prepare auxiliary data used when the operator acts as a preconditioner.

virtual void	getEBFluxes (const Vector< MF * > &a_flux, const Vector< MF * > &a_sol) const
	Extract embedded-boundary fluxes.

virtual void	postSolve (Vector< MF * > const &sol) const
	Optional hook invoked after the main solve completes.

virtual void	avgDownResMG (int clev, MF &cres, MF const &fres) const
	Average residuals from fine to coarse MG levels (FMG helper).

bool	isMFIterSafe (int amrlev, int mglev1, int mglev2) const
	Check whether mixing MFIter loops for different MG levels is safe.

int	NAMRLevels () const noexcept
	Return the number of AMR levels.

int	NMGLevels (int amrlev) const noexcept
	Return the number of MG levels at given AMR level.

const Geometry &	Geom (int amr_lev, int mglev=0) const noexcept
	Geometry accessor for (`amr_lev`,`mglev`).

Protected Attributes
bool	m_has_kappa = false

Vector< Vector< Array< MultiFab, 3 > > >	m_kappa

Protected Attributes inherited from amrex::MLABecLaplacianT< MF >
Vector< int >	m_is_singular

Protected Attributes inherited from amrex::MLCellABecLapT< MF >
Vector< Vector< std::unique_ptr< iMultiFab > > >	m_overset_mask

LPInfo	m_lpinfo_arg

Protected Attributes inherited from amrex::MLCellLinOpT< MF >
bool	m_has_metric_term = false

Vector< std::unique_ptr< MLMGBndryT< MF > > >	m_bndry_sol

Vector< std::unique_ptr< BndryRegisterT< MF > > >	m_crse_sol_br

Vector< std::unique_ptr< MLMGBndryT< MF > > >	m_bndry_cor

Vector< std::unique_ptr< BndryRegisterT< MF > > >	m_crse_cor_br

Vector< std::unique_ptr< MLMGBndryT< MF > > >	m_bndry_sol_zero

Vector< Vector< std::unique_ptr< BndryCondLoc > > >	m_bcondloc

Vector< Vector< BndryRegisterT< MF > > >	m_undrrelxr

Vector< Vector< Array< MultiMask, 2 *3 > > >	m_maskvals

Vector< std::unique_ptr< iMultiFab > >	m_norm_fine_mask

Vector< YAFluxRegisterT< MF > >	m_fluxreg

bool	m_use_gauss_seidel = true

Protected Attributes inherited from amrex::MLLinOpT< MF >
int	mg_domain_min_width = 2

LPInfo	info

int	verbose = 0

int	maxorder = 3

bool	enforceSingularSolvable = true

int	m_num_amr_levels = 0

Vector< int >	m_amr_ref_ratio

Vector< int >	m_num_mg_levels

const MLLinOpT< MF > *	m_parent = nullptr

IntVect	m_ixtype

bool	m_do_agglomeration = false

bool	m_do_consolidation = false

bool	m_do_semicoarsening = false

Vector< IntVect >	mg_coarsen_ratio_vec

Vector< Vector< Geometry > >	m_geom
	first Vector is for amr level and second is mg level

Vector< Vector< BoxArray > >	m_grids

Vector< Vector< DistributionMapping > >	m_dmap

Vector< Vector< std::unique_ptr< FabFactory< FAB > > > >	m_factory

Vector< int >	m_domain_covered

MPI_Comm	m_default_comm = MPI_COMM_NULL

MPI_Comm	m_bottom_comm = MPI_COMM_NULL

std::unique_ptr< CommContainer >	m_raii_comm

Array< Real, 3 >	m_domain_bloc_lo {{ 0._rt , 0._rt , 0._rt }}

Array< Real, 3 >	m_domain_bloc_hi {{ 0._rt , 0._rt , 0._rt }}

bool	m_needs_coarse_data_for_bc = false

LinOpBCType	m_coarse_fine_bc_type = LinOpBCType::Dirichlet

IntVect	m_coarse_data_crse_ratio = IntVect(-1)

RealVect	m_coarse_bc_loc

const MF *	m_coarse_data_for_bc = nullptr

MF	m_coarse_data_for_bc_raii

bool	m_precond_mode = false

Additional Inherited Members
Public Types inherited from amrex::MLABecLaplacianT< MF >
using	FAB = typename MF::fab_type

using	RT = typename MF::value_type

using	BCType = LinOpBCType

using	Location = typename MLLinOpT< MF >::Location

Public Types inherited from amrex::MLCellABecLapT< MF >
using	FAB = typename MF::fab_type

using	RT = typename MF::value_type

using	Location = typename MLLinOpT< MF >::Location

Public Types inherited from amrex::MLCellLinOpT< MF >
using	FAB = typename FabDataType< MF >::fab_type

using	RT = typename FabDataType< MF >::value_type

using	BCType = LinOpBCType

using	BCMode = typename MLLinOpT< MF >::BCMode

using	StateMode = typename MLLinOpT< MF >::StateMode

using	Location = typename MLLinOpT< MF >::Location

Public Types inherited from amrex::MLLinOpT< MF >
using	MFType = MF

using	FAB = typename FabDataType< MF >::fab_type

using	RT = typename FabDataType< MF >::value_type

using	BCType = LinOpBCType

using	BCMode = LinOpEnumType::BCMode

using	StateMode = LinOpEnumType::StateMode

using	Location = LinOpEnumType::Location

Static Public Member Functions inherited from amrex::MLABecLaplacianT< MF >
static void	FFlux (Box const &box, Real const dxinv, RT bscalar, Array< FAB const , 3 > const &bcoef, Array< FAB *, 3 > const &flux, FAB const &sol, int face_only, int ncomp)
	Helper that computes face fluxes for a single tilebox.

Public Attributes inherited from amrex::MLABecLaplacianT< MF >
RT	m_a_scalar = std::numeric_limits<RT>::quiet_NaN()

RT	m_b_scalar = std::numeric_limits<RT>::quiet_NaN()

Vector< Vector< MF > >	m_a_coeffs

Vector< Vector< Array< MF, 3 > > >	m_b_coeffs

bool	m_scalars_set = false

bool	m_acoef_set = false

Public Attributes inherited from amrex::MLCellLinOpT< MF >
Vector< std::unique_ptr< MF > >	m_robin_bcval

Public Attributes inherited from amrex::MLLinOpT< MF >
Vector< Array< BCType, 3 > >	m_lobc

Vector< Array< BCType, 3 > >	m_hibc

Vector< Array< BCType, 3 > >	m_lobc_orig

Vector< Array< BCType, 3 > >	m_hibc_orig

Protected Types inherited from amrex::MLCellLinOpT< MF >
using	RealTuple = Array< RT, 2 *3 >

using	BCTuple = Array< BoundCond, 2 *3 >

Protected Member Functions inherited from amrex::MLABecLaplacianT< MF >
bool	supportRobinBC () const noexcept override

Protected Member Functions inherited from amrex::MLCellABecLapT< MF >
bool	supportInhomogNeumannBC () const noexcept override

Protected Member Functions inherited from amrex::MLLinOpT< MF >
const Vector< int > &	AMRRefRatio () const noexcept
	Return AMR refinement ratios.

int	AMRRefRatio (int amr_lev) const noexcept
	Return AMR refinement ratio at given AMR level.

IntVect	AMRRefRatioVect (int amr_lev) const noexcept
	Return AMR refinement ratio as IntVect (1 in hidden direction)

FabFactory< FAB > const *	Factory (int amr_lev, int mglev=0) const noexcept

GpuArray< BCType, 3 >	LoBC (int icomp=0) const noexcept

GpuArray< BCType, 3 >	HiBC (int icomp=0) const noexcept

bool	hasBC (BCType bct) const noexcept

bool	hasInhomogNeumannBC () const noexcept

bool	hasRobinBC () const noexcept

bool	isBottomActive () const noexcept

MPI_Comm	BottomCommunicator () const noexcept

MPI_Comm	Communicator () const noexcept

void	setCoarseFineBCLocation (const RealVect &cloc) noexcept

bool	doAgglomeration () const noexcept

bool	doConsolidation () const noexcept

bool	doSemicoarsening () const noexcept

bool	isCellCentered () const noexcept

virtual IntVect	getNGrowVectRestriction () const

virtual void	make (Vector< Vector< MF > > &mf, IntVect const &ng) const

virtual MF	make (int amrlev, int mglev, IntVect const &ng) const

virtual MF	makeAlias (MF const &mf) const

virtual MF	makeCoarseMG (int amrlev, int mglev, IntVect const &ng) const
	Allocate an `MF` on the next coarser MG level (`mglev+1`) with grow cells `ng`.

virtual MF	makeCoarseAmr (int famrlev, IntVect const &ng) const
	Allocate an `MF` on the next coarser AMR level (`famrlev-1`) with grow cells `ng`.

virtual std::unique_ptr< FabFactory< FAB > >	makeFactory (int, int) const

virtual void	resizeMultiGrid (int new_size)

bool	hasHiddenDimension () const noexcept

int	hiddenDirection () const noexcept

Box	compactify (Box const &b) const noexcept

template<typename T >
Array4< T >	compactify (Array4< T > const &a) const noexcept

template<typename T >
T	get_d0 (T const &dx, T const &dy, T const &) const noexcept

template<typename T >
T	get_d1 (T const &, T const &dy, T const &dz) const noexcept

Static Protected Attributes inherited from amrex::MLLinOpT< MF >
static constexpr int	mg_coarsen_ratio = 2

static constexpr int	mg_box_min_width = 2

Detailed Description

Tensor solver with shear/bulk viscosity inputs for high-Reynolds flows.

Constructor & Destructor Documentation

◆ MLTensorOp() [1/5]

amrex::MLTensorOp::MLTensorOp ( )

Construct an empty tensor operator; call define() before solving.

◆ MLTensorOp() [2/5]

amrex::MLTensorOp::MLTensorOp	(	const Vector< Geometry > &	a_geom,
		const Vector< BoxArray > &	a_grids,
		const Vector< DistributionMapping > &	a_dmap,
		const LPInfo &	a_info = `LPInfo()`,
		const Vector< FabFactory< FArrayBox > const * > &	a_factory = `{}`
	)

Convenience constructor without overset masks (for cell-only domains).

Internally calls define() with the provided metadata.

◆ MLTensorOp() [3/5]

amrex::MLTensorOp::MLTensorOp	(	const Vector< Geometry > &	a_geom,
		const Vector< BoxArray > &	a_grids,
		const Vector< DistributionMapping > &	a_dmap,
		const Vector< iMultiFab const * > &	a_overset_mask,
		const LPInfo &	a_info = `LPInfo()`,
		const Vector< FabFactory< FArrayBox > const * > &	a_factory = `{}`
	)

Convenience constructor that includes overset-mask support (1 = unknown, 0 = known).

◆ ~MLTensorOp()

amrex::MLTensorOp::~MLTensorOp ( )

overridedefault

◆ MLTensorOp() [4/5]

amrex::MLTensorOp::MLTensorOp ( const MLTensorOp & )

delete

◆ MLTensorOp() [5/5]

amrex::MLTensorOp::MLTensorOp ( MLTensorOp && )

delete

Member Function Documentation

◆ apply()

void amrex::MLTensorOp::apply	(	int	amrlev,
		int	mglev,
		MultiFab &	out,
		MultiFab &	in,
		BCMode	bc_mode,
		StateMode	s_mode,
		const MLMGBndry *	bndry = `nullptr`
	)		const

final

Apply the tensor operator on AMR level amrlev / MG level mglev.

Parameters

amrlev	AMR level targeted by the smoothing/apply stage.
mglev	Multigrid level within that AMR level.
out	Destination MultiFab receiving `L(in)`.
in	Source MultiFab holding the field being operated on.
bc_mode	Boundary-condition mode (homogeneous vs inhomogeneous).
s_mode	State mode (solution vs correction).
bndry	Optional boundary metadata supplied by MLMG.

◆ applyBCTensor()

void amrex::MLTensorOp::applyBCTensor	(	int	amrlev,
		int	mglev,
		MultiFab &	vel,
		BCMode	bc_mode,
		StateMode	s_mode,
		const MLMGBndry *	bndry
	)		const

Apply tensor-specific boundary conditions on (amrlev,mglev).

Parameters

amrlev	AMR level index.
mglev	Multigrid level index.
vel	MultiFab updated in place with boundary values.
bc_mode	Boundary-condition mode requested by MLMG.
s_mode	State mode (solution/correction).
bndry	Boundary metadata supplied by MLMG (may be null).

◆ compFlux()

void amrex::MLTensorOp::compFlux	(	int	amrlev,
		const Array< MultiFab *, 3 > &	fluxes,
		MultiFab &	sol,
		Location	loc
	)		const

override

Assemble viscous fluxes on AMR level amrlev.

Parameters

amrlev	Level whose fluxes are built.
fluxes	Destination arrays (one per direction).
sol	Solution values defining the flux.
loc	Sampling location for the stencil (faces, centroids, etc.).

◆ compVelGrad()

void amrex::MLTensorOp::compVelGrad	(	int	amrlev,
		const Array< MultiFab *, 3 > &	fluxes,
		MultiFab &	sol,
		Location	loc
	)		const

Compute velocity gradients (symmetrized) at the requested location.

Parameters

amrlev	AMR level index.
fluxes	Destination gradient MultiFabs per direction.
sol	Solution whose gradients are computed.
loc	Sampling location (cell/face center, centroid, etc.).

◆ define() [1/2]

void amrex::MLTensorOp::define	(	const Vector< Geometry > &	a_geom,
		const Vector< BoxArray > &	a_grids,
		const Vector< DistributionMapping > &	a_dmap,
		const LPInfo &	a_info = `LPInfo()`,
		const Vector< FabFactory< FArrayBox > const * > &	a_factory = `{}`
	)

Define the hierarchy for standard tensor solves.

Parameters

a_geom	Per-level geometries (fine to coarse).
a_grids	Per-level grids.
a_dmap	Distribution mappings.
a_info	Optional LPInfo overrides.
a_factory	Optional FAB factories.

◆ define() [2/2]

void amrex::MLTensorOp::define	(	const Vector< Geometry > &	a_geom,
		const Vector< BoxArray > &	a_grids,
		const Vector< DistributionMapping > &	a_dmap,
		const Vector< iMultiFab const * > &	a_overset_mask,
		const LPInfo &	a_info = `LPInfo()`,
		const Vector< FabFactory< FArrayBox > const * > &	a_factory = `{}`
	)

Define the hierarchy with overset masks (1 = unknown, 0 = known).

Parameters

a_geom	Per-level geometries.
a_grids	Per-level grids.
a_dmap	Distribution mappings.
a_overset_mask	Overset masks per AMR level.
a_info	Optional LPInfo overrides.
a_factory	Optional FAB factories.

◆ getNComp()

int amrex::MLTensorOp::getNComp ( ) const

inlinefinalvirtual

Return number of components.

Reimplemented from amrex::MLABecLaplacianT< MF >.

◆ isBottomSingular()

bool amrex::MLTensorOp::isBottomSingular ( ) const

inlinefinalvirtual

Convenience helper for the coarsest level singularity flag.

Reimplemented from amrex::MLABecLaplacianT< MF >.

◆ isCrossStencil()

bool amrex::MLTensorOp::isCrossStencil ( ) const

inlinefinalvirtual

Whether the stencil is the cross shape.

Reimplemented from amrex::MLCellLinOpT< MF >.

◆ isSingular()

bool amrex::MLTensorOp::isSingular ( int amrlev ) const

inlinefinalvirtual

Query whether AMR level amrlev is singular (null space present).

Reimplemented from amrex::MLABecLaplacianT< MF >.

◆ isTensorOp()

bool amrex::MLTensorOp::isTensorOp ( ) const

inlinefinalvirtual

Whether this operator is a tensor solve.

Reimplemented from amrex::MLCellLinOpT< MF >.

◆ needsUpdate()

bool amrex::MLTensorOp::needsUpdate ( ) const

inlinefinalvirtual

True if viscosity coefficients need to be rebuilt before apply().

Reimplemented from amrex::MLABecLaplacianT< MF >.

◆ operator=() [1/2]

MLTensorOp & amrex::MLTensorOp::operator= ( const MLTensorOp & )

delete

◆ operator=() [2/2]

MLTensorOp & amrex::MLTensorOp::operator= ( MLTensorOp && )

delete

◆ prepareForSolve()

void amrex::MLTensorOp::prepareForSolve ( )

finalvirtual

Assemble viscosity data/BC caches before solving.

Reimplemented from amrex::MLABecLaplacianT< MF >.

◆ setBCoeffs()

void amrex::MLTensorOp::setBCoeffs	(	int	amrlev,
		const Array< MultiFab const *, 3 > &	beta
	)

delete

◆ setBulkViscosity() [1/2]

void amrex::MLTensorOp::setBulkViscosity	(	int	amrlev,
		const Array< MultiFab const *, 3 > &	kappa
	)

Provide bulk viscosity kappa as face-centered MultiFabs.

Parameters

amrlev	AMR level receiving the data.
kappa	Array of MultiFabs, one per spatial direction.

◆ setBulkViscosity() [2/2]

void amrex::MLTensorOp::setBulkViscosity	(	int	amrlev,
		Real	kappa
	)

Populate kappa with a constant scalar.

Parameters

amrlev	AMR level receiving the value.
kappa	Scalar bulk viscosity.

◆ setShearViscosity() [1/2]

void amrex::MLTensorOp::setShearViscosity	(	int	amrlev,
		const Array< MultiFab const *, 3 > &	eta
	)

Provide shear viscosity eta as face-centered MultiFabs.

Parameters

amrlev	AMR level receiving the data.
eta	Array of MultiFabs, one per spatial direction.

◆ setShearViscosity() [2/2]

void amrex::MLTensorOp::setShearViscosity	(	int	amrlev,
		Real	eta
	)

Populate eta with a constant scalar.

Parameters

amrlev	AMR level receiving the value.
eta	Scalar shear viscosity.

◆ update()

void amrex::MLTensorOp::update ( )

inlinefinalvirtual

Rebuild coefficient averages (not yet implemented).

Reimplemented from amrex::MLABecLaplacianT< MF >.

Member Data Documentation

◆ m_has_kappa

bool amrex::MLTensorOp::m_has_kappa = false

protected

◆ m_kappa

Vector<Vector<Array<MultiFab,3> > > amrex::MLTensorOp::m_kappa

protected

The documentation for this class was generated from the following files:

/home/runner/work/amrex/amrex/Src/LinearSolvers/MLMG/AMReX_MLTensorOp.H
/home/runner/work/amrex/amrex/Src/LinearSolvers/MLMG/AMReX_MLTensorOp.cpp
/home/runner/work/amrex/amrex/Src/LinearSolvers/MLMG/AMReX_MLTensorOp_grad.cpp

Public Member Functions

Protected Attributes

Additional Inherited Members

Detailed Description

Constructor & Destructor Documentation

◆ MLTensorOp() [1/5]

◆ MLTensorOp() [2/5]

◆ MLTensorOp() [3/5]

◆ ~MLTensorOp()

◆ MLTensorOp() [4/5]

◆ MLTensorOp() [5/5]

Member Function Documentation

◆ apply()

◆ applyBCTensor()

◆ compFlux()

◆ compVelGrad()

◆ define() [1/2]

◆ define() [2/2]

◆ getNComp()

◆ isBottomSingular()

◆ isCrossStencil()

◆ isSingular()

◆ isTensorOp()

◆ needsUpdate()

◆ operator=() [1/2]

◆ operator=() [2/2]

◆ prepareForSolve()

◆ setBCoeffs()

◆ setBulkViscosity() [1/2]

◆ setBulkViscosity() [2/2]

◆ setShearViscosity() [1/2]

◆ setShearViscosity() [2/2]

◆ update()

Member Data Documentation

◆ m_has_kappa

◆ m_kappa