E:/project/Anoria2/bigworld/anoria/res/scripts/client/Helpers/PyGUI/VisualStateComponent.py

Go to the documentation of this file.

'''
'''

import sys
from bwdebug import ERROR_MSG

import ResMgr


# TODO: put docstrings on these

class VisualState( object ):

        def onSave( self, dataSection ):
                pass

        def onLoad( self, dataSection ):
                pass

        def apply( self, componentScript ):
                pass


        def _readMappingSection( self, dataSection ):
                mappingName = dataSection.asString.strip()
                mappingType = mappingName if mappingName else 'UV'

                mapping = [None, None, None, None]
                mapping[0] = dataSection.readVector2( "coords0" )
                mapping[1] = dataSection.readVector2( "coords1" )
                mapping[2] = dataSection.readVector2( "coords2" )
                mapping[3] = dataSection.readVector2( "coords3" )
                mapping = tuple( mapping )

                return mappingType, mapping


        def _writeMappingSection( self, dataSection, mappingType, mapping ):
                dataSection.asString = mappingType
                dataSection.writeVector2( "coords0", mapping[0] )
                dataSection.writeVector2( "coords1", mapping[1] )
                dataSection.writeVector2( "coords2", mapping[2] )
                dataSection.writeVector2( "coords3", mapping[3] )



class VisualStateComponent( object ):

        def __init__( self, component, visualStateClassName ):
                self.visualStateClassName = visualStateClassName
                self._visualStates = {}


        def onSave( self, dataSection ):
                dataSection.writeString( "visualStates", self.visualStateClassName )

                visualStatesSection = dataSection._visualStates
                for (stateName, state) in self._visualStates.items():
                        stateSection = visualStatesSection.createSection( stateName )
                        state.onSave( stateSection )


        def onLoad( self, dataSection ):
                if dataSection.has_key( "visualStates" ):
                        visualStatesSection = dataSection._visualStates

                        if visualStatesSection.has_key( "external" ):
                                extName = visualStatesSection._external.asString
                                ext = ResMgr.openSection( extName )
                                if ext is not None:
                                        visualStatesSection = ext
                                else:
                                        ERROR_MSG( "Failed to open external visual state '%s'." % extName )

                        self.visualStateClassName = visualStatesSection.asString
                        components = self.visualStateClassName.strip('"').split('.')

                        module = __import__('__main__')
                        for comp in components[:-1]:
                                module = getattr( module, comp )
                        visualStateClass = getattr( module, components[-1] )

                        for (stateName, stateSection) in visualStatesSection.items():
                                visualState = visualStateClass()
                                visualState.onLoad( stateSection )
                                self._visualStates[ stateName ] = visualState


        def setVisualState( self, stateName ):
                state = self._visualStates.get( stateName, None )
                if state:
                        state.apply( self )
                else:
                        ERROR_MSG( "No Visual State '%s' on %s" % (stateName, self) )