src/image-load-tiff.cc

   1 /*
   2  * Copyright (C) 1999 Mark Crichton
   3  * Copyright (C) 1999 The Free Software Foundation
   4  * Copyright (C) 2004 John Ellis
   5  * Copyright (C) 2008 - 2016 The Geeqie Team
   6  *
   7  * Authors: Mark Crichton <crichton@gimp.org>
   8  *          Federico Mena-Quintero <federico@gimp.org>
   9  *          Jonathan Blandford <jrb@redhat.com>
  10  *          S�ren Sandmann <sandmann@daimi.au.dk>
  11  *          Vladimir Nadvornik
  12  *
  13  * This program is free software; you can redistribute it and/or modify
  14  * it under the terms of the GNU General Public License as published by
  15  * the Free Software Foundation; either version 2 of the License, or
  16  * (at your option) any later version.
  17  *
  18  * This program is distributed in the hope that it will be useful,
  19  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  20  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  21  * GNU General Public License for more details.
  22  *
  23  * You should have received a copy of the GNU General Public License along
  24  * with this program; if not, write to the Free Software Foundation, Inc.,
  25  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  26  */
  27
  28 #include "main.h"
  29
  30 #include "image-load.h"
  31 #include "image-load-tiff.h"
  32
  33 #ifdef HAVE_TIFF
  34
  35 #include <tiffio.h>
  36
  37 struct ImageLoaderTiff {
  38         ImageLoaderBackendCbAreaUpdated area_updated_cb;
  39         ImageLoaderBackendCbSize size_cb;
  40         ImageLoaderBackendCbAreaPrepared area_prepared_cb;
  41
  42         gpointer data;
  43
  44         GdkPixbuf *pixbuf;
  45         guint requested_width;
  46         guint requested_height;
  47
  48         gboolean abort;
  49
  50         const guchar *buffer;
  51         toff_t used;
  52         toff_t pos;
  53         gint page_num;
  54         gint page_total;
  55 };
  56
  57 static void free_buffer (guchar *pixels, gpointer)
  58 {
  59         g_free (pixels);
  60 }
  61
  62 static tsize_t
  63 tiff_load_read (thandle_t handle, tdata_t buf, tsize_t size)
  64 {
  65         auto context = static_cast<ImageLoaderTiff *>(handle);
  66
  67         if (context->pos + size > context->used)
  68                 return 0;
  69
  70         memcpy (buf, context->buffer + context->pos, size);
  71         context->pos += size;
  72         return size;
  73 }
  74
  75 static tsize_t
  76 tiff_load_write (thandle_t, tdata_t, tsize_t)
  77 {
  78         return -1;
  79 }
  80
  81 static toff_t
  82 tiff_load_seek (thandle_t handle, toff_t offset, int whence)
  83 {
  84         auto context = static_cast<ImageLoaderTiff *>(handle);
  85
  86         switch (whence)
  87                 {
  88                 case SEEK_SET:
  89                         if (offset > context->used)
  90                                 return -1;
  91                         context->pos = offset;
  92                         break;
  93                 case SEEK_CUR:
  94                         if (offset + context->pos >= context->used)
  95                                 return -1;
  96                         context->pos += offset;
  97                         break;
  98                 case SEEK_END:
  99                         if (offset + context->used > context->used)
 100                                 return -1;
 101                         context->pos = context->used + offset;
 102                         break;
 103                 default:
 104                         return -1;
 105                 }
 106
 107         return context->pos;
 108 }
 109
 110 static int
 111 tiff_load_close (thandle_t)
 112 {
 113         return 0;
 114 }
 115
 116 static toff_t
 117 tiff_load_size (thandle_t handle)
 118 {
 119         auto context = static_cast<ImageLoaderTiff *>(handle);
 120         return context->used;
 121 }
 122
 123 static int
 124 tiff_load_map_file (thandle_t handle, tdata_t *buf, toff_t *size)
 125 {
 126         auto context = static_cast<ImageLoaderTiff *>(handle);
 127
 128         *buf = const_cast<guchar *>(context->buffer);
 129         *size = context->used;
 130
 131         return 0;
 132 }
 133
 134 static void
 135 tiff_load_unmap_file (thandle_t, tdata_t, toff_t)
 136 {
 137 }
 138
 139 static gboolean image_loader_tiff_load (gpointer loader, const guchar *buf, gsize count, GError **)
 140 {
 141         auto lt = static_cast<ImageLoaderTiff *>(loader);
 142
 143         TIFF *tiff;
 144         guchar *pixels = nullptr;
 145         gint width, height, rowstride;
 146         size_t bytes;
 147         guint32 rowsperstrip;
 148         gint dircount = 0;
 149
 150         lt->buffer = buf;
 151         lt->used = count;
 152         lt->pos = 0;
 153
 154         TIFFSetWarningHandler(nullptr);
 155
 156         tiff = TIFFClientOpen ( "libtiff-geeqie", "r", lt,
 157                                                         tiff_load_read, tiff_load_write,
 158                                                         tiff_load_seek, tiff_load_close,
 159                                                         tiff_load_size,
 160                                                         tiff_load_map_file, tiff_load_unmap_file);
 161         if (!tiff)
 162                 {
 163                 DEBUG_1("Failed to open TIFF image");
 164                 return FALSE;
 165                 }
 166
 167         do      {
 168                 dircount++;
 169                 } while (TIFFReadDirectory(tiff));
 170
 171         lt->page_total = dircount;
 172
 173     if (!TIFFSetDirectory(tiff, lt->page_num))
 174                 {
 175                 DEBUG_1("Failed to open TIFF image");
 176                 TIFFClose(tiff);
 177                 return FALSE;
 178                 }
 179
 180         if (!TIFFGetField (tiff, TIFFTAG_IMAGEWIDTH, &width))
 181                 {
 182                 DEBUG_1("Could not get image width (bad TIFF file)");
 183                 TIFFClose(tiff);
 184                 return FALSE;
 185                 }
 186
 187         if (!TIFFGetField (tiff, TIFFTAG_IMAGELENGTH, &height))
 188                 {
 189                 DEBUG_1("Could not get image height (bad TIFF file)");
 190                 TIFFClose(tiff);
 191                 return FALSE;
 192                 }
 193
 194         if (width <= 0 || height <= 0)
 195                 {
 196                 DEBUG_1("Width or height of TIFF image is zero");
 197                 TIFFClose(tiff);
 198                 return FALSE;
 199                 }
 200
 201         rowstride = width * 4;
 202         if (rowstride / 4 != width)
 203                 { /* overflow */
 204                 DEBUG_1("Dimensions of TIFF image too large: width %d", width);
 205                 TIFFClose(tiff);
 206                 return FALSE;
 207                 }
 208
 209         bytes = static_cast<size_t>(height) * rowstride;
 210         if (bytes / rowstride != static_cast<size_t>(height))
 211                 { /* overflow */
 212                 DEBUG_1("Dimensions of TIFF image too large: height %d", height);
 213                 TIFFClose(tiff);
 214                 return FALSE;
 215                 }
 216
 217         lt->requested_width = width;
 218         lt->requested_height = height;
 219         lt->size_cb(loader, lt->requested_width, lt->requested_height, lt->data);
 220
 221         pixels = static_cast<guchar *>(g_try_malloc (bytes));
 222
 223         if (!pixels)
 224                 {
 225                 DEBUG_1("Insufficient memory to open TIFF file: need %zu", bytes);
 226                 TIFFClose(tiff);
 227                 return FALSE;
 228                 }
 229
 230         lt->pixbuf = gdk_pixbuf_new_from_data (pixels, GDK_COLORSPACE_RGB, TRUE, 8,
 231                                                                                    width, height, rowstride,
 232                                                                                    free_buffer, nullptr);
 233         if (!lt->pixbuf)
 234                 {
 235                 g_free (pixels);
 236                 DEBUG_1("Insufficient memory to open TIFF file");
 237                 TIFFClose(tiff);
 238                 return FALSE;
 239                 }
 240
 241         lt->area_prepared_cb(loader, lt->data);
 242
 243         if (TIFFGetField(tiff, TIFFTAG_ROWSPERSTRIP, &rowsperstrip))
 244                 {
 245                 /* read by strip */
 246                 ptrdiff_t row;
 247                 const size_t line_bytes = width * sizeof(guint32);
 248                 auto wrk_line = static_cast<guchar *>(g_malloc(line_bytes));
 249
 250                 for (row = 0; row < height; row += rowsperstrip)
 251                         {
 252                         int rows_to_write, i_row;
 253
 254                         if (lt->abort) {
 255                                 break;
 256                         }
 257
 258                         /* Read the strip into an RGBA array */
 259                         if (!TIFFReadRGBAStrip(tiff, row, reinterpret_cast<guint32 *>(pixels + row * rowstride))) {
 260                                 break;
 261                         }
 262
 263                         /*
 264                          * Figure out the number of scanlines actually in this strip.
 265                         */
 266                         if (row + static_cast<int>(rowsperstrip) > height)
 267                                 rows_to_write = height - row;
 268                         else
 269                                 rows_to_write = rowsperstrip;
 270
 271                         /*
 272                          * For some reason the TIFFReadRGBAStrip() function chooses the
 273                          * lower left corner as the origin.  Vertically mirror scanlines.
 274                          */
 275                         for (i_row = 0; i_row < rows_to_write / 2; i_row++)
 276                                 {
 277                                 guchar *top_line, *bottom_line;
 278
 279                                 top_line = pixels + (row + i_row) * rowstride;
 280                                 bottom_line = pixels + (row + rows_to_write - i_row - 1) * rowstride;
 281
 282                                 memcpy(wrk_line, top_line, line_bytes);
 283                                 memcpy(top_line, bottom_line, line_bytes);
 284                                 memcpy(bottom_line, wrk_line, line_bytes);
 285                                 }
 286                         lt->area_updated_cb(loader, 0, row, width, rows_to_write, lt->data);
 287                         }
 288                 g_free(wrk_line);
 289                 }
 290         else
 291                 {
 292                 /* fallback, tiled tiff */
 293                 if (!TIFFReadRGBAImageOriented (tiff, width, height, reinterpret_cast<guint32 *>(pixels), ORIENTATION_TOPLEFT, 1))
 294                         {
 295                         TIFFClose(tiff);
 296                         return FALSE;
 297                         }
 298
 299 #if G_BYTE_ORDER == G_BIG_ENDIAN
 300                 /* Turns out that the packing used by TIFFRGBAImage depends on
 301                  * the host byte order...
 302                  */
 303                 {
 304                 guchar *ptr = pixels;
 305                 while (ptr < pixels + bytes)
 306                         {
 307                         guint32 pixel = *(guint32 *)ptr;
 308                         int r = TIFFGetR(pixel);
 309                         int g = TIFFGetG(pixel);
 310                         int b = TIFFGetB(pixel);
 311                         int a = TIFFGetA(pixel);
 312                         *ptr++ = r;
 313                         *ptr++ = g;
 314                         *ptr++ = b;
 315                         *ptr++ = a;
 316                         }
 317                 }
 318 #endif
 319
 320                 lt->area_updated_cb(loader, 0, 0, width, height, lt->data);
 321                 }
 322         TIFFClose(tiff);
 323
 324         return TRUE;
 325 }
 326
 327
 328 static gpointer image_loader_tiff_new(ImageLoaderBackendCbAreaUpdated area_updated_cb, ImageLoaderBackendCbSize size_cb, ImageLoaderBackendCbAreaPrepared area_prepared_cb, gpointer data)
 329 {
 330         auto loader = g_new0(ImageLoaderTiff, 1);
 331
 332         loader->area_updated_cb = area_updated_cb;
 333         loader->size_cb = size_cb;
 334         loader->area_prepared_cb = area_prepared_cb;
 335         loader->data = data;
 336         return loader;
 337 }
 338
 339
 340 static void image_loader_tiff_set_size(gpointer loader, int width, int height)
 341 {
 342         auto lt = static_cast<ImageLoaderTiff *>(loader);
 343         lt->requested_width = width;
 344         lt->requested_height = height;
 345 }
 346
 347 static GdkPixbuf* image_loader_tiff_get_pixbuf(gpointer loader)
 348 {
 349         auto lt = static_cast<ImageLoaderTiff *>(loader);
 350         return lt->pixbuf;
 351 }
 352
 353 static gchar* image_loader_tiff_get_format_name(gpointer)
 354 {
 355         return g_strdup("tiff");
 356 }
 357 static gchar** image_loader_tiff_get_format_mime_types(gpointer)
 358 {
 359         static const gchar *mime[] = {"image/tiff", nullptr};
 360         return g_strdupv(const_cast<gchar **>(mime));
 361 }
 362
 363 static gboolean image_loader_tiff_close(gpointer, GError **)
 364 {
 365         return TRUE;
 366 }
 367
 368 static void image_loader_tiff_abort(gpointer loader)
 369 {
 370         auto lt = static_cast<ImageLoaderTiff *>(loader);
 371         lt->abort = TRUE;
 372 }
 373
 374 static void image_loader_tiff_free(gpointer loader)
 375 {
 376         auto lt = static_cast<ImageLoaderTiff *>(loader);
 377         if (lt->pixbuf) g_object_unref(lt->pixbuf);
 378         g_free(lt);
 379 }
 380
 381 static void image_loader_tiff_set_page_num(gpointer loader, gint page_num)
 382 {
 383         auto lt = static_cast<ImageLoaderTiff *>(loader);
 384
 385         lt->page_num = page_num;
 386 }
 387
 388 static gint image_loader_tiff_get_page_total(gpointer loader)
 389 {
 390         auto lt = static_cast<ImageLoaderTiff *>(loader);
 391
 392         return lt->page_total;
 393 }
 394
 395 void image_loader_backend_set_tiff(ImageLoaderBackend *funcs)
 396 {
 397         funcs->loader_new = image_loader_tiff_new;
 398         funcs->set_size = image_loader_tiff_set_size;
 399         funcs->load = image_loader_tiff_load;
 400         funcs->write = nullptr;
 401         funcs->get_pixbuf = image_loader_tiff_get_pixbuf;
 402         funcs->close = image_loader_tiff_close;
 403         funcs->abort = image_loader_tiff_abort;
 404         funcs->free = image_loader_tiff_free;
 405
 406         funcs->get_format_name = image_loader_tiff_get_format_name;
 407         funcs->get_format_mime_types = image_loader_tiff_get_format_mime_types;
 408
 409         funcs->set_page_num = image_loader_tiff_set_page_num;
 410         funcs->get_page_total = image_loader_tiff_get_page_total;
 411 }
 412
 413
 414
 415 #endif
 416 /* vim: set shiftwidth=8 softtabstop=0 cindent cinoptions={1s: */