程序博客网 > 变频器模拟软件

ffmpeg学习---3.Outputting to the Screen

来源：互联网 发布：变频器模拟软件 编辑：程序博客网 时间：2024/05/17 07:36
1. 2output.c
  1. // tutorial01.c
  2. // Code based on a tutorial by Martin Bohme (boehme@inb.uni-luebeckREMOVETHIS.de)
  3. // Tested on Gentoo, CVS version 5/01/07 compiled with GCC 4.1.1

  5. // A small sample program that shows how to use libavformat and libavcodec to
  6. // read video from a file.
  7. //
  8. // Use
  9. //
  10. // gcc -o tutorial01 tutorial01.c -lavformat -lavcodec -lz
  11. //
  12. // to build (assuming libavformat and libavcodec are correctly installed
  13. // your system).
  14. //
  15. // Run using
  16. //
  17. // tutorial01 myvideofile.mpg
  18. //
  19. // to write the first five frames from "myvideofile.mpg" to disk in PPM
  20. // format.
  21. #include <stdio.h>
  22. #include <libavformat/avformat.h>
  23. #include <libswscale/swscale.h>
  24. #include <SDL.h>
  25. int main(int argc, char *argv[]) {
  26.     AVFormatContext *pFormatCtx;
  27.     int i, videoStream;
  28.     AVCodecContext *pCodecCtx;
  29.     AVCodec *pCodec;
  30.     AVFrame *pFrame; 
  31.     AVFrame *pFrameRGB;
  32.     AVPacket packet;
  33.     int frameFinished;
  34.     int numBytes;
  35.     uint8_t *buffer;

  37.     if(argc < 2) {
  38.         printf("Please provide a movie file\n");
  39.         return -1;
  40.     }
  41.     
  42.     if(SDL_Init(SDL_INIT_VIDEO|SDL_INIT_AUDIO|SDL_INIT_TIMER))
  43.     {
  44.         fprintf(stderr, "SD_init error \n");
  45.         exit(1);
  46.     }
  47.     // Register all formats and codecs
  48.     av_register_all();

  50.     pFormatCtx = avformat_alloc_context();

  52.     // Open video file
  53.     //if(av_open_input_file(&pFormatCtx, argv[1], NULL, 0, NULL)!=0)
  54.     if(avformat_open_input(&pFormatCtx, argv[1], NULL, NULL)!=0)
  55.         return -1; // Couldn't open file

  57.     // Retrieve stream information
  58.     if(avformat_find_stream_info(pFormatCtx,NULL)<0)
  59.         return -1; // Couldn't find stream information

  61.     // Dump information about file onto standard error
  62.     av_dump_format(pFormatCtx, 0, argv[1], 0);

  64.     // Find the first video stream
  65.     videoStream=-1;
  66.     for(i=0; i<pFormatCtx->nb_streams; i++)
  67.         if(pFormatCtx->streams[i]->codec->codec_type==AVMEDIA_TYPE_VIDEO) 
  68.         {
  69.             videoStream=i;
  70.             break;
  71.         }
  72.     if(videoStream==-1)
  73.         return -1; // Didn't find a video stream

  75.     // Get a pointer to the codec context for the video stream
  76.     pCodecCtx=pFormatCtx->streams[videoStream]->codec;

  78.     // Find the decoder for the video stream
  79.     pCodec=avcodec_find_decoder(pCodecCtx->codec_id);
  80.     if(pCodec==NULL) {
  81.         fprintf(stderr, "Unsupported codec!\n");
  82.         return -1; // Codec not found
  83.     }

  85.     // Open codec
  86.     if(avcodec_open2(pCodecCtx, pCodec, NULL)<0)
  87.         return -1; // Could not open codec

  89.     //Set up a screen
  90.     SDL_Surface * screen;
  91.     screen = SDL_SetVideoMode(pCodecCtx->width, pCodecCtx->height, 0, 0);
  92.     if(!screen)
  93.     {
  94.         fprintf(stderr, "SetVideoMode error\n");
  95.         exit(1);
  96.     }

  98.     //create a YUV overlay
  99.     SDL_Overlay *bmp;
  100.     SDL_Rect rect;
  101.     SDL_Event event;
  102.     bmp = SDL_CreateYUVOverlay(pCodecCtx->width, pCodecCtx->height,
  103.             SDL_YV12_OVERLAY, screen);
  104.     // Allocate video frame
  105.     pFrame=avcodec_alloc_frame();

  107.     // Allocate an AVFrame structure
  108.     pFrameRGB=avcodec_alloc_frame();
  109.     if(pFrameRGB==NULL)
  110.         return -1;

  112.     // Determine required buffer size and allocate buffer
  113.     numBytes=avpicture_get_size(PIX_FMT_RGB24, pCodecCtx->width,
  114.             pCodecCtx->height);
  115.     buffer=(uint8_t *)av_malloc(numBytes*sizeof(uint8_t));

  117.     // Assign appropriate parts of buffer to image planes in pFrameRGB
  118.     // Note that pFrameRGB is an AVFrame, but AVFrame is a superset
  119.     // of AVPicture
  120.     avpicture_fill((AVPicture *)pFrameRGB, buffer, PIX_FMT_RGB24,
  121.             pCodecCtx->width, pCodecCtx->height);

  123.     // Read frames and display on screen 
  124.     while(av_read_frame(pFormatCtx, &packet)>=0) {
  125.         // Is this a packet from the video stream?
  126.         if(packet.stream_index==videoStream) {
  127.             // Decode video frame
  128.             avcodec_decode_video2(pCodecCtx, pFrame, &frameFinished, &packet);

  130.             // Did we get a video frame?
  131.             if(frameFinished) {
  132.                 SDL_LockYUVOverlay(bmp);
  133.                 AVPicture pict;
  134.                 pict.data[0] = bmp->pixels[0];
  135.                 pict.data[1] = bmp->pixels[2];
  136.                 pict.data[2] = bmp->pixels[1];

  138.                 pict.linesize[0] = bmp->pitches[0];
  139.                 pict.linesize[1] = bmp->pitches[2];
  140.                 pict.linesize[2] = bmp->pitches[1];

  142.                // img_convert(&pict, PIX_FMT_YUV420P, (AVPicture*)pFrame, pCodecCtx->pix_fmt,
  143.                // pCodecCtx->width, pCodecCtx->height);
  144.                 struct SwsContext *img_convert_ctx;
  145.                 img_convert_ctx = sws_getContext(pCodecCtx->width, pCodecCtx->height, pCodecCtx->pix_fmt, pCodecCtx->width, pCodecCtx->height, PIX_FMT_YUV420P, SWS_BICUBIC, NULL, NULL, NULL);
  146.                 if(img_convert_ctx == NULL)
  147.                 {
  148.                     fprintf(stderr, "Cannot initialize the conversion context!\n");
  149.                             exit(1);
  150.                 }

  152.                 sws_scale(img_convert_ctx, (const uint8_t* const*)pFrame->data, pFrame->linesize, 0, pCodecCtx->height, pict.data, pict.linesize);
  153.                 SDL_UnlockYUVOverlay(bmp);
  154.                 rect.x = 0;
  155.                 rect.y = 0;
  156.                 rect.w = pCodecCtx->width;
  157.                 rect.h = pCodecCtx->height;
  158.                 SDL_DisplayYUVOverlay(bmp, &rect);

  160.             }
  161.         }

  163.         // Free the packet that was allocated by av_read_frame
  164.         av_free_packet(&packet);
  165.         SDL_PollEvent(&event);
  166.         switch(event.type)
  167.         {
  168.             case SDL_QUIT:
  169.                 SDL_Quit();
  170.                 exit(0);
  171.                 break;
  172.             default:
  173.                 break;
  174.         }
  175.     }

  177.     // Free the RGB image
  178.     av_free(buffer);
  179.     av_free(pFrameRGB);

  181.     // Free the YUV frame
  182.     av_free(pFrame);

  184.     // Close the codec
  185.     avcodec_close(pCodecCtx);

  187.     // Close the video file
  188.     avformat_close_input(&pFormatCtx);

  190.     return 0;
  191. }
