root/trunk/libtinymailui-gtk/tny-gtk-folder-store-tree-model-iterator.c

/* libtinymail - The Tiny Mail base library
 * Copyright (C) 2006-2007 Philip Van Hoof <pvanhoof@gnome.org>
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with self library; if not, write to the
 * Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
 * Boston, MA 02110-1301, USA.
 */

#include <config.h>

#include <tny-gtk-folder-store-tree-model.h>

static GObjectClass *parent_class = NULL;

#include "tny-gtk-folder-store-tree-model-iterator-priv.h"

GType _tny_gtk_folder_store_tree_model_iterator_get_type (void);


void 
_tny_gtk_folder_store_tree_model_iterator_set_model (TnyGtkFolderStoreTreeModelIterator *self, TnyGtkFolderStoreTreeModel *model)
{
        if (self->model)
                g_object_unref (self->model);

        self->model = g_object_ref (model);
        self->current = model->first;

        return;
}



TnyGtkFolderStoreTreeModelIterator*
_tny_gtk_folder_store_tree_model_iterator_new (TnyGtkFolderStoreTreeModel *model)
{
        TnyGtkFolderStoreTreeModelIterator *self = g_object_new (TNY_TYPE_GTK_FOLDER_STORE_TREE_MODEL_ITERATOR, NULL);

        _tny_gtk_folder_store_tree_model_iterator_set_model (self, model);

        return self;
}

static void
tny_gtk_folder_store_tree_model_iterator_instance_init (GTypeInstance *instance, gpointer g_class)
{
        TnyGtkFolderStoreTreeModelIterator *self = (TnyGtkFolderStoreTreeModelIterator *)instance;

        self->model = NULL;
        self->current = NULL;

        return;
}

static void
tny_gtk_folder_store_tree_model_iterator_finalize (GObject *object)
{
        TnyGtkFolderStoreTreeModelIterator *self = (TnyGtkFolderStoreTreeModelIterator *) object;

        if (self->model)
                g_object_unref (self->model);

        (*parent_class->finalize) (object);

        return;
}


static void
tny_gtk_folder_store_tree_model_iterator_next (TnyIterator *self)
{
        TnyGtkFolderStoreTreeModelIterator *me = (TnyGtkFolderStoreTreeModelIterator*) self;

        if (G_UNLIKELY (!me || !me->current || !me->model))
                return;

        /* Move the iterator to the next node */

        g_mutex_lock (me->model->iterator_lock);
        me->current = g_list_next (me->current);
        g_mutex_unlock (me->model->iterator_lock);

        return;
}

static void 
tny_gtk_folder_store_tree_model_iterator_prev (TnyIterator *self)
{
        TnyGtkFolderStoreTreeModelIterator *me = (TnyGtkFolderStoreTreeModelIterator*) self;

        if (G_UNLIKELY (!me || !me->current || !me->model))
                return;

        /* Move the iterator to the previous node */

        g_mutex_lock (me->model->iterator_lock);
        me->current = g_list_previous (me->current);
        g_mutex_unlock (me->model->iterator_lock);

        return;
}


static gboolean 
tny_gtk_folder_store_tree_model_iterator_is_done (TnyIterator *self)
{
        TnyGtkFolderStoreTreeModelIterator *me = (TnyGtkFolderStoreTreeModelIterator*) self;
        
        if (G_UNLIKELY (!me || !me->model))
                return TRUE;

        return me->current == NULL;
}



static void 
tny_gtk_folder_store_tree_model_iterator_first (TnyIterator *self)
{
        TnyGtkFolderStoreTreeModelIterator *me = (TnyGtkFolderStoreTreeModelIterator*) self;

        if (G_UNLIKELY (!me || !me->current || !me->model))
                return;

        /* Move the iterator to the first node. We know that model always 
           keeps a reference to the first node, there's nothing wrong with 
           using that one. */

        g_mutex_lock (me->model->iterator_lock);
        me->current = me->model->first;
        g_mutex_unlock (me->model->iterator_lock);

        return;
}


