DtHdrDrawable.h

Go to the documentation of this file.

/******************************************************************************
** Copyright (c) 2020 MAK Technologies, Inc.
** All rights reserved.
******************************************************************************/
// This file is derived from the HDR NVIDIA gameworks SDK example
//----------------------------------------------------------------------------------
// Multiple Files in folder:        es3aep-kepler\HDR/
// SDK Version: v2.11 
// Email:       gameworks@nvidia.com
// Site:        http://developer.nvidia.com/
//
// Copyright (c) 2014-2015, NVIDIA CORPORATION. All rights reserved.
//
// Redistribution and use in source and binary forms, with or without
// modification, are permitted provided that the following conditions
// are met:
//  * Redistributions of source code must retain the above copyright
//    notice, this list of conditions and the following disclaimer.
//  * Redistributions in binary form must reproduce the above copyright
//    notice, this list of conditions and the following disclaimer in the
//    documentation and/or other materials provided with the distribution.
//  * Neither the name of NVIDIA CORPORATION nor the names of its
//    contributors may be used to endorse or promote products derived
//    from this software without specific prior written permission.
//
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS ``AS IS'' AND ANY
// EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
// PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE COPYRIGHT OWNER OR
// CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
// EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
// PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
// PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
// OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
//
//----------------------------------------------------------------------------------

#pragma once



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

#include <osg/Drawable>
#include <osg/Program>
#include <osg/Texture>
#include <osg/Texture2D>

namespace osg
{
   class Object;
   class State;
   class RenderInfo;
   class CopyOp;
}

namespace makVrv 
{ 
   class DtDe;
   class DtGlTexture;
   class DtGlBuffer;

   class DT_DLL_VRVPOSTPROCESSOR DtHdrDrawable : public DtPostProcessorDrawable
   {
   public:

      enum HDR_PASS 
      { //Program ID enum
         DOWNSAMPLE = 0,
         DOWNSAMPLE4X,
         BLURH4,
         BLURV4,
         BLURH6,
         BLURV6,
         BLURH10,
         BLURV10,
         BLURH12,
         BLURV12,
         TONEMAPPING,
         EXTRACTHL,
         GAUSCOMP,
         STARSTREAK,
         GLARECOMP,
         STARCOMP,
         GHOSTIMAGE,
         CALCADAPTEDLUM,
         PROGSIZE
      };

      enum BUFFER_PYRAMID 
      {
         LEVEL_0 = 0, 
         LEVEL_1, 
         LEVEL_2,     
         LEVEL_3,
         LEVEL_4,
         LEVEL_5,
         LEVEL_TOTAL
      };

      struct DtHdrOptions
      {
         int chanWidth;
         int chanHeight;
         int hdrWidth;
         int hdrHeight;
         float lumThreshold;
         float lumScaler;
         BUFFER_PYRAMID starGenLevel;
         float blendAmount;
         float gamma;
         int numStreaks;
         float streakScale;
         float streakRotationOffset;
         float streakDec;
         float streakStride;
         float timeScale; //0.0 for instant
         bool adaptiveLuminance;
         bool exposureCorrelation;
         bool enableBloom;
         bool enableStreaks;
         bool enableLensFlare;
         float adaptiveLumMaxDiff;
         float adaptiveLumMinDiff;
         osg::Vec4f ghostScalar1;
         osg::Vec4f ghostScalar2;
         osg::Vec4f cameraMixCoeff;
         osg::Vec4f gausCompCoeff;
         osg::Vec4f blurScalar;
         float adaptiveLumPreScale;
         float adaptiveLumPostScale;
         float luminanceMaxValue;
         float luminanceMinValue;
         float saturationScale;
         bool multiview;
      };

   public:
      DtHdrDrawable(DtDe &de, DtHdrOptions options, osg::Texture *lensFlareTexture);

      DtHdrDrawable(const DtHdrDrawable& orig, const osg::CopyOp& op);

      virtual ~DtHdrDrawable();
   private:
      DtHdrDrawable();

      DtHdrDrawable(const DtHdrDrawable&);

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

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

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

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

      virtual void setEnabled( bool enabled );
      virtual bool enabled();

      virtual DtHdrOptions& options();

      virtual void setLumOverride(bool lumOverride);

      // should just pass through without applying any effects
      // used by SensorFX to turn off HDR on a per channel basis.
      virtual void setHdrPassthrough(bool enable);
      virtual bool hdrPassthrough();

      virtual void cleanupGpuMemory();

      virtual float localHdrOutputScale();

   protected:      
      virtual void initRenderTextures() const;

      virtual void downSample(DtGlTexture* src, DtGlTexture* dest, osg::State *state) const;

      virtual void downsample4x(DtGlTexture* src, DtGlTexture* dest, osg::State *state) const;

      virtual void extractHL(DtGlTexture* src, DtGlTexture* dest, osg::State *state) const;

      virtual void runPass(HDR_PASS prog, DtGlTexture* src, DtGlTexture* dest,
         osg::State *state) const;

