DtShaderEffectSnow.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 DT_DLL_VRVOSG DtShaderEffectSnow : public DtShaderEffect
   {
   public:
      // Inner class for generating texture for snow blowing across the ground effect
      class SnowEffectGeneration : public osg::Operation
      {
      public:
         SnowEffectGeneration(DtShaderEffectSnow& effectManager, DtDe& de);

         virtual ~SnowEffectGeneration();

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

      protected:

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

         virtual void release();

         virtual GLuint genComputeProg();

         virtual void windSpeedChanged(float speed);

         virtual void windDirectionChanged(float direction);
      protected:
         DtRandomNumberGenerator myRandomNumberGen;

         DtShaderEffectSnow& myShaderEffectManager;
         DtDe& myDe;

         GLuint myComputeId;

         bool myClearedLines;
         int myOutputTextureUniformLoc;
         int mySimTimeUniformLoc;
         int myBlownSnowTexUniformLoc;
         int myWindUniformLoc;
         int myCurLine;
         double myLastSimTime;
         double myLastWindTime;
         double myLastDropTime;
         std::vector<float> myBlowingSnowLine;
         //x,y wind direction, z wind speed.
         osg::Vec3 myWindUniform;

         DtSignalConnectionManager mySignals;
      };


   public:
      virtual ~DtShaderEffectSnow();

   protected:
      DtShaderEffectSnow(DtDe& de);

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

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

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

      virtual void registerWeatherEffectsObject(DtOsgWeatherEffectsObject* weatherEffectsObject);

      virtual void reloadShader();
      
      virtual void updateImGui();

      virtual void updateUniforms();

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

      static const std::string& theClassName();

   protected:

      virtual void slot_snowAccumulationChanged(double depth);

      virtual void slot_snowRoadAccumulationChanged(double depth);

      virtual void slot_snowMinimumElevationChanged(double elevation);

      virtual void slot_snowElevationTransitionChagned(double elevation);

      virtual SnowEffectGeneration* createSnowEffectJob();

   protected:

      DtSignalConnectionManager myConnections;

      osg::ref_ptr<osg::Uniform> myEnableSnowUniform;
      osg::ref_ptr<osg::Uniform> mySnowUpThresholdUniform;
      osg::ref_ptr<osg::Uniform> mySnowColorUniform;
      osg::ref_ptr<osg::Uniform> mySnowMaskedRoadUniform;

      GLuint mySnowEffectTextureId;
      GLuint64 mySnowEffectImageHandle;
      osg::ref_ptr<osg::Texture2D> mySnowEffectTexture;

      bool mySnowEnabled;
      float mySnowUpThresholdRange;
      float mySnowUpThresholdMin;
      float mySnowElevationMin;
      float mySnowElevationRange;

      float mySnowMaskedRoadMax;
      float mySnowMaskedRoadRange;

      bool myEnableBlowingSnow;

      osg::Vec4 mySnowColor;

      DtOsgWeatherEffectsObject* myWeatherEffectsObject;

      SnowEffectGeneration* mySnowEffectGenerationJob;

      DtOsgRenderer& myOsgRenderer;
   };
}