Nektar::Collections::BwdTrans_SumFac_Prism Class Reference

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

Inheritance diagram for Nektar::Collections::BwdTrans_SumFac_Prism:

Collaboration diagram for Nektar::Collections::BwdTrans_SumFac_Prism:

Public Member Functions
virtual	~BwdTrans_SumFac_Prism ()
virtual void	operator() (const Array< OneD, const NekDouble > &input, Array< OneD, NekDouble > &output, Array< OneD, NekDouble > &output1, Array< OneD, NekDouble > &output2, Array< OneD, NekDouble > &wsp)
	Perform operation. More...
virtual void	operator() (int dir, const Array< OneD, const NekDouble > &input, Array< OneD, NekDouble > &output, Array< OneD, NekDouble > &wsp)
Public Member Functions inherited from Nektar::Collections::Operator
	Operator (std::vector< StdRegions::StdExpansionSharedPtr > pCollExp, boost::shared_ptr< CoalescedGeomData > GeomData)
	Constructor. More...
virtual COLLECTIONS_EXPORT	~Operator ()
int	GetWspSize ()
	Get the size of the required workspace. More...

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
StdRegions::StdExpansionSharedPtr	m_stdExp
unsigned int	m_numElmt
unsigned int	m_wspSize

Private Member Functions
	BwdTrans_SumFac_Prism (vector< StdRegions::StdExpansionSharedPtr > pCollExp, CoalescedGeomDataSharedPtr pGeomData)

Detailed Description

Backward transform operator using sum-factorisation (Prism)

Definition at line 883 of file BwdTrans.cpp.

Constructor & Destructor Documentation

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

inlinevirtual

Definition at line 888 of file BwdTrans.cpp.

        {
        }

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

inlineprivate

Definition at line 985 of file BwdTrans.cpp.

References Nektar::LibUtilities::eModified_A.

            : Operator  (pCollExp, pGeomData),
              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;
            }

        }

Member Function Documentation

virtual 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  )

inlinevirtual

Perform operation.

Implements Nektar::Collections::Operator.

Definition at line 892 of file BwdTrans.cpp.

References ASSERTL1, and Vmath::Zero().

        {
            ASSERTL1(wsp.num_elements() == 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[0]+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, output.get(), m_nquad0);
        }

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

inlinevirtual

Implements Nektar::Collections::Operator.

Definition at line 962 of file BwdTrans.cpp.

References ASSERTL0.

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

Member Data Documentation

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

protected

Definition at line 979 of file BwdTrans.cpp.

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

protected

Definition at line 980 of file BwdTrans.cpp.

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

protected

Definition at line 981 of file BwdTrans.cpp.

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

protected

Definition at line 976 of file BwdTrans.cpp.

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

protected

Definition at line 977 of file BwdTrans.cpp.

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

protected

Definition at line 978 of file BwdTrans.cpp.

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

protected

Definition at line 973 of file BwdTrans.cpp.

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

protected

Definition at line 974 of file BwdTrans.cpp.

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

protected

Definition at line 975 of file BwdTrans.cpp.

bool Nektar::Collections::BwdTrans_SumFac_Prism::m_sortTopVertex

protected

Definition at line 982 of file BwdTrans.cpp.