ffmpeg解碼視頻存為BMP文件

分類：?ffmpeg2011-07-28 12:13?8人閱讀?評論(0)?收藏?舉報 view plain

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

轉載于:https://www.cnblogs.com/moonvan/archive/2011/09/11/2173467.html

總結

以上是生活随笔為你收集整理的ffmpeg解码视频存为BMP文件的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。