1 /* === S Y N F I G ========================================================= */
3 ** \brief ppm Target Module
8 ** Copyright (c) 2002-2005 Robert B. Quattlebaum Jr., Adrian Bentley
10 ** This package is free software; you can redistribute it and/or
11 ** modify it under the terms of the GNU General Public License as
12 ** published by the Free Software Foundation; either version 2 of
13 ** the License, or (at your option) any later version.
15 ** This package is distributed in the hope that it will be useful,
16 ** but WITHOUT ANY WARRANTY; without even the implied warranty of
17 ** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
18 ** General Public License for more details.
21 ** === N O T E S ===========================================================
23 ** ========================================================================= */
26 ** \todo Support 16 bit PNG files
27 ** \todo Support GAMMA correction
28 ** \todo Fix memory leaks
31 /* === H E A D E R S ======================================================= */
41 #include <synfig/importer.h>
42 #include <synfig/time.h>
43 #include <synfig/general.h>
51 /* === M A C R O S ========================================================= */
53 using namespace synfig;
57 #define PNG_CHECK_BYTES 8
59 /* === G L O B A L S ======================================================= */
61 SYNFIG_IMPORTER_INIT(png_mptr);
62 SYNFIG_IMPORTER_SET_NAME(png_mptr,"png");
63 SYNFIG_IMPORTER_SET_EXT(png_mptr,"png");
64 SYNFIG_IMPORTER_SET_VERSION(png_mptr,"0.1");
65 SYNFIG_IMPORTER_SET_CVS_ID(png_mptr,"$Id$");
67 /* === M E T H O D S ======================================================= */
70 png_mptr::png_out_error(png_struct */*png_data*/,const char *msg)
72 //png_mptr *me=(png_mptr*)png_data->error_ptr;
73 synfig::error(strprintf("png_mptr: error: %s",msg));
78 png_mptr::png_out_warning(png_struct */*png_data*/,const char *msg)
80 //png_mptr *me=(png_mptr*)png_data->error_ptr;
81 synfig::warning(strprintf("png_mptr: warning: %s",msg));
86 png_mptr::read_chunk_callback(png_struct */*png_data*/, png_unknown_chunkp /*chunk*/)
88 /* The unknown chunk structure contains your
93 /* Note that libpng has already taken care of
96 /* put your code here. Return one of the
99 //return (-n); /* chunk had an error */
100 return (0); /* did not recognize */
101 //return (n); /* success */
104 png_mptr::png_mptr(const char *file_name)
108 /* Open the file pointer */
109 FILE *file = fopen(file_name, "rb");
112 //! \todo THROW SOMETHING
113 throw strprintf("Unable to physically open %s",file_name);
118 /* Make sure we are dealing with a PNG format file */
119 png_byte header[PNG_CHECK_BYTES];
120 fread(header, 1, PNG_CHECK_BYTES, file);
121 bool is_png = !png_sig_cmp(header, 0, PNG_CHECK_BYTES);
124 //! \todo THROW SOMETHING
125 throw strprintf("This (\"%s\") doesn't appear to be a PNG file",file_name);
130 png_structp png_ptr = png_create_read_struct
131 (PNG_LIBPNG_VER_STRING, (png_voidp)this,
132 &png_mptr::png_out_error, &png_mptr::png_out_warning);
135 //! \todo THROW SOMETHING
136 throw String("error on importer construction, *WRITEME*3");
140 png_infop info_ptr = png_create_info_struct(png_ptr);
143 png_destroy_read_struct(&png_ptr,
144 (png_infopp)NULL, (png_infopp)NULL);
145 //! \todo THROW SOMETHING
146 throw String("error on importer construction, *WRITEME*4");
150 png_infop end_info = png_create_info_struct(png_ptr);
153 png_destroy_read_struct(&png_ptr, &info_ptr,
155 //! \todo THROW SOMETHING
156 throw String("error on importer construction, *WRITEME*4");
162 png_init_io(png_ptr, file);
163 png_set_sig_bytes(png_ptr,PNG_CHECK_BYTES);
165 png_read_info(png_ptr, info_ptr);
167 int bit_depth,color_type,interlace_type, compression_type,filter_method;
168 png_uint_32 width,height;
170 png_get_IHDR(png_ptr, info_ptr, &width, &height,
171 &bit_depth, &color_type, &interlace_type,
172 &compression_type, &filter_method);
175 png_set_strip_16(png_ptr);
178 png_set_packing(png_ptr);
181 if (png_get_gAMA(png_ptr, info_ptr, &fgamma))
183 synfig::info("PNG: Image gamma is %f",fgamma);
184 png_set_gamma(png_ptr, gamma().get_gamma(), fgamma);
189 if (setjmp(png_jmpbuf(png_ptr)))
191 synfig::error("Unable to setup longjump");
192 png_destroy_read_struct(&png_ptr, &info_ptr, &end_info);
194 //! \todo THROW SOMETHING
195 throw String("error on importer construction, *WRITEME*5");
200 png_set_read_user_chunk_fn(png_ptr, this, &png_mptr::read_chunk_callback);
202 // man libpng tells me:
203 // You must use png_transforms and not call any
204 // png_set_transform() functions when you use png_read_png().
205 // but we used png_set_gamma(), which may be why we were seeing a crash at the end
206 // png_read_png(png_ptr, info_ptr, PNG_TRANSFORM_PACKING|PNG_TRANSFORM_STRIP_16, NULL);
208 png_read_update_info(png_ptr, info_ptr);
209 png_uint_32 rowbytes = png_get_rowbytes(png_ptr, info_ptr);
211 // allocate buffer to read image data into
212 png_bytep *row_pointers=new png_bytep[height];
213 png_byte *data = new png_byte[rowbytes*height];
214 for (png_uint_32 i = 0; i < height; i++)
215 row_pointers[i] = &(data[rowbytes*i]);
217 png_read_image(png_ptr, row_pointers);
221 surface_buffer.set_wh(width,height);
225 case PNG_COLOR_TYPE_RGB:
226 for(y=0;y<surface_buffer.get_h();y++)
227 for(x=0;x<surface_buffer.get_w();x++)
229 float r=gamma().r_U8_to_F32((unsigned char)row_pointers[y][x*3+0]);
230 float g=gamma().g_U8_to_F32((unsigned char)row_pointers[y][x*3+1]);
231 float b=gamma().b_U8_to_F32((unsigned char)row_pointers[y][x*3+2]);
232 surface_buffer[y][x]=Color(
239 surface_buffer[y][x]=Color(
240 (float)(unsigned char)row_pointers[y][x*3+0]*(1.0/255.0),
241 (float)(unsigned char)row_pointers[y][x*3+1]*(1.0/255.0),
242 (float)(unsigned char)row_pointers[y][x*3+2]*(1.0/255.0),
249 case PNG_COLOR_TYPE_RGB_ALPHA:
250 for(y=0;y<surface_buffer.get_h();y++)
251 for(x=0;x<surface_buffer.get_w();x++)
253 float r=gamma().r_U8_to_F32((unsigned char)row_pointers[y][x*4+0]);
254 float g=gamma().g_U8_to_F32((unsigned char)row_pointers[y][x*4+1]);
255 float b=gamma().b_U8_to_F32((unsigned char)row_pointers[y][x*4+2]);
256 surface_buffer[y][x]=Color(
260 (float)(unsigned char)row_pointers[y][x*4+3]*(1.0/255.0)
263 surface_buffer[y][x]=Color(
264 (float)(unsigned char)row_pointers[y][x*4+0]*(1.0/255.0),
265 (float)(unsigned char)row_pointers[y][x*4+1]*(1.0/255.0),
266 (float)(unsigned char)row_pointers[y][x*4+2]*(1.0/255.0),
267 (float)(unsigned char)row_pointers[y][x*4+3]*(1.0/255.0)
273 case PNG_COLOR_TYPE_GRAY:
274 for(y=0;y<surface_buffer.get_h();y++)
275 for(x=0;x<surface_buffer.get_w();x++)
277 float gray=gamma().g_U8_to_F32((unsigned char)row_pointers[y][x]);
278 //float gray=(float)(unsigned char)row_pointers[y][x]*(1.0/255.0);
279 surface_buffer[y][x]=Color(
288 case PNG_COLOR_TYPE_GRAY_ALPHA:
289 for(y=0;y<surface_buffer.get_h();y++)
290 for(x=0;x<surface_buffer.get_w();x++)
292 float gray=gamma().g_U8_to_F32((unsigned char)row_pointers[y][x*2]);
293 // float gray=(float)(unsigned char)row_pointers[y][x*2]*(1.0/255.0);
294 surface_buffer[y][x]=Color(
298 (float)(unsigned char)row_pointers[y][x*2+1]*(1.0/255.0)
303 case PNG_COLOR_TYPE_PALETTE:
304 synfig::warning("png_mptr: Paletted PNGs aren't yet fully supported.");
305 for(y=0;y<surface_buffer.get_h();y++)
306 for(x=0;x<surface_buffer.get_w();x++)
308 float r=gamma().r_U8_to_F32((unsigned char)png_ptr->palette[row_pointers[y][x]].red);
309 float g=gamma().g_U8_to_F32((unsigned char)png_ptr->palette[row_pointers[y][x]].green);
310 float b=gamma().b_U8_to_F32((unsigned char)png_ptr->palette[row_pointers[y][x]].blue);
311 surface_buffer[y][x]=Color(
320 synfig::error("png_mptr: error: Unsupported color type");
321 //! \todo THROW SOMETHING
322 throw String("error on importer construction, *WRITEME*6");
326 png_read_end(png_ptr, end_info);
327 png_destroy_read_struct(&png_ptr, &info_ptr, &end_info);
330 delete [] row_pointers;
334 png_mptr::~png_mptr()
339 png_mptr::get_frame(synfig::Surface &surface,Time, synfig::ProgressCallback */*cb*/)
341 surface.mirror(surface_buffer);
342 // surface=surface_buffer;