--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/src/contrib/visualizer/plugins/olsr.py	Sat Nov 08 10:29:18 2008 +0000
@@ -0,0 +1,102 @@
+import gtk
+import ns3
+from visualizer.base import InformationWindow
+
+class ShowOlsrRoutingTable(InformationWindow):
+    (
+        COLUMN_DESTINATION,
+        COLUMN_NEXT_HOP,
+        COLUMN_INTERFACE,
+        COLUMN_NUM_HOPS,
+        ) = range(4)
+
+    def __init__(self, visualizer, node_index):
+        InformationWindow.__init__(self)
+        self.win = gtk.Dialog(parent=visualizer.window,
+                              flags=gtk.DIALOG_DESTROY_WITH_PARENT|gtk.DIALOG_NO_SEPARATOR,
+                              buttons=(gtk.STOCK_CLOSE, gtk.RESPONSE_CLOSE))
+        self.win.set_default_size(gtk.gdk.screen_width()/2, gtk.gdk.screen_height()/2)
+        self.win.connect("response", self._response_cb)
+        self.win.set_title("OLSR routing table for node %i" % node_index) 
+        self.visualizer = visualizer
+        self.node_index = node_index
+
+        self.table_model = gtk.ListStore(str, str, str, int)
+
+        treeview = gtk.TreeView(self.table_model)
+        treeview.show()
+        sw = gtk.ScrolledWindow()
+        sw.set_properties(hscrollbar_policy=gtk.POLICY_AUTOMATIC,
+                          vscrollbar_policy=gtk.POLICY_AUTOMATIC)
+        sw.show()
+        sw.add(treeview)
+        self.win.vbox.add(sw)
+        
+        # Dest.
+        column = gtk.TreeViewColumn('Destination', gtk.CellRendererText(),
+                                    text=self.COLUMN_DESTINATION)
+        treeview.append_column(column)
+
+        # Next hop
+        column = gtk.TreeViewColumn('Next hop', gtk.CellRendererText(),
+                                    text=self.COLUMN_NEXT_HOP)
+        treeview.append_column(column)
+
+        # Interface
+        column = gtk.TreeViewColumn('Interface', gtk.CellRendererText(),
+                                    text=self.COLUMN_INTERFACE)
+        treeview.append_column(column)
+
+        # Num. Hops
+        column = gtk.TreeViewColumn('Num. Hops', gtk.CellRendererText(),
+                                    text=self.COLUMN_NUM_HOPS)
+        treeview.append_column(column)
+
+        self.visualizer.add_information_window(self)
+        self.win.show()
+
+    def _response_cb(self, win, response):
+        self.win.destroy()
+        self.visualizer.remove_information_window(self)
+    
+    def update(self):
+        node = ns3.NodeList.GetNode(self.node_index)
+        olsr = node.GetObject(ns3.olsr.Agent.GetTypeId())
+        ipv4 = node.GetObject(ns3.Ipv4.GetTypeId())
+        if olsr is None:
+            return
+        self.table_model.clear()
+        for route in olsr.GetRoutingTable().GetEntries():
+            tree_iter = self.table_model.append()
+            netdevice = ipv4.GetNetDevice(route.interface)
+            if netdevice is None:
+                interface_name = 'lo'
+            else:
+                interface_name = netdevice.GetName()
+                if not interface_name:
+                    interface_name = "(interface %i)" % route.interface
+            self.table_model.set(tree_iter,
+                                 self.COLUMN_DESTINATION, str(route.destAddr),
+                                 self.COLUMN_NEXT_HOP, str(route.nextAddr),
+                                 self.COLUMN_INTERFACE, interface_name,
+                                 self.COLUMN_NUM_HOPS, route.distance)
+
+
+def populate_node_menu(viz, node, menu):
+    ns3_node = ns3.NodeList.GetNode(node.node_index)
+    olsr = ns3_node.GetObject(ns3.olsr.Agent.GetTypeId())
+    if olsr is None:
+        print "No OLSR"
+        return
+    
+    menu_item = gtk.MenuItem("Show OLSR Routing Table")
+    menu_item.show()
+
+    def _show_ipv4_routing_table(dummy_menu_item):
+        ShowOlsrRoutingTable(viz, node.node_index)
+
+    menu_item.connect("activate", _show_ipv4_routing_table)
+    menu.add(menu_item)
+
+def register(viz):
+    viz.connect("populate-node-menu", populate_node_menu)