DtTritonShaderUniformManager.h

Go to the documentation of this file.

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


#pragma once

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

#include <vrvUtil/DtVirtualBaseClassCreator.h>

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

#include <osg/Matrixf>

#include <set>
#include <array>

// avoid including glew
typedef int GLint;

namespace osg
{
   class State;
}

namespace makVrv
{
   class DtDe;
   class DtTritonRenderingParameters;

   class DtGlBuffer;

   class DtTritonShaderUniformManager;

   typedef DtVirtualBaseClassCreator<DtTritonShaderUniformManager> DtTritonShaderUniformManagerCreator;

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

   class DT_DLL_VRVTRITON DtTritonShaderUniformManager
   {
   public:
      friend DtTritonShaderUniformManagerCreator;

      virtual ~DtTritonShaderUniformManager();

   protected:
      DtTritonShaderUniformManager(DtDe& de);

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

      virtual void updateShadingUniforms(GLint shader, int shaderType, const DtTritonRenderingParameters& renderingParameters);

      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 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);

      virtual void resetToValue(int shaderType, int val);

      virtual void bindUbo();
      
      virtual void unbindUbo();
   protected:

      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 myModelViewMatrixUniformLoc;
         int myModelViewInverseTransposeMatrixUniformLoc;

         int myTritRegularShadowActiveLoc;

         int myVrvTritDebugValueLoc;

         int myVrvTritUseGGXLoc;

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

         int myVrvTritEnvInReflectLoc;
         int myVrvTritReflectionPowerLoc;

         int myVrvTritReverseZLoc;
      };

      DtVrvTritUboConstantLocations myTritUboConstantLocations;

      struct DtVrvTritUniformLocations
      {
      public:
         int myOceanMaskTextureUnitUniform;

         int myLatLongMinMax;

         int myOsgViewMatrixInverse;
         int myOsgModelViewMatrix;
      };

      std::array<DtVrvTritUniformLocations, NUM_TRITON_SHADER_TYPES> myTritUniformLocations;

      DtGlBuffer* myGpuShaderUbo;
      int myGpuShaderUboBinding;
   };

} // namespace makVrv