ERF_ComputeStress_S.cpp File Reference

#include <ERF_Diffusion.H>
#include <ERF_TerrainMetrics.H>

Include dependency graph for ERF_ComputeStress_S.cpp:

Functions
void	ComputeStressConsVisc_S (Box bxcc, Box tbxxy, Box tbxxz, Box tbxyz, Real mu_eff, const Array4< const Real > &cell_data, Array4< Real > &tau11, Array4< Real > &tau22, Array4< Real > &tau33, Array4< Real > &tau12, Array4< Real > &tau21, Array4< Real > &tau13, Array4< Real > &tau31, Array4< Real > &tau23, Array4< Real > &tau32, const Array4< const Real > &er_arr, const Array4< const Real > &mf_mx, const Array4< const Real > &mf_ux, const Array4< const Real > &mf_vx, const Array4< const Real > &mf_my, const Array4< const Real > &mf_uy, const Array4< const Real > &mf_vy, Array4< Real > &tau13i, Array4< Real > &tau23i, Array4< Real > &tau33i)
void	ComputeStressVarVisc_S (Box bxcc, Box tbxxy, Box tbxxz, Box tbxyz, Real mu_eff, const Array4< const Real > &mu_turb, const Array4< const Real > &cell_data, Array4< Real > &tau11, Array4< Real > &tau22, Array4< Real > &tau33, Array4< Real > &tau12, Array4< Real > &tau21, Array4< Real > &tau13, Array4< Real > &tau31, Array4< Real > &tau23, Array4< Real > &tau32, const Array4< const Real > &er_arr, const Array4< const Real > &mf_mx, const Array4< const Real > &mf_ux, const Array4< const Real > &mf_vx, const Array4< const Real > &mf_my, const Array4< const Real > &mf_uy, const Array4< const Real > &mf_vy, Array4< Real > &tau13i, Array4< Real > &tau23i, Array4< Real > &tau33i)

Function Documentation

ComputeStressConsVisc_S()

void ComputeStressConsVisc_S	(	Box	bxcc,
		Box	tbxxy,
		Box	tbxxz,
		Box	tbxyz,
		Real	mu_eff,
		const Array4< const Real > &	cell_data,
		Array4< Real > &	tau11,
		Array4< Real > &	tau22,
		Array4< Real > &	tau33,
		Array4< Real > &	tau12,
		Array4< Real > &	tau21,
		Array4< Real > &	tau13,
		Array4< Real > &	tau31,
		Array4< Real > &	tau23,
		Array4< Real > &	tau32,
		const Array4< const Real > &	er_arr,
		const Array4< const Real > &	mf_mx,
		const Array4< const Real > &	mf_ux,
		const Array4< const Real > &	mf_vx,
		const Array4< const Real > &	mf_my,
		const Array4< const Real > &	mf_uy,
		const Array4< const Real > &	mf_vy,
		Array4< Real > &	tau13i,
		Array4< Real > &	tau23i,
		Array4< Real > &	tau33i
	)

Function for computing the stress with constant viscosity on a stretched grid.

Parameters

[in]	bxcc	cell center box for tau_ii
[in]	tbxxy	nodal xy box for tau_12
[in]	tbxxz	nodal xz box for tau_13
[in]	tbxyz	nodal yz box for tau_23
[in]	mu_eff	constant molecular viscosity
[in]	cell_data	to access rho if ConstantAlpha
[in,out]	tau11	11 strain -> stress
[in,out]	tau22	22 strain -> stress
[in,out]	tau33	33 strain -> stress
[in,out]	tau12	12 strain -> stress
[in,out]	tau13	13 strain -> stress
[in,out]	tau21	21 strain -> stress
[in,out]	tau23	23 strain -> stress
[in,out]	tau31	31 strain -> stress
[in,out]	tau32	32 strain -> stress
[in]	er_arr	expansion rate
[in]	mf_mx	x map factor at cell centers
[in]	mf_ux	x map factor at x-faces
[in]	mf_vx	x map factor at y-faces
[in]	mf_my	y map factor at cell centers
[in]	mf_uy	y map factor at x-faces
[in]	mf_vy	y map factor at y-faces
[in,out]	tau13i	contribution to stress from du/dz
[in,out]	tau23i	contribution to stress from dv/dz
[in,out]	tau33i	contribution to stress from dw/dz

