DtOsgEffectTextureVisitor.h

Go to the documentation of this file.

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


#pragma once

#include <vrvOsg/vrvOsg.h>
#include <vrvOsg/DtPossibleFileNamesCache.h>

#include <osgDB/Options>

#include <osg/NodeVisitor>
#include <osg/ExtendedMaterial>

#include <string>

namespace makVrv
{
   class DtOsgEffectTextureController;
   class DtOsgFileCache;
   class DtOsgShaderManager;
   class DtEffectControllerContainer;

   class DT_DLL_VRVOSG DtOsgEffectTextureVisitor : public osg::NodeVisitor
   {
   public:
      DtOsgEffectTextureVisitor(DtDe& de, DtOsgFileCache& fileCache);

      DtOsgEffectTextureVisitor(DtDe& de, DtOsgFileCache&, const osgDB::Options* options);

      virtual void setDbOptions(const osgDB::Options* options);

      virtual ~DtOsgEffectTextureVisitor();

      virtual void reset();

      virtual unsigned numEffectTexturesInstalled() const;

      static const char* ignoreHint(void);

   public: // osg::NodeVisitor overrides

      void apply(osg::Node& node);
      void apply(osg::Geode& geode);

   protected:

      void initialize();

      void apply(osg::StateSet* stateSet);

      virtual bool isEffectTexture(osg::Texture* tex, const std::string& texName) const;

      // check is we already visited this stateSet.
      // \param extendedMaterial osg::ExtendedMaterial in this stateSet
      virtual bool checkIfAlreadyVisited(osg::StateSet* stateSet, osg::ExtendedMaterial* extendedMaterial);

      //Install the effect texture, it will first trying to read the effect texture from the 
      //extended material, if it's not supported in extended material or failed to find it in extended material
      //we try the postfix way.
      virtual void installEffectTexture(osg::StateSet* stateSet, osg::Texture* baseTexture, int baseUnit,
         osg::ExtendedMaterial* extendedMaterial);

      //Install the effect texture from the osg::ExtendedMaterial
      virtual bool installExtendedEffectTexture(const DtOsgEffectTextureController* controller,
         osg::StateSet* stateSet, osg::Texture* baseTexture, osg::Texture* effectTexture, int baseUnit, int effectUnit);

      virtual bool installExtendedEffectTexture(const DtOsgEffectTextureController* controller
         , osg::StateSet* stateSet, osg::Texture* baseTexture, int baseUnit, osg::ExtendedMaterial* extendedMaterial);

      virtual bool installPostfixEffectTexturesFromTextureLayers(const DtOsgEffectTextureController* controller, osg::Texture* baseTex,
         int baseTexUnit, osg::StateSet* stateSet);

      virtual bool installPostfixEffectTexturesFromFileSystem(const DtOsgEffectTextureController* controller, osg::Texture* baseTex,
         int baseTexUnit, osg::StateSet* stateSet);

      virtual bool checkForGlassMaterial(osg::StateSet* stateSet);

   protected:
      unsigned      myCount;

      osg::ref_ptr<const osgDB::Options> myDbOptions;
      DtOsgFileCache& myFileCache;

      DtPossibleFileNamesCache myPossibleFileNamesCache;

      DtEffectControllerContainer& myEffectControllerContainer;

      static const char* theIgnoreHint;

      DtDe& myDe;
   };
}