2002-10-12 11:37:38 +00:00
|
|
|
/**
|
|
|
|
*
|
2008-04-16 22:40:48 +00:00
|
|
|
* ***** BEGIN GPL LICENSE BLOCK *****
|
2002-10-12 11:37:38 +00:00
|
|
|
*
|
|
|
|
* This program is free software; you can redistribute it and/or
|
|
|
|
* modify it under the terms of the GNU General Public License
|
|
|
|
* as published by the Free Software Foundation; either version 2
|
2008-04-16 22:40:48 +00:00
|
|
|
* of the License, or (at your option) any later version.
|
2002-10-12 11:37:38 +00:00
|
|
|
*
|
|
|
|
* This program is distributed in the hope that it will be useful,
|
|
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
|
|
* GNU General Public License for more details.
|
|
|
|
*
|
|
|
|
* You should have received a copy of the GNU General Public License
|
|
|
|
* along with this program; if not, write to the Free Software Foundation,
|
2010-02-12 13:34:04 +00:00
|
|
|
* Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
2002-10-12 11:37:38 +00:00
|
|
|
*
|
|
|
|
* The Original Code is Copyright (C) 2001-2002 by NaN Holding BV.
|
|
|
|
* All rights reserved.
|
|
|
|
*
|
|
|
|
* The Original Code is: all of this file.
|
|
|
|
*
|
|
|
|
* Contributor(s): none yet.
|
|
|
|
*
|
2008-04-16 22:40:48 +00:00
|
|
|
* ***** END GPL LICENSE BLOCK *****
|
2002-10-12 11:37:38 +00:00
|
|
|
* $Id$
|
|
|
|
*/
|
|
|
|
|
2003-12-04 18:18:05 +00:00
|
|
|
|
2002-10-12 11:37:38 +00:00
|
|
|
#include "png.h"
|
|
|
|
|
|
|
|
#include "BLI_blenlib.h"
|
2010-08-16 05:46:10 +00:00
|
|
|
#include "MEM_guardedalloc.h"
|
2002-10-12 11:37:38 +00:00
|
|
|
|
|
|
|
#include "imbuf.h"
|
|
|
|
|
|
|
|
#include "IMB_imbuf_types.h"
|
|
|
|
#include "IMB_imbuf.h"
|
|
|
|
|
|
|
|
#include "IMB_allocimbuf.h"
|
Merge image related changes from the render branch. This includes the image
tile cache code in imbuf, but it is not hooked up to the render engine.
Imbuf module: some small refactoring and removing a lot of unused or old code
(about 6.5k lines).
* Added a ImFileType struct with callbacks to make adding an file format type,
or making changes to the API easier.
* Move imbuf init/exit code into IMB_init()/IMB_exit() functions.
* Increased mipmap levels from 10 to 20, you run into this limit already with
a 2k image.
* Removed hamx, amiga, anim5 format support.
* Removed colormap saving, only simple colormap code now for reading tga.
* Removed gen_dynlibtiff.py, editing this is almost as much work as just
editing the code directly.
* Functions removed that were only used for sequencer plugin API:
IMB_anim_nextpic, IMB_clever_double, IMB_antialias, IMB_gamwarp,
IMB_scalefieldImBuf, IMB_scalefastfieldImBuf, IMB_onethird, IMB_halflace,
IMB_dit0, IMB_dit2, IMB_cspace
* Write metadata info into OpenEXR images. Can be viewed with the command
line utility 'exrheader'
For the image tile cache code, see this page:
http://wiki.blender.org/index.php/Dev:2.5/Source/Imaging/ImageTileCache
2010-05-07 15:18:04 +00:00
|
|
|
#include "IMB_metadata.h"
|
|
|
|
#include "IMB_filetype.h"
|
2002-10-12 11:37:38 +00:00
|
|
|
|
2003-12-04 18:18:05 +00:00
|
|
|
typedef struct PNGReadStruct {
|
2005-01-03 19:53:04 +00:00
|
|
|
unsigned char *data;
|
|
|
|
unsigned int size;
|
|
|
|
unsigned int seek;
|
2003-12-04 18:18:05 +00:00
|
|
|
}PNGReadStruct;
|
2002-10-12 11:37:38 +00:00
|
|
|
|
2003-12-04 18:18:05 +00:00
|
|
|
static void ReadData( png_structp png_ptr, png_bytep data, png_size_t length);
|
|
|
|
static void WriteData( png_structp png_ptr, png_bytep data, png_size_t length);
|
|
|
|
static void Flush( png_structp png_ptr);
|
2002-10-12 11:37:38 +00:00
|
|
|
|
Merge image related changes from the render branch. This includes the image
tile cache code in imbuf, but it is not hooked up to the render engine.
Imbuf module: some small refactoring and removing a lot of unused or old code
(about 6.5k lines).
* Added a ImFileType struct with callbacks to make adding an file format type,
or making changes to the API easier.
* Move imbuf init/exit code into IMB_init()/IMB_exit() functions.
* Increased mipmap levels from 10 to 20, you run into this limit already with
a 2k image.
* Removed hamx, amiga, anim5 format support.
* Removed colormap saving, only simple colormap code now for reading tga.
* Removed gen_dynlibtiff.py, editing this is almost as much work as just
editing the code directly.
* Functions removed that were only used for sequencer plugin API:
IMB_anim_nextpic, IMB_clever_double, IMB_antialias, IMB_gamwarp,
IMB_scalefieldImBuf, IMB_scalefastfieldImBuf, IMB_onethird, IMB_halflace,
IMB_dit0, IMB_dit2, IMB_cspace
* Write metadata info into OpenEXR images. Can be viewed with the command
line utility 'exrheader'
For the image tile cache code, see this page:
http://wiki.blender.org/index.php/Dev:2.5/Source/Imaging/ImageTileCache
2010-05-07 15:18:04 +00:00
|
|
|
int imb_is_a_png(unsigned char *mem)
|
2002-10-12 11:37:38 +00:00
|
|
|
{
|
2005-01-03 19:53:04 +00:00
|
|
|
int ret_val = 0;
|
2002-10-12 11:37:38 +00:00
|
|
|
|
2005-01-03 19:53:04 +00:00
|
|
|
if (mem) ret_val = !png_sig_cmp(mem, 0, 8);
|
|
|
|
return(ret_val);
|
2002-10-12 11:37:38 +00:00
|
|
|
}
|
|
|
|
|
2003-12-04 18:18:05 +00:00
|
|
|
static void Flush(png_structp png_ptr)
|
|
|
|
{
|
2002-10-12 11:37:38 +00:00
|
|
|
}
|
|
|
|
|
2003-12-04 18:18:05 +00:00
|
|
|
static void WriteData( png_structp png_ptr, png_bytep data, png_size_t length)
|
|
|
|
{
|
|
|
|
ImBuf *ibuf = (ImBuf *) png_get_io_ptr(png_ptr);
|
|
|
|
|
|
|
|
// if buffer is to small increase it.
|
|
|
|
while (ibuf->encodedsize + length > ibuf->encodedbuffersize) {
|
|
|
|
imb_enlargeencodedbufferImBuf(ibuf);
|
|
|
|
}
|
2002-10-12 11:37:38 +00:00
|
|
|
|
2003-12-04 18:18:05 +00:00
|
|
|
memcpy(ibuf->encodedbuffer + ibuf->encodedsize, data, length);
|
|
|
|
ibuf->encodedsize += length;
|
|
|
|
}
|
|
|
|
|
|
|
|
static void ReadData( png_structp png_ptr, png_bytep data, png_size_t length)
|
2002-10-12 11:37:38 +00:00
|
|
|
{
|
|
|
|
PNGReadStruct *rs= (PNGReadStruct *) png_get_io_ptr(png_ptr);
|
|
|
|
|
|
|
|
if (rs) {
|
|
|
|
if (length <= rs->size - rs->seek) {
|
|
|
|
memcpy(data, rs->data + rs->seek, length);
|
|
|
|
rs->seek += length;
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
printf("Reached EOF while decoding PNG\n");
|
|
|
|
longjmp(png_jmpbuf(png_ptr), 1);
|
|
|
|
}
|
|
|
|
|
Merge image related changes from the render branch. This includes the image
tile cache code in imbuf, but it is not hooked up to the render engine.
Imbuf module: some small refactoring and removing a lot of unused or old code
(about 6.5k lines).
* Added a ImFileType struct with callbacks to make adding an file format type,
or making changes to the API easier.
* Move imbuf init/exit code into IMB_init()/IMB_exit() functions.
* Increased mipmap levels from 10 to 20, you run into this limit already with
a 2k image.
* Removed hamx, amiga, anim5 format support.
* Removed colormap saving, only simple colormap code now for reading tga.
* Removed gen_dynlibtiff.py, editing this is almost as much work as just
editing the code directly.
* Functions removed that were only used for sequencer plugin API:
IMB_anim_nextpic, IMB_clever_double, IMB_antialias, IMB_gamwarp,
IMB_scalefieldImBuf, IMB_scalefastfieldImBuf, IMB_onethird, IMB_halflace,
IMB_dit0, IMB_dit2, IMB_cspace
* Write metadata info into OpenEXR images. Can be viewed with the command
line utility 'exrheader'
For the image tile cache code, see this page:
http://wiki.blender.org/index.php/Dev:2.5/Source/Imaging/ImageTileCache
2010-05-07 15:18:04 +00:00
|
|
|
int imb_savepng(struct ImBuf *ibuf, char *name, int flags)
|
2003-12-04 18:18:05 +00:00
|
|
|
{
|
|
|
|
png_structp png_ptr;
|
|
|
|
png_infop info_ptr;
|
2007-09-02 17:25:03 +00:00
|
|
|
|
2003-12-04 18:18:05 +00:00
|
|
|
unsigned char *pixels = 0;
|
|
|
|
unsigned char *from, *to;
|
|
|
|
png_bytepp row_pointers = 0;
|
|
|
|
int i, bytesperpixel, color_type = PNG_COLOR_TYPE_GRAY;
|
|
|
|
FILE *fp = 0;
|
|
|
|
|
2010-07-17 13:29:55 +00:00
|
|
|
/* use the jpeg quality setting for compression */
|
|
|
|
int compression;
|
|
|
|
compression= (int)(((float)(ibuf->ftype & 0xff) / 11.1111f));
|
|
|
|
compression= compression < 0 ? 0 : (compression > 9 ? 9 : compression);
|
|
|
|
|
2003-12-04 18:18:05 +00:00
|
|
|
bytesperpixel = (ibuf->depth + 7) >> 3;
|
|
|
|
if ((bytesperpixel > 4) || (bytesperpixel == 2)) {
|
|
|
|
printf("imb_savepng: unsupported bytes per pixel: %d\n", bytesperpixel);
|
|
|
|
return (0);
|
|
|
|
}
|
|
|
|
|
|
|
|
png_ptr = png_create_write_struct(PNG_LIBPNG_VER_STRING,
|
|
|
|
NULL, NULL, NULL);
|
|
|
|
if (png_ptr == NULL) {
|
|
|
|
printf("Cannot png_create_write_struct\n");
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
info_ptr = png_create_info_struct(png_ptr);
|
|
|
|
if (info_ptr == NULL) {
|
|
|
|
png_destroy_write_struct(&png_ptr, (png_infopp)NULL);
|
|
|
|
printf("Cannot png_create_info_struct\n");
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (setjmp(png_jmpbuf(png_ptr))) {
|
|
|
|
png_destroy_write_struct(&png_ptr, &info_ptr);
|
|
|
|
if (pixels) MEM_freeN(pixels);
|
|
|
|
if (row_pointers) MEM_freeN(row_pointers);
|
|
|
|
// printf("Aborting\n");
|
|
|
|
if (fp) {
|
|
|
|
fflush(fp);
|
|
|
|
fclose(fp);
|
|
|
|
}
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
// copy image data
|
|
|
|
|
|
|
|
pixels = MEM_mallocN(ibuf->x * ibuf->y * bytesperpixel * sizeof(unsigned char), "pixels");
|
|
|
|
if (pixels == NULL) {
|
|
|
|
printf("Cannot allocate pixels array\n");
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
from = (unsigned char *) ibuf->rect;
|
|
|
|
to = pixels;
|
|
|
|
|
|
|
|
switch (bytesperpixel) {
|
|
|
|
case 4:
|
|
|
|
color_type = PNG_COLOR_TYPE_RGBA;
|
|
|
|
for (i = ibuf->x * ibuf->y; i > 0; i--) {
|
|
|
|
to[0] = from[0];
|
|
|
|
to[1] = from[1];
|
|
|
|
to[2] = from[2];
|
|
|
|
to[3] = from[3];
|
|
|
|
to += 4; from += 4;
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
case 3:
|
|
|
|
color_type = PNG_COLOR_TYPE_RGB;
|
|
|
|
for (i = ibuf->x * ibuf->y; i > 0; i--) {
|
|
|
|
to[0] = from[0];
|
|
|
|
to[1] = from[1];
|
|
|
|
to[2] = from[2];
|
|
|
|
to += 3; from += 4;
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
case 1:
|
|
|
|
color_type = PNG_COLOR_TYPE_GRAY;
|
|
|
|
for (i = ibuf->x * ibuf->y; i > 0; i--) {
|
|
|
|
to[0] = from[0];
|
|
|
|
to++; from += 4;
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (flags & IB_mem) {
|
|
|
|
// create image in memory
|
|
|
|
imb_addencodedbufferImBuf(ibuf);
|
|
|
|
ibuf->encodedsize = 0;
|
|
|
|
|
|
|
|
png_set_write_fn(png_ptr,
|
|
|
|
(png_voidp) ibuf,
|
|
|
|
WriteData,
|
|
|
|
Flush);
|
|
|
|
} else {
|
2004-12-09 13:09:11 +00:00
|
|
|
fp = fopen(name, "wb");
|
2010-03-22 09:30:00 +00:00
|
|
|
if (!fp) {
|
2007-01-22 05:07:16 +00:00
|
|
|
MEM_freeN(pixels);
|
|
|
|
return 0;
|
|
|
|
}
|
2003-12-04 18:18:05 +00:00
|
|
|
png_init_io(png_ptr, fp);
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
png_set_filter(png_ptr, 0,
|
|
|
|
PNG_FILTER_NONE | PNG_FILTER_VALUE_NONE |
|
|
|
|
PNG_FILTER_SUB | PNG_FILTER_VALUE_SUB |
|
|
|
|
PNG_FILTER_UP | PNG_FILTER_VALUE_UP |
|
|
|
|
PNG_FILTER_AVG | PNG_FILTER_VALUE_AVG |
|
|
|
|
PNG_FILTER_PAETH | PNG_FILTER_VALUE_PAETH|
|
|
|
|
PNG_ALL_FILTERS);
|
|
|
|
*/
|
|
|
|
|
2010-07-17 13:29:55 +00:00
|
|
|
png_set_compression_level(png_ptr, compression);
|
|
|
|
|
2003-12-04 18:18:05 +00:00
|
|
|
// png image settings
|
|
|
|
png_set_IHDR(png_ptr,
|
|
|
|
info_ptr,
|
|
|
|
ibuf->x,
|
|
|
|
ibuf->y,
|
|
|
|
8,
|
|
|
|
color_type,
|
|
|
|
PNG_INTERLACE_NONE,
|
|
|
|
PNG_COMPRESSION_TYPE_DEFAULT,
|
|
|
|
PNG_FILTER_TYPE_DEFAULT);
|
|
|
|
|
2007-09-02 17:25:03 +00:00
|
|
|
/* image text info */
|
Merge image related changes from the render branch. This includes the image
tile cache code in imbuf, but it is not hooked up to the render engine.
Imbuf module: some small refactoring and removing a lot of unused or old code
(about 6.5k lines).
* Added a ImFileType struct with callbacks to make adding an file format type,
or making changes to the API easier.
* Move imbuf init/exit code into IMB_init()/IMB_exit() functions.
* Increased mipmap levels from 10 to 20, you run into this limit already with
a 2k image.
* Removed hamx, amiga, anim5 format support.
* Removed colormap saving, only simple colormap code now for reading tga.
* Removed gen_dynlibtiff.py, editing this is almost as much work as just
editing the code directly.
* Functions removed that were only used for sequencer plugin API:
IMB_anim_nextpic, IMB_clever_double, IMB_antialias, IMB_gamwarp,
IMB_scalefieldImBuf, IMB_scalefastfieldImBuf, IMB_onethird, IMB_halflace,
IMB_dit0, IMB_dit2, IMB_cspace
* Write metadata info into OpenEXR images. Can be viewed with the command
line utility 'exrheader'
For the image tile cache code, see this page:
http://wiki.blender.org/index.php/Dev:2.5/Source/Imaging/ImageTileCache
2010-05-07 15:18:04 +00:00
|
|
|
if (ibuf->metadata) {
|
|
|
|
png_text* metadata;
|
|
|
|
ImMetaData* iptr;
|
2007-09-02 17:25:03 +00:00
|
|
|
int num_text = 0;
|
Merge image related changes from the render branch. This includes the image
tile cache code in imbuf, but it is not hooked up to the render engine.
Imbuf module: some small refactoring and removing a lot of unused or old code
(about 6.5k lines).
* Added a ImFileType struct with callbacks to make adding an file format type,
or making changes to the API easier.
* Move imbuf init/exit code into IMB_init()/IMB_exit() functions.
* Increased mipmap levels from 10 to 20, you run into this limit already with
a 2k image.
* Removed hamx, amiga, anim5 format support.
* Removed colormap saving, only simple colormap code now for reading tga.
* Removed gen_dynlibtiff.py, editing this is almost as much work as just
editing the code directly.
* Functions removed that were only used for sequencer plugin API:
IMB_anim_nextpic, IMB_clever_double, IMB_antialias, IMB_gamwarp,
IMB_scalefieldImBuf, IMB_scalefastfieldImBuf, IMB_onethird, IMB_halflace,
IMB_dit0, IMB_dit2, IMB_cspace
* Write metadata info into OpenEXR images. Can be viewed with the command
line utility 'exrheader'
For the image tile cache code, see this page:
http://wiki.blender.org/index.php/Dev:2.5/Source/Imaging/ImageTileCache
2010-05-07 15:18:04 +00:00
|
|
|
iptr = ibuf->metadata;
|
2007-09-02 17:25:03 +00:00
|
|
|
while (iptr) {
|
|
|
|
num_text++;
|
|
|
|
iptr = iptr->next;
|
|
|
|
}
|
|
|
|
|
Merge image related changes from the render branch. This includes the image
tile cache code in imbuf, but it is not hooked up to the render engine.
Imbuf module: some small refactoring and removing a lot of unused or old code
(about 6.5k lines).
* Added a ImFileType struct with callbacks to make adding an file format type,
or making changes to the API easier.
* Move imbuf init/exit code into IMB_init()/IMB_exit() functions.
* Increased mipmap levels from 10 to 20, you run into this limit already with
a 2k image.
* Removed hamx, amiga, anim5 format support.
* Removed colormap saving, only simple colormap code now for reading tga.
* Removed gen_dynlibtiff.py, editing this is almost as much work as just
editing the code directly.
* Functions removed that were only used for sequencer plugin API:
IMB_anim_nextpic, IMB_clever_double, IMB_antialias, IMB_gamwarp,
IMB_scalefieldImBuf, IMB_scalefastfieldImBuf, IMB_onethird, IMB_halflace,
IMB_dit0, IMB_dit2, IMB_cspace
* Write metadata info into OpenEXR images. Can be viewed with the command
line utility 'exrheader'
For the image tile cache code, see this page:
http://wiki.blender.org/index.php/Dev:2.5/Source/Imaging/ImageTileCache
2010-05-07 15:18:04 +00:00
|
|
|
metadata = MEM_callocN(num_text*sizeof(png_text), "png_metadata");
|
|
|
|
iptr = ibuf->metadata;
|
2007-09-02 17:25:03 +00:00
|
|
|
num_text = 0;
|
|
|
|
while (iptr) {
|
2007-10-20 16:17:27 +00:00
|
|
|
|
Merge image related changes from the render branch. This includes the image
tile cache code in imbuf, but it is not hooked up to the render engine.
Imbuf module: some small refactoring and removing a lot of unused or old code
(about 6.5k lines).
* Added a ImFileType struct with callbacks to make adding an file format type,
or making changes to the API easier.
* Move imbuf init/exit code into IMB_init()/IMB_exit() functions.
* Increased mipmap levels from 10 to 20, you run into this limit already with
a 2k image.
* Removed hamx, amiga, anim5 format support.
* Removed colormap saving, only simple colormap code now for reading tga.
* Removed gen_dynlibtiff.py, editing this is almost as much work as just
editing the code directly.
* Functions removed that were only used for sequencer plugin API:
IMB_anim_nextpic, IMB_clever_double, IMB_antialias, IMB_gamwarp,
IMB_scalefieldImBuf, IMB_scalefastfieldImBuf, IMB_onethird, IMB_halflace,
IMB_dit0, IMB_dit2, IMB_cspace
* Write metadata info into OpenEXR images. Can be viewed with the command
line utility 'exrheader'
For the image tile cache code, see this page:
http://wiki.blender.org/index.php/Dev:2.5/Source/Imaging/ImageTileCache
2010-05-07 15:18:04 +00:00
|
|
|
metadata[num_text].compression = PNG_TEXT_COMPRESSION_NONE;
|
|
|
|
metadata[num_text].key = iptr->key;
|
|
|
|
metadata[num_text].text = iptr->value;
|
2007-09-02 17:25:03 +00:00
|
|
|
num_text++;
|
|
|
|
iptr = iptr->next;
|
|
|
|
}
|
|
|
|
|
Merge image related changes from the render branch. This includes the image
tile cache code in imbuf, but it is not hooked up to the render engine.
Imbuf module: some small refactoring and removing a lot of unused or old code
(about 6.5k lines).
* Added a ImFileType struct with callbacks to make adding an file format type,
or making changes to the API easier.
* Move imbuf init/exit code into IMB_init()/IMB_exit() functions.
* Increased mipmap levels from 10 to 20, you run into this limit already with
a 2k image.
* Removed hamx, amiga, anim5 format support.
* Removed colormap saving, only simple colormap code now for reading tga.
* Removed gen_dynlibtiff.py, editing this is almost as much work as just
editing the code directly.
* Functions removed that were only used for sequencer plugin API:
IMB_anim_nextpic, IMB_clever_double, IMB_antialias, IMB_gamwarp,
IMB_scalefieldImBuf, IMB_scalefastfieldImBuf, IMB_onethird, IMB_halflace,
IMB_dit0, IMB_dit2, IMB_cspace
* Write metadata info into OpenEXR images. Can be viewed with the command
line utility 'exrheader'
For the image tile cache code, see this page:
http://wiki.blender.org/index.php/Dev:2.5/Source/Imaging/ImageTileCache
2010-05-07 15:18:04 +00:00
|
|
|
png_set_text(png_ptr, info_ptr, metadata, num_text);
|
|
|
|
MEM_freeN(metadata);
|
2007-09-02 17:25:03 +00:00
|
|
|
|
|
|
|
}
|
|
|
|
|
2003-12-04 18:18:05 +00:00
|
|
|
// write the file header information
|
|
|
|
png_write_info(png_ptr, info_ptr);
|
|
|
|
|
|
|
|
// allocate memory for an array of row-pointers
|
|
|
|
row_pointers = (png_bytepp) MEM_mallocN(ibuf->y * sizeof(png_bytep), "row_pointers");
|
|
|
|
if (row_pointers == NULL) {
|
2007-01-22 05:07:16 +00:00
|
|
|
printf("Cannot allocate row-pointers array\n");
|
|
|
|
MEM_freeN(pixels);
|
|
|
|
return 0;
|
2003-12-04 18:18:05 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// set the individual row-pointers to point at the correct offsets
|
|
|
|
for (i = 0; i < ibuf->y; i++) {
|
|
|
|
row_pointers[ibuf->y-1-i] = (png_bytep)
|
|
|
|
((unsigned char *)pixels + (i * ibuf->x) * bytesperpixel * sizeof(unsigned char));
|
|
|
|
}
|
|
|
|
|
|
|
|
// write out the entire image data in one call
|
|
|
|
png_write_image(png_ptr, row_pointers);
|
|
|
|
|
|
|
|
// write the additional chunks to the PNG file (not really needed)
|
|
|
|
png_write_end(png_ptr, info_ptr);
|
|
|
|
|
|
|
|
// clean up
|
|
|
|
MEM_freeN(pixels);
|
|
|
|
MEM_freeN(row_pointers);
|
|
|
|
png_destroy_write_struct(&png_ptr, &info_ptr);
|
|
|
|
|
|
|
|
if (fp) {
|
|
|
|
fflush(fp);
|
|
|
|
fclose(fp);
|
|
|
|
}
|
|
|
|
|
|
|
|
return(1);
|
|
|
|
}
|
2002-10-12 11:37:38 +00:00
|
|
|
|
2010-09-15 06:43:36 +00:00
|
|
|
struct ImBuf *imb_loadpng(unsigned char *mem, size_t size, int flags)
|
2002-10-12 11:37:38 +00:00
|
|
|
{
|
|
|
|
struct ImBuf *ibuf = 0;
|
|
|
|
png_structp png_ptr;
|
|
|
|
png_infop info_ptr;
|
|
|
|
unsigned char *pixels = 0;
|
|
|
|
png_bytepp row_pointers = 0;
|
|
|
|
png_uint_32 width, height;
|
|
|
|
int bit_depth, color_type;
|
|
|
|
PNGReadStruct ps;
|
|
|
|
|
|
|
|
unsigned char *from, *to;
|
|
|
|
int i, bytesperpixel;
|
|
|
|
|
2003-12-04 18:18:05 +00:00
|
|
|
if (imb_is_a_png(mem) == 0) return(0);
|
2002-10-12 11:37:38 +00:00
|
|
|
|
|
|
|
png_ptr = png_create_read_struct(PNG_LIBPNG_VER_STRING,
|
|
|
|
NULL, NULL, NULL);
|
|
|
|
if (png_ptr == NULL) {
|
|
|
|
printf("Cannot png_create_read_struct\n");
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
info_ptr = png_create_info_struct(png_ptr);
|
|
|
|
if (info_ptr == NULL) {
|
2003-12-04 18:18:05 +00:00
|
|
|
png_destroy_read_struct(&png_ptr, (png_infopp)NULL,
|
|
|
|
(png_infopp)NULL);
|
2002-10-12 11:37:38 +00:00
|
|
|
printf("Cannot png_create_info_struct\n");
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2010-09-15 06:43:36 +00:00
|
|
|
ps.size = size; /* XXX, 4gig limit! */
|
2002-10-12 11:37:38 +00:00
|
|
|
ps.data = mem;
|
|
|
|
ps.seek = 0;
|
|
|
|
|
|
|
|
png_set_read_fn(png_ptr, (void *) &ps, ReadData);
|
|
|
|
|
|
|
|
if (setjmp(png_jmpbuf(png_ptr))) {
|
|
|
|
png_destroy_read_struct(&png_ptr, &info_ptr, (png_infopp)NULL);
|
|
|
|
if (pixels) MEM_freeN(pixels);
|
|
|
|
if (row_pointers) MEM_freeN(row_pointers);
|
|
|
|
if (ibuf) IMB_freeImBuf(ibuf);
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
// png_set_sig_bytes(png_ptr, 8);
|
|
|
|
|
|
|
|
png_read_info(png_ptr, info_ptr);
|
2003-12-04 18:18:05 +00:00
|
|
|
png_get_IHDR(png_ptr, info_ptr, &width, &height, &bit_depth,
|
|
|
|
&color_type, NULL, NULL, NULL);
|
2002-10-12 11:37:38 +00:00
|
|
|
|
|
|
|
if (bit_depth == 16) {
|
|
|
|
png_set_strip_16(png_ptr);
|
|
|
|
bit_depth = 8;
|
|
|
|
}
|
|
|
|
|
|
|
|
bytesperpixel = png_get_channels(png_ptr, info_ptr);
|
|
|
|
|
|
|
|
switch(color_type) {
|
|
|
|
case PNG_COLOR_TYPE_RGB:
|
|
|
|
case PNG_COLOR_TYPE_RGB_ALPHA:
|
|
|
|
break;
|
|
|
|
case PNG_COLOR_TYPE_PALETTE:
|
|
|
|
png_set_palette_to_rgb(png_ptr);
|
|
|
|
if (png_get_valid(png_ptr, info_ptr, PNG_INFO_tRNS)) {
|
|
|
|
bytesperpixel = 4;
|
|
|
|
} else {
|
|
|
|
bytesperpixel = 3;
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
case PNG_COLOR_TYPE_GRAY:
|
|
|
|
case PNG_COLOR_TYPE_GRAY_ALPHA:
|
|
|
|
if (bit_depth < 8) {
|
|
|
|
png_set_expand(png_ptr);
|
|
|
|
bit_depth = 8;
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
default:
|
|
|
|
printf("PNG format not supported\n");
|
|
|
|
longjmp(png_jmpbuf(png_ptr), 1);
|
|
|
|
}
|
|
|
|
|
|
|
|
ibuf = IMB_allocImBuf(width, height, 8 * bytesperpixel, 0, 0);
|
|
|
|
|
|
|
|
if (ibuf) {
|
|
|
|
ibuf->ftype = PNG;
|
2010-01-09 00:16:35 +00:00
|
|
|
ibuf->profile = IB_PROFILE_SRGB;
|
2002-10-12 11:37:38 +00:00
|
|
|
} else {
|
|
|
|
printf("Couldn't allocate memory for PNG image\n");
|
|
|
|
}
|
|
|
|
|
|
|
|
if (ibuf && ((flags & IB_test) == 0)) {
|
|
|
|
imb_addrectImBuf(ibuf);
|
|
|
|
|
|
|
|
pixels = MEM_mallocN(ibuf->x * ibuf->y * bytesperpixel * sizeof(unsigned char), "pixels");
|
|
|
|
if (pixels == NULL) {
|
|
|
|
printf("Cannot allocate pixels array\n");
|
|
|
|
longjmp(png_jmpbuf(png_ptr), 1);
|
|
|
|
}
|
|
|
|
|
|
|
|
// allocate memory for an array of row-pointers
|
|
|
|
row_pointers = (png_bytepp) MEM_mallocN(ibuf->y * sizeof(png_bytep), "row_pointers");
|
|
|
|
if (row_pointers == NULL) {
|
|
|
|
printf("Cannot allocate row-pointers array\n");
|
|
|
|
longjmp(png_jmpbuf(png_ptr), 1);
|
|
|
|
}
|
|
|
|
|
|
|
|
// set the individual row-pointers to point at the correct offsets
|
|
|
|
for (i = 0; i < ibuf->y; i++) {
|
|
|
|
row_pointers[ibuf->y-1-i] = (png_bytep)
|
|
|
|
((unsigned char *)pixels + (i * ibuf->x) * bytesperpixel * sizeof(unsigned char));
|
|
|
|
}
|
|
|
|
|
|
|
|
png_read_image(png_ptr, row_pointers);
|
|
|
|
|
|
|
|
// copy image data
|
|
|
|
|
|
|
|
to = (unsigned char *) ibuf->rect;
|
|
|
|
from = pixels;
|
|
|
|
|
|
|
|
switch (bytesperpixel) {
|
|
|
|
case 4:
|
|
|
|
for (i = ibuf->x * ibuf->y; i > 0; i--) {
|
|
|
|
to[0] = from[0];
|
|
|
|
to[1] = from[1];
|
|
|
|
to[2] = from[2];
|
|
|
|
to[3] = from[3];
|
|
|
|
to += 4; from += 4;
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
case 3:
|
|
|
|
for (i = ibuf->x * ibuf->y; i > 0; i--) {
|
|
|
|
to[0] = from[0];
|
|
|
|
to[1] = from[1];
|
|
|
|
to[2] = from[2];
|
|
|
|
to[3] = 0xff;
|
|
|
|
to += 4; from += 3;
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
case 2:
|
|
|
|
for (i = ibuf->x * ibuf->y; i > 0; i--) {
|
|
|
|
to[0] = to[1] = to[2] = from[0];
|
|
|
|
to[3] = from[1];
|
|
|
|
to += 4; from += 2;
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
case 1:
|
|
|
|
for (i = ibuf->x * ibuf->y; i > 0; i--) {
|
|
|
|
to[0] = to[1] = to[2] = from[0];
|
|
|
|
to[3] = 0xff;
|
|
|
|
to += 4; from++;
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
Merge image related changes from the render branch. This includes the image
tile cache code in imbuf, but it is not hooked up to the render engine.
Imbuf module: some small refactoring and removing a lot of unused or old code
(about 6.5k lines).
* Added a ImFileType struct with callbacks to make adding an file format type,
or making changes to the API easier.
* Move imbuf init/exit code into IMB_init()/IMB_exit() functions.
* Increased mipmap levels from 10 to 20, you run into this limit already with
a 2k image.
* Removed hamx, amiga, anim5 format support.
* Removed colormap saving, only simple colormap code now for reading tga.
* Removed gen_dynlibtiff.py, editing this is almost as much work as just
editing the code directly.
* Functions removed that were only used for sequencer plugin API:
IMB_anim_nextpic, IMB_clever_double, IMB_antialias, IMB_gamwarp,
IMB_scalefieldImBuf, IMB_scalefastfieldImBuf, IMB_onethird, IMB_halflace,
IMB_dit0, IMB_dit2, IMB_cspace
* Write metadata info into OpenEXR images. Can be viewed with the command
line utility 'exrheader'
For the image tile cache code, see this page:
http://wiki.blender.org/index.php/Dev:2.5/Source/Imaging/ImageTileCache
2010-05-07 15:18:04 +00:00
|
|
|
if (flags & IB_metadata) {
|
2007-09-02 17:25:03 +00:00
|
|
|
png_text* text_chunks;
|
|
|
|
int count = png_get_text(png_ptr, info_ptr, &text_chunks, NULL);
|
|
|
|
for(i = 0; i < count; i++) {
|
Merge image related changes from the render branch. This includes the image
tile cache code in imbuf, but it is not hooked up to the render engine.
Imbuf module: some small refactoring and removing a lot of unused or old code
(about 6.5k lines).
* Added a ImFileType struct with callbacks to make adding an file format type,
or making changes to the API easier.
* Move imbuf init/exit code into IMB_init()/IMB_exit() functions.
* Increased mipmap levels from 10 to 20, you run into this limit already with
a 2k image.
* Removed hamx, amiga, anim5 format support.
* Removed colormap saving, only simple colormap code now for reading tga.
* Removed gen_dynlibtiff.py, editing this is almost as much work as just
editing the code directly.
* Functions removed that were only used for sequencer plugin API:
IMB_anim_nextpic, IMB_clever_double, IMB_antialias, IMB_gamwarp,
IMB_scalefieldImBuf, IMB_scalefastfieldImBuf, IMB_onethird, IMB_halflace,
IMB_dit0, IMB_dit2, IMB_cspace
* Write metadata info into OpenEXR images. Can be viewed with the command
line utility 'exrheader'
For the image tile cache code, see this page:
http://wiki.blender.org/index.php/Dev:2.5/Source/Imaging/ImageTileCache
2010-05-07 15:18:04 +00:00
|
|
|
IMB_metadata_add_field(ibuf, text_chunks[i].key, text_chunks[i].text);
|
|
|
|
ibuf->flags |= IB_metadata;
|
2010-03-22 09:30:00 +00:00
|
|
|
}
|
2007-09-02 17:25:03 +00:00
|
|
|
}
|
|
|
|
|
2002-10-12 11:37:38 +00:00
|
|
|
png_read_end(png_ptr, info_ptr);
|
|
|
|
}
|
|
|
|
|
|
|
|
// clean up
|
|
|
|
MEM_freeN(pixels);
|
|
|
|
MEM_freeN(row_pointers);
|
|
|
|
png_destroy_read_struct(&png_ptr, &info_ptr, (png_infopp)NULL);
|
|
|
|
|
|
|
|
return(ibuf);
|
|
|
|
}
|
|
|
|
|