python hier/top block holds refs of connected blocks

1 files changed, 2 insertions, 9 deletions

diff --git a/lib/hier_block.cpp b/lib/hier_block.cpp
index 36e1b9d..6c95e9b 100644
--- a/lib/hier_block.cpp
+++ b/lib/hier_block.cpp
@@ -45,13 +45,6 @@ void HierBlock::disconnect(const Element &elem)
     (*this)->topology->remove_topology(elem->topology.get());
 }
 
-static Apology::Wax get_elem_ref(const Element &elem)
-{
-    boost::shared_ptr<Element> shared_elem = elem.weak_self.lock();
-    if (shared_elem) return shared_elem;
-    return elem;
-}
-
 void HierBlock::connect(
     const Element &src,
     const size_t src_index,
@@ -60,8 +53,8 @@ void HierBlock::connect(
 ){
     //TODO, this is the perfect place to validate IO sigs
     const Apology::Flow flow(
-        Apology::Port(src->get_elem(), src_index, get_elem_ref(src)),
-        Apology::Port(sink->get_elem(), sink_index, get_elem_ref(sink))
+        Apology::Port(src->get_elem(), src_index, src.weak_self.lock()),
+        Apology::Port(sink->get_elem(), sink_index, sink.weak_self.lock())
     );
     (*this)->topology->add_flow(flow);
 }