reset the parent of tools when invoke AlgBase::setParent()

SNiPER-Framework · Oct 29, 2021 · 58d21f0 · 58d21f0
1 parent 74728d5
commit 58d21f0
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 1 deletion.
diff --git a/SniperKernel/SniperKernel/AlgBase.h b/SniperKernel/SniperKernel/AlgBase.h
@@ -42,6 +42,9 @@ class AlgBase : public DLElement
     template <typename Type>
     Type *tool(const std::string &toolName);
 
+    // set the parent (ExecUnit implementation) pointer
+    virtual void setParent(ExecUnit *parent) override;
+
     //Declared in base class DLElement
     //virtual bool initialize() = 0;
     //virtual bool finalize() = 0;

diff --git a/SniperKernel/SniperKernel/DLElement.h b/SniperKernel/SniperKernel/DLElement.h
@@ -51,7 +51,7 @@ class DLElement : public NamedElement
     void setDescription(const std::string &desc) { m_description = desc; }
 
     // set the parent (ExecUnit implementation) pointer
-    void setParent(ExecUnit *parent);
+    virtual void setParent(ExecUnit *parent);
 
     // get a property via its key-name
     Property *property(const std::string &key);

diff --git a/SniperKernel/src/AlgBase.cc b/SniperKernel/src/AlgBase.cc
@@ -74,6 +74,15 @@ ToolBase *AlgBase::findTool(const std::string &toolName)
     return nullptr;
 }
 
+void AlgBase::setParent(ExecUnit *parent)
+{
+    DLElement::setParent(parent);
+    for (auto &it : m_tools)
+    {
+        it.second->setParentAlg(this);
+    }
+}
+
 SniperJSON AlgBase::json()
 {
     SniperJSON j = DLElement::json();