{
    // NOTE: mu_eff includes factor of 2
 
    // Handle constant alpha case, in which the provided mu_eff is actually
    // "alpha" and the viscosity needs to be scaled by rho. This can be further
    // optimized with if statements below instead of creating a new FAB,
    // but this is implementation is cleaner.
    FArrayBox temp;
    Box gbx = bxcc; // Note: bxcc have been grown in x/y only.
    gbx.grow(IntVect(0,0,1));
    temp.resize(gbx,1, The_Async_Arena());
    Array4<Real> rhoAlpha = temp.array();
 
    if (cell_data)
    // constant alpha (stored in mu_eff)
    {
        ParallelFor(gbx,
        [=] AMREX_GPU_DEVICE (int i, int j, int k) noexcept {
            rhoAlpha(i,j,k) = cell_data(i, j, k, Rho_comp) * mu_eff;
        });
    }
    else
    // constant mu_eff
    {
        ParallelFor(gbx,
        [=] AMREX_GPU_DEVICE (int i, int j, int k) noexcept {
            rhoAlpha(i,j,k) = mu_eff;
        });
    }
 
    // First block: cell centered stresses
    //***********************************************************************************
    Real OneThird   = (one/three);
    ParallelFor(bxcc, [=] AMREX_GPU_DEVICE (int i, int j, int k) noexcept
    {
        Real mfx = mf_mx(i,j,0);
        Real mfy = mf_my(i,j,0);
 
        Real mu_tot     = rhoAlpha(i,j,k);
 
        if (tau33i) tau33i(i,j,k) = -mu_tot * tau33(i,j,k);
 
        tau11(i,j,k) = -mu_tot / mfy * ( tau11(i,j,k) - OneThird*er_arr(i,j,k) );
        tau22(i,j,k) = -mu_tot / mfx * ( tau22(i,j,k) - OneThird*er_arr(i,j,k) );
        tau33(i,j,k) = -mu_tot       * ( tau33(i,j,k) - OneThird*er_arr(i,j,k) );
    });
 
    // Second block: off diagonal stresses
    //***********************************************************************************
    ParallelFor(tbxxy,tbxxz,tbxyz,
    [=] AMREX_GPU_DEVICE (int i, int j, int k) noexcept
    {
        Real mfx = myhalf * (mf_ux(i,j,0) + mf_ux(i,j-1,0));
        Real mfy = myhalf * (mf_vy(i,j,0) + mf_vy(i-1,j,0));
 
        Real mu_tot = fourth*( rhoAlpha(i-1, j  , k) + rhoAlpha(i, j  , k)
                           + rhoAlpha(i-1, j-1, k) + rhoAlpha(i, j-1, k) );
 
        tau12(i,j,k) *= -mu_tot / mfx;
        tau21(i,j,k) *= -mu_tot / mfy;
    },
    [=] AMREX_GPU_DEVICE (int i, int j, int k) noexcept
    {
        Real mfy = mf_uy(i,j,0);
 
        Real mu_tot = fourth * ( rhoAlpha(i-1, j  , k  ) + rhoAlpha(i  , j  , k  )
                             + rhoAlpha(i-1, j  , k-1) + rhoAlpha(i  , j  , k-1) );
 
        tau13(i,j,k) *= -mu_tot;
        tau31(i,j,k) *= -mu_tot / mfy;
 
        if (tau13i) tau13i(i,j,k) *= -mu_tot;
    },
    [=] AMREX_GPU_DEVICE (int i, int j, int k) noexcept
    {
        Real mfx = mf_vx(i,j,0);
 
        Real mu_tot = fourth * ( rhoAlpha(i  , j-1, k  ) + rhoAlpha(i  , j  , k  )
                             + rhoAlpha(i  , j-1, k-1) + rhoAlpha(i  , j  , k-1) );
 
        tau23(i,j,k) *= -mu_tot;
        tau32(i,j,k) *= -mu_tot / mfx;
 
        if (tau23i) tau23i(i,j,k) *= -mu_tot;
    });
}

