ffmpeg-인코딩 h264

H264

코드는 다음과 같습니다.

#define   _WIDTH   352   
#define    _HEIGHT 288   
void CEncodeYUVDlg::OnButEncode()   
{  
 // TODO: Add your control notification handler code here   
  
 uint8_t *video_outbuf;  
 int video_outbuf_size, out_size;  
    /* alloc image and output buffer */  
    video_outbuf_size = 10000000;  
    video_outbuf = (unsigned char *)malloc(video_outbuf_size);  
  
    av_register_all();  
 avcodec_init();  
 unsigned char  sYUV_data[_WIDTH*_HEIGHT*3/2];  
 AVCodecContext *pContext = avcodec_alloc_context();  
 /* put sample parameters */  
    pContext->codec_id = CODEC_ID_H264;  //       ， ：CODEC_ID_MPEG1VIDEO，      ， ：400K  ，10   ，    。 
 pContext->codec_type = CODEC_TYPE_VIDEO;   
 pContext->bit_rate = 400000;   
 pContext->width = _WIDTH;   
 pContext->height = _HEIGHT;   
 pContext->time_base.den = 25;   
 pContext->time_base.num = 1;   
 pContext->gop_size = 25;   
    pContext->pix_fmt = PIX_FMT_YUV420P;  
 pContext->dct_algo = 0;   
 pContext->me_pre_cmp=2;   
 pContext->cqp = 26;   
 pContext->me_method =7;   
 pContext->qmin = 3;   
 pContext->qmax = 31;   
 pContext->max_qdiff = 3;   
 pContext->qcompress = 0.5;   
 pContext->qblur = 0.5;   
 pContext->nsse_weight = 8;   
 pContext->i_quant_factor = (float)0.8;   
 pContext->b_quant_factor = 1.25;   
 pContext->b_quant_offset = 1.25;  
 /* find the mpeg1 video encoder */  
    AVCodec* codec = avcodec_find_encoder(pContext->codec_id);  
    avcodec_open(pContext, codec);  
      
    AVFrame* g_pInFrame = avcodec_alloc_frame();  
  
 int isize = pContext->width * pContext->height;  
 g_pInFrame->data[0] = sYUV_data;  
    g_pInFrame->data[1] = g_pInFrame->data[0] + isize;  
 g_pInFrame->data[2] = g_pInFrame->data[1] + isize / 4;  
 g_pInFrame->linesize[0] = pContext->width;  
 g_pInFrame->linesize[1] = pContext->width / 2;  
 g_pInFrame->linesize[2] = pContext->width / 2;  
  
 FILE* fp = fopen("01.mov","rb");  
 int nread = -1;  
 if (fp!=NULL)  
 {  
  while(nread!=0)  
  {  
        nread = fread(sYUV_data,1,_WIDTH*_HEIGHT*3/2,fp);  
     out_size = avcodec_encode_video(pContext, video_outbuf, video_outbuf_size, g_pInFrame);   
     if (out_size>0)  
     {  
      FILE* ff = fopen("1.mov","ab");  
      if (ff!=NULL)  
      {  
       fwrite(video_outbuf,1,out_size,ff);  
       fclose(ff);  
      }  
     }  
  }  
  fclose(fp);  
  AfxMessageBox("Encode Over!");  
 }  
}  

 

H264 - YUV

H264 코드 흐름이 있는데, 어떻게 YUV 데이터로 변하는지.ffmpeg 디코딩
코드는 다음과 같습니다.

avcodec_init();   
av_register_all();  
AVCodec* m_pCodec=avcodec_find_decoder(CODEC_ID_H264);  
if(NULL==m_pCodec)  
return FALSE;  
AVCodecContext* m_pContext = avcodec_alloc_context();  
AVFrame* m_pFrame = avcodec_alloc_frame();  
if(NULL==m_pContext ||NULL==m_pFrame)  
return FALSE;  
if (avcodec_open(m_pContext, m_pCodec) < 0)  
{  
	return FALSE;  
}  
 int nWidth = 352;
 int nHeight = 288;
int uSize;
int iLen=avcodec_decode_video(m_pContext,m_pFrame,&uSize,pInBuffer,iBufferSize);
if((uSize>0) &&(iLen>0))
{
	LPBYTE PtrY = m_pFrame->data[0];			
	LPBYTE PtrU = m_pFrame->data[1];				
	LPBYTE PtrV = m_pFrame->data[2];

	int iSizeY = m_pFrame->linesize[0];	
	int iSizeU = m_pFrame->linesize[1];		
	int iSizeV = m_pFrame->linesize[2];

	FILE*fp = fopen("1.yuv","ab");
	if (fp!=NULL)
	{
		//copy y
		for (int i =0;i<nHeight;i++)
		{
			fwrite(PtrY,1,nWidth,fp);
			PtrY+=iSizeY;
		}
		//copy u
		for (int i =0;i<nHeight/2;i++)
		{
			fwrite(PtrU,1,nWidth/2,fp);
			PtrU+=iSizeU;
		}
		//copy v
		for (int i =0;i<nHeight/2;i++)
		{
			fwrite(PtrV,1,nWidth/2,fp);
			PtrV+=iSizeV;
		}
		////copy to file
		fclose(fp);
	}
}