Python/StdExpansion.cpp File Reference

#include <StdRegions/StdExpansion.h>
#include <LibUtilities/Python/BasicUtils/SharedArray.hpp>
#include <LibUtilities/Python/LinearAlgebra/NekMatrix.hpp>
#include <LibUtilities/Python/NekPyConfig.hpp>

Go to the source code of this file.

Functions
Array< OneD, NekDouble >	StdExpansion_FwdTrans (StdExpansionSharedPtr exp, const Array< OneD, const NekDouble > &in)
Array< OneD, NekDouble >	StdExpansion_BwdTrans (StdExpansionSharedPtr exp, const Array< OneD, const NekDouble > &in)
Array< OneD, NekDouble >	StdExpansion_IProductWRTBase (StdExpansionSharedPtr exp, const Array< OneD, const NekDouble > &in)
NekDouble	StdExpansion_PhysEvaluate (StdExpansionSharedPtr exp, const Array< OneD, const NekDouble > &coords, const Array< OneD, const NekDouble > &physvals)
NekDouble	StdExpansion_L2 (StdExpansionSharedPtr exp, const Array< OneD, const NekDouble > &in)
NekDouble	StdExpansion_L2_Error (StdExpansionSharedPtr exp, const Array< OneD, const NekDouble > &in, const Array< OneD, const NekDouble > &err)
py::tuple	StdExpansion_GetCoords (StdExpansionSharedPtr exp)
py::tuple	StdExpansion_PhysDeriv (StdExpansionSharedPtr exp, const Array< OneD, const NekDouble > &inarray)
py::array	StdExpansion_GenMatrix (std::shared_ptr< StdExpansion > &stdExp, const StdMatrixKey &mkey)
py::array	StdExpansion_GetStdMatrix (std::shared_ptr< StdExpansion > &stdExp, const StdMatrixKey &mkey)
void	export_StdExpansion (py::module &m)

Function Documentation

export_StdExpansion()

void export_StdExpansion

(

py::module &

m

)

Definition at line 170 of file Python/StdExpansion.cpp.

{
    py::class_<StdExpansion, std::shared_ptr<StdExpansion>>(m, "StdExpansion")
 
        .def("GetNcoeffs", &StdExpansion::GetNcoeffs)
        .def("GetTotPoints", &StdExpansion::GetTotPoints)
        .def("GetBasisType", &StdExpansion::GetBasisType)
        .def("GetPointsType", &StdExpansion::GetPointsType)
        .def("GetNverts", &StdExpansion::GetNverts)
        .def("GetNtraces", &StdExpansion::GetNtraces)
        .def("DetShapeType", &StdExpansion::DetShapeType)
        .def("GetShapeDimension", &StdExpansion::GetShapeDimension)
        .def("Integral", &StdExpansion::Integral)
 
        .def("GetBasis", &StdExpansion::GetBasis)
 
        .def("GenMatrix", &StdExpansion_GenMatrix)
        .def("GetStdMatrix", &StdExpansion_GetStdMatrix)
 
        .def("FwdTrans", &StdExpansion_FwdTrans)
        .def("BwdTrans", &StdExpansion_BwdTrans)
        .def("IProductWRTBase", &StdExpansion_IProductWRTBase)
 
        .def("PhysEvaluate", &StdExpansion_PhysEvaluate)
        .def("L2", &StdExpansion_L2)
        .def("L2", &StdExpansion_L2_Error)
 
        .def("GetCoords", &StdExpansion_GetCoords)
 
        .def("PhysDeriv", &StdExpansion_PhysDeriv);
}

References StdExpansion_BwdTrans(), StdExpansion_FwdTrans(), StdExpansion_GenMatrix(), StdExpansion_GetCoords(), StdExpansion_GetStdMatrix(), StdExpansion_IProductWRTBase(), StdExpansion_L2(), StdExpansion_L2_Error(), StdExpansion_PhysDeriv(), and StdExpansion_PhysEvaluate().

Referenced by PYBIND11_MODULE().

StdExpansion_BwdTrans()

Array< OneD, NekDouble > StdExpansion_BwdTrans	(	StdExpansionSharedPtr	exp,
		const Array< OneD, const NekDouble > &	in
	)

Definition at line 52 of file Python/StdExpansion.cpp.

{
    Array<OneD, NekDouble> out(exp->GetTotPoints());
    exp->BwdTrans(in, out);
    return out;
}

Referenced by export_StdExpansion().

StdExpansion_FwdTrans()

Array< OneD, NekDouble > StdExpansion_FwdTrans	(	StdExpansionSharedPtr	exp,
		const Array< OneD, const NekDouble > &	in
	)

Definition at line 44 of file Python/StdExpansion.cpp.

{
    Array<OneD, NekDouble> out(exp->GetNcoeffs());
    exp->FwdTrans(in, out);
    return out;
}

Referenced by export_StdExpansion().

StdExpansion_GenMatrix()

