使用FFMpeg进行H264编码
使用FFMpeg可以很方便的对音视频进行编码,并且写文件。
下面的代码是将5幅1280*720大小的图片进行编码,并且写到文件中。
代码有些乱,但希望能抛砖引玉,对学习这方面的朋友有帮助。
- CFile file[5];
- BYTE *szTxt[5];
- int nWidth = 0;
- int nHeight= 0;
- int nDataLen=0;
- int nLen;
- CString csFileName;
- for (int fileI = 1; fileI <= 5; fileI ++)
- {
- csFileName.Format(“e:\\pics\\%d.bmp”, fileI);
- file[fileI – 1].Open(csFileName,CFile::modeRead | CFile::typeBinary);
- nLen = file[fileI – 1].GetLength();
- szTxt[fileI -1] = new BYTE[nLen];
- file[fileI – 1].Read(szTxt[fileI – 1], nLen);
- file[fileI – 1].Close();
- //BMP bmi;//BITMAPINFO bmi;
- //int nHeadLen = sizeof(BMP);
- BITMAPFILEHEADER bmpFHeader;
- BITMAPINFOHEADER bmpIHeader;
- memcpy(&bmpFHeader,szTxt[fileI -1],sizeof(BITMAPFILEHEADER));
- int nHeadLen = bmpFHeader.bfOffBits – sizeof(BITMAPFILEHEADER);
- memcpy(&bmpIHeader,szTxt[fileI – 1]+sizeof(BITMAPFILEHEADER),nHeadLen);
- nWidth = bmpIHeader.biWidth;// 464;// bmi.bmpInfo.bmiHeader.biWidth;// ;
- nHeight = bmpIHeader.biHeight;//362;// bmi.bmpInfo.bmiHeader.biHeight;// ;
- szTxt[fileI – 1] += bmpFHeader.bfOffBits;
- nDataLen = nLen-bmpFHeader.bfOffBits;
- }
- av_register_all();
- avcodec_register_all();
- AVFrame *m_pRGBFrame = new AVFrame[1]; //RGB帧数据
- AVFrame *m_pYUVFrame = new AVFrame[1];; //YUV帧数据
- AVCodecContext *c= NULL;
- AVCodecContext *in_c= NULL;
- AVCodec *pCodecH264; //编码器
- uint8_t * yuv_buff;//
- //查找h264编码器
- pCodecH264 = avcodec_find_encoder(CODEC_ID_H264);
- if(!pCodecH264)
- {
- fprintf(stderr, “h264 codec not found\n”);
- exit(1);
- }
- c= avcodec_alloc_context3(pCodecH264);
- c->bit_rate = 3000000;// put sample parameters
- c->width =nWidth;//
- c->height = nHeight;//
- // frames per second
- AVRational rate;
- rate.num = 1;
- rate.den = 25;
- c->time_base= rate;//(AVRational){1,25};
- c->gop_size = 10; // emit one intra frame every ten frames
- c->max_b_frames=1;
- c->thread_count = 1;
- c->pix_fmt = PIX_FMT_YUV420P;//PIX_FMT_RGB24;
- //av_opt_set(c->priv_data, /*”preset”*/”libvpx-1080p.ffpreset”, /*”slow”*/NULL, 0);
- //打开编码器
- if(avcodec_open2(c,pCodecH264,NULL)<0)
- TRACE(“不能打开编码库”);
- int size = c->width * c->height;
- yuv_buff = (uint8_t *) malloc((size * 3) / 2); // size for YUV 420
- //将rgb图像数据填充rgb帧
- uint8_t * rgb_buff = new uint8_t[nDataLen];
- //图象编码
- int outbuf_size=100000;
- uint8_t * outbuf= (uint8_t*)malloc(outbuf_size);
- int u_size = 0;
- FILE *f=NULL;
- char * filename = “e:\\pics\\myData.h264”;
- f = fopen(filename, “wb”);
- if (!f)
- {
- TRACE( “could not open %s\n”, filename);
- exit(1);
- }
- //初始化SwsContext
- SwsContext * scxt = sws_getContext(c->width,c->height,PIX_FMT_BGR24,c->width,c->height,PIX_FMT_YUV420P,SWS_POINT,NULL,NULL,NULL);
- AVPacket avpkt;
- //AVFrame *pTFrame=new AVFrame
- for (int i=0;i<250;++i)
- {
- //AVFrame *m_pYUVFrame = new AVFrame[1];
- int index = (i / 25) % 5;
- memcpy(rgb_buff,szTxt[index],nDataLen);
- avpicture_fill((AVPicture*)m_pRGBFrame, (uint8_t*)rgb_buff, PIX_FMT_RGB24, nWidth, nHeight);
- //将YUV buffer 填充YUV Frame
- avpicture_fill((AVPicture*)m_pYUVFrame, (uint8_t*)yuv_buff, PIX_FMT_YUV420P, nWidth, nHeight);
- // 翻转RGB图像
- m_pRGBFrame->data[0] += m_pRGBFrame->linesize[0] * (nHeight – 1);
- m_pRGBFrame->linesize[0] *= -1;
- m_pRGBFrame->data[1] += m_pRGBFrame->linesize[1] * (nHeight / 2 – 1);
- m_pRGBFrame->linesize[1] *= -1;
- m_pRGBFrame->data[2] += m_pRGBFrame->linesize[2] * (nHeight / 2 – 1);
- m_pRGBFrame->linesize[2] *= -1;
- //将RGB转化为YUV
- sws_scale(scxt,m_pRGBFrame->data,m_pRGBFrame->linesize,0,c->height,m_pYUVFrame->data,m_pYUVFrame->linesize);
- int got_packet_ptr = 0;
- av_init_packet(&avpkt);
- avpkt.data = outbuf;
- avpkt.size = outbuf_size;
- u_size = avcodec_encode_video2(c, &avpkt, m_pYUVFrame, &got_packet_ptr);
- if (u_size == 0)
- {
- fwrite(avpkt.data, 1, avpkt.size, f);
- }
- }
- fclose(f);
- delete []m_pRGBFrame;
- delete []m_pYUVFrame;
- delete []rgb_buff;
- free(outbuf);
- avcodec_close(c);
- av_free(c);