Linux下C语言实现LCD屏幕截图

From: http://blog.chinaunix.net/uid-24789420-id-3191806.html

一、概述

        最近看到网上有人问怎么用C语言实现屏幕截图，刚好自己也在研究Linux驱动，于是花了半天时间把Linux的FrameBuffer驱动看懂了个七八，接着就动手写了个LCD屏幕截图的应用程序。

     一些说明：

     1. 板子LCD的分辨率为480x272，像素格式为RGB565。

     2. 程序首先获取LCD的一些参数 ，然后将获取到的一帧数据转换为RGB24格式，最后用jpeglib库压缩为jpg格式的图片。

　  3. 在嵌入式Linux的板子上测试通过。

 

二、实现

       程序中有注释，不多说，直接上程序：

#include <stdio.h>
#include <stdlib.h>
#include <sys/ioctl.h>
#include <unistd.h>
#include <fcntl.h>
#include <malloc.h>
#include <linux/fb.h>
#include <jpeglib.h>
#include <jerror.h>


/*RGB565转RGB24函数
 *@rgb565: 指向存放rgb565数据的起始地址
 *@rgb24：指向存放rgb24数据的起始地址
 *@width：屏幕(分辨率)的宽度
 *@height：屏幕(分辨率)的高度
 */
int RGB565_to_RGB24(unsigned char *rgb565, unsigned char *rgb24, int width, int height)
{
	int i;
	int whole = width*height;
	unsigned char r, g, b;
	unsigned short int *pix565;

	pix565 = (unsigned short int *)rgb565;

	for(i = 0;i < whole;i++)
	{    
		r = ((*pix565)>>11)&0x1f;
		*rgb24 = (r<<3) | (r>>2);
		rgb24++;
		g = ((*pix565)>>5)&0x3f;
		*rgb24 = (g<<2) | (g>>4);
		rgb24++;
		b = (*pix565)&0x1f;
		*rgb24 = (b<<3) | (b>>2);
		rgb24++;
		pix565++;    

	}
	return 1;
}

/*jpeg压缩函数
 *@rgb：指向存放rgb24数据的起始地址
 *@width：屏幕(分辨率)的宽度
 *@height：屏幕(分辨率)的高度
*/
int jpeg_compress(unsigned char *rgb, int width, int height)
{
	struct jpeg_compress_struct cinfo;
	struct jpeg_error_mgr jerr;
	FILE * outfile;
	JSAMPROW row_pointer[1];
	int row_stride;
	cinfo.err = jpeg_std_error(&jerr);
	jpeg_create_compress(&cinfo);
	// 输出文件名为：out.jpg
	if ((outfile = fopen("out.jpg", "wb")) == NULL)
	{
		printf("can not open out.jpg\n");
		return -1;
	}
	jpeg_stdio_dest(&cinfo, outfile);

	cinfo.image_width = width;
	cinfo.image_height = height;
	cinfo.input_components = 3;				// 1-灰度图，3-彩色图
	// 输入数据格式为RGB
	cinfo.in_color_space = JCS_RGB;			// JCS_GRAYSCALE-灰度图，JCS_RGB-彩色图

	jpeg_set_defaults(&cinfo);
	jpeg_set_quality(&cinfo, 80, TRUE);		// 设置压缩质量：80
	jpeg_start_compress(&cinfo, TRUE);		// 开始压缩过程
	row_stride = width * 3;					// row_stride: 每一行的字节数

	while (cinfo.next_scanline < cinfo.image_height)
	{
		row_pointer[0] = &rgb[cinfo.next_scanline * row_stride];
		(void) jpeg_write_scanlines(&cinfo, row_pointer, 1);
	}

	jpeg_finish_compress(&cinfo);			// 完成压缩过程
	fclose(outfile);

	jpeg_destroy_compress(&cinfo);			// 释放资源

	return 1;
}


int main()
{
	int fd;
	struct fb_var_screeninfo fb_var_info;
	struct fb_fix_screeninfo fb_fix_info;
	unsigned char *trgb;
	unsigned char *rgb;
	int buffer_size;

	// 打开framebuffer设备
	fd = open("/dev/fb0", O_RDONLY);
	if(fd < 0)
	{
		printf("can not open dev\n");
		exit(1);
	}

	// 获取LCD的可变参数
	ioctl(fd, FBIOGET_VSCREENINFO, &fb_var_info);
	// 一个像素多少位    
	printf("bits_per_pixel: %d\n", fb_var_info.bits_per_pixel);
	// x分辨率
	printf("xres: %d\n", fb_var_info.xres);
	// y分辨率
	printf("yres: %d\n", fb_var_info.yres);
	// r分量长度(bit)
	printf("red_length: %d\n", fb_var_info.red.length);
	// g分量长度(bit)
	printf("green_length: %d\n", fb_var_info.green.length);
	// b分量长度(bit)
	printf("blue_length: %d\n", fb_var_info.blue.length);
	// t(透明度)分量长度(bit)
	printf("transp_length: %d\n", fb_var_info.transp.length);
	// r分量偏移
	printf("red_offset: %d\n", fb_var_info.red.offset);
	// g分量偏移
	printf("green_offset: %d\n", fb_var_info.green.offset);
	// b分量偏移
	printf("blue_offset: %d\n", fb_var_info.blue.offset);
	// t分量偏移
	printf("transp_offset: %d\n", fb_var_info.transp.offset);

	// 获取LCD的固定参数
	ioctl(fd, FBIOGET_FSCREENINFO, &fb_fix_info);
	// 一帧大小
	printf("smem_len: %d\n", fb_fix_info.smem_len);
	// 一行大小
	printf("line_length: %d\n", fb_fix_info.line_length);
	// 一帧大小
	buffer_size = (fb_var_info.xres * fb_var_info.yres * fb_var_info.bits_per_pixel / 8);

	trgb = (unsigned char *)malloc(buffer_size);
	if(trgb == NULL)
		exit(0);
	rgb = (unsigned char *)malloc(fb_var_info.xres * fb_var_info.yres * 3);
	if(rgb == NULL)
	{
		goto here;
	}
	if(read(fd, trgb, buffer_size) < 0)		// 获取一帧数据
	{
		printf("read failed!\n");
		goto read_fail;
	}
	// 格式转换
	RGB565_to_RGB24(trgb, rgb, fb_var_info.xres, fb_var_info.yres);
	// jpeg压缩
	if(jpeg_compress(rgb, fb_var_info.xres, fb_var_info.yres) < 0)
		printf("compress failed!\n");        

read_fail:
	free(rgb);
here:
	free(trgb);

	close(fd);

	return 0;
}

注意：编译时加上" -ljpeg"