pans

2016-12-30 eb68a1e3ad5fb24b4c674b8f2fd529b8302c7d97

 RtspFace/PL_H264Decoder.cpp

@@ -1,5 +1,6 @@
#include "PL_H264Decoder.h"

#include "MaterialBuffer.h"

#include "logger.h"



#include <H264VideoRTPSource.hh> // for SPropRecord

#include <libbase64.h>

@@ -121,7 +122,7 @@


   if (!avCodec)   

   {  

      printf("codec not found!\n");  
      LOG(WARN) << "codec not found!";  
      return false;  

   }  



@@ -162,16 +163,18 @@
   return true;

}



bool decodeH264(H264Decoder_Internal* in, uint8_t* buffer, size_t buffSize)  
bool decodeH264(H264Decoder_Internal* in, uint8_t* buffer, size_t buffSize, timeval pts)  
{

   AVPacket packet = {0};

   int gotPicture = buffSize; // frameFinished



   if (av_packet_from_data(&packet, buffer, buffSize) != 0)

   {  

      printf("av_packet_from_data error\n");

      LOG(WARN) << "av_packet_from_data error";

      return false;

   }  
   }

	
   packet.pts = packet.dts = (pts.tv_sec * 1000 * 1000 + pts.tv_usec) / 90000;

   

   // decode

   avcodec_decode_video2(in->pAVCodecContext, in->pAVFrame, &gotPicture, &packet);

@@ -182,7 +185,7 @@
   }

   else

   {

      printf("incomplete frame\n");

      LOG(WARN) << "incomplete frame";

      return false;

   }

}

@@ -212,27 +215,41 @@


      bool ret = initH264DecoderEnv(in, sps.sPropBytes, sps.sPropLength, pps.sPropBytes, pps.sPropLength);

      if (!ret)

      {

         LOG(ERROR) << "PL_H264Decoder::pay initH264DecoderEnv error";

         return false; // #todo log

      }

      else

         in->fmtp_set_to_context = true;

   }

   

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

   if (pm.buffer == nullptr)

      return false;

   

   bool ret = false;

   if (pm.type == PipeMaterial::PMT_BYTES)

      ret = decodeH264(in, pm.buffer, pm.buffSize);

   {

      if (pm.buffSize <= 0)

         return false;



      timeval pts = {0};

      ret = decodeH264(in, (uint8_t*)pm.buffer, pm.buffSize, pts);

   }

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

   {

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

      

      ret = decodeH264(in, frame->buffer, frame->buffSize);

      if (frame->buffSize <= 0)

         return false;

		
      ret = decodeH264(in, (uint8_t*)frame->buffer, frame->buffSize, frame->pts);

      if (ret)

      {

         in->lastFrame.type = MB_Frame::MBFT_PTR_AVFRAME;

         in->lastFrame.buffer = (uint8_t*)(in->pAVFrame);

         in->lastFrame.buffSize = sizeof(in->pAVFrame);

         in->lastFrame.width = in->pAVFrame->width;

         in->lastFrame.height = in->pAVFrame->height;

         //in->lastFrame.pts = frame->pts;//#todo

         gettimeofday(&(in->lastFrame.pts),NULL);

      }

@@ -249,8 +266,8 @@
   if (!in->payError)

   {

      pm.type = PipeMaterial::PMT_FRAME;

      pm.buffer = (uint8_t*)(&(in->lastFrame));

      pm.buffSize = sizeof(uint8_t*);

      pm.buffer = &(in->lastFrame);

      pm.buffSize = 0;

   }

   pm.former = this;

   return !in->payError;