LightShaft.h

#pragma once
#include "Scene.h"
#include "LightShaftGeometry.h"

namespace osgCloudyDay
{
        class ViewerLightCallback2 : public osg::NodeCallback
        {
        public:
                ViewerLightCallback2(void)
                {

                }

                virtual void operator()(osg::Node* node, osg::NodeVisitor* nv)
                {
                        osgUtil::CullVisitor* cv = dynamic_cast<osgUtil::CullVisitor*>(nv);        
                        osg::ref_ptr<osg::Geode> geometry = dynamic_cast<osg::Geode*> (node);

                        if(geometry && cv)
                        {
                                osg::Matrixd view(cv->getCurrentCamera()->getViewMatrix());
                                osg::Matrixd invViewMatrix = osg::Matrixd::inverse(view);
                                osg::Matrixd model(*cv->getModelViewMatrix());
                                model = model*invViewMatrix;
                                osg::Matrixd proj(cv->getCurrentCamera()->getProjectionMatrix());
                
                                geometry->getOrCreateStateSet()->getUniform("ModelMatrix")->set(model);
                                geometry->getOrCreateStateSet()->getUniform("ViewMatrix")->set(Scene::GetViewMatrix_View());
                                geometry->getOrCreateStateSet()->getUniform("ProjectionMatrix")->set(Scene::GetProjectionMatrix_View());                                        
                        
                                geometry->getOrCreateStateSet()->addUniform(new osg::Uniform("v3LightPos", Scene::m_skydome->GetLightPosition()));
                                Scene::m_skydome->SetupUniform(geometry.get()->getOrCreateStateSet());
                                geometry->getOrCreateStateSet()->addUniform(new osg::Uniform("color_tex", 0));  
                        
                                std::cout << std::endl << std::endl << "ath callback" << std::endl << std::endl ;

        #ifdef SHADOW_MAPPING
                                geometry->getOrCreateStateSet()->getUniform("light_proj_matrix")->set(Scene::GetProjectionMatrix_Light());
                                geometry->getOrCreateStateSet()->getUniform("light_mv_matrix")->set(Scene::GetLightCamera()->getViewMatrix());
        #endif
                        }
                        traverse(node, nv);
                }
        };

        class LightShaft
        {
        public:
                LightShaft(void);
                ~LightShaft(void);

                void Initialize();
                osg::ref_ptr<osg::Camera> GetCamera();

                osg::ref_ptr<osg::Texture2D> GetLightshaft1();
                osg::ref_ptr<osg::Texture2D> GetLightshaft2();
                
        protected:
                osg::ref_ptr<osg::Texture2D> fbo_lightshaft1;
                osg::ref_ptr<osg::Texture2D> fbo_lightshaft2;

                osg::ref_ptr<osg::Texture2D> fbo_lightshaft_d;  
                osg::Camera* camera_lightshaft;
        };
}