LightShaft2.h

#pragma once
#include "Scene.h"

#include <osg/Geode>
#include <osg/Geometry>
#include <osg/Array>
#include <osg/CullFace>
#include <osg/Depth>

namespace osgCloudyDay
{
        class LightShaft2Callback : public osg::NodeCallback
        {
        public:
                LightShaft2Callback(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 modelview(*cv->getModelViewMatrix());
                                osg::Matrixd model = modelview*invViewMatrix;
                                osg::Matrixd proj(cv->getCurrentCamera()->getProjectionMatrix());
                
                                geometry->getOrCreateStateSet()->getUniform("ModelMatrix")->set(model);
                                geometry->getOrCreateStateSet()->getUniform("ViewMatrix")->set(view);
                                geometry->getOrCreateStateSet()->getUniform("ProjectionMatrix")->set(proj);             

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

        class LightShaft2
        {
        public:
                LightShaft2();
                ~LightShaft2();

                void Initialize();
                osg::Geode* GetGeode(); 
                void SetUniforms(osg::ref_ptr<osg::StateSet> nodess4);

        protected:
                osg::ref_ptr<osg::Geode> geode;
                osg::ref_ptr<osg::Geometry> geometry;
                osg::ref_ptr<osg::Vec3Array> m_vertices;
                osg::ref_ptr<osg::UIntArray> m_indices; 
                osg::ref_ptr<osg::Program> m_forestProg;
        };
}