DtReadFramebufferCallback.h

Go to the documentation of this file.

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


#pragma once
#include <vrvVideoStream/vrvVideoStream.h>
#include <vrvVideoStream/DtVideoStream.h>
#include <vrvVideoStream/DtProcessImageTask.h>
#include <vrvVideoStream/DtReadFramebufferProcessData.h>

#include <osg/Camera>
#include <vector>
#include <list>

namespace makVrv
{
   class DtProcessImageTask;
   class DtOsgChannel;
   class DtPostProcessManager;
   class DtVideoStreamPostProcessor;

   class DtEncoderIO;
   class DtEncoderIOCallback;

   class DT_DLL_VRVVIDEOSTREAM DtReadFramebufferCallback 
      : public osg::Camera::DrawCallback
   {
   public:
      friend class DtVideoStream;

      virtual ~DtReadFramebufferCallback();

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

      virtual void process(const std::vector<std::vector<uint8_t>>& encodedPackets);
   protected:

      DtReadFramebufferCallback( DtDe& de, DtVideoStream& stream, DtOsgChannel* channel );

      virtual bool init();

      virtual void unInit();

      virtual void setupPixelBufferObjects( int numBuffers );

      virtual bool createProcessImageTask( DtDe& de, const DtVideoStream& stream );

      virtual bool setPixelFormatFromString( const DtUnicode& pxfString );

      virtual bool setReadBufferFromString( const DtUnicode& rbString );

      virtual bool setDataTypeFromString( const DtUnicode& dtString );

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

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

      virtual DtVideoStreamPostProcessor* findVideoStreamPostProcessor(void) const;

      virtual void registerPbosWithHardwareEncoderIfNecessary(void);
   private:
      DtReadFramebufferCallback() = delete;

      DtReadFramebufferCallback( const DtReadFramebufferCallback& rhs ) = delete;

      DtReadFramebufferCallback& operator=( const DtReadFramebufferCallback& orig ) = delete;

   protected:
      DtVideoStream& myStream;

      DtDe& myDe;

      DtOsgChannel* myChannel;

      double myFrequency;

      bool myUseReadPixels;

      int myWidth;

      int myHeight;

      int myX;

      int myY;

      GLenum myPixelFormat;

      GLenum myDataType;

      GLenum myReadBuffer;

      std::vector<unsigned int> myPbos;

      mutable int myMutableWriteIndex;

      mutable int myMutableReadIndex;

      osg::ref_ptr<osg::Image> myImage;

      DtProcessImageTask* myTask;

      bool myIsInitialized;

      bool myIsUsingHardware;
      
      bool myReadFromTexture;

      DtEncoderIO* myEncoderIO;

      DtReadFramebufferProcessData myProcessData;

      DtPostProcessManager& myPostProcessManager;

      int myId = 0;

      static int theRunningId;

      DtEncoderIOCallback* myEncoderCallback = nullptr;
      
      double myTimeNextFrame = 0.0;
   };
}