图像解码之二——使用libpng解码png图片

目录(?)[+]libpng的数据结构
libpng的使用
判断是否为libpng数据
初始化libpng
创建图像信息png_infop变量
设置错误返回点
设置libpng的数据源
png图像处理
高层处理
底层处理
释放libpng的内存
总结
原文地址::http://my.unix-center.net/~Simon_fu/?p=1030

上文《图像解码之一——使用libjpeg解码jpeg图片》介绍了使用libjpeg解码jpeg图片。png图片应用也非常广泛，本文将会简单介绍怎样使用开源libpng库解码png图片。

libpng的数据结构

png_structp变量是在libpng初始化的时候创建，由libpng库内部使用，代表libpng的是调用上下文，库的使用者不应该对这个变量进行访问。调用libpng的API的时候，需要把这个参数作为第一个参数传入。 png_infop变量，初始化完成libpng之后，可以从libpng中获得该类型变量指针。这个变量保存了png图片数据的信息，库的使用者可以修改和查阅该变量，比如：查阅图片信息，修改图片解码参数。在早期的版本中直接访问该变量的成员，最新的版本建议是通过API来访问这些成员。

libpng的使用

0、判断是否为libpng数据

这步是可选的，在利用libpng继续数据处理之前，可以调用png_sig_cmp函数来检查是否为png数据，请参阅libpng手册了解详细内容。

1、初始化libpng

1: /* Create and initialize the png_struct with the desired error handler

2:  * functions.  If you want to use the default stderr and longjump method,

3:  * you can supply NULL for the last three parameters.  We also supply the

4:  * the compiler header file version, so that we know if the application

5:  * was compiled with a compatible version of the library.  REQUIRED

6:  */

7: png_ptr = png_create_read_struct(PNG_LIBPNG_VER_STRING,

8:    png_voidp user_error_ptr, user_error_fn, user_warning_fn);

初始化libpng的时候，用户可以指定自定义错误处理函数，如果不需要指定自定义错误处理函数，则传NULL即可。png_create_read_struct函数返回一个png_structp变量，前面已经提到该变量不应该被用户访问，应该在以后调用libpng的函数时传递给libpng库。 如果你需要提供自定义内存管理模块则需要调用png_create_read_struct_2来完成对libpng的初始化：

1: png_structp png_ptr = png_create_read_struct_2

2: (PNG_LIBPNG_VER_STRING, (png_voidp)user_error_ptr,

3: user_error_fn, user_warning_fn, (png_voidp)

4: user_mem_ptr, user_malloc_fn, user_free_fn)

2、创建图像信息——png_infop变量

1: /* Allocate/initialize the memory for image information.  REQUIRED. */

2: info_ptr = png_create_info_struct(png_ptr);

3: if (info_ptr == NULL)

4: {

5:    png_destroy_read_struct(&png_ptr, png_infopp_NULL, png_infopp_NULL);

6:    return (ERROR);

7: }

如前面所说，用户将会通过png_infop变量来获得图片的信息，设置图片解码参数等。

3、设置错误返回点

上文libjpeg解码jpeg图片中提到用setjmp/longjmp函数来处理异常。libpng库默认集成这种机制来完成异常处理，如下代码初始化错误返回点：

1: /* Set error handling if you are using the setjmp/longjmp method (this is

2:  * the normal method of doing things with libpng).  REQUIRED unless you

3:  * set up your own error handlers in the png_create_read_struct() earlier.

4:  */

5: if (setjmp(png_jmpbuf(png_ptr)))

6: {

7:    /* Free all of the memory associated with the png_ptr and info_ptr */

8:    png_destroy_read_struct(&png_ptr, &info_ptr, png_infopp_NULL);

9:    /* If we get here, we had a problem reading the file */

10:    return (ERROR);

11: }

正如上面注释中提到情况，只有在初始化libpng的时候未指定用户自定义的错误处理函数情况下，才需要设置错误返回点。如果设置了用户自定义的错误处理函数，libpng将会调用用户自定义错误处理函数，而不会返回到这个调用点。 当libpng库出现错误的时候，libpng将会自动调用longjmp函数返回到这个点。在这个点我们可以进行必要的清理工作。

