Allow selecting node via ns3.Node instance
authorGustavo J. A. M. Carneiro <gjc@inescporto.pt>
Thu, 29 Jan 2009 16:28:59 +0000
changeset 3910 c13700aa65e0
parent 3909 c60167cd0242
child 3911 e070ed523ee9
Allow selecting node via ns3.Node instance
src/contrib/visualizer/__init__.py
src/contrib/visualizer/core.py
--- a/src/contrib/visualizer/__init__.py	Thu Jan 29 16:04:14 2009 +0000
+++ b/src/contrib/visualizer/__init__.py	Thu Jan 29 16:28:59 2009 +0000
@@ -1,3 +1,3 @@
 
-from core import start, register_plugin, set_bounds
+from core import start, register_plugin, set_bounds, add_initialization_hook
 
--- a/src/contrib/visualizer/core.py	Thu Jan 29 16:04:14 2009 +0000
+++ b/src/contrib/visualizer/core.py	Thu Jan 29 16:28:59 2009 +0000
@@ -1149,6 +1149,9 @@
     
 
     def select_node(self, node):
+        if isinstance(node, ns3.Node):
+            node = self.nodes[node.GetId()]
+
         if node is self.selected_node:
             return