2.Makefile
  1. CC=gcc
  2. CFLAGS = -g -I/home/sun/code/ffmpeg-1.0/install/include/ -I/home/sun/code/SDL-1.2.15/install/include/SDL/
  3. LDFLAGS = -L/home/sun/code/ffmpeg-1.0/install/lib/ -lavutil -lavformat -lavcodec -lavutil -lm -lswscale
  4. LDFLAGS += -L/home/sun/code/SDL-1.2.15/install/lib/ -lSDLmain -lSDL

  6. TARGETS=2output
  7. all: $(TARGETS)
  8. 2output.o:2output.c
  9.     $(CC) $(CFLAGS) -o $@ -c $^

  11. 2output:2output.o
  12.     $(CC) -o $@ $^ $(LDFLAGS)
  13. clean:
  14.     rm -rf *.o $(TARGETS)
二. 更新版
1. 代码
现在ffmpeg发布了ffmpeg-2.7.2,下面基于这个版本更新一下,同时加上注解,下次看的时候理解就更深了
  1. cong@msi:/work/ffmpeg/test/1view$ cat view.c 
  2. #include "utils.h"
  3. #include <libavformat/avformat.h>
  4. #include <libswscale/swscale.h>
  5. #include <SDL/SDL.h>

  7. SDL_mutex *affmutex;
  8. SDL_Event sdlevent;
  9. int signal_quit = 1;

  11. static int eventThread(void* data)
  12. {
  13.     while(signal_quit)
  14.     {
  15.         SDL_LockMutex(affmutex);
  16.         while(SDL_PollEvent(&sdlevent))
  17.         {
  18.             switch(sdlevent.type)
  19.             {
  20.                 case SDL_QUIT:
  21.                     {
  22.                         signal_quit = 0;
  23.                     }
  24.                     break;
  25.                 default:
  26.                     break;

  28.             }
  29.         }
  30.         SDL_UnlockMutex(affmutex);
  31.     } 
  32. }

  34. int main(int argc, char **argv)
  35. {
  36.     int i=0;
  37.     int ret;
  38.     int videoindex= -1;
  39.     int frameFinished;
  40.     AVFormatContext *pFormatCtx = NULL;
  41.     AVCodecContext * pCodecCtx;
  42.     AVCodec * pCodec;
  43.     AVFrame * pFrame;
  44.     AVFrame * pFrameYUV;
  45.     AVPacket * packet; 
  46.     struct SwsContext *img_convert_ctx;

  48.     SDL_Surface* psscreen;
  49.     SDL_Overlay* overlay;
  50.     SDL_Rect rect;
  51.     SDL_Thread* sdl_thread;
  52.     //a. ffmpeg的初始化(虽然名字是register就这么说吧)
  53.     avcodec_register_all();
  54.     avfilter_register_all();
  55.     av_register_all();
  56.     
  57.     //b.打开视频文件
  58.     pFormatCtx = avformat_alloc_context();
  59.     if(avformat_open_input(&pFormatCtx, argv[1], NULL, NULL)!=0)
  60.         return -1; 
  61.     //c.获取视频文件的流信息,(即查看有几个视频流几个音频流)    
  62.     if(avformat_find_stream_info(pFormatCtx, NULL)<0)
  63.         return -1; 
  64.     av_dump_format(pFormatCtx,0, 0, 0);
  65.     //d.获取了视频流的index,这样以后读取一帧之后,根据索引号才能判断这一帧是不是视频帧
  66.     for(i=0; i<pFormatCtx->nb_streams; i++)
  67.     {
  68.         if(pFormatCtx->streams[i]->codec->codec_type == AVMEDIA_TYPE_VIDEO)
  69.         {
  70.             videoindex= i;
  71.             break;
  72.         }
  73.     }
  74.     if(videoindex== -1)
  75.     {
  76.         dbmsg("no video stream found!");
  77.         return -1;
  78.     }
  79.     dbmsg("videoindex=%d", videoindex);
  80.     //e.为视频流寻找解码器:在c中不仅有视频流的index,还有视频流的编码codec_id
  81.     //通过这个codec_id就可以寻到视频流的解码器
  82.     pCodecCtx = pFormatCtx->streams[videoindex]->codec;
  83.     pCodec = avcodec_find_decoder(pCodecCtx->codec_id);
  84.     if(pCodec == NULL)
  85.     {
  86.         dbmsg("Codec not found");
  87.         return -1;
  88.     }
  89.     if(avcodec_open2(pCodecCtx, pCodec, NULL) < 0)   //找到解码器之后打开解码器
  90.         return -1;

  92.     pFrame = av_frame_alloc();         //以前的avcodec_alloc_frame函数现在不用了
  93.     pFrameYUV = av_frame_alloc();

  95.     //显示的准备:SDL初始化,设置显示模式,创建画布
  96.     SDL_Init(SDL_INIT_EVERYTHING);
  97.     psscreen = SDL_SetVideoMode(pCodecCtx->width, pCodecCtx->height, 0, SDL_SWSURFACE);
  98.     SDL_WM_SetCaption( "FFMPEG Window", NULL);
  99.     //注意这儿的参数SDL_YU12_OVERLAY与SDL_YUY2_OVERLAY一定要与下面sws_scale中的参数配套
  100.     //overlay = SDL_CreateYUVOverlay(pCodecCtx->width, pCodecCtx->height, SDL_YV12_OVERLAY, psscreen);
  101.     overlay = SDL_CreateYUVOverlay(pCodecCtx->width, pCodecCtx->height, SDL_YUY2_OVERLAY, psscreen);

  103.     //与SDL的退出相关
  104.     affmutex = SDL_CreateMutex();
  105.     sdl_thread = SDL_CreateThread(eventThread, NULL);    //创建SDL线程监测退出信号

  107.     rect.x = 0;
  108.     rect.y = 0;
  109.     rect.w = pCodecCtx->width;
  110.     rect.h = pCodecCtx->height;

  112.     packet = (AVPacket*)av_malloc(sizeof(AVPacket));
  113.     //注意这儿的参数PIX_FMT_YUV420P或PIX_FMT_YUYU422一定要与上面SDL_CreateYUVOVerlay中的参数配套
  114.     //img_convert_ctx = sws_getContext(pCodecCtx->width, pCodecCtx->height, pCodecCtx->pix_fmt, pCodecCtx->width, pCodecCtx->height, PIX_FMT_YUV420P, SWS_BICUBIC, NULL, NULL, NULL);
  115.     img_convert_ctx = sws_getContext(pCodecCtx->width, pCodecCtx->height, pCodecCtx->pix_fmt, pCodecCtx->width, pCodecCtx->height, PIX_FMT_YUYV422, SWS_BICUBIC, NULL, NULL, NULL);
  116.     if(img_convert_ctx == NULL)
  117.     {
  118.         dbmsg("img_convert error");
  119.         return -1;
  120.     }
  121.     //f.循环读取视频文件所有的帧,包括音频帧与视频帧:读取帧到packet
  122.     while( (av_read_frame(pFormatCtx, packet)>=0) && (signal_quit))
  123.     {
  124.         //g.判断是视频帧,则对视频帧解码
  125.         if(packet->stream_index == videoindex)   
  126.         {    //解码packet中的视频帧到pFrame中
  127.             if((ret=avcodec_decode_video2(pCodecCtx, pFrame, &frameFinished, packet)) < 0)
  128.             {
  129.                 dbmsg("decocode video error");
  130.                 return -1;
  131.             }
  132.             if(frameFinished)
  133.             {
  134.                 SDL_LockYUVOverlay(overlay);
  135.                 pFrameYUV->data[0] = overlay->pixels[0];      //Y
  136.                 pFrameYUV->data[1] = overlay->pixels[2];      //U   //对overlay中的数据组织不清楚,为什么这儿要把uv交换呢?有知道的告诉我一下,谢谢
  137.                 pFrameYUV->data[2] = overlay->pixels[1];      //V
  138.                 pFrameYUV->linesize[0] = overlay->pitches[0];
  139.                 pFrameYUV->linesize[1] = overlay->pitches[2];
  140.                 pFrameYUV->linesize[2] = overlay->pitches[1];
  141.                 sws_scale(img_convert_ctx, (const uint8_t* const*)pFrame->data, pFrame->linesize, 0, 
  142.                         pCodecCtx->height, pFrameYUV->data, pFrameYUV->linesize);    //将解码后pFrame中的数据转为sws_getContext中设定的格式,这样就可以通过sdl进行显示了 
  143.                 SDL_UnlockYUVOverlay(overlay);
  144.                 SDL_DisplayYUVOverlay(overlay, &rect);
  145.                 SDL_Delay(40);
  146.             }
  147.         }
  148.     }
  149.     SDL_WaitThread(sdl_thread, &ret);
  150.     SDL_DestroyMutex(affmutex);
  151.     return 0;
  152. }
