DtOsgLightingComponent.h

Go to the documentation of this file.

/****************************************************************************** 
** Copyright (c) 2016 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 DeDe;

   class TextureApplyAttribute;
   class TextureBufferObjectApplyAttribute;

   class DT_DLL_VRVOSG DtOsgLightingComponent : public DtOsgStateComponent
   {
   public:
      DtOsgLightingComponent(int maxLights, DtDe& de);

      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:

      void setLightPointScale(double scale);
      void setDynamicLightScale(double scale);
      void setEmissiveTextureScale(double scale);
      void setSpecularScale(double scale);
      void setShininessScale(double scale);

      void setHdrOutputScale(double scale);

      osg::Uniform* createUniform(const std::string& name, osg::Uniform::Type type, int count=1);

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

      void UpdateTileLightIndexListTexture();
      void UpdateTileLightGridCountOffsetsTexture();

   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> 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> myUseGlobalAmbientUniform;
      osg::ref_ptr<osg::Uniform> myDiffuseFactorUniform;
      bool myUseGlobalDiffuse;
      osg::ref_ptr<osg::Uniform> myUseGlobalDiffuseUniform;
      bool myUseHDR;
      bool myDirty;
      osg::ref_ptr<osg::Uniform> myEmissionFactorUniform;

      osg::ref_ptr<osg::Uniform> myTransmittanceFactorUniform;
      osg::ref_ptr<osg::Uniform> myModelTransmittanceUniform;

      osg::ref_ptr<osg::Uniform> myLightPointScaleUniform;
      osg::ref_ptr<osg::Uniform> myDynamicLightScaleUniform;
      osg::ref_ptr<osg::Uniform> myEmissiveTextureScaleUniform;
      osg::ref_ptr<osg::Uniform> mySpecularScaleUniform;
      osg::ref_ptr<osg::Uniform> myShininessScaleUniform;

      osg::ref_ptr<osg::Uniform> myHdrOutputScaleUniform;
      osg::ref_ptr<osg::Uniform> myUpVectorUniform;

      VectorLights myViewSpaceLightsScratchPad;

      int32* myTileLightGridCountOffsets;
      uint32 myTileLightGridCountOffsetsSize;

      bool myIsInstalled;

      static bool bColorRampTextureWasTried;
      DtDe& myDe;

   };
}