Referenced by erf_make_tau_terms().

Here is the call graph for this function:

Here is the caller graph for this function:

ComputeStressVarVisc_S()

void ComputeStressVarVisc_S	(	Box	bxcc,
		Box	tbxxy,
		Box	tbxxz,
		Box	tbxyz,
		Real	mu_eff,
		const Array4< const Real > &	mu_turb,
		const Array4< const Real > &	cell_data,
		Array4< Real > &	tau11,
		Array4< Real > &	tau22,
		Array4< Real > &	tau33,
		Array4< Real > &	tau12,
		Array4< Real > &	tau21,
		Array4< Real > &	tau13,
		Array4< Real > &	tau31,
		Array4< Real > &	tau23,
		Array4< Real > &	tau32,
		const Array4< const Real > &	er_arr,
		const Array4< const Real > &	mf_mx,
		const Array4< const Real > &	mf_ux,
		const Array4< const Real > &	mf_vx,
		const Array4< const Real > &	mf_my,
		const Array4< const Real > &	mf_uy,
		const Array4< const Real > &	mf_vy,
		Array4< Real > &	tau13i,
		Array4< Real > &	tau23i,
		Array4< Real > &	tau33i
	)

Function for computing the stress with variable viscosity on a stretched grid.

Parameters

[in]	bxcc	cell center box for tau_ii
[in]	tbxxy	nodal xy box for tau_12
[in]	tbxxz	nodal xz box for tau_13
[in]	tbxyz	nodal yz box for tau_23
[in]	mu_eff	constant molecular viscosity
[in]	mu_turb	variable turbulent viscosity
[in]	cell_data	to access rho if ConstantAlpha
[in,out]	tau11	11 strain -> stress
[in,out]	tau22	22 strain -> stress
[in,out]	tau33	33 strain -> stress
[in,out]	tau12	12 strain -> stress
[in,out]	tau13	13 strain -> stress
[in,out]	tau21	21 strain -> stress
[in,out]	tau23	23 strain -> stress
[in,out]	tau31	31 strain -> stress
[in,out]	tau32	32 strain -> stress
[in]	er_arr	expansion rate
[in]	mf_mx	x map factor at cell centers
[in]	mf_ux	x map factor at x-faces
[in]	mf_vx	x map factor at y-faces
[in]	mf_my	y map factor at cell centers
[in]	mf_uy	y map factor at x-faces
[in]	mf_vy	y map factor at y-faces
[in,out]	tau13i	contribution to stress from du/dz
[in,out]	tau23i	contribution to stress from dv/dz
[in,out]	tau33i	contribution to stress from dw/dz

