DtVolumetricDrawable.h

Go to the documentation of this file.

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

#pragma once


#include <osg/Drawable>

// Get proper local export symbol
#include <vrvPostProcessor/vrvPostProcessor.h>

#include <vrvPostProcessor/DtPostProcessorDrawable.h>
#include <vrvOsg/DtCrepuscularRaysParameters.h>

#include <vrvUtil/DtVirtualBaseClassCreator.h>

#include <array>

namespace osg
{
   class Texture2D;
   class State;
}

namespace makVrv
{
   class DtGlRenderTexture;
   class DtGlTexture;
   class DtDe;
   class DtGlProgram;
   struct DtCrepuscularRaysParameters;

   class DtVolumetricDrawable;
   typedef DtVirtualBaseClassCreator<DtVolumetricDrawable> DtVolumetricDrawableCreator;

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

   class DT_DLL_VRVPOSTPROCESSOR DtVolumetricDrawable : public DtPostProcessorDrawable
   {
   public:
      enum  VOLUMETRIC_PASS
      {
         DOWNSAMPLE4X = 0,
         VOLUMETRIC_RAYS,
         FINAL_PASS,
         PASSES_SIZE
      };
   public:
      virtual ~DtVolumetricDrawable();

   protected:
      DtVolumetricDrawable(DtDe &de);

      friend DtVolumetricDrawableCreator;

   private:
      DtVolumetricDrawable();

      DtVolumetricDrawable(const DtVolumetricDrawable&);

      DtVolumetricDrawable& operator=(const DtVolumetricDrawable&);

   public:
      virtual void reloadShaders();

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

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

      virtual void setSize(int width, int height);

      virtual void setOcclusionTexture(osg::Texture* inputTex);

      virtual void setBaseTexture(osg::Texture* inputTex);

      virtual void setVisible(bool visible) { myVisible = visible; }
      virtual bool visible() const { return myVisible; }

      virtual void setParameters(const DtCrepuscularRaysParameters& params);
   protected:
      virtual void initRenderTextures() const;

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

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

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

      virtual void createComputeProgramIfNecessary(VOLUMETRIC_PASS pass, const std::string& path, const std::string& name) const;

      virtual void createProgramIfNecessary(VOLUMETRIC_PASS pass, const std::string& path, const std::string& prefixName) const;

      virtual void initializeUniformLocations(VOLUMETRIC_PASS pass) const;

      virtual void zeroOutUniformLocations() const;

      virtual void deleteShaders() const;

   protected:
      int myChannelWidth;
      int myChannelHeight;

      osg::ref_ptr<osg::Texture> myOcclusionTex;
      osg::ref_ptr<osg::Texture> myBaseTex;

      mutable DtGlTexture *myMutableDownSampledOcclusionTexture;
      mutable DtGlTexture *myMutableVolumetricRaysBuffer;

      bool myVisible;

      mutable std::array<DtGlProgram*, PASSES_SIZE> myMutablePrograms;

      DtCrepuscularRaysParameters myParameters;

      mutable int myMutableSlLightPositionOnScreenLoc;
      mutable int myMutableSlParametersLoc;
      mutable int myMutableSlLightColorLoc;
      mutable int myMutableSlOutputScaleLoc;

      mutable int myMutableBaseStrengthLoc;
      mutable int myMutableBaseTextureLoc;
      mutable int myMutableVolumetricTextureLoc;

      mutable bool myMutableShadersDirtyFlag;
   };


}