add MaterialBuffer

houxiao

2016-12-28 017b7eb736ccc53c52f81486be8196d7fafc3289

 RtspFace/PL_AVFrameYUV420.cpp

@@ -1,4 +1,5 @@
#include "PL_AVFrameYUV420.h"

#include "MaterialBuffer.h"



extern "C"

{

@@ -12,9 +13,10 @@
   uint8_t buffer[1920*1080*3];

   size_t buffSize;

   size_t buffSizeMax;

   MB_Frame lastFrame;

   

   AVFrameYUV420_Internal() : 

      buffSize(0), buffSizeMax(sizeof(buffer))

      buffSize(0), buffSizeMax(sizeof(buffer)), lastFrame()

   {

   }

   

@@ -25,6 +27,9 @@
   void reset()

   {

      buffSize = 0;

		
      MB_Frame _lastFrame;

      lastFrame = _lastFrame;

   }

};



@@ -61,7 +66,23 @@
{

   AVFrameYUV420_Internal* in = (AVFrameYUV420_Internal*)internal;

   

   AVFrame* pAVFrame = (AVFrame*)pm.buffer;

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

   {

      printf("PL_H264Encoder::pay only support PMT_FRAME\n");

      return false;

   }

	
   if (pm.buffer == nullptr)

      return false;

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

   if (frame->type != MB_Frame::MBFT_PTR_AVFRAME)

   {

      printf("PL_H264Encoder::pay only support MBFT_PTR_AVFRAME\n");

      return false;

   }



   AVFrame* pAVFrame = (AVFrame*)frame->buffer;

   if (pAVFrame == nullptr)

      return false;

   

@@ -84,33 +105,20 @@
   

   in->buffSize = pBuff - in->buffer;

   

   // write yuv420

   //int a=0;

   //for (int i = 0; i < height; i++)

   //{

   //   memcpy(in->buffer + a, pAVFrame->data[0] + i * pAVFrame->linesize[0], width);

   //   a += width;

   //}

   //for (int i=0; i<height/2; i++)

   //{

   //   memcpy(in->buffer + a, pAVFrame->data[1] + i * pAVFrame->linesize[1], width / 2);

   //   a += width / 2;//#todo 4

   //}

   //for (int i=0; i<height/2; i++)

   //{

   //   memcpy(in->buffer + a, pAVFrame->data[2] + i * pAVFrame->linesize[2], width / 2);

   //   a += width / 2;

   //}



   //in->buffer readly



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

   in->lastFrame.buffer = in->buffer;

   in->lastFrame.buffSize = in->buffSize;

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



   //#test

   //static size_t f=0;

   //char fname[50];

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

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

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

   //fclose(pFile);

   static size_t f=0;

   char fname[50];

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

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

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

   fclose(pFile);



   return true;

}

@@ -119,8 +127,9 @@
{

   AVFrameYUV420_Internal* in = (AVFrameYUV420_Internal*)internal;



   pm.buffer = in->buffer;

   pm.buffSize = in->buffSize;

   pm.type = PipeMaterial::PMT_FRAME;

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

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

   pm.former = this;

   return true;

}