1 /* -*- Mode: C; tab-width: 8; indent-tabs-mode: t; c-basic-offset: 8 -*- */
2
3 /*
4 * GThumb
5 *
6 * Copyright (C) 2009 Free Software Foundation, Inc.
7 *
8 * This program is free software; you can redistribute it and/or modify
9 * it under the terms of the GNU General Public License as published by
10 * the Free Software Foundation; either version 2 of the License, or
11 * (at your option) any later version.
12 *
13 * This program is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 * GNU General Public License for more details.
17 *
18 * You should have received a copy of the GNU General Public License
19 * along with this program. If not, see <http://www.gnu.org/licenses/>.
20 */
21
22 #include <config.h>
23 #include <glib/gi18n.h>
24 #include <gtk/gtk.h>
25 #include "gio-utils.h"
26 #include "glib-utils.h"
27 #include "gth-error.h"
28 #include "gth-hook.h"
29 #include "gth-main.h"
30 #include "gth-image-saver.h"
31
32
G_DEFINE_TYPE(GthImageSaver,gth_image_saver,G_TYPE_OBJECT)33 G_DEFINE_TYPE (GthImageSaver, gth_image_saver, G_TYPE_OBJECT)
34
35
36 static GtkWidget *
37 base_get_control (GthImageSaver *self)
38 {
39 return gtk_label_new (_("No options available for this file type"));
40 }
41
42
43 static void
base_save_options(GthImageSaver * self)44 base_save_options (GthImageSaver *self)
45 {
46 /* void */
47 }
48
49
50 static gboolean
base_can_save(GthImageSaver * self,const char * mime_type)51 base_can_save (GthImageSaver *self,
52 const char *mime_type)
53 {
54 return FALSE;
55 }
56
57
58 static gboolean
base_save_image(GthImageSaver * self,GthImage * image,char ** buffer,gsize * buffer_size,const char * mime_type,GCancellable * cancellable,GError ** error)59 base_save_image (GthImageSaver *self,
60 GthImage *image,
61 char **buffer,
62 gsize *buffer_size,
63 const char *mime_type,
64 GCancellable *cancellable,
65 GError **error)
66 {
67 return FALSE;
68 }
69
70
71 static void
gth_image_saver_class_init(GthImageSaverClass * klass)72 gth_image_saver_class_init (GthImageSaverClass *klass)
73 {
74 klass->id = "";
75 klass->display_name = "";
76 klass->get_control = base_get_control;
77 klass->save_options = base_save_options;
78 klass->can_save = base_can_save;
79 klass->save_image = base_save_image;
80 }
81
82
83 static void
gth_image_saver_init(GthImageSaver * self)84 gth_image_saver_init (GthImageSaver *self)
85 {
86 /* void */
87 }
88
89
90 const char *
gth_image_saver_get_id(GthImageSaver * self)91 gth_image_saver_get_id (GthImageSaver *self)
92 {
93 return GTH_IMAGE_SAVER_GET_CLASS (self)->id;
94 }
95
96
97 const char *
gth_image_saver_get_display_name(GthImageSaver * self)98 gth_image_saver_get_display_name (GthImageSaver *self)
99 {
100 return GTH_IMAGE_SAVER_GET_CLASS (self)->display_name;
101 }
102
103
104 const char *
gth_image_saver_get_mime_type(GthImageSaver * self)105 gth_image_saver_get_mime_type (GthImageSaver *self)
106 {
107 return GTH_IMAGE_SAVER_GET_CLASS (self)->mime_type;
108 }
109
110
111 const char *
gth_image_saver_get_extensions(GthImageSaver * self)112 gth_image_saver_get_extensions (GthImageSaver *self)
113 {
114 return GTH_IMAGE_SAVER_GET_CLASS (self)->extensions;
115 }
116
117
118 const char *
gth_image_saver_get_default_ext(GthImageSaver * self)119 gth_image_saver_get_default_ext (GthImageSaver *self)
120 {
121 if (GTH_IMAGE_SAVER_GET_CLASS (self)->get_default_ext != NULL)
122 return GTH_IMAGE_SAVER_GET_CLASS (self)->get_default_ext (self);
123 else
124 return gth_image_saver_get_extensions (self);
125 }
126
127
128 GtkWidget *
gth_image_saver_get_control(GthImageSaver * self)129 gth_image_saver_get_control (GthImageSaver *self)
130 {
131 return GTH_IMAGE_SAVER_GET_CLASS (self)->get_control (self);
132 }
133
134
135 void
gth_image_saver_save_options(GthImageSaver * self)136 gth_image_saver_save_options (GthImageSaver *self)
137 {
138 GTH_IMAGE_SAVER_GET_CLASS (self)->save_options (self);
139 }
140
141
142 gboolean
gth_image_saver_can_save(GthImageSaver * self,const char * mime_type)143 gth_image_saver_can_save (GthImageSaver *self,
144 const char *mime_type)
145 {
146 return GTH_IMAGE_SAVER_GET_CLASS (self)->can_save (self, mime_type);
147 }
148
149
150 static gboolean
gth_image_saver_save_image(GthImageSaver * self,GthImage * image,char ** buffer,gsize * buffer_size,const char * mime_type,GCancellable * cancellable,GError ** error)151 gth_image_saver_save_image (GthImageSaver *self,
152 GthImage *image,
153 char **buffer,
154 gsize *buffer_size,
155 const char *mime_type,
156 GCancellable *cancellable,
157 GError **error)
158 {
159 return GTH_IMAGE_SAVER_GET_CLASS (self)->save_image (self,
160 image,
161 buffer,
162 buffer_size,
163 mime_type,
164 cancellable,
165 error);
166 }
167
168
169 static GthImageSaveData *
_gth_image_save_to_buffer_common(GthImage * image,const char * mime_type,GthFileData * file_data,GCancellable * cancellable,GError ** p_error)170 _gth_image_save_to_buffer_common (GthImage *image,
171 const char *mime_type,
172 GthFileData *file_data,
173 GCancellable *cancellable,
174 GError **p_error)
175 {
176 GthImageSaver *saver;
177 char *buffer;
178 gsize buffer_size;
179 GError *error = NULL;
180 GthImageSaveData *save_data = NULL;
181
182 saver = gth_main_get_image_saver (mime_type);
183 if (saver == NULL) {
184 if (p_error != NULL)
185 *p_error = g_error_new (GTH_ERROR, GTH_ERROR_GENERIC, _("Could not find a suitable module to save the image as “%s”"), mime_type);
186 return NULL;
187 }
188
189 if (gth_image_saver_save_image (saver,
190 image,
191 &buffer,
192 &buffer_size,
193 mime_type,
194 cancellable,
195 &error))
196 {
197 save_data = g_new0 (GthImageSaveData, 1);
198 save_data->file_data = _g_object_ref (file_data);
199 save_data->image = gth_image_copy (image);
200 save_data->mime_type = mime_type;
201 save_data->buffer = buffer;
202 save_data->buffer_size = buffer_size;
203 save_data->files = NULL;
204 save_data->error = NULL;
205 save_data->cancellable = _g_object_ref (cancellable);
206
207 if (save_data->file_data != NULL)
208 gth_hook_invoke ("save-image", save_data);
209
210 if ((save_data->error != NULL) && (p_error != NULL))
211 *p_error = g_error_copy (*save_data->error);
212 }
213 else {
214 if (p_error != NULL)
215 *p_error = error;
216 else
217 _g_error_free (error);
218 }
219
220 g_object_unref (saver);
221
222 return save_data;
223 }
224
225
226 static void
gth_image_save_file_free(GthImageSaveFile * file)227 gth_image_save_file_free (GthImageSaveFile *file)
228 {
229 g_object_unref (file->file);
230 g_free (file->buffer);
231 g_free (file);
232 }
233
234
235 static void
gth_image_save_data_free(GthImageSaveData * data)236 gth_image_save_data_free (GthImageSaveData *data)
237 {
238 _g_object_unref (data->cancellable);
239 _g_object_unref (data->file_data);
240 g_object_unref (data->image);
241 g_list_foreach (data->files, (GFunc) gth_image_save_file_free, NULL);
242 g_list_free (data->files);
243 g_free (data);
244 }
245
246
247 gboolean
gth_image_save_to_buffer(GthImage * image,const char * mime_type,GthFileData * file_data,char ** buffer,gsize * buffer_size,GCancellable * cancellable,GError ** p_error)248 gth_image_save_to_buffer (GthImage *image,
249 const char *mime_type,
250 GthFileData *file_data,
251 char **buffer,
252 gsize *buffer_size,
253 GCancellable *cancellable,
254 GError **p_error)
255 {
256 GthImageSaveData *save_data;
257
258 g_return_val_if_fail (image != NULL, FALSE);
259
260 save_data = _gth_image_save_to_buffer_common (image,
261 mime_type,
262 file_data,
263 cancellable,
264 p_error);
265
266 if (save_data != NULL) {
267 *buffer = save_data->buffer;
268 *buffer_size = save_data->buffer_size;
269 gth_image_save_data_free (save_data);
270 return TRUE;
271 }
272
273 return FALSE;
274 }
275
276
277 /* -- gth_image_save_to_buffer_async -- */
278
279
280 typedef struct {
281 GthImage *image;
282 char *mime_type;
283 GthFileData *file_data;
284 gboolean replace;
285 GCancellable *cancellable;
286 GthFileDataFunc ready_func;
287 gpointer user_data;
288 } SaveArguments;
289
290
291 static void
save_arguments_free(SaveArguments * arguments)292 save_arguments_free (SaveArguments *arguments)
293 {
294 _g_object_unref (arguments->image);
295 g_free (arguments->mime_type);
296 _g_object_unref (arguments->file_data);
297 _g_object_unref (arguments->cancellable);
298 g_free (arguments);
299 }
300
301
302 static void
303 gth_image_save_to_buffer_async (GthImage *image,
304 const char *mime_type,
305 GthFileData *file_data,
306 GCancellable *cancellable,
307 GAsyncReadyCallback callback,
308 gpointer user_data);
309
310
311 static gboolean
gth_image_save_to_buffer_finish(GAsyncResult * result,GthImageSaveData ** save_data,GError ** error)312 gth_image_save_to_buffer_finish (GAsyncResult *result,
313 GthImageSaveData **save_data,
314 GError **error)
315 {
316 GthImageSaveData *data;
317
318 data = g_task_propagate_pointer (G_TASK (result), error);
319 if (data == NULL)
320 return FALSE;
321
322 if (save_data != NULL)
323 *save_data = data;
324 else
325 gth_image_save_data_free (data);
326
327 return TRUE;
328 }
329
330
331 static void
save_to_buffer_thread(GTask * task,gpointer source_object,gpointer task_data,GCancellable * cancellable)332 save_to_buffer_thread (GTask *task,
333 gpointer source_object,
334 gpointer task_data,
335 GCancellable *cancellable)
336 {
337 SaveArguments *arguments;
338 GthImageSaveData *data;
339 GError *error = NULL;
340
341 arguments = g_task_get_task_data (task);
342 data = _gth_image_save_to_buffer_common (arguments->image,
343 arguments->mime_type,
344 arguments->file_data,
345 cancellable,
346 &error);
347 if (data != NULL)
348 g_task_return_pointer (task, data, (GDestroyNotify) gth_image_save_data_free);
349 else
350 g_task_return_error (task, error);
351 }
352
353
354 static void
gth_image_save_to_buffer_async(GthImage * image,const char * mime_type,GthFileData * file_data,GCancellable * cancellable,GAsyncReadyCallback callback,gpointer user_data)355 gth_image_save_to_buffer_async (GthImage *image,
356 const char *mime_type,
357 GthFileData *file_data,
358 GCancellable *cancellable,
359 GAsyncReadyCallback callback,
360 gpointer user_data)
361 {
362 GTask *task;
363 SaveArguments *arguments;
364
365 g_return_if_fail (image != NULL);
366 g_return_if_fail (file_data != NULL);
367
368 arguments = g_new0 (SaveArguments, 1);
369 arguments->image = g_object_ref (image);
370 arguments->mime_type = g_strdup (mime_type);
371 arguments->file_data = g_object_ref (file_data);
372
373 task = g_task_new (NULL, cancellable, callback, user_data);
374 g_task_set_task_data (task, arguments, (GDestroyNotify) save_arguments_free);
375 g_task_run_in_thread (task, save_to_buffer_thread);
376
377 g_object_unref (task);
378 }
379
380
381 /* -- gth_image_save_to_file -- */
382
383
384 typedef struct {
385 GthImageSaveData *data;
386 GthFileDataFunc ready_func;
387 gpointer ready_data;
388 GList *current;
389 } SaveData;
390
391
392 static void
save_completed(SaveData * save_data)393 save_completed (SaveData *save_data)
394 {
395 if (save_data->data->error != NULL)
396 (*save_data->ready_func) (save_data->data->file_data, *save_data->data->error, save_data->ready_data);
397 else
398 (*save_data->ready_func) (save_data->data->file_data, NULL, save_data->ready_data);
399 gth_image_save_data_free (save_data->data);
400 g_free (save_data);
401 }
402
403
404 static void save_current_file (SaveData *save_data);
405
406
407 static void
file_saved_cb(void ** buffer,gsize count,GError * error,gpointer user_data)408 file_saved_cb (void **buffer,
409 gsize count,
410 GError *error,
411 gpointer user_data)
412 {
413 SaveData *save_data = user_data;
414
415 *buffer = NULL; /* do not free the buffer, it's owned by file->buffer */
416
417 if (error != NULL) {
418 save_data->data->error = &error;
419 save_completed (save_data);
420 return;
421 }
422
423 save_data->current = save_data->current->next;
424 save_current_file (save_data);
425 }
426
427
428 static void
save_current_file(SaveData * save_data)429 save_current_file (SaveData *save_data)
430 {
431 GthImageSaveFile *file;
432
433 if (save_data->current == NULL) {
434 save_completed (save_data);
435 return;
436 }
437
438 file = save_data->current->data;
439 _g_file_write_async (file->file,
440 file->buffer,
441 file->buffer_size,
442 (g_file_equal (save_data->data->file_data->file, file->file) ? save_data->data->replace : TRUE),
443 G_PRIORITY_DEFAULT,
444 save_data->data->cancellable,
445 file_saved_cb,
446 save_data);
447 }
448
449
450 static void
save_files(GthImageSaveData * data,GthFileDataFunc ready_func,gpointer ready_data)451 save_files (GthImageSaveData *data,
452 GthFileDataFunc ready_func,
453 gpointer ready_data)
454 {
455 SaveData *save_data;
456
457 save_data = g_new0 (SaveData, 1);
458 save_data->data = data;
459 save_data->ready_func = ready_func;
460 save_data->ready_data = ready_data;
461
462 save_data->current = save_data->data->files;
463 save_current_file (save_data);
464 }
465
466
467 static void
save_to_buffer_ready_cb(GObject * source_object,GAsyncResult * result,gpointer user_data)468 save_to_buffer_ready_cb (GObject *source_object,
469 GAsyncResult *result,
470 gpointer user_data)
471 {
472 SaveArguments *arguments = user_data;
473 GthImageSaveData *data = NULL;
474 GError *error = NULL;
475 GthImageSaveFile *file;
476
477 if (! gth_image_save_to_buffer_finish (result, &data, &error)) {
478 gth_file_data_ready_with_error (arguments->file_data,
479 arguments->ready_func,
480 arguments->user_data,
481 error);
482 save_arguments_free (arguments);
483 return;
484 }
485
486 data->replace = arguments->replace;
487
488 file = g_new0 (GthImageSaveFile, 1);
489 file->file = g_object_ref (data->file_data->file);
490 file->buffer = data->buffer;
491 file->buffer_size = data->buffer_size;
492 data->files = g_list_prepend (data->files, file);
493
494 save_files (data, arguments->ready_func, arguments->user_data);
495
496 save_arguments_free (arguments);
497 }
498
499
500 void
gth_image_save_to_file(GthImage * image,const char * mime_type,GthFileData * file_data,gboolean replace,GCancellable * cancellable,GthFileDataFunc ready_func,gpointer user_data)501 gth_image_save_to_file (GthImage *image,
502 const char *mime_type,
503 GthFileData *file_data,
504 gboolean replace,
505 GCancellable *cancellable,
506 GthFileDataFunc ready_func,
507 gpointer user_data)
508 {
509 SaveArguments *arguments;
510
511 g_return_if_fail (image != NULL);
512 g_return_if_fail (file_data != NULL);
513
514 arguments = g_new0 (SaveArguments, 1);
515 arguments->file_data = g_object_ref (file_data);
516 arguments->replace = replace;
517 arguments->cancellable = _g_object_ref (cancellable);
518 arguments->ready_func = ready_func;
519 arguments->user_data = user_data;
520
521 gth_image_save_to_buffer_async (image,
522 mime_type,
523 file_data,
524 cancellable,
525 save_to_buffer_ready_cb,
526 arguments);
527 }
528