DtShaderEffectSnow.h

Go to the documentation of this file.

/******************************************************************************
** Copyright (c) 2022 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 <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 DtShaderEffectSnow;
   class DtIndirectTexturePrototype;
   class DtIndirectGlTextureObject;

   typedef DtVirtualBaseClassCreator<DtShaderEffectSnow> DtShaderEffectSnowCreator;

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

   class DT_DLL_VRVOSG DtShaderEffectSnow : public DtShaderEffect
   {
   public:
      virtual ~DtShaderEffectSnow();

      DtShaderEffectSnow(DtDe& de);

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

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

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

      virtual void registerWeatherEffectsObject(DtOsgWeatherEffectsObject* weatherEffectsObject);

      virtual void reloadShader();
      
      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 void setWindSpeed(float speed);

      virtual void setWindDirection(float direction);

   public:

      static const std::string& theClassName();

   protected:

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

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

      virtual void updateEnabledState();

      virtual void slot_snowAccumulationChanged(double depth);

      virtual void slot_snowRoadAccumulationChanged(double depth);

      virtual void slot_snowMinimumElevationChanged(double elevation);

      virtual void slot_snowElevationTransitionChanged(double elevation);

      virtual void slot_PreObserverUpdate(double simTime);
   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;
      osg::ref_ptr<osg::Uniform> mySnowWindUniform;
      osg::ref_ptr<osg::Uniform> mySnowPolyMaskUniform;

      GLuint mySnowEffectTextureId;
      GLuint64 mySnowEffectImageHandle;
      osg::ref_ptr<osg::Texture2D> mySnowEffectTexture;
      const DtIndirectTexturePrototype* mySnowEffectVrvTexture = nullptr;
      DtIndirectGlTextureObject* mySnowEffectTextureBindless = nullptr;

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

      float mySnowMaskedRoadMax;
      float mySnowMaskedRoadRange;

      bool myEnableBlowingSnow;

      osg::Vec4 mySnowColor;

      //x,y wind direction, z wind speed, w wind angle radians.
      osg::Vec4 myWindUniform;

      float myWindOffset;
      double myLastSimulationTime;

      DtVector myCameraLocation;
      double myCameraAltitude;

      DtOsgWeatherEffectsObject* myWeatherEffectsObject;

      friend class DtSnowEffectGeneration;
   };
}