X-Git-Url: http://mj.ucw.cz/gitweb/?a=blobdiff_plain;f=images%2Fio-libmagick.c;h=3636163174a183cbf1400713c965ccad5e6b9068;hb=1bc3bb66e47ec02003658fb3040aef0ffd7b7540;hp=d48553a4608db00a3fa633f8de8eedf4d2499dec;hpb=966305189fb3c543170117da57f110a8e9851ac2;p=libucw.git diff --git a/images/io-libmagick.c b/images/io-libmagick.c index d48553a4..36361631 100644 --- a/images/io-libmagick.c +++ b/images/io-libmagick.c @@ -1,5 +1,5 @@ /* - * Image Library -- GrapgicsMagick + * Image Library -- GraphicsMagick (slow fallback library) * * (c) 2006 Pavel Charvat * @@ -7,108 +7,415 @@ * of the GNU Lesser General Public License. */ -#define LOCAL_DEBUG +#undef LOCAL_DEBUG + +#include +#include +#include +#include +#include +#include +#include -#include "lib/lib.h" -#include "images/images.h" #include #include #include #include +#include + +#define MAX_FILE_SIZE (1 << 30) +#define QUANTUM_SCALE (QuantumDepth - 8) +#define QUANTUM_TO_BYTE(x) ((uint)(x) >> QUANTUM_SCALE) +#define BYTE_TO_QUANTUM(x) ((uint)(x) << QUANTUM_SCALE) +#define ALPHA_TO_BYTE(x) (255 - QUANTUM_TO_BYTE(x)) +#define BYTE_TO_ALPHA(x) (BYTE_TO_QUANTUM(255 - (x))) + +static pthread_mutex_t libmagick_mutex = PTHREAD_MUTEX_INITIALIZER; +static uint libmagick_counter; + +struct magick_read_data { + ExceptionInfo exception; + ImageInfo *info; + Image *image; +}; int -libmagick_read_header(struct image_io *io) +libmagick_init(struct image_io *io UNUSED) { - image_thread_err(io->thread, IMAGE_ERR_NOT_IMPLEMENTED, "GraphicsMagick read not implemented."); - return 0; + pthread_mutex_lock(&libmagick_mutex); + if (!libmagick_counter++) + InitializeMagick(NULL); + pthread_mutex_unlock(&libmagick_mutex); + return 1; } -int -libmagick_read_data(struct image_io *io UNUSED) +void +libmagick_cleanup(struct image_io *io UNUSED) { - ASSERT(0); + pthread_mutex_lock(&libmagick_mutex); + if (!--libmagick_counter) + DestroyMagick(); + pthread_mutex_unlock(&libmagick_mutex); } -int -libmagick_write(struct image_io *io) +static void +libmagick_destroy_read_data(struct magick_read_data *rd) { - image_thread_err(io->thread, IMAGE_ERR_NOT_IMPLEMENTED, "GraphicsMagick write not implemented."); - return 0; + if (rd->image) + DestroyImage(rd->image); + DestroyImageInfo(rd->info); + DestroyExceptionInfo(&rd->exception); } -#if 0 -struct magick_internals { - ExceptionInfo exception; - QuantizeInfo quantize; - ImageInfo *info; -}; - -static inline void -magick_cleanup(struct image_io *io) +static void +libmagick_read_cancel(struct image_io *io) { - DestroyImageInfo(io->internals->info); - DestroyExceptionInfo(&io->internals->exception); - DestroyMagick(); + DBG("libmagick_read_cancel()"); + + struct magick_read_data *rd = io->read_data; + libmagick_destroy_read_data(rd); } -static int -magick_read_header(struct image_io *io) +int +libmagick_read_header(struct image_io *io) { - DBG("magick_read_header()"); - struct magick_internals *i = io->internals = mp_alloc(io->pool, sizeof(*i)); + DBG("libmagick_read_header()"); - InitializeMagick(NULL); - GetExceptionInfo(&i->exception); - i->info = CloneImageInfo(NULL); - i->info->subrange = 1; - GetQuantizeInfo(&i->quantize); - i->quantize.colorspace = RGBColorspace; + /* Read entire stream */ + ucw_off_t file_size = bfilesize(io->fastbuf) - btell(io->fastbuf); + if (unlikely(file_size > MAX_FILE_SIZE)) + { + IMAGE_ERROR(io->context, IMAGE_ERROR_READ_FAILED, "Too long stream."); + return 0; + } + uint buf_size = file_size; + byte *buf = xmalloc(buf_size); + breadb(io->fastbuf, buf, buf_size); - uns len = bfilesize(io->fastbuf); - byte *buf = mp_alloc(io->pool, len); - len = bread(io->fastbuf, buf, len); + /* Allocate read structure */ + struct magick_read_data *rd = io->read_data = mp_alloc_zero(io->internal_pool, sizeof(*rd)); - Image *image = BlobToImage(magick_info, imo->thumb_data, imo->thumb_size, &magick_exception); - if (unlikely(!image)) - goto error; + /* Initialize GraphicsMagick */ + GetExceptionInfo(&rd->exception); + rd->info = CloneImageInfo(NULL); + rd->info->subrange = 1; + + /* Read the image */ + rd->image = BlobToImage(rd->info, buf, buf_size, &rd->exception); + xfree(buf); + if (unlikely(!rd->image)) + { + IMAGE_ERROR(io->context, IMAGE_ERROR_READ_FAILED, "GraphicsMagick failed to read the image."); + goto err; + } + if (unlikely(rd->image->columns > image_max_dim || rd->image->rows > image_max_dim)) + { + IMAGE_ERROR(io->context, IMAGE_ERROR_INVALID_DIMENSIONS, "Image too large."); + goto err; + } - // FIXME + /* Fill image parameters */ + io->cols = rd->image->columns; + io->rows = rd->image->rows; + switch (rd->image->colorspace) + { + case GRAYColorspace: + io->flags = COLOR_SPACE_GRAYSCALE; + break; + default: + io->flags = COLOR_SPACE_RGB; + break; + } + if (rd->image->matte) + io->flags |= IMAGE_ALPHA; + io->number_of_colors = rd->image->colors; + if (rd->image->storage_class == PseudoClass && rd->image->compression != JPEGCompression) + io->flags |= IMAGE_IO_HAS_PALETTE; + + io->read_cancel = libmagick_read_cancel; return 1; -error: - magick_cleanup(io); + +err: + libmagick_destroy_read_data(rd); return 0; } -static int -magick_read_data(struct image_io *io) +static inline byte +libmagick_pixel_to_gray(PixelPacket *pixel) +{ + return rgb_to_gray_func(pixel->red, pixel->green, pixel->blue) >> QUANTUM_SCALE; +} + +int +libmagick_read_data(struct image_io *io) { - DBG("magick_read_data()"); + DBG("libmagick_read_data()"); - // FIXME + struct magick_read_data *rd = io->read_data; - magick_cleanup(io); - return 1; + /* Quantize image */ + switch (rd->image->colorspace) + { + case RGBColorspace: + case GRAYColorspace: + break; + default: ; + QuantizeInfo quantize; + GetQuantizeInfo(&quantize); + quantize.colorspace = RGBColorspace; + QuantizeImage(&quantize, rd->image); + break; + } + + /* Prepare the image */ + struct image_io_read_data_internals rdi; + uint read_flags = io->flags; + uint cs = read_flags & IMAGE_COLOR_SPACE; + if (cs != COLOR_SPACE_GRAYSCALE && cs != COLOR_SPACE_RGB) + read_flags = (read_flags & ~IMAGE_COLOR_SPACE & IMAGE_PIXEL_FORMAT) | COLOR_SPACE_RGB; + if ((read_flags & IMAGE_IO_USE_BACKGROUND) && !(read_flags & IMAGE_ALPHA)) + read_flags = (read_flags & IMAGE_CHANNELS_FORMAT) | IMAGE_ALPHA; + if (unlikely(!image_io_read_data_prepare(&rdi, io, rd->image->columns, rd->image->rows, read_flags))) + { + libmagick_destroy_read_data(rd); + return 0; + } + + /* Acquire pixels */ + PixelPacket *src = (PixelPacket *)AcquireImagePixels(rd->image, 0, 0, rd->image->columns, rd->image->rows, &rd->exception); + if (unlikely(!src)) + { + IMAGE_ERROR(io->context, IMAGE_ERROR_READ_FAILED, "Cannot acquire image pixels."); + libmagick_destroy_read_data(rd); + image_io_read_data_break(&rdi, io); + return 0; + } + + /* Convert pixels */ + switch (rdi.image->pixel_size) + { + case 1: +# define IMAGE_WALK_PREFIX(x) walk_##x +# define IMAGE_WALK_INLINE +# define IMAGE_WALK_IMAGE (rdi.image) +# define IMAGE_WALK_UNROLL 4 +# define IMAGE_WALK_COL_STEP 1 +# define IMAGE_WALK_DO_STEP do{ \ + walk_pos[0] = libmagick_pixel_to_gray(src); \ + src++; }while(0) +# include + break; + + case 2: +# define IMAGE_WALK_PREFIX(x) walk_##x +# define IMAGE_WALK_INLINE +# define IMAGE_WALK_IMAGE (rdi.image) +# define IMAGE_WALK_UNROLL 4 +# define IMAGE_WALK_COL_STEP 2 +# define IMAGE_WALK_DO_STEP do{ \ + walk_pos[0] = libmagick_pixel_to_gray(src); \ + walk_pos[1] = ALPHA_TO_BYTE(src->opacity); \ + src++; }while(0) +# include + break; + + case 3: +# define IMAGE_WALK_PREFIX(x) walk_##x +# define IMAGE_WALK_INLINE +# define IMAGE_WALK_IMAGE (rdi.image) +# define IMAGE_WALK_UNROLL 4 +# define IMAGE_WALK_COL_STEP 3 +# define IMAGE_WALK_DO_STEP do{ \ + walk_pos[0] = QUANTUM_TO_BYTE(src->red); \ + walk_pos[1] = QUANTUM_TO_BYTE(src->green); \ + walk_pos[2] = QUANTUM_TO_BYTE(src->blue); \ + src++; }while(0) +# include + break; + + case 4: +# define IMAGE_WALK_PREFIX(x) walk_##x +# define IMAGE_WALK_INLINE +# define IMAGE_WALK_IMAGE (rdi.image) +# define IMAGE_WALK_UNROLL 4 +# define IMAGE_WALK_COL_STEP 4 +# define IMAGE_WALK_DO_STEP do{ \ + walk_pos[0] = QUANTUM_TO_BYTE(src->red); \ + walk_pos[1] = QUANTUM_TO_BYTE(src->green); \ + walk_pos[2] = QUANTUM_TO_BYTE(src->blue); \ + walk_pos[3] = ALPHA_TO_BYTE(src->opacity); \ + src++; }while(0) +# include + break; + + default: + ASSERT(0); + } + + /* Free GraphicsMagick structures */ + libmagick_destroy_read_data(rd); + + /* Finish the image */ + return image_io_read_data_finish(&rdi, io); } -static int -magick_decompress_thumbnail(struct image_obj *imo) +int +libmagick_write(struct image_io *io) { - DBG("Quantizing image"); - QuantizeImage(&magick_quantize, image); - DBG("Converting pixels"); - PixelPacket *pixels = (PixelPacket *)AcquireImagePixels(image, 0, 0, image->columns, image->rows, &magick_exception); - ASSERT(pixels); - uns size = image->columns * image->rows; - byte *p = imo->thumb.pixels = mp_alloc(imo->pool, imo->thumb.size = size * 3); - for (uns i = 0; i < size; i++) - { - p[0] = pixels->red >> (QuantumDepth - 8); - p[1] = pixels->green >> (QuantumDepth - 8); - p[2] = pixels->blue >> (QuantumDepth - 8); - p += 3; - pixels++; + DBG("libmagick_write()"); + + /* Initialize GraphicsMagick */ + int result = 0; + ExceptionInfo exception; + ImageInfo *info; + GetExceptionInfo(&exception); + info = CloneImageInfo(NULL); + + /* Setup image parameters and allocate the image*/ + struct image *img = io->image; + switch (img->flags & IMAGE_COLOR_SPACE) + { + case COLOR_SPACE_GRAYSCALE: + info->colorspace = GRAYColorspace; + break; + case COLOR_SPACE_RGB: + info->colorspace = RGBColorspace; + break; + default: + IMAGE_ERROR(io->context, IMAGE_ERROR_WRITE_FAILED, "Unsupported color space."); + goto err; + } + switch (io->format) + { + case IMAGE_FORMAT_JPEG: + strcpy(info->magick, "JPEG"); + if (io->jpeg_quality) + info->quality = MIN(io->jpeg_quality, 100); + break; + case IMAGE_FORMAT_PNG: + strcpy(info->magick, "PNG"); + break; + case IMAGE_FORMAT_GIF: + strcpy(info->magick, "GIF"); + break; + default: + ASSERT(0); + } + Image *image = AllocateImage(info); + if (unlikely(!image)) + { + IMAGE_ERROR(io->context, IMAGE_ERROR_WRITE_FAILED, "GraphicsMagick failed to allocate the image."); + goto err; } + image->columns = img->cols; + image->rows = img->rows; + + /* Get pixels */ + PixelPacket *pixels = SetImagePixels(image, 0, 0, img->cols, img->rows), *dest = pixels; + if (unlikely(!pixels)) + { + IMAGE_ERROR(io->context, IMAGE_ERROR_WRITE_FAILED, "Cannot get GraphicsMagick pixels."); + goto err2; + } + + /* Convert pixels */ + switch (img->pixel_size) + { + case 1: +# define IMAGE_WALK_PREFIX(x) walk_##x +# define IMAGE_WALK_INLINE +# define IMAGE_WALK_IMAGE img +# define IMAGE_WALK_UNROLL 4 +# define IMAGE_WALK_COL_STEP 1 +# define IMAGE_WALK_DO_STEP do{ \ + dest->red = BYTE_TO_QUANTUM(walk_pos[0]); \ + dest->green = BYTE_TO_QUANTUM(walk_pos[0]); \ + dest->blue = BYTE_TO_QUANTUM(walk_pos[0]); \ + dest->opacity = 0; \ + dest++; }while(0) +# include + break; + + case 2: +# define IMAGE_WALK_PREFIX(x) walk_##x +# define IMAGE_WALK_INLINE +# define IMAGE_WALK_IMAGE img +# define IMAGE_WALK_UNROLL 4 +# define IMAGE_WALK_COL_STEP 2 +# define IMAGE_WALK_DO_STEP do{ \ + dest->red = BYTE_TO_QUANTUM(walk_pos[0]); \ + dest->green = BYTE_TO_QUANTUM(walk_pos[0]); \ + dest->blue = BYTE_TO_QUANTUM(walk_pos[0]); \ + dest->opacity = BYTE_TO_ALPHA(walk_pos[1]); \ + dest++; }while(0) +# include + break; + + case 3: +# define IMAGE_WALK_PREFIX(x) walk_##x +# define IMAGE_WALK_INLINE +# define IMAGE_WALK_IMAGE img +# define IMAGE_WALK_UNROLL 4 +# define IMAGE_WALK_COL_STEP 3 +# define IMAGE_WALK_DO_STEP do{ \ + dest->red = BYTE_TO_QUANTUM(walk_pos[0]); \ + dest->green = BYTE_TO_QUANTUM(walk_pos[1]); \ + dest->blue = BYTE_TO_QUANTUM(walk_pos[2]); \ + dest->opacity = 0; \ + dest++; }while(0) +# include + break; + + case 4: +# define IMAGE_WALK_PREFIX(x) walk_##x +# define IMAGE_WALK_INLINE +# define IMAGE_WALK_IMAGE img +# define IMAGE_WALK_UNROLL 4 +# define IMAGE_WALK_COL_STEP 4 +# define IMAGE_WALK_DO_STEP do{ \ + dest->red = BYTE_TO_QUANTUM(walk_pos[0]); \ + dest->green = BYTE_TO_QUANTUM(walk_pos[1]); \ + dest->blue = BYTE_TO_QUANTUM(walk_pos[2]); \ + dest->opacity = BYTE_TO_ALPHA(walk_pos[3]); \ + dest++; }while(0) +# include + break; + + default: + ASSERT(0); + } + + /* Store pixels */ + if (unlikely(!SyncImagePixels(image))) + { + IMAGE_ERROR(io->context, IMAGE_ERROR_WRITE_FAILED, "Cannot sync GraphicsMagick pixels."); + goto err2; + } + + /* Write image */ + size_t buf_len = 0; + void *buf = ImageToBlob(info, image, &buf_len, &exception); + if (unlikely(!buf)) + { + IMAGE_ERROR(io->context, IMAGE_ERROR_WRITE_FAILED, "GraphicsMagick failed to compress the image."); + goto err2; + } + if (unlikely(buf_len > MAX_FILE_SIZE)) + { + IMAGE_ERROR(io->context, IMAGE_ERROR_WRITE_FAILED, "Image too large."); + goto err2; + } + + /* Write to stream */ + bwrite(io->fastbuf, buf, buf_len); + + /* Success */ + result = 1; + +err2: DestroyImage(image); - return 1; +err: + DestroyImageInfo(info); + DestroyExceptionInfo(&exception); + return result; } -#endif