add h264 encoder and rtsp server

houxiao

2016-12-23 1abced42eb3997ef9ef675bfe990f7913ea73f2f

 RtspFace/PL_H264Encoder.cpp

@@ -2,24 +2,33 @@


extern "C"

{

   #include <libyuv.h>

   #include <libavcodec/avcodec.h>

   #include <libavutil/frame.h>

   #include <libavformat/avformat.h>

   #include "libavutil/imgutils.h"

}



struct PL_H264Encoder_Internal

struct H264Encoder_Internal

{

   uint8_t buffer[1920*1080*4];

   uint8_t buffer[1920*1080*3];

   size_t buffSize;

   size_t buffSizeMax;



   bool payError;

   bool ffmpegInited;

   size_t frameCount;



   AVCodecContext* pAVCodecContext;

   AVFrame* pAVFrame;//#todo delete

   

   PL_H264Encoder_Internal() : 
   H264Encoder_Internal() : 
      buffSize(0), buffSizeMax(sizeof(buffer)), 

      payError(true)

      payError(true), ffmpegInited(false), frameCount(0), 
      pAVCodecContext(nullptr), pAVFrame(nullptr)

		
   {

   }

   

   ~PL_H264Encoder_Internal()

   ~H264Encoder_Internal()

   {

   }

   

@@ -27,6 +36,11 @@
   {

      buffSize = 0;

      payError = true;

      ffmpegInited = false;

      frameCount = 0;

		
      pAVCodecContext = nullptr;

      pAVFrame = nullptr;

   }

};



@@ -35,52 +49,180 @@
   return new PL_H264Encoder;

}



PL_H264Encoder::PL_H264Encoder() : internal(new PL_H264Encoder_Internal)

PL_H264Encoder::PL_H264Encoder() : internal(new H264Encoder_Internal)

{

}



PL_H264Encoder::~PL_H264Encoder()

{

   delete (PL_H264Encoder_Internal*)internal;

   delete (H264Encoder_Internal*)internal;

   internal= nullptr;

}



bool PL_H264Encoder::init(void* args)

{

   PL_H264Encoder_Internal* in = (PL_H264Encoder_Internal*)internal;

   H264Encoder_Internal* in = (H264Encoder_Internal*)internal;

   in->reset();



	
   return true;

}



void PL_H264Encoder::finit()

{

   PL_H264Encoder_Internal* in = (PL_H264Encoder_Internal*)internal;

   H264Encoder_Internal* in = (H264Encoder_Internal*)internal;

   

}



bool initH264EncoderEnv(H264Encoder_Internal* in)

{

   av_register_all();



   // find the video encoder

   AVCodec* avCodec = avcodec_find_encoder(AV_CODEC_ID_H264);



   if (!avCodec)   
   {  
      printf("codec not found!\n");  
      return false;  
   }  


   in->pAVCodecContext = avcodec_alloc_context3(avCodec);



   in->pAVCodecContext->bit_rate = 3*1024*1024*8; // 3MB

    in->pAVCodecContext->width = 1920;

    in->pAVCodecContext->height = 1080;//#todo from config

    in->pAVCodecContext->time_base.num=1;

    in->pAVCodecContext->time_base.den=25;

    in->pAVCodecContext->gop_size = 20;

    in->pAVCodecContext->max_b_frames = 0;

    in->pAVCodecContext->pix_fmt = AV_PIX_FMT_YUV420P;



   if(avcodec_open2(in->pAVCodecContext, avCodec, NULL) >= 0)

   {

      in->pAVFrame = av_frame_alloc(); // Allocate video frame

		
      in->pAVFrame->format = in->pAVCodecContext->pix_fmt;  
      in->pAVFrame->width  = in->pAVCodecContext->width;  
      in->pAVFrame->height = in->pAVCodecContext->height;

		
      int ret = av_image_alloc(in->pAVFrame->data, in->pAVFrame->linesize, in->pAVCodecContext->width, in->pAVCodecContext->height, 
                     in->pAVCodecContext->pix_fmt, 16);  
      if (ret < 0)

      {  
         printf("av_image_alloc error\n");

         return false;

      } 
   }

   else

   {

      printf("avcodec_open2 error\n");

      return false;

   }

	
   return true;

}



void copyAVFrame(AVFrame* dest, AVFrame* src)

{

   int height = dest->height;

   int width = dest->width;

	
   memcpy(dest->data[0], src->data[0], height * width); // Y

   memcpy(dest->data[1], src->data[1], height * width / 4); // U

   memcpy(dest->data[2], src->data[2], height * width / 4); // V

}



bool encodeH264(H264Encoder_Internal* in, AVFrame* pAVFrame, size_t buffSize)  
{

   in->buffSize = 0;

   in->frameCount++;



   copyAVFrame(in->pAVFrame, pAVFrame);

   in->pAVFrame->pts = in->frameCount;

	
   AVPacket pAVPacket = {0};

   av_init_packet(&pAVPacket);

	
   // encode the image

   int gotPacket = 0;

   int ret = avcodec_encode_video2(in->pAVCodecContext, &pAVPacket, in->pAVFrame, &gotPacket);  
   if (ret < 0)

   {

      printf("avcodec_encode_video2 (1) error=%d\n", ret);

      return false;

   }

	
   if (gotPacket > 0)

   {

      printf("Succeed to encode (1) frame=%d, size=%d\n", in->pAVFrame->pts, pAVPacket.size);

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

      in->buffSize += pAVPacket.size;

      av_free_packet(&pAVPacket);

   }

	
   //#todo finit

   //Flush Encoder  
   //while (gotPacket > 0)

   //{  
   //   ret = avcodec_encode_video2(in->pAVCodecContext, &pAVPacket, NULL, &gotPacket);

   //   if (ret < 0)

   //   {

   //      printf("avcodec_encode_video2 (2) error=%d\n", ret);

   //      return false;

   //   }  
   //   if (gotPacket > 0)

   //   {  
   //      printf("Succeed to encode (2) frame=%d, size=%d\n", in->pAVFrame->pts, pAVPacket.size);

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

   //      in->buffSize += pAVPacket.size; 
   //      av_free_packet(&pAVPacket);  
   //   }  
   //} 
	
   //#test

   if (in->buffSize > 0)

   {

      static FILE * pFile = fopen("out.h264","wba+");

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

      fflush(pFile);

   }

	
   in->payError = (in->buffSize == 0);

   return !(in->payError);

}



bool PL_H264Encoder::pay(const PipeMaterial& pm)

{

   PL_H264Encoder_Internal* in = (PL_H264Encoder_Internal*)internal;

   H264Encoder_Internal* in = (H264Encoder_Internal*)internal;

   

   //in->buffer readly



   //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);



   return true;

   in->payError = true;

	
   if (!in->ffmpegInited)

   {

      bool ret = initH264EncoderEnv(in);

      if (!ret)

      {

         printf("initH264EncoderEnv error");

         return false;

      }

      else

         in->ffmpegInited = true;

   }

	
   bool ret = encodeH264(in, (AVFrame*)pm.buffer, pm.buffSize);

   in->payError = !ret;

   return ret;

}



bool PL_H264Encoder::gain(PipeMaterial& pm)

{

   PL_H264Encoder_Internal* in = (PL_H264Encoder_Internal*)internal;

   H264Encoder_Internal* in = (H264Encoder_Internal*)internal;



   pm.buffer = in->buffer;

   pm.buffSize = in->buffSize;

   if (!in->payError)

   {

      pm.buffer = in->buffer;

      pm.buffSize = in->buffSize;

   }

   pm.former = this;

   return true;

   return !in->payError;

}