ffmpeg解码视频存为BMP文件

1. #include <windows.h>
2. #include <stdio.h>
3. #include <stdlib.h>
4. #include <string.h>
5. #pragma once
6. #ifdef __cplusplus
7. extern "C" {
8. #endif
9. #include <libavcodec/avcodec.h>
10. #include <libavformat/avformat.h>
11. #include <libswscale/swscale.h>
12. #ifdef __cplusplus
13. }
14. #endif
15. //定义BMP文件头
16. #ifndef _WINGDI_
17. #define _WINGDI_
18. typedef struct tagBITMAPFILEHEADER {
19. WORD    bfType;
20. DWORD   bfSize;
21. WORD    bfReserved1;
22. WORD    bfReserved2;
23. DWORD   bfOffBits;
24. } BITMAPFILEHEADER, FAR *LPBITMAPFILEHEADER, *PBITMAPFILEHEADER;
25. typedef struct tagBITMAPINFOHEADER{
26. DWORD      biSize;
27. LONG       biWidth;
28. LONG       biHeight;
29. WORD       biPlanes;
30. WORD       biBitCount;
31. DWORD      biCompression;
32. DWORD      biSizeImage;
33. LONG       biXPelsPerMeter;
34. LONG       biYPelsPerMeter;
35. DWORD      biClrUsed;
36. DWORD      biClrImportant;
37. } BITMAPINFOHEADER, FAR *LPBITMAPINFOHEADER, *PBITMAPINFOHEADER;
38. #endif
39. //保存BMP文件的函数
40. void SaveAsBMP (AVFrame *pFrameRGB, int width, int height, int index, int bpp)
41. {
42. char buf[5] = {0};
43. BITMAPFILEHEADER bmpheader;
44. BITMAPINFOHEADER bmpinfo;
45. FILE *fp;
46. char *filename = new char[255];
47. //文件存放路径，根据自己的修改
48. sprintf_s(filename,255,"%s%d.bmp","D:/My Documents/Visual Studio 2008/Projects/WriteVideo/Debug/test",index);
49. if ( (fp=fopen(filename,"wb+")) == NULL )
50. {
51. printf ("open file failed!\n");
52. return;
53. }
54. bmpheader.bfType = 0x4d42;
55. bmpheader.bfReserved1 = 0;
56. bmpheader.bfReserved2 = 0;
57. bmpheader.bfOffBits = sizeof(BITMAPFILEHEADER) + sizeof(BITMAPINFOHEADER);
58. bmpheader.bfSize = bmpheader.bfOffBits + width*height*bpp/8;
59. bmpinfo.biSize = sizeof(BITMAPINFOHEADER);
60. bmpinfo.biWidth = width;
61. bmpinfo.biHeight = height;
62. bmpinfo.biPlanes = 1;
63. bmpinfo.biBitCount = bpp;
64. bmpinfo.biCompression = BI_RGB;
65. bmpinfo.biSizeImage = (width*bpp+31)/32*4*height;
66. bmpinfo.biXPelsPerMeter = 100;
67. bmpinfo.biYPelsPerMeter = 100;
68. bmpinfo.biClrUsed = 0;
69. bmpinfo.biClrImportant = 0;
70. fwrite (&bmpheader, sizeof(bmpheader), 1, fp);
71. fwrite (&bmpinfo, sizeof(bmpinfo), 1, fp);
72. fwrite (pFrameRGB->data[0], width*height*bpp/8, 1, fp);
73. fclose(fp);
74. }
75. //主函数
76. int main (void)
77. {
78. unsigned int i = 0, videoStream = -1;
79. AVCodecContext *pCodecCtx;
80. AVFormatContext *pFormatCtx;
81. AVCodec *pCodec;
82. AVFrame *pFrame, *pFrameRGB;
83. struct SwsContext *pSwsCtx;
84. const char *filename = "D:/My Documents/Visual Studio 2008/Projects/WriteVideo/Debug/DELTA.MPG";
85. AVPacket packet;
86. int frameFinished;
87. int PictureSize;
88. uint8_t *buf;
89. //注册编解码器
90. av_register_all();
91. //打开视频文件
92. if ( av_open_input_file(&pFormatCtx, filename, NULL, 0, NULL) != 0 )
93. {
94. printf ("av open input file failed!\n");
95. exit (1);
96. }
97. //获取流信息
98. if ( av_find_stream_info(pFormatCtx) < 0 )
99. {
100. printf ("av find stream info failed!\n");
101. exit (1);
102. }
103. //获取视频流
104. for ( i=0; i<pFormatCtx->nb_streams; i++ )
105. if ( pFormatCtx->streams[i]->codec->codec_type == CODEC_TYPE_VIDEO )
106. {
107. videoStream = i;
108. break;
109. }
110. if (videoStream == -1)
111. {
112. printf ("find video stream failed!\n");
113. exit (1);
114. }
115. pCodecCtx = pFormatCtx->streams[videoStream]->codec;
116. pCodec = avcodec_find_decoder (pCodecCtx->codec_id);
117. if (pCodec == NULL)
118. {
119. printf ("avcode find decoder failed!\n");
120. exit (1);
121. }
122. //打开解码器
123. if ( avcodec_open(pCodecCtx, pCodec)<0 )
124. {
125. printf ("avcode open failed!\n");
126. exit (1);
127. }
128. //为每帧图像分配内存
129. pFrame = avcodec_alloc_frame();
130. pFrameRGB = avcodec_alloc_frame();
131. if ( (pFrame==NULL)||(pFrameRGB==NULL) )
132. {
133. printf("avcodec alloc frame failed!\n");
134. exit (1);
135. }
136. PictureSize = avpicture_get_size (PIX_FMT_BGR24, pCodecCtx->width, pCodecCtx->height);
137. buf = (uint8_t*)av_malloc(PictureSize);
138. if ( buf == NULL )
139. {
140. printf( "av malloc failed!\n");
141. exit(1);
142. }
143. avpicture_fill ( (AVPicture *)pFrameRGB, buf, PIX_FMT_BGR24, pCodecCtx->width, pCodecCtx->height);
144. //设置图像转换上下文
145. pSwsCtx = sws_getContext (pCodecCtx->width,
146. pCodecCtx->height,
147. pCodecCtx->pix_fmt,
148. pCodecCtx->width,
149. pCodecCtx->height,
150. PIX_FMT_BGR24,
151. SWS_BICUBIC,
152. NULL, NULL, NULL);
153. i = 0;
154. while(av_read_frame(pFormatCtx, &packet) >= 0)
155. {
156. if(packet.stream_index==videoStream)
157. {
158. avcodec_decode_video(pCodecCtx, pFrame, &frameFinished,
159. packet.data, packet.size);
160. if(frameFinished)
161. {
162. //反转图像 ，否则生成的图像是上下调到的
163. pFrame->data[0] += pFrame->linesize[0] * (pCodecCtx->height - 1);
164. pFrame->linesize[0] *= -1;
165. pFrame->data[1] += pFrame->linesize[1] * (pCodecCtx->height / 2 - 1);
166. pFrame->linesize[1] *= -1;
167. pFrame->data[2] += pFrame->linesize[2] * (pCodecCtx->height / 2 - 1);
168. pFrame->linesize[2] *= -1;
169. //转换图像格式，将解压出来的YUV420P的图像转换为BRG24的图像
170. sws_scale (pSwsCtx, pFrame->data, pFrame->linesize, 0, pCodecCtx->height, pFrameRGB->data, pFrameRGB->linesize);
171. SaveAsBMP (pFrameRGB, pCodecCtx->width, pCodecCtx->height, i++, 24);
172. }
173. }
174. av_free_packet(&packet);
175. }
176. sws_freeContext (pSwsCtx);
177. av_free (pFrame);
178. av_free (pFrameRGB);
179. avcodec_close (pCodecCtx);
180. av_close_input_file (pFormatCtx);
181. return 0;
182. }