DtShaderEffectManager.h

Go to the documentation of this file.

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


#pragma once

#include <vrvOsg/vrvOsg.h>

#include <vrvOsg/DtGpuOperation.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 DtOsgRenderer;

   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 DtTextureAddressOp : public DtGpuOperation
      {
      public:
         DtTextureAddressOp(DtShaderEffectManager& effectManager, DtDe& de);
         virtual ~DtTextureAddressOp();
      public:
         virtual void execute(void);        
      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 updateTextureUniforms();

      virtual void update();

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

      virtual void registerWeatherEffectsObject(DtOsgWeatherEffectsObject* weatherEffectsObject);

      virtual void slot_reloadShader();

      virtual void slot_windSpeedChanged(float metersPerSecond);

      virtual void slot_windDirectionChanged(float radians);

      virtual DtOsgWeatherEffectsObject* findWeatherEffectsObject();

      virtual const DtOsgWeatherEffectsObject* findWeatherEffectsObject() const;

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

      virtual bool anyEffectDirty(osg::Camera* camera);

      virtual void installEffectsToCamera(osg::Camera* camera);

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

      static const std::string& theClassName();

   protected:

      virtual void updateImGui();

      virtual void updateUniforms();

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

      virtual void initializeChannelEffects();

      virtual DtTextureAddressOp* createTextureAddressOp();
   protected:

      DtDe& myDe;

      DtSignalConnectionManager myConnections;

      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;

      DtTextureAddressOp* myTextureAddressOp = nullptr;

      float myWindSpeed;
      float myWindDirection;
   };
}