LinearAdvectionDiffusionReaction operator using LocalRegions implementation. More...

Inheritance diagram for Nektar::Collections::LinearAdvectionDiffusionReaction_IterPerExp:

Public Member Functions
	~LinearAdvectionDiffusionReaction_IterPerExp () final=default

void	operator() (const Array< OneD, const NekDouble > &input, Array< OneD, NekDouble > &output, Array< OneD, NekDouble > &output1, Array< OneD, NekDouble > &output2, Array< OneD, NekDouble > &wsp) final
	Perform operation. More...

void	operator() (int dir, const Array< OneD, const NekDouble > &input, Array< OneD, NekDouble > &output, Array< OneD, NekDouble > &wsp) final

void	UpdateFactors (StdRegions::FactorMap factors) override
	Check the validity of supplied constant factors. More...

void	UpdateVarcoeffs (StdRegions::VarCoeffMap &varcoeffs) override
	Check whether necessary advection velocities are supplied and copy into local array for operator. More...

Public Member Functions inherited from Nektar::Collections::Operator
	Operator (std::vector< StdRegions::StdExpansionSharedPtr > pCollExp, std::shared_ptr< CoalescedGeomData > GeomData, StdRegions::FactorMap factors)
	Constructor. More...

virtual	~Operator ()=default

virtual COLLECTIONS_EXPORT void	operator() (const Array< OneD, const NekDouble > &input, Array< OneD, NekDouble > &output0, Array< OneD, NekDouble > &output1, Array< OneD, NekDouble > &output2, Array< OneD, NekDouble > &wsp=NullNekDouble1DArray)=0
	Perform operation. More...

virtual COLLECTIONS_EXPORT void	operator() (int dir, const Array< OneD, const NekDouble > &input, Array< OneD, NekDouble > &output, Array< OneD, NekDouble > &wsp=NullNekDouble1DArray)=0

virtual COLLECTIONS_EXPORT void	UpdateFactors (StdRegions::FactorMap factors)
	Update the supplied factor map. More...

virtual COLLECTIONS_EXPORT void	UpdateVarcoeffs (StdRegions::VarCoeffMap &varcoeffs)
	Update the supplied variable coefficients. More...

unsigned int	GetWspSize ()
	Get the size of the required workspace. More...

unsigned int	GetNumElmt ()
	Get number of elements. More...

StdRegions::StdExpansionSharedPtr	GetExpSharedPtr ()
	Get expansion pointer. More...

unsigned int	GetInputSize (bool defaultIn=true)

unsigned int	GetOutputSize (bool defaultOut=true)

Protected Attributes
Array< TwoD, const NekDouble >	m_derivFac

Array< OneD, const NekDouble >	m_jac

int	m_dim

int	m_coordim

StdRegions::FactorMap	m_factors

StdRegions::VarCoeffMap	m_varcoeffs

NekDouble	m_lambda

Array< OneD, Array< OneD, NekDouble > >	m_advVel

const StdRegions::VarCoeffType	advVelTypes [3]

Protected Attributes inherited from Nektar::Collections::Operator
bool	m_isDeformed

StdRegions::StdExpansionSharedPtr	m_stdExp

unsigned int	m_numElmt
	number of elements that the operator is applied on More...

unsigned int	m_nqe

unsigned int	m_wspSize

unsigned int	m_inputSize
	number of modes or quadrature points that are passed as input to an operator More...

unsigned int	m_outputSize
	number of modes or quadrature points that are taken as output from an operator More...

unsigned int	m_inputSizeOther
	Number of modes or quadrature points, opposite to m_inputSize. More...

unsigned int	m_outputSizeOther
	Number of modes or quadrature points, opposite to m_outputSize. More...

Private Member Functions
	LinearAdvectionDiffusionReaction_IterPerExp (vector< StdRegions::StdExpansionSharedPtr > pCollExp, CoalescedGeomDataSharedPtr pGeomData, StdRegions::FactorMap factors)

Private Member Functions inherited from Nektar::Collections::LinearAdvectionDiffusionReaction_Helper
	LinearAdvectionDiffusionReaction_Helper ()

Detailed Description

LinearAdvectionDiffusionReaction operator using LocalRegions implementation.

