graphics/FBD_Objects.py
changeset 1130 f96e0254f0ce
parent 1054 ef514eaacd8c
child 1133 d81d99fd1932
--- a/graphics/FBD_Objects.py	Mon May 13 23:10:10 2013 +0200
+++ b/graphics/FBD_Objects.py	Mon May 13 23:15:28 2013 +0200
@@ -179,15 +179,16 @@
             output.MoveConnected(exclude)
     
     # Returns the block connector that starts with the point given if it exists 
-    def GetConnector(self, position, name = None):
-        # if a name is given
-        if name is not None:
-            # Test each input and output connector
-            #for input in self.Inputs:
-            #    if name == input.GetName():
-            #        return input
+    def GetConnector(self, position, output_name = None, input_name = None):
+        if input_name is not None:
+            # Test each input connector
+            for input in self.Inputs:
+                if input_name == input.GetName():
+                    return input
+        if output_name is not None:
+            # Test each output connector
             for output in self.Outputs:
-                if name == output.GetName():
+                if output_name == output.GetName():
                     return output
         return self.FindNearestConnector(position, self.Inputs + self.Outputs)