X-Git-Url: http://mj.ucw.cz/gitweb/?a=blobdiff_plain;f=images%2Fimage.c;h=914d0f17c4f52112f3fabe20e637b3a2ff52d0d3;hb=04cd223644f534d556f12e01fe1baa4e21b7cc1b;hp=e04d839c573f86a4cbc05acc1343b4bb22d12b2b;hpb=35d602a8f8676b8988eca13086c63f2026c1b952;p=libucw.git diff --git a/images/image.c b/images/image.c index e04d839c..914d0f17 100644 --- a/images/image.c +++ b/images/image.c @@ -41,17 +41,10 @@ image_thread_err_format(struct image_thread *it, uns code, char *msg, ...) va_end(args); } -struct image * -image_new(struct image_thread *it, uns cols, uns rows, uns flags, struct mempool *pool) +static inline uns +flags_to_pixel_size(uns flags) { - DBG("image_new(cols=%u rows=%u flags=0x%x pool=%p)", cols, rows, flags, pool); - if (cols > IMAGE_MAX_SIZE || rows > IMAGE_MAX_SIZE) - { - image_thread_err(it, IMAGE_ERR_INVALID_DIMENSIONS, "Image dimension(s) too large"); - return NULL; - } - struct image *img; - uns pixel_size, row_size, image_size, align; + uns pixel_size; switch (flags & IMAGE_COLOR_SPACE) { case COLOR_SPACE_GRAYSCALE: @@ -65,6 +58,22 @@ image_new(struct image_thread *it, uns cols, uns rows, uns flags, struct mempool } if (flags & IMAGE_ALPHA) pixel_size++; + return pixel_size; +} + +struct image * +image_new(struct image_thread *it, uns cols, uns rows, uns flags, struct mempool *pool) +{ + DBG("image_new(cols=%u rows=%u flags=0x%x pool=%p)", cols, rows, flags, pool); + flags &= IMAGE_NEW_FLAGS; + if (unlikely(!image_dimensions_valid(cols, rows))) + { + image_thread_err_format(it, IMAGE_ERR_INVALID_DIMENSIONS, "Invalid image dimensions (%ux%u)", cols, rows); + return NULL; + } + struct image *img; + uns pixel_size, row_size, align; + pixel_size = flags_to_pixel_size(flags); switch (pixel_size) { case 1: @@ -89,17 +98,18 @@ image_new(struct image_thread *it, uns cols, uns rows, uns flags, struct mempool row_size = ALIGN(row_size, align); u64 image_size_64 = (u64)row_size * rows; u64 bytes_64 = image_size_64 + (sizeof(struct image) + IMAGE_SSE_ALIGN_SIZE - 1 + sizeof(uns)); - if (bytes_64 > MAX_IMAGE_BYTES) + if (unlikely(bytes_64 > MAX_IMAGE_BYTES)) { image_thread_err(it, IMAGE_ERR_INVALID_DIMENSIONS, "Image does not fit in memory"); return NULL; } - if (!(image_size = image_size_64)) + if (pool) + img = mp_alloc(pool, bytes_64); + else { - image_thread_err(it, IMAGE_ERR_INVALID_DIMENSIONS, "Zero dimension(s)"); - return NULL; + img = xmalloc(bytes_64); + flags |= IMAGE_NEED_DESTROY; } - img = pool ? mp_alloc(pool, (uns)bytes_64) : xmalloc((uns)bytes_64); bzero(img, sizeof(struct image)); byte *p = (byte *)img + sizeof(struct image); img->pixels = ALIGN_PTR(p, IMAGE_SSE_ALIGN_SIZE); @@ -108,7 +118,7 @@ image_new(struct image_thread *it, uns cols, uns rows, uns flags, struct mempool img->cols = cols; img->rows = rows; img->row_size = row_size; - img->image_size = image_size; + img->image_size = image_size_64; DBG("img=%p flags=0x%x pixel_size=%u row_size=%u image_size=%u pixels=%p", img, img->flags, img->pixel_size, img->row_size, img->image_size, img->pixels); return img; @@ -119,18 +129,20 @@ image_clone(struct image_thread *it, struct image *src, uns flags, struct mempoo { DBG("image_clone(src=%p flags=0x%x pool=%p)", src, src->flags, pool); struct image *img; - flags &= ~IMAGE_CHANNELS_FORMAT; + flags &= IMAGE_NEW_FLAGS & ~IMAGE_CHANNELS_FORMAT; flags |= src->flags & IMAGE_CHANNELS_FORMAT; if (!(img = image_new(it, src->cols, src->rows, flags, pool))) return NULL; if (img->image_size) { - if (src->pixel_size != img->pixel_size) + if (src->pixel_size != img->pixel_size) /* conversion between aligned and unaligned RGB */ { - struct image *sec_img = src; +# define IMAGE_WALK_PREFIX(x) walk_##x # define IMAGE_WALK_INLINE +# define IMAGE_WALK_IMAGE img +# define IMAGE_WALK_SEC_IMAGE src # define IMAGE_WALK_DOUBLE -# define IMAGE_WALK_DO_STEP do{ *(u32 *)pos = *(u32 *)sec_pos; }while(0) +# define IMAGE_WALK_DO_STEP do{ walk_pos[0] = walk_sec_pos[0]; walk_pos[1] = walk_sec_pos[1]; walk_pos[2] = walk_sec_pos[2]; }while(0) # include "images/image-walk.h" } else if (src->row_size != img->row_size) @@ -155,7 +167,8 @@ void image_destroy(struct image *img) { DBG("image_destroy(img=%p)", img); - xfree((byte *)img); + if (img->flags & IMAGE_NEED_DESTROY) + xfree(img); } void @@ -166,6 +179,41 @@ image_clear(struct image_thread *it UNUSED, struct image *img) bzero(img->pixels, img->image_size); } +struct image * +image_init_matrix(struct image_thread *it, struct image *img, byte *pixels, uns cols, uns rows, uns row_size, uns flags) +{ + DBG("image_init_matrix(img=%p pixels=%p cols=%u rows=%u row_size=%u flags=0x%x)", img, pixels, cols, rows, row_size, flags); + if (unlikely(!image_dimensions_valid(cols, rows))) + { + image_thread_err_format(it, IMAGE_ERR_INVALID_DIMENSIONS, "Invalid image dimensions (%ux%u)", cols, rows); + return NULL; + } + img->pixels = pixels; + img->cols = cols; + img->rows = rows; + img->pixel_size = flags_to_pixel_size(flags); + img->row_size = row_size; + img->image_size = rows * row_size; + img->flags = flags & (IMAGE_NEW_FLAGS | IMAGE_GAPS_PROTECTED); + return img; +} + +struct image * +image_init_subimage(struct image_thread *it UNUSED, struct image *img, struct image *src, uns left, uns top, uns cols, uns rows) +{ + DBG("image_init_subimage(img=%p src=%p left=%u top=%u cols=%u rows=%u)", img, src, left, top, cols, rows); + ASSERT(left + cols <= src->cols && top + rows <= src->rows); + img->pixels = src->pixels + left * src->pixel_size + top * src->row_size; + img->cols = cols; + img->rows = rows; + img->pixel_size = src->pixel_size; + img->row_size = src->row_size; + img->image_size = src->row_size * rows; + img->flags = src->flags & IMAGE_NEW_FLAGS; + img->flags |= IMAGE_GAPS_PROTECTED; + return img; +} + byte * color_space_to_name(enum color_space cs) {