Generate sRGB thumbnails for images with embedded color profile data
[geeqie.git] / src / thumb_standard.c
1 /*
2  * Copyright (C) 2006 John Ellis
3  * Copyright (C) 2008 - 2016 The Geeqie Team
4  *
5  * Author: John Ellis
6  *
7  * This program is free software; you can redistribute it and/or modify
8  * it under the terms of the GNU General Public License as published by
9  * the Free Software Foundation; either version 2 of the License, or
10  * (at your option) any later version.
11  *
12  * This program is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15  * GNU General Public License for more details.
16  *
17  * You should have received a copy of the GNU General Public License along
18  * with this program; if not, write to the Free Software Foundation, Inc.,
19  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
20  */
21
22 #include "main.h"
23 #include "thumb_standard.h"
24
25 #include "cache.h"
26 #include "image-load.h"
27 #include "md5-util.h"
28 #include "pixbuf_util.h"
29 #include "ui_fileops.h"
30 #include "filedata.h"
31 #include "exif.h"
32 #include "metadata.h"
33 #include "color-man.h"
34
35
36 /**
37  * @file
38  * 
39  * This thumbnail caching implementation attempts to conform
40  * to the Thumbnail Managing Standard proposed on freedesktop.org
41  * The standard is documented here: \n
42  *   http://triq.net/~jens/thumbnail-spec/index.html \n
43  *  (why isn't it actually hosted on freedesktop.org?)
44  *
45  * This code attempts to conform to version 0.7.0 of the standard.
46  *
47  * Notes:
48  *   > Validation of the thumb's embedded uri is a simple strcmp between our
49  *   > version of the escaped uri and the thumb's escaped uri. But not all uri
50  *   > escape functions escape the same set of chars, comparing the unescaped
51  *   > versions may be more accurate. \n
52  *   > Only Thumb::URI and Thumb::MTime are stored in a thumb at this time.
53  *     Storing the Size, Width, Height should probably be implemented.
54  */
55
56
57 #define THUMB_SIZE_NORMAL 128
58 #define THUMB_SIZE_LARGE  256
59
60 #define THUMB_MARKER_URI    "tEXt::Thumb::URI"
61 #define THUMB_MARKER_MTIME  "tEXt::Thumb::MTime"
62 #define THUMB_MARKER_SIZE   "tEXt::Thumb::Size"
63 #define THUMB_MARKER_WIDTH  "tEXt::Thumb::Image::Width"
64 #define THUMB_MARKER_HEIGHT "tEXt::Thumb::Image::Height"
65 #define THUMB_MARKER_APP    "tEXt::Software"
66
67 #define THUMB_PERMS_FOLDER 0700
68 #define THUMB_PERMS_THUMB  0600
69
70
71
72 /*
73  *-----------------------------------------------------------------------------
74  * thumbnail loader
75  *-----------------------------------------------------------------------------
76  */
77
78
79 static void thumb_loader_std_error_cb(ImageLoader *il, gpointer data);
80 static gint thumb_loader_std_setup(ThumbLoaderStd *tl, FileData *fd);
81
82
83 ThumbLoaderStd *thumb_loader_std_new(gint width, gint height)
84 {
85         ThumbLoaderStd *tl;
86
87         tl = g_new0(ThumbLoaderStd, 1);
88
89         tl->standard_loader = TRUE;
90         tl->requested_width = width;
91         tl->requested_height = height;
92         tl->cache_enable = options->thumbnails.enable_caching;
93
94         return tl;
95 }
96
97 void thumb_loader_std_set_callbacks(ThumbLoaderStd *tl,
98                                     ThumbLoaderStdFunc func_done,
99                                     ThumbLoaderStdFunc func_error,
100                                     ThumbLoaderStdFunc func_progress,
101                                     gpointer data)
102 {
103         if (!tl) return;
104
105         tl->func_done = func_done;
106         tl->func_error = func_error;
107         tl->func_progress = func_progress;
108         tl->data = data;
109 }
110
111 static void thumb_loader_std_reset(ThumbLoaderStd *tl)
112 {
113         image_loader_free(tl->il);
114         tl->il = NULL;
115
116         file_data_unref(tl->fd);
117         tl->fd = NULL;
118
119         g_free(tl->thumb_path);
120         tl->thumb_path = NULL;
121
122         g_free(tl->thumb_uri);
123         tl->thumb_uri = NULL;
124         tl->local_uri = NULL;
125
126         tl->thumb_path_local = FALSE;
127
128         tl->cache_hit = FALSE;
129
130         tl->source_mtime = 0;
131         tl->source_size = 0;
132         tl->source_mode = 0;
133
134         tl->progress = 0.0;
135 }
136
137 static gchar *thumb_std_cache_path(const gchar *path, const gchar *uri, gboolean local,
138                                    const gchar *cache_subfolder)
139 {
140         gchar *result = NULL;
141         gchar *md5_text;
142         guchar digest[16];
143         gchar *name;
144
145         if (!path || !uri || !cache_subfolder) return NULL;
146
147         md5_get_digest((guchar *)uri, strlen(uri), digest);
148         md5_text = md5_digest_to_text(digest);
149
150         if (!md5_text) return NULL;
151
152         name = g_strconcat(md5_text, THUMB_NAME_EXTENSION, NULL);
153
154         if (local)
155                 {
156                 gchar *base = remove_level_from_path(path);
157
158                 result = g_build_filename(base, THUMB_FOLDER_LOCAL, cache_subfolder, name, NULL);
159                 g_free(base);
160                 }
161         else
162                 {
163                 result = g_build_filename(get_thumbnails_standard_cache_dir(),
164                                                                                                         cache_subfolder, name, NULL);
165                 }
166
167         g_free(name);
168         g_free(md5_text);
169
170         return result;
171 }
172
173 static gchar *thumb_loader_std_cache_path(ThumbLoaderStd *tl, gboolean local, GdkPixbuf *pixbuf, gboolean fail)
174 {
175         const gchar *folder;
176         gint w, h;
177
178         if (!tl->fd || !tl->thumb_uri) return NULL;
179
180         if (pixbuf)
181                 {
182                 w = gdk_pixbuf_get_width(pixbuf);
183                 h = gdk_pixbuf_get_height(pixbuf);
184                 }
185         else
186                 {
187                 w = tl->requested_width;
188                 h = tl->requested_height;
189                 }
190
191         if (fail)
192                 {
193                 folder = THUMB_FOLDER_FAIL;
194                 }
195         else if (w > THUMB_SIZE_NORMAL || h > THUMB_SIZE_NORMAL)
196                 {
197                 folder = THUMB_FOLDER_LARGE;
198                 }
199         else
200                 {
201                 folder = THUMB_FOLDER_NORMAL;
202                 }
203
204         return thumb_std_cache_path(tl->fd->path,
205                                     (local) ?  tl->local_uri : tl->thumb_uri,
206                                     local, folder);
207 }
208
209 static gboolean thumb_loader_std_fail_check(ThumbLoaderStd *tl)
210 {
211         gchar *fail_path;
212         gboolean result = FALSE;
213
214         fail_path = thumb_loader_std_cache_path(tl, FALSE, NULL, TRUE);
215         if (isfile(fail_path))
216                 {
217                 GdkPixbuf *pixbuf;
218
219                 if (tl->cache_retry)
220                         {
221                         pixbuf = NULL;
222                         }
223                 else
224                         {
225                         gchar *pathl;
226
227                         pathl = path_from_utf8(fail_path);
228                         pixbuf = gdk_pixbuf_new_from_file(pathl, NULL);
229                         g_free(pathl);
230                         }
231
232                 if (pixbuf)
233                         {
234                         const gchar *mtime_str;
235
236                         mtime_str = gdk_pixbuf_get_option(pixbuf, THUMB_MARKER_MTIME);
237                         if (mtime_str && strtol(mtime_str, NULL, 10) == tl->source_mtime)
238                                 {
239                                 result = TRUE;
240                                 DEBUG_1("thumb fail valid: %s", tl->fd->path);
241                                 DEBUG_1("           thumb: %s", fail_path);
242                                 }
243
244                         g_object_unref(G_OBJECT(pixbuf));
245                         }
246
247                 if (!result) unlink_file(fail_path);
248                 }
249         g_free(fail_path);
250
251         return result;
252 }
253
254 static gboolean thumb_loader_std_validate(ThumbLoaderStd *tl, GdkPixbuf *pixbuf)
255 {
256         const gchar *valid_uri;
257         const gchar *uri;
258         const gchar *mtime_str;
259         time_t mtime;
260         gint w, h;
261
262         if (!pixbuf) return FALSE;
263
264         w = gdk_pixbuf_get_width(pixbuf);
265         h = gdk_pixbuf_get_height(pixbuf);
266
267         if (w != THUMB_SIZE_NORMAL && w != THUMB_SIZE_LARGE &&
268             h != THUMB_SIZE_NORMAL && h != THUMB_SIZE_LARGE) return FALSE;
269
270         valid_uri = (tl->thumb_path_local) ? tl->local_uri : tl->thumb_uri;
271
272         uri = gdk_pixbuf_get_option(pixbuf, THUMB_MARKER_URI);
273         mtime_str = gdk_pixbuf_get_option(pixbuf, THUMB_MARKER_MTIME);
274
275         if (!mtime_str || !uri || !valid_uri) return FALSE;
276         if (strcmp(uri, valid_uri) != 0) return FALSE;
277
278         mtime = strtol(mtime_str, NULL, 10);
279         if (tl->source_mtime != mtime) return FALSE;
280
281         return TRUE;
282 }
283
284 static void thumb_loader_std_save(ThumbLoaderStd *tl, GdkPixbuf *pixbuf)
285 {
286         gchar *base_path;
287         gchar *tmp_path;
288         gboolean fail;
289
290         if (!tl->cache_enable || tl->cache_hit) return;
291         if (tl->thumb_path) return;
292
293         if (!pixbuf)
294                 {
295                 /* local failures are not stored */
296                 if (tl->cache_local) return;
297
298                 pixbuf = gdk_pixbuf_new(GDK_COLORSPACE_RGB, TRUE, 8, 1, 1);
299                 fail = TRUE;
300                 }
301         else
302                 {
303                 g_object_ref(G_OBJECT(pixbuf));
304                 fail = FALSE;
305                 }
306
307         tl->thumb_path = thumb_loader_std_cache_path(tl, tl->cache_local, pixbuf, fail);
308         if (!tl->thumb_path)
309                 {
310                 g_object_unref(G_OBJECT(pixbuf));
311                 return;
312                 }
313         tl->thumb_path_local = tl->cache_local;
314
315         /* create thumbnail dir if needed */
316         base_path = remove_level_from_path(tl->thumb_path);
317         if (tl->cache_local)
318                 {
319                 if (!isdir(base_path))
320                         {
321                         struct stat st;
322                         gchar *source_base;
323
324                         source_base = remove_level_from_path(tl->fd->path);
325                         if (stat_utf8(source_base, &st))
326                                 {
327                                 recursive_mkdir_if_not_exists(base_path, st.st_mode);
328                                 }
329                         g_free(source_base);
330                         }
331                 }
332         else
333                 {
334                 recursive_mkdir_if_not_exists(base_path, THUMB_PERMS_FOLDER);
335                 }
336         g_free(base_path);
337
338         DEBUG_1("thumb saving: %s", tl->fd->path);
339         DEBUG_1("       saved: %s", tl->thumb_path);
340
341         /* save thumb, using a temp file then renaming into place */
342         tmp_path = unique_filename(tl->thumb_path, ".tmp", "_", 2);
343         if (tmp_path)
344                 {
345                 const gchar *mark_uri;
346                 gchar *mark_app;
347                 gchar *mark_mtime;
348                 gchar *pathl;
349                 gboolean success;
350
351                 mark_uri = (tl->cache_local) ? tl->local_uri :tl->thumb_uri;
352
353                 mark_app = g_strdup_printf("%s %s", GQ_APPNAME, VERSION);
354                 mark_mtime = g_strdup_printf("%llu", (unsigned long long)tl->source_mtime);
355                 pathl = path_from_utf8(tmp_path);
356                 success = gdk_pixbuf_save(pixbuf, pathl, "png", NULL,
357                                           THUMB_MARKER_URI, mark_uri,
358                                           THUMB_MARKER_MTIME, mark_mtime,
359                                           THUMB_MARKER_APP, mark_app,
360                                           NULL);
361                 if (success)
362                         {
363                         chmod(pathl, (tl->cache_local) ? tl->source_mode : THUMB_PERMS_THUMB);
364                         success = rename_file(tmp_path, tl->thumb_path);
365                         }
366
367                 g_free(pathl);
368
369                 g_free(mark_mtime);
370                 g_free(mark_app);
371
372                 g_free(tmp_path);
373                 if (!success)
374                         {
375                         DEBUG_1("thumb save failed: %s", tl->fd->path);
376                         DEBUG_1("            thumb: %s", tl->thumb_path);
377                         }
378
379                 }
380
381         g_object_unref(G_OBJECT(pixbuf));
382 }
383
384 static void thumb_loader_std_set_fallback(ThumbLoaderStd *tl)
385 {
386         if (tl->fd->thumb_pixbuf) g_object_unref(tl->fd->thumb_pixbuf);
387         tl->fd->thumb_pixbuf = pixbuf_fallback(tl->fd, tl->requested_width, tl->requested_height);
388 }
389
390
391 void thumb_loader_std_calibrate_pixbuf(FileData *fd, GdkPixbuf *pixbuf) {
392         ColorMan *cm = NULL;
393         ExifData *exif = NULL;
394         gint color_profile_from_image = COLOR_PROFILE_NONE;
395         ColorManProfileType input_type = COLOR_PROFILE_MEM;
396         ColorManProfileType screen_type;
397         const gchar *input_file = NULL;
398         guchar *profile = NULL;
399         guint profile_len;
400         gint sw, sh;
401
402         sw = gdk_pixbuf_get_width(pixbuf);
403         sh = gdk_pixbuf_get_height(pixbuf);
404
405         exif = exif_read_fd(fd);
406
407         if (exif)
408                 {
409                 profile = exif_get_color_profile(exif, &profile_len);
410                 if (profile)
411                         {
412                         DEBUG_1("Found embedded color profile");
413                         color_profile_from_image = COLOR_PROFILE_MEM;
414                         }
415                 else
416                         {
417                         gchar *interop_index = exif_get_data_as_text(exif, "Exif.Iop.InteroperabilityIndex");
418
419                         if (interop_index)
420                                 {
421                                 /* Exif 2.21 specification */
422                                 if (!strcmp(interop_index, "R98"))
423                                         {
424                                         color_profile_from_image = COLOR_PROFILE_SRGB;
425                                         DEBUG_1("Found EXIF 2.21 ColorSpace of sRGB");
426                                         }
427                                 else if (!strcmp(interop_index, "R03"))
428                                         {
429                                         color_profile_from_image = COLOR_PROFILE_ADOBERGB;
430                                         DEBUG_1("Found EXIF 2.21 ColorSpace of AdobeRGB");
431                                         }
432                                 g_free(interop_index);
433                                 }
434                         else
435                                 {
436                                 gint cs;
437
438                                 /* ColorSpace == 1 specifies sRGB per EXIF 2.2 */
439                                 if (!exif_get_integer(exif, "Exif.Photo.ColorSpace", &cs)) cs = 0;
440                                 if (cs == 1)
441                                         {
442                                         color_profile_from_image = COLOR_PROFILE_SRGB;
443                                         DEBUG_1("Found EXIF 2.2 ColorSpace of sRGB");
444                                         }
445                                 else if (cs == 2)
446                                         {
447                                         /* non-standard way of specifying AdobeRGB (used by some software) */
448                                         color_profile_from_image = COLOR_PROFILE_ADOBERGB;
449                                         DEBUG_1("Found EXIF 2.2 ColorSpace of AdobeRGB");
450                                         }
451                                 }
452                         }
453
454                 if(color_profile_from_image != COLOR_PROFILE_NONE)
455                         {
456                                 // transform image, we always use sRGB as target for thumbnails
457                                 screen_type = COLOR_PROFILE_SRGB;
458
459                                 if (profile)
460                                         {
461                                         cm = color_man_new_embedded(NULL, pixbuf,
462                                                                         profile, profile_len,
463                                                                         screen_type, NULL, NULL, 0);
464                                         g_free(profile);
465                                         }
466                                 else
467                                         {
468                                         cm = color_man_new(NULL, pixbuf,
469                                                         input_type, input_file,
470                                                         screen_type, NULL, NULL, 0);
471                                         }
472
473                                 if(cm) {
474                                         color_man_correct_region(cm, cm->pixbuf, 0, 0, sw, sh);
475                                         g_free(cm);
476                                 }
477
478                         }
479                 exif_free_fd(fd, exif);
480                 }
481 }
482
483 static GdkPixbuf *thumb_loader_std_finish(ThumbLoaderStd *tl, GdkPixbuf *pixbuf, gboolean shrunk)
484 {
485         GdkPixbuf *pixbuf_thumb = NULL;
486         GdkPixbuf *result;
487         GdkPixbuf *rotated = NULL;
488         gint sw, sh;
489
490
491         if (!tl->cache_hit && options->image.exif_rotate_enable)
492                 {
493                 if (!tl->fd->exif_orientation)
494                         {
495                         tl->fd->exif_orientation = metadata_read_int(tl->fd, ORIENTATION_KEY, EXIF_ORIENTATION_TOP_LEFT);
496                         }
497
498                 if (tl->fd->exif_orientation != EXIF_ORIENTATION_TOP_LEFT)
499                         {
500                         rotated = pixbuf_apply_orientation(pixbuf, tl->fd->exif_orientation);
501                         pixbuf = rotated;
502                         }
503                 }
504
505         sw = gdk_pixbuf_get_width(pixbuf);
506         sh = gdk_pixbuf_get_height(pixbuf);
507
508         if (tl->cache_enable)
509                 {
510                 if (!tl->cache_hit)
511                         {
512                         gint cache_w, cache_h;
513
514                         if (tl->requested_width > THUMB_SIZE_NORMAL || tl->requested_height > THUMB_SIZE_NORMAL)
515                                 {
516                                 cache_w = cache_h = THUMB_SIZE_LARGE;
517                                 }
518                         else
519                                 {
520                                 cache_w = cache_h = THUMB_SIZE_NORMAL;
521                                 }
522
523                         if (sw > cache_w || sh > cache_h || shrunk)
524                                 {
525                                 gint thumb_w, thumb_h;
526                                 struct stat st;
527
528                                 if (pixbuf_scale_aspect(cache_w, cache_h, sw, sh,
529                                                                   &thumb_w, &thumb_h))
530                                         {
531                                         pixbuf_thumb = gdk_pixbuf_scale_simple(pixbuf, thumb_w, thumb_h,
532                                                                                (GdkInterpType)options->thumbnails.quality);
533                                         }
534                                 else
535                                         {
536                                         pixbuf_thumb = pixbuf;
537                                         g_object_ref(G_OBJECT(pixbuf_thumb));
538                                         }
539
540                                 /* do not save the thumbnail if the source file has changed meanwhile -
541                                    the thumbnail is most probably broken */
542                                 if (stat_utf8(tl->fd->path, &st) &&
543                                     tl->source_mtime == st.st_mtime &&
544                                     tl->source_size == st.st_size)
545                                         {
546                                         thumb_loader_std_save(tl, pixbuf_thumb);
547                                         }
548                                 }
549                         }
550                 else if (tl->cache_hit &&
551                          tl->cache_local && !tl->thumb_path_local)
552                         {
553                         /* A local cache save was requested, but a valid thumb is in $HOME,
554                          * so specifically save as a local thumbnail.
555                          */
556                         g_free(tl->thumb_path);
557                         tl->thumb_path = NULL;
558
559                         tl->cache_hit = FALSE;
560
561                         DEBUG_1("thumb copied: %s", tl->fd->path);
562
563                         thumb_loader_std_save(tl, pixbuf);
564                         }
565                 }
566
567         if (sw <= tl->requested_width && sh <= tl->requested_height)
568                 {
569                 result = pixbuf;
570                 g_object_ref(result);
571                 }
572         else
573                 {
574                 gint thumb_w, thumb_h;
575
576                 if (pixbuf_thumb)
577                         {
578                         pixbuf = pixbuf_thumb;
579                         sw = gdk_pixbuf_get_width(pixbuf);
580                         sh = gdk_pixbuf_get_height(pixbuf);
581                         }
582
583                 if (pixbuf_scale_aspect(tl->requested_width, tl->requested_height, sw, sh,
584                                                   &thumb_w, &thumb_h))
585                         {
586                         result = gdk_pixbuf_scale_simple(pixbuf, thumb_w, thumb_h,
587                                                          (GdkInterpType)options->thumbnails.quality);
588                         }
589                 else
590                         {
591                         result = pixbuf;
592                         g_object_ref(result);
593                         }
594                 }
595
596         // apply color correction, if required
597         thumb_loader_std_calibrate_pixbuf(tl->fd, result);
598
599         if (pixbuf_thumb) g_object_unref(pixbuf_thumb);
600         if (rotated) g_object_unref(rotated);
601
602         return result;
603 }
604
605 static gboolean thumb_loader_std_next_source(ThumbLoaderStd *tl, gboolean remove_broken)
606 {
607         image_loader_free(tl->il);
608         tl->il = NULL;
609
610         if (tl->thumb_path)
611                 {
612                 if (!tl->thumb_path_local && remove_broken)
613                         {
614                         DEBUG_1("thumb broken, unlinking: %s", tl->thumb_path);
615                         unlink_file(tl->thumb_path);
616                         }
617
618                 g_free(tl->thumb_path);
619                 tl->thumb_path = NULL;
620
621                 if (!tl->thumb_path_local)
622                         {
623                         tl->thumb_path = thumb_loader_std_cache_path(tl, TRUE, NULL, FALSE);
624                         if (isfile(tl->thumb_path))
625                                 {
626                                 FileData *fd = file_data_new_no_grouping(tl->thumb_path);
627                                 if (thumb_loader_std_setup(tl, fd))
628                                         {
629                                         file_data_unref(fd);
630                                         tl->thumb_path_local = TRUE;
631                                         return TRUE;
632                                         }
633                                 file_data_unref(fd);
634                                 }
635
636                         g_free(tl->thumb_path);
637                         tl->thumb_path = NULL;
638                         }
639
640                 if (thumb_loader_std_setup(tl, tl->fd)) return TRUE;
641                 }
642
643         thumb_loader_std_save(tl, NULL);
644         return FALSE;
645 }
646
647 static void thumb_loader_std_done_cb(ImageLoader *il, gpointer data)
648 {
649         ThumbLoaderStd *tl = data;
650         GdkPixbuf *pixbuf;
651
652         DEBUG_1("thumb image done: %s", tl->fd ? tl->fd->path : "???");
653         DEBUG_1("            from: %s", image_loader_get_fd(tl->il)->path);
654
655         pixbuf = image_loader_get_pixbuf(tl->il);
656         if (!pixbuf)
657                 {
658                 DEBUG_1("...but no pixbuf");
659                 thumb_loader_std_error_cb(il, data);
660                 return;
661                 }
662
663         if (tl->thumb_path && !thumb_loader_std_validate(tl, pixbuf))
664                 {
665                 if (thumb_loader_std_next_source(tl, TRUE)) return;
666
667                 if (tl->func_error) tl->func_error(tl, tl->data);
668                 return;
669                 }
670
671         tl->cache_hit = (tl->thumb_path != NULL);
672
673         if (tl->fd)
674                 {
675                 if (tl->fd->thumb_pixbuf) g_object_unref(tl->fd->thumb_pixbuf);
676                 tl->fd->thumb_pixbuf = thumb_loader_std_finish(tl, pixbuf, image_loader_get_shrunk(il));
677                 }
678
679         if (tl->func_done) tl->func_done(tl, tl->data);
680 }
681
682 static void thumb_loader_std_error_cb(ImageLoader *il, gpointer data)
683 {
684         ThumbLoaderStd *tl = data;
685
686         /* if at least some of the image is available, go to done */
687         if (image_loader_get_pixbuf(tl->il) != NULL)
688                 {
689                 thumb_loader_std_done_cb(il, data);
690                 return;
691                 }
692
693         DEBUG_1("thumb image error: %s", tl->fd->path);
694         DEBUG_1("             from: %s", image_loader_get_fd(tl->il)->path);
695
696         if (thumb_loader_std_next_source(tl, TRUE)) return;
697
698         thumb_loader_std_set_fallback(tl);
699
700         if (tl->func_error) tl->func_error(tl, tl->data);
701 }
702
703 static void thumb_loader_std_progress_cb(ImageLoader *il, gdouble percent, gpointer data)
704 {
705         ThumbLoaderStd *tl = data;
706
707         tl->progress = (gdouble)percent;
708
709         if (tl->func_progress) tl->func_progress(tl, tl->data);
710 }
711
712 static gboolean thumb_loader_std_setup(ThumbLoaderStd *tl, FileData *fd)
713 {
714         tl->il = image_loader_new(fd);
715         image_loader_set_priority(tl->il, G_PRIORITY_LOW);
716
717         /* this will speed up jpegs by up to 3x in some cases */
718         if (tl->requested_width <= THUMB_SIZE_NORMAL &&
719             tl->requested_height <= THUMB_SIZE_NORMAL)
720                 {
721                 image_loader_set_requested_size(tl->il, THUMB_SIZE_NORMAL, THUMB_SIZE_NORMAL);
722                 }
723         else
724                 {
725                 image_loader_set_requested_size(tl->il, THUMB_SIZE_LARGE, THUMB_SIZE_LARGE);
726                 }
727
728         g_signal_connect(G_OBJECT(tl->il), "error", (GCallback)thumb_loader_std_error_cb, tl);
729         if (tl->func_progress)
730                 {
731                 g_signal_connect(G_OBJECT(tl->il), "percent", (GCallback)thumb_loader_std_progress_cb, tl);
732                 }
733         g_signal_connect(G_OBJECT(tl->il), "done", (GCallback)thumb_loader_std_done_cb, tl);
734
735         if (image_loader_start(tl->il))
736                 {
737                 return TRUE;
738                 }
739
740         image_loader_free(tl->il);
741         tl->il = NULL;
742         return FALSE;
743 }
744
745 /*
746  * Note: Currently local_cache only specifies where to save a _new_ thumb, if
747  *       a valid existing thumb is found anywhere the local thumb will not be created.
748  */
749 void thumb_loader_std_set_cache(ThumbLoaderStd *tl, gboolean enable_cache, gboolean local, gboolean retry_failed)
750 {
751         if (!tl) return;
752
753         tl->cache_enable = enable_cache;
754         tl->cache_local = local;
755         tl->cache_retry = retry_failed;
756 }
757
758 gboolean thumb_loader_std_start(ThumbLoaderStd *tl, FileData *fd)
759 {
760         static gchar *thumb_cache = NULL;
761         struct stat st;
762
763         if (!tl || !fd) return FALSE;
764
765         thumb_loader_std_reset(tl);
766
767
768         tl->fd = file_data_ref(fd);
769         if (!stat_utf8(fd->path, &st) || (tl->fd->format_class != FORMAT_CLASS_IMAGE && tl->fd->format_class != FORMAT_CLASS_RAWIMAGE && tl->fd->format_class != FORMAT_CLASS_VIDEO && tl->fd->format_class != FORMAT_CLASS_COLLECTION && tl->fd->format_class != FORMAT_CLASS_DOCUMENT && !options->file_filter.disable))
770                 {
771                 thumb_loader_std_set_fallback(tl);
772                 return FALSE;
773                 }
774         tl->source_mtime = st.st_mtime;
775         tl->source_size = st.st_size;
776         tl->source_mode = st.st_mode;
777
778         if (!thumb_cache)
779                 {
780                 thumb_cache = g_strdup(get_thumbnails_standard_cache_dir());
781                 }
782
783         if (strncmp(tl->fd->path, thumb_cache, strlen(thumb_cache)) != 0)
784                 {
785                 gchar *pathl;
786
787                 pathl = path_from_utf8(fd->path);
788                 tl->thumb_uri = g_filename_to_uri(pathl, NULL, NULL);
789                 tl->local_uri = filename_from_path(tl->thumb_uri);
790                 g_free(pathl);
791                 }
792
793         if (tl->cache_enable)
794                 {
795                 gint found;
796
797                 tl->thumb_path = thumb_loader_std_cache_path(tl, FALSE, NULL, FALSE);
798                 tl->thumb_path_local = FALSE;
799
800                 found = isfile(tl->thumb_path);
801                 if (found)
802                         {
803                         FileData *fd = file_data_new_no_grouping(tl->thumb_path);
804                         if (thumb_loader_std_setup(tl, fd))
805                                 {
806                                 file_data_unref(fd);
807                                 return TRUE;
808                                 }
809                         file_data_unref(fd);
810                         }
811
812                 if (thumb_loader_std_fail_check(tl) ||
813                     !thumb_loader_std_next_source(tl, found))
814                         {
815                         thumb_loader_std_set_fallback(tl);
816                         return FALSE;
817                         }
818                 return TRUE;
819                 }
820
821         if (!thumb_loader_std_setup(tl, tl->fd))
822                 {
823                 thumb_loader_std_save(tl, NULL);
824                 thumb_loader_std_set_fallback(tl);
825                 return FALSE;
826                 }
827
828         return TRUE;
829 }
830
831 void thumb_loader_std_free(ThumbLoaderStd *tl)
832 {
833         if (!tl) return;
834
835         thumb_loader_std_reset(tl);
836         g_free(tl);
837 }
838
839 GdkPixbuf *thumb_loader_std_get_pixbuf(ThumbLoaderStd *tl)
840 {
841         GdkPixbuf *pixbuf;
842
843         if (tl && tl->fd && tl->fd->thumb_pixbuf)
844                 {
845                 pixbuf = tl->fd->thumb_pixbuf;
846                 g_object_ref(pixbuf);
847                 }
848         else
849                 {
850                 pixbuf = pixbuf_fallback(NULL, tl->requested_width, tl->requested_height);
851                 }
852
853         return pixbuf;
854 }
855
856
857 typedef struct _ThumbValidate ThumbValidate;
858 struct _ThumbValidate
859 {
860         ThumbLoaderStd *tl;
861         gchar *path;
862         gint days;
863
864         void (*func_valid)(const gchar *path, gboolean valid, gpointer data);
865         gpointer data;
866
867         guint idle_id; /* event source id */
868 };
869
870 static void thumb_loader_std_thumb_file_validate_free(ThumbValidate *tv)
871 {
872         thumb_loader_std_free(tv->tl);
873         g_free(tv->path);
874         g_free(tv);
875 }
876
877 void thumb_loader_std_thumb_file_validate_cancel(ThumbLoaderStd *tl)
878 {
879         ThumbValidate *tv;
880
881         if (!tl) return;
882
883         tv = tl->data;
884
885         if (tv->idle_id)
886                 {
887                 g_source_remove(tv->idle_id);
888                 tv->idle_id = 0;
889                 }
890
891         thumb_loader_std_thumb_file_validate_free(tv);
892 }
893
894 static void thumb_loader_std_thumb_file_validate_finish(ThumbValidate *tv, gboolean valid)
895 {
896         if (tv->func_valid) tv->func_valid(tv->path, valid, tv->data);
897
898         thumb_loader_std_thumb_file_validate_free(tv);
899 }
900
901 static void thumb_loader_std_thumb_file_validate_done_cb(ThumbLoaderStd *tl, gpointer data)
902 {
903         ThumbValidate *tv = data;
904         GdkPixbuf *pixbuf;
905         gboolean valid = FALSE;
906
907         /* get the original thumbnail pixbuf (unrotated, with original options)
908            this is called from image_loader done callback, so tv->tl->il must exist*/
909         pixbuf = image_loader_get_pixbuf(tv->tl->il);
910         if (pixbuf)
911                 {
912                 const gchar *uri;
913                 const gchar *mtime_str;
914
915                 uri = gdk_pixbuf_get_option(pixbuf, THUMB_MARKER_URI);
916                 mtime_str = gdk_pixbuf_get_option(pixbuf, THUMB_MARKER_MTIME);
917                 if (uri && mtime_str)
918                         {
919                         if (strncmp(uri, "file:", strlen("file:")) == 0)
920                                 {
921                                 struct stat st;
922                                 gchar *target;
923
924                                 target = g_filename_from_uri(uri, NULL, NULL);
925                                 if (stat(target, &st) == 0 &&
926                                     st.st_mtime == strtol(mtime_str, NULL, 10))
927                                         {
928                                         valid = TRUE;
929                                         }
930                                 g_free(target);
931                                 }
932                         else
933                                 {
934                                 struct stat st;
935
936                                 DEBUG_1("thumb uri foreign, doing day check: %s", uri);
937
938                                 if (stat_utf8(tv->path, &st))
939                                         {
940                                         time_t now;
941
942                                         now = time(NULL);
943                                         if (st.st_atime >= now - (time_t)tv->days * 24 * 60 * 60)
944                                                 {
945                                                 valid = TRUE;
946                                                 }
947                                         }
948                                 }
949                         }
950                 else
951                         {
952                         DEBUG_1("invalid image found in std cache: %s", tv->path);
953                         }
954                 }
955
956         thumb_loader_std_thumb_file_validate_finish(tv, valid);
957 }
958
959 static void thumb_loader_std_thumb_file_validate_error_cb(ThumbLoaderStd *tl, gpointer data)
960 {
961         ThumbValidate *tv = data;
962
963         thumb_loader_std_thumb_file_validate_finish(tv, FALSE);
964 }
965
966 static gboolean thumb_loader_std_thumb_file_validate_idle_cb(gpointer data)
967 {
968         ThumbValidate *tv = data;
969
970         tv->idle_id = 0;
971         thumb_loader_std_thumb_file_validate_finish(tv, FALSE);
972
973         return FALSE;
974 }
975
976 ThumbLoaderStd *thumb_loader_std_thumb_file_validate(const gchar *thumb_path, gint allowed_days,
977                                                      void (*func_valid)(const gchar *path, gboolean valid, gpointer data),
978                                                      gpointer data)
979 {
980         ThumbValidate *tv;
981
982         tv = g_new0(ThumbValidate, 1);
983
984         tv->tl = thumb_loader_std_new(THUMB_SIZE_LARGE, THUMB_SIZE_LARGE);
985         thumb_loader_std_set_callbacks(tv->tl,
986                                        thumb_loader_std_thumb_file_validate_done_cb,
987                                        thumb_loader_std_thumb_file_validate_error_cb,
988                                        NULL,
989                                        tv);
990         thumb_loader_std_reset(tv->tl);
991
992         tv->path = g_strdup(thumb_path);
993         tv->days = allowed_days;
994         tv->func_valid = func_valid;
995         tv->data = data;
996
997         FileData *fd = file_data_new_no_grouping(thumb_path);
998         if (!thumb_loader_std_setup(tv->tl, fd))
999                 {
1000                 tv->idle_id = g_idle_add(thumb_loader_std_thumb_file_validate_idle_cb, tv);
1001                 }
1002         else
1003                 {
1004                 tv->idle_id = 0;
1005                 }
1006
1007         file_data_unref(fd);
1008         return tv->tl;
1009 }
1010
1011 static void thumb_std_maint_remove_one(const gchar *source, const gchar *uri, gboolean local,
1012                                        const gchar *subfolder)
1013 {
1014         gchar *thumb_path;
1015
1016         thumb_path = thumb_std_cache_path(source,
1017                                           (local) ? filename_from_path(uri) : uri,
1018                                           local, subfolder);
1019         if (isfile(thumb_path))
1020                 {
1021                 DEBUG_1("thumb removing: %s", thumb_path);
1022                 unlink_file(thumb_path);
1023                 }
1024         g_free(thumb_path);
1025 }
1026
1027 /* this also removes local thumbnails (the source is gone so it makes sense) */
1028 void thumb_std_maint_removed(const gchar *source)
1029 {
1030         gchar *uri;
1031         gchar *sourcel;
1032
1033         sourcel = path_from_utf8(source);
1034         uri = g_filename_to_uri(sourcel, NULL, NULL);
1035         g_free(sourcel);
1036
1037         /* all this to remove a thumbnail? */
1038
1039         thumb_std_maint_remove_one(source, uri, FALSE, THUMB_FOLDER_NORMAL);
1040         thumb_std_maint_remove_one(source, uri, FALSE, THUMB_FOLDER_LARGE);
1041         thumb_std_maint_remove_one(source, uri, FALSE, THUMB_FOLDER_FAIL);
1042         thumb_std_maint_remove_one(source, uri, TRUE, THUMB_FOLDER_NORMAL);
1043         thumb_std_maint_remove_one(source, uri, TRUE, THUMB_FOLDER_LARGE);
1044
1045         g_free(uri);
1046 }
1047
1048 typedef struct _TMaintMove TMaintMove;
1049 struct _TMaintMove
1050 {
1051         gchar *source;
1052         gchar *dest;
1053
1054         ThumbLoaderStd *tl;
1055         gchar *source_uri;
1056         gchar *thumb_path;
1057
1058         gint pass;
1059 };
1060
1061 static GList *thumb_std_maint_move_list = NULL;
1062 static GList *thumb_std_maint_move_tail = NULL;
1063
1064
1065 static void thumb_std_maint_move_step(TMaintMove *tm);
1066 static gboolean thumb_std_maint_move_idle(gpointer data);
1067
1068
1069 static void thumb_std_maint_move_validate_cb(const gchar *path, gboolean valid, gpointer data)
1070 {
1071         TMaintMove *tm = data;
1072         GdkPixbuf *pixbuf;
1073
1074         /* get the original thumbnail pixbuf (unrotated, with original options)
1075            this is called from image_loader done callback, so tm->tl->il must exist*/
1076         pixbuf = image_loader_get_pixbuf(tm->tl->il);
1077         if (pixbuf)
1078                 {
1079                 const gchar *uri;
1080                 const gchar *mtime_str;
1081
1082                 uri = gdk_pixbuf_get_option(pixbuf, THUMB_MARKER_URI);
1083                 mtime_str = gdk_pixbuf_get_option(pixbuf, THUMB_MARKER_MTIME);
1084
1085                 if (uri && mtime_str && strcmp(uri, tm->source_uri) == 0)
1086                         {
1087                         gchar *pathl;
1088
1089                         /* The validation utility abuses ThumbLoader, and we
1090                          * abuse the utility just to load the thumbnail,
1091                          * but the loader needs to look sane for the save to complete.
1092                          */
1093
1094                         tm->tl->cache_enable = TRUE;
1095                         tm->tl->cache_hit = FALSE;
1096                         tm->tl->cache_local = FALSE;
1097                         file_data_unref(tm->tl->fd);
1098                         tm->tl->fd = file_data_new_group(tm->dest);
1099                         tm->tl->source_mtime = strtol(mtime_str, NULL, 10);
1100
1101                         pathl = path_from_utf8(tm->tl->fd->path);
1102                         g_free(tm->tl->thumb_uri);
1103                         tm->tl->thumb_uri = g_filename_to_uri(pathl, NULL, NULL);
1104                         tm->tl->local_uri = filename_from_path(tm->tl->thumb_uri);
1105                         g_free(pathl);
1106
1107                         g_free(tm->tl->thumb_path);
1108                         tm->tl->thumb_path = NULL;
1109                         tm->tl->thumb_path_local = FALSE;
1110
1111                         DEBUG_1("thumb move attempting save:");
1112
1113                         thumb_loader_std_save(tm->tl, pixbuf);
1114                         }
1115
1116                 DEBUG_1("thumb move unlink: %s", tm->thumb_path);
1117                 unlink_file(tm->thumb_path);
1118                 }
1119
1120         thumb_std_maint_move_step(tm);
1121 }
1122
1123 static void thumb_std_maint_move_step(TMaintMove *tm)
1124 {
1125         const gchar *folder;
1126
1127         tm->pass++;
1128         if (tm->pass > 2)
1129                 {
1130                 g_free(tm->source);
1131                 g_free(tm->dest);
1132                 g_free(tm->source_uri);
1133                 g_free(tm->thumb_path);
1134                 g_free(tm);
1135
1136                 if (thumb_std_maint_move_list)
1137                         {
1138                         g_idle_add_full(G_PRIORITY_LOW, thumb_std_maint_move_idle, NULL, NULL);
1139                         }
1140
1141                 return;
1142                 }
1143
1144         folder = (tm->pass == 1) ? THUMB_FOLDER_NORMAL : THUMB_FOLDER_LARGE;
1145
1146         g_free(tm->thumb_path);
1147         tm->thumb_path = thumb_std_cache_path(tm->source, tm->source_uri, FALSE, folder);
1148         tm->tl = thumb_loader_std_thumb_file_validate(tm->thumb_path, 0,
1149                                                       thumb_std_maint_move_validate_cb, tm);
1150 }
1151
1152 static gboolean thumb_std_maint_move_idle(gpointer data)
1153 {
1154         TMaintMove *tm;
1155         gchar *pathl;
1156
1157         if (!thumb_std_maint_move_list) return FALSE;
1158
1159         tm = thumb_std_maint_move_list->data;
1160
1161         thumb_std_maint_move_list = g_list_remove(thumb_std_maint_move_list, tm);
1162         if (!thumb_std_maint_move_list) thumb_std_maint_move_tail = NULL;
1163
1164         pathl = path_from_utf8(tm->source);
1165         tm->source_uri = g_filename_to_uri(pathl, NULL, NULL);
1166         g_free(pathl);
1167
1168         tm->pass = 0;
1169
1170         thumb_std_maint_move_step(tm);
1171
1172         return FALSE;
1173 }
1174
1175 /* This will schedule a move of the thumbnail for source image to dest when idle.
1176  * We do this so that file renaming or moving speed is not sacrificed by
1177  * moving the thumbnails at the same time because:
1178  *
1179  * This cache design requires the tedious task of loading the png thumbnails and saving them.
1180  *
1181  * The thumbnails are processed when the app is idle. If the app
1182  * exits early well too bad - they can simply be regenerated from scratch.
1183  *
1184  * This does not manage local thumbnails (fixme ?)
1185  */
1186 void thumb_std_maint_moved(const gchar *source, const gchar *dest)
1187 {
1188         TMaintMove *tm;
1189
1190         tm = g_new0(TMaintMove, 1);
1191         tm->source = g_strdup(source);
1192         tm->dest = g_strdup(dest);
1193
1194         if (!thumb_std_maint_move_list)
1195                 {
1196                 g_idle_add_full(G_PRIORITY_LOW, thumb_std_maint_move_idle, NULL, NULL);
1197                 }
1198
1199         if (thumb_std_maint_move_tail)
1200                 {
1201                 thumb_std_maint_move_tail = g_list_append(thumb_std_maint_move_tail, tm);
1202                 thumb_std_maint_move_tail = thumb_std_maint_move_tail->next;
1203                 }
1204         else
1205                 {
1206                 thumb_std_maint_move_list = g_list_append(thumb_std_maint_move_list, tm);
1207                 thumb_std_maint_move_tail = thumb_std_maint_move_list;
1208                 }
1209 }
1210 /* vim: set shiftwidth=8 softtabstop=0 cindent cinoptions={1s: */