DtTritonShaderUniformManager.h

Go to the documentation of this file.

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


#pragma once

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

#include <vrvCore/DtVirtualBaseClassCreator.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 DtGlBuffer;

   class DtBindingManager;

   class DtTritonShaderUniformManager;

   typedef DtVirtualBaseClassCreator<DtTritonShaderUniformManager, DtTritonDrawable*> DtTritonShaderUniformManagerCreator;

   template <>
   inline const char* creatorTypeName<DtTritonShaderUniformManager>(void) { return "DtTritonShaderUniformManagerCreator"; }

   class DT_DLL_VRVTRITON DtTritonShaderUniformManager
   {
   public:
      friend DtTritonShaderUniformManagerCreator;

      virtual ~DtTritonShaderUniformManager();

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

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

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

      virtual void updateShadingUniforms(GLint shader, int shaderType);

      virtual void updateReverseZ(GLint shader, int shaderType, bool isReverseZ);

      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;

      virtual void bindUbo();
      
      virtual void unbindUbo();
   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

      struct DtVrvTritUboConstantLocations
      {
      public:
         int myMutableModelViewMatrixUniformLoc;
         int myMutableModelViewInverseTransposeMatrixUniformLoc;

         int myMutableCloudShadowTexGenUniformLoc;
         int myMutableCloudShadowTexGenUniform2Loc;

         int myMutableCloudShadowActiveLoc;

         int myMutableVrvTritDebugValueLoc;

         int myMutableVrvTritUseGGXLoc;

         int myVrvTritRoughnessLoc;
         int myVrvTritRoughfloorLoc;
         int myVrvTritSparkleMixFactorLoc;
         int myVrvTritWindToRoughnessLoc;
         int myVrvTritIncludeGLoc;
         int myVrvTritFinalSpecscaleLoc;

         int myVrvTritEnvInReflectLoc;
         int myVrvTritReflectionPowerLoc;

         int myVrvTritReverseZLoc;
      };

      mutable DtVrvTritUboConstantLocations myTritUboConstantLocations;

      struct DtVrvTritUniformLocations
      {
      public:
         int myMutableColorRampTextureUniform;

         int myMutableCloudShadowTextureUniform;
         int myMutableCloudShadowTextureUniform2;

         int myMutableOceanMaskTextureUnitUniform;

         int myMutableLatLongMinMax;

         int myMutableOsgViewMatrixInverse;
         int myMutableOsgModelViewMatrix;
      };

      mutable std::array<DtVrvTritUniformLocations, NUM_TRITON_SHADER_TYPES> myTritUniformLocations;

      mutable DtGlBuffer* myGpuShaderUbo;
      mutable int myGpuShaderUboBinding;
      
      DtBindingManager& myBindingManager;
   };

} // namespace makVrv