image_list.c 37.3 KB
Newer Older
1
/*
2
 *  Copyright (c) Stephan Arts 2009-2011 <stephan@xfce.org>
3 4 5 6 7
 *
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 2 of the License, or
 *  (at your option) any later version.
Stephan Arts's avatar
Stephan Arts committed
8 9
 *
 *  This program is distributed in the hope that it will be useful,
10 11 12 13 14 15 16
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU Library General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program; if not, write to the Free Software
 *  Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
Stephan Arts's avatar
Stephan Arts committed
17
 *
Stephan Arts's avatar
Stephan Arts committed
18 19
 *  Sorting-algorithm taken from the thunar filemanager. 
 *    Copyright (c) Benedict Meurer <benny@xfce.org>
20 21 22 23 24 25 26
 */

#include <config.h>

#include <gtk/gtk.h>
#include <gtk/gtkmarshal.h>

27
#include <stdlib.h>
28 29 30 31
#include <string.h>

#include <libexif/exif-data.h>

Stephan Arts's avatar
Stephan Arts committed
32
#include "util.h"
Stephan Arts's avatar
Stephan Arts committed
33
#include "file.h"
34
#include "image_list.h"
35
#include "thumbnailer.h"
36
#include "settings.h"
37

Stephan Arts's avatar
Stephan Arts committed
38 39 40 41 42
static void
rstto_image_list_tree_model_init (GtkTreeModelIface *iface);
static void
rstto_image_list_tree_sortable_init(GtkTreeSortableIface *iface);

43 44 45 46 47 48 49
static void 
rstto_image_list_init(RsttoImageList *);
static void
rstto_image_list_class_init(RsttoImageListClass *);
static void
rstto_image_list_dispose(GObject *object);

Stephan Arts's avatar
Stephan Arts committed
50 51 52 53 54 55 56 57
static void
cb_file_monitor_changed (
        GFileMonitor      *monitor,
        GFile             *file,
        GFile             *other_file,
        GFileMonitorEvent  event_type,
        gpointer           user_data );

58 59 60 61 62 63 64
static void 
rstto_image_list_iter_init(RsttoImageListIter *);
static void
rstto_image_list_iter_class_init(RsttoImageListIterClass *);
static void
rstto_image_list_iter_dispose(GObject *object);

65 66 67 68 69 70
static void
cb_rstto_wrap_images_changed (
        GObject *settings,
        GParamSpec *pspec,
        gpointer user_data);

71 72 73 74 75 76
static void
cb_rstto_thumbnailer_ready(
        RsttoThumbnailer *thumbnailer,
        RsttoFile *file,
        gpointer user_data);

77 78 79 80 81 82 83 84 85
static void
rstto_image_list_monitor_dir (
        RsttoImageList *image_list,
        GFile *dir );

static void
rstto_image_list_remove_all (
        RsttoImageList *image_list);

86 87 88 89
static gboolean
iter_next (
        RsttoImageListIter *iter,
        gboolean sticky);
90

91 92 93 94
static gboolean
iter_previous (
        RsttoImageListIter *iter,
        gboolean sticky);
95

96 97 98 99 100 101
static void 
iter_set_position (
        RsttoImageListIter *iter,
        gint pos,
        gboolean sticky);

Stephan Arts's avatar
Stephan Arts committed
102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172
/***************************************/
/*  Begin TreeModelIface Functions     */
/***************************************/

static GtkTreeModelFlags
image_list_model_get_flags ( GtkTreeModel *tree_model );

static gint
image_list_model_get_n_columns ( GtkTreeModel *tree_model );

static GType
image_list_model_get_column_type (
        GtkTreeModel *tree_model,
        gint index_ );

static gboolean
image_list_model_get_iter (
        GtkTreeModel *tree_model,
        GtkTreeIter *iter,
        GtkTreePath *path );

static GtkTreePath *
image_list_model_get_path (
        GtkTreeModel *tree_model,
        GtkTreeIter *iter );

static void 
image_list_model_get_value (
        GtkTreeModel *tree_model,
        GtkTreeIter *iter,
        gint column,
        GValue *value );

static gboolean
image_list_model_iter_children (
        GtkTreeModel *tree_model,
        GtkTreeIter *iter,
        GtkTreeIter *parent );

static gboolean
image_list_model_iter_has_child (
        GtkTreeModel *tree_model,
        GtkTreeIter *iter );

static gboolean
image_list_model_iter_parent (
        GtkTreeModel *tree_model,
        GtkTreeIter *iter,
        GtkTreeIter *child );

static gint
image_list_model_iter_n_children (
        GtkTreeModel *tree_model,
        GtkTreeIter *iter );

static gboolean 
image_list_model_iter_nth_child (
        GtkTreeModel *tree_model,
        GtkTreeIter *iter,
        GtkTreeIter *parent,
        gint n );

static gboolean
image_list_model_iter_next (
        GtkTreeModel *tree_model,
        GtkTreeIter *iter );

/***************************************/
/*  End TreeModelIface Functions       */
/***************************************/

173 174 175
static RsttoImageListIter * rstto_image_list_iter_new ();

static gint
Stephan Arts's avatar
Stephan Arts committed
176
cb_rstto_image_list_image_name_compare_func (RsttoFile *a, RsttoFile *b);
Stephan Arts's avatar
Stephan Arts committed
177
static gint
Stephan Arts's avatar
Stephan Arts committed
178
cb_rstto_image_list_exif_date_compare_func (RsttoFile *a, RsttoFile *b);
179 180 181 182 183 184

static GObjectClass *parent_class = NULL;
static GObjectClass *iter_parent_class = NULL;

enum
{
Stephan Arts's avatar
Stephan Arts committed
185
    RSTTO_IMAGE_LIST_SIGNAL_REMOVE_IMAGE = 0,
186 187 188 189 190 191 192
    RSTTO_IMAGE_LIST_SIGNAL_REMOVE_ALL,
    RSTTO_IMAGE_LIST_SIGNAL_COUNT
};

