7c3d8ad0c7
problem we're seeing (at least I couldn't reproduce the problem anymore). * SaveToPNG() now returns an error code. * cleanup. git-svn-id: file:///srv/svn/repos/haiku/haiku/trunk@14813 a95241bf-73f2-0310-859d-f6bbb57e9c96
92 lines
1.9 KiB
C++
92 lines
1.9 KiB
C++
/*
|
|
* Copyright 2001-2005, Haiku.
|
|
* Distributed under the terms of the MIT License.
|
|
*
|
|
* Authors:
|
|
* DarkWyrm <bpmagic@columbus.rr.com>
|
|
*/
|
|
|
|
/** Function for saving a generic framebuffer to a PNG file */
|
|
|
|
#include "PNGDump.h"
|
|
|
|
#include <Rect.h>
|
|
|
|
#include <png.h>
|
|
|
|
#include <errno.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <unistd.h>
|
|
|
|
|
|
#define TRACE_PNGDUMP
|
|
#ifdef TRACE_PNGDUMP
|
|
# define TRACE(x) printf x
|
|
#else
|
|
# define TRACE(x) ;
|
|
#endif
|
|
|
|
|
|
status_t
|
|
SaveToPNG(const char* filename, const BRect& bounds, color_space space,
|
|
const void* bits, int32 bitsLength, int32 bytesPerRow)
|
|
{
|
|
int32 width = bounds.IntegerWidth() + 1;
|
|
int32 height = bounds.IntegerHeight() + 1;
|
|
|
|
TRACE(("SaveToPNG: %s (%ldx%ld)\n", filename, width, height));
|
|
|
|
FILE *file = fopen(filename, "wb");
|
|
if (file == NULL) {
|
|
TRACE(("Couldn't open file: %s\n", strerror(errno)));
|
|
return errno;
|
|
}
|
|
|
|
png_structp png = png_create_write_struct(PNG_LIBPNG_VER_STRING,
|
|
NULL, NULL, NULL);
|
|
if (png == NULL) {
|
|
TRACE(("Couldn't create write struct\n"));
|
|
fclose(file);
|
|
return B_NO_MEMORY;
|
|
}
|
|
|
|
png_infop info = png_create_info_struct(png);
|
|
if (info == NULL) {
|
|
TRACE(("Couldn't create info struct\n"));
|
|
png_destroy_write_struct(&png, NULL);
|
|
fclose(file);
|
|
return B_NO_MEMORY;
|
|
}
|
|
|
|
if (setjmp(png->jmpbuf)) {
|
|
png_destroy_write_struct(&png, NULL);
|
|
fclose(file);
|
|
return B_ERROR;
|
|
}
|
|
|
|
png_init_io(png, file);
|
|
png_set_bgr(png);
|
|
|
|
png_set_IHDR(png, info, width, height, 8, PNG_COLOR_TYPE_RGB_ALPHA,
|
|
PNG_INTERLACE_NONE, PNG_COMPRESSION_TYPE_DEFAULT,
|
|
PNG_FILTER_TYPE_DEFAULT);
|
|
png_set_strip_alpha(png);
|
|
|
|
png_write_info(png, info);
|
|
|
|
png_byte* rowPointers[height];
|
|
png_byte* index = (png_byte*)bits;
|
|
for (int32 i = 0; i < height; i++) {
|
|
rowPointers[i] = index;
|
|
index += bytesPerRow;
|
|
}
|
|
|
|
png_write_image(png, rowPointers);
|
|
png_write_end(png, info);
|
|
|
|
png_destroy_write_struct(&png, NULL);
|
|
fclose(file);
|
|
return B_OK;
|
|
}
|