DtPostProcessor.h

Go to the documentation of this file.

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

 
#pragma once

#include <vrvOsg/vrvOsg.h>

#include <osg/Camera>
#include <osg/Texture>
#include <osg/Program>
#include <osg/Shader>
#include <osg/Texture2D>
#include <osg/Geode>

#include <map>

#include <boost/unordered_map.hpp>

namespace makVrv 
{ 
   class DtChannel;

   class DT_DLL_VRVOSG DtPostProcessor
   {
   public:
      friend class DtOsgPostProcessManager;

      DtPostProcessor(const std::string& postProcessType, bool handles2D);

      virtual ~DtPostProcessor();

      void setFragmentShaderSource(const char *source);

      void setVertexShaderSource(const char *source);

      void setFragmentShaderFile(const std::string& fileName);

      void setVertexShaderFile(const std::string& fileName);


      void setInputTexture(const DtChannel *channel, osg::Texture2D* texture);

      virtual void update(DtChannel *channel) = 0;

      osg::Texture2D* outputTexture();

      void setIsFinalEffect(const DtChannel *channel, bool final);

      osg::Camera* cameraForChannel(const DtChannel *channel);

      virtual void releaseGLObjects();

      bool hasEffect(std::string effect);

      bool effectEnabled(std::string effect);

      bool anyEffectEnabled();
      



      virtual bool enableEffect(std::string effect);
      virtual bool disableEffect(std::string effect);

      std::list<std::string> supportedEffects();

      bool handles2D() const;

      const std::string& type() const;


      virtual void enablePostProcessor();
      virtual void disablePostProcessor();

      bool postProcessorEnabled() const;

   protected:
      void initialize();

      void addEffect(std::string effect);

      void setEnablePostProcessor(bool enabled);

   protected:

      osg::ref_ptr<osg::Texture2D> myOutputTexture, myInputTexture;
      osg::ref_ptr<osg::Program> myProgram;
      osg::ref_ptr<osg::Shader> myFragmentShader;
      osg::ref_ptr<osg::Shader> myVertexShader;
      osg::ref_ptr<osg::Geode> myQuad;

      char *myCustomFragmentSource;

      typedef std::map<const DtChannel*, osg::ref_ptr<osg::Camera> > CameraMap;
      CameraMap myCameraMap;

      bool myHandles2DFlag;
      bool myEnabledFlag;

      // The category of this post effect ex. "dof", "wateronscreen"
      std::string myType;

      // A list of sub-effects that can be enabled/disabled
      typedef boost::unordered_map<std::string, bool> EffectMap;

      EffectMap myEffects; // ex. { {"rainsplash",true}, {"oceanspray",false} }
   };

   class DT_DLL_VRVOSG DtPostProcessorCreator
   {
   public:
       friend class DtPostProcessorFactory;

       DtPostProcessorCreator();

       virtual ~DtPostProcessorCreator();

   protected:
       virtual DtPostProcessor* create(DtDe& de, const std::string& postProcessorType) = 0;
   };
}