enum
{
    RSTTO_IMAGE_LIST_ITER_SIGNAL_CHANGED = 0,
193
    RSTTO_IMAGE_LIST_ITER_SIGNAL_PREPARE_CHANGE,
194 195 196 197 198 199
    RSTTO_IMAGE_LIST_ITER_SIGNAL_COUNT
};

struct _RsttoImageListIterPriv
{
    RsttoImageList *image_list;
200 201 202 203
    RsttoFile      *file;
    
    /* This is set if the iter-position is chosen by the user */
    gboolean        sticky; 
204 205 206 207
};

struct _RsttoImageListPriv
{
Stephan Arts's avatar
Stephan Arts committed
208 209
    gint           stamp;
    GFileMonitor  *monitor;
210
    RsttoSettings *settings;
211
    RsttoThumbnailer *thumbnailer;
212
    GtkFileFilter *filter;
Stephan Arts's avatar
Stephan Arts committed
213

Stephan Arts's avatar
Stephan Arts committed
214 215 216 217 218
    GList        *images;
    gint          n_images;

    GSList       *iterators;
    GCompareFunc  cb_rstto_image_list_compare_func;
219 220

    gboolean      wrap_images;
221 222 223 224 225 226
};

static gint rstto_image_list_signals[RSTTO_IMAGE_LIST_SIGNAL_COUNT];
static gint rstto_image_list_iter_signals[RSTTO_IMAGE_LIST_ITER_SIGNAL_COUNT];

GType
Stephan Arts's avatar
Stephan Arts committed
227
rstto_image_list_get_type (void)
228
{
Stephan Arts's avatar
Stephan Arts committed
229 230 231 232 233 234 235 236 237 238 239 240
    static const GInterfaceInfo tree_model_info =
    {
        (GInterfaceInitFunc) rstto_image_list_tree_model_init,
            NULL,
            NULL
    };
    static const GInterfaceInfo tree_sort_info =
    {
        (GInterfaceInitFunc) rstto_image_list_tree_sortable_init,
            NULL,
            NULL
    };
241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259
    static GType rstto_image_list_type = 0;

    if (!rstto_image_list_type)
    {
        static const GTypeInfo rstto_image_list_info = 
        {
            sizeof (RsttoImageListClass),
            (GBaseInitFunc) NULL,
            (GBaseFinalizeFunc) NULL,
            (GClassInitFunc) rstto_image_list_class_init,
            (GClassFinalizeFunc) NULL,
            NULL,
            sizeof (RsttoImageList),
            0,
            (GInstanceInitFunc) rstto_image_list_init,
            NULL
        };

        rstto_image_list_type = g_type_register_static (G_TYPE_OBJECT, "RsttoImageList", &rstto_image_list_info, 0);
Stephan Arts's avatar
Stephan Arts committed
260 261 262 263

        g_type_add_interface_static (rstto_image_list_type, GTK_TYPE_TREE_MODEL, &tree_model_info);

        g_type_add_interface_static (rstto_image_list_type, GTK_TYPE_TREE_SORTABLE, &tree_sort_info);
264
    }
Stephan Arts's avatar
Stephan Arts committed
265 266


267 268 269
    return rstto_image_list_type;
}

Stephan Arts's avatar
Stephan Arts committed
270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298
static void
rstto_image_list_tree_model_init (GtkTreeModelIface *iface)
{
    iface->get_flags       = image_list_model_get_flags;
    iface->get_n_columns   = image_list_model_get_n_columns;
    iface->get_column_type = image_list_model_get_column_type;
    iface->get_iter        = image_list_model_get_iter;
    iface->get_path        = image_list_model_get_path;
    iface->get_value       = image_list_model_get_value;
    iface->iter_children   = image_list_model_iter_children;
    iface->iter_has_child  = image_list_model_iter_has_child;
    iface->iter_n_children = image_list_model_iter_n_children;
    iface->iter_nth_child  = image_list_model_iter_nth_child;
    iface->iter_parent     = image_list_model_iter_parent;
    iface->iter_next       = image_list_model_iter_next;
}

static void
rstto_image_list_tree_sortable_init(GtkTreeSortableIface *iface)
{
#if 0 
    iface->get_sort_column_id    = sq_archive_store_get_sort_column_id;
    iface->set_sort_column_id    = sq_archive_store_set_sort_column_id;
    iface->set_sort_func         = sq_archive_store_set_sort_func;            /*NOT SUPPORTED*/
    iface->set_default_sort_func = sq_archive_store_set_default_sort_func;    /*NOT SUPPORTED*/
    iface->has_default_sort_func = sq_archive_store_has_default_sort_func;
#endif 
}

299 300 301
static void
rstto_image_list_init(RsttoImageList *image_list)
{
302

303
    image_list->priv = g_new0 (RsttoImageListPriv, 1);
Stephan Arts's avatar
Stephan Arts committed
304
    image_list->priv->stamp = g_random_int();
305
    image_list->priv->settings = rstto_settings_new ();
306
    image_list->priv->thumbnailer = rstto_thumbnailer_new();
307 308 309
    image_list->priv->filter = gtk_file_filter_new ();
    g_object_ref_sink (image_list->priv->filter);
    gtk_file_filter_add_pixbuf_formats (image_list->priv->filter);
310

Stephan Arts's avatar
Stephan Arts committed
311
    image_list->priv->cb_rstto_image_list_compare_func = (GCompareFunc)cb_rstto_image_list_image_name_compare_func;
312 313 314 315 316 317 318 319 320 321 322

    image_list->priv->wrap_images = rstto_settings_get_boolean_property (
            image_list->priv->settings,
            "wrap-images");

    g_signal_connect (
            G_OBJECT(image_list->priv->settings),
            "notify::wrap-images",
            G_CALLBACK (cb_rstto_wrap_images_changed),
            image_list);

323 324 325 326 327 328
    g_signal_connect (
            G_OBJECT(image_list->priv->thumbnailer),
            "ready",
            G_CALLBACK (cb_rstto_thumbnailer_ready),
            image_list);

329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366
}

