DtTritonShaderUniformManager.h

Go to the documentation of this file.

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


#pragma once

#include <vrvTriton/vrvTriton.h>
#include <vrvTriton/DtTritonConfig.h>

#include <vrvUtil/DtVirtualBaseClass.h>

#include <vrvGraphicsUtils/DtVector4.h>
#include <vrvGraphicsUtils/DtMatrix4.h>
#include <vrvGraphicsUtils/DtVector2.h>

#include <osg/Matrixf>

#include <set>

// avoid including glew
typedef int GLint;

namespace osg
{
   class State;
}

namespace makVrv
{
   class DtDe;
   class DtTritonDrawable;

   class DT_DLL_VRVTRITON DtTritonShaderUniformManager
   {
   public:
      virtual ~DtTritonShaderUniformManager();

   protected:
      friend class DtTritonShaderUniformManagerCreator;
      DtTritonShaderUniformManager(DtTritonDrawable* tritonDrawable, DtDe& de);

   private:
      DtTritonShaderUniformManager();
      DtTritonShaderUniformManager(const DtTritonShaderUniformManager &other);
      DtTritonShaderUniformManager &operator=(const DtTritonShaderUniformManager &other);
   
   public:
      virtual bool initShaderUniformHandles(GLint shader, int shaderType) const;

      virtual void updateCloudShadowUniforms(GLint shader, int shaderType
         , GLboolean bIsCloudShadowActive
         , unsigned int cloudshadowTextureUnit
         , unsigned int cloudshadowTextureIndex
         , const osg::Matrixf& textureViewProjMatrix
         , unsigned int cloudshadowTextureUnit2
         , unsigned int cloudshadowTextureIndex2
         , const osg::Matrixf& textureViewProjMatrix2) const;

      virtual void updateViewingUniforms(GLint shader, int shaderType
         , const Matrix4_32& view_matrix_32
         , const Matrix4_32& view_inverse_transpose_matrix_32);
      
      virtual void updateLightingUniforms(osg::State &state
         , const Vector2_uint32& tileSize
         , const Vector2_uint32& maxDims
         , int textureStartOffset
         , GLint shader, int shaderType);

      virtual void updateOceanMaskingUniforms(GLint shader, int shaderType, unsigned int maskUnit, const DtVector4<double>& latLongMinMax);

      virtual void updateUpdateMatricesForSensorFX(GLint shader, int shaderType
      , const osg::Matrixf& osgInverseViewMatrix, const osg::Matrixf& osgModelViewMatrix);

      void resetToValue(int shaderType, int val) const;

   protected:
      DtTritonDrawable* myTritonDrawable;

      DtDe& myDe;  // De reference

      // set to track which program ID's we've bound our interface blocks to, to avoid redundancies.
      std::set<unsigned int> myProgramIds;
      
      //Arrays for caching the shader uniform handles. Mutable since they are 
      // updated in drawImplemtation of the DtTritonDrawable which osg requires to be const
      mutable int myMutableViewPortSizeUniform[NUM_TRITON_SHADER_TYPES];

      mutable int myMutableColorRampTextureUniform[NUM_TRITON_SHADER_TYPES];

      mutable int myMutableModelViewMatrixUniform[NUM_TRITON_SHADER_TYPES];
      mutable int myMutableModelViewInverseTransposeMatrixUniform[NUM_TRITON_SHADER_TYPES];

      mutable int myMutableLightGridTileDimUniform[NUM_TRITON_SHADER_TYPES];
      mutable int myMutableLightGridMaxDimUniform[NUM_TRITON_SHADER_TYPES];

      mutable int myMutableCloudShadowTextureUniform[NUM_TRITON_SHADER_TYPES];
      mutable int myMutableCloudShadowTextureUniform2[NUM_TRITON_SHADER_TYPES];
      mutable int myMutableCloudShadowTexGenUniform[NUM_TRITON_SHADER_TYPES];
      mutable int myMutableCloudShadowTexGenUniform2[NUM_TRITON_SHADER_TYPES];
      mutable int myMutableCloudShadowActive[NUM_TRITON_SHADER_TYPES];

      mutable int myMutableOceanMaskTextureUnitUniform[NUM_TRITON_SHADER_TYPES];
      mutable int myMutableLatLongMinMax[NUM_TRITON_SHADER_TYPES];

      mutable int myMutableOsgViewMatrixInverse[NUM_TRITON_SHADER_TYPES];
      mutable int myMutableOsgModelViewMatrix[NUM_TRITON_SHADER_TYPES];
   };

   class DT_DLL_VRVTRITON DtTritonShaderUniformManagerCreator : public DtVirtualBaseClass
   {
   protected:
      DtTritonShaderUniformManagerCreator();

      virtual ~DtTritonShaderUniformManagerCreator();

   private:
      DtTritonShaderUniformManagerCreator(const DtTritonShaderUniformManagerCreator &other);
      DtTritonShaderUniformManagerCreator &operator=(const DtTritonShaderUniformManagerCreator &other);

   public:
      static DtTritonShaderUniformManagerCreator& instance(DtDe& de);

      static void registerInstance(DtDe& de,
         DtTritonShaderUniformManagerCreator* instance);

   public:
      virtual DtTritonShaderUniformManager* create(DtTritonDrawable* parentDrawable,
                     DtDe& de);
   };


} // namespace makVrv