source: trunk/gtk/file-list.c @ 11804

Last change on this file since 11804 was 11804, checked in by jordan, 11 years ago

(trunk gtk) #3876 "Give filesize its own column in the Files tree" -- fix minor memory leak introduced by r11642

r11642 allocated a pango font description with pango_font_description_copy() but never freed it.

  • Property svn:keywords set to Date Rev Author Id
File size: 35.3 KB
Line 
1/*
2 * This file Copyright (C) Mnemosyne LLC
3 *
4 * This file is licensed by the GPL version 2. Works owned by the
5 * Transmission project are granted a special exemption to clause 2(b)
6 * so that the bulk of its code can remain under the MIT license.
7 * This exemption does not extend to derived works not owned by
8 * the Transmission project.
9 *
10 * $Id: file-list.c 11804 2011-02-01 17:42:50Z jordan $
11 */
12
13#include <stddef.h>
14#include <stdio.h>
15#include <stdlib.h>
16#include <string.h>
17#include <glib/gi18n.h>
18#include <gtk/gtk.h>
19
20#include <libtransmission/transmission.h>
21
22#include "file-list.h"
23#include "hig.h"
24#include "icons.h"
25#include "tr-prefs.h"
26
27#define TR_DOWNLOAD_KEY  "tr-download-key"
28#define TR_COLUMN_ID_KEY "tr-model-column-id-key"
29#define TR_PRIORITY_KEY  "tr-priority-key"
30
31enum
32{
33    /* these two fields could be any number at all so long as they're not
34     * TR_PRI_LOW, TR_PRI_NORMAL, TR_PRI_HIGH, TRUE, or FALSE */
35    NOT_SET = 1000,
36    MIXED = 1001
37};
38
39enum
40{
41    FC_ICON,
42    FC_LABEL,
43    FC_PROG,
44    FC_INDEX,
45    FC_SIZE,
46    FC_SIZE_STR,
47    FC_HAVE,
48    FC_PRIORITY,
49    FC_ENABLED,
50    N_FILE_COLS
51};
52
53typedef struct
54{
55    TrCore        * core;
56    GtkWidget     * top;
57    GtkWidget     * view;
58    GtkTreeModel  * model; /* same object as store, but recast */
59    GtkTreeStore  * store; /* same object as model, but recast */
60    int             torrentId;
61    guint           timeout_tag;
62}
63FileData;
64
65static void
66clearData( FileData * data )
67{
68    data->torrentId = -1;
69
70    if( data->timeout_tag ) {
71        g_source_remove( data->timeout_tag );
72        data->timeout_tag = 0;
73    }
74}
75
76static void
77freeData( gpointer data )
78{
79    clearData( data );
80    g_free( data );
81}
82
83/***
84****
85***/
86
87struct RefreshData
88{
89    int sort_column_id;
90    gboolean resort_needed;
91
92    tr_file_stat  * refresh_file_stat;
93    tr_torrent * tor;
94
95    FileData * file_data;
96};
97
98static gboolean
99refreshFilesForeach( GtkTreeModel * model,
100                     GtkTreePath  * path UNUSED,
101                     GtkTreeIter  * iter,
102                     gpointer       gdata )
103{
104    struct RefreshData * refresh_data = gdata;
105    FileData * data = refresh_data->file_data;
106    unsigned int index;
107    uint64_t size;
108    uint64_t old_have;
109    int old_prog;
110    int old_priority;
111    int old_enabled;
112    const gboolean is_file = !gtk_tree_model_iter_has_child( model, iter );
113
114    gtk_tree_model_get( model, iter, FC_ENABLED, &old_enabled,
115                                     FC_PRIORITY, &old_priority,
116                                     FC_INDEX, &index,
117                                     FC_HAVE, &old_have,
118                                     FC_SIZE, &size,
119                                     FC_PROG, &old_prog,
120                                     -1  );
121
122    if( is_file )
123    {
124        tr_torrent * tor = refresh_data->tor;
125        const tr_info * inf = tr_torrentInfo( tor );
126        const int enabled = !inf->files[index].dnd;
127        const int priority = inf->files[index].priority;
128        const uint64_t have = refresh_data->refresh_file_stat[index].bytesCompleted;
129        const int prog = size ? (int)((100.0*have)/size) : 1;
130
131        if( (priority!=old_priority) || (enabled!=old_enabled) || (have!=old_have) || (prog!=old_prog) )
132        {
133            /* Changing a value in the sort column can trigger a resort
134             * which breaks this foreach() call. (See #3529)
135             * As a workaround: if that's about to happen, temporarily disable
136             * sorting until we finish walking the tree. */
137            if( !refresh_data->resort_needed )
138            {
139                if(( refresh_data->resort_needed =
140                    (( refresh_data->sort_column_id==FC_PRIORITY ) && ( priority!=old_priority )) ||
141                    (( refresh_data->sort_column_id==FC_ENABLED ) && ( enabled!=old_enabled ))))
142                {
143                    refresh_data->resort_needed = TRUE;
144                    gtk_tree_sortable_set_sort_column_id( GTK_TREE_SORTABLE( data->model ),
145                                                          GTK_TREE_SORTABLE_UNSORTED_SORT_COLUMN_ID,
146                                                          GTK_SORT_ASCENDING );
147                }
148            }
149
150            gtk_tree_store_set( data->store, iter, FC_PRIORITY, priority,
151                                                   FC_ENABLED, enabled,
152                                                   FC_HAVE, have,
153                                                   FC_PROG, prog,
154                                                   -1 );
155        }
156    }
157    else
158    {
159        GtkTreeIter child;
160        uint64_t sub_size = 0;
161        uint64_t have = 0;
162        int prog;
163        int enabled = NOT_SET;
164        int priority = NOT_SET;
165
166        /* since gtk_tree_model_foreach() is depth-first, we can
167         * get the `sub' info by walking the immediate children */
168
169        if( gtk_tree_model_iter_children( model, &child, iter ) ) do
170        {
171            int child_enabled;
172            int child_priority;
173            int64_t child_have, child_size;
174
175            gtk_tree_model_get( model, &child, FC_SIZE, &child_size,
176                                               FC_HAVE, &child_have,
177                                               FC_PRIORITY, &child_priority,
178                                               FC_ENABLED, &child_enabled,
179                                               -1 );
180
181            sub_size += child_size;
182            have += child_have;
183
184            if( enabled == NOT_SET )
185                enabled = child_enabled;
186            else if( enabled != child_enabled )
187                enabled = MIXED;
188
189            if( priority == NOT_SET )
190                priority = child_priority;
191            else if( priority != child_priority )
192                priority = MIXED;
193        }
194        while( gtk_tree_model_iter_next( model, &child ) );
195
196        prog = sub_size ? (int)((100.0*have)/sub_size) : 1;
197
198        if( (size!=sub_size) || (have!=old_have)
199                             || (priority!=old_priority)
200                             || (enabled!=old_enabled)
201                             || (prog!=old_prog) )
202        {
203            char size_str[64];
204            tr_strlsize( size_str, sub_size, sizeof size_str );
205            gtk_tree_store_set( data->store, iter, FC_SIZE, sub_size,
206                                                   FC_SIZE_STR, size_str,
207                                                   FC_HAVE, have,
208                                                   FC_PRIORITY, priority,
209                                                   FC_ENABLED, enabled,
210                                                   FC_PROG, prog,
211                                                   -1 );
212        }
213    }
214
215    return FALSE; /* keep walking */
216}
217
218static void
219gtr_tree_model_foreach_postorder_subtree( GtkTreeModel            * model,
220                                          GtkTreeIter             * parent,
221                                          GtkTreeModelForeachFunc   func,
222                                          gpointer                  data )
223{
224    GtkTreeIter child;
225    if( gtk_tree_model_iter_children( model, &child, parent ) ) do
226        gtr_tree_model_foreach_postorder_subtree( model, &child, func, data );
227    while( gtk_tree_model_iter_next( model, &child ) );
228    if( parent )
229        func( model, NULL, parent, data );
230}
231
232static void
233gtr_tree_model_foreach_postorder( GtkTreeModel            * model,
234                                  GtkTreeModelForeachFunc   func,
235                                  gpointer                  data )
236{
237    GtkTreeIter iter;
238    if( gtk_tree_model_get_iter_first( model, &iter ) ) do
239        gtr_tree_model_foreach_postorder_subtree( model, &iter, func, data );
240    while( gtk_tree_model_iter_next( model, &iter ) );
241}
242
243static void
244refresh( FileData * data )
245{
246    tr_torrent * tor = NULL;
247    tr_session * session = tr_core_session( data->core );
248
249    if( session != NULL )
250        tor = tr_torrentFindFromId( session, data->torrentId );
251
252    if( tor == NULL )
253    {
254        gtr_file_list_clear( data->top );
255    }
256    else
257    {
258        GtkSortType order;
259        int sort_column_id;
260        tr_file_index_t fileCount;
261        struct RefreshData refresh_data;
262        GtkTreeSortable * sortable = GTK_TREE_SORTABLE( data->model );
263        gtk_tree_sortable_get_sort_column_id( sortable, &sort_column_id, &order );
264
265        refresh_data.sort_column_id = sort_column_id;
266        refresh_data.resort_needed = FALSE;
267        refresh_data.refresh_file_stat = tr_torrentFiles( tor, &fileCount );
268        refresh_data.tor = tr_torrentFindFromId( session, data->torrentId );
269        refresh_data.file_data = data;
270
271        gtr_tree_model_foreach_postorder( data->model, refreshFilesForeach, &refresh_data );
272
273        if( refresh_data.resort_needed )
274            gtk_tree_sortable_set_sort_column_id( sortable, sort_column_id, order );
275
276        tr_torrentFilesFree( refresh_data.refresh_file_stat, fileCount );
277    }
278}
279
280static gboolean
281refreshModel( gpointer file_data )
282{
283    refresh( file_data );
284    return TRUE;
285}
286
287/***
288****
289***/
290
291struct ActiveData
292{
293    GtkTreeSelection  * sel;
294    GArray            * array;
295};
296
297static gboolean
298getSelectedFilesForeach( GtkTreeModel * model,
299                         GtkTreePath  * path UNUSED,
300                         GtkTreeIter  * iter,
301                         gpointer       gdata )
302{
303    const gboolean is_file = !gtk_tree_model_iter_has_child( model, iter );
304
305    if( is_file )
306    {
307        struct ActiveData * data = gdata;
308
309        /* active means: if it's selected or any ancestor is selected */
310
311        gboolean is_active = gtk_tree_selection_iter_is_selected( data->sel, iter );
312
313        if( !is_active )
314        {
315            GtkTreeIter walk = *iter;
316            GtkTreeIter parent;
317            while( !is_active && gtk_tree_model_iter_parent( model, &parent, &walk ) )
318            {
319                is_active = gtk_tree_selection_iter_is_selected( data->sel, &parent );
320                walk = parent;
321            }
322        }
323
324        if( is_active )
325        {
326            unsigned int i;
327            gtk_tree_model_get( model, iter, FC_INDEX, &i, -1 );
328            g_array_append_val( data->array, i );
329        }
330    }
331
332    return FALSE; /* keep walking */
333}
334
335static GArray*
336getSelectedFilesAndDescendants( GtkTreeView * view )
337{
338    struct ActiveData data;
339
340    data.sel = gtk_tree_view_get_selection( view );
341    data.array = g_array_new( FALSE, FALSE, sizeof( tr_file_index_t ) );
342    gtk_tree_model_foreach( gtk_tree_view_get_model( view ),
343                            getSelectedFilesForeach, &data );
344    return data.array;
345}
346
347struct SubtreeForeachData
348{
349    GArray       * array;
350    GtkTreePath  * path;
351};
352
353static gboolean
354getSubtreeForeach( GtkTreeModel   * model,
355                   GtkTreePath    * path,
356                   GtkTreeIter    * iter,
357                   gpointer         gdata )
358{
359    const gboolean is_file = !gtk_tree_model_iter_has_child( model, iter );
360
361    if( is_file )
362    {
363        struct SubtreeForeachData * data = gdata;
364
365        if( !gtk_tree_path_compare( path, data->path ) || gtk_tree_path_is_descendant( path, data->path ) )
366        {
367            unsigned int i;
368            gtk_tree_model_get( model, iter, FC_INDEX, &i, -1 );
369            g_array_append_val( data->array, i );
370        }
371    }
372
373    return FALSE; /* keep walking */
374}
375
376static void
377getSubtree( GtkTreeView * view, GtkTreePath * path, GArray * indices )
378{
379    struct SubtreeForeachData tmp;
380    tmp.array = indices;
381    tmp.path = path;
382    gtk_tree_model_foreach( gtk_tree_view_get_model( view ), getSubtreeForeach, &tmp );
383}
384
385/* if `path' is a selected row, all selected rows are returned.
386 * otherwise, only the row indicated by `path' is returned.
387 * this is for toggling all the selected rows' states in a batch.
388 */
389static GArray*
390getActiveFilesForPath( GtkTreeView * view, GtkTreePath * path )
391{
392    GArray * indices;
393    GtkTreeSelection * sel = gtk_tree_view_get_selection( view );
394
395    if( gtk_tree_selection_path_is_selected( sel, path ) )
396    {
397        /* clicked in a selected row... use the current selection */
398        indices = getSelectedFilesAndDescendants( view );
399    }
400    else
401    {
402        /* clicked OUTSIDE of the selected row... just use the clicked row */
403        indices = g_array_new( FALSE, FALSE, sizeof( tr_file_index_t ) );
404        getSubtree( view, path, indices );
405    }
406
407    return indices;
408}
409
410/***
411****
412***/
413
414void
415gtr_file_list_clear( GtkWidget * w )
416{
417    gtr_file_list_set_torrent( w, -1 );
418}
419
420struct build_data
421{
422    GtkWidget    * w;
423    tr_torrent   * tor;
424    GtkTreeIter  * iter;
425    GtkTreeStore * store;
426};
427
428struct row_struct
429{
430    uint64_t    length;
431    char      * name;
432    int         index;
433};
434
435static void
436buildTree( GNode * node, gpointer gdata )
437{
438    char size_str[64];
439    GtkTreeIter child_iter;
440    struct build_data * build = gdata;
441    struct row_struct *child_data = node->data;
442    const gboolean isLeaf = node->children == NULL;
443
444    const char * mime_type = isLeaf ? gtr_get_mime_type_from_filename( child_data->name ) : DIRECTORY_MIME_TYPE;
445    GdkPixbuf * icon = gtr_get_mime_type_icon( mime_type, GTK_ICON_SIZE_MENU, build->w );
446    const tr_info * inf = tr_torrentInfo( build->tor );
447    const int priority = isLeaf ? inf->files[ child_data->index ].priority : 0;
448    const gboolean enabled = isLeaf ? !inf->files[ child_data->index ].dnd : TRUE;
449
450    tr_strlsize( size_str, child_data->length, sizeof size_str );
451
452#if GTK_CHECK_VERSION(2,10,0)
453    gtk_tree_store_insert_with_values( build->store, &child_iter, build->iter, INT_MAX,
454                                       FC_INDEX, child_data->index,
455                                       FC_LABEL, child_data->name,
456                                       FC_SIZE, child_data->length,
457                                       FC_SIZE_STR, size_str,
458                                       FC_ICON, icon,
459                                       FC_PRIORITY, priority,
460                                       FC_ENABLED, enabled,
461                                       -1 );
462#else
463    gtk_tree_store_append( build->store, &child_iter, build->iter );
464    gtk_tree_store_set( build->store, &child_iter,
465                        FC_INDEX, child_data->index,
466                        FC_LABEL, child_data->name,
467                        FC_SIZE, child_data->length,
468                        FC_SIZE_STR, size_str,
469                        FC_ICON, icon,
470                        FC_PRIORITY, priority,
471                        FC_ENABLED, enabled,
472                        -1 );
473#endif
474
475    if( !isLeaf )
476    {
477        struct build_data b = *build;
478        b.iter = &child_iter;
479        g_node_children_foreach( node, G_TRAVERSE_ALL, buildTree, &b );
480    }
481
482    g_object_unref( icon );
483
484    /* we're done with this node */
485    g_free( child_data->name );
486    g_free( child_data );
487}
488
489static GNode*
490find_child( GNode* parent, const char * name )
491{
492    GNode * child = parent->children;
493    while( child ) {
494        const struct row_struct * child_data = child->data;
495        if( ( *child_data->name == *name ) && !strcmp( child_data->name, name ) )
496            break;
497        child = child->next;
498    }
499    return child;
500}
501
502void
503gtr_file_list_set_torrent( GtkWidget * w, int torrentId )
504{
505    GtkTreeStore * store;
506    FileData * data = g_object_get_data( G_OBJECT( w ), "file-data" );
507
508    /* unset the old fields */
509    clearData( data );
510
511    /* instantiate the model */
512    store = gtk_tree_store_new ( N_FILE_COLS,
513                                 GDK_TYPE_PIXBUF,  /* icon */
514                                 G_TYPE_STRING,    /* label */
515                                 G_TYPE_INT,       /* prog [0..100] */
516                                 G_TYPE_UINT,      /* index */
517                                 G_TYPE_UINT64,    /* size */
518                                 G_TYPE_STRING,    /* size str */
519                                 G_TYPE_UINT64,    /* have */
520                                 G_TYPE_INT,       /* priority */
521                                 G_TYPE_INT );     /* dl enabled */
522
523    data->store = store;
524    data->model = GTK_TREE_MODEL( store );
525    data->torrentId = torrentId;
526
527    /* populate the model */
528    if( torrentId > 0 )
529    {
530        tr_session * session = tr_core_session( data->core );
531        tr_torrent * tor = tr_torrentFindFromId( session, torrentId );
532        if( tor != NULL )
533        {
534            tr_file_index_t i;
535            const tr_info * inf = tr_torrentInfo( tor );
536            struct row_struct * root_data;
537            GNode * root;
538            struct build_data build;
539
540            /* build a GNode tree of the files */
541            root_data = g_new0( struct row_struct, 1 );
542            root_data->name = g_strdup( inf->name );
543            root_data->index = -1;
544            root_data->length = 0;
545            root = g_node_new( root_data );
546            for( i=0; i<inf->fileCount; ++i ) {
547                int j;
548                GNode * parent = root;
549                const tr_file * file = &inf->files[i];
550                char ** tokens = g_strsplit( file->name, G_DIR_SEPARATOR_S, 0 );
551                for( j=0; tokens[j]; ++j ) {
552                    const gboolean isLeaf = tokens[j+1] == NULL;
553                    const char * name = tokens[j];
554                    GNode * node = find_child( parent, name );
555                    if( node == NULL ) {
556                        struct row_struct * row = g_new( struct row_struct, 1 );
557                        row->name = g_strdup( name );
558                        row->index = isLeaf ? (int)i : -1;
559                        row->length = isLeaf ? file->length : 0;
560                        node = g_node_new( row );
561                        g_node_append( parent, node );
562                    }
563                    parent = node;
564                }
565                g_strfreev( tokens );
566            }
567
568            /* now, add them to the model */
569            build.w = w;
570            build.tor = tor;
571            build.store = data->store;
572            build.iter = NULL;
573            g_node_children_foreach( root, G_TRAVERSE_ALL, buildTree, &build );
574
575            /* cleanup */
576            g_node_destroy( root );
577            g_free( root_data->name );
578            g_free( root_data );
579        }
580
581        refresh( data );
582        data->timeout_tag = gtr_timeout_add_seconds( SECONDARY_WINDOW_REFRESH_INTERVAL_SECONDS, refreshModel, data );
583    }
584
585    gtk_tree_view_set_model( GTK_TREE_VIEW( data->view ), data->model );
586    gtk_tree_view_expand_all( GTK_TREE_VIEW( data->view ) );
587}
588
589/***
590****
591***/
592
593static void
594renderDownload( GtkTreeViewColumn  * column UNUSED,
595                GtkCellRenderer    * renderer,
596                GtkTreeModel       * model,
597                GtkTreeIter        * iter,
598                gpointer             data   UNUSED )
599{
600    gboolean enabled;
601    gtk_tree_model_get( model, iter, FC_ENABLED, &enabled, -1 );
602    g_object_set( renderer, "inconsistent", (enabled==MIXED),
603                            "active", (enabled==TRUE),
604                            NULL );
605}
606
607static void
608renderPriority( GtkTreeViewColumn  * column UNUSED,
609                GtkCellRenderer    * renderer,
610                GtkTreeModel       * model,
611                GtkTreeIter        * iter,
612                gpointer             data   UNUSED )
613{
614    int priority;
615    const char * text;
616    gtk_tree_model_get( model, iter, FC_PRIORITY, &priority, -1 );
617    switch( priority ) {
618        case TR_PRI_HIGH:   text = _( "High" ); break;
619        case TR_PRI_NORMAL: text = _( "Normal" ); break;
620        case TR_PRI_LOW:    text = _( "Low" ); break;
621        default:            text = _( "Mixed" ); break;
622    }
623    g_object_set( renderer, "text", text, NULL );
624}
625
626/* build a filename from tr_torrentGetCurrentDir() + the model's FC_LABELs */
627static char*
628buildFilename( tr_torrent * tor, GtkTreeModel * model,
629               GtkTreePath * path, GtkTreeIter * iter )
630{
631    char * ret;
632    GtkTreeIter child;
633    GtkTreeIter parent = *iter;
634    int n = gtk_tree_path_get_depth( path );
635    char ** tokens = g_new0( char*, n + 2 );
636    tokens[0] = g_strdup( tr_torrentGetCurrentDir( tor ) );
637    do {
638        child = parent;
639        gtk_tree_model_get( model, &child, FC_LABEL, &tokens[n--], -1 );
640    } while( gtk_tree_model_iter_parent( model, &parent, &child ) );
641    ret = g_build_filenamev( tokens );
642    g_strfreev( tokens );
643    return ret;
644}
645
646static gboolean
647onRowActivated( GtkTreeView * view, GtkTreePath * path,
648                GtkTreeViewColumn * col UNUSED, gpointer gdata )
649{
650    gboolean handled = FALSE;
651    FileData * data = gdata;
652    tr_torrent * tor = tr_torrentFindFromId( tr_core_session( data->core ), data->torrentId );
653
654    if( tor != NULL )
655    {
656        GtkTreeIter iter;
657        GtkTreeModel * model = gtk_tree_view_get_model( view );
658
659        if( gtk_tree_model_get_iter( model, &iter, path ) )
660        {
661            int prog;
662            char * filename = buildFilename( tor, model, path, &iter );
663            gtk_tree_model_get( model, &iter, FC_PROG, &prog, -1 );
664
665            /* if the file's not done, walk up the directory tree until we find
666             * an ancestor that exists, and open that instead */
667            if( filename && ( prog<100 || !g_file_test( filename, G_FILE_TEST_EXISTS ) ) ) do
668            {
669                char * tmp = g_path_get_dirname( filename );
670                g_free( filename );
671                filename = tmp;
672            }
673            while( filename && *filename && !g_file_test( filename, G_FILE_TEST_EXISTS ) );
674
675            if(( handled = filename && *filename ))
676                gtr_open_file( filename );
677        }
678    }
679
680    return handled;
681}
682
683static void
684fileMenuSetDownload( GtkWidget * item, gpointer gdata )
685{
686    tr_torrent * tor;
687    FileData * data = gdata;
688
689    if(( tor = tr_torrentFindFromId( tr_core_session( data->core ), data->torrentId )))
690    {
691        const tr_bool download_flag = g_object_get_data( G_OBJECT( item ), TR_DOWNLOAD_KEY ) != NULL;
692        GArray * indices = getSelectedFilesAndDescendants( GTK_TREE_VIEW( data->view ) );
693
694        tr_torrentSetFileDLs( tor,
695                              (tr_file_index_t *) indices->data,
696                              (tr_file_index_t) indices->len,
697                              download_flag );
698
699        refresh( data );
700        g_array_free( indices, TRUE );
701    }
702}
703
704static void
705fileMenuSetPriority( GtkWidget * item, gpointer gdata )
706{
707    tr_torrent * tor;
708    FileData * data = gdata;
709
710    if(( tor = tr_torrentFindFromId( tr_core_session( data->core ), data->torrentId )))
711    {
712        const int priority = GPOINTER_TO_INT( g_object_get_data( G_OBJECT( item ), TR_PRIORITY_KEY ) );
713        GArray * indices = getSelectedFilesAndDescendants( GTK_TREE_VIEW( data->view ) );
714        tr_torrentSetFilePriorities( tor,
715                                     (tr_file_index_t *) indices->data,
716                                     (tr_file_index_t) indices->len,
717                                     priority );
718        refresh( data );
719        g_array_free( indices, TRUE );
720    }
721}
722
723static void
724fileMenuPopup( GtkWidget * w, GdkEventButton * event, gpointer filedata )
725{
726    GtkWidget * item;
727    GtkWidget * menu;
728
729    menu = gtk_menu_new( );
730    item = gtk_menu_item_new_with_label( _( "Set Priority High" ) );
731    g_object_set_data( G_OBJECT( item ), TR_PRIORITY_KEY,
732                       GINT_TO_POINTER( TR_PRI_HIGH ) );
733    g_signal_connect( item, "activate",
734                      G_CALLBACK( fileMenuSetPriority ), filedata );
735    gtk_menu_shell_append( GTK_MENU_SHELL( menu ), item );
736
737    item = gtk_menu_item_new_with_label( _( "Set Priority Normal" ) );
738    g_object_set_data( G_OBJECT( item ), TR_PRIORITY_KEY,
739                       GINT_TO_POINTER( TR_PRI_NORMAL ) );
740    g_signal_connect( item, "activate",
741                      G_CALLBACK( fileMenuSetPriority ), filedata );
742    gtk_menu_shell_append( GTK_MENU_SHELL( menu ), item );
743
744    item = gtk_menu_item_new_with_label( _( "Set Priority Low" ) );
745    g_object_set_data( G_OBJECT( item ), TR_PRIORITY_KEY,
746                       GINT_TO_POINTER( TR_PRI_LOW ) );
747    g_signal_connect( item, "activate",
748                      G_CALLBACK( fileMenuSetPriority ), filedata );
749    gtk_menu_shell_append( GTK_MENU_SHELL( menu ), item );
750
751    item = gtk_separator_menu_item_new( );
752    gtk_menu_shell_append( GTK_MENU_SHELL( menu ), item );
753
754    item = gtk_menu_item_new_with_label( _( "Download" ) );
755    g_object_set_data( G_OBJECT( item ), TR_DOWNLOAD_KEY,
756                       GINT_TO_POINTER( TRUE ) );
757    g_signal_connect( item, "activate",
758                      G_CALLBACK( fileMenuSetDownload ), filedata );
759    gtk_menu_shell_append( GTK_MENU_SHELL( menu ), item );
760
761    item = gtk_menu_item_new_with_label( _( "Do Not Download" ) );
762    g_object_set_data( G_OBJECT( item ), TR_DOWNLOAD_KEY,
763                       GINT_TO_POINTER( FALSE ) );
764    g_signal_connect( item, "activate",
765                      G_CALLBACK( fileMenuSetDownload ), filedata );
766    gtk_menu_shell_append( GTK_MENU_SHELL( menu ), item );
767
768    gtk_widget_show_all( menu );
769    gtk_menu_attach_to_widget( GTK_MENU( menu ), w, NULL );
770    gtk_menu_popup( GTK_MENU( menu ), NULL, NULL, NULL, NULL,
771                    event ? event->button : 0,
772                    event ? event->time : gtk_get_current_event_time( ) );
773}
774
775static gboolean
776onViewPathToggled( GtkTreeView       * view,
777                   GtkTreeViewColumn * col,
778                   GtkTreePath       * path,
779                   FileData          * data )
780{
781    int cid;
782    tr_torrent * tor;
783    gboolean handled = FALSE;
784
785    if( !col || !path )
786        return FALSE;
787
788    cid = GPOINTER_TO_INT( g_object_get_data( G_OBJECT( col ), TR_COLUMN_ID_KEY ) );
789    tor = tr_torrentFindFromId( tr_core_session( data->core ), data->torrentId );
790    if( ( tor != NULL ) && ( ( cid == FC_PRIORITY ) || ( cid == FC_ENABLED ) ) )
791    {
792        GtkTreeIter iter;
793        GArray * indices = getActiveFilesForPath( view, path );
794        GtkTreeModel * model = data->model;
795
796        gtk_tree_model_get_iter( model, &iter, path );
797
798        if( cid == FC_PRIORITY )
799        {
800            int priority;
801            gtk_tree_model_get( model, &iter, FC_PRIORITY, &priority, -1 );
802            switch( priority ) {
803                case TR_PRI_NORMAL: priority = TR_PRI_HIGH; break;
804                case TR_PRI_HIGH:   priority = TR_PRI_LOW; break;
805                default:            priority = TR_PRI_NORMAL; break;
806            }
807            tr_torrentSetFilePriorities( tor,
808                                         (tr_file_index_t *) indices->data,
809                                         (tr_file_index_t) indices->len,
810                                         priority );
811        }
812        else
813        {
814            int enabled;
815            gtk_tree_model_get( model, &iter, FC_ENABLED, &enabled, -1 );
816            enabled = !enabled;
817
818            tr_torrentSetFileDLs( tor,
819                                  (tr_file_index_t *) indices->data,
820                                  (tr_file_index_t) indices->len,
821                                  enabled );
822        }
823
824        refresh( data );
825        g_array_free( indices, TRUE );
826        handled = TRUE;
827    }
828
829    return handled;
830}
831
832/**
833 * @note 'col' and 'path' are assumed not to be NULL.
834 */
835static gboolean
836getAndSelectEventPath( GtkTreeView        * treeview,
837                       GdkEventButton     * event,
838                       GtkTreeViewColumn ** col,
839                       GtkTreePath       ** path )
840{
841    GtkTreeSelection * sel;
842
843    if( gtk_tree_view_get_path_at_pos( treeview,
844                                       event->x, event->y,
845                                       path, col, NULL, NULL ) )
846    {
847        sel = gtk_tree_view_get_selection( treeview );
848        if( !gtk_tree_selection_path_is_selected( sel, *path ) )
849        {
850            gtk_tree_selection_unselect_all( sel );
851            gtk_tree_selection_select_path( sel, *path );
852        }
853        return TRUE;
854    }
855
856    return FALSE;
857}
858
859static gboolean
860onViewButtonPressed( GtkWidget * w, GdkEventButton * event, gpointer gdata )
861{
862    tr_torrent        * tor;
863    GtkTreeViewColumn * col = NULL;
864    GtkTreePath       * path = NULL;
865    FileData          * data = gdata;
866    gboolean            handled = FALSE;
867    GtkTreeView       * treeview = GTK_TREE_VIEW( w );
868
869    tor = tr_torrentFindFromId( tr_core_session( data->core ),
870                                data->torrentId );
871    if( tor == NULL )
872        return FALSE;
873
874    if( event->type == GDK_BUTTON_PRESS && event->button == 1
875        && !( event->state & ( GDK_SHIFT_MASK | GDK_CONTROL_MASK ) )
876        && getAndSelectEventPath( treeview, event, &col, &path ) )
877    {
878        handled = onViewPathToggled( treeview, col, path, data );
879    }
880    else if( event->type == GDK_BUTTON_PRESS && event->button == 3
881             && getAndSelectEventPath( treeview, event, &col, &path ) )
882    {
883        GtkTreeSelection * sel = gtk_tree_view_get_selection( treeview );
884        if( gtk_tree_selection_count_selected_rows( sel ) > 0 )
885        {
886            fileMenuPopup( w, event, data );
887            handled = TRUE;
888        }
889    }
890
891    gtk_tree_path_free( path );
892    return handled;
893}
894
895static gboolean
896onViewPopupMenu( GtkWidget * w, gpointer gdata )
897{
898    fileMenuPopup( w, NULL, gdata );
899    return TRUE;
900}
901
902GtkWidget *
903gtr_file_list_new( TrCore * core, int torrentId )
904{
905    int size;
906    int width;
907    GtkWidget * ret;
908    GtkWidget * view;
909    GtkWidget * scroll;
910    GtkCellRenderer * rend;
911    GtkTreeSelection * sel;
912    GtkTreeViewColumn * col;
913    GtkTreeView * tree_view;
914    const char * title;
915    PangoLayout * pango_layout;
916    PangoContext * pango_context;
917    PangoFontDescription * pango_font_description;
918    FileData * data = g_new0( FileData, 1 );
919
920    data->core = core;
921
922    /* create the view */
923    view = gtk_tree_view_new( );
924    tree_view = GTK_TREE_VIEW( view );
925    gtk_tree_view_set_rules_hint( tree_view, TRUE );
926    gtk_container_set_border_width( GTK_CONTAINER( view ), GUI_PAD_BIG );
927    g_signal_connect( view, "button-press-event",
928                      G_CALLBACK( onViewButtonPressed ), data );
929    g_signal_connect( view, "popup-menu",
930                      G_CALLBACK( onViewPopupMenu ), data );
931    g_signal_connect( view, "row_activated",
932                      G_CALLBACK( onRowActivated ), data );
933    g_signal_connect( view, "button-release-event",
934                      G_CALLBACK( on_tree_view_button_released ), NULL );
935
936
937    pango_context = gtk_widget_create_pango_context( view );
938    pango_font_description = pango_font_description_copy( pango_context_get_font_description( pango_context ) );
939    size = pango_font_description_get_size( pango_font_description );
940    pango_font_description_set_size( pango_font_description, size * 0.8 );
941    g_object_unref( G_OBJECT( pango_context ) );
942
943    /* set up view */
944    sel = gtk_tree_view_get_selection( tree_view );
945    gtk_tree_selection_set_mode( sel, GTK_SELECTION_MULTIPLE );
946    gtk_tree_view_expand_all( tree_view );
947    gtk_tree_view_set_search_column( tree_view, FC_LABEL );
948
949    /* add file column */
950    col = GTK_TREE_VIEW_COLUMN ( g_object_new ( GTK_TYPE_TREE_VIEW_COLUMN,
951                                                "expand", TRUE,
952                                                "title", _( "Name" ),
953                                                NULL ) );
954    gtk_tree_view_column_set_resizable( col, TRUE );
955    rend = gtk_cell_renderer_pixbuf_new( );
956    gtk_tree_view_column_pack_start( col, rend, FALSE );
957    gtk_tree_view_column_add_attribute( col, rend, "pixbuf", FC_ICON );
958    /* add text renderer */
959    rend = gtk_cell_renderer_text_new( );
960    g_object_set( rend, "ellipsize", PANGO_ELLIPSIZE_END, "font-desc", pango_font_description, NULL );
961    gtk_tree_view_column_pack_start( col, rend, TRUE );
962    gtk_tree_view_column_set_attributes( col, rend, "text", FC_LABEL, NULL );
963    gtk_tree_view_column_set_sort_column_id( col, FC_LABEL );
964    gtk_tree_view_append_column( tree_view, col );
965
966    /* add "size" column */
967    title = _( "Size" );
968    rend = gtk_cell_renderer_text_new( );
969    g_object_set( rend, "alignment", PANGO_ALIGN_RIGHT,
970                        "font-desc", pango_font_description,
971                        "xpad", GUI_PAD,
972                        "xalign", 1.0f,
973                        "yalign", 0.5f,
974                        NULL );
975    col = gtk_tree_view_column_new_with_attributes( title, rend, NULL );
976    gtk_tree_view_column_set_sizing( col, GTK_TREE_VIEW_COLUMN_GROW_ONLY );
977    gtk_tree_view_column_set_sort_column_id( col, FC_SIZE );
978    gtk_tree_view_column_set_attributes( col, rend, "text", FC_SIZE_STR, NULL );
979    gtk_tree_view_append_column( tree_view, col );
980
981    /* add "progress" column */
982    title = _( "Have" );
983    pango_layout = gtk_widget_create_pango_layout( view, title );
984    pango_layout_get_pixel_size( pango_layout, &width, NULL );
985    width += 30; /* room for the sort indicator */
986    g_object_unref( G_OBJECT( pango_layout ) );
987    rend = gtk_cell_renderer_progress_new( );
988    col = gtk_tree_view_column_new_with_attributes( title, rend, "value", FC_PROG, NULL );
989    gtk_tree_view_column_set_fixed_width( col, width );
990    gtk_tree_view_column_set_sizing( col, GTK_TREE_VIEW_COLUMN_FIXED );
991    gtk_tree_view_column_set_sort_column_id( col, FC_PROG );
992    gtk_tree_view_append_column( tree_view, col );
993
994    /* add "enabled" column */
995    title = _( "Download" );
996    pango_layout = gtk_widget_create_pango_layout( view, title );
997    pango_layout_get_pixel_size( pango_layout, &width, NULL );
998    width += 30; /* room for the sort indicator */
999    g_object_unref( G_OBJECT( pango_layout ) );
1000    rend = gtk_cell_renderer_toggle_new( );
1001    col = gtk_tree_view_column_new_with_attributes( title, rend, NULL );
1002    g_object_set_data( G_OBJECT( col ), TR_COLUMN_ID_KEY,
1003                       GINT_TO_POINTER( FC_ENABLED ) );
1004    gtk_tree_view_column_set_fixed_width( col, width );
1005    gtk_tree_view_column_set_sizing( col, GTK_TREE_VIEW_COLUMN_FIXED );
1006    gtk_tree_view_column_set_cell_data_func( col, rend, renderDownload, NULL, NULL );
1007    gtk_tree_view_column_set_sort_column_id( col, FC_ENABLED );
1008    gtk_tree_view_append_column( tree_view, col );
1009
1010    /* add priority column */
1011    title = _( "Priority" );
1012    pango_layout = gtk_widget_create_pango_layout( view, title );
1013    pango_layout_get_pixel_size( pango_layout, &width, NULL );
1014    width += 30; /* room for the sort indicator */
1015    g_object_unref( G_OBJECT( pango_layout ) );
1016    rend = gtk_cell_renderer_text_new( );
1017    g_object_set( rend, "xalign", (gfloat)0.5, "yalign", (gfloat)0.5, NULL );
1018    col = gtk_tree_view_column_new_with_attributes( title, rend, NULL );
1019    g_object_set_data( G_OBJECT( col ), TR_COLUMN_ID_KEY,
1020                       GINT_TO_POINTER( FC_PRIORITY ) );
1021    gtk_tree_view_column_set_fixed_width( col, width );
1022    gtk_tree_view_column_set_sizing( col, GTK_TREE_VIEW_COLUMN_FIXED );
1023    gtk_tree_view_column_set_sort_column_id( col, FC_PRIORITY );
1024    gtk_tree_view_column_set_cell_data_func( col, rend, renderPriority, NULL, NULL );
1025    gtk_tree_view_append_column( tree_view, col );
1026
1027    /* create the scrolled window and stick the view in it */
1028    scroll = gtk_scrolled_window_new( NULL, NULL );
1029    gtk_scrolled_window_set_policy( GTK_SCROLLED_WINDOW( scroll ),
1030                                    GTK_POLICY_AUTOMATIC,
1031                                    GTK_POLICY_AUTOMATIC );
1032    gtk_scrolled_window_set_shadow_type ( GTK_SCROLLED_WINDOW( scroll ),
1033                                          GTK_SHADOW_IN );
1034    gtk_container_add( GTK_CONTAINER( scroll ), view );
1035    gtk_widget_set_size_request ( scroll, -1, 200 );
1036
1037    ret = scroll;
1038    data->view = view;
1039    data->top = scroll;
1040    g_object_set_data_full( G_OBJECT( ret ), "file-data", data, freeData );
1041    gtr_file_list_set_torrent( ret, torrentId );
1042
1043    pango_font_description_free( pango_font_description );
1044    return ret;
1045}
Note: See TracBrowser for help on using the repository browser.