haiku/src/servers/app/PNGDump.cpp
Stephan Aßmus 151ebb14dd just a little cleanup, made PNG screenshots work
git-svn-id: file:///srv/svn/repos/haiku/haiku/trunk@12665 a95241bf-73f2-0310-859d-f6bbb57e9c96
2005-05-15 00:22:55 +00:00

116 lines
3.3 KiB
C++

//------------------------------------------------------------------------------
// Copyright (c) 2001-2002, Haiku, Inc.
//
// Permission is hereby granted, free of charge, to any person obtaining a
// copy of this software and associated documentation files (the "Software"),
// to deal in the Software without restriction, including without limitation
// the rights to use, copy, modify, merge, publish, distribute, sublicense,
// and/or sell copies of the Software, and to permit persons to whom the
// Software is furnished to do so, subject to the following conditions:
//
// The above copyright notice and this permission notice shall be included in
// all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
// FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
// DEALINGS IN THE SOFTWARE.
//
// File Name: PNGDump.cpp
// Author: DarkWyrm <bpmagic@columbus.rr.com>
// Description: Function for saving a generic framebuffer to a PNG file
//
//------------------------------------------------------------------------------
#include <Rect.h>
#include <Bitmap.h>
#include <stdio.h>
#include <unistd.h>
#include <png.h>
#include <ColorUtils.h>
#include "PNGDump.h"
#define DEBUG_PNGDUMP
void SaveToPNG(const char *filename, const BRect &bounds, color_space space,
const void *bits, const int32 &bitslength, const int32 bytesperrow)
{
#ifdef DEBUG_PNGDUMP
printf("SaveToPNG: %s ",filename);
bounds.PrintToStream();
#endif
FILE *fp;
png_structp png_ptr;
png_infop info_ptr;
fp=fopen(filename, "wb");
if(fp==NULL)
{
#ifdef DEBUG_PNGDUMP
printf("Couldn't open file\n");
#endif
return;
}
png_ptr=png_create_write_struct(PNG_LIBPNG_VER_STRING, NULL, NULL, NULL);
if(png_ptr==NULL)
{
#ifdef DEBUG_PNGDUMP
printf("Couldn't create write struct\n");
#endif
fclose(fp);
return;
}
info_ptr=png_create_info_struct(png_ptr);
if(info_ptr==NULL)
{
#ifdef DEBUG_PNGDUMP
printf("Couldn't create info struct\n");
#endif
fclose(fp);
png_destroy_write_struct(&png_ptr, (png_infopp)NULL);
return;
}
if(setjmp(png_ptr->jmpbuf))
{
#ifdef DEBUG_PNGDUMP
printf("Couldn't set jump\n");
#endif
fclose(fp);
png_destroy_write_struct(&png_ptr, (png_infopp)NULL);
return;
}
png_init_io(png_ptr, fp);
// png_set_compression_level(png_ptr, Z_NO_COMPRESSION);
png_set_bgr(png_ptr);
int32 width = bounds.IntegerWidth() + 1;
int32 height = bounds.IntegerHeight() + 1;
png_set_IHDR(png_ptr, info_ptr, width, height, 8, PNG_COLOR_TYPE_RGB_ALPHA,
// PNG_INTERLACE_NONE, PNG_COMPRESSION_TYPE_BASE, PNG_FILTER_TYPE_BASE);
PNG_INTERLACE_NONE, PNG_COMPRESSION_TYPE_DEFAULT, PNG_FILTER_TYPE_DEFAULT);
png_write_info(png_ptr, info_ptr);
png_byte* row_pointers[height];
png_byte* index = (png_byte*)bits;
for (int32 i = 0; i < height; i++) {
row_pointers[i] = index;
index += bytesperrow;
}
png_write_image(png_ptr, row_pointers);
png_write_end(png_ptr, info_ptr);
png_destroy_write_struct(&png_ptr, (png_infopp)NULL);
fclose(fp);
}