Nektar::Utilities::ProcessDisplacement Class Reference

#include <ProcessDisplacement.h>

Inheritance diagram for Nektar::Utilities::ProcessDisplacement:

Collaboration diagram for Nektar::Utilities::ProcessDisplacement:

Public Member Functions
	ProcessDisplacement (FieldSharedPtr f)
virtual	~ProcessDisplacement ()
virtual void	Process (po::variables_map &vm)
	Write mesh to output file. More...
virtual std::string	GetModuleName ()
Public Member Functions inherited from Nektar::Utilities::ProcessModule
	ProcessModule ()
	ProcessModule (FieldSharedPtr p_f)
	ProcessModule (MeshSharedPtr p_m)
Public Member Functions inherited from Nektar::Utilities::Module
	Module (FieldSharedPtr p_f)
void	RegisterConfig (string key, string value)
	Register a configuration option with a module. More...
void	PrintConfig ()
	Print out all configuration options for a module. More...
void	SetDefaults ()
	Sets default configuration options for those which have not been set. More...
bool	GetRequireEquiSpaced (void)
void	SetRequireEquiSpaced (bool pVal)
void	EvaluateTriFieldAtEquiSpacedPts (LocalRegions::ExpansionSharedPtr &exp, const Array< OneD, const NekDouble > &infield, Array< OneD, NekDouble > &outfield)
	Module (MeshSharedPtr p_m)
virtual void	Process ()=0
void	RegisterConfig (std::string key, std::string value)
void	PrintConfig ()
void	SetDefaults ()
MeshSharedPtr	GetMesh ()
virtual void	ProcessVertices ()
	Extract element vertices. More...
virtual void	ProcessEdges (bool ReprocessEdges=true)
	Extract element edges. More...
virtual void	ProcessFaces (bool ReprocessFaces=true)
	Extract element faces. More...
virtual void	ProcessElements ()
	Generate element IDs. More...
virtual void	ProcessComposites ()
	Generate composites. More...
virtual void	ClearElementLinks ()

Static Public Member Functions
static boost::shared_ptr< Module >	create (FieldSharedPtr f)
	Creates an instance of this class. More...

Static Public Attributes
static ModuleKey	className

Additional Inherited Members
Protected Member Functions inherited from Nektar::Utilities::Module
	Module ()
void	ReorderPrisms (PerMap &perFaces)
	Reorder node IDs so that prisms and tetrahedra are aligned correctly. More...
void	PrismLines (int prism, PerMap &perFaces, std::set< int > &prismsDone, std::vector< ElementSharedPtr > &line)
Protected Attributes inherited from Nektar::Utilities::Module
FieldSharedPtr	m_f
	Field object. More...
map< string, ConfigOption >	m_config
	List of configuration values. More...
bool	m_requireEquiSpaced
MeshSharedPtr	m_mesh
	Mesh object. More...
std::map< std::string, ConfigOption >	m_config
	List of configuration values. More...

Detailed Description

Definition at line 46 of file ProcessDisplacement.h.

Constructor & Destructor Documentation

Nektar::Utilities::ProcessDisplacement::ProcessDisplacement

(

FieldSharedPtr

f

)

Definition at line 107 of file ProcessDisplacement.cpp.

References Nektar::Utilities::Module::m_config.

                                                             :
        ProcessModule(f)
    {
        m_config["to"] = ConfigOption(
            false, "", "Name of file containing high order boundary");
        m_config["id"] = ConfigOption(
            false, "", "Boundary ID to calculate displacement for");
        m_config["usevertexids"] = ConfigOption(
            false, "0", "Use vertex IDs instead of face IDs for matching");
        f->m_declareExpansionAsContField = true;
        f->m_writeBndFld = true;
        f->m_fldToBnd = false;
    }

Nektar::Utilities::ProcessDisplacement::~ProcessDisplacement ( )

virtual

Definition at line 121 of file ProcessDisplacement.cpp.

    {
    }

Member Function Documentation

static boost::shared_ptr<Module> Nektar::Utilities::ProcessDisplacement::create ( FieldSharedPtr  f )

inlinestatic

Creates an instance of this class.

Definition at line 50 of file ProcessDisplacement.h.

References Nektar::MemoryManager< DataType >::AllocateSharedPtr().

                                                            {
        return MemoryManager<ProcessDisplacement>::AllocateSharedPtr(f);
    }

virtual std::string Nektar::Utilities::ProcessDisplacement::GetModuleName ( )

inlinevirtual

Implements Nektar::Utilities::Module.

Definition at line 61 of file ProcessDisplacement.h.

    {
        return "ProcessDisplacement";
    }

