DtShaderEffectManager.h

Go to the documentation of this file.

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


#pragma once

#include <vrvOsg/vrvOsg.h>

#include <vrvOsg/DtIndirectTextureProcessor.h>

#include <vrvCore/DtEnvironmentInterface.h>

#include <vrvUtil/DtVirtualBaseClass.h>
#include <vrvUtil/DtSignalConnectionManager.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
   {
   protected:

      typedef std::set<DtShaderEffect*> ShaderEffectSet;

   public:
      
      struct DT_DLL_VRVOSG DtShaderEffectPerChannelData
      {
      public:
         osg::Vec3d myLocalPostion;
         osg::Matrixd myLocalMatrixTranformInverse;
         ShaderEffectSet myShaderEffectSet;
      };

      // 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();

      DtShaderEffectManager() = delete;
      DtShaderEffectManager(const DtShaderEffectManager&) = delete;
      DtShaderEffectManager &operator=(const DtShaderEffectManager&) = delete;

   protected:
      DtShaderEffectManager(DtDe& de);

   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 void slot_elevationalConfigurationUpdated(DtEnvironmentInterface::DtElevationalConfigurationRecord rec);

      virtual DtOsgWeatherEffectsObject* findWeatherEffectsObject();

      virtual const DtOsgWeatherEffectsObject* findWeatherEffectsObject() const;

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

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

      static const std::string& theClassName();

   protected:

      virtual void updateImGui();

      virtual void updateUniforms();

      virtual DtShaderEffectPerChannelData& getOrCreatePerChannelData(osg::Camera* camera);
  
      virtual void flushInvalidChannels();

   protected:

      DtDe& myDe;

      DtSignalConnectionManager myConnections;

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

      typedef std::map< osg::observer_ptr<osg::Camera>, DtShaderEffectPerChannelData*> PerChannelDataMap;
      PerChannelDataMap myPerChannelDataMap;

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

      DtOsgWeatherEffectsObject* myWeatherEffectsObject;

      osg::ref_ptr<TextureAddressJob> myTextureAddressJob;

      float myWindBaseMSAltitude;
      float myWindSpeed;
      float myWindDirection;
      float myVisibilityDistance;
      DtEnvironmentInterface::VisibilityObscurant myVisibilityObscurant;
   };
}