merge home base

houxiao

2017-08-17 0250edfd9c4d453f25a2f9827698ccf87ff5afff

 RtspFace/PL_RTSPServer.cpp

@@ -1,28 +1,167 @@
#include "PL_RTSPServer.h"

#include "MaterialBuffer.h"

#include "logger.h"



struct PL_RTSPServer_Internal

#include <liveMedia/liveMedia.hh>

#include <BasicUsageEnvironment/BasicUsageEnvironment.hh>



#include "FFmpegRTSPServer/IEncoder.h"

#include "FFmpegRTSPServer/LiveRTSPServer.h"

#include "FFmpegRTSPServer/FFmpegH264Source.h"

#include "FFmpegRTSPServer/LiveServerMediaSubsession.h"



class MyEncoderStub;



struct RTSPServer_Internal

{

   uint8_t buffer[1920*1080*4];

   uint8_t buffer[1920*1080*3];

   size_t buffSize;

   size_t buffSizeMax;

	
   RTSPServerConfig config;



   bool payError;

   pthread_t live_daemon_thid;

   pthread_mutex_t* frame_mutex;

   bool live_daemon_running;

   

   PL_RTSPServer_Internal() : 
      buffSize(0), buffSizeMax(sizeof(buffer)), 
      payError(true)

   MESAI::LiveRTSPServer * server;

   MyEncoderStub * encoderStub;



   RTSPServer_Internal() : 
      buffSize(0), buffSizeMax(sizeof(buffer)), config(), 
      payError(true), live_daemon_thid(0), frame_mutex(new pthread_mutex_t), live_daemon_running(false), 
      server(nullptr), encoderStub(nullptr)

   {

      pthread_mutex_init(frame_mutex, NULL);

   }

   

   ~PL_RTSPServer_Internal()

   ~RTSPServer_Internal()

   {

      if (frame_mutex != nullptr)

      {

         pthread_mutex_destroy(frame_mutex);

         delete frame_mutex;

         frame_mutex = nullptr;

      }

   }

   

   void reset()

   {

      buffSize = 0;

		
      RTSPServerConfig _config;

      config =_config;



      payError = true;



      if (frame_mutex != nullptr)

      {

         pthread_mutex_destroy(frame_mutex);

         delete frame_mutex;

         frame_mutex = nullptr;

      }

		
      frame_mutex = new pthread_mutex_t;

      pthread_mutex_init(frame_mutex, NULL);



      live_daemon_thid = 0;

      live_daemon_running = false;

		
      server = nullptr;

      encoderStub = nullptr;

   }

};



class MyEncoderStub : public MESAI::IEncoder

{

public:

   MyEncoderStub(RTSPServer_Internal& _in) : in(_in)

   {

   }

	
   virtual ~MyEncoderStub()

   {

   }

	
   virtual void setCallbackFunctionFrameIsReady(std::function<void()> func)

   {

      onFrame = func;

   }

	
   virtual char GetFrame(u_int8_t** FrameBuffer, unsigned int *FrameSize)

   {

      if (in.buffer == nullptr || in.buffSize <= 0)

      {

         ReleaseFrame();

         return 0;

      }



      //static size_t f = 0;

      //static FILE *pFile = fopen("/data/bb.264", "wb");

      //fwrite(in.buffer, sizeof(char), in.buffSize, pFile);

      //if (++f > 400){

      //   fclose(pFile);

      //   exit(0);

      //}



      uint8_t* pBuffer = in.buffer;

      size_t newBufferSize = in.buffSize;



      if (in.config.payWithAux)

      {

         if (newBufferSize <= 4)

         {

            ReleaseFrame();

            return 0;

         }

         pBuffer += 4;

         newBufferSize -= 4;

      }



      *FrameBuffer = pBuffer;

      *FrameSize = newBufferSize;



      LOG_DEBUG << "send frame size=" << in.buffSize << std::endl;

      return 1;

   }

	
   virtual char ReleaseFrame()

   {

      in.buffSize = 0;

		
      if (in.config.syncDeliverFrame)

      {

         int ret = pthread_mutex_unlock(in.frame_mutex);

         if(ret != 0)

         {

            LOG_WARN << "pthread_mutex_unlock frame_mutex: " << strerror(ret) << std::endl;

            return 0;

         }

      }

		
      return 1;

   }