static void
rstto_image_list_class_init(RsttoImageListClass *nav_class)
{
    GObjectClass *object_class = G_OBJECT_CLASS(nav_class);

    parent_class = g_type_class_peek_parent(nav_class);

    object_class->dispose = rstto_image_list_dispose;

    rstto_image_list_signals[RSTTO_IMAGE_LIST_SIGNAL_REMOVE_IMAGE] = g_signal_new("remove-image",
            G_TYPE_FROM_CLASS(nav_class),
            G_SIGNAL_RUN_LAST | G_SIGNAL_ACTION,
            0,
            NULL,
            NULL,
            g_cclosure_marshal_VOID__OBJECT,
            G_TYPE_NONE,
            1,
            G_TYPE_OBJECT,
            NULL);

    rstto_image_list_signals[RSTTO_IMAGE_LIST_SIGNAL_REMOVE_ALL] = g_signal_new("remove-all",
            G_TYPE_FROM_CLASS(nav_class),
            G_SIGNAL_RUN_LAST | G_SIGNAL_ACTION,
            0,
            NULL,
            NULL,
            g_cclosure_marshal_VOID__VOID,
            G_TYPE_NONE,
            0,
            NULL);
}

static void
rstto_image_list_dispose(GObject *object)
{
367 368 369 370 371 372 373 374
    RsttoImageList *image_list = RSTTO_IMAGE_LIST(object);
    if (NULL != image_list->priv)
    {
        if (image_list->priv->settings)
        {
            g_object_unref (image_list->priv->settings);
            image_list->priv->settings = NULL;
        }
375

376 377 378 379 380 381
        if (image_list->priv->thumbnailer)
        {
            g_object_unref (image_list->priv->thumbnailer);
            image_list->priv->thumbnailer = NULL;
        }

382 383 384 385 386
        if (image_list->priv->filter)
        {
            g_object_unref (image_list->priv->filter);
            image_list->priv->filter= NULL;
        }
387 388 389 390
        
        g_free (image_list->priv);
        image_list->priv = NULL;
    }
391 392 393
}

RsttoImageList *
Stephan Arts's avatar
Stephan Arts committed
394
rstto_image_list_new (void)
395 396 397 398 399 400 401 402 403
{
    RsttoImageList *image_list;

    image_list = g_object_new(RSTTO_TYPE_IMAGE_LIST, NULL);

    return image_list;
}

gboolean
404 405 406 407
rstto_image_list_add_file (
        RsttoImageList *image_list,
        RsttoFile *file,
        GError **error )
408
{
409
    GtkFileFilterInfo filter_info;
Stephan Arts's avatar
Stephan Arts committed
410
    GList *image_iter = g_list_find (image_list->priv->images, file);
411
    GSList *iter = image_list->priv->iterators;
Stephan Arts's avatar
Stephan Arts committed
412 413 414
    gint i = 0;
    GtkTreePath *path = NULL;
    GtkTreeIter t_iter;
415

416 417 418
    g_return_val_if_fail ( NULL != file , FALSE);
    g_return_val_if_fail ( RSTTO_IS_FILE (file) , FALSE);

Stephan Arts's avatar
Stephan Arts committed
419 420
    if (!image_iter)
    {
421
        if (file)
Stephan Arts's avatar
Stephan Arts committed
422
        {
423 424 425
            filter_info.contains =  GTK_FILE_FILTER_MIME_TYPE | GTK_FILE_FILTER_URI;
            filter_info.uri = rstto_file_get_uri (file);
            filter_info.mime_type = rstto_file_get_content_type (file);
Stephan Arts's avatar
Stephan Arts committed
426

427
            if ( TRUE == gtk_file_filter_filter (image_list->priv->filter, &filter_info))
Stephan Arts's avatar
Stephan Arts committed
428
            {
429 430
                g_object_ref (G_OBJECT (file));

431 432 433
                image_list->priv->images = g_list_insert_sorted (image_list->priv->images, file, rstto_image_list_get_compare_func (image_list));

                image_list->priv->n_images++;
Stephan Arts's avatar
Stephan Arts committed
434 435 436 437 438 439 440 441 442
                i = g_list_index (image_list->priv->images, file);

                path = gtk_tree_path_new();
                gtk_tree_path_append_index (path, i);
                t_iter.stamp = image_list->priv->stamp;
                t_iter.user_data = file;
                t_iter.user_data3 = GINT_TO_POINTER(i);

                gtk_tree_model_row_inserted(GTK_TREE_MODEL(image_list), path, &t_iter);
443 444 445

                /** TODO: update all iterators */
                while (iter)
Stephan Arts's avatar
Stephan Arts committed
446
                {
447 448 449 450 451
                    if (FALSE == RSTTO_IMAGE_LIST_ITER(iter->data)->priv->sticky)
                    {
                        rstto_image_list_iter_find_file (iter->data, file);
                    }
                    iter = g_slist_next (iter);
Stephan Arts's avatar
Stephan Arts committed
452
                }
Stephan Arts's avatar
Stephan Arts committed
453

454 455 456 457 458
                return TRUE;
            }
            else
            {
                return FALSE;
Stephan Arts's avatar
Stephan Arts committed
459 460
            }
        }
Stephan Arts's avatar
Stephan Arts committed
461
        return FALSE;
462
    }
463

Stephan Arts's avatar
Stephan Arts committed
464
    return TRUE;
465 466 467 468 469 470 471 472
}

gint
rstto_image_list_get_n_images (RsttoImageList *image_list)
{
    return g_list_length (image_list->priv->images);
}

