DtSwapChain.h

Go to the documentation of this file.

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


#pragma once

#include "vrvVirtualReality/vrvVirtualReality.h"

#include <vrvGraphicsUtils/DtGlImageType.h>
#include <osg/StateSet>
#include <osg/Geometry>
#include <osg/Texture2D>

#include <vector>
#include <map>

namespace makVrv
{
   class DtDe;
   class DtGlProgram;

   enum DtSwapChainTextureType
   {
      COLOR,
      DEPTH,
      MASK,
      NONE
   };

   class DT_DLL_VRVVR DtSwapChain
   {
   private:
      DtSwapChain();

      DtSwapChain(const DtSwapChain& orig);

      DtSwapChain& operator=(const DtSwapChain&);

   public:
      DtSwapChain(DtDe& de, vrvVirtualReality::DtEye eye);

      virtual ~DtSwapChain();

      virtual bool initalise(unsigned int width, unsigned int height, unsigned int layers, DtSwapChainTextureType type, DtGlImageType format);

      virtual bool addFormat(DtSwapChainTextureType type, DtGlImageType format);

      virtual void acquireTextures() = 0;

      unsigned int getAcquiredIndex(DtSwapChainTextureType type) const;

      virtual unsigned int getTexture(DtSwapChainTextureType type, int index) const = 0;

      virtual void releaseTextures() = 0;

      bool isInitialized() const;

      unsigned int length() const;

      DtSwapChainTextureType type(unsigned int index) const;

      DtGlImageType format(unsigned int index) const;

      DtGlImageType format(DtSwapChainTextureType type) const;

      unsigned int width() const;

      unsigned int height() const;

      unsigned int layers() const;

      vrvVirtualReality::DtEye eye() const;

      GLuint getFbo(unsigned int index) const;

      virtual void copyTextures(osg::RenderInfo& renderInfo, unsigned int srcColorId, unsigned int srcDepthId, unsigned int srcStencilId, bool copySrcAlpha = false, bool srcIsArray = true, int mode = 0) const;

      virtual void blitTextureToScreen(int swidth, int sheight, bool showleftEye, int overrideIndex = -1);

      virtual void blendSwapChains(osg::RenderInfo& renderInfo, DtSwapChain* base, DtSwapChain* overlay) const;

   protected:

      virtual bool initaliseFbos() const;

      virtual bool initCopyProgram(bool copyDepth, bool copyStencil, bool copySrcAlpha, bool srcIsArray) const;

      virtual  bool initBlendProgram() const;

   protected:
    
      // !\brief have the textue swap chains been initalised
      bool myIsInitialized;

      // !\brief the eye this swapchain represents
      vrvVirtualReality::DtEye myEye;

      mutable unsigned int myLength;

      std::vector<DtSwapChainTextureType> myTextureTypes;

      std::vector<DtGlImageType> myFormats;

      mutable unsigned int myWidth;

      mutable unsigned int myHeight;

      mutable unsigned int myLayers;

      mutable std::map<DtSwapChainTextureType, unsigned int> mySwapChainIndexes;

      mutable std::vector<GLuint> myFbos;

      mutable DtGlProgram* myCopyProgram;

      mutable int myLastCopyProgramConfig;

      mutable osg::ref_ptr<osg::StateSet> myCopyStateSet;

      mutable unsigned int myCopyQuadVbo;

      mutable unsigned int myLastStencilTexId;

      mutable unsigned int mySrcStencilTexId;

      mutable DtGlProgram* myBlendProgram;

      DtDe& myDe;
   };

   class DtVirtualRealityDriver;

   class DT_DLL_VRVVR DtExternalTextureAttribute : public osg::Texture2D
   {
   public:
      DtExternalTextureAttribute()
         : osg::Texture2D()
         , myTextureId(0)
      {
      }

      virtual osg::Object* cloneType() const
      {
         return new DtExternalTextureAttribute();
      }
      virtual osg::Object* clone(const osg::CopyOp& copyop) const
      {
         return new DtExternalTextureAttribute();// (*this, copyop);
      }
      virtual int compare(const StateAttribute& sa) const
      {
         COMPARE_StateAttribute_Types(DtExternalTextureAttribute, sa)
            rhs;
         return 0;
      }

      virtual void apply(osg::State& state) const
      {
         if (myTextureId)
         {
            glBindTexture(GL_TEXTURE_2D, myTextureId);
         }
      }

      GLuint myTextureId;
   };

   class DT_DLL_VRVVR DtSwapChainPreviewDrawable : public osg::Geometry
   {
   public:
      DtSwapChainPreviewDrawable(DtDe& de, DtVirtualRealityDriver& driver);

      virtual ~DtSwapChainPreviewDrawable();

      virtual void drawImplementation(osg::RenderInfo& info) const;
      virtual void releaseGLObjects(osg::State* state/* =0 */) const;

      virtual void setSwapChain(DtSwapChain* swapchain);

      virtual void setOverlayTexture(osg::Texture2D* overlay);
      virtual void setSceneTextureOffset(osg::Vec2 uvOffset);

   protected:

      virtual void initGlObjects() const;

   protected:

      DtDe& myDe;
      DtVirtualRealityDriver& myDriver;

      bool myUseBlitMode;

      DtSwapChain* mySwapChain;
      mutable int myCurrentSwapChainIndex;

      mutable std::map<vrvVirtualReality::DtEye, osg::ref_ptr<DtExternalTextureAttribute>> myLayerTextures;
      osg::ref_ptr<osg::StateSet> myStateSet;

   };
}