def __init__(self):
self.fm = FileManager()
self.tm = TraceManager()
self.fm.start()
self.tm.start()
self.gui = gtk.Builder()
self.gui.add_from_file('ui/traceanalyser.ui')
win_traceanalyser = self.gui.get_object('win_traceanalyser')
win_traceanalyser.show()
self.gui.connect_signals(self)
self.tm.connect('operation-completed', self.on_operations_completed)
self.tvw_nodes = self.gui.get_object('tvw_nodes')
tvc_nodes = gtk.TreeViewColumn('Id')
cell = gtk.CellRendererText()
tvc_nodes.pack_start(cell)
tvc_nodes.set_attributes(cell, text=0)
self.tvw_nodes.append_column(tvc_nodes)
self.node_list = gtk.ListStore(str)
self.tvw_nodes.set_model(self.node_list)
self.tvw_flows = self.gui.get_object('tvw_flows')
tvc_flowid = gtk.TreeViewColumn('FlowId')
tvc_src = gtk.TreeViewColumn('Source')
tvc_dst = gtk.TreeViewColumn('Destination')
tvc_type = gtk.TreeViewColumn('Type')
cel_flowid = gtk.CellRendererText()
cel_src = gtk.CellRendererText()
cel_dst = gtk.CellRendererText()
cel_type = gtk.CellRendererText()
tvc_flowid.pack_start(cel_flowid)
tvc_src.pack_start(cel_src)
tvc_dst.pack_start(cel_dst)
tvc_type.pack_start(cel_type)
tvc_flowid.set_attributes(cel_flowid, text=0)
tvc_src.set_attributes(cel_src, text=1)
tvc_dst.set_attributes(cel_dst, text=2)
tvc_type.set_attributes(cel_type, text=3)
self.tvw_flows.append_column(tvc_flowid)
self.tvw_flows.append_column(tvc_src)
self.tvw_flows.append_column(tvc_dst)
self.tvw_flows.append_column(tvc_type)
self.flow_list = gtk.ListStore(str,str,str,str)
self.tvw_flows.set_model(self.flow_list)
self.flt_trace = gtk.FileFilter()
self.flt_trace.set_name('NS2 New Trace File')
self.flt_trace.add_pattern('*.tr')
self.flt_prj = gtk.FileFilter()
self.flt_prj.set_name('Project Trace Analyser')
self.flt_prj.add_pattern('*.pta')
self.gui.get_object('dlg_newfile').add_filter(self.flt_trace)
self.gui.get_object('dlg_openfile').add_filter(self.flt_prj)
self.pgb_loading = self.gui.get_object('pgb_loading')
self.pgb_loading.set_pulse_step(0.3)
self.timeoutId = 0
class Gui:
def __init__(self):
self.fm = FileManager()
self.tm = TraceManager()
self.fm.start()
self.tm.start()
self.gui = gtk.Builder()
self.gui.add_from_file('ui/traceanalyser.ui')
win_traceanalyser = self.gui.get_object('win_traceanalyser')
win_traceanalyser.show()
self.gui.connect_signals(self)
self.tm.connect('operation-completed', self.on_operations_completed)
self.tvw_nodes = self.gui.get_object('tvw_nodes')
tvc_nodes = gtk.TreeViewColumn('Id')
cell = gtk.CellRendererText()
tvc_nodes.pack_start(cell)
tvc_nodes.set_attributes(cell, text=0)
self.tvw_nodes.append_column(tvc_nodes)
self.node_list = gtk.ListStore(str)
self.tvw_nodes.set_model(self.node_list)
self.tvw_flows = self.gui.get_object('tvw_flows')
tvc_flowid = gtk.TreeViewColumn('FlowId')
tvc_src = gtk.TreeViewColumn('Source')
tvc_dst = gtk.TreeViewColumn('Destination')
tvc_type = gtk.TreeViewColumn('Type')
cel_flowid = gtk.CellRendererText()
cel_src = gtk.CellRendererText()
cel_dst = gtk.CellRendererText()
cel_type = gtk.CellRendererText()
tvc_flowid.pack_start(cel_flowid)
tvc_src.pack_start(cel_src)
tvc_dst.pack_start(cel_dst)
tvc_type.pack_start(cel_type)
tvc_flowid.set_attributes(cel_flowid, text=0)
tvc_src.set_attributes(cel_src, text=1)
tvc_dst.set_attributes(cel_dst, text=2)
tvc_type.set_attributes(cel_type, text=3)
self.tvw_flows.append_column(tvc_flowid)
self.tvw_flows.append_column(tvc_src)
self.tvw_flows.append_column(tvc_dst)
self.tvw_flows.append_column(tvc_type)
self.flow_list = gtk.ListStore(str,str,str,str)
self.tvw_flows.set_model(self.flow_list)
self.flt_trace = gtk.FileFilter()
self.flt_trace.set_name('NS2 New Trace File')
self.flt_trace.add_pattern('*.tr')
self.flt_prj = gtk.FileFilter()
self.flt_prj.set_name('Project Trace Analyser')
self.flt_prj.add_pattern('*.pta')
self.gui.get_object('dlg_newfile').add_filter(self.flt_trace)
self.gui.get_object('dlg_openfile').add_filter(self.flt_prj)
self.pgb_loading = self.gui.get_object('pgb_loading')
self.pgb_loading.set_pulse_step(0.3)
self.timeoutId = 0
def on_mnuitm_new_activate(self, widget):
self.gui.get_object('dlg_newprj').show()
def on_mnuitm_open_activate(self, widget):
self.gui.get_object('dlg_openfile').show()
def on_mnuitm_save_activate(self, widget):
self.gui.get_object('dlg_savefile').show()
def on_btn_newprj_clicked(self, widget):
self.gui.get_object('dlg_newprj').hide()
self.gui.get_object('dlg_newfile').show()
def on_btn_new_open_clicked(self, widget):
self.gui.get_object('dlg_newfile').hide()
filename = self.gui.get_object('dlg_newfile').get_filename()
newprjname = self.gui.get_object('ent_newprjname').get_text() + '.pta'
self.fm.open_tracefile(newprjname, filename)
def on_btn_open_clicked(self, widget):
self.gui.get_object('dlg_openfile').hide()
filename = self.gui.get_object('dlg_openfile').get_filename()
self.tm.open_db(filename)
self.gui.get_object('mnuitm_stats').show()
def on_mnuitm_get_nodes_activate(self, widget):
self.timeoutId = gobject.timeout_add(300,self.on_operations_running)
self.gui.get_object('dlg_loading').show()
self.tm.query_nodes()
self.tm.close_operations(self.display_nodes)
def on_mnuitm_get_flows_activate(self, widget):
self.timeoutId = gobject.timeout_add(300,self.on_operations_running)
self.gui.get_object('dlg_loading').show()
self.tm.query_flows()
self.tm.query_src_dst_per_flow()
self.tm.query_flow_types()
self.tm.close_operations(self.display_flows)
def on_mnuitm_avgthroughput_activate(self, widget):
sel = self.tvw_flows.get_selection()
if sel:
self.timeoutId = gobject.timeout_add(300,self.on_operations_running)
self.gui.get_object('dlg_loading').show()
model, sel_iter = sel.get_selected()
flow_id, ip_src, ip_dst = self.flow_list.get(sel_iter,0,1,2)
flow_id = int(flow_id)
ip_src = int(ip_src)
ip_dst = int(ip_dst)
self.tm.query_sent_pkts_times_at(ip_src, flow_id)
self.tm.query_recv_pkts_times_at(ip_dst, flow_id)
self.tm.query_recv_flow_total_size_at(ip_dst, flow_id, 20)
self.tm.close_operations(self.calc_avgthroughput)
def display_nodes(self):
node_ids = self.tm.get_result()
for node_id in node_ids:
self.node_list.append([node_id])
def display_flows(self):
flow_ids = self.tm.get_result()
flow_ids.sort()
flow_src_dst = self.tm.get_result()
flow_types = self.tm.get_result()
for flow_id in flow_ids:
(src,dst) = flow_src_dst[flow_id]
flow_type = flow_types[flow_id]
row = flow_id,src,dst,flow_type
self.flow_list.append(row)
def calc_avgthroughput(self):
sent_pkts = self.tm.get_result()
recv_pkts = self.tm.get_result()
tota_size = self.tm.get_result()
start_time = Decimal(sent_pkts[0][1])
stop_time = Decimal(recv_pkts[-1][1])
delta = stop_time - start_time
data_size = Decimal(tota_size)
th_bps = data_size / delta
conv = Decimal(8) / Decimal(1000)
avg_tput = th_bps*conv
print 'Avg tput', avg_tput
def on_btn_open_cancel_clicked(self, widget):
self.gui.get_object('dlg_openfile').hide()
def on_btn_save_cancel_clicked(self, widget):
self.gui.get_object('dlg_savefile').hide()
def on_operations_completed(self, tm, operations_handler):
self.gui.get_object('dlg_loading').hide()
gobject.source_remove(self.timeoutId)
operations_handler()
def on_operations_running(self):
self.pgb_loading.pulse()
return True
def on_mnuitm_exit_activate(self, widget):
self.tm.stop()
self.fm.stop()
self.tm.join()
self.fm.join()
gtk.main_quit()
