Fix 2nd part of STR #2520: added Fl_PNG_Image in-memory constructor.
This constructor has a 3rd argument, the size of the array holding the in-memory PNG image. This allows the constructor to check for errors in the image. git-svn-id: file:///fltk/svn/fltk/branches/branch-1.3@8287 ea41ed52-d2ee-0310-a9c1-e6b18d33e121
This commit is contained in:
parent
5af1531cd2
commit
0ae1054d07
@ -40,9 +40,12 @@
|
||||
*/
|
||||
class FL_EXPORT Fl_PNG_Image : public Fl_RGB_Image {
|
||||
|
||||
public:
|
||||
public:
|
||||
|
||||
Fl_PNG_Image(const char* filename);
|
||||
Fl_PNG_Image (const char *name_png, const unsigned char *buffer, int datasize);
|
||||
private:
|
||||
void load_png_(const char *name_png, const unsigned char *buffer_png, int datasize);
|
||||
};
|
||||
|
||||
#endif
|
||||
|
@ -42,37 +42,76 @@
|
||||
#include <stdlib.h>
|
||||
#include <FL/fl_utf8.h>
|
||||
|
||||
#if defined(HAVE_LIBPNG) && defined(HAVE_LIBZ)
|
||||
extern "C"
|
||||
{
|
||||
#if defined(HAVE_LIBPNG) && defined(HAVE_LIBZ)
|
||||
# include <zlib.h>
|
||||
# ifdef HAVE_PNG_H
|
||||
# include <png.h>
|
||||
# else
|
||||
# include <libpng/png.h>
|
||||
# endif // HAVE_PNG_H
|
||||
#endif // HAVE_LIBPNG && HAVE_LIBZ
|
||||
}
|
||||
|
||||
typedef struct {
|
||||
png_structp pp;
|
||||
const unsigned char *current;
|
||||
const unsigned char *last;
|
||||
} fl_png_memory;
|
||||
|
||||
static void png_read_data_from_mem( png_structp png_ptr, //pointer to our data
|
||||
png_bytep data, // where to copy the image data for libpng computing
|
||||
png_size_t length) // length of data to copy
|
||||
{
|
||||
fl_png_memory *png_mem_data = (fl_png_memory*)png_get_io_ptr(png_ptr); // get the pointer to our struct
|
||||
if (png_mem_data->current + length > png_mem_data->last) {
|
||||
png_error(png_mem_data->pp, "Invalid attempt to read row data");
|
||||
return;
|
||||
}
|
||||
/* copy data from image buffer */
|
||||
memcpy (data, png_mem_data->current, length);
|
||||
/* advance in the memory data */
|
||||
png_mem_data->current += length;
|
||||
}
|
||||
#endif // HAVE_LIBPNG && HAVE_LIBZ
|
||||
|
||||
|
||||
/**
|
||||
The constructor loads the named PNG image from the given png filename.
|
||||
<P>The destructor free all memory and server resources that are used by
|
||||
<P>The destructor frees all memory and server resources that are used by
|
||||
the image.
|
||||
*/
|
||||
Fl_PNG_Image::Fl_PNG_Image(const char *png) // I - File to read
|
||||
: Fl_RGB_Image(0,0,0) {
|
||||
Fl_PNG_Image::Fl_PNG_Image (const char *filename): Fl_RGB_Image(0,0,0)
|
||||
{
|
||||
load_png_(filename, NULL, 0);
|
||||
}
|
||||
|
||||
/** Constructor that reads a PNG image from memory.
|
||||
\param name_png A name given to this image
|
||||
\param buffer Pointer to the start of the PNG image in memory
|
||||
\param maxsize Size in bytes of the memory buffer containing the PNG image
|
||||
*/
|
||||
Fl_PNG_Image::Fl_PNG_Image (
|
||||
const char *name_png, const unsigned char *buffer, int maxsize): Fl_RGB_Image(0,0,0)
|
||||
{
|
||||
load_png_(name_png, buffer, maxsize);
|
||||
}
|
||||
|
||||
void Fl_PNG_Image::load_png_(const char *name_png, const unsigned char *buffer_png, int maxsize)
|
||||
{
|
||||
#if defined(HAVE_LIBPNG) && defined(HAVE_LIBZ)
|
||||
int i; // Looping var
|
||||
FILE *fp; // File pointer
|
||||
int channels; // Number of color channels
|
||||
png_structp pp; // PNG read pointer
|
||||
png_infop info; // PNG info pointers
|
||||
png_bytep *rows; // PNG row pointers
|
||||
int i; // Looping var
|
||||
FILE *fp; // File pointer
|
||||
int channels; // Number of color channels
|
||||
png_structp pp; // PNG read pointer
|
||||
png_infop info; // PNG info pointers
|
||||
png_bytep *rows;// PNG row pointers
|
||||
fl_png_memory png_mem_data;
|
||||
int from_memory = (buffer_png != NULL); // true iff reading image from memory
|
||||
|
||||
|
||||
// Open the PNG file...
|
||||
if ((fp = fl_fopen(png, "rb")) == NULL) return;
|
||||
if (!from_memory) {
|
||||
if ((fp = fl_fopen(name_png, "rb")) == NULL) return;
|
||||
}
|
||||
|
||||
// Setup the PNG data structures...
|
||||
pp = png_create_read_struct(PNG_LIBPNG_VER_STRING, NULL, NULL, NULL);
|
||||
@ -80,12 +119,19 @@ Fl_PNG_Image::Fl_PNG_Image(const char *png) // I - File to read
|
||||
|
||||
if (setjmp(png_jmpbuf(pp)))
|
||||
{
|
||||
Fl::warning("PNG file \"%s\" contains errors!\n", png);
|
||||
Fl::warning("PNG file \"%s\" contains errors!\n", name_png);
|
||||
return;
|
||||
}
|
||||
|
||||
// Initialize the PNG read "engine"...
|
||||
png_init_io(pp, fp);
|
||||
if (from_memory) {
|
||||
png_mem_data.current = buffer_png;
|
||||
png_mem_data.last = buffer_png + maxsize;
|
||||
png_mem_data.pp = pp;
|
||||
// Initialize the function pointer to the PNG read "engine"...
|
||||
png_set_read_fn (pp, (png_voidp) &png_mem_data, png_read_data_from_mem);
|
||||
} else {
|
||||
png_init_io(pp, fp); // Initialize the PNG file read "engine"...
|
||||
}
|
||||
|
||||
// Get the image dimensions and convert to grayscale or RGB...
|
||||
png_read_info(pp, info);
|
||||
@ -101,7 +147,7 @@ Fl_PNG_Image::Fl_PNG_Image(const char *png) // I - File to read
|
||||
int num_trans = 0;
|
||||
png_get_tRNS(pp, info, 0, &num_trans, 0);
|
||||
if ((png_get_color_type(pp, info) & PNG_COLOR_MASK_ALPHA) || (num_trans != 0))
|
||||
channels ++;
|
||||
channels ++;
|
||||
|
||||
w((int)(png_get_image_width(pp, info)));
|
||||
h((int)(png_get_image_height(pp, info)));
|
||||
@ -150,7 +196,7 @@ Fl_PNG_Image::Fl_PNG_Image(const char *png) // I - File to read
|
||||
png_read_end(pp, info);
|
||||
png_destroy_read_struct(&pp, &info, NULL);
|
||||
|
||||
fclose(fp);
|
||||
if (!from_memory) fclose(fp);
|
||||
#endif // HAVE_LIBPNG && HAVE_LIBZ
|
||||
}
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user