add disconnect to elements, and disconnect renderer from sources on screen close

[enigma2.git] / lib / python / Components / Element.py

diff --git a/lib/python/Components/Element.py b/lib/python/Components/Element.py
index 6f812b207a6f9a4d7b5e0849c0068dd214ea18f3..fc709139eb8802e6c9bb5b6320e71fca4801fffd 100644 (file)
--- a/lib/python/Components/Element.py
+++ b/lib/python/Components/Element.py
@@ -25,6 +25,24 @@ class Element:
                self.connectUpstream(upstream)
                upstream.connectDownstream(self)
 
+       # we disconnect from down to up
+       def disconnectAll(self):
+               # we should not disconnect from upstream if
+               # there are still elements depending on us.
+               assert len(self.downstream_elements) == 0, "there are still downstream elements left"
+               
+               # disconnect all upstream elements from us
+               for upstream in self.upstream_elements:
+                       upstream.disconnectDownstream(self)
+       
+       def disconnectDownstream(self, downstream):
+               self.downstream_elements.remove(downstream)
+               if self.master == downstream:
+                       self.master = None
+               
+               if len(self.downstream_elements) == 0:
+                       self.disconnectAll()
+
        # default action: push downstream
        def changed(self):
                self.downstream_elements.changed()