Python/ExpList.cpp File Reference

#include <LibUtilities/Python/BasicUtils/SharedArray.hpp>
#include <LibUtilities/Python/NekPyConfig.hpp>
#include <LocalRegions/MatrixKey.h>
#include <MultiRegions/ExpList.h>
#include <fstream>
#include <sstream>
#include <string>

Go to the source code of this file.

Functions
ExpansionSharedPtr	ExpList_GetExp (ExpListSharedPtr exp, int i)
void	ExpList_WriteVTK (ExpListSharedPtr exp, std::string filename)
Array< OneD, NekDouble >	ExpList_FwdTrans (ExpListSharedPtr exp, const Array< OneD, const NekDouble > &in)
Array< OneD, NekDouble >	ExpList_BwdTrans (ExpListSharedPtr exp, const Array< OneD, const NekDouble > &in)
Array< OneD, NekDouble >	ExpList_IProductWRTBase (ExpListSharedPtr exp, const Array< OneD, const NekDouble > &in)
Array< OneD, NekDouble >	ExpList_MultiplyByInvMassMatrix (ExpListSharedPtr exp, const Array< OneD, const NekDouble > &in)
Array< OneD, NekDouble >	ExpList_HelmSolve (ExpListSharedPtr exp, const Array< OneD, const NekDouble > &in, const StdRegions::ConstFactorMap &constFactorMap, const StdRegions::VarCoeffMap &varCoeffMap)
NekDouble	ExpList_L2 (ExpListSharedPtr exp, const Array< OneD, const NekDouble > &in)
NekDouble	ExpList_L2_Error (ExpListSharedPtr exp, const Array< OneD, const NekDouble > &in, const Array< OneD, const NekDouble > &err)
NekDouble	ExpList_Linf (ExpListSharedPtr exp, const Array< OneD, const NekDouble > &in)
NekDouble	ExpList_Linf_Error (ExpListSharedPtr exp, const Array< OneD, const NekDouble > &in, const Array< OneD, const NekDouble > &err)
py::tuple	ExpList_GetCoords (ExpListSharedPtr exp)
void	ExpList_SetPhysArray (ExpListSharedPtr exp, Array< OneD, NekDouble > inarray)
void	ExpList_SetPhys (ExpListSharedPtr exp, const Array< OneD, const NekDouble > &inarray)
const Array< OneD, const NekDouble >	ExpList_GetPhys (ExpListSharedPtr exp)
void	ExpList_SetCoeffsArray (ExpListSharedPtr exp, Array< OneD, NekDouble > inarray)
const Array< OneD, const NekDouble >	ExpList_GetCoeffs (ExpListSharedPtr exp)
NekDouble	ExpList_Integral (ExpListSharedPtr exp)
std::string	ExpList_GetPhysAddress (ExpListSharedPtr exp)
void	ExpList_ResetManagers (ExpListSharedPtr exp)
void	ExpList_LoadField (ExpListSharedPtr exp, std::string filename, std::string varName)
void	export_ExpList (py::module &m)

Function Documentation

ExpList_BwdTrans()

Array< OneD, NekDouble > ExpList_BwdTrans	(	ExpListSharedPtr	exp,
		const Array< OneD, const NekDouble > &	in
	)

Definition at line 80 of file Python/ExpList.cpp.

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

Referenced by export_ExpList().

ExpList_FwdTrans()

Array< OneD, NekDouble > ExpList_FwdTrans	(	ExpListSharedPtr	exp,
		const Array< OneD, const NekDouble > &	in
	)

Definition at line 72 of file Python/ExpList.cpp.

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

Referenced by export_ExpList().

ExpList_GetCoeffs()

const Array< OneD, const NekDouble > ExpList_GetCoeffs

(

ExpListSharedPtr

exp

)

Definition at line 192 of file Python/ExpList.cpp.

{
    return exp->GetCoeffs();
}

