root/trunk/RBRapier/RBRapier/Tools/Visualizers/AxisSets.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 
#~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

from OpenGL import GL
from RBRapier.Renderer.AttributeMgr import AttributeChangeElement

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

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

    AttributeChange = AttributeChangeElement(GL.GL_LINE_BIT)
    LineWidth = 4.
    AxisLength = 1.
    xColor = (1., 0., 0., 1.)
    yColor = (0., 1., 0., 1.)
    zColor = (0., 0., 1., 1.)

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

    def __init__(self, AxisLength=None, **kw):
        if AxisLength is not None:
            self.AxisLength = AxisLength 
        for name, value in kw.iteritems():
            setattr(self, name, value)

    def GLExecute(self, context):
        AxisLength = self.AxisLength

        GL.glLineWidth(self.LineWidth)
        GL.glNormal3f(0.577,0.577,0.577)
        GL.glBegin(GL.GL_LINES)

        # X Axis
        GL.glColor4f(*self.xColor)
        GL.glVertex3f(0., 0., 0.)
        GL.glVertex3f(AxisLength, 0., 0.)
        # Y Axis
        GL.glColor4f(*self.yColor)
        GL.glVertex3f(0., 0., 0.)
        GL.glVertex3f(0., AxisLength, 0.)
        # Z Axis
        GL.glColor4f(*self.zColor)
        GL.glVertex3f(0., 0., 0.)
        GL.glVertex3f(0., 0., AxisLength)

        GL.glEnd()
        GL.glLineWidth(1.)

        context.Statistics['lines'] = context.Statistics.get('lines', 0) + 3