Stephan Arts's avatar
Stephan Arts committed
473 474 475 476 477 478 479 480
/**
 * rstto_image_list_get_iter:
 * @image_list:
 *
 * TODO: track iterators
 *
 * return iter;
 */
481 482 483
RsttoImageListIter *
rstto_image_list_get_iter (RsttoImageList *image_list)
{
Stephan Arts's avatar
Stephan Arts committed
484
    RsttoFile *file = NULL;
Stephan Arts's avatar
Stephan Arts committed
485
    RsttoImageListIter *iter = NULL;
486
    if (image_list->priv->images)
487
        file = image_list->priv->images->data;
488

489
    iter = rstto_image_list_iter_new (image_list, file);
490

Stephan Arts's avatar
Stephan Arts committed
491 492
    image_list->priv->iterators = g_slist_prepend (image_list->priv->iterators, iter);

493 494 495 496 497
    return iter;
}


void
Stephan Arts's avatar
Stephan Arts committed
498
rstto_image_list_remove_file (RsttoImageList *image_list, RsttoFile *file)
499
{
Stephan Arts's avatar
Stephan Arts committed
500
    GSList *iter = NULL;
Stephan Arts's avatar
Stephan Arts committed
501
    RsttoFile *afile = NULL;
Stephan Arts's avatar
Stephan Arts committed
502
    GtkTreePath *path_ = NULL;
Stephan Arts's avatar
Stephan Arts committed
503

504
    if (g_list_find(image_list->priv->images, file))
505
    {
Stephan Arts's avatar
Stephan Arts committed
506

Stephan Arts's avatar
Stephan Arts committed
507
        iter = image_list->priv->iterators;
Stephan Arts's avatar
Stephan Arts committed
508 509
        while (iter)
        {
Stephan Arts's avatar
Stephan Arts committed
510
            if (rstto_file_equal(rstto_image_list_iter_get_file (iter->data), file))
Stephan Arts's avatar
Stephan Arts committed
511
            {
Stephan Arts's avatar
Stephan Arts committed
512 513 514 515 516 517
                
                path_ = gtk_tree_path_new();
                gtk_tree_path_append_index(path_,rstto_image_list_iter_get_position (iter->data));

                gtk_tree_model_row_deleted(GTK_TREE_MODEL(image_list), path_);

518
                if (rstto_image_list_iter_get_position (iter->data) == rstto_image_list_get_n_images (image_list)-1)
519
                {
520
                    iter_previous (iter->data, FALSE);
521 522 523
                }
                else
                {
524
                    iter_next (iter->data, FALSE);
525
                }
526 527 528 529
                /* If the image is still the same, 
                 * it's a single item list,
                 * and we should force the image in this iter to NULL
                 */
Stephan Arts's avatar
Stephan Arts committed
530
                if (rstto_file_equal(rstto_image_list_iter_get_file (iter->data), file))
531
                {
532
                    ((RsttoImageListIter *)(iter->data))->priv->file = NULL;
533
                    g_signal_emit (G_OBJECT (iter->data), rstto_image_list_iter_signals[RSTTO_IMAGE_LIST_ITER_SIGNAL_CHANGED], 0, NULL);
534
                }
Stephan Arts's avatar
Stephan Arts committed
535 536 537
            }
            iter = g_slist_next (iter);
        }
538

539
        image_list->priv->images = g_list_remove (image_list->priv->images, file);
540 541 542
        iter = image_list->priv->iterators;
        while (iter)
        {
543 544
            afile = rstto_image_list_iter_get_file(iter->data);
            if (NULL != afile)
545
            {
Stephan Arts's avatar
Stephan Arts committed
546
                if (rstto_file_equal(afile, file))
547
                {
548
                    iter_next (iter->data, FALSE);
549
                }
550 551 552 553
            }
            iter = g_slist_next (iter);
        }

554 555
        g_signal_emit (G_OBJECT (image_list), rstto_image_list_signals[RSTTO_IMAGE_LIST_SIGNAL_REMOVE_IMAGE], 0, file, NULL);
        g_object_unref(file);
556 557 558
    }
}

559
static void
560 561
rstto_image_list_remove_all (RsttoImageList *image_list)
{
Stephan Arts's avatar
Stephan Arts committed
562
    GSList *iter = NULL;
Stephan Arts's avatar
Stephan Arts committed
563 564 565 566 567 568 569 570 571 572 573 574 575 576 577 578
    GList *image_iter = image_list->priv->images;
    GtkTreePath *path_ = NULL;
    gint i = g_list_length (image_iter);

    while (image_iter)
    {
        i--;
        path_ = gtk_tree_path_new();
        gtk_tree_path_append_index(path_, i);

        gtk_tree_model_row_deleted(GTK_TREE_MODEL(image_list), path_);

        
        image_iter = g_list_next (image_iter);     
    }

579 580 581
    g_list_foreach (image_list->priv->images, (GFunc)g_object_unref, NULL);
    g_list_free (image_list->priv->images);
    image_list->priv->images = NULL;
Stephan Arts's avatar
Stephan Arts committed
582

Stephan Arts's avatar
Stephan Arts committed
583
    iter = image_list->priv->iterators;
Stephan Arts's avatar
Stephan Arts committed
584 585
    while (iter)
    {
586
        iter_set_position (iter->data, -1, FALSE);
Stephan Arts's avatar
Stephan Arts committed
587 588 589
        iter = g_slist_next (iter);
    }
    g_signal_emit (G_OBJECT (image_list), rstto_image_list_signals[RSTTO_IMAGE_LIST_SIGNAL_REMOVE_ALL], 0, NULL);
590 591
}

