ChangeColorVisitor.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 CHANGECOLORVISITOR_H_
#define CHANGECOLORVISITOR_H_

#include <osg/NodeVisitor>
#include <osg/StateSet>
#include <osg/Vec3>
#include <osg/Vec4>

class osg::Geode;


class ChangeColorVisitor : public osg::NodeVisitor
{
public:
        ChangeColorVisitor( osg::Vec3 color=osg::Vec3(1.0,1.0,1.0) );
        
        void reset();

        void changeOnlyAlpha( bool b );
        
        void setColor( osg::Vec4 color );
        
        void setColor( osg::Vec3 color );

        void setAlpha( float alpha );
        
        virtual void apply( osg::Geode& geode );
        
        int getNumberOfChangedDrawables() const;
        
        void setChangeBlendState( bool b ) { m_dontChangeBlendState = !b; }
        
protected:
        void ChangeColorVisitor::setBlendState( osg::StateSet* stateSet );

        osg::Vec4       m_color;
        int             m_numberOfChangedDrawables;
        bool            m_AlphaOnly;

        bool m_dontChangeBlendState;
};

#endif /*CHANGECOLORVISITOR_H_*/