DtOsgShaderUtils.h

Go to the documentation of this file.

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


#pragma once

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

#include <vrvGraphicsUtils/DtCommonTypes.h>

#include <boost/unordered_map.hpp>

#include <osg/StateAttribute>

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

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

   class DT_DLL_VRVOSG DtOsgShaderUtils
   {
      
   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);

      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(osg::State& state, int stateToSet);
      static void restoreOpenGLStateAttribute(osg::State& state, osg::StateAttribute::Type type);
      static void restoreOpenGLProgram(osg::State& state);
      static void restoreOpenGLFBO(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);

   private:

      static ShaderUniLocMap s_shaderUniLocMap; 
   };

  
   
}