V4L2摄像头获取单幅图片测试程序（MMAP模式）

1. #include <stdio.h>   
2. #include <stdlib.h>   
3. #include <string.h>   
4. #include <assert.h>   
5. #include <getopt.h>              
6. #include <fcntl.h>                
7. #include <unistd.h>   
8. #include <errno.h>   
9. #include <malloc.h>   
10. #include <sys/stat.h>   
11. #include <sys/types.h>   
12. #include <sys/time.h>   
13. #include <sys/mman.h>   
14. #include <sys/ioctl.h>   
15. #include <asm/types.h>            
16. #include <linux/videodev2.h>   
17. #define CLEAR(x) memset (&(x), 0, sizeof (x))   
18. struct buffer {  
19.         void *                  start;  
20.         size_t                  length;  
21. };  
22. static char *           dev_name        = "/dev/video0";//摄像头设备名   
23. static int              fd              = -1;  
24. struct buffer *         buffers         = NULL;  
25. static unsigned int     n_buffers       = 0;  
26. FILE *file_fd;  
27. static unsigned long file_length;  
28. static unsigned char *file_name;  
29. //   
30. //获取一帧数据   
31. //   
32. static int read_frame (void)  
33. {  
34. struct v4l2_buffer buf;  
35. unsigned int i;  
36. CLEAR (buf);  
37. buf.type = V4L2_BUF_TYPE_VIDEO_CAPTURE;  
38. buf.memory = V4L2_MEMORY_MMAP;  
39. ioctl (fd, VIDIOC_DQBUF, &buf); //出列采集的帧缓冲   
40. assert (buf.index < n_buffers);  
41.    printf ("buf.index dq is %d,\n",buf.index);  
42. fwrite(buffers[buf.index].start, buffers[buf.index].length, 1, file_fd); //将其写入文件中   
43.     
44. ioctl (fd, VIDIOC_QBUF, &buf); //再将其入列   
45. return 1;  
46. }  
47. int main (int argc,char ** argv)  
48. {  
49. struct v4l2_capability cap;   
50. struct v4l2_format fmt;  
51. unsigned int i;  
52. enum v4l2_buf_type type;  
53. file_fd = fopen("test-mmap.jpg", "w");//图片文件名   
54. fd = open (dev_name, O_RDWR /* required */ | O_NONBLOCK, 0);//打开设备   
55. ioctl (fd, VIDIOC_QUERYCAP, &cap);//获取摄像头参数   
56. CLEAR (fmt);  
57. fmt.type                = V4L2_BUF_TYPE_VIDEO_CAPTURE;  
58. fmt.fmt.pix.width       = 640;   
59. fmt.fmt.pix.height      = 480;  
60. fmt.fmt.pix.pixelformat = V4L2_PIX_FMT_YUYV;  
61. fmt.fmt.pix.field       = V4L2_FIELD_INTERLACED;  
62. ioctl (fd, VIDIOC_S_FMT, &fmt); //设置图像格式   
63. file_length = fmt.fmt.pix.bytesperline * fmt.fmt.pix.height; //计算图片大小   
64. struct v4l2_requestbuffers req;  
65. CLEAR (req);  
66. req.count               = 4;  
67. req.type                = V4L2_BUF_TYPE_VIDEO_CAPTURE;  
68. req.memory              = V4L2_MEMORY_MMAP;  
69. ioctl (fd, VIDIOC_REQBUFS, &req); //申请缓冲，count是申请的数量   
70. if (req.count < 2)  
71.    printf("Insufficient buffer memory\n");  
72. buffers = calloc (req.count, sizeof (*buffers));//内存中建立对应空间   
73. for (n_buffers = 0; n_buffers < req.count; ++n_buffers)   
74. {  
75.    struct v4l2_buffer buf;   //驱动中的一帧   
76.    CLEAR (buf);  
77.    buf.type        = V4L2_BUF_TYPE_VIDEO_CAPTURE;  
78.    buf.memory      = V4L2_MEMORY_MMAP;  
79.    buf.index       = n_buffers;  
80.    if (-1 == ioctl (fd, VIDIOC_QUERYBUF, &buf)) //映射用户空间   
81.     printf ("VIDIOC_QUERYBUF error\n");  
82.    buffers[n_buffers].length = buf.length;  
83.    buffers[n_buffers].start =  
84.    mmap (NULL /* start anywhere */,    //通过mmap建立映射关系   
85.     buf.length,  
86.     PROT_READ | PROT_WRITE /* required */,  
87.     MAP_SHARED /* recommended */,  
88.     fd, buf.m.offset);  
89.    if (MAP_FAILED == buffers[n_buffers].start)  
90.     printf ("mmap failed\n");  
91.         }  
92. for (i = 0; i < n_buffers; ++i)   
93. {  
94.    struct v4l2_buffer buf;  
95.    CLEAR (buf);  
96.    buf.type        = V4L2_BUF_TYPE_VIDEO_CAPTURE;  
97.    buf.memory      = V4L2_MEMORY_MMAP;  
98.    buf.index       = i;  
99.    if (-1 == ioctl (fd, VIDIOC_QBUF, &buf))//申请到的缓冲进入列队   
100.     printf ("VIDIOC_QBUF failed\n");  
101. }  
102.                   
103. type = V4L2_BUF_TYPE_VIDEO_CAPTURE;  
104. if (-1 == ioctl (fd, VIDIOC_STREAMON, &type)) //开始捕捉图像数据   
105.    printf ("VIDIOC_STREAMON failed\n");  
106. for (;;) //这一段涉及到异步IO   
107. {  
108.    fd_set fds;  
109.    struct timeval tv;  
110.    int r;  
111.    FD_ZERO (&fds);//将指定的文件描述符集清空   
112.    FD_SET (fd, &fds);//在文件描述符集合中增加一个新的文件描述符   
113.    /* Timeout. */  
114.    tv.tv_sec = 2;  
115.    tv.tv_usec = 0;  
116.    r = select (fd + 1, &fds, NULL, NULL, &tv);//判断是否可读（即摄像头是否准备好），tv是定时   
117.    if (-1 == r) {  
118.     if (EINTR == errno)  
119.      continue;  
120.     printf ("select err\n");  
121.                         }  
122.    if (0 == r) {  
123.     fprintf (stderr, "select timeout\n");  
124.     exit (EXIT_FAILURE);  
125.                         }  
126.    if (read_frame ())//如果可读，执行read_frame ()函数，并跳出循环   
127.    break;  
128. }  
129. unmap:  
130. for (i = 0; i < n_buffers; ++i)  
131.    if (-1 == munmap (buffers[i].start, buffers[i].length))  
132.     printf ("munmap error");  
133. close (fd);  
134. fclose (file_fd);  
135. exit (EXIT_SUCCESS);  
136. return 0;  
137. }