DtLightComputeProgram.h

Go to the documentation of this file.

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


#pragma once

#include <vrvOsg/vrvOsg.h>

#include <vrvMath/DtVector2.h>
#include <vrvMath/DtVector3.h>
#include <vrvMath/DtVector4.h>
#include <vrvMath/DtMatrix4.h>

#include <vrvUtil/DtSignalConnectionManager.h>

#include <atomic>
#include <array>

namespace makVrv
{
   class DtDe;
   class DtGlBuffer;
   class DtLight;
   class DtTileSpaceLightGrid;
   class DtGlProgram;
   class DtGlShader;

   #define CPU_SIDE_COMPILATION 1
   #include <../data/shaders/lights/DtLightIndexData.h>
   #include <../data/shaders/lights/DtLightIntersectionData.h>
   #include <../data/shaders/lights/DtLightIndexBufferInfo.h>
   #undef CPU_SIDE_COMPILATION

   class DtLightComputeProgramShaderConstants;

   class DT_DLL_VRVOSG DtLightComputeProgram
   {
   public:
      DtLightComputeProgram(DtDe& de);

      virtual ~DtLightComputeProgram();
   private:
      DtLightComputeProgram() = delete;
      DtLightComputeProgram(const DtLightComputeProgram&) = delete;
      DtLightComputeProgram& operator=(const DtLightComputeProgram&) = delete;

   public:
      virtual void slot_reloadShader(void);

      virtual bool failedCompilation(void) const;

      virtual void execute(const DtTileSpaceLightGrid* tileSpaceLightGrid
         , const int tileLightIndexTotalCount
         , const size_t lightCount
         , const int numChunks
         , const Matrix4_64& projectionMatrix
         , const Vector3_64& vWorldEye
         , const Vector3_64& vWorldCenter
         , const Vector3_64& vWorldUp
         , const Vector2_uint32& viewportSize
         , const int viewOffset
         , const unsigned int sensorMask);

   protected:
      virtual void initializeShadersIfNecessary(void);

      virtual void deleteShaders(void);
   protected:
      static const int theNumLightSourceComputeShaders = 2;

      std::array<DtGlProgram*, theNumLightSourceComputeShaders> myComputePrograms;

      std::array<DtLightComputeProgramShaderConstants*, theNumLightSourceComputeShaders> myLightSourceComputeShaderConstants;

      std::array<DtGlBuffer*, theNumLightSourceComputeShaders> myGpuShaderUbo;
      std::array<int, theNumLightSourceComputeShaders> myGpuShaderUboBinding;

      static const std::array<std::string, theNumLightSourceComputeShaders> theLightSourcesComputeShaderFilePaths;
      static const std::string theLightingFunctionsShaderFilePath;

      bool myShadersFailedToCompile;

      DtDe& myDe;

      DtSignalConnectionManager myConnections;

      // We cannot delete shaders from the cull thread, so setting this true
      // signals the main thread to do that.
      std::atomic<bool> myDeleteShaders;

      int myWorkGroupLocalSizeX;
   };

}