py::array StdExpansion_GenMatrix	(	std::shared_ptr< StdExpansion > &	stdExp,
		const StdMatrixKey &	mkey
	)

Definition at line 149 of file Python/StdExpansion.cpp.

{
    auto mat = stdExp->GenMatrix(mkey);
    // A bit wasteful but at the moment we can't easily keep hold of the
    // NekMatrix returned as a std::shared_ptr, because pybind11 doesn't like
    // std::shared_ptr around types that have custom casters (see issue #787).
    return py::array({mat->GetRows(), mat->GetColumns()},
                     {sizeof(NekDouble), mat->GetRows() * sizeof(NekDouble)},
                     mat->GetRawPtr());
}

Referenced by export_StdExpansion().

StdExpansion_GetCoords()

py::tuple StdExpansion_GetCoords

(

StdExpansionSharedPtr

exp

)

Definition at line 88 of file Python/StdExpansion.cpp.

{
    int nPhys   = exp->GetTotPoints();
    int coordim = exp->GetCoordim();
 
    std::vector<Array<OneD, NekDouble>> coords(coordim);
    for (int i = 0; i < coordim; ++i)
    {
        coords[i] = Array<OneD, NekDouble>(nPhys);
    }
 
    switch (coordim)
    {
        case 1:
            exp->GetCoords(coords[0]);
            return py::make_tuple(coords[0]);
            break;
        case 2:
            exp->GetCoords(coords[0], coords[1]);
            return py::make_tuple(coords[0], coords[1]);
            break;
        case 3:
            exp->GetCoords(coords[0], coords[1], coords[2]);
            return py::make_tuple(coords[0], coords[1], coords[2]);
            break;
    }
 
    return py::tuple();
}

Referenced by export_StdExpansion().

StdExpansion_GetStdMatrix()

py::array StdExpansion_GetStdMatrix	(	std::shared_ptr< StdExpansion > &	stdExp,
		const StdMatrixKey &	mkey
	)

Definition at line 161 of file Python/StdExpansion.cpp.

{
    auto mat = stdExp->GetStdMatrix(mkey);
    return py::array({mat->GetRows(), mat->GetColumns()},
                     {sizeof(NekDouble), mat->GetRows() * sizeof(NekDouble)},
                     mat->GetRawPtr());
}

Referenced by export_StdExpansion().

StdExpansion_IProductWRTBase()

Array< OneD, NekDouble > StdExpansion_IProductWRTBase	(	StdExpansionSharedPtr	exp,
		const Array< OneD, const NekDouble > &	in
	)

Definition at line 60 of file Python/StdExpansion.cpp.

{
    Array<OneD, NekDouble> out(exp->GetNcoeffs());
    exp->IProductWRTBase(in, out);
    return out;
}

Referenced by export_StdExpansion().

StdExpansion_L2()

NekDouble StdExpansion_L2	(	StdExpansionSharedPtr	exp,
		const Array< OneD, const NekDouble > &	in
	)

Definition at line 75 of file Python/StdExpansion.cpp.

{
    return exp->L2(in);
}

Referenced by export_StdExpansion().

StdExpansion_L2_Error()

NekDouble StdExpansion_L2_Error	(	StdExpansionSharedPtr	exp,
		const Array< OneD, const NekDouble > &	in,
		const Array< OneD, const NekDouble > &	err
	)

Definition at line 81 of file Python/StdExpansion.cpp.

{
    return exp->L2(in, err);
}

Referenced by export_StdExpansion().

StdExpansion_PhysDeriv()

py::tuple StdExpansion_PhysDeriv	(	StdExpansionSharedPtr	exp,
		const Array< OneD, const NekDouble > &	inarray
	)

Definition at line 118 of file Python/StdExpansion.cpp.

{
    int nPhys   = exp->GetTotPoints();
    int coordim = exp->GetCoordim();
 
    std::vector<Array<OneD, NekDouble>> derivs(coordim);
    for (int i = 0; i < coordim; ++i)
    {
        derivs[i] = Array<OneD, NekDouble>(nPhys);
    }
 
    switch (coordim)
    {
        case 1:
            exp->PhysDeriv(inarray, derivs[0]);
            return py::make_tuple(derivs[0]);
            break;
        case 2:
            exp->PhysDeriv(inarray, derivs[0], derivs[1]);
            return py::make_tuple(derivs[0], derivs[1]);
            break;
        case 3:
            exp->PhysDeriv(inarray, derivs[0], derivs[1], derivs[2]);
            return py::make_tuple(derivs[0], derivs[1], derivs[2]);
            break;
    }
 
    return py::tuple();
}

Referenced by export_StdExpansion().

StdExpansion_PhysEvaluate()

NekDouble StdExpansion_PhysEvaluate	(	StdExpansionSharedPtr	exp,
		const Array< OneD, const NekDouble > &	coords,
		const Array< OneD, const NekDouble > &	physvals
	)

Definition at line 68 of file Python/StdExpansion.cpp.

{
    return exp->PhysEvaluate(coords, physvals);
}

Referenced by export_StdExpansion().