DtOsgShaderUtils.h

Go to the documentation of this file.

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


#pragma once

#include <vrvOsg/vrvOsg.h>
#include <vrvUtil/DtVirtualBaseClass.h>

#include <vrvGraphicsUtils/DtCommonTypes.h>

#include <osg/StateAttribute>

#include <unordered_map>

// forward declarations
namespace osg
{
   class State;
   class StateSet;
   class Texture;
   class Node;
   class Uniform;
}

namespace makVrv
{
   class DtOsgFileCache;

   class DT_DLL_VRVOSG DtOsgShaderUtils
   {
   private:
      DtOsgShaderUtils() = delete;

      DtOsgShaderUtils(const DtOsgShaderUtils& rhs) = delete;

      DtOsgShaderUtils& operator = (const DtOsgShaderUtils& rhs) = delete;

      virtual ~DtOsgShaderUtils() = delete;

   public:
      static void bindAttributes(const osg::GLExtensions * ext, const osg::StateSet *ss, int shader, int contextID);

      static bool getVPShaderVector(osg::StateSet *ss, osg::State& state, unsigned int &shaderHash,
         std::vector<unsigned int> &vertShaderVector, std::vector<unsigned int> &fragShaderVector, bool skip_shadow_shaders = false, bool skip_multiview_shaders = false);

      static bool addDegenerateShadersIfNeeded(osg::StateSet *ss, osg::State& state,
         std::vector<unsigned int> &vertShaderVector, std::vector<unsigned int> &fragShaderVector, bool skip_shadow_shaders = false);

      static void restoreEverything(osg::State& state);

      static void restoreOpenGLMode(const osg::State& state, int stateToSet);

      static void restoreOpenGLStateAttribute(osg::State& state, osg::StateAttribute::Type type);

      static void restoreOpenGLProgram(const osg::State& state);

      static void restoreOpenGLFBO(const osg::State& state);

      static void markAllTextureAttributesDirty(osg::State & state);

      static void generateModelInstanceShaders(osg::Node *node, const std::string& shaderName);

      static unsigned getNumGPUTextureImageUnits();

      static void clearCachedProgramsAndUniformLocations(void);

   protected:

      struct CachedUniformData
      {
         CachedUniformData() : location(-1), modCount(0), uniform(0) { ; }
         int location;
         int modCount;
         void * uniform;
      };
      typedef std::unordered_map<unsigned int, CachedUniformData > UniformLocationMap;
      typedef std::unordered_map<int, UniformLocationMap> ShaderUniLocMap;

      static ShaderUniLocMap theShaderUniLocMap;
   };



}