更新产品原型

houxiao

2017-02-23 77513d0db1242f30821f94ae97e272055e5ae0bc

 RtspFace/PL_H264Encoder.cpp

@@ -9,9 +9,9 @@
   #include <libavutil/imgutils.h>

   #include <libavutil/opt.h>

   #include <libavformat/avformat.h>

	
   #include <libyuv.h>

}



#include <libyuv.h>



PL_H264Encoder_Config::PL_H264Encoder_Config() : 

   inBufferSize(2*1024*1024), // 2MByte

@@ -47,6 +47,8 @@
   

   ~H264Encoder_Internal()

   {

      delete[] buffer;

      buffer = nullptr;

   }

   

   void reset()

@@ -67,8 +69,10 @@
      pAVFormatContext = nullptr;

      

      if (buffer != nullptr)

      {

         delete[] buffer;

      buffer = new uint8_t[config.inBufferSize];

         buffer = nullptr;

      }

   }

};



@@ -98,6 +102,8 @@
      in->config = *config;

   }

   

   in->buffer = new uint8_t[in->config.inBufferSize];

	
   return true;

}



@@ -116,7 +122,7 @@


   if (!avCodec)   

   {  

      LOG_ERROR << "codec not found!";  
      LOG_ERROR << "codec not found!" << std::endl;  
      return false;  

   }  



@@ -150,13 +156,13 @@
                     in->pAVCodecContext->pix_fmt, 16);  

      if (ret < 0)

      {  

         LOG_ERROR << "av_image_alloc error";

         LOG_ERROR << "av_image_alloc error" << std::endl;

         return false;

      } 

   }

   else

   {

      LOG_ERROR << "avcodec_open2 error";

      LOG_ERROR << "avcodec_open2 error" << std::endl;

      return false;

   }



@@ -227,7 +233,7 @@
   int ret = avcodec_encode_video2(in->pAVCodecContext, &pAVPacket, in->pAVFrame, &gotPacket);  

   if (ret < 0)

   {

      LOG_WARN << "avcodec_encode_video2 (1) error=" << ret;

      LOG_WARN << "avcodec_encode_video2 (1) error=" << ret << std::endl;

      return false;

   }

   

@@ -235,7 +241,7 @@
   {

      in->frameCount++;

      LOGP(DEBUG, "Succeed to encode (1) frame=%d, size=%d", in->frameCount, pAVPacket.size);

      memcpy(in->buffer, pAVPacket.data, pAVPacket.size);

      memcpy(in->buffer, pAVPacket.data, pAVPacket.size);//#todo check inBufferSize

      in->buffSize = pAVPacket.size;

      av_free_packet(&pAVPacket);

   }

@@ -321,7 +327,7 @@
      bool ret = initH264EncoderEnv(in);

      if (!ret)

      {

         LOG_ERROR << "initH264EncoderEnv error";

         LOG_ERROR << "initH264EncoderEnv error" << std::endl;

         return false;

      }

      else

@@ -330,7 +336,7 @@
   

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

   {

      LOG_ERROR << "Only support PMT_FRAME";

      LOG_ERROR << "Only support PMT_FRAME" << std::endl;

      return false;

   }

   

@@ -347,7 +353,7 @@
      ret = encodeH264(in, (uint8_t*)(frame->buffer), frame->pts);

   else

   {

      LOG_ERROR << "Only support MBFT_PTR_AVFRAME / MBFT_YUV420";

      LOG_ERROR << "Only support MBFT_PTR_AVFRAME / MBFT_YUV420" << std::endl;

      in->payError = true;

      return false;

   }

@@ -361,8 +367,11 @@
      in->lastFrame.buffSize = in->buffSize;

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

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

      in->lastFrame.pts = frame->pts;

      //#todo resetPts



      if (in->config.resetPTS)

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

      else

         in->lastFrame.pts = frame->pts;

   }

   

   return ret;