libmmanager-gtk/mm-gtk-hit-store.c

   1 /* MManager - a Desktop wide manager for multimedia applications.
   2  *
   3  * Copyright (C) 2008 Cosimo Cecchi <cosimoc@gnome.org>
   4  *
   5  * This library is free software; you can redistribute it and/or
   6  * modify it under the terms of the GNU Lesser General Public
   7  * License as published by the Free Software Foundation; either
   8  * version 2 of the License, or (at your option) any later version.
   9  *
  10  * This library is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13  * Lesser General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU Lesser General Public
  16  * License along with this library; if not, write to the
  17  * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  18  * Boston, MA 02111-1307, USA.
  19  */
  20
  21 #include "mm-gtk-hit-store.h"
  22
  23 #include "libmmanager/mm-attribute-manager.h"
  24 #include "libmmanager/mm-attribute-base-manager.h"
  25 #include "libmmanager/mm-attribute.h"
  26 #include "libmmanager/mm-hit.h"
  27 #include "libmmanager/mm-hit-collection.h"
  28
  29 #include <gtk/gtk.h>
  30 #include <glib.h>
  31 #include <glib-object.h>
  32
  33 /* our parent's model iface */
  34 static GtkTreeModelIface parent_iface = { 0, };
  35 static void mm_gtk_hit_store_tree_model_iface_init (GtkTreeModelIface *iface);
  36
  37 G_DEFINE_TYPE_EXTENDED (MMGtkHitStore, mm_gtk_hit_store,
  38                         GTK_TYPE_LIST_STORE, 0,
  39                         G_IMPLEMENT_INTERFACE (GTK_TYPE_TREE_MODEL,
  40                                                mm_gtk_hit_store_tree_model_iface_init));
  41
  42 static void
  43 mm_gtk_hit_store_class_init (MMGtkHitStoreClass *klass)
  44 {
  45   /* do nothing */
  46 }
  47
  48 static void
  49 mm_gtk_hit_store_init (MMGtkHitStore *self)
  50 {
  51   GType types [] = { MM_TYPE_HIT };
  52
  53   gtk_tree_store_set_column_types (GTK_TREE_STORE (self), 1, types);
  54 }
  55
  56 static void
  57 populate_from_hit_collection (MMGtkHitStore *self,
  58                               MMHitCollection *collection)
  59 {
  60   MMHit *hit;
  61   GtkTreeIter iter;
  62
  63   /* just append the hit collection to the model */
  64   while ((hit = mm_hit_collection_get_next_hit (collection)) != NULL) {
  65     gtk_list_store_append (GTK_LIST_STORE (self), &iter);
  66     gtk_list_store_set (GTK_LIST_STORE (self), &iter,
  67                         MM_GTK_HIT_STORE_HIT_COL, hit, -1);
  68   }
  69 }
  70
  71 /* interface implementation */
  72
  73 static GType
  74 impl_get_column_type (GtkTreeModel *self, gint column)
  75 {
  76   GType types [] = {
  77     MM_TYPE_HIT,
  78     G_TYPE_FILE,
  79     G_TYPE_STRING,
  80     G_TYPE_ICON
  81   };
  82
  83   g_return_val_if_fail (MM_GTK_IS_HIT_STORE (self), G_TYPE_INVALID);
  84   g_return_val_if_fail (column >= 0 && column < MM_GTK_HIT_STORE_N_COL, G_TYPE_INVALID);
  85
  86   return types[column];
  87 }
  88
  89 static int
  90 impl_get_n_columns (GtkTreeModel *self)
  91 {
  92   g_return_val_if_fail (MM_GTK_IS_HIT_STORE (self), 0);
  93
  94   /* we have four columns:
  95    * - hit
  96    * - gfile
  97    * - name
  98    * - gicon
  99    */
 100   return MM_GTK_HIT_STORE_N_COL;
 101 }
 102
 103 static MMHit *
 104 get_hit (GtkTreeModel *self, GtkTreeIter *iter)
 105 {
 106   GValue val = { 0, };
 107   MMHit *hit;
 108
 109   /* validate our parameters */
 110   g_return_val_if_fail (MM_GTK_IS_HIT_STORE (self), NULL);
 111   g_return_val_if_fail (iter != NULL, NULL);
 112
 113   /* retreive the object using our parent's interface */
 114   parent_iface.get_value (self, iter, MM_GTK_HIT_STORE_HIT_COL, &val);
 115
 116   hit = MM_HIT (g_value_dup_object (&val));
 117   g_value_unset (&val);
 118
 119   return hit;
 120 }
 121
 122 static GFile *
 123 get_gfile_from_hit (MMHit *hit)
 124 {
 125   GFile *file;
 126   GHashTable *pairs;
 127   GValue *uri_val;
 128   MMAttribute *uri_attr;
 129   const char *uri;
 130
 131
 132   pairs = mm_hit_get_values (hit,
 133                              MM_ATTRIBUTE_BASE_URI);
 134   uri_attr = mm_attribute_manager_lookup_attribute (mm_attribute_base_manager_get (),
 135                                                     MM_ATTRIBUTE_BASE_URI);
 136   uri_val = g_hash_table_lookup (pairs, uri_attr);
 137   uri = g_value_get_string (uri_val);
 138   file = g_file_new_for_uri (uri);
 139
 140   g_hash_table_destroy (pairs);
 141
 142   return file;
 143 }
 144
 145 static void
 146 value_set_gicon_from_hit (GValue *val, MMHit *hit)
 147 {
 148   /* TODO: maybe we should support custom icons in some way... */
 149   GFile *file;
 150   GFileInfo *info;
 151
 152   file = get_gfile_from_hit (hit);
 153   info = g_file_query_info (file, G_FILE_ATTRIBUTE_STANDARD_ICON,
 154                             0, NULL, NULL);
 155   if (info) {
 156     GIcon *icon;
 157
 158     icon = g_file_info_get_icon (info);
 159     g_value_take_object (val, icon);
 160     g_object_unref (info);
 161   }
 162
 163   g_object_unref (file);
 164 }
 165
 166 static void
 167 value_set_name_from_hit (GValue *val, MMHit *hit)
 168 {
 169   GHashTable *pairs;
 170   MMAttribute *name_attr;
 171   GValue *name_val;
 172
 173   pairs = mm_hit_get_values (hit,
 174                              MM_ATTRIBUTE_BASE_NAME);
 175   name_attr = mm_attribute_manager_lookup_attribute (mm_attribute_base_manager_get (),
 176                                                      MM_ATTRIBUTE_BASE_NAME);
 177   name_val = g_hash_table_lookup (pairs, name_attr);
 178
 179   if (name_val) {
 180     g_value_copy (name_val, val);
 181   } else {
 182     /* fallback */
 183     GFile *file;
 184     GFileInfo *info;
 185
 186     file = get_gfile_from_hit (hit);
 187     info = g_file_query_info (file, G_FILE_ATTRIBUTE_STANDARD_DISPLAY_NAME,
 188                               0, NULL, NULL);
 189     if (info) {
 190       g_value_set_string (val, g_file_info_get_display_name (info));
 191       g_object_unref (info);
 192     }
 193     g_object_unref (file);
 194   }
 195
 196   g_hash_table_destroy (pairs);
 197 }
 198
 199 static void
 200 value_set_gfile_from_hit (GValue *val, MMHit *hit)
 201 {
 202   GFile *file;
 203
 204   file = get_gfile_from_hit (hit);
 205   g_value_take_object (val, file);
 206 }
 207
 208 static void
 209 impl_get_value (GtkTreeModel *self, GtkTreeIter *iter,
 210                 gint column, GValue *value)
 211 {
 212   MMHit *hit;
 213
 214   g_return_if_fail (MM_GTK_IS_HIT_STORE (self));
 215   g_return_if_fail (iter != NULL);
 216   g_return_if_fail (column >= 0 && column < MM_GTK_HIT_STORE_N_COL);
 217   g_return_if_fail (value != NULL);
 218
 219   hit = get_hit (self, iter);
 220   value = g_value_init (value, impl_get_column_type (self, column));
 221
 222   switch (column) {
 223     case MM_GTK_HIT_STORE_HIT_COL:
 224       g_value_set_object (value, hit);
 225       break;
 226     case MM_GTK_HIT_STORE_FILE_COL:
 227       value_set_gfile_from_hit (value, hit);
 228       break;
 229     case MM_GTK_HIT_STORE_NAME_COL:
 230       value_set_name_from_hit (value, hit);
 231       break;
 232     case MM_GTK_HIT_STORE_ICON_COL:
 233       value_set_gicon_from_hit (value, hit);
 234       break;
 235     default:
 236       g_assert_not_reached ();
 237       break;
 238   }
 239
 240   g_object_unref (hit);
 241 }
 242
 243 static void
 244 mm_gtk_hit_store_tree_model_iface_init (GtkTreeModelIface *iface)
 245 {
 246   /* save a copy of our parent's iface */
 247   parent_iface = *iface;
 248
 249   /* override the iface methods */
 250   iface->get_n_columns = impl_get_n_columns;
 251   iface->get_column_type = impl_get_column_type;
 252   iface->get_value = impl_get_value;
 253 }
 254
 255 /* public methods */
 256
 257 MMGtkHitStore*
 258 mm_gtk_hit_store_new (MMHitCollection *hit_collection)
 259 {
 260   MMGtkHitStore * self;
 261
 262   self = MM_GTK_HIT_STORE (g_object_new (MM_GTK_TYPE_HIT_STORE, NULL));
 263   populate_from_hit_collection (self, hit_collection);
 264
 265   return self;
 266 }