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

void	CheckFactors (StdRegions::FactorMap factors, int coll_phys_offset) override
	Check the validity of the supplied factor map. 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	CheckFactors (StdRegions::FactorMap factors, int coll_phys_offset)=0
	Check the validity of the supplied factor map. 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 ()

unsigned int	GetOutputSize ()

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...

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::Operator
virtual int	v_GetInputSize ()
	This purely virtual function needs to be set-up for every operator inside Collections. It is responsible for returning the size of input collection, that the operator is applied on either in physical or coefficient space. More...

virtual int	v_GetOutputSize ()
	This purely virtual function needs to be set-up for every operator inside Collections. It is responsible for returning the output size either in physical or coefficient space of an operator inside Collections. More...

Protected Member Functions inherited from Nektar::Collections::BwdTrans_Helper
	BwdTrans_Helper ()

Detailed Description

Backward transform operator using sum-factorisation (Prism)

Definition at line 1003 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 1108 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

◆ CheckFactors()

void Nektar::Collections::BwdTrans_SumFac_Prism::CheckFactors	(	StdRegions::FactorMap	factors,
		int	coll_phys_offset
	)

inlineoverridevirtual

Check the validity of the supplied factor map.

Implements Nektar::Collections::Operator.

Definition at line 1089 of file BwdTrans.cpp.

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

References ASSERTL0.

◆ 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 1011 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 1081 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 1102 of file BwdTrans.cpp.

Referenced by operator()().

◆ m_base1

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

protected

Definition at line 1103 of file BwdTrans.cpp.

Referenced by operator()().

◆ m_base2

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

protected

Definition at line 1104 of file BwdTrans.cpp.

Referenced by operator()().

◆ m_nmodes0

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

protected

Definition at line 1099 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 1100 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 1101 of file BwdTrans.cpp.

Referenced by operator()().

◆ m_nquad0

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

protected

Definition at line 1096 of file BwdTrans.cpp.

Referenced by operator()().

◆ m_nquad1

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

protected

Definition at line 1097 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 1098 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 1105 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

◆ CheckFactors()

◆ 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