{
    // NOTE: mu_eff includes factor of 2
 
    // Handle constant alpha case, in which the provided mu_eff is actually
    // "alpha" and the viscosity needs to be scaled by rho. This can be further
    // optimized with if statements below instead of creating a new FAB,
    // but this is implementation is cleaner.
    FArrayBox temp;
    Box gbx = bxcc; // Note: bxcc have been grown in x/y only.
    gbx.grow(IntVect(0,0,1));
    temp.resize(gbx,1, The_Async_Arena());
    Array4<Real> rhoAlpha = temp.array();
 
    if (cell_data)
    // constant alpha (stored in mu_eff)
    {
        ParallelFor(gbx,
        [=] AMREX_GPU_DEVICE (int i, int j, int k) noexcept {
            rhoAlpha(i,j,k) = cell_data(i, j, k, Rho_comp) * mu_eff;
        });
    }
    else
    // constant mu_eff
    {
        ParallelFor(gbx,
        [=] AMREX_GPU_DEVICE (int i, int j, int k) noexcept {
            rhoAlpha(i,j,k) = mu_eff;
        });
    }
 
    // First block: cell centered stresses
    //***********************************************************************************
    Real OneThird   = (one/three);
    ParallelFor(bxcc, [=] AMREX_GPU_DEVICE (int i, int j, int k) noexcept
    {
        Real mfx = mf_mx(i,j,0);
        Real mfy = mf_my(i,j,0);
 
        Real mu_11 = rhoAlpha(i,j,k) + two * mu_turb(i, j, k, EddyDiff::Mom_h);
        Real mu_22 = mu_11;
        Real mu_33 = rhoAlpha(i,j,k) + two * mu_turb(i, j, k, EddyDiff::Mom_v);
 
        if (tau33i) tau33i(i,j,k) = -mu_33 * tau33(i,j,k);
 
        tau11(i,j,k) = -mu_11 / mfy * ( tau11(i,j,k) - OneThird*er_arr(i,j,k) );
        tau22(i,j,k) = -mu_22 / mfx * ( tau22(i,j,k) - OneThird*er_arr(i,j,k) );
        tau33(i,j,k) = -mu_33       * ( tau33(i,j,k) - OneThird*er_arr(i,j,k) );
    });
 
    // Second block: off diagonal stresses
    //***********************************************************************************
    ParallelFor(tbxxy,tbxxz,tbxyz,
    [=] AMREX_GPU_DEVICE (int i, int j, int k) noexcept
    {
        Real mfx = myhalf * (mf_ux(i,j,0) + mf_ux(i,j-1,0));
        Real mfy = myhalf * (mf_vy(i,j,0) + mf_vy(i-1,j,0));
 
        Real mu_bar = fourth*( mu_turb(i-1, j  , k, EddyDiff::Mom_h) + mu_turb(i, j  , k, EddyDiff::Mom_h)
                           + mu_turb(i-1, j-1, k, EddyDiff::Mom_h) + mu_turb(i, j-1, k, EddyDiff::Mom_h) );
        Real rhoAlpha_bar = fourth*( rhoAlpha(i-1, j  , k) + rhoAlpha(i, j  , k)
                                 + rhoAlpha(i-1, j-1, k) + rhoAlpha(i, j-1, k) );
        Real mu_tot = rhoAlpha_bar + two*mu_bar;
 
        tau12(i,j,k) *= -mu_tot / mfx;
        tau21(i,j,k) *= -mu_tot / mfy;
    },
    [=] AMREX_GPU_DEVICE (int i, int j, int k) noexcept
    {
        Real mfy = mf_uy(i,j,0);
 
        Real mu_bar = fourth*( mu_turb(i-1, j, k  , EddyDiff::Mom_v) + mu_turb(i, j, k  , EddyDiff::Mom_v)
                           + mu_turb(i-1, j, k-1, EddyDiff::Mom_v) + mu_turb(i, j, k-1, EddyDiff::Mom_v) );
        Real rhoAlpha_bar = fourth*( rhoAlpha(i-1, j, k  ) + rhoAlpha(i, j, k  )
                                 + rhoAlpha(i-1, j, k-1) + rhoAlpha(i, j, k-1) );
        Real mu_tot = rhoAlpha_bar + two*mu_bar;
 
        tau13(i,j,k) *= -mu_tot;
        tau31(i,j,k) *= -mu_tot / mfy;
 
        if (tau13i) tau13i(i,j,k) *= -mu_tot;
    },
    [=] AMREX_GPU_DEVICE (int i, int j, int k) noexcept
    {
        Real mfx = mf_vx(i,j,0);
 
        Real mu_bar = fourth*( mu_turb(i, j-1, k  , EddyDiff::Mom_v) + mu_turb(i, j, k  , EddyDiff::Mom_v)
                           + mu_turb(i, j-1, k-1, EddyDiff::Mom_v) + mu_turb(i, j, k-1, EddyDiff::Mom_v) );
        Real rhoAlpha_bar = fourth*( rhoAlpha(i, j-1, k  ) + rhoAlpha(i, j, k  )
                                 + rhoAlpha(i, j-1, k-1) + rhoAlpha(i, j, k-1) );
        Real mu_tot = rhoAlpha_bar + two*mu_bar;
 
        tau23(i,j,k) *= -mu_tot;
        tau32(i,j,k) *= -mu_tot / mfx;
 
        if (tau23i) tau23i(i,j,k) *= -mu_tot;
    });
}

Referenced by erf_make_tau_terms().

Here is the call graph for this function:

Here is the caller graph for this function: