123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264 |
- /*!
- * \file src/utils/png.cpp
- * \brief PNG image reader
- *
- * https://github.com/DavidEGrayson/ahrs-visualizer/blob/master/png_texture.cpp
- * http://zarb.org/~gc/html/libpng.html
- *
- * \author xythobuz
- */
-
- #include <png.h>
- #include <cstdio>
-
- #include "global.h"
- #include "utils/pixel.h"
- #include "utils/png.h"
-
- #include "Console.h"
- #define pngPrint getConsole().print
-
- int pngCheck(const char *filename) {
- png_byte header[8];
-
- assert(filename != NULL);
- assert(filename[0] != '\0');
-
- FILE *fp = fopen(filename, "rb");
- if (fp == NULL) {
- pngPrint("Could not open %s", filename);
- return -1;
- }
-
- fread(header, 1, 8, fp);
- fclose(fp);
-
- if (png_sig_cmp(header, 0, 8)) {
- pngPrint("File %s is not a PNG.", filename);
- return -2;
- }
-
- return 0;
- }
-
- int pngLoad(const char *filename, unsigned char **image,
- unsigned int *width, unsigned int *height, ColorMode *mode, unsigned int *bpp) {
- png_byte header[8];
-
- assert(filename != NULL);
- assert(filename[0] != '\0');
- assert(image != NULL);
- assert(width != NULL);
- assert(height != NULL);
- assert(mode != NULL);
- assert(bpp != NULL);
-
- FILE *fp = fopen(filename, "rb");
- if (fp == NULL) {
- pngPrint("Could not open %s", filename);
- return -1;
- }
-
- fread(header, 1, 8, fp);
-
- if (png_sig_cmp(header, 0, 8)) {
- pngPrint("File %s is not a PNG.", filename);
- fclose(fp);
- return -2;
- }
-
- png_structp png_ptr = png_create_read_struct(PNG_LIBPNG_VER_STRING, NULL, NULL, NULL);
- if (!png_ptr) {
- pngPrint("png_create_read_struct returned 0.");
- fclose(fp);
- return -3;
- }
-
- png_infop info_ptr = png_create_info_struct(png_ptr);
- if (!info_ptr) {
- pngPrint("png_create_info_struct returned 0.");
- png_destroy_read_struct(&png_ptr, (png_infopp)NULL, (png_infopp)NULL);
- fclose(fp);
- return -4;
- }
-
- png_infop end_info = png_create_info_struct(png_ptr);
- if (!end_info) {
- pngPrint("png_create_info_struct returned 0.");
- png_destroy_read_struct(&png_ptr, &info_ptr, (png_infopp) NULL);
- fclose(fp);
- return -5;
- }
-
- if (setjmp(png_jmpbuf(png_ptr))) {
- pngPrint("Error from libpng");
- png_destroy_read_struct(&png_ptr, &info_ptr, &end_info);
- fclose(fp);
- return -6;
- }
-
- png_init_io(png_ptr, fp);
- png_set_sig_bytes(png_ptr, 8);
- png_read_info(png_ptr, info_ptr);
-
- int bit_depth, color_type;
- png_uint_32 tmpWidth, tmpHeight;
-
- png_get_IHDR(png_ptr, info_ptr, &tmpWidth, &tmpHeight, &bit_depth, &color_type,
- NULL, NULL, NULL);
-
- *width = tmpWidth;
- *height = tmpHeight;
-
- if (bit_depth != 8) {
- pngPrint("%s: Unsupported bit depth %d. Must be 8.", filename, bit_depth);
- return -7;
- }
-
- png_read_update_info(png_ptr, info_ptr);
-
- png_size_t rowbytes = png_get_rowbytes(png_ptr, info_ptr);
-
- png_byte *image_data = new png_byte[(rowbytes * *height) + 15];
- png_bytep *row_pointers = new png_bytep[*height];
-
- for (unsigned int i = 0; i < *height; i++)
- row_pointers[*height - 1 - i] = image_data + i * rowbytes;
-
- png_read_image(png_ptr, row_pointers);
-
- png_destroy_read_struct(&png_ptr, &info_ptr, &end_info);
- delete [] row_pointers;
- fclose(fp);
-
- if (color_type == PNG_COLOR_TYPE_GRAY) {
- *mode = GREYSCALE;
- *bpp = 8;
- } else if (color_type == PNG_COLOR_TYPE_RGB) {
- *mode = RGB;
- *bpp = 24;
- } else if (color_type == PNG_COLOR_TYPE_RGB_ALPHA) {
- *mode = RGBA;
- *bpp = 32;
- } else {
- pngPrint("%s: Unknown libpng color type %d.", filename, color_type);
- delete [] image_data;
- return -8;
- }
-
- // Flip
- *image = new unsigned char[*width * *height * (*bpp / 8)];
- for (unsigned int y = 0; y < (*height); y++) {
- for (unsigned int x = 0; x < (*width); x++) {
- (*image)[((y * *width) + x) * (*bpp / 8)]
- = image_data[(((*height - y - 1) * *width) + x) * (*bpp / 8)];
- (*image)[(((y * *width) + x) * (*bpp / 8)) + 1]
- = image_data[((((*height - y - 1) * *width) + x) * (*bpp / 8)) + 1];
- (*image)[(((y * *width) + x) * (*bpp / 8)) + 2]
- = image_data[((((*height - y - 1) * *width) + x) * (*bpp / 8)) + 2];
- (*image)[(((y * *width) + x) * (*bpp / 8)) + 3]
- = image_data[((((*height - y - 1) * *width) + x) * (*bpp / 8)) + 3];
- }
- }
-
- delete [] image_data;
- return 0;
- }
-
- int pngSave(const char *filename, unsigned char *image,
- unsigned int width, unsigned int height, ColorMode mode, unsigned int bpp) {
- assert(filename != NULL);
- assert(filename[0] != '\0');
- assert(image != NULL);
- assert(width > 0);
- assert(height > 0);
-
- FILE *fp = fopen(filename, "wb");
- if (!fp) {
- pngPrint("File %s could not be opened for writing", filename);
- return -1;
- }
-
- png_structp png_ptr = png_create_write_struct(PNG_LIBPNG_VER_STRING, NULL, NULL, NULL);
-
- if (!png_ptr) {
- pngPrint("png_create_write_struct failed");
- fclose(fp);
- return -2;
- }
-
- png_infop info_ptr = png_create_info_struct(png_ptr);
- if (!info_ptr) {
- pngPrint("png_create_info_struct failed");
- fclose(fp);
- return -3;
- }
-
- if (setjmp(png_jmpbuf(png_ptr))) {
- pngPrint("Error during init_io");
- fclose(fp);
- return -4;
- }
-
- png_init_io(png_ptr, fp);
-
- if (setjmp(png_jmpbuf(png_ptr))) {
- pngPrint("Error during writing header");
- fclose(fp);
- return -5;
- }
-
- int color_type;
- if ((mode == GREYSCALE) && (bpp == 8)) {
- color_type = PNG_COLOR_TYPE_GRAY;
- } else if (((mode == RGB) || (mode == BGR)) && (bpp == 24)) {
- if (mode == BGR) {
- bgr2rgb24(image, width, height);
- }
- color_type = PNG_COLOR_TYPE_RGB;
- } else if (((mode == RGBA) || (mode == BGRA)) && (bpp == 32)) {
- if (mode == BGRA) {
- bgra2rgba32(image, width, height);
- }
- color_type = PNG_COLOR_TYPE_RGB_ALPHA;
- } else {
- pngPrint("Error invalid color mode");
- fclose(fp);
- return -6;
- }
-
- png_set_IHDR(png_ptr, info_ptr, width, height,
- 8, color_type, PNG_INTERLACE_NONE,
- PNG_COMPRESSION_TYPE_BASE, PNG_FILTER_TYPE_BASE);
-
- png_write_info(png_ptr, info_ptr);
-
- png_bytep *row_pointers = new png_bytep[height];
-
- for (unsigned int i = 0; i < height; i++)
- row_pointers[height - 1 - i] = image + (i * width * 4);
-
- if (setjmp(png_jmpbuf(png_ptr))) {
- pngPrint("Error during writing bytes");
- delete [] row_pointers;
- fclose(fp);
- return -7;
- }
-
- png_write_image(png_ptr, row_pointers);
-
- if (setjmp(png_jmpbuf(png_ptr))) {
- pngPrint("Error during end of write");
- delete [] row_pointers;
- fclose(fp);
- return -8;
- }
-
- png_write_end(png_ptr, NULL);
-
- delete [] row_pointers;
- fclose(fp);
-
- return 0;
- }
|