Source code for direct.particles.SpriteParticleRendererExt

from acron.core import ConfigVariableString
from acron.physics import SpriteParticleRenderer




class SpriteParticleRendererExt(SpriteParticleRenderer):
    """
    Contains methods to extend functionality
    of the SpriteParticleRenderer class
    """

    # Initialize class variables for texture, source file and node for texture and
    # node path textures to None.  These will be initialized to a hardcoded default
    # or whatever the user specifies in his/her Configrc variable the first time they
    # are accessed
    # Will use instance copy of this in functions below
    sourceTextureName = None
    sourceFileName = None
    sourceNodeName = None

    def getSourceTextureName(self):
        if self.sourceTextureName is None:
            SpriteParticleRendererExt.sourceTextureName = ConfigVariableString(
                'particle-sprite-texture', 'maps/lightbulb.rgb').value
        # Return instance copy of class variable
        return self.sourceTextureName

    def setSourceTextureName(self, name):
        # Set instance copy of class variable
        self.sourceTextureName = name

    def setTextureFromFile(self, fileName = None):
        if fileName is None:
            fileName = self.getSourceTextureName()

        t = base.loader.loadTexture(fileName)
        if t is not None:
            self.setTexture(t, t.getYSize())
            self.setSourceTextureName(fileName)
            return True
        else:
            print("Couldn't find rendererSpriteTexture file: %s" % fileName)
            return False

    def addTextureFromFile(self, fileName = None):
        if self.getNumAnims() == 0:
            return self.setTextureFromFile(fileName)

        if fileName is None:
            fileName = self.getSourceTextureName()

        t = base.loader.loadTexture(fileName)
        if t is not None:
            self.addTexture(t, t.getYSize())
            return True
        else:
            print("Couldn't find rendererSpriteTexture file: %s" % fileName)
            return False

    def getSourceFileName(self):
        if self.sourceFileName is None:
            SpriteParticleRendererExt.sourceFileName = ConfigVariableString(
                'particle-sprite-model', 'models/misc/smiley').value
        # Return instance copy of class variable
        return self.sourceFileName

    def setSourceFileName(self, name):
        # Set instance copy of class variable
        self.sourceFileName = name

    def getSourceNodeName(self):
        if self.sourceNodeName is None:
            SpriteParticleRendererExt.sourceNodeName = ConfigVariableString(
                'particle-sprite-node', '**/*').value
        # Return instance copy of class variable
        return self.sourceNodeName

    def setSourceNodeName(self, name):
        # Set instance copy of class variable
        self.sourceNodeName = name

    def setTextureFromNode(self, modelName = None, nodeName = None, sizeFromTexels = False):
        if modelName is None:
            modelName = self.getSourceFileName()
            if nodeName is None:
                nodeName = self.getSourceNodeName()

        # Load model and get texture
        m = base.loader.loadModel(modelName)
        if m is None:
            print("SpriteParticleRendererExt: Couldn't find model: %s!" % modelName)
            return False

        np = m.find(nodeName)
        if np.isEmpty():
            print("SpriteParticleRendererExt: Couldn't find node: %s!" % nodeName)
            m.removeNode()
            return False

        self.setFromNode(np, modelName, nodeName, sizeFromTexels)
        self.setSourceFileName(modelName)
        self.setSourceNodeName(nodeName)
        m.removeNode()
        return True

    def addTextureFromNode(self, modelName = None, nodeName = None, sizeFromTexels = False):
        if self.getNumAnims() == 0:
            return self.setTextureFromNode(modelName, nodeName, sizeFromTexels)

        if modelName is None:
            modelName = self.getSourceFileName()
            if nodeName is None:
                nodeName = self.getSourceNodeName()

        # Load model and get texture
        m = base.loader.loadModel(modelName)
        if m is None:
            print("SpriteParticleRendererExt: Couldn't find model: %s!" % modelName)
            return False

        np = m.find(nodeName)
        if np.isEmpty():
            print("SpriteParticleRendererExt: Couldn't find node: %s!" % nodeName)
            m.removeNode()
            return False

        self.addFromNode(np, modelName, nodeName, sizeFromTexels)
        m.removeNode()

        return True