#[MIT license:]
#
# Copyright (c) 2004  Dave Pape
#
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documentation files (the "Software"), to deal
# in the Software without restriction, including without limitation the rights
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
# copies of the Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included in
# all copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL THE
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
# SOFTWARE.


class Node:
    def __init__(self, name=''):
        self.name = name
        self.children = []
        
    def attach(self,other):
        self.children.append(other)

    def draw(self):
        pass

    def predraw(self):
        pass

    def postdraw(self):
        pass

    def drawAll(self):
        self.predraw()
        self.draw()
        for child in self.children:
            child.drawAll()
        self.postdraw()

    def traverse(self, func):
        if hasattr(self, 'pre' + func):
            getattr(self,'pre' + func)()
        if hasattr(self, func):
            getattr(self,func)()
        for child in self.children:
            child.traverse(func)
        if hasattr(self, 'post' + func):
            getattr(self,'post' + func)()


class TransformNode(Node):
    def __init__(self, transform=None, name=''):
        Node.__init__(self,name)
        self.transform = transform

    def draw(self):
        if self.transform:
            self.transform.pushApply()

    def postdraw(self):
        if self.transform:
            self.transform.pop()



class LightNode(Node):
    def __init__(self, light=None, name=''):
        Node.__init__(self,name)
        self.light = light

    def draw(self):
        if self.light:
            self.light.apply()



import types

class ObjectNode(Node):
    def __init__(self, state=None, name=''):
        Node.__init__(self,name)
        self.state = state

    def predraw(self):
        if self.state:
            if hasattr(self.state, 'apply'):
                self.state.apply()
            elif isinstance(self.state, types.ListType):
                for s in self.state:
                    s.apply()

    def postdraw(self):
        if self.state:
            if hasattr(self.state, 'disable'):
                self.state.disable()
            elif isinstance(self.state, types.ListType):
                for s in self.state:
                    s.disable()