592 593 594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 626 627 628 629 630 631 632 633 634 635 636 637
gboolean
rstto_image_list_set_directory (
        RsttoImageList *image_list,
        GFile *dir,
        GError **error )
{
    /* Declare variables */
    GFileEnumerator *file_enumerator = NULL;
    GFileInfo *file_info;
    const gchar *filename;
    const gchar *content_type;
    GFile *child_file;

    /* Source code block */
    rstto_image_list_remove_all (image_list);

    /* Allow all images to be removed by providing NULL to dir */
    if ( NULL != dir )
    {
        file_enumerator = g_file_enumerate_children (dir, "standard::*", 0, NULL, NULL);

        if (NULL != file_enumerator)
        {
            for(file_info = g_file_enumerator_next_file (file_enumerator, NULL, NULL);
                NULL != file_info;
                file_info = g_file_enumerator_next_file (file_enumerator, NULL, NULL))
            {
                filename = g_file_info_get_name (file_info);
                content_type  = g_file_info_get_content_type (file_info);
                child_file = g_file_get_child (dir, filename);
                if (strncmp (content_type, "image/", 6) == 0)
                {
                    rstto_image_list_add_file (image_list, rstto_file_new (child_file), NULL);
                }
            }
            g_object_unref (file_enumerator);
            file_enumerator = NULL;
        }
    }

    rstto_image_list_monitor_dir ( image_list, dir );

    return TRUE;
}

static void
Stephan Arts's avatar
Stephan Arts committed
638 639 640 641 642 643 644 645 646 647 648 649
rstto_image_list_monitor_dir (
        RsttoImageList *image_list,
        GFile *dir )
{
    GFileMonitor *monitor = NULL;

    if ( NULL != image_list->priv->monitor )
    {
        g_object_unref (image_list->priv->monitor);
        image_list->priv->monitor = NULL;
    }

650 651 652 653 654 655 656 657 658 659 660 661 662 663 664
    /* Allow a monitor to be removed by providing NULL to dir */
    if ( NULL != dir )
    {
        monitor = g_file_monitor_directory (
                dir,
                G_FILE_MONITOR_NONE,
                NULL,
                NULL);

        g_signal_connect (
                G_OBJECT(monitor),
                "changed",
                G_CALLBACK (cb_file_monitor_changed),
                image_list);
    }
Stephan Arts's avatar
Stephan Arts committed
665 666 667 668 669 670 671 672 673 674 675 676 677 678 679 680 681 682 683 684 685 686 687 688 689 690 691

    image_list->priv->monitor = monitor;
}

static void
cb_file_monitor_changed (
        GFileMonitor      *monitor,
        GFile             *file,
        GFile             *other_file,
        GFileMonitorEvent  event_type,
        gpointer           user_data )
{
    RsttoImageList *image_list = RSTTO_IMAGE_LIST (user_data);
    RsttoFile *r_file = rstto_file_new (file);

    g_return_if_fail ( monitor == image_list->priv->monitor);

    switch ( event_type )
    {
        case G_FILE_MONITOR_EVENT_DELETED:
            rstto_image_list_remove_file ( image_list, r_file );
            break;
        case G_FILE_MONITOR_EVENT_CREATED:
            rstto_image_list_add_file (image_list, r_file, NULL);
            break;
        case G_FILE_MONITOR_EVENT_MOVED:
            rstto_image_list_remove_file ( image_list, r_file );
692 693 694 695

            /* Remove our reference, reusing pointer */
            g_object_unref (r_file);

Stephan Arts's avatar
Stephan Arts committed
696 697 698 699 700 701 702 703 704 705 706 707 708
            r_file = rstto_file_new (other_file);
            rstto_image_list_add_file (image_list, r_file, NULL);
            break;
        default:
            break;
    }

    if ( NULL != r_file )
    {
        g_object_unref (r_file);
    }
}

709 710 711


GType
Stephan Arts's avatar
Stephan Arts committed
712
rstto_image_list_iter_get_type (void)
713 714 715 716 717 718 719 720 721 722 723 724 725 726 727 728 729 730 731 732 733 734 735 736 737 738 739 740 741 742 743 744 745 746 747 748 749 750 751
{
    static GType rstto_image_list_iter_type = 0;

    if (!rstto_image_list_iter_type)
    {
        static const GTypeInfo rstto_image_list_iter_info = 
        {
            sizeof (RsttoImageListIterClass),
            (GBaseInitFunc) NULL,
            (GBaseFinalizeFunc) NULL,
            (GClassInitFunc) rstto_image_list_iter_class_init,
            (GClassFinalizeFunc) NULL,
            NULL,
            sizeof (RsttoImageListIter),
            0,
            (GInstanceInitFunc) rstto_image_list_iter_init,
            NULL
        };

        rstto_image_list_iter_type = g_type_register_static (G_TYPE_OBJECT, "RsttoImageListIter", &rstto_image_list_iter_info, 0);
    }
    return rstto_image_list_iter_type;
}

static void
rstto_image_list_iter_init (RsttoImageListIter *iter)
{
    iter->priv = g_new0 (RsttoImageListIterPriv, 1);
}

static void
rstto_image_list_iter_class_init(RsttoImageListIterClass *iter_class)
{
    GObjectClass *object_class = G_OBJECT_CLASS(iter_class);

    iter_parent_class = g_type_class_peek_parent(iter_class);

    object_class->dispose = rstto_image_list_iter_dispose;

752 753 754 755 756 757 758 759 760 761 762
    rstto_image_list_iter_signals[RSTTO_IMAGE_LIST_ITER_SIGNAL_PREPARE_CHANGE] = g_signal_new("prepare-change",
            G_TYPE_FROM_CLASS(iter_class),
            G_SIGNAL_RUN_LAST | G_SIGNAL_ACTION,
            0,
            NULL,
            NULL,
            g_cclosure_marshal_VOID__VOID,
            G_TYPE_NONE,
            0,
            NULL);

763 764 765 766 767 768 769 770 771 772 773 774 775 776 777 778 779
    rstto_image_list_iter_signals[RSTTO_IMAGE_LIST_ITER_SIGNAL_CHANGED] = g_signal_new("changed",
            G_TYPE_FROM_CLASS(iter_class),
            G_SIGNAL_RUN_LAST | G_SIGNAL_ACTION,
            0,
            NULL,
            NULL,
            g_cclosure_marshal_VOID__VOID,
            G_TYPE_NONE,
            0,
            NULL);

}

