V4L2通过摄像头采集图片

[python]  view plain copy
  1. V4L2通过摄像头采集图片  
[python]  view plain copy
  1. #include <stdio.h>  
  2. #include <stdlib.h>  
  3. #include <string.h>  
  4.   
  5.   
  6. #include <assert.h>  
  7. #include <getopt.h>             
  8. #include <fcntl.h>              
  9. #include <unistd.h>  
  10. #include <errno.h>  
  11. #include <malloc.h>  
  12. #include <sys/stat.h>  
  13. #include <sys/types.h>  
  14. #include <sys/time.h>  
  15. #include <sys/mman.h>  
  16. #include <sys/ioctl.h>  
  17.   
  18.   
  19.   
  20.   
  21. #include <asm/types.h>          
  22. #include <linux/videodev2.h>  
  23.   
  24.   
  25. #define CAMERA_DEVICE "/dev/video0"  
  26.   
  27.   
  28. #define CAPTURE_FILE "frame_yuyv_new.jpg"  
  29. #define CAPTURE_RGB_FILE "frame_rgb_new.bmp"  
  30. #define CAPTURE_show_FILE "a.bmp"  
  31.   
  32.   
  33. #define VIDEO_WIDTH 640  
  34. #define VIDEO_HEIGHT 480  
  35. #define VIDEO_FORMAT V4L2_PIX_FMT_YUYV  
  36. #define BUFFER_COUNT 4  
  37.   
  38.   
  39. typedef struct VideoBuffer {  
  40.     void   *start; //视频缓冲区的起始地址  
  41.     size_t  length;//缓冲区的长度  
  42. } VideoBuffer;  
  43.   
  44.   
  45. /*  
  46. void *calloc(unsigned n,unsigned size) 功 能: 在内存的动态存储区中分配n个长度为size的连续空间,函数返回一个指向分配起始地址的指针;如果分配不成功,返回NULL。跟malloc的区别:calloc在动态分配完内存后,自动初始化该内存空间为零,而malloc不初始化,里边数据是随机的垃圾数据  
  47. */  
  48. //位图文件头数据结构含有位图文件的类型,大小和打印格式等信息  
  49. //进行数据字节的对齐  
  50. #pragma pack(1)  
  51. typedef struct BITMAPFILEHEADER  
  52. {  
  53.   unsigned short bfType;//位图文件的类型,  
  54.   unsigned long bfSize;//位图文件的大小,以字节为单位  
  55.   unsigned short bfReserved1;//位图文件保留字,必须为0  
  56.   unsigned short bfReserved2;//同上  
  57.   unsigned long bfOffBits;//位图阵列的起始位置,以相对于位图文件   或者说是头的偏移量表示,以字节为单位  
  58. } BITMAPFILEHEADER;  
  59. #pragma pack()  
  60.   
  61.   
  62. typedef struct BITMAPINFOHEADER//位图信息头类型的数据结构,用于说明位图的尺寸  
  63. {  
  64.   unsigned long biSize;//位图信息头的长度,以字节为单位  
  65.   unsigned long biWidth;//位图的宽度,以像素为单位  
  66.   unsigned long biHeight;//位图的高度,以像素为单位  
  67.   unsigned short biPlanes;//目标设备的级别,必须为1  
  68.   unsigned short biBitCount;//每个像素所需的位数,必须是1(单色),4(16色),8(256色)或24(2^24色)之一  
  69.   unsigned long biCompression;//位图的压缩类型,必须是0-不压缩,1-BI_RLE8压缩类型或2-BI_RLE4压缩类型之一  
  70.   unsigned long biSizeImage;//位图大小,以字节为单位  
  71.   unsigned long biXPelsPerMeter;//位图目标设备水平分辨率,以每米像素数为单位  
  72.   unsigned long biYPelsPerMeter;//位图目标设备垂直分辨率,以每米像素数为单位  
  73.   unsigned long biClrUsed;//位图实际使用的颜色表中的颜色变址数  
  74.   unsigned long biClrImportant;//位图显示过程中被认为重要颜色的变址数  
  75. } BITMAPINFOHEADER;  
  76.   
  77.   
  78.   
  79.   
  80. VideoBuffer framebuf[BUFFER_COUNT];   //修改了错误,2012-5.21  
  81. int fd;  
  82. struct v4l2_capability cap;  
  83. struct v4l2_fmtdesc fmtdesc;  
  84. struct v4l2_format fmt;  
  85. struct v4l2_requestbuffers reqbuf;  
  86. struct v4l2_buffer buf;  
  87. unsigned char *starter;  
  88. unsigned char *newBuf;  
  89. struct BITMAPFILEHEADER bfh;  
  90. struct BITMAPINFOHEADER bih;  
  91.   
  92.   
  93. void create_bmp_header()  
  94. {  
  95.   bfh.bfType = (unsigned short)0x4D42;  
  96.   bfh.bfSize = (unsigned long)(14 + 40 + VIDEO_WIDTH * VIDEO_HEIGHT*3);  
  97.   bfh.bfReserved1 = 0;  
  98.   bfh.bfReserved2 = 0;  
  99.   bfh.bfOffBits = (unsigned long)(14 + 40);  
  100.   
  101.   
  102.   bih.biBitCount = 24;  
  103.   bih.biWidth = VIDEO_WIDTH;  
  104.   bih.biHeight = VIDEO_HEIGHT;  
  105.   bih.biSizeImage = VIDEO_WIDTH * VIDEO_HEIGHT * 3;  
  106.   bih.biClrImportant = 0;  
  107.   bih.biClrUsed = 0;  
  108.   bih.biCompression = 0;  
  109.   bih.biPlanes = 1;  
  110.   bih.biSize = 40;//sizeof(bih);  
  111.   bih.biXPelsPerMeter = 0x00000ec4;  
  112.   bih.biYPelsPerMeter = 0x00000ec4;  
  113. }  
  114.   
  115.   
  116. int open_device()  
  117. {  
  118. /*  
  119. 在linux下设备都是以文件的形式进行管理的  
  120. ioctl是设备驱动程序中对设备的I/O通道进行管理的函数int ioctl(int fd,int cmd,...)?  
  121. 成功返回0,出错返回-1  
  122. 其中fd--就是用户程序打开设备使用open函数返回的文件标识符  
  123.     cmd--就是用户程序对设备的控制命令,至于后面都省略号,有或没有和cmd的意义相关  
  124. */  
  125.     int fd;  
  126.     fd = open(CAMERA_DEVICE, O_RDWR, 0);//  
  127.     if (fd < 0) {  
  128.         printf("Open %s failed\n", CAMERA_DEVICE);  
  129.         return -1;  
  130.     }  
  131.     return fd;  
  132. }  
  133.   
  134.   
  135. void get_capability()  
  136. {// 获取驱动信息  
  137. /*  
  138. 控制命令VIDIOC_QUERYCAP  
  139. 功能:查询设备驱动的功能;  
  140. 参数说明:参数类型为V4L2的能力描述类型struct v4l2_capability;  
  141. struct v4l2_capability {  
  142.         __u8    driver[16];     //i.e. "bttv"            //驱动名称,  
  143.         __u8    card[32];       // i.e. "Hauppauge WinTV"         //  
  144.         __u8    bus_info[32];   // "PCI:" + pci_name(pci_dev)     //PCI总线信息  
  145.         __u32   version;        // should use KERNEL_VERSION()   
  146.         __u32   capabilities;   // Device capabilities         //设备能力  
  147.         __u32   reserved[4];  
  148. };  
  149. 返回值说明: 执行成功时,函数返回值为 0;  
  150. 函数执行成功后,struct v4l2_capability 结构体变量中的返回当前视频设备所支持的功能  
  151. 例如支持视频捕获功能V4L2_CAP_VIDEO_CAPTURE或V4L2_CAP_STREAMING  
  152. */  
  153.     int ret = ioctl(fd, VIDIOC_QUERYCAP, &cap);  
  154.     if (ret < 0) {  
  155.         printf("VIDIOC_QUERYCAP failed (%d)\n", ret);  
  156.         return;  
  157.     }  
  158.     // Print capability infomations  
  159.     printf("------------VIDIOC_QUERYCAP-----------\n");  
  160.     printf("Capability Informations:\n");  
  161.     printf(" driver: %s\n", cap.driver);  
  162.     printf(" card: %s\n", cap.card);  
  163.     printf(" bus_info: %s\n", cap.bus_info);  
  164.     printf(" version: %08X\n", cap.version);  
  165.     printf(" capabilities: %08X\n\n", cap.capabilities);  
  166.     return;  
  167. }  
  168.   
  169.   
  170. void get_format()  
  171. {  
  172. /*获取当前视频设备支持的视频格式  
  173. 控制命令 VIDIOC_ENUM_FMT  
  174. 功能: 获取当前视频设备支持的视频格式 。  
  175. 参数说明:参数类型为V4L2的视频格式描述符类型 struct v4l2_fmtdesc  
  176. struct v4l2_fmtdesc {  
  177.         __u32               index;             // Format number        
  178.         enum v4l2_buf_type  type;              // buffer type          
  179.         __u32               flags;  
  180.         __u8                description[32];   // Description string   
  181.         __u32               pixelformat;       // Format fourcc        
  182.         __u32               reserved[4];  
  183. };  
  184. 返回值说明: 执行成功时,函数返回值为 0;  
  185. */  
  186.     int ret;  
  187.     fmtdesc.index=0;  
  188.     fmtdesc.type = V4L2_BUF_TYPE_VIDEO_CAPTURE;  
  189.     ret=ioctl(fd, VIDIOC_ENUM_FMT, &fmtdesc);  
  190.     while (ret != 0)  
  191.     {  
  192.         fmtdesc.index++;  
  193.         ret=ioctl(fd, VIDIOC_ENUM_FMT, &fmtdesc);  
  194.     }  
  195.     printf("--------VIDIOC_ENUM_FMT---------\n");  
  196.     printf("get the format what the device support\n{ pixelformat = ''%c%c%c%c'', description = ''%s'' }\n",fmtdesc.pixelformat & 0xFF, (fmtdesc.pixelformat >> 8) & 0xFF, (fmtdesc.pixelformat >> 16) & 0xFF,(fmtdesc.pixelformat >> 24) & 0xFF, fmtdesc.description);  
  197.       
  198.     return;  
  199. }  
  200.   
  201.   
  202. int set_format()  
  203. {  
  204. /*  
  205. 控制命令VIDIOC_S_FMT  
  206. 功能:设置视频设备的视频数据格式,例如设置视频图像数据的长、宽,图像格式JPEG、YUYV格式);  
  207. 参数说明:参数类型为V4L2的视频数据格式类型struct v4l2_format;  
  208. struct v4l2_format {  
  209.         enum v4l2_buf_type type;    //数据流类型,必须永远是V4L2_BUF_TYPE_VIDEO_CAPTURE  
  210.         union {  
  211.                 struct v4l2_pix_format          pix;     // V4L2_BUF_TYPE_VIDEO_CAPTURE   
  212.                 struct v4l2_window              win;     // V4L2_BUF_TYPE_VIDEO_OVERLAY   
  213.                 struct v4l2_vbi_format          vbi;     // V4L2_BUF_TYPE_VBI_CAPTURE   
  214.                 struct v4l2_sliced_vbi_format   sliced;  // V4L2_BUF_TYPE_SLICED_VBI_CAPTURE   
  215.                 __u8    raw_data[200];                   // user-defined   
  216.         } fmt;  
  217. };  
  218. struct v4l2_pix_format {  
  219.         __u32                   width;         // 宽,必须是16的倍数  
  220.         __u32                   height;        // 高,必须是16的倍数  
  221.         __u32                   pixelformat;   // 视频数据存储类型,例如是YUV4:2:2还是RGB  
  222.         enum v4l2_field       field;  
  223.         __u32                   bytesperline;  
  224.         __u32                   sizeimage;  
  225.         enum v4l2_colorspace colorspace;  
  226.         __u32                   priv;  
  227. };  
  228. 返回值说明: 执行成功时,函数返回值为 0;  
  229.   
  230.   
  231.   
  232.   
  233. 注意:如果该视频设备驱动不支持你所设定的图像格式,视频驱动会重新修改struct v4l2_format结构体变量的值为该视频设备所支持的图像格式,所以在程序设计中,设定完所有的视频格式后,要获取实际的视频格式,要重新读取 struct v4l2_format结构体变量。  
  234. 使用VIDIOC_G_FMT设置视频设备的视频数据格式,VIDIOC_TRY_FMT验证视频设备的视频数据格式  
  235. */  
  236.     fmt.type                = V4L2_BUF_TYPE_VIDEO_CAPTURE;  
  237.     fmt.fmt.pix.width       = VIDEO_WIDTH;  
  238.     fmt.fmt.pix.height      = VIDEO_HEIGHT;  
  239.     fmt.fmt.pix.pixelformat = fmtdesc.pixelformat;//V4L2_PIX_FMT_YUYV;  
  240.     fmt.fmt.pix.field       = V4L2_FIELD_INTERLACED;  
  241.     int ret = ioctl(fd, VIDIOC_S_FMT, &fmt);  
  242.     if (ret < 0) {  
  243.         printf("VIDIOC_S_FMT failed (%d)\n", ret);  
  244.         return;  
  245.     }  
  246.   
  247.   
  248.   /*  // 设置视频格式VIDIOC_G_FMT和VIDIOC_S_FMT相同  
  249.     ret = ioctl(fd, VIDIOC_G_FMT, &fmt);  
  250.     if (ret < 0) {  
  251.         printf("VIDIOC_G_FMT failed (%d)\n", ret);  
  252.         return ret;  
  253.     }*/  
  254.     // Print Stream Format  
  255.     printf("------------VIDIOC_S_FMT---------------\n");  
  256.     printf("Stream Format Informations:\n");  
  257.     printf(" type: %d\n", fmt.type);  
  258.     printf(" width: %d\n", fmt.fmt.pix.width);  
  259.     printf(" height: %d\n", fmt.fmt.pix.height);  
  260.   
  261.   
  262.     char fmtstr[8];  
  263.     memset(fmtstr, 08);  
  264. /*  
  265. void *memcpy(void *dest, const void *src, size_t n);  
  266. 从源src所指的内存地址的起始位置开始拷贝n个字节到目标dest所指的内存地址的起始位置中  
  267. 所需头文件include <string.h>  
  268. */  
  269.     memcpy(fmtstr, &fmt.fmt.pix.pixelformat, 4);  
  270.     printf(" pixelformat: %s\n", fmtstr);  
  271.     printf(" field: %d\n", fmt.fmt.pix.field);  
  272.     printf(" bytesperline: %d\n", fmt.fmt.pix.bytesperline);  
  273.     printf(" sizeimage: %d\n", fmt.fmt.pix.sizeimage);  
  274.     printf(" colorspace: %d\n", fmt.fmt.pix.colorspace);  
  275.     printf(" priv: %d\n", fmt.fmt.pix.priv);  
  276.     printf(" raw_date: %s\n", fmt.fmt.raw_data);  
  277.     return;  
  278. }  
  279.   
  280.   
  281. void request_buf()  
  282. {  
  283. /*  
  284. 控制命令VIDIOC_REQBUFS  
  285. 功能: 请求V4L2驱动分配视频缓冲区(申请V4L2视频驱动分配内存),V4L2是视频设备的驱动层,位于内核空间,所以通过VIDIOC_REQBUFS控制命令字申请的内存位于内核空间,应用程序不能直接访问,需要通过调用mmap内存映射函数把内核空间内存映射到用户空间后,应用程序通过访问用户空间地址来访问内核空间。  
  286. 参数说明:参数类型为V4L2的申请缓冲区数据结构体类型struct v4l2_requestbuffers;  
  287. struct v4l2_requestbuffers {  
  288.         u32                   count;        //缓存数量,也就是说在缓存队列里保持多少张照片  
  289.         enum v4l2_buf_type    type;         //数据流类型,必须永远是V4L2_BUF_TYPE_VIDEO_CAPTURE  
  290.         enum v4l2_memory      memory;       //V4L2_MEMORY_MMAP或V4L2_MEMORY_USERPTR  
  291.         u32                   reserved[2];  
  292. };  
  293. 返回值说明: 执行成功时,函数返回值为 0,V4L2驱动层分配好了视频缓冲区;  
  294.   
  295.   
  296. 注意:VIDIOC_REQBUFS会修改tV4L2_reqbuf的count值,tV4L2_reqbuf的count值返回实际申请成功的视频缓冲区数目;  
  297. */  
  298.     reqbuf.count = BUFFER_COUNT;  
  299.     reqbuf.type = V4L2_BUF_TYPE_VIDEO_CAPTURE;  
  300.     reqbuf.memory = V4L2_MEMORY_MMAP;  
  301.       
  302.     int ret = ioctl(fd , VIDIOC_REQBUFS, &reqbuf);  
  303.     if(ret < 0) {  
  304.         printf("VIDIOC_REQBUFS failed (%d)\n", ret);  
  305.         return;  
  306.     }  
  307.     printf("the buffer has been assigned successfully!\n");  
  308.     return;  
  309. }  
  310.   
  311.   
  312. void query_map_qbuf()  
  313. {  
  314. /*  
  315. 控制命令VIDIOC_QUERYBUF  
  316. 功能:查询已经分配的V4L2的视频缓冲区的相关信息,包括视频缓冲区的使用状态、在内核空间的偏移地址、缓冲区长度等。在应用程序设计中通过调VIDIOC_QUERYBUF来获取内核空间的视频缓冲区信息,然后调用函数mmap把内核空间地址映射到用户空间,这样应用程序才能够访问位于内核空间的视频缓冲区  
  317. 参数说明:参数类型为V4L2缓冲区数据结构类型struct v4l2_buffer;  
  318. struct v4l2_buffer {  
  319.         __u32                   index;  
  320.         enum v4l2_buf_type      type;  
  321.         __u32                   bytesused;  
  322.         __u32                   flags;  
  323.         enum v4l2_field         field;  
  324.         struct timeval          timestamp;  
  325.         struct v4l2_timecode    timecode;  
  326.         __u32                   sequence;  
  327.         // memory location   
  328.         enum v4l2_memory        memory;  
  329.         union {  
  330.                 __u32           offset;  
  331.                 unsigned long   userptr;  
  332.         } m;  
  333.         __u32                   length;  
  334.         __u32                   input;  
  335.         __u32                   reserved;  
  336. };  
  337. 返回值说明: 执行成功时,函数返回值为 0;  
  338. struct v4l2_buffer结构体变量中保存了指令的缓冲区的相关信息;一般情况下,应用程序中调用VIDIOC_QUERYBUF取得了内核缓冲区信息后,紧接着调用mmap函数把内核空间地址映射到用户空间方便用户空间应用程序的访问  
  339. */  
  340.     int i,ret;  
  341.     for (i = 0; i < reqbuf.count; i++)  
  342.     {  
  343.         buf.index = i;  
  344.         buf.type = V4L2_BUF_TYPE_VIDEO_CAPTURE;  
  345.         buf.memory = V4L2_MEMORY_MMAP;  
  346.         ret = ioctl(fd , VIDIOC_QUERYBUF, &buf);//buf取得内存缓冲区的信息  
  347.         if(ret < 0) {  
  348.             printf("VIDIOC_QUERYBUF (%d) failed (%d)\n", i, ret);  
  349.             return;  
  350.         }  
  351.   
  352.   
  353.         // mmap buffer  
  354.         framebuf[i].length = buf.length;//framebuf是程序最前面定义的一个结构体类型的数据  
  355. /*  
  356. #include <sys/mman.h>  
  357. void *mmap(void *start, size_t length, int prot, int flags,int fd, off_t offset);  
  358. int munmap(void *start, size_t length);  
  359. mmap将一个文件或者其它对象映射进内存。文件被映射到多个页上,如果文件的大小不是所有页的大小之和,最后一个页不被使用的空间将会清零  
  360. start:映射区的开始地址,设置为0时表示由系统决定映射区的起始地址  
  361. length:映射区的长度  
  362. prot:期望的内存保护标志,不能与文件的打开模式冲突。是以下的某个值,可以通过or运算合理地组合在一起    PROT_EXEC //页内容可以被执行  
  363.     PROT_READ //页内容可以被读取  
  364.     PROT_WRITE //页可以被写入  
  365.     PROT_NONE //页不可访问  
  366. flags:指定映射对象的类型,映射选项和映射页是否可以共享  
  367.     MAP_SHARED //与其它所有映射这个对象的进程共享映射空间。对共享区的写入,相当于输出到文件。直到msync()或者munmap()被调用,文件实际上不会被更新  
  368. fd:有效的文件描述词。一般是由open()函数返回,其值也可以设置为-1,此时需要指定flags参数中的MAP_ANON,表明进行的是匿名映射  
  369. offset:被映射对象内容的起点  
  370.   
  371.   
  372. 成功执行时,mmap()返回被映射区的指针,munmap()返回0。失败时,mmap()返回MAP_FAILED[其值为(void *)-1],munmap返回-1  
  373. */  
  374.         framebuf[i].start = (char *) mmap(0, buf.length, PROT_READ|PROT_WRITE, MAP_SHARED, fd, buf.m.offset);  
  375.         if (framebuf[i].start == MAP_FAILED) {  
  376.             printf("mmap (%d) failed: %s\n", i, strerror(errno));  
  377.             return;  
  378.         }  
  379.       
  380.         // Queen buffer  
  381. /*  
  382. 控制命令VIDIOC_QBUF  
  383. 功能:投放一个空的视频缓冲区到视频缓冲区输入队列中  
  384. 参数说明:参数类型为V4L2缓冲区数据结构类型struct v4l2_buffer;  
  385. 返回值说明: 执行成功时,函数返回值为 0;  
  386. 函数执行成功后,指令(指定)的视频缓冲区进入视频输入队列,在启动视频设备拍摄图像时,相应的视频数据被保存到视频输入队列相应的视频缓冲区中  
  387. */  
  388.         ret = ioctl(fd , VIDIOC_QBUF, &buf);  
  389.         if (ret < 0) {  
  390.             printf("VIDIOC_QBUF (%d) failed (%d)\n", i, ret);  
  391.             return;  
  392.         }  
  393.   
  394.   
  395.         printf("Frame buffer %d: address=0x%x, length=%d\n", i, (unsigned int)framebuf[i].start, framebuf[i].length);  
  396.     }//空的视频缓冲区都已经在视频缓冲的输入队列中了  
  397.     return;  
  398. }  
  399.   
  400.   
  401.   
  402.   
  403. void yuyv2rgb()  
  404. {  
  405.     unsigned char YUYV[4],RGB[6];  
  406.     int j,k,i;     
  407.     unsigned int location=0;  
  408.     j=0;  
  409.     for(i=0;i < framebuf[buf.index].length;i+=4)  
  410.     {  
  411.         YUYV[0]=starter[i];//Y0  
  412.         YUYV[1]=starter[i+1];//U  
  413.         YUYV[2]=starter[i+2];//Y1  
  414.         YUYV[3]=starter[i+3];//V  
  415.         if(YUYV[0]<1)  
  416.         {  
  417.             RGB[0]=0;  
  418.             RGB[1]=0;  
  419.             RGB[2]=0;  
  420.         }  
  421.         else  
  422.         {  
  423.             RGB[0]=YUYV[0]+1.772*(YUYV[1]-128);//b  
  424.             RGB[1]=YUYV[0]-0.34413*(YUYV[1]-128)-0.71414*(YUYV[3]-128);//g  
  425.             RGB[2]=YUYV[0]+1.402*(YUYV[3]-128);//r  
  426.         }  
  427.         if(YUYV[2]<0)  
  428.         {  
  429.             RGB[3]=0;  
  430.             RGB[4]=0;  
  431.             RGB[5]=0;  
  432.         }  
  433.         else  
  434.         {  
  435.             RGB[3]=YUYV[2]+1.772*(YUYV[1]-128);//b  
  436.             RGB[4]=YUYV[2]-0.34413*(YUYV[1]-128)-0.71414*(YUYV[3]-128);//g  
  437.             RGB[5]=YUYV[2]+1.402*(YUYV[3]-128);//r  
  438.   
  439.   
  440.         }  
  441.   
  442.   
  443.         for(k=0;k<6;k++)  
  444.         {  
  445.             if(RGB[k]<0)  
  446.                 RGB[k]=0;  
  447.             if(RGB[k]>255)  
  448.                 RGB[k]=255;  
  449.         }  
  450.   
  451.   
  452.         //请记住:扫描行在位图文件中是反向存储的!  
  453.         if(j%(VIDEO_WIDTH*3)==0)//定位存储位置  
  454.         {  
  455.             location=(VIDEO_HEIGHT-j/(VIDEO_WIDTH*3))*(VIDEO_WIDTH*3);  
  456.         }  
  457.         bcopy(RGB,newBuf+location+(j%(VIDEO_WIDTH*3)),sizeof(RGB));  
  458.   
  459.   
  460.         j+=6;         
  461.     }  
  462.     return;  
  463. }  
  464.   
  465.   
  466. void move_noise()  
  467. {//双滤波器  
  468.     int i,j,k,temp[3],temp1[3];  
  469.     unsigned char BGR[13*3];  
  470.     unsigned int sq,sq1,loc,loc1;  
  471.     int h=VIDEO_HEIGHT,w=VIDEO_WIDTH;  
  472.     for(i=2;i<h-2;i++)  
  473.     {  
  474.         for(j=2;j<w-2;j++)  
  475.         {  
  476.             memcpy(BGR,newBuf+(i-1)*w*3+3*(j-1),9);  
  477.             memcpy(BGR+9,newBuf+i*w*3+3*(j-1),9);  
  478.             memcpy(BGR+18,newBuf+(i+1)*w*3+3*(j-1),9);  
  479.             memcpy(BGR+27,newBuf+(i-2)*w*3+3*j,3);  
  480.             memcpy(BGR+30,newBuf+(i+2)*w*3+3*j,3);  
  481.             memcpy(BGR+33,newBuf+i*w*3+3*(j-2),3);  
  482.             memcpy(BGR+36,newBuf+i*w*3+3*(j+2),3);  
  483.   
  484.   
  485.             memset(temp,0,4*3);  
  486.               
  487.             for(k=0;k<9;k++)  
  488.             {  
  489.                 temp[0]+=BGR[k*3];  
  490.                 temp[1]+=BGR[k*3+1];  
  491.                 temp[2]+=BGR[k*3+2];  
  492.             }  
  493.             temp1[0]=temp[0];  
  494.             temp1[1]=temp[1];  
  495.             temp1[2]=temp[2];  
  496.             for(k=9;k<13;k++)  
  497.             {  
  498.                 temp1[0]+=BGR[k*3];  
  499.                 temp1[1]+=BGR[k*3+1];  
  500.                 temp1[2]+=BGR[k*3+2];  
  501.             }  
  502.             for(k=0;k<3;k++)  
  503.             {  
  504.                 temp[k]/=9;  
  505.                 temp1[k]/=13;  
  506.             }  
  507.             sq=0xffffffff;loc=0;  
  508.             sq1=0xffffffff;loc1=0;  
  509.             unsigned int a;           
  510.             for(k=0;k<9;k++)  
  511.             {  
  512.                 a=abs(temp[0]-BGR[k*3])+abs(temp[1]-BGR[k*3+1])+abs(temp[2]-BGR[k*3+2]);  
  513.                 if(a<sq)  
  514.                 {  
  515.                     sq=a;  
  516.                     loc=k;  
  517.                 }  
  518.             }  
  519.             for(k=0;k<13;k++)  
  520.             {  
  521.                 a=abs(temp1[0]-BGR[k*3])+abs(temp1[1]-BGR[k*3+1])+abs(temp1[2]-BGR[k*3+2]);  
  522.                 if(a<sq1)  
  523.                 {  
  524.                     sq1=a;  
  525.                     loc1=k;  
  526.                 }  
  527.             }  
  528.               
  529.             newBuf[i*w*3+3*j]=(unsigned char)((BGR[3*loc]+BGR[3*loc1])/2);  
  530.             newBuf[i*w*3+3*j+1]=(unsigned char)((BGR[3*loc+1]+BGR[3*loc1+1])/2);  
  531.             newBuf[i*w*3+3*j+2]=(unsigned char)((BGR[3*loc+2]+BGR[3*loc1+2])/2);  
  532.             /*还是有些许的噪点  
  533.             temp[0]=(BGR[3*loc]+BGR[3*loc1])/2;  
  534.             temp[1]=(BGR[3*loc+1]+BGR[3*loc1+1])/2;  
  535.             temp[2]=(BGR[3*loc+2]+BGR[3*loc1+2])/2;  
  536.             sq=abs(temp[0]-BGR[loc*3])+abs(temp[1]-BGR[loc*3+1])+abs(temp[2]-BGR[loc*3+2]);  
  537.             sq1=abs(temp[0]-BGR[loc1*3])+abs(temp[1]-BGR[loc1*3+1])+abs(temp[2]-BGR[loc1*3+2]);  
  538.             if(sq1<sq) loc=loc1;  
  539.             newBuf[i*w*3+3*j]=BGR[3*loc];  
  540.             newBuf[i*w*3+3*j+1]=BGR[3*loc+1];  
  541.             newBuf[i*w*3+3*j+2]=BGR[3*loc+2];*/  
  542.         }  
  543.     }  
  544.     return;  
  545. }  
  546.   
  547.   
  548. void yuyv2rgb1()  
  549. {  
  550.     unsigned char YUYV[3],RGB[3];  
  551.     memset(YUYV,0,3);  
  552.     int j,k,i;     
  553.     unsigned int location=0;  
  554.     j=0;  
  555.     for(i=0;i < framebuf[buf.index].length;i+=2)  
  556.     {  
  557.         YUYV[0]=starter[i];//Y0  
  558.         if(i%4==0)  
  559.             YUYV[1]=starter[i+1];//U  
  560.         //YUYV[2]=starter[i+2];//Y1  
  561.         if(i%4==2)  
  562.             YUYV[2]=starter[i+1];//V  
  563.         if(YUYV[0]<1)  
  564.         {  
  565.             RGB[0]=0;  
  566.             RGB[1]=0;  
  567.             RGB[2]=0;  
  568.         }  
  569.         else  
  570.         {  
  571.             RGB[0]=YUYV[0]+1.772*(YUYV[1]-128);//b  
  572.             RGB[1]=YUYV[0]-0.34413*(YUYV[1]-128)-0.71414*(YUYV[2]-128);//g  
  573.             RGB[2]=YUYV[0]+1.402*(YUYV[2]-128);//r  
  574.         }  
  575.   
  576.   
  577.         for(k=0;k<3;k++)  
  578.         {  
  579.             if(RGB[k]<0)  
  580.                 RGB[k]=0;  
  581.             if(RGB[k]>255)  
  582.                 RGB[k]=255;  
  583.         }  
  584.   
  585.   
  586.         //请记住:扫描行在位图文件中是反向存储的!  
  587.         if(j%(VIDEO_WIDTH*3)==0)//定位存储位置  
  588.         {  
  589.             location=(VIDEO_HEIGHT-j/(VIDEO_WIDTH*3))*(VIDEO_WIDTH*3);  
  590.         }  
  591.         bcopy(RGB,newBuf+location+(j%(VIDEO_WIDTH*3)),sizeof(RGB));  
  592.   
  593.   
  594.         j+=3;         
  595.     }  
  596.     return;  
  597. }  
  598.   
  599.   
  600. void store_yuyv()  
  601. {  
  602.     FILE *fp = fopen(CAPTURE_FILE, "wb");  
  603.     if (fp < 0) {  
  604.         printf("open frame data file failed\n");  
  605.         return;  
  606.     }  
  607.     fwrite(framebuf[buf.index].start, 1, buf.length, fp);  
  608.     fclose(fp);  
  609.     printf("Capture one frame saved in %s\n", CAPTURE_FILE);  
  610.     return;  
  611. }  
  612.   
  613.   
  614.   
  615.   
  616. void store_bmp(int n_len)  
  617. {  
  618.     FILE *fp1 = fopen(CAPTURE_RGB_FILE, "wb");  
  619.     if (fp1 < 0) {  
  620.         printf("open frame data file failed\n");  
  621.         return;  
  622.     }  
  623.     fwrite(&bfh,sizeof(bfh),1,fp1);  
  624.     fwrite(&bih,sizeof(bih),1,fp1);  
  625.     fwrite(newBuf, 1, n_len, fp1);  
  626.     fclose(fp1);  
  627.     printf("Change one frame saved in %s\n", CAPTURE_RGB_FILE);  
  628.     return;  
  629. }  
  630.   
  631.   
  632. int main()  
  633. {  
  634.     int i, ret;  
  635.   
  636.   
  637.     // 打开设备  
  638.     fd=open_device();  
  639.       
  640.     // 获取驱动信息  
  641.     //struct v4l2_capability cap;  
  642.     get_capability();  
  643.      
  644.     //获取当前视频设备支持的视频格式  
  645.     //struct v4l2_fmtdesc fmtdesc;  
  646.     memset(&fmtdesc,0,sizeof(fmtdesc));  
  647.     get_format();  
  648.       
  649.     // 设置视频格式  
  650.     //struct v4l2_format fmt;  
  651.     //memset在一段内存块中填充某个给定的值,它是对较大的结构体或数组进行清零操作的一种最快的方法  
  652.     memset(&fmt, 0, sizeof(fmt));//将fmt中的前sizeof(fmt)字节用0替换并返回fmt  
  653.     set_format();  
  654.       
  655.     // 请求分配内存  
  656.     //struct v4l2_requestbuffers reqbuf;  
  657.     request_buf();  
  658.   
  659.   
  660.     // 获取空间,并将其映射到用户空间,然后投放到视频输入队列  
  661.     //struct v4l2_buffer buf;  
  662.     query_map_qbuf();  
  663.   
  664.   
  665.   
  666.   
  667.     // 开始录制  
  668.     enum v4l2_buf_type type = V4L2_BUF_TYPE_VIDEO_CAPTURE;  
  669. /*  
  670. 控制命令VIDIOC_STREAMON  
  671. 功能:启动视频采集命令,应用程序调用VIDIOC_STREAMON启动视频采集命令后,视频设备驱动程序开始采集视频数据,并把采集到的视频数据保存到视频驱动的视频缓冲区中  
  672. 参数说明:参数类型为V4L2的视频缓冲区类型 enum v4l2_buf_type ;  
  673. enum v4l2_buf_type {  
  674.         V4L2_BUF_TYPE_VIDEO_CAPTURE        = 1,  
  675.         V4L2_BUF_TYPE_VIDEO_OUTPUT         = 2,  
  676.         V4L2_BUF_TYPE_VIDEO_OVERLAY        = 3,  
  677.         V4L2_BUF_TYPE_VBI_CAPTURE          = 4,  
  678.         V4L2_BUF_TYPE_VBI_OUTPUT           = 5,  
  679.         V4L2_BUF_TYPE_SLICED_VBI_CAPTURE   = 6,  
  680.         V4L2_BUF_TYPE_SLICED_VBI_OUTPUT    = 7,  
  681. #if 1  
  682.          Experimental   
  683.         V4L2_BUF_TYPE_VIDEO_OUTPUT_OVERLAY = 8,  
  684. #endif  
  685.         V4L2_BUF_TYPE_PRIVATE              = 0x80,  
  686. };  
  687. 返回值说明: 执行成功时,函数返回值为 0;函数执行成功后,视频设备驱动程序开始采集视频数据,此时应用程序一般通过调用select函数来判断一帧视频数据是否采集完成,当视频设备驱动完成一帧视频数据采集并保存到视频缓冲区中时,select函数返回,应用程序接着可以读取视频数据;否则select函数阻塞直到视频数据采集完成  
  688. */  
  689.     ret = ioctl(fd, VIDIOC_STREAMON, &type);  
  690.     if (ret < 0) {  
  691.         printf("VIDIOC_STREAMON failed (%d)\n", ret);  
  692.         return ret;  
  693.     }  
  694.   
  695.   
  696.     // Get frame  
  697. /*  
  698. 控制命令VIDIOC_DQBUF  
  699. 功能:从视频缓冲区的输出队列中取得一个已经保存有一帧视频数据的视频缓冲区  
  700. 参数说明:参数类型为V4L2缓冲区数据结构类型struct v4l2_buffer;  
  701. 返回值说明: 执行成功时,函数返回值为 0;函数执行成功后,相应的内核视频缓冲区中保存有当前拍摄到的视频数据,应用程序可以通过访问用户空间来读取该视频数据(前面已经通过调用函数 mmap做了用户空间和内核空间的内存映射).  
  702.   
  703.   
  704. 说明: VIDIOC_DQBUF命令结果, 使从队列删除的缓冲帧信息传给了此buf  
  705. V4L2_buffer结构体的作用就相当于申请的缓冲帧的代理,找缓冲帧的都要先问问它,通过它来联系缓冲帧,起了中间桥梁的作用  
  706. */  
  707.     ret = ioctl(fd, VIDIOC_DQBUF, &buf);//VIDIOC_DQBUF命令结果, 使从队列删除的缓冲帧信息传给了此buf  
  708.     if (ret < 0) {  
  709.         printf("VIDIOC_DQBUF failed (%d)\n", ret);  
  710.         return ret;  
  711.     }  
  712.   
  713.   
  714.     // Process the frame 此时我们需要进行数据格式的改变  
  715.     store_yuyv();  
  716.       
  717.       
  718.     //对采集的数据进行转变,变换成RGB24模式,然后进行存储  
  719. /*  
  720. 1)开辟出来一段内存区域来存放转换后的数据  
  721. 2)循环读取buf内存段的内容,进行转换,转换后放入到新开辟的内存区域中  
  722. 3)将新开辟出来的内存区的内容读到文件中  
  723. */  
  724.     printf("********************************************\n");  
  725.     int n_len;  
  726.     n_len=framebuf[buf.index].length*3/2;  
  727.     newBuf=calloc((unsigned int)n_len,sizeof(unsigned char));  
  728.     
  729.     if(!newBuf)  
  730.     {  
  731.         printf("cannot assign the memory !\n");  
  732.         exit(0);  
  733.     }  
  734.   
  735.   
  736.     printf("the information about the new buffer:\n start Address:0x%x,length=%d\n\n",(unsigned int)newBuf,n_len);  
  737.   
  738.   
  739.     printf("----------------------------------\n");  
  740.       
  741.     //YUYV to RGB  
  742.     starter=(unsigned char *)framebuf[buf.index].start;  
  743.     yuyv2rgb();//还是这个采集的图片的效果比较好  
  744.     move_noise();  
  745.     //yuyv2rgb1();  
  746.     //设置bmp文件的头和bmp文件的一些信息  
  747.     create_bmp_header();  
  748.       
  749.     store_bmp(n_len);  
  750.       
  751.       
  752.     // Re-queen buffer  
  753.     ret = ioctl(fd, VIDIOC_QBUF, &buf);  
  754.     if (ret < 0) {  
  755.         printf("VIDIOC_QBUF failed (%d)\n", ret);  
  756.         return ret;  
  757.     }  
  758.     printf("re-queen buffer end\n");  
  759.     // Release the resource  
  760. /*  
  761. 表头文件 #include<unistd.h>  
  762.         #include<sys/mman.h>  
  763.         定义函数 int munmap(void *start,size_t length);  
  764.         函数说明 munmap()用来取消参数start所指的映射内存起始地址,参数length则是欲取消的内存大小。当进程结束或利用exec相关函数来执行其他程序时,映射内存会自动解除,但关闭对应的文件描述词时不会解除映射  
  765.         返回值 如果解除映射成功则返回0,否则返回-1  
  766. */  
  767.     for (i=0; i< 4; i++)  
  768.     {  
  769.       
  770.         munmap(framebuf[i].start, framebuf[i].length);  
  771.     }  
  772.     //free(starter);  
  773. printf("free starter end\n");  
  774.     //free(newBuf);  
  775. printf("free newBuf end\n");  
  776.     close(fd);  
  777.   
  778.   
  779.       
  780.     printf("Camera test Done.\n");  
  781.     return 0;  
  782. }




FROM:  http://blog.csdn.net/bg2bkk/article/details/8946429

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值