4、设置libpng的数据源

我在上文《图像解码之一——使用libjpeg解码jpeg图片》中提到，一个好的代码库应该能够运行用户输入各式各样的数据，而不能把输入数据定死。libpng在这方面做得非常的好，它提供了默认的文件输入流的支持，并且提供了用户自定义回调函数来完成png数据的输入。 对于文件流数据数据设置代码如下：

1: /* One of the following I/O initialization methods is REQUIRED */

2: def streams /* PNG file I/O method 1 */

3: /* Set up the input control if you are using standard C streams */

4: png_init_io(png_ptr, fp);

用户自定义回调函数设置libpng数据源的代码如下：

1: /* If you are using replacement read functions, instead of calling

2:  * png_init_io() here you would call:

3:  */

4: png_set_read_fn(png_ptr, (void *)user_io_ptr, user_read_fn);

5: /* where user_io_ptr is a structure you want available to the callbacks */

如果你已经使用png_sig_cmp函数来检查了png数据，需要调用png_set_sig_bytes函数来告诉libpng库，这样库处理数据的时候将会跳过相应的数据，具体请参考libpng手册。

5、png图像处理

这步有两种设置方案一种称为高层处理，一种称为底层处理。高层处理 当用户的内存足够大，可以一次性读入所有的png数据，并且输出数据格式为如下libpng预定义数据类型时，可以用高层函数，下libpng预定义数据类型为：PNG_TRANSFORM_IDENTITY No transformation

PNG_TRANSFORM_STRIP_16 Strip 16-bit samples to 8 bitsPNG_TRANSFORM_STRIP_ALPHA Discard the alpha channelPNG_TRANSFORM_PACKING Expand 1, 2 and 4-bit samples to bytesPNG_TRANSFORM_PACKSWAP Change order of packed pixels to LSB firstPNG_TRANSFORM_EXPAND Perform set_expand()PNG_TRANSFORM_INVERT_MONO Invert monochrome imagesPNG_TRANSFORM_SHIFT Normalize pixels to the sBIT depthPNG_TRANSFORM_BGR Flip RGB to BGR, RGBA to BGRAPNG_TRANSFORM_SWAP_ALPHA Flip RGBA to ARGB or GA to AGPNG_TRANSFORM_INVERT_ALPHA Change alpha from opacity to transparencyPNG_TRANSFORM_SWAP_ENDIAN Byte-swap 16-bit samplesPNG_TRANSFORM_GRAY_TO_RGB Expand grayscale samples to RGB (or GA to RGBA) 高层读取函数如下：

1: /*

2:  * If you have enough memory to read in the entire image at once,

3:  * and you need to specify only transforms that can be controlled

4:  * with one of the PNG_TRANSFORM_* bits (this presently excludes

5:  * dithering, filling, setting background, and doing gamma

6:  * adjustment), then you can read the entire image (including

7:  * pixels) into the info structure with this call:

8:  */

9: png_read_png(png_ptr, info_ptr, png_transforms, png_voidp_NULL);

该函数将会把所有的图片数据解码到info_ptr数据结构中。png_transforms为整型参数，为上面libpng预定义的数据类型进行or操作得到。调用了该函数，就不可以再调用png_set_transform函数来设置输出数据。 该函数相当于调用底层函数（下文将会介绍）如下调用顺序：a)调用png_read_info函数获得图片信息。b)根据png_transforms所指示的，调用png_set_transform设置输出格式转换的函数。c)调用png_read_image来解码整个图片的数据到内存。d)调用png_read_end结束图片解码。 当你调用png_read_png之后，则可以调用如下函数得到png数据：

1: row_pointers = png_get_rows(png_ptr, info_ptr);

底层处理a)读取输入png数据的图片信息：

1: /* The call to png_read_info() gives us all of the information from the

2:  * PNG file before the first IDAT (image data chunk).  REQUIRED

3:  */

4: png_read_info(png_ptr, info_ptr);

该函数将会把输入png数据的信息读入到info_ptr数据结构中。b)查询图像信息 前面提到png_read_info将会把输入png数据的信息读入到info_ptr数据结构中，接下来需要调用API查询该信息。

