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;

   
   typedef boost::unordered_map<void *, int> UniformLocationMap;
   typedef boost::unordered_map<int, UniformLocationMap> ShaderUniLocMap; 


   class DT_DLL_VRVOSG DtOsgShaderUtils
   {
      
   public:
      static std::string loadShaderSource(const std::string& strFileName);

      static void saveShaderSource(const std::string& strFileName, const std::string& strSource);

      static std::string fixVersionDirective(const std::string &shaderString);

      static unsigned getNumGPUTextureImageUnits();

      //         hashPtr is just a pointer to use for the location hash. it should be unique for each uniform set this way
      //         for an example look at how DtSpeedTreeForest uses this
      static void bindSingleAttribute(int shader, const std::string& uniName, bool value, void *hashPtr);

      static void bindAttributes(osg::Node *node, 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);


      static void deleteShaders(const std::vector<unsigned int>& shaderVector);

      static bool compileShader(const char* source, unsigned int shaderType, unsigned int &shaderID);

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

   private:

      static ShaderUniLocMap s_shaderUniLocMap; 
   };

  
   
}