DtWeatherEffectDrawable.h

Go to the documentation of this file.

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


#pragma once

#include <osg/Camera>
#include <osg/Drawable>

#include <vrvCore/DtDe.h>

#include <vrvOsg/DtWeatherEffect.h>
#include <vrvOsg/DtOsgCullVisitor.h>
#include <vrvOsg/vrvOsg.h>

#include <vrvUtil/DtVirtualBaseClass.h>


namespace makVrv
{
   // Forward declarations.
   class DtWeatherLayerGroup;
   class osg::Texture2D;


   class DT_DLL_VRVOSG DtWeatherEffectDrawable : public osg::Drawable
   {
   public:

      DtWeatherEffectDrawable(makVrv::DtDe& de, DtWeatherLayerGroup* weatherLayerGroup);

      DtWeatherEffectDrawable(const DtWeatherEffectDrawable& orig, const osg::CopyOp& opr);

      virtual ~DtWeatherEffectDrawable();

   private:

      DtWeatherEffectDrawable() = delete;

      DtWeatherEffectDrawable(const DtWeatherEffectDrawable &other) = delete;

      DtWeatherEffectDrawable &operator=(const DtWeatherEffectDrawable &other) = delete;

   public:

      virtual osg::Object* clone(const osg::CopyOp& copyop) const { return new DtWeatherEffectDrawable(*this, copyop); }
      virtual osg::Object* cloneType() const { return NULL; }

      virtual bool init();

      virtual void SetCameraBasis(const osg::Matrixd& basisRot);

      virtual osg::Texture2D* getTexture(const char* imagePath);

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

      virtual void drawImplementation(osg::RenderInfo& renderInfo) const;

      virtual void applyState(DtWeatherEffect::DrawData& drawData) const;

      virtual void applyHeightMap(DtWeatherEffect::DrawData& drawData) const;

   protected:

      virtual bool loadShaders();

      virtual void connectSignals();

      virtual void disconnectSignals();

      virtual void slot_reloadShaders();

      virtual void setupSoftness(osg::Camera* camera) const;

      virtual float computeNearClip(osg::Camera* camera) const;

      virtual void clearTextureLibrary() const;

   protected:

      struct TextureEntry
      {
         osg::ref_ptr<osg::Texture2D> myTexture;
         std::string myName;
      };
      
      DtDe& myDe;
      DtWeatherLayerGroup* myWeatherLayerGroup;

      bool myInitialized;

      osg::ref_ptr<osg::Texture2D> myTexture;
      osg::ref_ptr<osg::StateSet> myStateSet;

      osg::Matrixd myBasisRot;

   private:

      mutable osg::ref_ptr<osg::Uniform> myMutableInvPersMatrixUniform;
      mutable osg::ref_ptr<osg::Uniform> myMutablePersMatrixUniform;
      mutable osg::ref_ptr<osg::Uniform> myMutableViewportUniform;
      mutable osg::ref_ptr<osg::Uniform> myMutableIsReflectionUniform;
      mutable osg::ref_ptr<osg::Uniform> myMutableSoftnessUniform;

      // weather effect specific
      mutable osg::ref_ptr<osg::Uniform> myMutableModelAxisXUniform;
      mutable osg::ref_ptr<osg::Uniform> myMutableModelAxisZUniform;
      mutable osg::ref_ptr<osg::Uniform> myMutableViewMatrixInvUniform;

      mutable std::vector<TextureEntry*> myTextureLibrary;
   };


   class DT_DLL_VRVOSG DtWeatherEffectDrawableCreator : public DtVirtualBaseClass
   {
   public:
      DtWeatherEffectDrawableCreator(DtDe &de);

      virtual ~DtWeatherEffectDrawableCreator();

   private:
      DtWeatherEffectDrawableCreator(const DtWeatherEffectDrawableCreator& other);
      DtWeatherEffectDrawableCreator &operator=(const DtWeatherEffectDrawableCreator& other);

   public:
      virtual DtWeatherEffectDrawable* create(DtWeatherLayerGroup* weatherLayerGroup) const;

      static DtWeatherEffectDrawableCreator& instance(DtDe& de);

      static const std::string& theClassName();

   protected:
      DtDe& myDe;
   };
}