TypedSwitch.h

Go to the documentation of this file.

/******************************************************************************
 *
 *  Copyright (C) 2008 Boris Iven, Max Hermann
 *
 *  This file is part of the "CEDRIC Event Display" application.
 *
 *****************************************************************************/

#ifndef TYPEDSWITCH_H
#define TYPEDSWITCH_H

#define TYPEDSWITCH_ADDCHILD_MACRO( OSGADDFUNC )                    \
        T* node = dynamic_cast<T*>( child );                            \
        if( node && !childrenAllowed()                                  \
                         && !(node->getClassName().compare(getClassName())==0) )\
                return false;                                               \
    if( !OSGADDFUNC ) return false;                                 \
        if( node ) updateType( node->getType() );                       \
        return true;

#include <osg/Switch>
#include <osg/Node>

#include <string>

class TypedSwitch : public osg::Switch
{
public:
        typedef TypedSwitch T;

        enum ReservedTypes { UNSPECIFIED=-1, GROUP=1 };
        
        TypedSwitch( std::string className ): m_className(className) {}

        virtual bool childrenAllowed() { return true; }
        
        const std::string& getClassName() const { return m_className; }
        int getType() const { return m_type; }


        virtual bool addChild( osg::Node* child )                        { TYPEDSWITCH_ADDCHILD_MACRO( Switch::addChild(child) ) }
        virtual bool insertChild( unsigned int index, osg::Node *child ) { TYPEDSWITCH_ADDCHILD_MACRO( Switch::insertChild(index,child) ) }
        virtual bool setChild( unsigned int index, osg::Node *child )    { TYPEDSWITCH_ADDCHILD_MACRO( Switch::setChild(index,child) ) }
        
        virtual bool addChild( osg::Node* child, bool value )                        { TYPEDSWITCH_ADDCHILD_MACRO( Switch::addChild(child,value) ) }
        virtual bool insertChild( unsigned int index, osg::Node *child, bool value ) { TYPEDSWITCH_ADDCHILD_MACRO( Switch::insertChild(index,child,value) ) }
        
        virtual bool replaceChild( osg::Node *origChild, osg::Node *child )
        {
                if( !Switch::replaceChild( origChild, child ) )
                        return false;
                
                updateType();           
                return true;
        }
        
        virtual bool removeChildren( unsigned int pos, unsigned int numChildrenToRemove )
        {
                if( !( (pos+numChildrenToRemove-1) < _children.size() && numChildrenToRemove > 0 ) )
                        return false;
        
                bool atLeastOneTNodeRemoved = false;
                for( unsigned int i=pos; i < pos+numChildrenToRemove; ++i )
                {
                        T* child = dynamic_cast<T*>( getChild( i ) );
                        if( child )
                        {
                                atLeastOneTNodeRemoved = true;
                                break; // leave for-loop
                        }
                }
                
                if( !Switch::removeChildren( pos, numChildrenToRemove ) )
                        return false;
                
                if( atLeastOneTNodeRemoved ) updateType();              
                return true;
        }
        
protected:
        void updateType( int type=UNSPECIFIED )
        {
                // if type is not specified acquire it from children
                if( type == UNSPECIFIED )
                {
                        int acquired_type = ((getNumChildren()>0) ? GROUP : UNSPECIFIED);
                        for( unsigned int i=0; i < getNumChildren(); ++i )
                        {
                                T* child = dynamic_cast<T*>( getChild(i) );
                                if( child )
                                        acquired_type |= child->getType();
                        }                       
                        m_type = acquired_type;
                }
                else
                        // add to own type
                        m_type |= type;
                
                // tell all parents
                for( unsigned int i=0; i < getNumParents(); ++i )
                {
                        T* parent = dynamic_cast<T*>( getParent(i) );
                        if( parent )
                                parent->updateType( m_type );
                }
        }
        
        int m_type;
        std::string m_className;
};

#endif