root/trunk/RBRapier/RBRapier/Renderer/StateMgr.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.Aspects import Aspect
from RBRapier.Renderer.AttributeMgr import AttributeChangeElement, ClientAttributeChangeElement

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

class StateManagerBase(object):
    #~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
    #~ Definitions 
    #~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

    def __init__(self, statevector=None):
        if statevector is not None:
            self._statevector = statevector
        else: self._statevector = {}

    def _UpdateState(self, state, enabled):
        self._statevector[state] = enabled

    def GetState(self, state):
        return self._statevector.get(state, None)
    def SetState(self, state, enabled):
        current = self._statevector.get(state, None)
        if current != enabled:
            self._SetState(state, enabled)
            self._UpdateState(state, enabled)
    def DelState(self, state):
        try: del self._statevector[state]
        except KeyError: pass

    def Reset(self, state=None):
        if state is None:
            self._statevector.clear()
        else: self.DelState(state)

    def IsEnabled(self, state): 
        return self.GetState(state) == 1
    def IsNotEnabled(self, state): 
        return self.GetState(state) != 1
    def Enable(self, state): 
        self.SetState(state, 1)

    def IsDisabled(self, state): 
        return self.GetState(state) == 0
    def IsNotDisabled(self, state): 
        return self.GetState(state) != 0
    def Disable(self, state): 
        self.SetState(state, 0)

    def IsKnown(self, state): 
        return self.GetState(state) != None
    def IsUnknown(self, state): 
        return self.GetState(state) == None

#~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

class StateManagerStatsAspect(Aspect):
    #~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
    #~ Constants / Variables / Etc. 
    #~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

    _statechanges = None

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

    def _UpdateState(self, state, enabled):
        sc = self._statechanges
        if sc is None: self._statechanges = {state: 1}
        else: sc[state] = sc.get(state, 0) + 1

        self._statevector[state] = enabled

#~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

class StateManager(StateManagerBase):
    AttributeChange = AttributeChangeElement(GL.GL_ENABLE_BIT)

    def _SetState(self, state, enabled):
        if enabled: GL.glEnable(state)
        else: GL.glDisable(state)

#~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

class ClientStateManager(StateManagerBase):
    ClientAttributeChange = ClientAttributeChangeElement(GL.GL_CLIENT_VERTEX_ARRAY_BIT)

    def _SetState(self, state, enabled):
        if enabled: GL.glEnableClientState(state)
        else: GL.glDisableClientState(state)