      virtual void blur(DtGlTexture* src, DtGlTexture* dest, DtGlTexture *temp,
         int blurWidth, osg::State *state) const;

      virtual void getBufferPyramidSize(BUFFER_PYRAMID level, int* w, int* h) const;

      virtual void genStarStreak(int streak, float dir, float length) const;

      virtual void genGhostImage(const float* ghost_modulation1st,const float* ghost_modulation2nd, osg::State *state) const;

      virtual void composeEffect(void) const;

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

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

      virtual float adaptiveLuminance(osg::State* state) const;

      virtual bool adaptiveLuminanceCompute(osg::State* state) const;

      virtual float adaptiveLuminanceRead(osg::State* state) const;

      virtual float regularLuminance(osg::State* state) const;

      virtual float silverLiningLuminance(void) const;

      virtual void initShaders();

      virtual void computeColorModulationRedShift(float* color, float r, float g, float b, int num);
      virtual void computeColorModulation(float* color, float r, float g, float b, int num);
      virtual void computeColorModulationCoefficients();

      virtual void createComputeProgramIfNecessary(HDR_PASS pass, const std::string& name) const;
      virtual void createBlurProgramsIfNecessary() const;

      virtual unsigned int inputTextureId(unsigned int& target, const DtGlTexture* src, osg::State* state) const;

      virtual void streakPass0(int streak, float dec, const float step[2], float stride) const;
      virtual void streakPass1(int streak, float dec, const float step[2], float stride) const;
      virtual void streakPass2(int streak, float dec, const float step[2], float stride) const;

      virtual void ghostPass0(const float* ghost_modulation1st) const;
      virtual void ghostPass1(const float* ghost_modulation2nd) const;

      virtual void gaussianBlurComposePass(void) const;
      virtual void starStreakCompose(void) const;
      virtual void finalGlarePass(void) const;

      virtual void initializeUniformLocations(HDR_PASS pass) const;
      virtual void zeroOutUniformLocations();

      virtual void createToneMappingProgramIfNecessary(void) const;
      
   protected:
      bool myEnabled;

      osg::ref_ptr<osg::Texture> myInputTex;

      static double theLastHdrFrameTime;

      float myAspectRatio;

      mutable unsigned int myMutableShaders[PROGSIZE];

      osg::ref_ptr<osg::Texture> myLensMask;

      DtHdrOptions myOptions;

      bool myLumOverride;
      bool myEnableHdrPassthrough;

      float myStarModulation1st[16];
      float myStarModulation2nd[16];
      float myStarModulation3rd[16];

      float myFilmicGhostModulation1st[16];
      float myFilmicGhostModulation2nd[16];
      float myCameraGhostModulation1st[16];
      float myCameraGhostModulation2nd[16];

      float myHoriModulation1st[16];
      float myHoriModulation2nd[16];
      float myHoriModulation3rd[16];


      mutable DtGlTexture* myMutableComposebuffer[LEVEL_TOTAL];
      mutable DtGlTexture* myMutableBlurbufferA[LEVEL_TOTAL];
      mutable DtGlTexture* myMutableBlurbufferB[LEVEL_TOTAL];

      mutable DtGlTexture* myMutableStreakbufferA[8];
      mutable DtGlTexture* myMutableStreakbufferB[8];
      mutable DtGlTexture* myMutableStreakbufferFinal;

      mutable DtGlTexture* myMutableGhost1stBuffer;
      mutable DtGlTexture* myMutableGhost2ndBuffer;

      mutable DtGlTexture*  myMutableGlareBuffer;

      mutable DtGlTexture* myMutableLumcurrent;

      mutable unsigned int myMutableLumNext;
      mutable const float* myMutableLumNextPtr;

      mutable DtGlTexture* myMutableInitialBuffer;

      mutable double myMutableLastAdpativeLumFrameTime;

      mutable int myMutableExtractHlTresholdLoc;
      mutable int myMutableExtractHlScalarLoc;

      mutable int myMutableStarStreakStepSizeLoc;
      mutable int myMutableStarStreakStrideLoc;
      mutable int myMutableStarStreakColorCoeffLoc;

      mutable int myMutableGhostImageScalarLoc;
      mutable int myMutableGhostImageColorCoeffLoc;

      mutable int myMutableGaussianBlurComposeCoeffLoc;
      mutable int myMutableStarStreakComposeNumSamplersLoc;
      mutable int myMutableFinalGlarecomposeMixCoeffLoc;

      mutable int myMutableElapsedTimeLoc;

      mutable int myBlurAmountLoc;
      mutable int myGammaLoc;
      mutable int mySceneTexLoc;
      mutable int myBlurTexLoc;
      mutable int myLuminanceScaleLoc;
      mutable int myBlurTexScaleLoc;
      mutable int mySaturationScaleLoc;
      mutable int myEnablePassthroughLoc;

      mutable float myMutableLastLuminanceValue;
   };
}