Newer
Older
Martin Christoph Hierholzer
committed
/*
* EntityOwner.cc
*
* Created on: Nov 15, 2016
* Author: Martin Hierholzer
*/
#include <cassert>
#include <iostream>
#include "EntityOwner.h"
#include "Module.h"
Martin Christoph Hierholzer
committed
#include "VirtualModule.h"
Martin Christoph Hierholzer
committed
#include "VariableGroup.h"
Martin Christoph Hierholzer
committed
namespace ChimeraTK {
EntityOwner::EntityOwner(EntityOwner *owner, const std::string &name, const std::string &description,
bool eliminateHierarchy, const std::unordered_set<std::string> &tags)
: _name(name), _description(description), _owner(owner), _eliminateHierarchy(eliminateHierarchy), _tags(tags)
Martin Christoph Hierholzer
committed
{
if(owner != nullptr) {
auto thisMustBeAModule = static_cast<Module*>(this); /// @todo TODO FIXME this is a bit dangerous...
owner->registerModule(thisMustBeAModule);
}
}
Martin Christoph Hierholzer
committed
/*********************************************************************************************************************/
EntityOwner::~EntityOwner() {
if(_owner != nullptr) {
auto thisMustBeAModule = static_cast<Module*>(this); /// @todo TODO FIXME this is a bit dangerous...
_owner->unregisterModule(thisMustBeAModule);
}
}
Martin Christoph Hierholzer
committed
/*********************************************************************************************************************/
Martin Christoph Hierholzer
committed
void EntityOwner::registerModule(Module *module) {
for(auto &tag : _tags) module->addTag(tag);
Martin Christoph Hierholzer
committed
moduleList.push_back(module);
}
Martin Christoph Hierholzer
committed
/*********************************************************************************************************************/
void EntityOwner::unregisterModule(Module *module) {
moduleList.remove(module);
}
Martin Christoph Hierholzer
committed
/*********************************************************************************************************************/
Martin Christoph Hierholzer
committed
std::list<VariableNetworkNode> EntityOwner::getAccessorListRecursive(bool includeSubmodules) {
Martin Christoph Hierholzer
committed
// add accessors of this instance itself
Martin Christoph Hierholzer
committed
std::list<VariableNetworkNode> list = getAccessorList();
Martin Christoph Hierholzer
committed
// iterate through submodules
for(auto submodule : getSubmoduleList()) {
Martin Christoph Hierholzer
committed
// ignore anything that is not a VariableGroup if submodules should not be included
/// @todo Add test for this!
if(!includeSubmodules && dynamic_cast<VariableGroup*>(submodule) == nullptr) continue;
// obtail list of accessors from the submodule/group and insert it into the list
auto sublist = submodule->getAccessorListRecursive(includeSubmodules);
Martin Christoph Hierholzer
committed
list.insert(list.end(), sublist.begin(), sublist.end());
}
return list;
}
Martin Christoph Hierholzer
committed
/*********************************************************************************************************************/
std::list<Module*> EntityOwner::getSubmoduleListRecursive() {
// add modules of this instance itself
std::list<Module*> list = getSubmoduleList();
// iterate through submodules
for(auto submodule : getSubmoduleList()) {
auto sublist = submodule->getSubmoduleListRecursive();
list.insert(list.end(), sublist.begin(), sublist.end());
}
return list;
}
Martin Christoph Hierholzer
committed
/*********************************************************************************************************************/
VirtualModule EntityOwner::findTag(const std::string &tag, bool eliminateAllHierarchies) const {
// create new module to return
Martin Christoph Hierholzer
committed
VirtualModule module{_name, _description};
Martin Christoph Hierholzer
committed
// add everything matching the tag to the virtual module and return it
findTagAndAppendToModule(module, tag, eliminateAllHierarchies, true);
return module;
}
/*********************************************************************************************************************/
void EntityOwner::findTagAndAppendToModule(VirtualModule &module, const std::string &tag, bool eliminateAllHierarchies,
bool eliminateFirstHierarchy) const {
Martin Christoph Hierholzer
committed
VirtualModule nextmodule{_name, _description};
Martin Christoph Hierholzer
committed
VirtualModule *moduleToAddTo;
bool needToAddSubModule = false;
Martin Christoph Hierholzer
committed
if(!getEliminateHierarchy() && !eliminateAllHierarchies && !eliminateFirstHierarchy) {
moduleToAddTo = &nextmodule;
needToAddSubModule = true;
Martin Christoph Hierholzer
committed
}
else {
moduleToAddTo = &module;
}
// add nodes to the module if matching the tag
for(auto node : getAccessorList()) {
if(node.getTags().count(tag) > 0) {
moduleToAddTo->registerAccessor(node);
}
}
// iterate through submodules
for(auto submodule : getSubmoduleList()) {
submodule->findTagAndAppendToModule(*moduleToAddTo, tag, eliminateAllHierarchies);
}
if(needToAddSubModule) {
if( nextmodule.getAccessorList().size() > 0 || nextmodule.getSubmoduleList().size() > 0 ) {
module.addSubModule(nextmodule);
}
}
Martin Christoph Hierholzer
committed
}
/*********************************************************************************************************************/
void EntityOwner::dump(const std::string &prefix) const {
if(prefix == "") {
std::cout << "==== Hierarchy dump of module '" << _name << "':" << std::endl;
}
for(auto &node : getAccessorList()) {
Martin Christoph Hierholzer
committed
std::cout << prefix << "+ ";
node.dump();
}
for(auto &submodule : getSubmoduleList()) {
Martin Christoph Hierholzer
committed
std::cout << prefix << "| " << submodule->getName() << std::endl;
submodule->dump(prefix+"| ");
}
}
/*********************************************************************************************************************/
void EntityOwner::addTag(const std::string &tag) {
for(auto &node : getAccessorList()) node.addTag(tag);
for(auto &submodule : getSubmoduleList()) submodule->addTag(tag);
_tags.insert(tag);
Martin Christoph Hierholzer
committed
/*********************************************************************************************************************/
VirtualModule EntityOwner::flatten() {
Martin Christoph Hierholzer
committed
VirtualModule nextmodule{_name, _description};
Martin Christoph Hierholzer
committed
for(auto &node : getAccessorListRecursive(true)) {
Martin Christoph Hierholzer
committed
nextmodule.registerAccessor(node);
}
return nextmodule;
}
Martin Christoph Hierholzer
committed
Martin Christoph Hierholzer
committed
} /* namespace ChimeraTK */