void Nektar::Utilities::ProcessDisplacement::Process ( po::variables_map &  vm )

virtual

Write mesh to output file.

Implements Nektar::Utilities::Module.

Definition at line 125 of file ProcessDisplacement.cpp.

References Nektar::LibUtilities::SessionReader::CreateInstance(), Nektar::StdRegions::StdExpansion::GetTotPoints(), Nektar::iterator, Nektar::Utilities::Module::m_config, Nektar::Utilities::Module::m_f, Nektar::SpatialDomains::MeshGraph::Read(), and Vmath::Vsub().

    {
        if (m_f->m_verbose)
        {
            if(m_f->m_comm->GetRank() == 0)
            {
                cout << "ProcessDisplacement: Calculating displacement..."
                     << endl;
            }
        }

        string toFile = m_config["to"].as<string>();

        if (toFile == "")
        {
            cout << "ProcessDisplacement: you must provide a file" << endl;
            return;
        }

        bool useVertexIds = m_config["usevertexids"].m_beenSet;

        vector<string> files;
        files.push_back(toFile);
        LibUtilities::SessionReaderSharedPtr bndSession =
            LibUtilities::SessionReader::CreateInstance(0, NULL, files);
        SpatialDomains::MeshGraphSharedPtr bndGraph =
            SpatialDomains::MeshGraph::Read(bndSession);

        // Try to find boundary condition expansion.
        int bndCondId = m_config["id"].as<int>();

        // FIXME: We should be storing boundary condition IDs
        // somewhere...
        m_f->m_bndRegionsToWrite.push_back(bndCondId);

        if (bndGraph->GetMeshDimension() == 1)
        {
            m_f->m_exp.push_back(m_f->AppendExpList(0, "v"));

            MultiRegions::ExpListSharedPtr bndCondExpU =
                m_f->m_exp[0]->GetBndCondExpansions()[bndCondId];
            MultiRegions::ExpListSharedPtr bndCondExpV =
                m_f->m_exp[1]->GetBndCondExpansions()[bndCondId];

            map<int, int> bndCondIds;
            for (int i = 0; i < bndCondExpU->GetExpSize(); ++i)
            {
                bndCondIds[bndCondExpU->GetExp(i)->GetGeom()->GetGlobalID()]
                    = i;
            }

            const SpatialDomains::SegGeomMap &tmp =
                bndGraph->GetAllSegGeoms();
            SpatialDomains::SegGeomMap::const_iterator sIt;

            for (sIt = tmp.begin(); sIt != tmp.end(); ++sIt)
            {
                map<int, int>::iterator mIt = bndCondIds.find(sIt->first);

                if (mIt == bndCondIds.end())
                {
                    cout << "Warning: couldn't find element "
                         << sIt->first << endl;
                    continue;
                }

                int e = mIt->second;

                SpatialDomains::SegGeomSharedPtr from =
                    boost::dynamic_pointer_cast<SpatialDomains::SegGeom>(
                        bndCondExpU->GetExp(e)->GetGeom());

                SpatialDomains::SegGeomSharedPtr to = sIt->second;

                // Create temporary SegExp
                LocalRegions::SegExpSharedPtr toSeg = MemoryManager<
                    LocalRegions::SegExp>::AllocateSharedPtr(
                        bndCondExpU->GetExp(e)->GetBasis(0)->GetBasisKey(),
                        to);

                const int offset = bndCondExpU->GetPhys_Offset(e);
                const int nq     = toSeg->GetTotPoints();

                Array<OneD, NekDouble> xL(nq), xC(nq), yL(nq), yC(nq), tmp;

                bndCondExpU->GetExp(e)->GetCoords(xC, yC);
                toSeg->GetCoords(xL, yL);

                Vmath::Vsub(nq, xL, 1, xC, 1,
                            tmp = bndCondExpU->UpdatePhys() + offset, 1);
                Vmath::Vsub(nq, yL, 1, yC, 1,
                            tmp = bndCondExpV->UpdatePhys() + offset, 1);
            }

            // bndconstrained?
            bndCondExpU->FwdTrans_BndConstrained(
                bndCondExpU->GetPhys(), bndCondExpU->UpdateCoeffs());
            bndCondExpV->FwdTrans_BndConstrained(
                bndCondExpV->GetPhys(), bndCondExpV->UpdateCoeffs());
        }
        else if (bndGraph->GetMeshDimension() == 2)
        {
            m_f->m_exp.push_back(m_f->AppendExpList(0, "v"));
            m_f->m_exp.push_back(m_f->AppendExpList(0, "w"));

            MultiRegions::ExpListSharedPtr bndCondExpU =
                m_f->m_exp[0]->GetBndCondExpansions()[bndCondId];
            MultiRegions::ExpListSharedPtr bndCondExpV =
                m_f->m_exp[1]->GetBndCondExpansions()[bndCondId];
            MultiRegions::ExpListSharedPtr bndCondExpW =
                m_f->m_exp[2]->GetBndCondExpansions()[bndCondId];

            map<int, int> bndCondIds;
            for (int i = 0; i < bndCondExpU->GetExpSize(); ++i)
            {
                bndCondIds[bndCondExpU->GetExp(i)->GetGeom()->GetGlobalID()]
                    = i;
            }

            TriFaceMap vertexFaceMap;

            if (useVertexIds)
            {
                for (int i = 0; i < bndCondExpU->GetExpSize(); ++i)
                {
                    SpatialDomains::TriGeomSharedPtr from =
                        boost::dynamic_pointer_cast<SpatialDomains::TriGeom>(
                                                                             bndCondExpU->GetExp(i)->GetGeom());

                    TriFaceIDs t(from->GetVid(0), from->GetVid(1),
                                 from->GetVid(2));
                    vertexFaceMap[t] = i;
                }
            }

            const SpatialDomains::TriGeomMap &tmp =
                bndGraph->GetAllTriGeoms();
            SpatialDomains::TriGeomMap::const_iterator sIt;

            for (sIt = tmp.begin(); sIt != tmp.end(); ++sIt)
            {
                TriFaceMap::iterator tIt;
                int e;

                if (useVertexIds)
                {
                    TriFaceIDs t(sIt->second->GetVid(0),
                                 sIt->second->GetVid(1),
                                 sIt->second->GetVid(2));

                    tIt = vertexFaceMap.find(t);
                    e = tIt == vertexFaceMap.end() ? -1 : tIt->second;
                }
                else
                {
                    map<int, int>::iterator mIt;
                    mIt = bndCondIds.find(sIt->first);
                    e = mIt == bndCondIds.end() ? -1 : mIt->second;
                }

                if (e == -1)
                {
                    cout << "Warning: couldn't find element "
                         << sIt->first << endl;
                    continue;
                }

                SpatialDomains::TriGeomSharedPtr from =
                    boost::dynamic_pointer_cast<SpatialDomains::TriGeom>(
                        bndCondExpU->GetExp(e)->GetGeom());

                SpatialDomains::TriGeomSharedPtr to = sIt->second;

                // Create temporary SegExp
                LocalRegions::TriExpSharedPtr toSeg = MemoryManager<
                    LocalRegions::TriExp>::AllocateSharedPtr(
                        bndCondExpU->GetExp(e)->GetBasis(0)->GetBasisKey(),
                        bndCondExpV->GetExp(e)->GetBasis(1)->GetBasisKey(),
                        to);

                const int offset = bndCondExpU->GetPhys_Offset(e);
                const int nq     = toSeg->GetTotPoints();

                Array<OneD, NekDouble> xL(nq), xC(nq), yL(nq), yC(nq), tmp;
                Array<OneD, NekDouble> zL(nq), zC(nq);

                bndCondExpU->GetExp(e)->GetCoords(xC, yC, zC);
                toSeg->GetCoords(xL, yL, zL);

                Vmath::Vsub(nq, xL, 1, xC, 1,
                            tmp = bndCondExpU->UpdatePhys() + offset, 1);
                Vmath::Vsub(nq, yL, 1, yC, 1,
                            tmp = bndCondExpV->UpdatePhys() + offset, 1);
                Vmath::Vsub(nq, zL, 1, zC, 1,
                            tmp = bndCondExpW->UpdatePhys() + offset, 1);
            }

            // bndconstrained?
            bndCondExpU->FwdTrans_BndConstrained(
                bndCondExpU->GetPhys(), bndCondExpU->UpdateCoeffs());
            bndCondExpV->FwdTrans_BndConstrained(
                bndCondExpV->GetPhys(), bndCondExpV->UpdateCoeffs());
            bndCondExpW->FwdTrans_BndConstrained(
                bndCondExpW->GetPhys(), bndCondExpW->UpdateCoeffs());
        }
    }

Member Data Documentation

ModuleKey Nektar::Utilities::ProcessDisplacement::className

static

Initial value:

=
    GetModuleFactory().RegisterCreatorFunction(
        ModuleKey(eProcessModule, "displacement"), ProcessDisplacement::create,
        "Deform a mesh given an input field defining displacement")

Definition at line 53 of file ProcessDisplacement.h.