root/trunk/RBRapier/RBRapier/Tools/Animation/ProgressBar.py

#!/usr/bin/env python
##~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
##~ License 
##~ 
##- The RuneBlade Foundation library is intended to ease some 
##- aspects of writing intricate Jabber, XML, and User Interface (wxPython, etc.) 
##- applications, while providing the flexibility to modularly change the 
##- architecture. Enjoy.
##~ 
##~ Copyright (C) 2002  TechGame Networks, LLC.
##~ 
##~ This library is free software; you can redistribute it and/or
##~ modify it under the terms of the BSD style License as found in the 
##~ LICENSE file included with this distribution.
##~ 
##~ TechGame Networks, LLC can be reached at:
##~ 3578 E. Hartsel Drive #211
##~ Colorado Springs, Colorado, USA, 80920
##~
##~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

#~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
#~ Imports 
#~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

import Numeric
from OpenGL import GL

#~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
#~ Definitions 
#~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

class ProgressBar(object):
    #~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
    #~ Constants / Variables / Etc. 
    #~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

    NumericType = Numeric.Float16
    StartColor = Numeric.array([.2, .2, .6, 0.8], NumericType)
    EndColor = Numeric.array([.3, .7, .3, 0.8], NumericType)
    OutlineColor = Numeric.array([1., 1., 1., 0.8], NumericType)

    #~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
    #~ Public Methods 
    #~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

    def __init__(self, progress=0., pos0=(-0.9, -0.05), pos1=(0.9, 0.05), scaledim=(0.99, 0.8)):
        self.pos0 = Numeric.asarray(pos0, self.NumericType)
        self.pos1 = Numeric.asarray(pos1, self.NumericType)
        self.scaledim = Numeric.asarray(scaledim, self.NumericType)
        self.SetProgress(progress)

    def GetProgress(self, progress):
        return self.progress
    def SetProgress(self, progress):
        self.progress = progress

    def GLExecute(self, context):
        dim = self.pos1-self.pos0
        x0, y0 = 0.5*(self.pos1+self.pos0-self.scaledim*dim)
        dx, dy = self.scaledim*dim
        dx *= self.progress
        self._DrawProgressBar((x0,y0), (x0+dx,y0+dy))
        self._DrawOutline(self.pos0, self.pos1)

    def _DrawProgressBar(self, (x0, y0), (x1, y1)):
        color = (1.-self.progress)*self.StartColor + self.progress*self.EndColor
        GL.glColor4fv(color)
        GL.glRectf(x0, y0, x1, y1)

    def _DrawOutline(self, (x0, y0), (x1, y1)):
        GL.glColor4fv(self.OutlineColor)
        GL.glBegin(GL.GL_LINE_LOOP)
        GL.glVertex2f(x0,y0)
        GL.glVertex2f(x0,y1)
        GL.glVertex2f(x1,y1)
        GL.glVertex2f(x1,y0)
        GL.glEnd()

class CenteredProgressBar(ProgressBar):
    def GLExecute(self, context):
        dim = self.pos1-self.pos0
        cx, cy = 0.5*(self.pos1+self.pos0)
        dx, dy = 0.5*self.scaledim*dim
        dx *= self.progress
        self._DrawProgressBar((cx-dx,cy-dy),(cx+dx,cy+dy))
        self._DrawOutline(self.pos0, self.pos1)