DtShaderEffectManager.h

Go to the documentation of this file.

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


#pragma once

#include <vrvOsg/vrvOsg.h>

#include <vrvUtil/DtVirtualBaseClass.h>
#include <vrvUtil/DtSignalConnectionManager.h>

#include <vrvOsg/DtIndirectTextureProcessor.h>

#include <string>

#include <osg/ref_ptr>
#include <osg/Vec3d>
#include <osg/Matrixd>
#include <osg/OperationThread>
#include <osg/Object>
#include <osg/State>
#include <osg/RenderInfo>

namespace makVrv
{
   class DtOsgWeatherEffectsObject;
   class DtShaderEffect;
   class DtOsgChannel;

   class DT_DLL_VRVOSG DtShaderEffectManager : public DtVirtualBaseClass
   {
   public:
      
      struct DT_DLL_VRVOSG DtShaderEffectPerChannelData
      {
      public:
         osg::Vec3d myLocalPostion;
         osg::Matrixd myLocalMatrixTranformInverse;
      };

      // Inner class for passing the current render info from the main thread.
      // Calls findTextureAddresses in order to setup the texture addresses as they are 
      //  made resident
      class TextureAddressJob : public osg::Operation
      {
      public:
         TextureAddressJob(DtShaderEffectManager& effectManager);
         virtual ~TextureAddressJob();

         virtual void operator () (osg::Object* sv);

         virtual void release();

      protected:
         DtShaderEffectManager& myShaderEffectManager;

      };

   public:
      virtual ~DtShaderEffectManager();

   protected:
      DtShaderEffectManager(DtDe& de);

   private:
      DtShaderEffectManager();
      DtShaderEffectManager(const DtShaderEffectManager&);

      
      DtShaderEffectManager &operator=(const DtShaderEffectManager&);

   public:
      virtual void initializeEffects();

      virtual void findTextureAddresses(osg::RenderInfo& renderInfo);

      virtual TextureAddressJob* createTextureAddressJob();

      virtual void update();

      virtual void updateChannel(DtOsgChannel* channel, osg::Vec3d pos);

      virtual void registerWeatherEffectsObject(DtOsgWeatherEffectsObject* weatherEffectsObject);

      virtual void slot_reloadShader();

      virtual DtOsgWeatherEffectsObject* findWeatherEffectsObject();

      virtual const DtOsgWeatherEffectsObject* findWeatherEffectsObject() const;

      virtual void releaseGLObjects(osg::State* state);

      void addShaderEffect(DtShaderEffect& shaderEffect);

   public:
      static DtShaderEffectManager& instance(DtDe& de);

      static const std::string& theClassName();

   protected:

      virtual void updateImGui();

      virtual void updateUniforms();
  

   protected:

      DtDe& myDe;

      DtSignalConnectionManager myConnections;

      osg::ref_ptr<DtIndirectTextureProcessor> myTextureProcessor;
               
      DtOsgRenderer& myOsgRenderer;

      typedef std::map<DtOsgChannel*, DtShaderEffectPerChannelData> PerChannelDataMap;
      PerChannelDataMap myPerChannelDataMap;

      typedef osg::ref_ptr<osg::CoordinateSystemNode> CoordinateSystemNodeRef;
      CoordinateSystemNodeRef myCoordSystemNode;

      DtOsgWeatherEffectsObject* myWeatherEffectsObject;

      typedef std::set<DtShaderEffect*> ShaderEffectsSet;
      ShaderEffectsSet myShaderEffects;

      osg::ref_ptr<TextureAddressJob> myTextureAddressJob;

   };
}