1: png_get_IHDR(png_ptr, info_ptr, &width, &height, &bit_depth, &color_type,

2:     &interlace_type, int_p_NULL, int_p_NULL);

c)设置png输出参数（转换参数） 这步非常重要，用户可以指定输出数据的格式，比如RGB888，ARGB8888等等输出数据格式。通过png_set_xxxxx函数来实现，例如如下代码：

1: // expand images of all color-type and bit-depth to 3x8 bit RGB images

2: // let the library process things like alpha, transparency, background

3: if (bit_depth == 16)

4:     png_set_strip_16(png_ptr);

5: if (color_type == PNG_COLOR_TYPE_PALETTE)

6:     png_set_expand(png_ptr);

7: if (bit_depth<8)

8:     png_set_expand(png_ptr);

9: if (png_get_valid(png_ptr, info_ptr, PNG_INFO_tRNS))

10:     png_set_expand(png_ptr);

11: if (color_type == PNG_COLOR_TYPE_GRAY ||

12:     color_type == PNG_COLOR_TYPE_GRAY_ALPHA)

13:     png_set_gray_to_rgb(png_ptr);

如上代码将会把图像转换成RGB888的数据格式。这种转换函数还很多，请参阅libpng手册了解他们的作用。 虽然有很多设置输出参数的函数可以调用，但是用户的需求是无限的，很多输出格式libpng并不是原生支持的，比如YUV565，RGB565，YUYV等等。幸好libpng提供了自定义转换函数的功能，可以让用户注册转换回调函数给libpng库，在libpng对输出数据进行转换的时候，先对png_set_xxxxx函数设置的参数进行转换，最后将会调用用户自定义的转换函数进行转换。

1: png_set_read_user_transform_fn(png_ptr,

2:     read_transform_fn);

read_transform_fn为用户自定义的数据转换函数。具体实现可以参考pngtest.c中的实现。 另外你可以通过png_set_user_transform_info告诉libpng你的转换函数的用户自定义数据结构和输出数据的详细信息，比如颜色深度，颜色通道（channel）等等。你可能会问为什么要告诉libpng呢？libpng将会根据这些信息来更新png图片详细信息，后面会介绍。定义如下：

1: png_set_user_transform_info(png_ptr, user_ptr,

2:     user_depth, user_channels);

usr_ptr是用户自定义的数据结构，在用户自定义转换函数read_transform_fn中可以通过png_get_user_transform_ptr函数得到该数据结构，例如：

1: voidp read_user_transform_ptr =

2:     png_get_user_transform_ptr(png_ptr);

d)更新png数据的详细信息 经过前面的设置png数据的图片信息肯定会有一些变化，则需要调用png_read_update_info函数更新图片的详细信息：

1: png_read_update_info(png_ptr, info_ptr);

该函数将会更新保存于info_ptr变量中的图片数据信息，然后可以再调用png_get_IHDR重新查询图片信息。e)读取png数据 可以到用png_read_image函数，一次性把所有的数据读入内存，例如：

1: png_read_image(png_ptr, row_pointers);

也可以调用png_read_rows一次读入1行或多行到内存中，比如：

1: for (y = 0; y < height; y++)

2: {

3:    png_read_rows(png_ptr, &row_pointers[y], png_bytepp_NULL, 1);

4: }

f)结束读取数据通过png_read_end结束读取png数据，代码如下：

1: /* Read rest of file, and get additional chunks in info_ptr - REQUIRED */

2: png_read_end(png_ptr, info_ptr);

6、释放libpng的内存

调用png_destroy_read_struct来释放libpng的内存，代码如下：

1: /* Clean up after the read, and free any memory allocated - REQUIRED */

2: png_destroy_read_struct(&png_ptr, &info_ptr, png_infopp_NULL);

至此png数据解码，全部完成了。

总结：

通过上面的介绍，我们可以发现相对于libjpeg，libpng的扩展性非常好。我们基本上没有任何修改libpng库的需求，它的对外接口提供了足够的灵活性，允许我们扩展。从这个角度来讲，libpng库非常值得我们学习它的对外接口的定义。