说明1: SDL_CreateYUVOverlay与sws_getContext中的参数是相对应的
    SDL_CreateYUVOverlay   SDL_YV12_OVERLAY
                                       SDL_YUY2_OVERLAY     
    sws_getContext              PIX_FMT_YUV420P
                                       PIX_FMT_YUYV422    
     SDL_YV12_OVERLAY + PIX_FMT_YUYV422: 则显示不正常 
    SDL_YUY2_OVERLAY + PIX_FMT_YUV420P: 则sws_scale会报错bad dst image pointers
说明2:sws_scale之后就可以把yuv420p数据写到文件中
FILE *fp_yuv;
fp_yuv=fopen("output.yuv","wb+");

int y_size=pCodecCtx->width*pCodecCtx->height;    
fwrite(pFrameYUV->data[0],1,y_size,fp_yuv);    //Y   
fwrite(pFrameYUV->data[1],1,y_size/4,fp_yuv);  //U  
fwrite(pFrameYUV->data[2],1,y_size/4,fp_yuv);  //V 

1.2 Makefile
  1. cong@msi:/work/ffmpeg/test/1view$ cat Makefile 
  2. EXE=view
  3. CC=gcc
  4. FFMPEG=/work/ffmpeg/out
  5. CFLAGS=-g -O0 -I$(FFMPEG)/include
  6. LDFLAGS = -L$(FFMPEG)/lib/ -lswscale -lswresample -lavformat -lavdevice -lavcodec -lavutil -lavfilter -lm -lSDL
  7. SRC=$(wildcard *.c)
  8. OBJ=$(patsubst %.c,%.o,$(SRC))
  9. DEP=$(patsubst %.c,.%.d,$(SRC))
  10. $(EXE):$(OBJ)
  11.     $(CC) $(CFLAGS) $^ -o $@ $(LDFLAGS)

  13. $(DEP):.%.d:%.c
  14.     @set -e; rm -f $@; \
  15.     $(CC) -MM $< > $@.$$$$; \
  16.     sed 's,/($*/)/.o[ :]*,/1.o $@ : ,g' < $@.$$$$ > $@; \
  17.     rm -f $@.$$$$

  19. -include $(DEP)
  20. clean:
  21.     @rm $(EXE) $(OBJ) $(DEP) -f
  22. run:
  23.     export LD_LIBRARY_PATH=$(FFMPEG)/lib/ \
  24.     && ./$(EXE) ../resource/bing.rmvb

