DtOsgEffectTextureController.h

Go to the documentation of this file.

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


#pragma once

#include <vrvOsg/vrvOsg.h>
#include <osg/Referenced>
#include <osgDB/Options>

#include <set>
#include <string>
#include <vector>
#include <boost/thread.hpp>
namespace osg
{
   class StateSet;
   class Texture;
}
namespace osgEarth
{
    class VirtualProgram;
}

namespace makVrv
{
   class DtOsgShaderManager;
   class DtOsgFileCache;

   class DT_DLL_VRVOSG DtOsgEffectTextureController : public osg::Referenced
   {
   public:
      DtOsgEffectTextureController(const DtOsgFileCache& fileCache);

   public: // Implementation methods

      virtual std::string getEffectTypeCode() const =0;

      typedef std::map<std::string, std::string> FilenameMap;

      virtual osg::Texture* installEffectTexture(osg::Texture* baseTexture, int texUnit, osg::StateSet* stateSet, const osgDB::Options* dbOptions, const FilenameMap& possibleFilenames);

      virtual void findEffectTexture(const std::string& baseTextureName, const osgDB::Options* dbOptions, const FilenameMap &possibleFilenames, const osg::Texture* baseTexture, osg::ref_ptr<osg::Texture>& effectTexture);

      virtual bool generateShaders(int baseUnit, int effectUnit, osg::Texture* tex, osg::StateSet* stateSet, const std::string& baseTextureName) { return true; }

   public: // Utility methods

      virtual bool isEffectTexture(osg::Texture* tex) const;

      virtual bool effectTextureInstalled(const std::string& baseTextureName, osg::StateSet* stateSet) const;

      // find the file path for a texture.
      static std::string getTextureFilePath(osg::Texture* texture);

      // find the extended name for a texture file.
      std::string getExtendedFileName(const std::string& baseFileName) const;
      
      // Given the file name of a base texture, populate the output
      // vector with possible file names for corresponding effect textures.
      void getPossibleEffectTextureFileNames(const std::string& baseFileName, std::vector<std::string>& output) const;

   protected:

      std::string myVertSource;
      std::string myFragSource;
      float       myDefaultMaxRange;

      virtual ~DtOsgEffectTextureController();

      // applies default shader LOD ranges, if applicable
      void applyDefaultShaderLODRanges(osgEarth::VirtualProgram* vp, const std::string& function) const;
      
      const DtOsgFileCache& myFileCache;

      const std::string myShaderFolderPath;

      static boost::mutex s_controllerMutex;

   };
}