1. 测试代码:
#include <libavcodec/avcodec.h>
#include <libavformat/avformat.h>
#include <libswscale/swscale.h>
#include <libswresample/swresample.h>
#include <stdio.h>
void SaveFrame(AVFrame *pFrame, int width, int height, int iFrame) {
FILE *pFile;
char szFilename[32];
int y;
printf("start sws_scale\n");
// Open file
sprintf(szFilename, "frame%d.ppm", iFrame);
pFile=fopen(szFilename, "wb");
if(pFile==NULL){
printf("pFile is null");
return;
}
fprintf(pFile, "P6\n%d %d\n255\n", width, height);
for(y=0; y<height; y++)
fwrite(pFrame->data[0]+y*pFrame->linesize[0], 1, width*3, pFile);
fclose(pFile);
}
int main (int argc, char **argv)
{
AVFormatContext *fmt_ctx = NULL;
AVDictionaryEntry *tag = NULL;
int ret;
if (argc != 2) {
printf("usage: %s <input_file>\n"
"example program to demonstrate the use of the libavformat metadata API.\n"
"\n", argv[0]);
return 1;
}
av_register_all();
if ((ret = avformat_open_input(&fmt_ctx, argv[1], NULL, NULL)))
return ret;
if(avformat_find_stream_info(fmt_ctx, NULL)<0)
return 1;
av_dump_format(fmt_ctx, 0, argv[1], 0);
int i;
int videoStream = -1;
for(i=0;i<fmt_ctx->nb_streams; i++) {
if(fmt_ctx->streams[i]->codec->codec_type==AVMEDIA_TYPE_VIDEO) {
videoStream=i;
break;
}
}
printf("videoStream: %d\n", videoStream);
if(videoStream == -1) return 1;
AVCodecContext *pCodecCtx;
pCodecCtx = fmt_ctx->streams[videoStream]->codec;
AVCodec *pCodec=avcodec_find_decoder(pCodecCtx->codec_id);
if(pCodec==NULL) {
fprintf(stderr, "Unsupported codec!\n");
return -1; // Codec not found
}
// Open codec
if(avcodec_open2(pCodecCtx, pCodec, NULL)<0) return -1; // Could not open codec
fprintf(stdout, "open stream, codec_id:%d\n", pCodecCtx->codec_id);
AVFrame *pFrame;
pFrame=av_frame_alloc();
//保存frame
AVFrame *pFrameRGB;
pFrameRGB=av_frame_alloc();
if(pFrameRGB==NULL) return -1;
uint8_t *buffer;
int numBytes;
numBytes=avpicture_get_size(PIX_FMT_RGB24, pCodecCtx->width, pCodecCtx->height);
buffer=(uint8_t *)av_malloc(numBytes*sizeof(uint8_t));
avpicture_fill((AVPicture *)pFrameRGB, buffer, PIX_FMT_RGB24, pCodecCtx->width, pCodecCtx->height);
int frameFinished;
AVPacket packet;
i=0;
while(av_read_frame(fmt_ctx, &packet)>=0){
if(packet.stream_index==videoStream) {
int result;
avcodec_decode_video2(pCodecCtx,pFrame,&frameFinished, &packet);
if(frameFinished) {
struct SwsContext *img_convert_ctx = sws_getContext(pCodecCtx->width, pCodecCtx->height, pCodecCtx->pix_fmt, pCodecCtx->width,
pCodecCtx->height, PIX_FMT_RGB24, SWS_BICUBIC,NULL, NULL,NULL);
result = sws_scale(img_convert_ctx, (const uint8_t* const*)pFrame->data, pFrame->linesize,
0, pCodecCtx->height, pFrameRGB->data, pFrameRGB->linesize);
printf("get result is %d~~~~~~~~~\n",result);
// Save the frame to disk
printf("i is %d \n",i);
if(++i<=5){
SaveFrame(pFrameRGB, pCodecCtx->width, pCodecCtx->height, i);
}
}
}
av_free_packet(&packet);
}
return 0;
}
2. 编译命令
gcc test.c -o test -L/usr/local/ffmpeg/lib -lavcodec -lavformat -lavutil -lswresample -lswscale