Definition at line 224 of file LinearAdvectionDiffusionReaction.cpp.

Constructor & Destructor Documentation

◆ ~LinearAdvectionDiffusionReaction_IterPerExp()

Nektar::Collections::LinearAdvectionDiffusionReaction_IterPerExp::~LinearAdvectionDiffusionReaction_IterPerExp ( )

finaldefault

◆ LinearAdvectionDiffusionReaction_IterPerExp()

Nektar::Collections::LinearAdvectionDiffusionReaction_IterPerExp::LinearAdvectionDiffusionReaction_IterPerExp	(	vector< StdRegions::StdExpansionSharedPtr >	pCollExp,
		CoalescedGeomDataSharedPtr	pGeomData,
		StdRegions::FactorMap	factors
	)

inlineprivate

Definition at line 469 of file LinearAdvectionDiffusionReaction.cpp.

        : Operator(pCollExp, pGeomData, factors),
          LinearAdvectionDiffusionReaction_Helper()
    {
        m_dim     = pCollExp[0]->GetShapeDimension();
        m_coordim = pCollExp[0]->GetCoordim();
        int nqtot = m_stdExp->GetTotPoints();
 
        m_derivFac = pGeomData->GetDerivFactors(pCollExp);
        m_jac      = pGeomData->GetJac(pCollExp);
        m_wspSize  = (2 * m_coordim + 1) * nqtot;
 
        m_lambda    = 1.0;
        m_advVel    = Array<OneD, Array<OneD, NekDouble>>(m_coordim);
        m_factors   = StdRegions::NullFactorMap;
        m_varcoeffs = StdRegions::NullVarCoeffMap;
        this->UpdateFactors(factors);
    }

References m_advVel, m_coordim, m_derivFac, m_dim, m_factors, m_jac, m_lambda, Nektar::Collections::Operator::m_stdExp, m_varcoeffs, Nektar::Collections::Operator::m_wspSize, Nektar::StdRegions::NullFactorMap, Nektar::StdRegions::NullVarCoeffMap, and UpdateFactors().

Member Function Documentation

◆ operator()() [1/2]

void Nektar::Collections::LinearAdvectionDiffusionReaction_IterPerExp::operator()	(	const Array< OneD, const NekDouble > &	input,
		Array< OneD, NekDouble > &	output0,
		Array< OneD, NekDouble > &	output1,
		Array< OneD, NekDouble > &	output2,
		Array< OneD, NekDouble > &	wsp
	)

inlinefinalvirtual

Perform operation.

Mass term

Add advection term

Determine Laplacian term

Implements Nektar::Collections::Operator.

