Nektar::Collections::BwdTrans_SumFac_Hex Class Reference

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

Inheritance diagram for Nektar::Collections::BwdTrans_SumFac_Hex:

Collaboration diagram for Nektar::Collections::BwdTrans_SumFac_Hex:

Public Member Functions
virtual	~BwdTrans_SumFac_Hex ()
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
const bool	m_colldir0
const bool	m_colldir1
const bool	m_colldir2
Protected Attributes inherited from Nektar::Collections::Operator
StdRegions::StdExpansionSharedPtr	m_stdExp
unsigned int	m_numElmt
unsigned int	m_wspSize

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

Detailed Description

Backward transform operator using sum-factorisation (Hex)

Definition at line 596 of file BwdTrans.cpp.

Constructor & Destructor Documentation

virtual Nektar::Collections::BwdTrans_SumFac_Hex::~BwdTrans_SumFac_Hex ( )

inlinevirtual

Definition at line 601 of file BwdTrans.cpp.

        {
        }

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

inlineprivate

Definition at line 677 of file BwdTrans.cpp.

            : Operator  (pCollExp, pGeomData),
              m_nquad0  (pCollExp[0]->GetNumPoints(0)),
              m_nquad1  (pCollExp[0]->GetNumPoints(1)),
              m_nquad2  (pCollExp[0]->GetNumPoints(2)),
              m_nmodes0 (pCollExp[0]->GetBasisNumModes(0)),
              m_nmodes1 (pCollExp[0]->GetBasisNumModes(1)),
              m_nmodes2 (pCollExp[0]->GetBasisNumModes(2)),
              m_base0   (pCollExp[0]->GetBasis(0)->GetBdata()),
              m_base1   (pCollExp[0]->GetBasis(1)->GetBdata()),
              m_base2   (pCollExp[0]->GetBasis(2)->GetBdata()),
              m_colldir0(pCollExp[0]->GetBasis(0)->Collocation()),
              m_colldir1(pCollExp[0]->GetBasis(1)->Collocation()),
              m_colldir2(pCollExp[0]->GetBasis(2)->Collocation())
        {
            m_wspSize =  m_numElmt*m_nmodes0*(m_nmodes1*m_nquad2 +
                                              m_nquad1*m_nquad2);
        }

Member Function Documentation

virtual void Nektar::Collections::BwdTrans_SumFac_Hex::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 605 of file BwdTrans.cpp.

References ASSERTL1, and Vmath::Vcopy().

        {

            if(m_colldir0 && m_colldir1 && m_colldir2)
            {
                Vmath::Vcopy(m_numElmt * m_nmodes0 * m_nmodes1 * m_nmodes2,
                             input.get(),  1,
                             output.get(), 1);
            }
            else
            {
                ASSERTL1(wsp.num_elements() == m_wspSize,
                         "Incorrect workspace size");

                // Assign second half of workspace for 2nd DGEMM operation.
                int totmodes  = m_nmodes0*m_nmodes1*m_nmodes2;

                Array<OneD, NekDouble> wsp2
                                = wsp + m_nmodes0*m_nmodes1*m_nquad2*m_numElmt;

                //loop over elements  and do bwd trans wrt c
                for(int n = 0; n < m_numElmt; ++n)
                {
                    Blas::Dgemm('N', 'T', m_nquad2, m_nmodes0*m_nmodes1,
                                m_nmodes2, 1.0, m_base2.get(), m_nquad2,
                                &input[n*totmodes], m_nmodes0*m_nmodes1, 0.0,
                                &wsp[n*m_nquad2], m_nquad2*m_numElmt);
                }

                // trans wrt b
                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);

                // trans wrt a
                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_Hex::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 653 of file BwdTrans.cpp.

References ASSERTL0.

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

Member Data Documentation

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

protected

Definition at line 669 of file BwdTrans.cpp.

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

protected

Definition at line 670 of file BwdTrans.cpp.

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

protected

Definition at line 671 of file BwdTrans.cpp.

const bool Nektar::Collections::BwdTrans_SumFac_Hex::m_colldir0

protected

Definition at line 672 of file BwdTrans.cpp.

const bool Nektar::Collections::BwdTrans_SumFac_Hex::m_colldir1

protected

Definition at line 673 of file BwdTrans.cpp.

const bool Nektar::Collections::BwdTrans_SumFac_Hex::m_colldir2

protected

Definition at line 674 of file BwdTrans.cpp.

const int Nektar::Collections::BwdTrans_SumFac_Hex::m_nmodes0

protected

Definition at line 666 of file BwdTrans.cpp.

const int Nektar::Collections::BwdTrans_SumFac_Hex::m_nmodes1

protected

Definition at line 667 of file BwdTrans.cpp.

const int Nektar::Collections::BwdTrans_SumFac_Hex::m_nmodes2

protected

Definition at line 668 of file BwdTrans.cpp.

const int Nektar::Collections::BwdTrans_SumFac_Hex::m_nquad0

protected

Definition at line 663 of file BwdTrans.cpp.

const int Nektar::Collections::BwdTrans_SumFac_Hex::m_nquad1

protected

Definition at line 664 of file BwdTrans.cpp.

const int Nektar::Collections::BwdTrans_SumFac_Hex::m_nquad2

protected

Definition at line 665 of file BwdTrans.cpp.