deluge/plugins/NetworkGraph/plugin.py

# netgraph plugin

class plugin_NetGraph:
    def __init__(self, path, deluge_core, deluge_interface):
        import gtk
        self.parent   = deluge_interface
        self.location = path
        self.core = deluge_core

        self.image  = gtk.Image()

        self.viewPort = gtk.Viewport()
        self.viewPort.add(self.image)

        self.scrolledWindow = gtk.ScrolledWindow()
        self.scrolledWindow.add(self.viewPort)
        self.scrolledWindow.set_policy(gtk.POLICY_AUTOMATIC, gtk.POLICY_AUTOMATIC)

        self.topWidget = self.scrolledWindow

        self.parentNotebook = self.parent.notebook
#        print "Parent NOTEBOOK:", self.parentNotebook
        self.parentNotebook.append_page(self.topWidget, gtk.Label("Graph"))
#        print "My INDEX in parentNoteBook:", self.index

        self.image.show()
        self.viewPort.show()
        self.scrolledWindow.show()

        self.length = 60

        self.width  = -1
        self.height = -1

        import pango

        self.pangoContext = self.parent.window.get_pango_context()
        self.pangoLayout = pango.Layout(self.pangoContext)

        self.savedUpSpeeds   = []
        self.savedDownSpeeds = []

        self.bootupRuns = 3  # This ensures that we pass the resizing phase, with scrollbars, etc.
                                    # So the first time it is viewed, we are all ready

    def unload(self): # Shutdown is called when the plugin is deactivated
        numPages = self.parentNotebook.get_n_pages()
        for page in range(numPages):
            if self.parentNotebook.get_nth_page(page) == self.topWidget:
                self.parentNotebook.remove_page(page)
                break

    def update(self):
        import gtk
        session_info = self.core.get_state()
        self.savedUpSpeeds.insert(0, session_info['upload_rate'])
        if len(self.savedUpSpeeds) > self.length:
            self.savedUpSpeeds.pop()
        self.savedDownSpeeds.insert(0, session_info['download_rate'])
        if len(self.savedDownSpeeds) > self.length:
            self.savedDownSpeeds.pop()

        if not self.parentNotebook.get_nth_page(self.parentNotebook.get_current_page()) == \
                 self.topWidget and not self.bootupRuns > 0:
            return

        self.bootupRuns = max(self.bootupRuns - 1, 0)

        extraWidth  = self.scrolledWindow.get_vscrollbar().get_allocation().width  * 1.5
        extraHeight = self.scrolledWindow.get_hscrollbar().get_allocation().height * 1.5
        allocation = self.scrolledWindow.get_allocation()
        allocation.width  = int(allocation.width)  - extraWidth
        allocation.height = int(allocation.height) - extraHeight

        # Don't try to allocate a size too small, or you might crash
        if allocation.width < 2 or allocation.height < 2:
            return

#        savedDownSpeeds = [1,2,3,2,1]
#        savedUpSpeeds = [5,8,0,0,1,2]

#        allocation = self.image.get_allocation()
#        allocation.width  = 300
#        allocation.height = 200

        if not allocation.width == self.width or not allocation.height == self.height:
#            print "New Pixmap!"
            self.width  = allocation.width
            self.height = allocation.height

            self.networkPixmap = gtk.gdk.Pixmap(None, self.width, self.height, 24)
            self.image.set_from_pixmap(self.networkPixmap, None)
            self.ctx = self.networkPixmap.cairo_create()

        self.networkPixmap.draw_rectangle(self.image.get_style().white_gc,True, 0, 0, self.width, self.height)

        maxSpeed = max(max(self.savedDownSpeeds),max(self.savedUpSpeeds))

        if maxSpeed == 0:
            return

        maxSpeed = maxSpeed*1.1 # Give some extra room on top

        self.drawSpeedPoly(self.savedDownSpeeds, (0.5,1,   0.5, 1.0),    maxSpeed, True)
        self.drawSpeedPoly(self.savedDownSpeeds, (0,  0.75,0,   1.0),    maxSpeed, False)

        self.drawSpeedPoly(self.savedUpSpeeds,   (0.33,0.33,1.0,  0.5),  maxSpeed, True)
        self.drawSpeedPoly(self.savedUpSpeeds,   (0,   0,   1.0,  0.75), maxSpeed, False)

        meanUpSpeed   = sum(self.savedUpSpeeds)  /len(self.savedUpSpeeds)
        meanDownSpeed = sum(self.savedDownSpeeds)/len(self.savedDownSpeeds)
        shownSpeed    = max(meanUpSpeed, meanDownSpeed)
        
        import deluge.common

        self.pangoLayout.set_text(deluge.common.fspeed(shownSpeed))
        self.networkPixmap.draw_layout(self.image.get_style().black_gc,
                                                 4,
                                                 int(self.height - 1 - (self.height*shownSpeed/maxSpeed)),
                                                 self.pangoLayout)

        self.networkPixmap.draw_line(self.image.get_style().black_gc,
                                              0,              int(self.height - (self.height*shownSpeed/maxSpeed)),
                                              self.width, int(self.height - (self.height*shownSpeed/maxSpeed)))

        self.networkPixmap.draw_rectangle(self.image.get_style().black_gc,False, 0, 0, self.width-1, self.height-1)

        self.image.queue_draw()

    def tracePath(self, speeds, maxSpeed):
        lineWidth = 4

        self.ctx.set_line_width(lineWidth)

        self.ctx.move_to(self.width + lineWidth,self.height + lineWidth)
        self.ctx.line_to(self.width + lineWidth,int(self.height-(self.height*speeds[0]/maxSpeed)))

        for i in range(len(speeds)):
            self.ctx.line_to(int(self.width-1-((i*self.width)/(self.length-1))),
                                    int(self.height-1-(self.height*speeds[i]/maxSpeed)))

        self.ctx.line_to(int(self.width-1-(((len(speeds)-1)*self.width)/(self.length-1))),
                                int(self.height)-1 + lineWidth)

        self.ctx.close_path()

    def drawSpeedPoly(self, speeds, color, maxSpeed, fill):

        self.tracePath(speeds, maxSpeed)
        self.ctx.set_source_rgba(color[0],color[1],color[2], color[3])

        if fill:
            self.ctx.fill()
        else:
            self.ctx.stroke()