Definition at line 233 of file LinearAdvectionDiffusionReaction.cpp.

    {
        const int nCoeffs = m_stdExp->GetNcoeffs();
        const int nPhys   = m_stdExp->GetTotPoints();
 
        ASSERTL1(input.size() >= m_numElmt * nCoeffs,
                 "input array size is insufficient");
        ASSERTL1(output.size() >= m_numElmt * nCoeffs,
                 "output array size is insufficient");
 
        Array<OneD, NekDouble> tmpphys, t1;
        Array<OneD, Array<OneD, NekDouble>> dtmp(3);
        Array<OneD, Array<OneD, NekDouble>> tmp(3);
 
        tmpphys = wsp;
        for (int i = 1; i < m_coordim + 1; ++i)
        {
            dtmp[i - 1] = wsp + i * nPhys;
            tmp[i - 1]  = wsp + (i + m_coordim) * nPhys;
        }
 
        for (int i = 0; i < m_numElmt; ++i)
        {
            // Std u
            m_stdExp->BwdTrans(input + i * nCoeffs, tmpphys);
 
            // Std \nabla u
            m_stdExp->PhysDeriv(tmpphys, dtmp[0], dtmp[1], dtmp[2]);
 
            // Transform Std \nabla u -> Local \nabla u
            // tmp[0] = dxi1/dx du/dxi1 + dxi2/dx du/dxi2 = du / dx
            // tmp[1] = dxi1/dy du/dxi1 + dxi2/dy du/dxi2 = du / dy
            if (m_isDeformed)
            {
                for (int j = 0; j < m_coordim; ++j)
                {
                    Vmath::Vmul(nPhys,
                                m_derivFac[j * m_dim].origin() + i * nPhys, 1,
                                &dtmp[0][0], 1, &tmp[j][0], 1);
 
                    for (int k = 1; k < m_dim; ++k)
                    {
                        Vmath::Vvtvp(
                            nPhys,
                            m_derivFac[j * m_dim + k].origin() + i * nPhys, 1,
                            &dtmp[k][0], 1, &tmp[j][0], 1, &tmp[j][0], 1);
                    }
                }
            }
            else
            {
                for (int j = 0; j < m_coordim; ++j)
                {
                    Vmath::Smul(nPhys, m_derivFac[j * m_dim][i], &dtmp[0][0], 1,
                                &tmp[j][0], 1);
 
                    for (int k = 1; k < m_dim; ++k)
                    {
                        Vmath::Svtvp(nPhys, m_derivFac[j * m_dim + k][i],
                                     &dtmp[k][0], 1, &tmp[j][0], 1, &tmp[j][0],
                                     1);
                    }
                }
            }
 
            /// Mass term
            // tmpphys *= \lambda = \lambda u
            Vmath::Smul(nPhys, -m_lambda, tmpphys, 1, tmpphys, 1);
 
            /// Add advection term
            // tmpphys += V[0] * du / dx + V[1] * du / dy
            for (int j = 0; j < m_coordim; ++j)
            {
                Vmath::Vvtvp(nPhys, m_advVel[j] + i * nPhys, 1, tmp[j], 1,
                             tmpphys, 1, tmpphys, 1);
            }
 
            // Multiply by Jacobian for inner product
            if (m_isDeformed)
            {
                Vmath::Vmul(nPhys, m_jac + i * nPhys, 1, tmpphys, 1, tmpphys,
                            1);
            }
            else
            {
                Vmath::Smul(nPhys, m_jac[i], tmpphys, 1, tmpphys, 1);
            }
 
            // Compute inner product wrt base
            m_stdExp->IProductWRTBase(tmpphys, t1 = output + i * nCoeffs);
 
            /// Determine Laplacian term
            // add derivFactors for IProdWRTDerivBase below
            // dtmp[0] = dxi1/dx du/dx + dxi1/dy du/dy
            // dtmp[1] = dxi2/dx du/dx + dxi2/dy du/dy
            if (m_isDeformed)
            {
                for (int j = 0; j < m_dim; ++j)
                {
                    Vmath::Vmul(nPhys, m_derivFac[j].origin() + i * nPhys, 1,
                                &tmp[0][0], 1, &dtmp[j][0], 1);
 
                    for (int k = 1; k < m_coordim; ++k)
                    {
                        Vmath::Vvtvp(
                            nPhys,
                            m_derivFac[j + k * m_dim].origin() + i * nPhys, 1,
                            &tmp[k][0], 1, &dtmp[j][0], 1, &dtmp[j][0], 1);
                    }
                }
 
                // calculate Iproduct WRT Std Deriv
                for (int j = 0; j < m_dim; ++j)
                {
 
                    // multiply by Jacobian
                    Vmath::Vmul(nPhys, m_jac + i * nPhys, 1, dtmp[j], 1,
                                dtmp[j], 1);
 
                    m_stdExp->IProductWRTDerivBase(j, dtmp[j], tmp[0]);
                    Vmath::Vadd(nCoeffs, tmp[0], 1, output + i * nCoeffs, 1,
                                t1 = output + i * nCoeffs, 1);
                }
            }
            else
            {
                for (int j = 0; j < m_dim; ++j)
                {
                    Vmath::Smul(nPhys, m_derivFac[j][i], &tmp[0][0], 1,
                                &dtmp[j][0], 1);
 
                    for (int k = 1; k < m_coordim; ++k)
                    {
                        Vmath::Svtvp(nPhys, m_derivFac[j + k * m_dim][i],
                                     &tmp[k][0], 1, &dtmp[j][0], 1, &dtmp[j][0],
                                     1);
                    }
                }
 
                // calculate Iproduct WRT Std Deriv
                for (int j = 0; j < m_dim; ++j)
                {
                    // multiply by Jacobian for integration
                    Vmath::Smul(nPhys, m_jac[i], dtmp[j], 1, dtmp[j], 1);
 
                    m_stdExp->IProductWRTDerivBase(j, dtmp[j], tmp[0]);
                    Vmath::Vadd(nCoeffs, tmp[0], 1, output + i * nCoeffs, 1,
                                t1 = output + i * nCoeffs, 1);
                }
            }
        }
    }

