-
Notifications
You must be signed in to change notification settings - Fork 1
/
PNGImage.hpp
95 lines (73 loc) · 2.35 KB
/
PNGImage.hpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
#pragma once
#include <cstdio>
#include <png.h>
#include "Debug.hpp"
#include "File.hpp"
#include "Logger.hpp"
#include "Image.hpp"
namespace img {
struct PNGImage : public Image {
PNGImage(const char *filename):
Image(filename)
{
init();
}
void init() {
unsigned char header[8]; // 8 is the maximum size that can be checked
format = Image::Format::RGBA;
/* open file and test for it being a png */
FILE *fp = fopen(filename.c_str(), "rb");
if(fp == nullptr) {
TERMINATE("png: unable to open file '%s'\n", filename.c_str());
}
sys::File::Lock fl(fp);
fread(header, 1, 8, fp);
if(png_sig_cmp(header, 0, 8)) {
TERMINATE("png: File %s is not recognized as a PNG file\n", filename.c_str());
}
/* initialize stuff */
png_structp png_ptr = png_create_read_struct(PNG_LIBPNG_VER_STRING, nullptr, nullptr, nullptr);
if(png_ptr == nullptr) {
TERMINATE("png: png_create_read_struct failed\n");
}
png_infop info_ptr = png_create_info_struct(png_ptr);
if(!info_ptr) {
TERMINATE("[png:png_create_info_struct failed\n");
return;
}
if(setjmp(png_jmpbuf(png_ptr))) {
Logger::Error("png: error during init_io\n");
}
png_init_io(png_ptr, fp);
png_set_sig_bytes(png_ptr, 8);
png_read_info(png_ptr, info_ptr);
width = png_get_image_width(png_ptr, info_ptr);
height = png_get_image_height(png_ptr, info_ptr);
int color_type = png_get_color_type(png_ptr, info_ptr);
int bit_depth = png_get_bit_depth(png_ptr, info_ptr);
int number_of_passes = png_set_interlace_handling(png_ptr);
png_read_update_info(png_ptr, info_ptr);
/* read file */
if(setjmp(png_jmpbuf(png_ptr))) {
TERMINATE("[read_png_file] read_image failed\n");
}
size_t bpp = png_get_channels(png_ptr, info_ptr);
data = new unsigned char[height * width * bpp];
png_bytep row_pointers[height];
for(size_t y = 0; y < height; ++y) {
row_pointers[y] = &data[y * width * bpp];
}
png_read_image(png_ptr, row_pointers);
fl.drop();
fclose(fp);
if(bpp == 3) {
format = Image::Format::RGB;
} else if(bpp == 4) {
format = Image::Format::RGBA;
} else {
TERMINATE("png: unknown image format %d for file %s\n", bpp, filename.c_str());
}
png_destroy_read_struct(&png_ptr, &info_ptr, nullptr);
}
};
}