static void 
tny_gtk_folder_store_tree_model_iterator_nth (TnyIterator *self, guint nth)
{
        TnyGtkFolderStoreTreeModelIterator *me = (TnyGtkFolderStoreTreeModelIterator*) self;

        if (G_UNLIKELY (!me || !me->current || !me->model))
                return;

        /* Move the iterator to the nth node. We'll count from zero,
           so we start with the first node of which we know the model
           stored a reference. */

        g_mutex_lock (me->model->iterator_lock);
        me->current = g_list_nth (me->model->first, nth);
        g_mutex_unlock (me->model->iterator_lock);

        return;
}


static GObject* 
tny_gtk_folder_store_tree_model_iterator_get_current (TnyIterator *self)
{
        TnyGtkFolderStoreTreeModelIterator *me = (TnyGtkFolderStoreTreeModelIterator*) self;
        gpointer ptr;

        if (G_UNLIKELY (!me || !me->model))
                return NULL;

        /* Give the data of the current node */

        g_mutex_lock (me->model->iterator_lock);
        ptr = (G_UNLIKELY (me->current)) ? me->current->data : NULL;
        g_mutex_unlock (me->model->iterator_lock);

        if (ptr)
                g_object_ref (G_OBJECT(ptr));
        
        return G_OBJECT(ptr);
}


static TnyList* 
tny_gtk_folder_store_tree_model_iterator_get_list (TnyIterator *self)
{
        TnyGtkFolderStoreTreeModelIterator *me = (TnyGtkFolderStoreTreeModelIterator*) self;

        /* Return the list */

        if (G_UNLIKELY (!me || !me->model))
                return NULL;
    
        g_object_ref (G_OBJECT (me->model));

        return TNY_LIST (me->model);
}

static void
tny_iterator_init (TnyIteratorIface *klass)
{
        klass->next= tny_gtk_folder_store_tree_model_iterator_next;
        klass->prev= tny_gtk_folder_store_tree_model_iterator_prev;
        klass->first= tny_gtk_folder_store_tree_model_iterator_first;
        klass->nth= tny_gtk_folder_store_tree_model_iterator_nth;
        klass->get_current= tny_gtk_folder_store_tree_model_iterator_get_current;
        klass->get_list= tny_gtk_folder_store_tree_model_iterator_get_list;
        klass->is_done  = tny_gtk_folder_store_tree_model_iterator_is_done;
        
        return;
}

static void 
tny_gtk_folder_store_tree_model_iterator_class_init (TnyGtkFolderStoreTreeModelIteratorClass *klass)
{
        GObjectClass *object_class;

        parent_class = g_type_class_peek_parent (klass);
        object_class = (GObjectClass*) klass;

        object_class->finalize = tny_gtk_folder_store_tree_model_iterator_finalize;

        return;
}

static gpointer
_tny_gtk_folder_store_tree_model_iterator_register_type (gpointer notused)
{
        GType type = 0;

        static const GTypeInfo info = 
                {
                  sizeof (TnyGtkFolderStoreTreeModelIteratorClass),
                  NULL,   /* base_init */
                  NULL,   /* base_finalize */
                  (GClassInitFunc) tny_gtk_folder_store_tree_model_iterator_class_init,   /* class_init */
                  NULL,   /* class_finalize */
                  NULL,   /* class_data */
                  sizeof (TnyGtkFolderStoreTreeModelIterator),
                  0,      /* n_preallocs */
                  tny_gtk_folder_store_tree_model_iterator_instance_init,    /* instance_init */
                  NULL
                };

        static const GInterfaceInfo tny_iterator_info = 
                {
                  (GInterfaceInitFunc) tny_iterator_init, /* interface_init */
                  NULL,         /* interface_finalize */
                  NULL          /* interface_data */
                };

        type = g_type_register_static (G_TYPE_OBJECT,
                                       "TnyGtkFolderStoreTreeModelIterator",
                                       &info, 0);

        g_type_add_interface_static (type, TNY_TYPE_ITERATOR, 
                                     &tny_iterator_info);

        return GUINT_TO_POINTER (type);
}

GType 
_tny_gtk_folder_store_tree_model_iterator_get_type (void)
{
        static GOnce once = G_ONCE_INIT;
        g_once (&once, _tny_gtk_folder_store_tree_model_iterator_register_type, NULL);
        return GPOINTER_TO_UINT (once.retval);
}