DtWebSocketTaskElement.h

Go to the documentation of this file.

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


#pragma once

#include <vrvWebSocket/vrvWebSocket.h>

#include <vrvVideoStream/DtImageTaskElement.h>

#include <string>

#include <libwebsockets.h>

namespace makVrv
{
   class DT_DLL_VRVWEBSOCKET DtWebSocketTaskElement 
      : public DtImageTaskElement
   {
   public: 
      friend class DtWebSocketTaskElementCreator;

      virtual ~DtWebSocketTaskElement();

      virtual void setFromRecord( const makArchives::DtImageTaskRecord& config );

      virtual bool configure();

      DtLocklessQueue<std::string>& queue() {return *myCommandQueue;}
      
      virtual const int port() const;

      virtual const std::string& htmlFileNameToServe() const;

      virtual void setTxPacketSize( unsigned int size );
      virtual unsigned int txPacketSize();
   protected:
      DtWebSocketTaskElement( DtDe& de, DtProcessImageTask* owningTask );

      void handlePreFrame();

      void handleProcessFrame(DtVideoFrameBuffer* frame);

      bool generateAndWriteHtml();

      void buildProtocols();
   protected:
      lws_context* myContext;
      lws_vhost* myVHost;
      struct lws_protocols myProtocols[3];
      DtLocklessQueue<std::string>* myCommandQueue;
      int myPort;
      unsigned int myTxPacketSize;
      std::string myHTMLFileNameToServe;
   };

   class DT_DLL_VRVWEBSOCKET DtWebSocketTaskElementCreator 
      : public DtImageTaskElementCreator
   {
   public:
      friend class DtVideoStreamManager;

      DtWebSocketTaskElementCreator();

      virtual ~DtWebSocketTaskElementCreator();

   protected:
      virtual DtImageTaskElement* create( DtDe& de, DtProcessImageTask* owningTask );
   };
}