References ASSERTL1, m_advVel, m_coordim, m_derivFac, m_dim, Nektar::Collections::Operator::m_isDeformed, m_jac, m_lambda, Nektar::Collections::Operator::m_numElmt, Nektar::Collections::Operator::m_stdExp, Nektar::MovementTests::origin, Vmath::Smul(), Vmath::Svtvp(), Vmath::Vadd(), Vmath::Vmul(), and Vmath::Vvtvp().

◆ operator()() [2/2]

void Nektar::Collections::LinearAdvectionDiffusionReaction_IterPerExp::operator()	(	int	dir,
		const Array< OneD, const NekDouble > &	input,
		Array< OneD, NekDouble > &	output,
		Array< OneD, NekDouble > &	wsp
	)

inlinefinalvirtual

Implements Nektar::Collections::Operator.

Definition at line 390 of file LinearAdvectionDiffusionReaction.cpp.

    {
        NEKERROR(ErrorUtil::efatal, "Not valid for this operator.");
    }

References Nektar::ErrorUtil::efatal, and NEKERROR.

◆ UpdateFactors()

void Nektar::Collections::LinearAdvectionDiffusionReaction_IterPerExp::UpdateFactors ( StdRegions::FactorMap factors )

inlineoverridevirtual

Check the validity of supplied constant factors.

Parameters

factors Map of factors

Reimplemented from Nektar::Collections::Operator.

Definition at line 403 of file LinearAdvectionDiffusionReaction.cpp.

    {
        m_factors = factors;
 
        // Check Lambda constant of LinearAdvectionDiffusionReaction operator
        auto x = factors.find(StdRegions::eFactorLambda);
        ASSERTL1(
            x != factors.end(),
            "Constant factor not defined: " +
                std::string(
                    StdRegions::ConstFactorTypeMap[StdRegions::eFactorLambda]));
        m_lambda = x->second;
    }

References ASSERTL1, Nektar::StdRegions::ConstFactorTypeMap, Nektar::StdRegions::eFactorLambda, Nektar::VarcoeffHashingTest::factors, m_factors, and m_lambda.

Referenced by LinearAdvectionDiffusionReaction_IterPerExp().

◆ UpdateVarcoeffs()

void Nektar::Collections::LinearAdvectionDiffusionReaction_IterPerExp::UpdateVarcoeffs ( StdRegions::VarCoeffMap & varcoeffs )

inlineoverridevirtual

Check whether necessary advection velocities are supplied and copy into local array for operator.

Parameters

varcoeffs Map of variable coefficients

Reimplemented from Nektar::Collections::Operator.

Definition at line 423 of file LinearAdvectionDiffusionReaction.cpp.

    {
        // Check whether any varcoeffs are provided
        if (varcoeffs.empty())
        {
            return;
        }
 
        // Check advection velocities count
        int ndir = 0;
        for (auto &x : advVelTypes)
        {
            if (varcoeffs.count(x))
            {
                ndir++;
            }
        }
        ASSERTL0(ndir, "Must define at least one advection velocity");
        ASSERTL1(ndir <= m_coordim,
                 "Number of constants is larger than coordinate dimensions");
 
        // Copy new varcoeffs
        m_varcoeffs = varcoeffs;
 
        // Hold advection velocity reference
        // separate copy required for operator
        for (int i = 0; i < m_coordim; i++)
        {
            m_advVel[i] = m_varcoeffs.find(advVelTypes[i])->second.GetValue();
        }
    }

References advVelTypes, ASSERTL0, ASSERTL1, m_advVel, m_coordim, and m_varcoeffs.

Member Data Documentation

◆ advVelTypes

const StdRegions::VarCoeffType Nektar::Collections::LinearAdvectionDiffusionReaction_IterPerExp::advVelTypes[3]

protected

Initial value:

= {StdRegions::eVarCoeffVelX,
                                                     StdRegions::eVarCoeffVelY,
                                                     StdRegions::eVarCoeffVelZ}