static void
rstto_image_list_iter_dispose (GObject *object)
{
    RsttoImageListIter *iter = RSTTO_IMAGE_LIST_ITER(object);
780
    if (iter->priv->file)
781
    {
782
        iter->priv->file = NULL;
783
    }
Stephan Arts's avatar
Stephan Arts committed
784 785 786 787 788 789

    if (iter->priv->image_list)
    {
        iter->priv->image_list->priv->iterators = g_slist_remove (iter->priv->image_list->priv->iterators, iter);
        iter->priv->image_list= NULL;
    }
790 791 792
}

static RsttoImageListIter *
Stephan Arts's avatar
Stephan Arts committed
793
rstto_image_list_iter_new (RsttoImageList *nav, RsttoFile *file)
794 795 796 797
{
    RsttoImageListIter *iter;

    iter = g_object_new(RSTTO_TYPE_IMAGE_LIST_ITER, NULL);
798
    iter->priv->file = file;
799 800 801 802 803 804
    iter->priv->image_list = nav;

    return iter;
}

gboolean
Stephan Arts's avatar
Stephan Arts committed
805
rstto_image_list_iter_find_file (RsttoImageListIter *iter, RsttoFile *file)
806
{
807
    gint pos = g_list_index (iter->priv->image_list->priv->images, file);
808 809
    if (pos > -1)
    {
810 811
        g_signal_emit (G_OBJECT (iter), rstto_image_list_iter_signals[RSTTO_IMAGE_LIST_ITER_SIGNAL_PREPARE_CHANGE], 0, NULL);

812
        if (iter->priv->file)
813
        {
814
            iter->priv->file = NULL;
815
        }
816
        iter->priv->file = file;
817
        iter->priv->sticky = TRUE;
818 819 820 821 822 823 824 825 826 827 828

        g_signal_emit (G_OBJECT (iter), rstto_image_list_iter_signals[RSTTO_IMAGE_LIST_ITER_SIGNAL_CHANGED], 0, NULL);

        return TRUE;
    }
    return FALSE;
}

gint
rstto_image_list_iter_get_position (RsttoImageListIter *iter)
{
829
    if ( NULL == iter->priv->file )
830
    {
831
        return -1;
832
    }
833
    return g_list_index (iter->priv->image_list->priv->images, iter->priv->file);
834 835
}

Stephan Arts's avatar
Stephan Arts committed
836
RsttoFile *
837
rstto_image_list_iter_get_file (RsttoImageListIter *iter)
838
{
839
    return iter->priv->file;
840 841
}

842 843 844 845 846
static void
iter_set_position (
        RsttoImageListIter *iter,
        gint pos,
        gboolean sticky )
847
{
848 849
    g_signal_emit (G_OBJECT (iter), rstto_image_list_iter_signals[RSTTO_IMAGE_LIST_ITER_SIGNAL_PREPARE_CHANGE], 0, NULL);

850
    if (iter->priv->file)
851
    {
852
        iter->priv->file = NULL;
853 854
    }

Stephan Arts's avatar
Stephan Arts committed
855
    if (pos >= 0)
Stephan Arts's avatar
Stephan Arts committed
856
    {
857
        iter->priv->file = g_list_nth_data (iter->priv->image_list->priv->images, pos); 
Stephan Arts's avatar
Stephan Arts committed
858
    }
859

860 861 862
    g_signal_emit (G_OBJECT (iter), rstto_image_list_iter_signals[RSTTO_IMAGE_LIST_ITER_SIGNAL_CHANGED], 0, NULL);
}

863 864 865 866 867 868 869 870 871 872
void
rstto_image_list_iter_set_position (RsttoImageListIter *iter, gint pos)
{
    iter_set_position ( iter, pos, TRUE );
}

static gboolean
iter_next (
        RsttoImageListIter *iter,
        gboolean sticky)
873
{
874
    GList *position = NULL;
875
    RsttoImageList *image_list = iter->priv->image_list;
876
    RsttoFile *file = iter->priv->file;
877
    gboolean ret_val = FALSE;
878 879 880

    g_signal_emit (G_OBJECT (iter), rstto_image_list_iter_signals[RSTTO_IMAGE_LIST_ITER_SIGNAL_PREPARE_CHANGE], 0, NULL);

881
    if (iter->priv->file)
882
    {
883 884
        position = g_list_find (iter->priv->image_list->priv->images, iter->priv->file);
        iter->priv->file = NULL;
885 886
    }

887 888
    iter->priv->sticky = sticky;

889 890
    position = g_list_next (position);
    if (position)
891
    {
892
        iter->priv->file = position->data; 
893 894 895 896

        /* We could move forward, set ret_val to TRUE */
        ret_val = TRUE;
    }
897 898
    else
    {
Stephan Arts's avatar
Stephan Arts committed
899

900
        if (TRUE == image_list->priv->wrap_images)
901
        {
902
            position = g_list_first (iter->priv->image_list->priv->images);
903 904 905 906 907 908

            /* We could move forward, wrapped back to the start of the
             * list, set ret_val to TRUE
             */
            ret_val = TRUE;
        }
909 910 911
        else
            position = g_list_last (iter->priv->image_list->priv->images);

912
        if (position)
913
            iter->priv->file = position->data; 
914
        else
915
            iter->priv->file = NULL;
916 917
    }

918 919 920 921
    if (file != iter->priv->file)
    {
        g_signal_emit (G_OBJECT (iter), rstto_image_list_iter_signals[RSTTO_IMAGE_LIST_ITER_SIGNAL_CHANGED], 0, NULL);
    }
922 923

    return ret_val;
924 925
}