Referenced by export_ExpList().

ExpList_GetCoords()

py::tuple ExpList_GetCoords

(

ExpListSharedPtr

exp

)

Definition at line 140 of file Python/ExpList.cpp.

{
    size_t nPhys   = exp->GetNpoints();
    size_t coordim = exp->GetCoordim(0);
 
    std::vector<Array<OneD, NekDouble>> coords(coordim);
    for (size_t 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_ExpList().

ExpList_GetExp()

ExpansionSharedPtr ExpList_GetExp	(	ExpListSharedPtr	exp,
		int	i
	)

Definition at line 54 of file Python/ExpList.cpp.

{
    return exp->GetExp(i);
}

Referenced by export_ExpList().

ExpList_GetPhys()

const Array< OneD, const NekDouble > ExpList_GetPhys

(

ExpListSharedPtr

exp

)

Definition at line 181 of file Python/ExpList.cpp.

{
    return exp->GetPhys();
}

Referenced by export_ExpList().

ExpList_GetPhysAddress()

std::string ExpList_GetPhysAddress

(

ExpListSharedPtr

exp

)

Definition at line 202 of file Python/ExpList.cpp.

{
    std::stringstream ss;
    ss << static_cast<const void *>(&(exp->GetPhys()[0]));
    return ss.str();
}

Referenced by export_ExpList().

ExpList_HelmSolve()

Array< OneD, NekDouble > ExpList_HelmSolve	(	ExpListSharedPtr	exp,
		const Array< OneD, const NekDouble > &	in,
		const StdRegions::ConstFactorMap &	constFactorMap,
		const StdRegions::VarCoeffMap &	varCoeffMap
	)

Definition at line 104 of file Python/ExpList.cpp.

{
    Array<OneD, NekDouble> out(exp->GetNcoeffs(), 0.0);
    exp->HelmSolve(in, out, constFactorMap, varCoeffMap);
    return out;
}

Referenced by export_ExpList().

ExpList_Integral()

NekDouble ExpList_Integral

(

ExpListSharedPtr

exp

)

Definition at line 197 of file Python/ExpList.cpp.

{
    return exp->Integral();
}

Referenced by export_ExpList().

ExpList_IProductWRTBase()

Array< OneD, NekDouble > ExpList_IProductWRTBase	(	ExpListSharedPtr	exp,
		const Array< OneD, const NekDouble > &	in
	)

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

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

Referenced by export_ExpList().

ExpList_L2()

NekDouble ExpList_L2	(	ExpListSharedPtr	exp,
		const Array< OneD, const NekDouble > &	in
	)

Definition at line 114 of file Python/ExpList.cpp.

{
    return exp->L2(in);
}

Referenced by export_ExpList().

ExpList_L2_Error()

NekDouble ExpList_L2_Error	(	ExpListSharedPtr	exp,
		const Array< OneD, const NekDouble > &	in,
		const Array< OneD, const NekDouble > &	err
	)

Definition at line 120 of file Python/ExpList.cpp.

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

Referenced by export_ExpList().

ExpList_Linf()

NekDouble ExpList_Linf	(	ExpListSharedPtr	exp,
		const Array< OneD, const NekDouble > &	in
	)

Definition at line 127 of file Python/ExpList.cpp.

{
    return exp->Linf(in);
}

Referenced by export_ExpList().

ExpList_Linf_Error()

NekDouble ExpList_Linf_Error	(	ExpListSharedPtr	exp,
		const Array< OneD, const NekDouble > &	in,
		const Array< OneD, const NekDouble > &	err
	)

Definition at line 133 of file Python/ExpList.cpp.

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

Referenced by export_ExpList().

ExpList_LoadField()

void ExpList_LoadField	(	ExpListSharedPtr	exp,
		std::string	filename,
		std::string	varName
	)

Definition at line 218 of file Python/ExpList.cpp.

{
    size_t nExp = exp->GetExpSize();
    Array<OneD, int> elementGIDs(nExp);
 
    // Construct a map from element locations to global IDs
    for (size_t i = 0; i < nExp; ++i)
    {
        elementGIDs[i] = exp->GetExp(i)->GetGeom()->GetGlobalID();
    }
 
    std::vector<LibUtilities::FieldDefinitionsSharedPtr> def;
    std::vector<std::vector<NekDouble>> data;
    LibUtilities::FieldIOSharedPtr fldIO =
        LibUtilities::FieldIO::CreateForFile(exp->GetSession(), filename);
    fldIO->Import(filename, def, data, LibUtilities::NullFieldMetaDataMap,
                  elementGIDs);
 
    int idx = -1;
 
    Vmath::Zero(exp->GetNcoeffs(), exp->UpdateCoeffs(), 1);
 
    // Loop over all the expansions
    for (size_t i = 0; i < def.size(); ++i)
    {
        // Find the index of the required field in the expansion segment
        for (size_t j = 0; j < def[i]->m_fields.size(); ++j)
        {
            if (def[i]->m_fields[j] == varName)
            {
                idx = j;
            }
        }
 
        if (idx >= 0)
        {
            exp->ExtractDataToCoeffs(def[i], data[i], def[i]->m_fields[idx],
                                     exp->UpdateCoeffs());
        }
        else
        {
            std::cout << "Field " + varName + " not found." << std::endl;
        }
    }
 
    exp->BwdTrans(exp->GetCoeffs(), exp->UpdatePhys());
}

References Nektar::LibUtilities::NullFieldMetaDataMap, and Vmath::Zero().

Referenced by export_ExpList().

ExpList_MultiplyByInvMassMatrix()

Array< OneD, NekDouble > ExpList_MultiplyByInvMassMatrix	(	ExpListSharedPtr	exp,
		const Array< OneD, const NekDouble > &	in
	)

Definition at line 96 of file Python/ExpList.cpp.

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

Referenced by export_ExpList().

ExpList_ResetManagers()

void ExpList_ResetManagers

(

ExpListSharedPtr

exp

)

Definition at line 209 of file Python/ExpList.cpp.

{
    exp->ClearGlobalLinSysManager();
    LibUtilities::NekManager<LocalRegions::MatrixKey, DNekScalMat,
                             LocalRegions::MatrixKey::opLess>::ClearManager();
    LibUtilities::NekManager<LocalRegions::MatrixKey, DNekScalBlkMat,
                             LocalRegions::MatrixKey::opLess>::ClearManager();
}

Referenced by export_ExpList().

ExpList_SetCoeffsArray()

void ExpList_SetCoeffsArray	(	ExpListSharedPtr	exp,
		Array< OneD, NekDouble >	inarray
	)

Definition at line 186 of file Python/ExpList.cpp.

{
    exp->SetCoeffsArray(inarray);
}

Referenced by export_ExpList().

ExpList_SetPhys()

void ExpList_SetPhys	(	ExpListSharedPtr	exp,
		const Array< OneD, const NekDouble > &	inarray
	)

Definition at line 175 of file Python/ExpList.cpp.

{
    exp->SetPhys(inarray);
}

Referenced by export_ExpList().

ExpList_SetPhysArray()

void ExpList_SetPhysArray	(	ExpListSharedPtr	exp,
		Array< OneD, NekDouble >	inarray
	)

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

{
    exp->SetPhysArray(inarray);
}

Referenced by export_ExpList().

ExpList_WriteVTK()

void ExpList_WriteVTK	(	ExpListSharedPtr	exp,
		std::string	filename
	)

Definition at line 59 of file Python/ExpList.cpp.

{
    std::ofstream out(filename.c_str());
    exp->WriteVtkHeader(out);
    size_t nExp = exp->GetExpSize();
    for (size_t i = 0; i < nExp; ++i)
    {
        exp->WriteVtkPieceHeader(out, i);
        exp->WriteVtkPieceFooter(out, i);
    }
    exp->WriteVtkFooter(out);
}

Referenced by export_ExpList().

export_ExpList()

void export_ExpList

(

py::module &

m

)

Definition at line 267 of file Python/ExpList.cpp.

{
    int (ExpList::*GetNcoeffs)() const = &ExpList::GetNcoeffs;
 
    py::class_<ExpList, std::shared_ptr<ExpList>>(m, "ExpList")
        .def(py::init<const LibUtilities::SessionReaderSharedPtr &,
                      const SpatialDomains::MeshGraphSharedPtr &>())
 
        // Query points and offset information
        .def("GetExp", &ExpList_GetExp)
        .def("GetExpSize", &ExpList::GetExpSize)
        .def("GetNpoints", &ExpList::GetNpoints)
        .def("GetNcoeffs", GetNcoeffs)
        .def("GetCoords", &ExpList_GetCoords)
        .def("GetPhys_Offset", &ExpList::GetPhys_Offset)
        .def("GetCoeff_Offset", &ExpList::GetCoeff_Offset)
 
        // Evaluations
        .def("PhysEvaluate", &ExpList::PhysEvaluate)
        .def("LoadField", &ExpList_LoadField)
 
        // Operators
        .def("FwdTrans", &ExpList_FwdTrans)
        .def("BwdTrans", &ExpList_BwdTrans)
        .def("IProductWRTBase", &ExpList_IProductWRTBase)
        .def("MultiplyByInvMassMatrix", &ExpList_MultiplyByInvMassMatrix)
        .def("HelmSolve", &ExpList_HelmSolve, py::arg("in"),
             py::arg("constFactorMap") = StdRegions::NullConstFactorMap,
             py::arg("varCoeffMap")    = StdRegions::NullVarCoeffMap)
 
        // Error norms
        .def("L2", &ExpList_L2)
        .def("L2", &ExpList_L2_Error)
        .def("Linf", &ExpList_Linf)
        .def("Linf", &ExpList_Linf_Error)
 
        // Storage setups
        .def("SetPhysArray", &ExpList_SetPhysArray)
        .def("SetPhys", &ExpList_SetPhys)
        .def("GetPhys", &ExpList_GetPhys)
        .def("SetCoeffsArray", &ExpList_SetCoeffsArray)
        .def("GetCoeffs", &ExpList_GetCoeffs)
        .def("SetPhysState", &ExpList::SetPhysState)
        .def("GetPhysState", &ExpList::GetPhysState)
        .def("Integral", &ExpList_Integral)
        .def("GetPhysAddress", &ExpList_GetPhysAddress)
 
        .def_property("phys", &ExpList_GetPhys, &ExpList_SetPhys)
        .def_property("coeffs", &ExpList_GetCoeffs, &ExpList_SetCoeffsArray)
 
        // Misc functions
        .def("WriteVTK", &ExpList_WriteVTK)
        .def("ResetManagers", &ExpList_ResetManagers);
}

References ExpList_BwdTrans(), ExpList_FwdTrans(), ExpList_GetCoeffs(), ExpList_GetCoords(), ExpList_GetExp(), ExpList_GetPhys(), ExpList_GetPhysAddress(), ExpList_HelmSolve(), ExpList_Integral(), ExpList_IProductWRTBase(), ExpList_L2(), ExpList_L2_Error(), ExpList_Linf(), ExpList_Linf_Error(), ExpList_LoadField(), ExpList_MultiplyByInvMassMatrix(), ExpList_ResetManagers(), ExpList_SetCoeffsArray(), ExpList_SetPhys(), ExpList_SetPhysArray(), ExpList_WriteVTK(), Nektar::StdRegions::NullConstFactorMap, and Nektar::StdRegions::NullVarCoeffMap.

Referenced by PYBIND11_MODULE().