DtShaderEffectRain.h

Go to the documentation of this file.

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


#pragma once

#include <vrvOsg/vrvOsg.h>

#include <vrvOsg/DtShaderEffect.h>

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

#include <osg/ref_ptr>
#include <osg/Texture2D>
#include <osg/Uniform>
#include <osg/Object>
#include <osg/StateSet>
#include <osg/OperationThread>
#include <osg/Vec4>
#include <osg/State>

#include <string>
#include <vector>

namespace makVrv
{
   class DtOsgWeatherEffectsObject;
   class DtIndirectTextureProcessor;

   class DT_DLL_VRVOSG DtShaderEffectRain : public DtShaderEffect
   {
   public:

      // Inner class for generating a texture for rain hitting the puddles
      class RainEffectGeneration : public osg::Operation
      {
      public:
         RainEffectGeneration(DtShaderEffectRain& effectManager, DtDe& de);
         
         virtual ~RainEffectGeneration();
         
      private:
         RainEffectGeneration();
         RainEffectGeneration(const RainEffectGeneration&);
         RainEffectGeneration &operator=(const RainEffectGeneration&);

      protected:

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

         virtual void release();

         virtual GLuint genComputeProg();

      protected:
         DtRandomNumberGenerator myRandomNumberGen;

         DtShaderEffectRain& myShaderEffectManager;
         DtDe& myDe;

         GLuint myComputeId;

         bool myClearedDrops;
         int myOutputTextureUniformLoc;
         int mySimTimeUniformLoc;
         int myRainDropsTexUniformLoc;
         int myCurrentRainDrop;
         double myLastSimTime;
         
         std::vector<float> myRainDrops;
      };

   public:
      virtual ~DtShaderEffectRain();

   protected:
      DtShaderEffectRain(DtDe& de);

   private:
      DtShaderEffectRain();
      DtShaderEffectRain(const DtShaderEffectRain&);
      DtShaderEffectRain &operator=(const DtShaderEffectRain&);

   public:
      virtual void initializeEffect(const std::vector<osg::StateSet*>& stateSets,
         DtIndirectTextureProcessor* textureProcessor, DtOsgRenderer& osgRenderer);

      virtual void registerWeatherEffectsObject(DtOsgWeatherEffectsObject* weatherEffectsObject);

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

      virtual void updateImGui();

      virtual void updateUniforms();

   protected:

      virtual RainEffectGeneration* createRainEffectJob();
      
      virtual void slot_rainAccumulationChanged(double depth);


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

      static const std::string& theClassName();

   protected:

      DtSignalConnectionManager myConnections;

      osg::ref_ptr<osg::Uniform> myEnableRainUniform;
      osg::ref_ptr<osg::Uniform> myRainUpThresholdUniform;
      osg::ref_ptr<osg::Uniform> myRainColorUniform;
      osg::ref_ptr<osg::Uniform> myRainMaskedRoadUniform;
      
      GLuint myRainEffectTextureId;
      osg::ref_ptr<osg::Texture2D> myRainEffectTexture;
         
      DtOsgRenderer& myOsgRenderer;

      bool myRainEnabled;
      float myRainUpThresholdRange;
      float myRainUpThresholdMin;

      float myRainMaskedRoadMax;
      float myRainMaskedRoadRange;
      
      bool myEnablePuddleEffects;

      osg::Vec4 myRainColor;

      DtOsgWeatherEffectsObject* myWeatherEffectsObject;

      RainEffectGeneration* myRainEffectGenerationJob;

   };
}