diff options
author | Alan Fitton <ajf@eth0.org.uk> | 2012-01-04 10:40:44 +0000 |
---|---|---|
committer | Alan Fitton <ajf@eth0.org.uk> | 2012-01-04 10:40:44 +0000 |
commit | a007678f344943d6ea86392723004a3a72320d9e (patch) | |
tree | 5d92b359914fc32906cd81c49dd86eacb0bd818d /src/trg-files-model-common.c | |
parent | 950b833d5c7c63fddd32da986a83a9eb7215c9f4 (diff) |
make the torrent add tree view and the files tree view fairly consistent, sharing code where possible. this allows changing priority and enabled using a directory. expand/collapse all buttons. the one taken from Transmission for torrent add dialog now has menus and the mime icons. may be bugs here, I know there's one with directories appearing as mixed and not updating upwards properly, but it's a start.
Diffstat (limited to 'src/trg-files-model-common.c')
-rw-r--r-- | src/trg-files-model-common.c | 192 |
1 files changed, 192 insertions, 0 deletions
diff --git a/src/trg-files-model-common.c b/src/trg-files-model-common.c new file mode 100644 index 0000000..5cf16a3 --- /dev/null +++ b/src/trg-files-model-common.c @@ -0,0 +1,192 @@ +/* + * transmission-remote-gtk - A GTK RPC client to Transmission + * Copyright (C) 2011 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 <stdlib.h> + +#include <glib.h> +#include <glib/gi18n.h> +#include <gtk/gtk.h> + +#include "protocol-constants.h" +#include "trg-files-model-common.h" + +struct SubtreeForeachData { + gint column; + gint new_value; + GtkTreePath *path; +}; + +static void set_wanted_foreachfunc(GtkTreeModel * model, + GtkTreePath * path G_GNUC_UNUSED, GtkTreeIter * iter, gpointer data) { + struct SubtreeForeachData* args = (struct SubtreeForeachData*) data; + + gtk_tree_store_set(GTK_TREE_STORE(model), iter, args->column, + args->new_value, -1); + + trg_files_tree_model_setSubtree(model, path, iter, args->column, + args->new_value); +} + +static void set_priority_foreachfunc(GtkTreeModel * model, GtkTreePath * path, + GtkTreeIter * iter, gpointer data) { + struct SubtreeForeachData* args = (struct SubtreeForeachData*) data; + GValue value = { 0 }; + + g_value_init(&value, G_TYPE_INT); + g_value_set_int(&value, args->new_value); + + gtk_tree_store_set_value(GTK_TREE_STORE(model), iter, args->column, &value); + + trg_files_tree_model_setSubtree(model, path, iter, args->column, + args->new_value); +} + +void trg_files_model_set_wanted(GtkTreeView *tv, gint column, + gint new_value) { + struct SubtreeForeachData args; + GtkTreeSelection *selection = gtk_tree_view_get_selection(tv); + + args.column = column; + args.new_value = new_value; + + gtk_tree_selection_selected_foreach(selection, set_wanted_foreachfunc, + &args); +} + +void trg_files_tree_model_set_priority(GtkTreeView *tv, gint column, + gint new_value) { + struct SubtreeForeachData args; + GtkTreeSelection *selection = gtk_tree_view_get_selection(tv); + + args.column = column; + args.new_value = new_value; + + gtk_tree_selection_selected_foreach(selection, set_priority_foreachfunc, + &args); + +} + +static gboolean setSubtreeForeach(GtkTreeModel * model, GtkTreePath * path, + GtkTreeIter * iter, gpointer gdata) { + struct SubtreeForeachData *data = gdata; + + if (!gtk_tree_path_compare(path, data->path) + || gtk_tree_path_is_descendant(path, data->path)) { + GValue value = { 0 }; + + g_value_init(&value, G_TYPE_INT); + g_value_set_int(&value, data->new_value); + + gtk_tree_store_set_value(GTK_TREE_STORE(model), iter, data->column, + &value); + } + + return FALSE; /* keep walking */ +} + +void trg_files_tree_model_propogateChangeUp(GtkTreeModel * model, + GtkTreeIter * iter, gint column, gint new_value) { + GtkTreeIter back_iter = *iter; + gint result; + + while (1) { + GtkTreeIter tmp_iter; + gint n_children, i; + + if (!gtk_tree_model_iter_parent(model, &tmp_iter, &back_iter)) + break; + + n_children = gtk_tree_model_iter_n_children(model, &tmp_iter); + + for (i = 0; i < n_children; i++) { + GtkTreeIter child; + gint current_value; + + if (!gtk_tree_model_iter_nth_child(model, &child, &tmp_iter, i)) + continue; + + gtk_tree_model_get(model, &child, column, ¤t_value, -1); + if (current_value != new_value) { + result = TR_PRI_MIXED; + break; + } + } + + gtk_tree_store_set(GTK_TREE_STORE(model), &tmp_iter, column, result, + -1); + + back_iter = tmp_iter; + } +} + +void trg_files_tree_model_setSubtree(GtkTreeModel * model, GtkTreePath * path, + GtkTreeIter * iter, gint column, gint new_value) { + GtkTreeIter back_iter = *iter; + + if (gtk_tree_model_iter_has_child(model, iter)) { + struct SubtreeForeachData tmp; + GtkTreeIter parent; + + tmp.column = column; + tmp.new_value = new_value; + tmp.path = path; + + gtk_tree_model_foreach(model, setSubtreeForeach, &tmp); + gtk_tree_model_iter_parent(model, &parent, iter); + } else { + gtk_tree_store_set(GTK_TREE_STORE(model), &back_iter, column, new_value, + -1); + } + + trg_files_tree_model_propogateChangeUp(model, iter, column, new_value); +} + +struct UpdateParentsSizeForeachData { + gint size_column; + GtkTreeIter *descendent_iter; +}; + +static gboolean trg_files_model_update_parents_size_foreach(GtkTreeModel *model, + GtkTreePath *path, GtkTreeIter *iter, gpointer data) { + struct UpdateParentsSizeForeachData *args = (struct UpdateParentsSizeForeachData*)data; + + GtkTreePath *descendentPath = gtk_tree_model_get_path(model, + args->descendent_iter); + + if (gtk_tree_path_is_ancestor(path, descendentPath) + && gtk_tree_model_get_iter(model, args->descendent_iter, descendentPath)) { + gint64 size, oldSize; + gtk_tree_model_get(model, args->descendent_iter, args->size_column, &size, -1); + gtk_tree_model_get(model, iter, args->size_column, &oldSize, -1); + gtk_tree_store_set(GTK_TREE_STORE(model), iter, args->size_column, + size + oldSize, -1); + } + + gtk_tree_path_free(descendentPath); + + return FALSE; +} + +void trg_files_model_update_parents(GtkTreeModel *model, GtkTreeIter *iter, gint size_column) { + struct UpdateParentsSizeForeachData args; + args.descendent_iter = iter; + args.size_column = size_column; + gtk_tree_model_foreach(model, + trg_files_model_update_parents_size_foreach, &args); +} |