2.运行
  1. cong@msi:/work/ffmpeg/test/1view$ make run
  2. export LD_LIBRARY_PATH=/work/ffmpeg/out/lib/ \     -->己经在Makefile中写好了视频文件的路径
  3.     && ./view ../resource/bing.rmvb
3. 代码打包
1view.rar (下载后改名为1view.tar.gz)
0 0
变频器模拟软件 变频器模拟软件
原创粉丝点击
热门IT博客
js treeview drag区域js treeview drag区域
知乎超过1000赞的回答知乎超过1000赞的回答
编辑排版软件
拳皇97java源代码
小米手机抢购软件
淘宝刷手被发现
2004年网络歌曲大全
软件测试流程图
oracle导入大批量数据
淘宝砍价技巧
陕西广电网络吧
新商盟软件下载
英语看图识音软件
wps表格怎么刷新数据
wampserver mac版
蛇口招商网络怎么样
ubuntu 修改文件指令
linux空间扩容
java 运算符
角色扮演网络
热门问题 老师的惩罚 人脸识别 我在镇武司摸鱼那些年 重生之率土为王 我在大康的咸鱼生活 盘龙之生命进化 天生仙种 凡人之先天五行 春回大明朝 姑娘不必设防，我是瞎子 微型混泥土泵车价格 微型隔膜泵 微型课怎么上 英语微型课 大众微型汽车2人座 双人微型汽车 两座微型轿车 微型自动档汽车 微型轿车排行榜 两座微型车报价 微型洗车器 微型踏板车 二手微型车价格 自动挡微型车哪个好 大众两座微型车 小型集装箱车图片 便宜微型车 哪款微型车好 哈弗微型车 自动挡微型车排行榜 微型车强制险多少钱 豪华微型车 奔奔微型车 两门微型车 微型车购置税 微型车对比 微型车暖风 江铃微型车 自动挡微型车推荐 微型车电动 排半小货车报价及图片 柴油小货车报价及图片 二手小箱货车报价及图片 五菱集装箱小货车报价 长安小箱货车大全 长安箱货车报价及图片 小型卡车图片报价 福田集装箱小货车 小型卡车 福田小卡车报价及图片 箱货车图片

程序博客网，程序员的互联网技术博客家园。csdn论坛精品 msdn技术资料都在这里