1 /**************************************************************************
3 * Copyright 2008-2010 VMware, Inc.
6 * Permission is hereby granted, free of charge, to any person obtaining a copy
7 * of this software and associated documentation files (the "Software"), to deal
8 * in the Software without restriction, including without limitation the rights
9 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
10 * copies of the Software, and to permit persons to whom the Software is
11 * furnished to do so, subject to the following conditions:
13 * The above copyright notice and this permission notice shall be included in
14 * all copies or substantial portions of the Software.
16 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
19 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
24 **************************************************************************/
58 uint32_t biCompression;
60 int32_t biXPelsPerMeter;
61 int32_t biYPelsPerMeter;
63 uint32_t biClrImportant;
75 Image::writeBMP(const char *filename) const {
76 struct FileHeader bmfh;
77 struct InfoHeader bmih;
81 bmfh.bfSize = 14 + 40 + height*width*4;
84 bmfh.bfOffBits = 14 + 40;
88 bmih.biHeight = height;
91 bmih.biCompression = 0;
92 bmih.biSizeImage = height*width*4;
93 bmih.biXPelsPerMeter = 0;
94 bmih.biYPelsPerMeter = 0;
96 bmih.biClrImportant = 0;
98 std::ofstream stream(filename, std::ofstream::binary);
103 stream.write((const char *)&bmfh, 14);
104 stream.write((const char *)&bmih, 40);
106 unsigned stride = width*4;
109 for (y = 0; y < height; ++y) {
110 const unsigned char *ptr = pixels + y * stride;
111 for (x = 0; x < width; ++x) {
113 pixel.rgbRed = ptr[x*4 + 0];
114 pixel.rgbGreen = ptr[x*4 + 1];
115 pixel.rgbBlue = ptr[x*4 + 2];
116 pixel.rgbAlpha = ptr[x*4 + 3];
117 stream.write((const char *)&pixel, 4);
123 const unsigned char *ptr = pixels + y * stride;
124 for (x = 0; x < width; ++x) {
126 pixel.rgbRed = ptr[x*4 + 0];
127 pixel.rgbGreen = ptr[x*4 + 1];
128 pixel.rgbBlue = ptr[x*4 + 2];
129 pixel.rgbAlpha = ptr[x*4 + 3];
130 stream.write((const char *)&pixel, 4);
141 Image::writePNG(const char *filename) const {
146 fp = fopen(filename, "wb");
150 png_ptr = png_create_write_struct(PNG_LIBPNG_VER_STRING, NULL, NULL, NULL);
154 info_ptr = png_create_info_struct(png_ptr);
156 png_destroy_write_struct(&png_ptr, NULL);
160 if (setjmp(png_jmpbuf(png_ptr))) {
161 png_destroy_write_struct(&png_ptr, &info_ptr);
165 png_init_io(png_ptr, fp);
167 png_set_IHDR(png_ptr, info_ptr, width, height, 8, PNG_COLOR_TYPE_RGB_ALPHA,
168 PNG_INTERLACE_NONE, PNG_COMPRESSION_TYPE_BASE, PNG_FILTER_TYPE_BASE);
170 png_set_compression_level(png_ptr, Z_DEFAULT_COMPRESSION);
172 png_write_info(png_ptr, info_ptr);
175 for (unsigned y = 0; y < height; ++y) {
176 png_bytep row = (png_bytep)(pixels + y*width*4);
177 png_write_rows(png_ptr, &row, 1);
182 png_bytep row = (png_bytep)(pixels + y*width*4);
183 png_write_rows(png_ptr, &row, 1);
187 png_write_end(png_ptr, info_ptr);
188 png_destroy_write_struct(&png_ptr, &info_ptr);
201 readPNG(const char *filename)
209 fp = fopen(filename, "rb");
213 png_ptr = png_create_read_struct(PNG_LIBPNG_VER_STRING, NULL, NULL, NULL);
217 info_ptr = png_create_info_struct(png_ptr);
219 png_destroy_read_struct(&png_ptr, NULL, NULL);
223 end_info = png_create_info_struct(png_ptr);
225 png_destroy_read_struct(&png_ptr, &info_ptr, NULL);
229 if (setjmp(png_jmpbuf(png_ptr))) {
230 png_destroy_read_struct(&png_ptr, &info_ptr, &end_info);
234 png_init_io(png_ptr, fp);
236 png_read_info(png_ptr, info_ptr);
238 png_uint_32 width, height;
239 int bit_depth, color_type, interlace_type, compression_type, filter_method;
241 png_get_IHDR(png_ptr, info_ptr,
243 &bit_depth, &color_type, &interlace_type,
244 &compression_type, &filter_method);
246 image = new Image(width, height);
250 /* Convert to RGBA8 */
251 if (color_type == PNG_COLOR_TYPE_PALETTE)
252 png_set_palette_to_rgb(png_ptr);
253 if (color_type == PNG_COLOR_TYPE_GRAY && bit_depth < 8)
254 png_set_expand_gray_1_2_4_to_8(png_ptr);
255 if (png_get_valid(png_ptr, info_ptr, PNG_INFO_tRNS))
256 png_set_tRNS_to_alpha(png_ptr);
258 png_set_strip_16(png_ptr);
260 for (unsigned y = 0; y < height; ++y) {
261 png_bytep row = (png_bytep)(image->pixels + y*width*4);
262 png_read_row(png_ptr, row, NULL);
265 png_read_end(png_ptr, info_ptr);
266 png_destroy_read_struct(&png_ptr, &info_ptr, &end_info);
271 png_destroy_read_struct(&png_ptr, &info_ptr, &end_info);
279 double Image::compare(Image &ref)
281 if (width != ref.width ||
282 height != ref.height) {
286 const unsigned char *pSrc = start();
287 const unsigned char *pRef = ref.start();
289 unsigned long long error = 0;
290 for (unsigned y = 0; y < height; ++y) {
291 for (unsigned x = 0; x < width; ++x) {
292 // FIXME: Ignore alpha channel until we are able to pick a visual
293 // that matches the traces
294 for (unsigned c = 0; c < 3; ++c) {
295 int delta = pSrc[x*4 + c] - pRef[x*4 + c];
296 error += delta*delta;
301 pRef += ref.stride();
304 double numerator = error*2 + 1;
305 double denominator = height*width*3ULL*255ULL*255ULL*2;
306 double quotient = numerator/denominator;
309 double precision = -log(quotient)/log(2.0);
314 struct png_tmp_buffer
321 pngWriteCallback(png_structp png_ptr, png_bytep data, png_size_t length)
323 struct png_tmp_buffer *buf = (struct png_tmp_buffer*) png_get_io_ptr(png_ptr);
324 size_t nsize = buf->size + length;
326 /* allocate or grow buffer */
328 buf->buffer = (char*)realloc(buf->buffer, nsize);
330 buf->buffer = (char*)malloc(nsize);
333 png_error(png_ptr, "Buffer allocation error");
335 memcpy(buf->buffer + buf->size, data, length);
339 bool writePixelsToBuffer(unsigned char *pixels,
340 unsigned width, unsigned height, unsigned numChannels,
345 struct png_tmp_buffer png_mem;
350 png_mem.buffer = NULL;
353 switch (numChannels) {
355 type = PNG_COLOR_TYPE_RGB_ALPHA;
358 type = PNG_COLOR_TYPE_RGB;
361 type = PNG_COLOR_TYPE_GRAY_ALPHA;
364 type = PNG_COLOR_TYPE_GRAY;
370 png_ptr = png_create_write_struct(PNG_LIBPNG_VER_STRING, NULL, NULL, NULL);
374 info_ptr = png_create_info_struct(png_ptr);
376 png_destroy_write_struct(&png_ptr, NULL);
380 if (setjmp(png_jmpbuf(png_ptr))) {
381 png_destroy_write_struct(&png_ptr, &info_ptr);
385 png_set_write_fn(png_ptr, &png_mem, pngWriteCallback, NULL);
387 png_set_IHDR(png_ptr, info_ptr, width, height, 8,
388 type, PNG_INTERLACE_NONE,
389 PNG_COMPRESSION_TYPE_BASE, PNG_FILTER_TYPE_BASE);
391 png_set_compression_level(png_ptr, Z_DEFAULT_COMPRESSION);
393 png_write_info(png_ptr, info_ptr);
396 for (unsigned y = 0; y < height; ++y) {
397 png_bytep row = (png_bytep)(pixels + y*width*numChannels);
398 png_write_rows(png_ptr, &row, 1);
403 png_bytep row = (png_bytep)(pixels + y*width*numChannels);
404 png_write_rows(png_ptr, &row, 1);
408 png_write_end(png_ptr, info_ptr);
409 png_destroy_write_struct(&png_ptr, &info_ptr);
411 *buffer = png_mem.buffer;
412 *size = png_mem.size;
421 free(png_mem.buffer);
425 } /* namespace Image */