generic method to get successors

dream/plugins/plugin.py

@@ -42,7 +42,7 @@ class InputPreparationPlugin(Plugin):
     return data
 
   # returns the predecessors of a node
-  def findPredecessors(self, data, node_id):
+  def getPredecessors(self, data, node_id):
     predecessors=[]
     from copy import copy    
     edges=copy(data['graph']['edge'])
@@ -51,6 +51,16 @@ class InputPreparationPlugin(Plugin):
             predecessors.append(edge['source'])
     return predecessors
 
+  # returns the successors of a node
+  def getSuccessors(self, data, node_id):
+    successors=[]
+    from copy import copy    
+    edges=copy(data['graph']['edge'])
+    for edge_id,edge in edges.iteritems():
+        if edge['source']==node_id:
+            successors.append(edge['destination'])
+    return successors
+
 class OutputPreparationPlugin(Plugin):
   def postprocess(self, data):
     """Postprocess the data after simulation run.