   void deliverFrame()

   {

      // write frame buffer of RTSPServer_Internal::buffer

      onFrame();



      if (in.config.syncDeliverFrame)

      {

         int ret = pthread_mutex_lock(in.frame_mutex);

         if(ret != 0)

         {

            LOG_WARN << "pthread_mutex_lock frame_mutex: " << strerror(ret) << std::endl;

            return;

         }

      }

   }



private:

   RTSPServer_Internal& in;

   std::function<void()> onFrame;

};



PipeLineElem* create_PL_RTSPServer()

@@ -30,52 +169,94 @@
   return new PL_RTSPServer;

}



PL_RTSPServer::PL_RTSPServer() : internal(new PL_RTSPServer_Internal)

PL_RTSPServer::PL_RTSPServer() : internal(new RTSPServer_Internal)

{

}



PL_RTSPServer::~PL_RTSPServer()

{

   delete (PL_RTSPServer_Internal*)internal;

   internal= nullptr;

   delete (RTSPServer_Internal*)internal;

   internal = nullptr;

}



static void* live_daemon_thd(void* arg)

{

   RTSPServer_Internal* in = (RTSPServer_Internal*)arg;



   MyEncoderStub encoder(*in);

   in->encoderStub = &encoder;

   in->server = new MESAI::LiveRTSPServer(&encoder, 8554, 8080);

	
   in->live_daemon_running = true;

   in->server->run(); // does not return

   in->encoderStub = nullptr;

   in->live_daemon_running = false;

}



bool PL_RTSPServer::init(void* args)

{

   PL_RTSPServer_Internal* in = (PL_RTSPServer_Internal*)internal;

   RTSPServer_Internal* in = (RTSPServer_Internal*)internal;

   in->reset();



   if (args)

   {

      RTSPServerConfig* config = (RTSPServerConfig*)args;

      in->config = *config;

   }



   int ret = pthread_create(&(in->live_daemon_thid), NULL, live_daemon_thd, in);

   if(ret != 0)

   {

      LOG_ERROR << "pthread_create: " << strerror(ret) << std::endl;

      return false;

   }



   return true;

}



void PL_RTSPServer::finit()

{

   PL_RTSPServer_Internal* in = (PL_RTSPServer_Internal*)internal;

	
   RTSPServer_Internal* in = (RTSPServer_Internal*)internal;



   pthread_join(in->live_daemon_thid, NULL);

}



bool PL_RTSPServer::pay(const PipeMaterial& pm)

{

   PL_RTSPServer_Internal* in = (PL_RTSPServer_Internal*)internal;

   RTSPServer_Internal* in = (RTSPServer_Internal*)internal;



   if (pm.buffer == nullptr)

      return false;

   

   //in->buffer readly

   if (pm.type != PipeMaterial::PMT_FRAME)

   {

      LOG_ERROR << "PL_RTSPServer::pay only support PMT_FRAME" << std::endl;

      return false;

   }

	
   if (in->buffSize > 0)

      LOG_WARN << "PL_RTSPServer::pay may lost data size=" << in->buffSize << std::endl;

	
   MB_Frame* frame = (MB_Frame*)pm.buffer;

   if (frame->buffer == nullptr)

      return false;

   memcpy(in->buffer, frame->buffer, frame->buffSize);

   in->buffSize = frame->buffSize;



   //static size_t f=0;

   //char fname[50];

   //sprintf(fname, "%u.bgra", ++f);

   //FILE * pFile = fopen (fname,"wb");

   //fwrite (in->buffer , sizeof(char), in->buffSize, pFile);

   //fclose(pFile);



   if (in->encoderStub == nullptr)

      return false;

	
   in->encoderStub->deliverFrame();

   return true;

}



bool PL_RTSPServer::gain(PipeMaterial& pm)

{

   PL_RTSPServer_Internal* in = (PL_RTSPServer_Internal*)internal;

   RTSPServer_Internal* in = (RTSPServer_Internal*)internal;



   pm.buffer = in->buffer;

   pm.buffSize = in->buffSize;

   pm.type = PipeMaterial::PMT_NONE;

   pm.buffer = nullptr;

   pm.buffSize = 0;

   pm.former = this;

   return true;

}