portelli/Grid
1
0
Fork 0
mirror of https://github.com/paboyle/Grid.git synced 2024-11-10 07:55:35 +00:00
Code Releases Activity

Hadrons: getModule with upcast

Browse Source
This commit is contained in:
Antonin Portelli 2016-05-12 18:33:36 +01:00
parent 7ae667c767
commit afe5a94745
1 changed files with 25 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
programs/Hadrons/Environment.hpp
View File

@ -78,6 +78,10 @@ public:
XmlReader &reader);
ModuleBase * getModule(const unsigned int address) const;
ModuleBase * getModule(const std::string name) const;
template <typename M>
M * getModule(const unsigned int address) const;
template <typename M>
M * getModule(const std::string name) const;
unsigned int getModuleAddress(const std::string name) const;
std::string getModuleName(const unsigned int address) const;
std::string getModuleType(const unsigned int address) const;
@ -187,6 +191,27 @@ private:
/******************************************************************************
* template implementation *
******************************************************************************/
template <typename M>
M * Environment::getModule(const unsigned int address) const
{
if (auto *pt = dynamic_cast<M *>(getModule(address)))
{
return pt;
}
else
{
HADRON_ERROR("module '" + moduleName_[address] + "' does not have type "
+ typeName<M>() + "(object type: "
+ typeName(*module_.at(address).get()) + ")");
}
}
template <typename M>
M * Environment::getModule(const std::string name) const
{
return getModule<M>(getModuleAddress(name));
}
template <typename T>
unsigned int Environment::lattice4dSize(void) const
{