Backward transform operator using sum-factorisation (Prism) More...

Inheritance diagram for Nektar::Collections::BwdTrans_SumFac_Prism:

Public Member Functions
	~BwdTrans_SumFac_Prism () final=default

void	operator() (const Array< OneD, const NekDouble > &input, Array< OneD, NekDouble > &output0, 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

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
const int	m_nquad0

const int	m_nquad1

const int	m_nquad2

const int	m_nmodes0

const int	m_nmodes1

const int	m_nmodes2

Array< OneD, const NekDouble >	m_base0

Array< OneD, const NekDouble >	m_base1

Array< OneD, const NekDouble >	m_base2

bool	m_sortTopVertex

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
	BwdTrans_SumFac_Prism (vector< StdRegions::StdExpansionSharedPtr > pCollExp, CoalescedGeomDataSharedPtr pGeomData, StdRegions::FactorMap factors)

Additional Inherited Members
Protected Member Functions inherited from Nektar::Collections::BwdTrans_Helper
	BwdTrans_Helper ()

Detailed Description

Backward transform operator using sum-factorisation (Prism)

Definition at line 949 of file BwdTrans.cpp.

Constructor & Destructor Documentation

◆ ~BwdTrans_SumFac_Prism()

Nektar::Collections::BwdTrans_SumFac_Prism::~BwdTrans_SumFac_Prism ( )

finaldefault

◆ BwdTrans_SumFac_Prism()

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

inlineprivate

Definition at line 1048 of file BwdTrans.cpp.

        : Operator(pCollExp, pGeomData, factors), BwdTrans_Helper(),
          m_nquad0(m_stdExp->GetNumPoints(0)),
          m_nquad1(m_stdExp->GetNumPoints(1)),
          m_nquad2(m_stdExp->GetNumPoints(2)),
          m_nmodes0(m_stdExp->GetBasisNumModes(0)),
          m_nmodes1(m_stdExp->GetBasisNumModes(1)),
          m_nmodes2(m_stdExp->GetBasisNumModes(2)),
          m_base0(m_stdExp->GetBasis(0)->GetBdata()),
          m_base1(m_stdExp->GetBasis(1)->GetBdata()),
          m_base2(m_stdExp->GetBasis(2)->GetBdata())
    {
        m_wspSize = m_numElmt * m_nmodes0 *
                    (m_nmodes1 * m_nquad2 + m_nquad1 * m_nquad2);
 
        if (m_stdExp->GetBasis(0)->GetBasisType() == LibUtilities::eModified_A)
        {
            m_sortTopVertex = true;
        }
        else
        {
            m_sortTopVertex = false;
        }
    }

References Nektar::LibUtilities::eModified_A, m_nmodes0, m_nmodes1, m_nquad1, m_nquad2, Nektar::Collections::Operator::m_numElmt, m_sortTopVertex, Nektar::Collections::Operator::m_stdExp, and Nektar::Collections::Operator::m_wspSize.

Member Function Documentation

◆ operator()() [1/2]

void Nektar::Collections::BwdTrans_SumFac_Prism::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.

Implements Nektar::Collections::Operator.

Definition at line 957 of file BwdTrans.cpp.

    {
        ASSERTL1(wsp.size() == m_wspSize, "Incorrect workspace size");
 
        // Assign second half of workspace for 2nd DGEMM operation.
        int totmodes = m_stdExp->GetNcoeffs();
 
        Array<OneD, NekDouble> wsp2 =
            wsp + m_nmodes0 * m_nmodes1 * m_nquad2 * m_numElmt;
 
        Vmath::Zero(m_nmodes0 * m_nmodes1 * m_nquad2 * m_numElmt, wsp, 1);
        int i     = 0;
        int j     = 0;
        int mode  = 0;
        int mode1 = 0;
        int cnt   = 0;
        for (i = mode = mode1 = 0; i < m_nmodes0; ++i)
        {
            cnt = i * m_nquad2 * m_numElmt;
            for (j = 0; j < m_nmodes1; ++j)
            {
                Blas::Dgemm('N', 'N', m_nquad2, m_numElmt, m_nmodes2 - i, 1.0,
                            m_base2.get() + mode * m_nquad2, m_nquad2,
                            input.get() + mode1, totmodes, 0.0,
                            &wsp[j * m_nquad2 * m_numElmt * m_nmodes0 + cnt],
                            m_nquad2);
                mode1 += m_nmodes2 - i;
            }
            mode += m_nmodes2 - i;
        }
 
        // fix for modified basis by splitting top vertex mode
        if (m_sortTopVertex)
        {
            for (j = 0; j < m_nmodes1; ++j)
            {
                for (i = 0; i < m_numElmt; ++i)
                {
                    Blas::Daxpy(m_nquad2,
                                input[1 + i * totmodes + j * m_nmodes2],
                                m_base2.get() + m_nquad2, 1,
                                &wsp[j * m_nquad2 * m_numElmt * m_nmodes0 +
                                     m_nquad2 * m_numElmt] +
                                    i * m_nquad2,
                                1);
                }
            }
            // Believe this could be made into a m_nmodes1
            // dgemv if we made an array of m_numElmt copies
            // of m_base2[m_quad2] (which are of size
            // m_nquad2.
        }
 
        // Perform summation over '1' direction
        Blas::Dgemm('N', 'T', m_nquad1, m_nquad2 * m_numElmt * m_nmodes0,
                    m_nmodes1, 1.0, m_base1.get(), m_nquad1, wsp.get(),
                    m_nquad2 * m_numElmt * m_nmodes0, 0.0, wsp2.get(),
                    m_nquad1);
 
        // Perform summation over '0' direction
        Blas::Dgemm('N', 'T', m_nquad0, m_nquad1 * m_nquad2 * m_numElmt,
                    m_nmodes0, 1.0, m_base0.get(), m_nquad0, wsp2.get(),
                    m_nquad1 * m_nquad2 * m_numElmt, 0.0, output0.get(),
                    m_nquad0);
    }

References ASSERTL1, Blas::Daxpy(), Blas::Dgemm(), m_base0, m_base1, m_base2, m_nmodes0, m_nmodes1, m_nmodes2, m_nquad0, m_nquad1, m_nquad2, Nektar::Collections::Operator::m_numElmt, m_sortTopVertex, Nektar::Collections::Operator::m_stdExp, Nektar::Collections::Operator::m_wspSize, and Vmath::Zero().

◆ operator()() [2/2]

void Nektar::Collections::BwdTrans_SumFac_Prism::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 1027 of file BwdTrans.cpp.

    {
        ASSERTL0(false, "Not valid for this operator.");
    }

References ASSERTL0.

Member Data Documentation

◆ m_base0

Array<OneD, const NekDouble> Nektar::Collections::BwdTrans_SumFac_Prism::m_base0

protected

Definition at line 1042 of file BwdTrans.cpp.

Referenced by operator()().

◆ m_base1

Array<OneD, const NekDouble> Nektar::Collections::BwdTrans_SumFac_Prism::m_base1

protected

Definition at line 1043 of file BwdTrans.cpp.

Referenced by operator()().

◆ m_base2

Array<OneD, const NekDouble> Nektar::Collections::BwdTrans_SumFac_Prism::m_base2

protected

Definition at line 1044 of file BwdTrans.cpp.

Referenced by operator()().

◆ m_nmodes0

const int Nektar::Collections::BwdTrans_SumFac_Prism::m_nmodes0

protected

Definition at line 1039 of file BwdTrans.cpp.

Referenced by BwdTrans_SumFac_Prism(), and operator()().

◆ m_nmodes1

const int Nektar::Collections::BwdTrans_SumFac_Prism::m_nmodes1

protected

Definition at line 1040 of file BwdTrans.cpp.

Referenced by BwdTrans_SumFac_Prism(), and operator()().

◆ m_nmodes2

const int Nektar::Collections::BwdTrans_SumFac_Prism::m_nmodes2

protected

Definition at line 1041 of file BwdTrans.cpp.

Referenced by operator()().

◆ m_nquad0

const int Nektar::Collections::BwdTrans_SumFac_Prism::m_nquad0

protected

Definition at line 1036 of file BwdTrans.cpp.

Referenced by operator()().

◆ m_nquad1

const int Nektar::Collections::BwdTrans_SumFac_Prism::m_nquad1

protected

Definition at line 1037 of file BwdTrans.cpp.

Referenced by BwdTrans_SumFac_Prism(), and operator()().

◆ m_nquad2

const int Nektar::Collections::BwdTrans_SumFac_Prism::m_nquad2

protected

Definition at line 1038 of file BwdTrans.cpp.

Referenced by BwdTrans_SumFac_Prism(), and operator()().

◆ m_sortTopVertex

bool Nektar::Collections::BwdTrans_SumFac_Prism::m_sortTopVertex

protected

Definition at line 1045 of file BwdTrans.cpp.

Referenced by BwdTrans_SumFac_Prism(), and operator()().

Public Member Functions

Protected Attributes

Private Member Functions

Additional Inherited Members

Detailed Description

Constructor & Destructor Documentation

◆ ~BwdTrans_SumFac_Prism()

◆ BwdTrans_SumFac_Prism()

Member Function Documentation

◆ operator()() [1/2]

◆ operator()() [2/2]

Member Data Documentation

◆ m_base0

◆ m_base1

◆ m_base2

◆ m_nmodes0

◆ m_nmodes1

◆ m_nmodes2

◆ m_nquad0

◆ m_nquad1

◆ m_nquad2

◆ m_sortTopVertex