DtOsgLightingComponent.h

Go to the documentation of this file.

/****************************************************************************** 
** Copyright (c) 2014 MAK Technologies, Inc. 
** All rights reserved. 
******************************************************************************/ 


#pragma once

#include <vrvOsg/DtOsgStateComponent.h>
#include <vrvOsg/DtOsgTextureApplyAttribute.h>
#include <vrvOsg/DtOsgTextureBufferObjectApplyAttribute.h>
#include <osg/Uniform>
#include <vector>

#include "vrvGraphicsUtils/DtLight.h"

namespace osg
{
   class Matrixd;
}

namespace makVrv
{
   class DtLight;
   class DtLightData;
   class DtTileSpaceLightGrid;
   class DtGLTexture;
   class DtGLTextureBufferObject;

   class TextureApplyAttribute;
   class TextureBufferObjectApplyAttribute;

   class DT_DLL_VRVOSG DtOsgLightingComponent : public DtOsgStateComponent
   {
   public:
      DtOsgLightingComponent(int maxLights);

      virtual ~DtOsgLightingComponent();

      void update(const VectorLights& lights, const osg::State& state);

      void setMaxLights(int maxNumLights);

      static int getTextureStartOffset(void);

      const Vector2_uint32& GetTileSize(void) const;
      const Vector2_uint32& GetGridMaxDims(void) const;

   public: // Access to internal uniforms

      osg::Uniform* getEnableLightingUniform() const;
     
     void setAmbientFactor(float ambientFactor);
     
     void setUseGlobalAmbient(bool useGlobalAmbient);
     
     void setDiffuseFactor(float diffuseFactor);
     
     void setUseGlobalDiffuse(bool useGlobalDiffuse);
     
     void setUseHDR(bool useHDR);

     void setDirty(bool isDirty) { myDirty = isDirty; }

     void setEmissionFactor(float emissionFactor);

     bool dirty() const { return myDirty; }

     void releaseGLObjects(void);

   public: // DtOsgStateComponent interface

      virtual bool install(osg::StateSet* stateset);

      virtual bool uninstall(osg::StateSet* stateset);

   protected:

      osg::ref_ptr<osg::Uniform> myEnableLightingUniform;

      std::vector<osg::Uniform*> myUniforms;

      // Parameters of lights are stored in this structure
      DtLightData* myLightGrid;

      DtGLTextureBufferObject* myLightTexture;
      osg::ref_ptr<osg::Uniform> myLightTextureUniform;

      DtTileSpaceLightGrid* myTileSpaceLightGrid;

      DtGLTextureBufferObject* myTileLightIndexListTexture;
      osg::ref_ptr<osg::Uniform> myTileLightIndexListTextureUniform;

      DtGLTextureBufferObject* myTileLightGridCountOffsetsTexture;
      osg::ref_ptr<osg::Uniform> myTileLightGridCountOffsetsTextureUniform;

      DtGLTexture* myColorRampTexture;
      osg::ref_ptr<osg::Uniform> myColorRampTextureUniform;

      osg::ref_ptr<TextureApplyAttribute> myTextureApplyAttribute;
      osg::ref_ptr<TextureBufferObjectApplyAttribute> myTextureBufferObjectApplyAttribute;
      
      osg::ref_ptr<osg::Uniform> myProjectionMatrixUniform;
      osg::ref_ptr<osg::Uniform> myViewportSizeUniform;

      osg::ref_ptr<osg::Uniform> myLightGridTileDimUniform;
      osg::ref_ptr<osg::Uniform> myLightGridMaxDimUniform;
     
      osg::ref_ptr<osg::Uniform> myAmbientFactorUniform;
     bool myUseGlobalAmbient;
      osg::ref_ptr<osg::Uniform> myDiffuseFactorUniform;
     bool myUseGlobalDiffuse;
     bool myUseHDR;
     bool myDirty;
      osg::ref_ptr<osg::Uniform> myEmissionFactorUniform;
     
      osg::Uniform* createUniform(const std::string& name, osg::Uniform::Type type, int count=1);

      void PackLights(const VectorLights& lights
         , const osg::State& state);
      void UpdateTextureLightDataGrid(void);
      void UpdateTiledShadingData(const VectorLights& lights
         , const osg::State& state);

      void UpdateTileLightIndexListTexture();
      void UpdateTileLightGridCountOffsetsTexture();

      VectorLights myViewSpaceLightsScratchPad;

      int32* myTileLightGridCountOffsets;
      uint32 myTileLightGridCountOffsetsSize;

      bool myIsInstalled;

      static bool bColorRampTextureWasTried;
   };
}