vcs.maemo.org Git - mussorgsky/blob - src/browse_panel.py

   1 #!/usr/bin/env python2.5
   2 import hildon
   3 import gtk, gobject
   4 from edit_panel_tm import MussorgskyEditPanel
   5 from utils import escape_html, Set, is_empty
   6
   7 # Shared with edit_panel_tm
   8 URI_COLUMN = 0
   9 ARTIST_COLUMN = 2
  10 TITLE_COLUMN = 3
  11 ALBUM_COLUMN = 4
  12 MIME_COLUMN = 5
  13 UI_COLUMN = 6
  14 SEARCH_COLUMN = 7
  15
  16 SHOW_ALL = 1
  17 SHOW_UNCOMPLETE = 2
  18 SHOW_MATCH = 3
  19
  20 import time
  21
  22 class MussorgskyBrowsePanel (hildon.StackableWindow):
  23
  24     def __init__ (self, songs_list):
  25         hildon.StackableWindow.__init__ (self)
  26         self.set_title ("Browse collection")
  27         self.set_border_width (12)
  28         self.__create_view ()
  29
  30         # Prepare cache of artists and albums
  31         self.artist_set = Set ()
  32         self.albums_set = Set ()
  33
  34         # (uri, "Music", artist, title, album, mimetype) + "string" + search_string
  35         full_model = gtk.ListStore (str, str, str, str, str, str, str, str)
  36         for (uri, category, artist, title, album, mime) in songs_list:
  37             if is_empty (artist) and is_empty (title) and is_empty (album):
  38                 text = "<small>%s</small>" % (escape_html (uri))
  39             else:
  40                 text = "<b>%s</b>\n<small>%s</small>" % (escape_html (title),
  41                                                          escape_html (artist) + " / " + escape_html (album))
  42             search_str = " ".join ([artist.lower (),
  43                                    title.lower (),
  44                                    album.lower ()])
  45             full_model.append ((uri, None, artist, title, album, mime, text, search_str))
  46             self.artist_set.insert (artist)
  47             self.albums_set.insert (album)
  48
  49         self.filtered_model = full_model.filter_new ()
  50         self.treeview.set_model (self.filtered_model)
  51         self.filter_mode = SHOW_ALL
  52         self.filtered_model.set_visible_func (self.filter_entry)
  53         self.__set_filter_mode (None, SHOW_ALL)
  54
  55         self.kpid = self.connect ("key-press-event", self.key_pressed_cb)
  56
  57     def __create_view (self):
  58         vbox = gtk.VBox (homogeneous=False)
  59
  60         menu = hildon.AppMenu ()
  61         self.all_items = gtk.RadioButton (None, "All")
  62         self.all_items.set_mode (False)
  63         self.all_items.connect_after ("toggled", self.__set_filter_mode, SHOW_ALL)
  64         menu.add_filter (self.all_items)
  65         self.broken_items = gtk.RadioButton (self.all_items, "Uncomplete")
  66         self.broken_items.set_mode (False)
  67         self.broken_items.connect_after ("toggled",
  68                                          self.__set_filter_mode, SHOW_UNCOMPLETE)
  69         menu.add_filter (self.broken_items)
  70         menu.show_all ()
  71         self.set_app_menu (menu)
  72
  73         self.treeview = gtk.TreeView ()
  74         self.treeview.connect ("row-activated", self.row_activated_cb)
  75         desc_column = gtk.TreeViewColumn ("Song", gtk.CellRendererText (), markup=6)
  76         desc_column.set_expand (True)
  77         self.treeview.append_column (desc_column)
  78
  79         pannable_area = hildon.PannableArea ()
  80         pannable_area.add (self.treeview)
  81
  82         vbox.pack_start (pannable_area, expand=True)
  83
  84         self.search_hbox = gtk.HBox ()
  85         self.search_entry = hildon.Entry (gtk.HILDON_SIZE_FINGER_HEIGHT)
  86         self.search_hbox.pack_start (self.search_entry, expand=True)
  87
  88         self.search_close = gtk.Button (stock=gtk.STOCK_CLOSE)
  89         self.search_hbox.pack_start (self.search_close, expand=False)
  90         self.search_close.connect ("clicked", self.close_search_cb)
  91
  92         # Hide it when the window is created
  93         self.search_box_visible = False
  94         self.search_hbox.set_no_show_all (True)
  95         self.search_hbox.hide ()
  96         vbox.pack_start (self.search_hbox, expand=False)
  97         self.add (vbox)
  98
  99     def search_type (self, widget):
 100         self.filtered_model.refilter ()
 101
 102     def close_search_cb (self, widget):
 103         assert not self.search_box_visible
 104         self.search_hbox.hide_all ()
 105         self.search_entry.set_text ("")
 106         self.search_box_visible = False
 107         if (self.all_items.get_active ()):
 108             self.filter_mode = SHOW_ALL
 109         else:
 110             self.filter_mode = SHOW_UNCOMPLETE
 111         self.filtered_model.refilter ()
 112         self.kpid = self.connect ("key-press-event", self.key_pressed_cb)
 113
 114     def key_pressed_cb (self, widget, event):
 115         if (event.type == gtk.gdk.KEY_PRESS):
 116             if (event.keyval == gtk.gdk.keyval_from_name ("Alt_L")):
 117                 return
 118
 119             if (not self.search_box_visible ):
 120                 self.filter_mode = SHOW_MATCH
 121                 self.search_hbox.set_no_show_all (False)
 122                 self.search_hbox.show_all ()
 123
 124             self.search_entry.grab_focus ()
 125             self.search_entry.connect ("changed", self.search_type)
 126             self.disconnect (self.kpid)
 127
 128
 129     def row_activated_cb (self, treeview, path, view_colum):
 130         edit_view = MussorgskyEditPanel ()
 131         edit_view.set_artist_alternatives (self.artist_set.as_list ())
 132         edit_view.set_album_alternatives (self.albums_set.as_list ())
 133         edit_view.set_model (self.treeview.get_model (), self.treeview.get_model ().get_iter (path))
 134         edit_view.show_all ()
 135
 136     def __set_filter_mode (self, button, filter_mode):
 137         """
 138         Parameter to use it as callback as well as regular function
 139         """
 140         if (filter_mode == self.filter_mode):
 141             # Don't refilter if there is no change!
 142             return
 143         self.filter_mode = filter_mode
 144
 145         start = time.time ()
 146         self.treeview.set_model (None)
 147         self.filtered_model.refilter ()
 148         self.treeview.set_model (self.filtered_model)
 149         end = time.time ()
 150         print "Refiltering ", end - start
 151
 152     def filter_entry (self, model, it):
 153         if self.filter_mode == SHOW_ALL:
 154             return True
 155         elif self.filter_mode == SHOW_UNCOMPLETE:
 156             return self.entry_uncomplete (model, it)
 157         elif self.filter_mode == SHOW_MATCH:
 158             return self.entry_with_text (model, it)
 159
 160     def entry_with_text (self, model, it):
 161         t = self.search_entry.get_text ()
 162         return t.lower () in model.get_value (it, SEARCH_COLUMN)
 163
 164     def entry_uncomplete (self, model, it):
 165         r = filter (lambda x: not x or len(x.strip()) == 0,
 166                     model.get (it, ARTIST_COLUMN, TITLE_COLUMN, ALBUM_COLUMN))
 167         return len (r) > 0
 168
 169 if __name__ == "__main__":
 170
 171     import random
 172     def get_random_path ():
 173         path = "file://"
 174         for i in range (0, random.randint (1, 8)):
 175             path = path + "/" + ("x"* random.randint (4, 12))
 176         return path
 177
 178     def get_some_empty_titles (i):
 179         if random.randint (0, 5) <= 1:
 180             return ""
 181         else:
 182             return "Title <%d>" % i
 183
 184
 185     songs = [(get_random_path (),
 186               "Music",
 187               "Artist%d" % i,
 188               get_some_empty_titles (i),
 189               "album <%d>" % i,
 190               "audio/mpeg") for i in range (0, 30000)]
 191
 192     songs.append (("file:///no/metadata/at/all",
 193                    "music",
 194                    "",
 195                    "",
 196                    "",
 197                    "audio/mpeg"))
 198
 199     window = MussorgskyBrowsePanel (songs)
 200     window.connect ("destroy", gtk.main_quit )
 201     window.show_all ()
 202     gtk.main ()