diff options
Diffstat (limited to 'src/trg-files-model.c')
| -rw-r--r-- | src/trg-files-model.c | 177 |
1 files changed, 177 insertions, 0 deletions
diff --git a/src/trg-files-model.c b/src/trg-files-model.c new file mode 100644 index 0000000..16f58f9 --- /dev/null +++ b/src/trg-files-model.c @@ -0,0 +1,177 @@ +/* + * transmission-remote-gtk - Transmission RPC client for GTK + * Copyright (C) 2010 Alan Fitton + + * 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. + + * This program 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 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., + * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. + */ + +#include <gtk/gtk.h> +#include <json-glib/json-glib.h> + +#include "trg-files-model.h" +#include "torrent.h" +#include "tfile.h" +#include "util.h" + +#include "trg-files-model.h" + +G_DEFINE_TYPE(TrgFilesModel, trg_files_model, GTK_TYPE_LIST_STORE) +#define TRG_FILES_MODEL_GET_PRIVATE(o) \ + (G_TYPE_INSTANCE_GET_PRIVATE ((o), TRG_TYPE_FILES_MODEL, TrgFilesModelPrivate)) +typedef struct _TrgFilesModelPrivate TrgFilesModelPrivate; + +struct _TrgFilesModelPrivate { + gint torrentId; + JsonArray *files; + JsonArray *wanted; + JsonArray *priorities; + gint64 updateBarrier; + gint64 currentSerial; +}; + +static void trg_files_model_iter_new(TrgFilesModel * model, + GtkTreeIter * iter, JsonObject * file, + int id) +{ + char size[32]; + + gtk_list_store_append(GTK_LIST_STORE(model), iter); + + trg_strlsize(size, file_get_length(file)); + gtk_list_store_set(GTK_LIST_STORE(model), iter, + FILESCOL_NAME, file_get_name(file), + FILESCOL_SIZE, size, FILESCOL_ID, id, -1); +} + +void trg_files_model_set_update_barrier(TrgFilesModel * model, + gint64 serial) +{ + TrgFilesModelPrivate *priv = TRG_FILES_MODEL_GET_PRIVATE(model); + priv->updateBarrier = serial; +} + +static void +trg_files_model_iter_update(TrgFilesModel * model, + GtkTreeIter * filesIter, JsonObject * file, + JsonArray * wantedArray, + JsonArray * prioritiesArray, int id) +{ + TrgFilesModelPrivate *priv = TRG_FILES_MODEL_GET_PRIVATE(model); + + gboolean wanted = json_node_get_int(json_array_get_element + (wantedArray, id)) == 1; + gint64 priority = + json_node_get_int(json_array_get_element(prioritiesArray, id)); + gdouble progress = file_get_progress(file); + + gtk_list_store_set(GTK_LIST_STORE(model), filesIter, + FILESCOL_PROGRESS, progress, -1); + + if (priv->updateBarrier < 0 + || priv->currentSerial >= priv->updateBarrier) { + gtk_list_store_set(GTK_LIST_STORE(model), filesIter, + FILESCOL_ICON, + wanted ? GTK_STOCK_FILE : + GTK_STOCK_STOP, FILESCOL_WANTED, wanted, + FILESCOL_PRIORITY, priority, -1); + } +} + +static void trg_files_model_class_init(TrgFilesModelClass * klass) +{ + g_type_class_add_private(klass, sizeof(TrgFilesModelPrivate)); +} + +static void trg_files_model_init(TrgFilesModel * self) +{ + TrgFilesModelPrivate *priv = TRG_FILES_MODEL_GET_PRIVATE(self); + GType column_types[FILESCOL_COLUMNS]; + + priv->updateBarrier = -1; + + column_types[FILESCOL_ICON] = G_TYPE_STRING; + column_types[FILESCOL_NAME] = G_TYPE_STRING; + column_types[FILESCOL_SIZE] = G_TYPE_STRING; + column_types[FILESCOL_PROGRESS] = G_TYPE_DOUBLE; + column_types[FILESCOL_ID] = G_TYPE_INT; + column_types[FILESCOL_WANTED] = G_TYPE_BOOLEAN; + column_types[FILESCOL_PRIORITY] = G_TYPE_INT; + + gtk_list_store_set_column_types(GTK_LIST_STORE(self), + FILESCOL_COLUMNS, column_types); +} + +gboolean +trg_files_model_update_foreach(GtkListStore * model, + GtkTreePath * path, GtkTreeIter * iter, + gpointer data) +{ + TrgFilesModelPrivate *priv = TRG_FILES_MODEL_GET_PRIVATE(model); + JsonObject *file; + gint id; + + gtk_tree_model_get(GTK_TREE_MODEL(model), iter, FILESCOL_ID, &id, -1); + + file = json_node_get_object(json_array_get_element(priv->files, id)); + trg_files_model_iter_update(TRG_FILES_MODEL(model), iter, file, + priv->wanted, priv->priorities, id); + + return FALSE; +} + +void +trg_files_model_update(TrgFilesModel * model, gint64 updateSerial, + JsonObject * t, gboolean first) +{ + TrgFilesModelPrivate *priv = TRG_FILES_MODEL_GET_PRIVATE(model); + int j; + + if (first == TRUE) + gtk_list_store_clear(GTK_LIST_STORE(model)); + + priv->torrentId = torrent_get_id(t); + priv->priorities = torrent_get_priorities(t); + priv->wanted = torrent_get_wanted(t); + priv->files = torrent_get_files(t); + priv->currentSerial = updateSerial; + + if (first == TRUE) { + for (j = 0; j < json_array_get_length(priv->files); j++) { + JsonObject *file = + json_node_get_object(json_array_get_element + (priv->files, j)); + GtkTreeIter filesIter; + trg_files_model_iter_new(model, &filesIter, file, j); + trg_files_model_iter_update(model, &filesIter, + file, priv->wanted, + priv->priorities, j); + } + } else { + gtk_tree_model_foreach(GTK_TREE_MODEL(model), + (GtkTreeModelForeachFunc) + trg_files_model_update_foreach, NULL); + } +} + +gint trg_files_model_get_torrent_id(TrgFilesModel * model) +{ + TrgFilesModelPrivate *priv = TRG_FILES_MODEL_GET_PRIVATE(model); + return priv->torrentId; +} + +TrgFilesModel *trg_files_model_new(void) +{ + return g_object_new(TRG_TYPE_FILES_MODEL, NULL); +} |