ffmpeg Windows下采集摄像头一帧数据，并保存为bmp图片

这里请注意，在编译ffmpeg时，不要使用--disable-devices选项。

使用

--enable-encoder=rawvideo
 --enable-decoder=rawvideo

启用rawvideo codec。

代码如下：

[cpp] view plaincopyprint?

1. #include <stdlib.h>
2. #include <stdio.h>
3. #include <string.h>
4. #include <libavformat/avformat.h>
5. #include <libavcodec/avcodec.h>
6. #include <libavdevice/avdevice.h>
7. #include <libswscale/swscale.h>
8. #include <windows.h>
9. #include <time.h>
10. #define MAX_INPUT_DEVICE_NUM 10
11. #ifdef _WIN32
12. int strcasecmp(const char *s1, const char *s2)
13. {
14. while ((*s1 != '\0')
15. && (tolower(*(unsigned char *) s1) ==
16. tolower(*(unsigned char *) s2)))
17. {
18. s1++;
19. s2++;
20. }
21. return tolower(*(unsigned char *) s1) - tolower(*(unsigned char *) s2);
22. }
23. int strncasecmp(const char *s1, const char *s2, unsigned int n)
24. {
25. if (n == 0)
26. return 0;
27. while ((n-- != 0)
28. && (tolower(*(unsigned char *) s1) ==
29. tolower(*(unsigned char *) s2))) {
30. if (n == 0 || *s1 == '\0' || *s2 == '\0')
31. return 0;
32. s1++;
33. s2++;
34. }
35. return tolower(*(unsigned char *) s1) - tolower(*(unsigned char *) s2);
36. }
37. #endif
38. void save_bmp(unsigned char * data,int data_size,int w,int h,FILE * out)
39. {
40. // 位图文件头
41. BITMAPFILEHEADER bmpheader;
42. BITMAPINFO bmpinfo;
43. int bit = 24;
44. bmpheader.bfType = ('M' <<8)|'B';
45. bmpheader.bfReserved1 = 0;
46. bmpheader.bfReserved2 = 0;
47. bmpheader.bfOffBits = sizeof(BITMAPFILEHEADER) + sizeof(BITMAPINFOHEADER);
48. bmpheader.bfSize = bmpheader.bfOffBits + w*h*bit/8;
49. bmpinfo.bmiHeader.biSize = sizeof(BITMAPINFOHEADER);
50. bmpinfo.bmiHeader.biWidth = w;
51. bmpinfo.bmiHeader.biHeight = 0-h;
52. bmpinfo.bmiHeader.biPlanes = 1;
53. bmpinfo.bmiHeader.biBitCount = bit;
54. bmpinfo.bmiHeader.biCompression = BI_RGB;
55. bmpinfo.bmiHeader.biSizeImage = 0;
56. bmpinfo.bmiHeader.biXPelsPerMeter = 100;
57. bmpinfo.bmiHeader.biYPelsPerMeter = 100;
58. bmpinfo.bmiHeader.biClrUsed = 0;
59. bmpinfo.bmiHeader.biClrImportant = 0;
60. fwrite(&bmpheader,sizeof(BITMAPFILEHEADER),1,out);
61. fwrite(&bmpinfo.bmiHeader,sizeof(BITMAPINFOHEADER),1,out);
62. fwrite(data,data_size,1,out);
63. }
64. int CaptureFromLocalCamera()
65. {
66. AVFormatContext *ic = NULL;
67. AVFormatParameters in_fmt_para={0};
68. AVPacket packet;
69. char buffer[MAX_PATH]={0};
70. int width = 0,height = 0;
71. int ret,video_stream = -1,i=0;
72. //查找输入（vfwcap）格式
73. AVInputFormat *in_fmt = av_find_input_format ("vfwcap");
74. if (in_fmt == NULL)
75. {
76. printf("not support input device vfwcap.\n");
77. return -1;
78. }
79. memset (&in_fmt_para, 0, sizeof(in_fmt_para));
80. //指定需要采集图像的高度
81. in_fmt_para.height = height;
82. //指定需要采集图像的宽度
83. in_fmt_para.width  = width;
84. //设置帧率
85. av_parse_video_frame_rate(&in_fmt_para.time_base,"20");
86. //打开摄像头设备,从"0"到MAX_INPUT_DEVICE_NUM依次尝试打开
87. for( i=0 ; i < MAX_INPUT_DEVICE_NUM + 1; i++ )
88. {
89. sprintf(buffer,"%d",i);
90. ret = av_open_input_file ( &ic, buffer, in_fmt,sizeof(in_fmt_para),&in_fmt_para);
91. if ( ret == 0 && ic)
92. {
93. break;
94. }
95. }
96. //open success?
97. if(!ic || ret != 0)
98. {
99. if(ic)
100. av_close_input_file(ic);
101. printf("can not open input file.\n");
102. return -2;
103. }
104. printf("input device no. is %d\n",i);
105. //find the video stream
106. for(i=0;i<ic ->nb_streams;i++)
107. {
108. if ( CODEC_TYPE_VIDEO == ic ->streams[i] ->codec ->codec_type )
109. {
110. video_stream = i;
111. break;
112. }
113. }
114. if(video_stream < 0)
115. {
116. av_close_input_file(ic);
117. printf("can not find a video stream.\n");
118. return -3;
119. }
120. //获取视频时间宽度和高度
121. width  = ic ->streams[video_stream] ->codec ->width;
122. height = ic ->streams[video_stream] ->codec ->height;
123. printf("video size: %dx%d\n",width,height);
124. //从摄像头获取图像数据
125. if( 0 == av_read_frame(ic,&packet))
126. {
127. //find the decode codec
128. AVCodec * decodec =  avcodec_find_decoder(ic ->streams[video_stream] ->codec ->codec_id);
129. if(decodec)
130. {
131. //open the decode codec
132. if( 0 == avcodec_open(ic ->streams[video_stream] ->codec,decodec) )
133. {
134. int got_picture = 0;
135. AVFrame * frame = avcodec_alloc_frame();
136. avcodec_decode_video2(ic ->streams[video_stream] ->codec,frame,&got_picture,&packet);
137. //decode success
138. if(got_picture)
139. {
140. uint8_t * buffer = NULL;
141. size_t buffer_size = 0;
142. struct SwsContext *pSwsCtx=NULL;
143. AVFrame * rgb_frame = avcodec_alloc_frame();
144. buffer_size = avpicture_get_size(PIX_FMT_BGR24,width,height);
145. buffer = (uint8_t *)av_malloc(buffer_size);
146. avpicture_fill((AVPicture*)rgb_frame,(uint8_t *)buffer,PIX_FMT_BGR24,width,height);
147. //get swscale ctx
148. pSwsCtx = sws_getContext(
149. ic ->streams[video_stream] ->codec ->width,
150. ic ->streams[video_stream] ->codec ->height,
151. ic ->streams[video_stream] ->codec ->pix_fmt,
152. width,
153. height,
154. PIX_FMT_BGR24,
155. SWS_BILINEAR,
156. NULL,
157. NULL,
158. NULL);
159. if(pSwsCtx)
160. {
161. FILE *fp = NULL;
162. SYSTEMTIME dt={0};
163. //图像格式转换
164. sws_scale(
165. pSwsCtx,
166. frame ->data,
167. frame ->linesize,
168. 0,
169. ic ->streams[video_stream] ->codec ->height,
170. rgb_frame ->data,
171. rgb_frame ->linesize);
172. //create the image file name
173. GetLocalTime(&dt);
174. srand(0);
175. sprintf(buffer,"imgs/%04d_%02d_%02d %02d_%02d_%02d %02d.bmp",dt.wYear,dt.wMonth,dt.wDay,dt.wHour,dt.wMinute,dt.wSecond,rand()%30);
176. //
177. CreateDirectoryA("imgs",NULL);
178. //open file
179. fp = fopen(buffer, "wb");
180. if(fp)
181. {
182. save_bmp(rgb_frame ->data[0],rgb_frame ->linesize[0]*height,width,height,fp);
183. fclose(fp);
184. }
185. //free sws ctx
186. sws_freeContext(pSwsCtx);
187. }
188. //free buffer
189. av_free(rgb_frame);
190. av_free(buffer);
191. }
192. //free buffer
193. av_free(frame);
194. //close the decode codec
195. avcodec_close(ic ->streams[video_stream] ->codec);
196. }
197. }
198. }
199. //close the input device
200. av_close_input_file(ic);
201. return 0;
202. }
203. int main()
204. {
205. //avcodec_init();
206. avcodec_register_all();
207. avdevice_register_all();
208. CaptureFromLocalCamera();
209. return 0;
210. }