Nektar::FieldUtils::ProcessDisplacement Class Reference

#include <ProcessDisplacement.h>

Inheritance diagram for Nektar::FieldUtils::ProcessDisplacement:

Collaboration diagram for Nektar::FieldUtils::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::FieldUtils::ProcessModule
	ProcessModule ()
	ProcessModule (FieldSharedPtr p_f)
Public Member Functions inherited from Nektar::FieldUtils::Module
FIELD_UTILS_EXPORT	Module (FieldSharedPtr p_f)
FIELD_UTILS_EXPORT void	RegisterConfig (string key, string value)
	Register a configuration option with a module. More...
FIELD_UTILS_EXPORT void	PrintConfig ()
	Print out all configuration options for a module. More...
FIELD_UTILS_EXPORT void	SetDefaults ()
	Sets default configuration options for those which have not been set. More...
FIELD_UTILS_EXPORT bool	GetRequireEquiSpaced (void)
FIELD_UTILS_EXPORT void	SetRequireEquiSpaced (bool pVal)
FIELD_UTILS_EXPORT void	EvaluateTriFieldAtEquiSpacedPts (LocalRegions::ExpansionSharedPtr &exp, const Array< OneD, const NekDouble > &infield, Array< OneD, NekDouble > &outfield)

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::FieldUtils::Module
	Module ()
Protected Attributes inherited from Nektar::FieldUtils::Module
FieldSharedPtr	m_f
	Field object. More...
map< string, ConfigOption >	m_config
	List of configuration values. More...
bool	m_requireEquiSpaced

Detailed Description

Definition at line 46 of file ProcessDisplacement.h.

Constructor & Destructor Documentation

Nektar::FieldUtils::ProcessDisplacement::ProcessDisplacement

(

FieldSharedPtr

f

)

Definition at line 108 of file ProcessDisplacement.cpp.

References Nektar::FieldUtils::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::FieldUtils::ProcessDisplacement::~ProcessDisplacement ( )

virtual

Definition at line 121 of file ProcessDisplacement.cpp.

{
}

Member Function Documentation

static boost::shared_ptr<Module> Nektar::FieldUtils::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::FieldUtils::ProcessDisplacement::GetModuleName ( )

inlinevirtual

Implements Nektar::FieldUtils::Module.

Definition at line 62 of file ProcessDisplacement.h.

    {
        return "ProcessDisplacement";
    }

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

virtual

Write mesh to output file.

Implements Nektar::FieldUtils::Module.

Definition at line 125 of file ProcessDisplacement.cpp.

References Nektar::MemoryManager< DataType >::AllocateSharedPtr(), Nektar::LibUtilities::SessionReader::CreateInstance(), Nektar::iterator, Nektar::FieldUtils::Module::m_config, Nektar::FieldUtils::Module::m_f, Nektar::SpatialDomains::MeshGraph::Read(), and Vmath::Vsub().

{
    if (m_f->m_verbose)
    {
        if (m_f->m_comm->TreatAsRankZero())
        {
            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::FieldUtils::ProcessDisplacement::className

static

Initial value:

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

Definition at line 54 of file ProcessDisplacement.h.