926
gboolean
927 928 929 930 931
rstto_image_list_iter_next (RsttoImageListIter *iter)
{
    return iter_next (iter, TRUE);
}

932 933 934 935 936 937 938 939 940 941 942 943 944 945 946 947 948 949 950 951 952 953 954
gboolean
rstto_image_list_iter_has_next (RsttoImageListIter *iter)
{
    RsttoImageList *image_list = iter->priv->image_list;

    if (image_list->priv->wrap_images)
    {
        return TRUE;
    }
    else
    {
        if (rstto_image_list_iter_get_position (iter) ==
            (rstto_image_list_get_n_images (image_list) -1))
        {
            return FALSE;
        }
        else
        {
            return TRUE;
        }
    }
}

955 956 957 958
static gboolean
iter_previous (
        RsttoImageListIter *iter,
        gboolean sticky)
959
{
960
    GList *position = NULL;
961
    RsttoImageList *image_list = iter->priv->image_list;
962
    RsttoFile *file = iter->priv->file;
963
    gboolean ret_val = FALSE;
964 965 966

    g_signal_emit (G_OBJECT (iter), rstto_image_list_iter_signals[RSTTO_IMAGE_LIST_ITER_SIGNAL_PREPARE_CHANGE], 0, NULL);

967
    if (iter->priv->file)
968
    {
969 970
        position = g_list_find (iter->priv->image_list->priv->images, iter->priv->file);
        iter->priv->file = NULL;
971 972
    }

973 974
    iter->priv->sticky = sticky;

975 976
    position = g_list_previous (position);
    if (position)
977
    {
978
        iter->priv->file = position->data; 
979
    }
980 981
    else
    {
982 983
        if (TRUE == image_list->priv->wrap_images)
        {
984
            position = g_list_last (iter->priv->image_list->priv->images);
985
        }
986
        else
987
        {
988
            position = g_list_first (iter->priv->image_list->priv->images);
989
        }
990

991
        if (position)
992
        {
993
            iter->priv->file = position->data; 
994
        }
995
        else
996
        {
997
            iter->priv->file = NULL;
998
        }
999
    }
1000

1001 1002 1003 1004
    if (file != iter->priv->file)
    {
        g_signal_emit (G_OBJECT (iter), rstto_image_list_iter_signals[RSTTO_IMAGE_LIST_ITER_SIGNAL_CHANGED], 0, NULL);
    }
1005 1006

    return ret_val;
1007 1008 1009 1010 1011 1012

}
gboolean
rstto_image_list_iter_previous (RsttoImageListIter *iter)
{
    return iter_previous (iter, TRUE);
1013 1014
}

1015 1016 1017 1018 1019 1020 1021 1022 1023 1024 1025 1026 1027 1028 1029 1030 1031 1032 1033 1034 1035 1036 1037 1038

gboolean
rstto_image_list_iter_has_previous (RsttoImageListIter *iter)
{
    RsttoImageList *image_list = iter->priv->image_list;

    if (image_list->priv->wrap_images)
    {
        return TRUE;
    }
    else
    {
        if (rstto_image_list_iter_get_position (iter) == 0)
        {
            return FALSE;
        }
        else
        {
            return TRUE;
        }
    }
}


1039 1040 1041
RsttoImageListIter *
rstto_image_list_iter_clone (RsttoImageListIter *iter)
{
1042
    RsttoImageListIter *new_iter = rstto_image_list_iter_new (iter->priv->image_list, iter->priv->file);
Stephan Arts's avatar
Stephan Arts committed
1043
    rstto_image_list_iter_set_position (new_iter, rstto_image_list_iter_get_position(iter));
1044 1045 1046

    return new_iter;
}
1047 1048 1049 1050

GCompareFunc
rstto_image_list_get_compare_func (RsttoImageList *image_list)
{
Stephan Arts's avatar
Stephan Arts committed
1051 1052 1053
    return (GCompareFunc)image_list->priv->cb_rstto_image_list_compare_func;
}

1054 1055 1056
void
rstto_image_list_set_compare_func (RsttoImageList *image_list, GCompareFunc func)
{
1057
    GSList *iter = NULL;
1058
    image_list->priv->cb_rstto_image_list_compare_func = func;
1059
    image_list->priv->images = g_list_sort (image_list->priv->images,  func);
1060

1061 1062 1063 1064
    for (iter = image_list->priv->iterators; iter != NULL; iter = g_slist_next (iter))
    {
        g_signal_emit (G_OBJECT (iter->data), rstto_image_list_iter_signals[RSTTO_IMAGE_LIST_ITER_SIGNAL_CHANGED], 0, NULL);
    }
1065 1066
}

Stephan Arts's avatar
Stephan Arts committed
1067 1068 1069 1070
/***********************/
/*  Compare Functions  */
/***********************/

1071 1072 1073
void
rstto_image_list_set_sort_by_name (RsttoImageList *image_list)
{
1074
    rstto_image_list_set_compare_func (image_list, (GCompareFunc)cb_rstto_image_list_image_name_compare_func);
1075 1076 1077 1078 1079
}

void
rstto_image_list_set_sort_by_date (RsttoImageList *image_list)
{
1080
    rstto_image_list_set_compare_func (image_list, (GCompareFunc)cb_rstto_image_list_exif_date_compare_func);
1081 1082
}

Stephan Arts's avatar
Stephan Arts committed
1083 1084 1085 1086 1087 1088 1089 1090 1091
/**
 * cb_rstto_image_list_image_name_compare_func:
 * @a:
 * @b:
 *
 *
 * Return value: (see strcmp)
 */
