src/cache-loader.c

   1 /*
   2  * Copyright (C) 2004 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 "cache-loader.h"
  24 #include "cache.h"
  25
  26 #include "filedata.h"
  27 #include "exif.h"
  28 #include "metadata.h"
  29 #include "md5-util.h"
  30 #include "ui_fileops.h"
  31
  32
  33 static gboolean cache_loader_process(CacheLoader *cl);
  34
  35
  36 static void cache_loader_done_cb(ImageLoader *il, gpointer data)
  37 {
  38         CacheLoader *cl = data;
  39
  40         cache_loader_process(cl);
  41 }
  42
  43 static void cache_loader_error_cb(ImageLoader *il, gpointer data)
  44 {
  45         CacheLoader *cl = data;
  46
  47         cl->error = TRUE;
  48         cache_loader_done_cb(il, data);
  49 }
  50
  51 static gboolean cache_loader_process(CacheLoader *cl)
  52 {
  53         if (cl->todo_mask & CACHE_LOADER_SIMILARITY &&
  54             !cl->cd->similarity)
  55                 {
  56                 GdkPixbuf *pixbuf;
  57
  58                 if (!cl->il && !cl->error)
  59                         {
  60                         cl->il = image_loader_new(cl->fd);
  61                         g_signal_connect(G_OBJECT(cl->il), "error", (GCallback)cache_loader_error_cb, cl);
  62                         g_signal_connect(G_OBJECT(cl->il), "done", (GCallback)cache_loader_done_cb, cl);
  63                         if (image_loader_start(cl->il))
  64                                 {
  65                                 return FALSE;
  66                                 }
  67
  68                         cl->error = TRUE;
  69                         }
  70
  71                 pixbuf = image_loader_get_pixbuf(cl->il);
  72                 if (pixbuf)
  73                         {
  74                         if (!cl->error)
  75                                 {
  76                                 ImageSimilarityData *sim;
  77
  78                                 sim = image_sim_new_from_pixbuf(pixbuf);
  79                                 cache_sim_data_set_similarity(cl->cd, sim);
  80                                 image_sim_free(sim);
  81
  82                                 cl->done_mask |= CACHE_LOADER_SIMILARITY;
  83                                 }
  84
  85                         /* we have the dimensions via pixbuf */
  86                         if (!cl->cd->dimensions)
  87                                 {
  88                                 cache_sim_data_set_dimensions(cl->cd, gdk_pixbuf_get_width(pixbuf),
  89                                                                       gdk_pixbuf_get_height(pixbuf));
  90                                 if (cl->todo_mask & CACHE_LOADER_DIMENSIONS)
  91                                         {
  92                                         cl->todo_mask &= ~CACHE_LOADER_DIMENSIONS;
  93                                         cl->done_mask |= CACHE_LOADER_DIMENSIONS;
  94                                         }
  95                                 }
  96                         }
  97
  98                 image_loader_free(cl->il);
  99                 cl->il = NULL;
 100
 101                 cl->todo_mask &= ~CACHE_LOADER_SIMILARITY;
 102                 }
 103         else if (cl->todo_mask & CACHE_LOADER_DIMENSIONS &&
 104                  !cl->cd->dimensions)
 105                 {
 106                 if (!cl->error &&
 107                     image_load_dimensions(cl->fd, &cl->cd->width, &cl->cd->height))
 108                         {
 109                         cl->cd->dimensions = TRUE;
 110                         cl->done_mask |= CACHE_LOADER_DIMENSIONS;
 111                         }
 112                 else
 113                         {
 114                         cl->error = TRUE;
 115                         }
 116
 117                 cl->todo_mask &= ~CACHE_LOADER_DIMENSIONS;
 118                 }
 119         else if (cl->todo_mask & CACHE_LOADER_MD5SUM &&
 120                  !cl->cd->have_md5sum)
 121                 {
 122                 if (md5_get_digest_from_file_utf8(cl->fd->path, cl->cd->md5sum))
 123                         {
 124                         cl->cd->have_md5sum = TRUE;
 125                         cl->done_mask |= CACHE_LOADER_MD5SUM;
 126                         }
 127                 else
 128                         {
 129                         cl->error = TRUE;
 130                         }
 131
 132                 cl->todo_mask &= ~CACHE_LOADER_MD5SUM;
 133                 }
 134         else if (cl->todo_mask & CACHE_LOADER_DATE &&
 135                  !cl->cd->have_date)
 136                 {
 137                 time_t date = -1;
 138                 gchar *text;
 139
 140                 text =  metadata_read_string(cl->fd, "formatted.DateTime", METADATA_FORMATTED);
 141                 if (text)
 142                         {
 143                         struct tm t;
 144
 145                         memset(&t, 0, sizeof(t));
 146
 147                         if (sscanf(text, "%d:%d:%d %d:%d:%d", &t.tm_year, &t.tm_mon, &t.tm_mday,
 148                                    &t.tm_hour, &t.tm_min, &t.tm_sec) == 6)
 149                                 {
 150                                 t.tm_year -= 1900;
 151                                 t.tm_mon -= 1;
 152                                 t.tm_isdst = -1;
 153                                 date = mktime(&t);
 154                                 }
 155                         g_free(text);
 156                         }
 157
 158                 cl->cd->date = date;
 159                 cl->cd->have_date = TRUE;
 160
 161                 cl->done_mask |= CACHE_LOADER_DATE;
 162                 cl->todo_mask &= ~CACHE_LOADER_DATE;
 163                 }
 164         else
 165                 {
 166                 /* done, save then call done function */
 167                 if (options->thumbnails.enable_caching &&
 168                     cl->done_mask != CACHE_LOADER_NONE)
 169                         {
 170                         gchar *base;
 171                         mode_t mode = 0755;
 172
 173                         base = cache_get_location(CACHE_TYPE_SIM, cl->fd->path, FALSE, &mode);
 174                         if (recursive_mkdir_if_not_exists(base, mode))
 175                                 {
 176                                 g_free(cl->cd->path);
 177                                 cl->cd->path = cache_get_location(CACHE_TYPE_SIM, cl->fd->path, TRUE, NULL);
 178                                 if (cache_sim_data_save(cl->cd))
 179                                         {
 180                                         filetime_set(cl->cd->path, filetime(cl->fd->path));
 181                                         }
 182                                 }
 183                         g_free(base);
 184                         }
 185
 186                 cl->idle_id = 0;
 187
 188                 if (cl->done_func)
 189                         {
 190                         cl->done_func(cl, cl->error, cl->done_data);
 191                         }
 192
 193                 return FALSE;
 194                 }
 195
 196         return TRUE;
 197 }
 198
 199 static gboolean cache_loader_idle_cb(gpointer data)
 200 {
 201         CacheLoader *cl = data;
 202
 203         return cache_loader_process(cl);
 204 }
 205
 206 CacheLoader *cache_loader_new(FileData *fd, CacheDataType load_mask,
 207                               CacheLoaderDoneFunc done_func, gpointer done_data)
 208 {
 209         CacheLoader *cl;
 210         gchar *found;
 211
 212         if (!fd || !isfile(fd->path)) return NULL;
 213
 214         cl = g_new0(CacheLoader, 1);
 215         cl->fd = file_data_ref(fd);
 216
 217         cl->done_func = done_func;
 218         cl->done_data = done_data;
 219
 220         found = cache_find_location(CACHE_TYPE_SIM, cl->fd->path);
 221         if (found && filetime(found) == filetime(cl->fd->path))
 222                 {
 223                 cl->cd = cache_sim_data_load(found);
 224                 }
 225         g_free(found);
 226
 227         if (!cl->cd) cl->cd = cache_sim_data_new();
 228
 229         cl->todo_mask = load_mask;
 230         cl->done_mask = CACHE_LOADER_NONE;
 231
 232         cl->il = NULL;
 233         cl->idle_id = g_idle_add(cache_loader_idle_cb, cl);
 234
 235         cl->error = FALSE;
 236
 237         return cl;
 238 }
 239
 240 void cache_loader_free(CacheLoader *cl)
 241 {
 242         if (!cl) return;
 243
 244         if (cl->idle_id)
 245                 {
 246                 g_source_remove(cl->idle_id);
 247                 cl->idle_id = 0;
 248                 }
 249
 250         image_loader_free(cl->il);
 251         cache_sim_data_free(cl->cd);
 252
 253         file_data_unref(cl->fd);
 254         g_free(cl);
 255 }
 256 /* vim: set shiftwidth=8 softtabstop=0 cindent cinoptions={1s: */