root/trunk/RBRapier/RBRapier/Renderer/Appearance/PointRasterization.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 RBFoundation.Objects.Properties import LazyProperty
from RBRapier.Tools import Vector
from RBRapier.Renderer.AttributeMgr import AttributeChangeElement

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

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

    AttributeChange = AttributeChangeElement(GL.GL_POINT_BIT)
    Size = 1.

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

    def __init__(self, Size=1.):
        self.Size = Size

    def GLSelect(self, context):
        GL.glPointSize(self.Size)

    def GLDeselect(self, context):
        GL.glPointSize(1.)

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

    AttributeChange = AttributeChangeElement(GL.GL_POINT_BIT | GL.GL_HINT_BIT)
    SmoothHint = GL.GL_DONT_CARE

    def GLSelect(self, context):
        GL.glHint(GL.GL_POLYGON_SMOOTH_HINT, self.SmoothHint)
        context.StateMgr.Enable(GL.GL_POINT_SMOOTH)

    def GLDeselect(self, context):
        context.StateMgr.Disable(GL.GL_POINT_SMOOTH)

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

    AttributeChange = AttributeChangeElement(GL.GL_POINT_BIT)
    Attenuation = Vector.Vector3Property('Attenuation', (1., 0., 0.))
    MinSize = 0.
    MaxSize = 64.
    FadeThreshold = 1.

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

    class _Extensions(object):
        def __init__(self):
            from OpenGL.GL.EXT import point_parameters
            self.point_parameters = point_parameters.glInitPointParametersEXT() and point_parameters or None
    _extensions = LazyProperty(_Extensions)

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

    def GLExecute(self, context):
        pp = self._extensions.point_parameters
        pp.glPointParameterfvEXT(pp.GL_POINT_DISTANCE_ATTENUATION_EXT, self.Attenuation)
        pp.glPointParameterfEXT(pp.GL_POINT_SIZE_MIN_EXT, self.MinSize)
        pp.glPointParameterfEXT(pp.GL_POINT_SIZE_MAX_EXT, self.MaxSize)
        pp.glPointParameterfEXT(pp.GL_POINT_FADE_THRESHOLD_SIZE_EXT, self.FadeThreshold)