static gint
Stephan Arts's avatar
Stephan Arts committed
1092
cb_rstto_image_list_image_name_compare_func (RsttoFile *a, RsttoFile *b)
Stephan Arts's avatar
Stephan Arts committed
1093
{
Stephan Arts's avatar
Stephan Arts committed
1094 1095
    const gchar *a_base = rstto_file_get_display_name (a);
    const gchar *b_base = rstto_file_get_display_name (b);
1096 1097 1098 1099 1100
    guint  ac;
    guint  bc;
    const gchar *ap = a_base;
    const gchar *bp = b_base;

Stephan Arts's avatar
Stephan Arts committed
1101
    gint result = 0;
1102 1103 1104 1105 1106 1107 1108 1109 1110 1111 1112 1113 1114 1115 1116 1117 1118 1119 1120 1121 1122 1123 1124 1125 1126 1127
    guint a_num = 0;
    guint b_num = 0;

    /* try simple (fast) ASCII comparison first */
    for (;; ++ap, ++bp)
    {
        /* check if the characters differ or we have a non-ASCII char
         */
        ac = *((const guchar *) ap);
        bc = *((const guchar *) bp);
        if (ac != bc || ac == 0 || ac > 127)
            break;
    }

    /* fallback to Unicode comparison */
    if (G_UNLIKELY (ac > 127 || bc > 127))
    {
        for (;; ap = g_utf8_next_char (ap), bp = g_utf8_next_char (bp))
        {
            /* check if characters differ or end of string */
            ac = g_utf8_get_char (ap);
            bc = g_utf8_get_char (bp);
            if (ac != bc || ac == 0)
                break;
        }
    }
Stephan Arts's avatar
Stephan Arts committed
1128

1129
    /* If both strings are equal, we're done */
Stephan Arts's avatar
Stephan Arts committed
1130 1131 1132 1133 1134
    if (ac == bc)
    {
        return 0;
    }
    else
1135 1136 1137 1138 1139 1140 1141 1142 1143 1144 1145 1146 1147 1148 1149 1150
    {
        if (G_UNLIKELY (g_ascii_isdigit (ac) || g_ascii_isdigit (bc)))
        {
            /* if both strings differ in a digit, we use a smarter comparison
             * to get sorting 'file1', 'file5', 'file10' done the right way.
             */
            if (g_ascii_isdigit (ac) && g_ascii_isdigit (bc))
            {
                a_num = strtoul (ap, NULL, 10); 
                b_num = strtoul (bp, NULL, 10); 

                if (a_num < b_num)
                    result = -1;
                if (a_num > b_num)
                    result = 1;
            }
1151 1152 1153 1154 1155 1156 1157 1158 1159 1160 1161 1162 1163 1164

            if (ap > a_base &&
                bp > b_base &&
                g_ascii_isdigit (*(ap -1)) &&
                g_ascii_isdigit (*(bp -1)) )
            {
                a_num = strtoul (ap-1, NULL, 10); 
                b_num = strtoul (bp-1, NULL, 10); 

                if (a_num < b_num)
                    result = -1;
                if (a_num > b_num)
                    result = 1;
            }
1165 1166
        }
    }
Stephan Arts's avatar
Stephan Arts committed
1167

1168 1169 1170 1171
    if (result == 0)
    {
        if (ac > bc)
            result = 1;
Stephan Arts's avatar
Stephan Arts committed
1172
        if (ac < bc)
1173 1174 1175 1176
            result = -1;
    }


Stephan Arts's avatar
Stephan Arts committed
1177 1178 1179 1180 1181 1182 1183 1184
    return result;
}

/**
 * cb_rstto_image_list_exif_date_compare_func:
 * @a:
 * @b:
 *
1185
 * TODO: Use EXIF data if available, not the last-modification-time.
Stephan Arts's avatar
Stephan Arts committed
1186 1187 1188 1189
 *
 * Return value: (see strcmp)
 */
static gint
Stephan Arts's avatar
Stephan Arts committed
1190
cb_rstto_image_list_exif_date_compare_func (RsttoFile *a, RsttoFile *b)
Stephan Arts's avatar
Stephan Arts committed
1191
{
1192 1193 1194 1195 1196 1197 1198 1199
    guint64 a_t = rstto_file_get_modified_time (a);
    guint64 b_t = rstto_file_get_modified_time (b);

    if (a_t < b_t)
    {
        return -1;
    }
    return 1;
1200
}
1201 1202 1203 1204 1205 1206 1207

gboolean
rstto_image_list_iter_get_sticky (
        RsttoImageListIter *iter)
{
    return iter->priv->sticky;
}
1208 1209 1210 1211 1212 1213 1214 1215 1216 1217 1218 1219 1220 1221 1222 1223 1224 1225 1226 1227

static void
cb_rstto_wrap_images_changed (
        GObject *settings,
        GParamSpec *pspec,
        gpointer user_data)
{
    GValue val_wrap_images = { 0, };

    RsttoImageList *image_list = RSTTO_IMAGE_LIST (user_data);

    g_value_init (&val_wrap_images, G_TYPE_BOOLEAN);

    g_object_get_property (
            settings,
            "wrap-images",
            &val_wrap_images);

    image_list->priv->wrap_images = g_value_get_boolean (&val_wrap_images);
}
Stephan Arts's avatar
Stephan Arts committed
1228 1229 1230 1231 1232 1233 1234 1235 1236 1237 1238 1239 1240 1241 1242 1243 1244 1245 1246 1247 1248 1249 1250 1251 1252 1253 1254 1255 1256 1257 1258 1259 1260 1261 1262 1263 1264 1265 1266 1267 1268 1269 1270 1271 1272 1273 1274 1275 1276 1277 1278 1279 1280 1281 1282 1283 1284 1285 1286 1287 1288 1289 1290 1291 1292 1293 1294 1295 1296 1297 1298 1299 1300