DtShaderEffectRain.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/DtShaderEffect.h>

#include <vrvUtil/DtVirtualBaseClassCreator.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 DtGlProgram;
   class DtShaderEffectRain;
   class DtIndirectTexturePrototype;
   class DtIndirectGlTextureObject;
   class DtRainEffectGeneration;

   typedef DtVirtualBaseClassCreator<DtShaderEffectRain> DtShaderEffectRainCreator;

   template <>
   inline const char* creatorTypeName<DtShaderEffectRain>(void) { return "DtShaderEffectRainCreator"; }

   class DT_DLL_VRVOSG DtShaderEffectRain : public DtShaderEffect
   {
   public:
      DtShaderEffectRain(DtDe& de);

      virtual ~DtShaderEffectRain();

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

   public:
      virtual void initializeEffect(const std::vector<osg::StateSet*>& stateSets);

      virtual void registerWeatherEffectsObject(DtOsgWeatherEffectsObject* weatherEffectsObject);

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

      virtual void updateImGui();

      virtual void updateUniforms();

      virtual void setCameraLocation(const DtVector& location);

      // \brief gets the location for localized effects.
      virtual const DtVector& cameraLocation() const;

      virtual int rainEffectTextureSize(void) const;

   protected:

      virtual void installShader(const std::vector<osg::StateSet*>& stateSets);

      virtual void uninstallShader(const std::vector<osg::StateSet*>& stateSets);

      virtual DtRainEffectGeneration* createRainEffectOp();

      virtual void removeRainEffectGenerationOp(void);
      
      virtual void slot_rainAccumulationChanged(double depth);

      virtual void updateEnabledState(const float depth);

   public:

      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;
      
      const DtIndirectTexturePrototype* myRainEffectVrvTexture = nullptr;
      DtIndirectGlTextureObject* myRainEffectTextureBindless = nullptr;
         
      bool myRainEnabled;
      float myRainUpThresholdRange;
      float myRainUpThresholdMin;

      float myRainMaskedRoadMax;
      float myRainMaskedRoadRange;
      float myRainAccumulationDepth;
      
      bool myEnablePuddleEffects;

      osg::Vec4 myRainColor;

      DtOsgWeatherEffectsObject* myWeatherEffectsObject;

      friend class DtRainEffectGeneration;

      DtRainEffectGeneration* myRainEffectGenerationOp = nullptr;

      DtVector myCameraLocation;
      double myCameraAltitude;

      float myPrecipitationIntensity;

      const int myRainEffectTextureSize = 512;

   };
}