DtOsgEffectTextureController.h

Go to the documentation of this file.

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


#pragma once

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

#include <set>
#include <string>
#include <vector>

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(DtOsgFileCache& fileCache, DtOsgShaderManager& shaderMgr);

   public: // Implementation methods

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

      typedef std::set<std::string> FilenameSet;

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

      virtual osg::Texture* findEffectTexture(const std::string& baseTextureName, const osgDB::Options* dbOptions, const FilenameSet &possibleFilenames, osg::Texture* baseTexture);

      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(osg::Texture* baseTexture, 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;
      
      DtOsgFileCache& myFileCache;
      DtOsgShaderManager& myShaderMgr;

      std::string myShaderFolderPath;
   };
}