def _viewErrorsButtonClickedCb(self, unused_button): """ Show a FileListErrorDialog to display import _errors. """ if len(self._errors) > 1: msgs = (_("Error while analyzing files"), _("The following files can not be used with Pitivi.")) else: msgs = (_("Error while analyzing a file"), _("The following file can not be used with Pitivi.")) error_dialogbox = FileListErrorDialog(*msgs) error_dialogbox.connect("close", self._errorDialogBoxCloseCb) error_dialogbox.connect("response", self._errorDialogBoxResponseCb) for uri, reason, extra in self._errors: error_dialogbox.addFailedFile(uri, reason, extra) error_dialogbox.window.show() # Reset the error list, since the user has read them. self._resetErrorList()
def _viewErrorsButtonClickedCb(self, unused_button): """ Show a FileListErrorDialog to display import _errors. """ if len(self._errors) > 1: msgs = (_("Error while analyzing files"), _("The following files can not be used with Pitivi.")) else: msgs = (_("Error while analyzing a file"), _("The following file can not be used with Pitivi.")) error_dialogbox = FileListErrorDialog(*msgs) error_dialogbox.connect("close", self._errorDialogBoxCloseCb) error_dialogbox.connect("response", self._errorDialogBoxResponseCb) for uri, reason, extra in self._errors: error_dialogbox.addFailedFile(uri, reason, extra) error_dialogbox.window.set_transient_for(self.app.gui) error_dialogbox.window.show() # Reset the error list, since the user has read them. self._resetErrorList()
class MediaLibraryWidget(Gtk.VBox, Loggable): """ Widget for listing sources """ __gsignals__ = { 'play': (GObject.SignalFlags.RUN_LAST, None, (GObject.TYPE_PYOBJECT,))} def __init__(self, instance, uiman): Gtk.VBox.__init__(self) Loggable.__init__(self) self.pending_rows = [] self.app = instance self._errors = [] self._project = None self._draggedPaths = None self.dragged = False self.clip_view = self.app.settings.lastClipView builder = Gtk.Builder() builder.add_from_file(os.path.join(get_ui_dir(), "medialibrary.ui")) builder.connect_signals(self) self._welcome_infobar = builder.get_object("welcome_infobar") self._import_warning_infobar = builder.get_object("warning_infobar") self._warning_label = builder.get_object("warning_label") self._view_error_button = builder.get_object("view_error_button") toolbar = builder.get_object("medialibrary_toolbar") toolbar.get_style_context().add_class("inline-toolbar") self._remove_button = builder.get_object("media_remove_button") self._clipprops_button = builder.get_object("media_props_button") self._insert_button = builder.get_object("media_insert_button") self._listview_button = builder.get_object("media_listview_button") searchEntry = builder.get_object("media_search_entry") # Store self.storemodel = Gtk.ListStore(*STORE_MODEL_STRUCTURE) self.storemodel.set_sort_func(COL_URI, compare_simple, user_data=COL_URI) # Prefer to sort the media library elements by URI # rather than show them randomly. self.storemodel.set_sort_column_id(COL_URI, Gtk.SortType.ASCENDING) # Scrolled Windows self.treeview_scrollwin = Gtk.ScrolledWindow() self.treeview_scrollwin.set_policy(Gtk.PolicyType.NEVER, Gtk.PolicyType.AUTOMATIC) self.treeview_scrollwin.set_shadow_type(Gtk.ShadowType.ETCHED_IN) self.iconview_scrollwin = Gtk.ScrolledWindow() self.iconview_scrollwin.set_policy(Gtk.PolicyType.AUTOMATIC, Gtk.PolicyType.AUTOMATIC) self.iconview_scrollwin.set_shadow_type(Gtk.ShadowType.ETCHED_IN) # import sources dialogbox self._importDialog = None # Filtering model for the search box. # Use this instead of using self.storemodel directly self.modelFilter = self.storemodel.filter_new() self.modelFilter.set_visible_func(self._setRowVisible, data=searchEntry) # TreeView # Displays icon, name, type, length self.treeview = Gtk.TreeView(self.modelFilter) self.treeview_scrollwin.add(self.treeview) self.treeview.connect("button-press-event", self._treeViewButtonPressEventCb) self.treeview.connect("button-release-event", self._treeViewButtonReleaseEventCb) self.treeview.connect("focus-in-event", self._disableKeyboardShortcutsCb) self.treeview.connect("focus-out-event", self._enableKeyboardShortcutsCb) self.treeview.connect("row-activated", self._itemOrRowActivatedCb) self.treeview.set_property("rules_hint", True) self.treeview.set_headers_visible(False) self.treeview.set_property("search_column", COL_SEARCH_TEXT) tsel = self.treeview.get_selection() tsel.set_mode(Gtk.SelectionMode.MULTIPLE) tsel.connect("changed", self._viewSelectionChangedCb) pixbufcol = Gtk.TreeViewColumn(_("Icon")) pixbufcol.set_expand(False) pixbufcol.set_spacing(SPACING) self.treeview.append_column(pixbufcol) pixcell = Gtk.CellRendererPixbuf() pixcell.props.xpad = 6 pixbufcol.pack_start(pixcell, True) pixbufcol.add_attribute(pixcell, 'pixbuf', COL_ICON) namecol = Gtk.TreeViewColumn(_("Information")) self.treeview.append_column(namecol) namecol.set_expand(True) namecol.set_spacing(SPACING) namecol.set_sizing(Gtk.TreeViewColumnSizing.GROW_ONLY) namecol.set_min_width(150) txtcell = Gtk.CellRendererText() txtcell.set_property("ellipsize", Pango.EllipsizeMode.END) namecol.pack_start(txtcell, True) namecol.add_attribute(txtcell, "markup", COL_INFOTEXT) namecol = Gtk.TreeViewColumn(_("Duration")) namecol.set_expand(False) self.treeview.append_column(namecol) txtcell = Gtk.CellRendererText() txtcell.set_property("yalign", 0.0) namecol.pack_start(txtcell, True) namecol.add_attribute(txtcell, "markup", COL_LENGTH) # IconView self.iconview = Gtk.IconView(self.modelFilter) self.iconview_scrollwin.add(self.iconview) self.iconview.connect("button-press-event", self._iconViewButtonPressEventCb) self.iconview.connect("button-release-event", self._iconViewButtonReleaseEventCb) self.iconview.connect("focus-in-event", self._disableKeyboardShortcutsCb) self.iconview.connect("focus-out-event", self._enableKeyboardShortcutsCb) self.iconview.connect("item-activated", self._itemOrRowActivatedCb) self.iconview.connect("selection-changed", self._viewSelectionChangedCb) self.iconview.set_item_orientation(Gtk.Orientation.VERTICAL) self.iconview.set_property("has_tooltip", True) self.iconview.set_tooltip_column(COL_INFOTEXT) self.iconview.props.item_padding = 3 self.iconview.props.margin = 3 self.iconview_cursor_pos = None cell = Gtk.CellRendererPixbuf() self.iconview.pack_start(cell, False) self.iconview.add_attribute(cell, "pixbuf", COL_ICON_LARGE) cell = Gtk.CellRendererText() cell.props.alignment = Pango.Alignment.CENTER cell.props.xalign = 0.5 cell.props.yalign = 0.0 cell.props.xpad = 0 cell.props.ypad = 0 cell.set_property("ellipsize", Pango.EllipsizeMode.END) self.iconview.pack_start(cell, False) self.iconview.add_attribute(cell, "markup", COL_SEARCH_TEXT) self.iconview.set_selection_mode(Gtk.SelectionMode.MULTIPLE) # The _progressbar that shows up when importing clips self._progressbar = Gtk.ProgressBar() self._progressbar.set_show_text(True) # Connect to project. We must remove and reset the callbacks when # changing project. self.project_signals = SignalGroup() self.app.connect("new-project-created", self._newProjectCreatedCb) self.app.connect("new-project-loaded", self._newProjectLoadedCb) self.app.connect("new-project-failed", self._newProjectFailedCb) # default pixbufs self.audiofilepixbuf = self._getIcon("audio-x-generic") self.videofilepixbuf = self._getIcon("video-x-generic") # Drag and Drop self.drag_dest_set(Gtk.DestDefaults.DROP | Gtk.DestDefaults.MOTION, [dnd.URI_TARGET_ENTRY, dnd.FILE_TARGET_ENTRY], Gdk.DragAction.COPY) self.drag_dest_add_uri_targets() self.connect("drag_data_received", self._dndDataReceivedCb) self._setup_view_for_drag_and_drop(self.treeview, [("pitivi/file-source", 0, TYPE_PITIVI_FILESOURCE)]) self._setup_view_for_drag_and_drop(self.iconview, [Gtk.TargetEntry.new("pitivi/file-source", 0, TYPE_PITIVI_FILESOURCE)]) # Hack so that the views have the same method as self self.treeview.getSelectedItems = self.getSelectedItems # Keyboard shortcuts for some items in the gtkbuilder file selection_actions = ( ("RemoveSources", Gtk.STOCK_DELETE, _("_Remove from Project"), "<Control>Delete", None, self._removeSourcesCb), ("InsertEnd", Gtk.STOCK_COPY, _("Insert at _End of Timeline"), "Insert", None, self._insertEndCb), ) self.selection_actions = Gtk.ActionGroup("medialibraryselection") self.selection_actions.add_actions(selection_actions) self.selection_actions.set_sensitive(False) uiman.insert_action_group(self.selection_actions, 0) uiman.add_ui_from_string(ui) # Set the state of the view mode toggle button. self._listview_button.set_active(self.clip_view == SHOW_TREEVIEW) # Make sure the proper view is displayed. self._displayClipView() # Add all the child widgets. self.pack_start(toolbar, False, False, 0) self.pack_start(self._welcome_infobar, False, False, 0) self.pack_start(self._import_warning_infobar, False, False, 0) self.pack_start(self.iconview_scrollwin, True, True, 0) self.pack_start(self.treeview_scrollwin, True, True, 0) self.pack_start(self._progressbar, False, True, 0) def _setup_view_for_drag_and_drop(self, view, target_entries): view.drag_source_set(0, [], Gdk.DragAction.COPY) view.enable_model_drag_source(Gdk.ModifierType.BUTTON1_MASK, target_entries, Gdk.DragAction.COPY) view.drag_source_set_target_list(None) view.drag_source_add_uri_targets() view.drag_source_add_text_targets() view.connect("drag_begin", self._dndDragBeginCb) view.connect("drag-end", self._dndDragEndCb) def _importSourcesCb(self, unused_action): self.showImportSourcesDialog() def _removeSourcesCb(self, unused_action): self._removeSources() def _insertEndCb(self, unused_action): self.app.gui.timeline_ui.insertEnd(self.getSelectedAssets()) def _disableKeyboardShortcutsCb(self, *unused_args): """ Disable the Delete keyboard shortcut and playback shortcuts to prevent accidents or being unable to type various characters. This is used when focusing the search entry, icon on tree view widgets. """ self.app.gui.setActionsSensitive(False) def _enableKeyboardShortcutsCb(self, *unused_args): """ When focusing out of media library widgets, re-enable the timeline keyboard shortcuts. """ self.app.gui.setActionsSensitive(True) def _trackElementAddedCb(self, source, unused_track_element): """ After an object has been added to the first track, position it correctly and request the next source to be processed. """ timeline = self.app.current.timeline layer = timeline.get_layers()[0] # FIXME Get the longest layer # Handle the case where we just inserted the first clip if len(layer.get_clips()) == 1: source.props.start = 0 else: source.props.start = timeline.props.duration # We only need one TrackElement to estimate the new duration. # Process the next source. source.disconnect_by_func(self._trackElementAddedCb) self._insertNextSource() def _searchEntryChangedCb(self, entry): # With many hundred clips in an iconview with dynamic columns and # ellipsizing, doing needless searches is very expensive. # Realistically, nobody expects to search for only one character, # and skipping that makes a huge difference in responsiveness. if len(entry.get_text()) != 1: self.modelFilter.refilter() def _searchEntryIconClickedCb(self, entry, icon_pos, unused_event): if icon_pos == Gtk.EntryIconPosition.SECONDARY: entry.set_text("") elif icon_pos == Gtk.EntryIconPosition.PRIMARY: self._selectUnusedSources() def _setRowVisible(self, model, iter, data): """ Toggle the visibility of a liststore row. Used for the search box. """ text = data.get_text().lower() if text == "": return True # Avoid silly warnings else: # We must convert to markup form to be able to search for &, ', etc. text = GLib.markup_escape_text(text) return text in model.get_value(iter, COL_INFOTEXT).lower() def _getIcon(self, iconname, alternate=None): icontheme = Gtk.IconTheme.get_default() pixdir = get_pixmap_dir() icon = None try: icon = icontheme.load_icon(iconname, 48, 0) except: # empty except clause is bad but load_icon raises Gio.Error. # Right, *gio*. if alternate: icon = GdkPixbuf.Pixbuf.new_from_file(os.path.join(pixdir, alternate)) else: icon = icontheme.load_icon("dialog-question", 48, 0) return icon def _connectToProject(self, project): """Connect signal handlers to a project. This first disconnects any handlers connected to an old project. If project is None, this just disconnects any connected handlers. """ self.project_signals.connect(project, "asset-added", None, self._assetAddedCb) self.project_signals.connect(project, "asset-removed", None, self._assetRemovedCb) self.project_signals.connect(project, "error-loading-asset", None, self._errorCreatingAssetCb) self.project_signals.connect(project, "done-importing", None, self._sourcesStoppedImportingCb) self.project_signals.connect(project, "start-importing", None, self._sourcesStartedImportingCb) # The start-importing signal would have already been emited at that # time, make sure to catch if it is the case if project.nb_remaining_file_to_import > 0: self._sourcesStartedImportingCb(project) def _setClipView(self, view_type): """ Set which clip view to use when medialibrary is showing clips. view_type: one of SHOW_TREEVIEW or SHOW_ICONVIEW """ self.app.settings.lastClipView = view_type # Gather some info before switching views paths = self.getSelectedPaths() self._viewUnselectAll() # Now that we've got all the info, we can actually change the view type self.clip_view = view_type self._displayClipView() for path in paths: self._viewSelectPath(path) def _displayClipView(self): if self.clip_view == SHOW_TREEVIEW: self.iconview_scrollwin.hide() self.treeview_scrollwin.show_all() elif self.clip_view == SHOW_ICONVIEW: self.treeview_scrollwin.hide() self.iconview_scrollwin.show_all() if not len(self.storemodel): self._welcome_infobar.show_all() def showImportSourcesDialog(self): """Pop up the "Import Sources" dialog box""" if self._importDialog: return chooser_action = Gtk.FileChooserAction.OPEN dialogtitle = _("Select One or More Files") close_after = Gtk.CheckButton(_("Close after importing files")) close_after.set_active(self.app.settings.closeImportDialog) self._importDialog = Gtk.FileChooserDialog(dialogtitle, None, chooser_action, (Gtk.STOCK_CLOSE, Gtk.ResponseType.CLOSE, Gtk.STOCK_ADD, Gtk.ResponseType.OK)) self._importDialog.set_icon_name("pitivi") self._importDialog.props.extra_widget = close_after self._importDialog.set_default_response(Gtk.ResponseType.OK) self._importDialog.set_select_multiple(True) self._importDialog.set_modal(False) self._importDialog.set_current_folder(self.app.settings.lastImportFolder) self._importDialog.connect('response', self._dialogBoxResponseCb) self._importDialog.connect('close', self._dialogBoxCloseCb) pw = PreviewWidget(self.app) self._importDialog.set_preview_widget(pw) self._importDialog.set_use_preview_label(False) self._importDialog.connect('update-preview', pw.add_preview_request) # Filter for the "known good" formats by default filt_supported = Gtk.FileFilter() filt_known = Gtk.FileFilter() filt_supported.set_name(_("Supported file formats")) for category in SUPPORTED_FILE_FORMATS: # Category can be "video", "audio", "image", "application" for mime in SUPPORTED_FILE_FORMATS[category]: filt_supported.add_mime_type(category + "/" + mime) filt_known.add_mime_type(category + "/" + mime) # Also allow showing known but not reliable demuxers filt_known.set_name(_("All known file formats")) for fullmime in OTHER_KNOWN_FORMATS: filt_known.add_mime_type(fullmime) # ...and allow the user to override our whitelists default = Gtk.FileFilter() default.set_name(_("All files")) default.add_pattern("*") self._importDialog.add_filter(filt_supported) self._importDialog.add_filter(filt_known) self._importDialog.add_filter(default) self._importDialog.show() def _updateProgressbar(self): """ Update the _progressbar with the ratio of clips imported vs the total """ current_clip_iter = self.app.current.nb_imported_files total_clips = self.app.current.nb_remaining_file_to_import + current_clip_iter progressbar_text = _("Importing clip %(current_clip)d of %(total)d" % {"current_clip": current_clip_iter, "total": total_clips}) self._progressbar.set_text(progressbar_text) if current_clip_iter == 0: self._progressbar.set_fraction(0.0) elif total_clips != 0: self._progressbar.set_fraction((current_clip_iter - 1) / float(total_clips)) def _getThumbnailInDir(self, dir, hash): thumb_path_normal = dir + "normal/" + hash + ".png" try: thumbnail = GdkPixbuf.Pixbuf.new_from_file(thumb_path_normal) thumbnail_large = thumbnail thumbnail_height = int(thumbnail.get_height() / 2) thumbnail = thumbnail.scale_simple(64, thumbnail_height, GdkPixbuf.InterpType.BILINEAR) return thumbnail, thumbnail_large except GError: return None, None def _addAsset(self, asset): info = asset.get_info() # The code below tries to read existing thumbnails from the freedesktop # thumbnails directory (~/.thumbnails). The filenames are simply # the file URI hashed with md5, so we can retrieve them easily. video_streams = [i for i in info.get_stream_list() if isinstance(i, DiscovererVideoInfo)] if len(video_streams) > 0: # From the freedesktop spec: "if the environment variable # $XDG_CACHE_HOME is set and not blank then the directory # $XDG_CACHE_HOME/thumbnails will be used, otherwise # $HOME/.cache/thumbnails will be used." # Older version of the spec also mentioned $HOME/.thumbnails thumbnail_hash = md5(quote_uri(info.get_uri())).hexdigest() try: thumb_dir = os.environ['XDG_CACHE_HOME'] thumbnail, thumbnail_large = self._getThumbnailInDir(thumb_dir, thumbnail_hash) except KeyError: thumbnail, thumbnail_large = (None, None) if thumbnail is None: thumb_dir = os.path.expanduser("~/.cache/thumbnails/") thumbnail, thumbnail_large = self._getThumbnailInDir(thumb_dir, thumbnail_hash) if thumbnail is None: thumb_dir = os.path.expanduser("~/.thumbnails/") thumbnail, thumbnail_large = self._getThumbnailInDir(thumb_dir, thumbnail_hash) if thumbnail is None: thumbnail = self.videofilepixbuf # TODO gst discoverer should create missing thumbnails. thumbnail_large = thumbnail else: thumbnail = self.audiofilepixbuf thumbnail_large = self.audiofilepixbuf if info.get_duration() == Gst.CLOCK_TIME_NONE: duration = '' else: duration = beautify_length(info.get_duration()) name = info_name(info) self.pending_rows.append((thumbnail, thumbnail_large, beautify_info(info), asset, info.get_uri(), duration, name)) if len(self.pending_rows) > 50: self.flush_pending_rows() def flush_pending_rows(self): for row in self.pending_rows: self.storemodel.append(row) del self.pending_rows[:] # medialibrary callbacks def _assetAddedCb(self, unused_project, asset, current_clip_iter=None, total_clips=None): """ a file was added to the medialibrary """ if isinstance(asset, GES.UriClipAsset): self._updateProgressbar() self._addAsset(asset) def _assetRemovedCb(self, unsued_project, asset): """ the given uri was removed from the medialibrary """ # find the good line in the storemodel and remove it model = self.storemodel uri = asset.get_id() for row in model: if uri == row[COL_URI]: model.remove(row.iter) break if not len(model): self._welcome_infobar.show_all() self.debug("Removing %s", uri) def _errorCreatingAssetCb(self, unsued_project, error, id, type): """ The given uri isn't a media file """ if GObject.type_is_a(type, GES.UriClip): error = (id, str(error.domain), error) self._errors.append(error) self._updateProgressbar() def _sourcesStartedImportingCb(self, unsued_project): self.import_start_time = time.time() self._welcome_infobar.hide() self._progressbar.show() def _sourcesStoppedImportingCb(self, unsued_project): self.debug("Importing took %.3f seconds" % (time.time() - self.import_start_time)) self.flush_pending_rows() self._progressbar.hide() if self._errors: if len(self._errors) > 1: self._warning_label.set_text(_("Errors occurred while importing.")) self._view_error_button.set_label(_("View errors")) else: self._warning_label.set_text(_("An error occurred while importing.")) self._view_error_button.set_label(_("View error")) self._import_warning_infobar.show_all() ## Error Dialog Box callbacks def _errorDialogBoxCloseCb(self, unused_dialog): self._error_dialogbox.destroy() self._error_dialogbox = None def _errorDialogBoxResponseCb(self, unused_dialog, unused_response): self._error_dialogbox.destroy() self._error_dialogbox = None ## Import Sources Dialog Box callbacks def _dialogBoxResponseCb(self, dialogbox, response): self.debug("response:%r", response) if response == Gtk.ResponseType.OK: lastfolder = dialogbox.get_current_folder() self.app.settings.lastImportFolder = lastfolder self.app.settings.closeImportDialog = \ dialogbox.props.extra_widget.get_active() filenames = dialogbox.get_uris() self.app.current.addUris(filenames) if self.app.settings.closeImportDialog: dialogbox.destroy() self._importDialog = None else: dialogbox.destroy() self._importDialog = None def _dialogBoxCloseCb(self, unused_dialogbox): self.debug("closing") self._importDialog = None def _removeSources(self): """ Determine which clips are selected in the icon or list view, and ask MediaLibrary to remove them from the project. """ model = self.treeview.get_model() paths = self.getSelectedPaths() if paths is None or paths < 1: return # use row references so we don't have to care if a path has been removed rows = [] for path in paths: row = Gtk.TreeRowReference.new(model, path) rows.append(row) self.app.action_log.begin("remove clip from source list") for row in rows: asset = model[row.get_path()][COL_ASSET] self.app.current.remove_asset(asset) self.app.action_log.commit() def _sourceIsUsed(self, asset): """Check if a given URI is present in the timeline""" layers = self.app.current.timeline.get_layers() for layer in layers: for clip in layer.get_clips(): if clip.get_asset() == asset: return True return False def _selectUnusedSources(self): """ Select, in the media library, unused sources in the project. """ assets = self.app.current.list_assets(GES.UriClip) unused_sources_uris = [] model = self.treeview.get_model() selection = self.treeview.get_selection() for asset in assets: if not self._sourceIsUsed(asset): unused_sources_uris.append(asset.get_id()) # Hack around the fact that making selections (in a treeview/iconview) # deselects what was previously selected if self.clip_view == SHOW_TREEVIEW: self.treeview.get_selection().select_all() elif self.clip_view == SHOW_ICONVIEW: self.iconview.select_all() for row in model: if row[COL_URI] not in unused_sources_uris: if self.clip_view == SHOW_TREEVIEW: selection.unselect_iter(row.iter) else: self.iconview.unselect_path(row.path) ## UI callbacks def _removeClickedCb(self, unused_widget=None): """ Called when a user clicks on the remove button """ self._removeSources() def _clipPropertiesCb(self, unused_widget=None): """ Show the clip properties (resolution, framerate, audio channels...) and allow setting them as the new project settings. """ paths = self.getSelectedPaths()[0] # Only use the first item model = self.treeview.get_model() info = model[paths][COL_ASSET].get_info() d = clipmediapropsDialog(self.app.current, info.get_audio_streams(), info.get_video_streams()) d.run() def _warningInfoBarDismissedCb(self, unused_button): self._resetErrorList() def _resetErrorList(self): self._errors = [] self._import_warning_infobar.hide() def _viewErrorsButtonClickedCb(self, unused_button): """ Show a FileListErrorDialog to display import _errors. """ if len(self._errors) > 1: msgs = (_("Error while analyzing files"), _("The following files can not be used with PiTiVi.")) else: msgs = (_("Error while analyzing a file"), _("The following file can not be used with PiTiVi.")) self._error_dialogbox = FileListErrorDialog(*msgs) self._error_dialogbox.connect("close", self._errorDialogBoxCloseCb) self._error_dialogbox.connect("response", self._errorDialogBoxResponseCb) for uri, reason, extra in self._errors: self._error_dialogbox.addFailedFile(uri, reason, extra) self._error_dialogbox.window.show() # Reset the error list, since the user has read them. self._resetErrorList() def _toggleViewTypeCb(self, widget): if widget.get_active(): self._setClipView(SHOW_TREEVIEW) else: self._setClipView(SHOW_ICONVIEW) def _rowUnderMouseSelected(self, view, event): if isinstance(view, Gtk.TreeView): path = None tup = view.get_path_at_pos(int(event.x), int(event.y)) if tup: path, column, x, y = tup if path: selection = view.get_selection() return selection.path_is_selected(path) and selection.count_selected_rows() > 0 elif isinstance(view, Gtk.IconView): path = view.get_path_at_pos(int(event.x), int(event.y)) if path: selection = view.get_selected_items() return view.path_is_selected(path) and len(selection) else: assert False return False def _nothingUnderMouse(self, view, event): return not bool(view.get_path_at_pos(int(event.x), int(event.y))) def _viewGetFirstSelected(self): paths = self.getSelectedPaths() return paths[0] def _viewHasSelection(self): paths = self.getSelectedPaths() return bool(len(paths)) def _viewGetPathAtPos(self, event): if self.clip_view == SHOW_TREEVIEW: pathinfo = self.treeview.get_path_at_pos(int(event.x), int(event.y)) return pathinfo[0] elif self.clip_view == SHOW_ICONVIEW: return self.iconview.get_path_at_pos(int(event.x), int(event.y)) def _viewSelectPath(self, path): if self.clip_view == SHOW_TREEVIEW: selection = self.treeview.get_selection() selection.select_path(path) elif self.clip_view == SHOW_ICONVIEW: self.iconview.select_path(path) def _viewUnselectAll(self): if self.clip_view == SHOW_TREEVIEW: selection = self.treeview.get_selection() selection.unselect_all() elif self.clip_view == SHOW_ICONVIEW: self.iconview.unselect_all() def _treeViewButtonPressEventCb(self, treeview, event): chain_up = True if event.type == getattr(Gdk.EventType, '2BUTTON_PRESS'): # It is possible to double-click outside of clips: if self.getSelectedPaths() != []: # Here we used to emit "play", but # this is now handled by _itemOrRowActivatedCb instead. pass chain_up = False elif not event.get_state() & (Gdk.ModifierType.CONTROL_MASK | Gdk.ModifierType.SHIFT_MASK): chain_up = not self._rowUnderMouseSelected(treeview, event) if not chain_up: self._draggedPaths = self.getSelectedPaths() else: self._draggedPaths = None Gtk.TreeView.do_button_press_event(treeview, event) ts = self.treeview.get_selection() if self._draggedPaths: for path in self._draggedPaths: ts.select_path(path) return True def _treeViewButtonReleaseEventCb(self, treeview, event): ts = self.treeview.get_selection() state = event.get_state() & (Gdk.ModifierType.CONTROL_MASK | Gdk.ModifierType.SHIFT_MASK) path = self.treeview.get_path_at_pos(event.x, event.y) if not state and not self.dragged: ts.unselect_all() if path: ts.select_path(path[0]) def _viewSelectionChangedCb(self, unused): selected_items = len(self.getSelectedPaths()) if selected_items: self.selection_actions.set_sensitive(True) self._remove_button.set_sensitive(True) self._insert_button.set_sensitive(True) # Some actions can only be done on a single item at a time: self._clipprops_button.set_sensitive(False) if selected_items == 1: self._clipprops_button.set_sensitive(True) else: self.selection_actions.set_sensitive(False) self._remove_button.set_sensitive(False) self._insert_button.set_sensitive(False) self._clipprops_button.set_sensitive(False) def _itemOrRowActivatedCb(self, unused_view, path, *unused_column): """ When an item is double-clicked, or Space, Shift+Space, Return or Enter is pressed, preview the clip. This method is the same for both iconview and treeview. """ asset = self.modelFilter[path][COL_ASSET] self.emit('play', asset) def _iconViewButtonPressEventCb(self, iconview, event): chain_up = True if event.type == getattr(Gdk.EventType, '2BUTTON_PRESS'): # It is possible to double-click outside of clips: if self.getSelectedPaths() != []: # Here we used to emit "play", but # this is now handled by _itemOrRowActivatedCb instead. pass chain_up = False elif not event.get_state() & (Gdk.ModifierType.CONTROL_MASK | Gdk.ModifierType.SHIFT_MASK): chain_up = not self._rowUnderMouseSelected(iconview, event) if not chain_up: self._draggedPaths = self.getSelectedPaths() else: self._draggedPaths = None Gtk.IconView.do_button_press_event(iconview, event) if self._draggedPaths: for path in self._draggedPaths: self.iconview.select_path(path) self._ignoreRelease = chain_up self.iconview_cursor_pos = self.iconview.get_path_at_pos(event.x, event.y) return True def _iconViewButtonReleaseEventCb(self, iconview, event): control_mask = event.get_state() & Gdk.ModifierType.CONTROL_MASK shift_mask = event.get_state() & Gdk.ModifierType.SHIFT_MASK modifier_active = control_mask or shift_mask if not modifier_active and self.iconview_cursor_pos: current_cursor_pos = self.iconview.get_path_at_pos(event.x, event.y) if current_cursor_pos == self.iconview_cursor_pos: if iconview.path_is_selected(current_cursor_pos): iconview.unselect_all() iconview.select_path(current_cursor_pos) def _newProjectCreatedCb(self, app, project): if not self._project is project: self._project = project self._resetErrorList() self.storemodel.clear() self._connectToProject(project) def _newProjectLoadedCb(self, unused_pitivi, project): if not self._project is project: self._project = project self.storemodel.clear() self._connectToProject(project) # Make sure that the sources added to the project are added added self.flush_pending_rows() def _newProjectFailedCb(self, unused_pitivi, unused_reason, unused_uri): self.storemodel.clear() self.project_signals.disconnectAll() self._project = None def _addUris(self, uris): if self.app.current: self.app.current.addUris(uris) else: self.warning("Adding uris to project, but the project has changed in the meantime") return False ## Drag and Drop def _dndDataReceivedCb(self, unused_widget, unused_context, unused_x, unused_y, selection, targettype, unused_time): self.debug("targettype:%d, selection.data:%r", targettype, selection.get_data()) directories = [] remote_files = [] filenames = [] uris = selection.get_data().split("\r\n") uris = filter(lambda x: x != "", uris) for uri in uris: uri = unquote(uri.strip('\x00')) if os.path.isfile(uri[7:]): filenames.append(uri) elif os.path.isdir(uri[7:]): directories.append(uri) elif "://" in uri: #FIXME Very dubious check. remote_files.append(uri) if len(directories): # Recursively import from folders that were dragged into the library self.app.threads.addThread(PathWalker, directories, self._addUris) if len(remote_files): #TODO waiting for remote files downloader support to be implemented self.fixme("Importing remote files is not implemented") if len(filenames): self.app.current.addUris(filenames) #used with TreeView and IconView def _dndDragBeginCb(self, unused_view, context): self.info("Drag operation begun") self.dragged = True paths = self.getSelectedPaths() if len(paths) < 1: context.drag_abort(int(time.time())) else: row = self.modelFilter[paths[0]] Gtk.drag_set_icon_pixbuf(context, row[COL_ICON], 0, 0) def _dndDragEndCb(self, unused_view, context): self.info("Drag operation ended") self.dragged = False def getSelectedPaths(self): """ Returns a list of selected treeview or iconview items """ if self.clip_view == SHOW_TREEVIEW: return self._getSelectedPathsTreeView() elif self.clip_view == SHOW_ICONVIEW: return self._getSelectedPathsIconView() def _getSelectedPathsTreeView(self): model, rows = self.treeview.get_selection().get_selected_rows() return rows def _getSelectedPathsIconView(self): paths = self.iconview.get_selected_items() paths.reverse() return paths def getSelectedItems(self): """ Returns a list of selected items URIs """ if self._draggedPaths: return [self.modelFilter[path][COL_URI] for path in self._draggedPaths] return [self.modelFilter[path][COL_URI] for path in self.getSelectedPaths()] def getSelectedAssets(self): """ Returns a list of selected items URIs """ if self._draggedPaths: return [self.modelFilter[path][COL_ASSET] for path in self._draggedPaths] return [self.modelFilter[path][COL_ASSET] for path in self.getSelectedPaths()]
class MediaLibraryWidget(Gtk.VBox, Loggable): """ Widget for listing sources """ __gsignals__ = { 'play': (GObject.SignalFlags.RUN_LAST, None, (GObject.TYPE_PYOBJECT,))} def __init__(self, instance, uiman): Gtk.VBox.__init__(self) Loggable.__init__(self) self.pending_rows = [] self.app = instance self._errors = [] self._project = None self._draggedPaths = None self.dragged = False self.clip_view = self.app.settings.lastClipView builder = Gtk.Builder() builder.add_from_file(os.path.join(get_ui_dir(), "medialibrary.ui")) builder.connect_signals(self) self._welcome_infobar = builder.get_object("welcome_infobar") self._import_warning_infobar = builder.get_object("warning_infobar") self._warning_label = builder.get_object("warning_label") self._view_error_button = builder.get_object("view_error_button") toolbar = builder.get_object("medialibrary_toolbar") toolbar.get_style_context().add_class("inline-toolbar") self._remove_button = builder.get_object("media_remove_button") self._clipprops_button = builder.get_object("media_props_button") self._insert_button = builder.get_object("media_insert_button") self._listview_button = builder.get_object("media_listview_button") searchEntry = builder.get_object("media_search_entry") # Store self.storemodel = Gtk.ListStore(*STORE_MODEL_STRUCTURE) self.storemodel.set_sort_func(COL_URI, compare_simple, user_data=COL_URI) # Prefer to sort the media library elements by URI # rather than show them randomly. self.storemodel.set_sort_column_id(COL_URI, Gtk.SortType.ASCENDING) # Scrolled Windows self.treeview_scrollwin = Gtk.ScrolledWindow() self.treeview_scrollwin.set_policy(Gtk.PolicyType.NEVER, Gtk.PolicyType.AUTOMATIC) self.treeview_scrollwin.set_shadow_type(Gtk.ShadowType.ETCHED_IN) self.iconview_scrollwin = Gtk.ScrolledWindow() self.iconview_scrollwin.set_policy(Gtk.PolicyType.AUTOMATIC, Gtk.PolicyType.AUTOMATIC) self.iconview_scrollwin.set_shadow_type(Gtk.ShadowType.ETCHED_IN) # import sources dialogbox self._importDialog = None # Filtering model for the search box. # Use this instead of using self.storemodel directly self.modelFilter = self.storemodel.filter_new() self.modelFilter.set_visible_func(self._setRowVisible, data=searchEntry) # TreeView # Displays icon, name, type, length self.treeview = Gtk.TreeView(self.modelFilter) self.treeview_scrollwin.add(self.treeview) self.treeview.connect("button-press-event", self._treeViewButtonPressEventCb) self.treeview.connect("button-release-event", self._treeViewButtonReleaseEventCb) self.treeview.connect("focus-in-event", self._disableKeyboardShortcutsCb) self.treeview.connect("focus-out-event", self._enableKeyboardShortcutsCb) self.treeview.connect("row-activated", self._itemOrRowActivatedCb) self.treeview.set_property("rules_hint", True) self.treeview.set_headers_visible(False) self.treeview.set_property("search_column", COL_SEARCH_TEXT) tsel = self.treeview.get_selection() tsel.set_mode(Gtk.SelectionMode.MULTIPLE) tsel.connect("changed", self._viewSelectionChangedCb) pixbufcol = Gtk.TreeViewColumn(_("Icon")) pixbufcol.set_expand(False) pixbufcol.set_spacing(SPACING) self.treeview.append_column(pixbufcol) pixcell = Gtk.CellRendererPixbuf() pixcell.props.xpad = 6 pixbufcol.pack_start(pixcell, True) pixbufcol.add_attribute(pixcell, 'pixbuf', COL_ICON) namecol = Gtk.TreeViewColumn(_("Information")) self.treeview.append_column(namecol) namecol.set_expand(True) namecol.set_spacing(SPACING) namecol.set_sizing(Gtk.TreeViewColumnSizing.GROW_ONLY) namecol.set_min_width(150) txtcell = Gtk.CellRendererText() txtcell.set_property("ellipsize", Pango.EllipsizeMode.END) namecol.pack_start(txtcell, True) namecol.add_attribute(txtcell, "markup", COL_INFOTEXT) namecol = Gtk.TreeViewColumn(_("Duration")) namecol.set_expand(False) self.treeview.append_column(namecol) txtcell = Gtk.CellRendererText() txtcell.set_property("yalign", 0.0) namecol.pack_start(txtcell, True) namecol.add_attribute(txtcell, "markup", COL_LENGTH) # IconView self.iconview = Gtk.IconView(self.modelFilter) self.iconview_scrollwin.add(self.iconview) self.iconview.connect("button-press-event", self._iconViewButtonPressEventCb) self.iconview.connect("button-release-event", self._iconViewButtonReleaseEventCb) self.iconview.connect("focus-in-event", self._disableKeyboardShortcutsCb) self.iconview.connect("focus-out-event", self._enableKeyboardShortcutsCb) self.iconview.connect("item-activated", self._itemOrRowActivatedCb) self.iconview.connect("selection-changed", self._viewSelectionChangedCb) self.iconview.set_item_orientation(Gtk.Orientation.VERTICAL) self.iconview.set_property("has_tooltip", True) self.iconview.set_tooltip_column(COL_INFOTEXT) self.iconview.props.item_padding = 3 self.iconview.props.margin = 3 self.iconview_cursor_pos = None cell = Gtk.CellRendererPixbuf() self.iconview.pack_start(cell, False) self.iconview.add_attribute(cell, "pixbuf", COL_ICON_LARGE) cell = Gtk.CellRendererText() cell.props.alignment = Pango.Alignment.CENTER cell.props.xalign = 0.5 cell.props.yalign = 0.0 cell.props.xpad = 0 cell.props.ypad = 0 cell.set_property("ellipsize", Pango.EllipsizeMode.END) self.iconview.pack_start(cell, False) self.iconview.add_attribute(cell, "markup", COL_SEARCH_TEXT) self.iconview.set_selection_mode(Gtk.SelectionMode.MULTIPLE) # The _progressbar that shows up when importing clips self._progressbar = Gtk.ProgressBar() self._progressbar.set_show_text(True) # Connect to project. We must remove and reset the callbacks when # changing project. self.project_signals = SignalGroup() self.app.connect("new-project-created", self._newProjectCreatedCb) self.app.connect("new-project-loaded", self._newProjectLoadedCb) self.app.connect("new-project-failed", self._newProjectFailedCb) # default pixbufs self.audiofilepixbuf = self._getIcon("audio-x-generic") self.videofilepixbuf = self._getIcon("video-x-generic") # Drag and Drop self.drag_dest_set(Gtk.DestDefaults.DROP | Gtk.DestDefaults.MOTION, [dnd.URI_TARGET_ENTRY, dnd.FILE_TARGET_ENTRY], Gdk.DragAction.COPY) self.drag_dest_add_uri_targets() self.connect("drag_data_received", self._dndDataReceivedCb) self._setup_view_for_drag_and_drop(self.treeview, [("pitivi/file-source", 0, TYPE_PITIVI_FILESOURCE)]) self._setup_view_for_drag_and_drop(self.iconview, [Gtk.TargetEntry.new("pitivi/file-source", 0, TYPE_PITIVI_FILESOURCE)]) # Hack so that the views have the same method as self self.treeview.getSelectedItems = self.getSelectedItems # Keyboard shortcuts for some items in the gtkbuilder file selection_actions = ( ("RemoveSources", Gtk.STOCK_DELETE, _("_Remove from Project"), "<Control>Delete", None, self._removeSourcesCb), ("InsertEnd", Gtk.STOCK_COPY, _("Insert at _End of Timeline"), "Insert", None, self._insertEndCb), ) self.selection_actions = Gtk.ActionGroup("medialibraryselection") self.selection_actions.add_actions(selection_actions) self.selection_actions.set_sensitive(False) uiman.insert_action_group(self.selection_actions, 0) uiman.add_ui_from_string(ui) # Set the state of the view mode toggle button. self._listview_button.set_active(self.clip_view == SHOW_TREEVIEW) # Make sure the proper view is displayed. self._displayClipView() # Add all the child widgets. self.pack_start(toolbar, False, False, 0) self.pack_start(self._welcome_infobar, False, False, 0) self.pack_start(self._import_warning_infobar, False, False, 0) self.pack_start(self.iconview_scrollwin, True, True, 0) self.pack_start(self.treeview_scrollwin, True, True, 0) self.pack_start(self._progressbar, False, True, 0) def getAssetForUri(self, uri): # Sanitization uri = filter(lambda c: c != '\n' and c != '\r', uri) for path in self.modelFilter: asset = path[COL_ASSET] info = asset.get_info() asset_uri = info.get_uri() if asset_uri == uri: return asset def _setup_view_for_drag_and_drop(self, view, target_entries): view.drag_source_set(0, [], Gdk.DragAction.COPY) view.enable_model_drag_source(Gdk.ModifierType.BUTTON1_MASK, target_entries, Gdk.DragAction.COPY) view.drag_source_set_target_list([]) view.drag_source_add_uri_targets() view.drag_source_add_text_targets() view.connect("drag-data-get", self._dndDragDataGetCb) view.connect("drag-begin", self._dndDragBeginCb) view.connect("drag-end", self._dndDragEndCb) def _importSourcesCb(self, unused_action): self.showImportSourcesDialog() def _removeSourcesCb(self, unused_action): self._removeSources() def _insertEndCb(self, unused_action): self.app.gui.timeline_ui.insertEnd(self.getSelectedAssets()) def _disableKeyboardShortcutsCb(self, *unused_args): """ Disable the Delete keyboard shortcut and playback shortcuts to prevent accidents or being unable to type various characters. This is used when focusing the search entry, icon on tree view widgets. """ self.app.gui.setActionsSensitive(False) def _enableKeyboardShortcutsCb(self, *unused_args): """ When focusing out of media library widgets, re-enable the timeline keyboard shortcuts. """ self.app.gui.setActionsSensitive(True) def _trackElementAddedCb(self, source, unused_track_element): """ After an object has been added to the first track, position it correctly and request the next source to be processed. """ timeline = self.app.current.timeline layer = timeline.get_layers()[0] # FIXME Get the longest layer # Handle the case where we just inserted the first clip if len(layer.get_clips()) == 1: source.props.start = 0 else: source.props.start = timeline.props.duration # We only need one TrackElement to estimate the new duration. # Process the next source. source.disconnect_by_func(self._trackElementAddedCb) self._insertNextSource() def _searchEntryChangedCb(self, entry): # With many hundred clips in an iconview with dynamic columns and # ellipsizing, doing needless searches is very expensive. # Realistically, nobody expects to search for only one character, # and skipping that makes a huge difference in responsiveness. if len(entry.get_text()) != 1: self.modelFilter.refilter() def _searchEntryIconClickedCb(self, entry, icon_pos, unused_event): if icon_pos == Gtk.EntryIconPosition.SECONDARY: entry.set_text("") elif icon_pos == Gtk.EntryIconPosition.PRIMARY: self._selectUnusedSources() def _setRowVisible(self, model, iter, data): """ Toggle the visibility of a liststore row. Used for the search box. """ text = data.get_text().lower() if text == "": return True # Avoid silly warnings else: # We must convert to markup form to be able to search for &, ', etc. text = GLib.markup_escape_text(text) return text in model.get_value(iter, COL_INFOTEXT).lower() def _getIcon(self, iconname, alternate=None): icontheme = Gtk.IconTheme.get_default() pixdir = get_pixmap_dir() icon = None try: icon = icontheme.load_icon(iconname, 48, 0) except: # empty except clause is bad but load_icon raises Gio.Error. # Right, *gio*. if alternate: icon = GdkPixbuf.Pixbuf.new_from_file(os.path.join(pixdir, alternate)) else: icon = icontheme.load_icon("dialog-question", 48, 0) return icon def _connectToProject(self, project): """Connect signal handlers to a project. This first disconnects any handlers connected to an old project. If project is None, this just disconnects any connected handlers. """ self.project_signals.connect(project, "asset-added", None, self._assetAddedCb) self.project_signals.connect(project, "asset-removed", None, self._assetRemovedCb) self.project_signals.connect(project, "error-loading-asset", None, self._errorCreatingAssetCb) self.project_signals.connect(project, "done-importing", None, self._sourcesStoppedImportingCb) self.project_signals.connect(project, "start-importing", None, self._sourcesStartedImportingCb) # The start-importing signal would have already been emited at that # time, make sure to catch if it is the case if project.nb_remaining_file_to_import > 0: self._sourcesStartedImportingCb(project) def _setClipView(self, view_type): """ Set which clip view to use when medialibrary is showing clips. view_type: one of SHOW_TREEVIEW or SHOW_ICONVIEW """ self.app.settings.lastClipView = view_type # Gather some info before switching views paths = self.getSelectedPaths() self._viewUnselectAll() # Now that we've got all the info, we can actually change the view type self.clip_view = view_type self._displayClipView() for path in paths: self._viewSelectPath(path) def _displayClipView(self): if self.clip_view == SHOW_TREEVIEW: self.iconview_scrollwin.hide() self.treeview_scrollwin.show_all() elif self.clip_view == SHOW_ICONVIEW: self.treeview_scrollwin.hide() self.iconview_scrollwin.show_all() if not len(self.storemodel): self._welcome_infobar.show_all() def showImportSourcesDialog(self): """Pop up the "Import Sources" dialog box""" if self._importDialog: return chooser_action = Gtk.FileChooserAction.OPEN dialogtitle = _("Select One or More Files") close_after = Gtk.CheckButton(_("Close after importing files")) close_after.set_active(self.app.settings.closeImportDialog) self._importDialog = Gtk.FileChooserDialog(dialogtitle, None, chooser_action, (Gtk.STOCK_CLOSE, Gtk.ResponseType.CLOSE, Gtk.STOCK_ADD, Gtk.ResponseType.OK)) self._importDialog.set_icon_name("pitivi") self._importDialog.props.extra_widget = close_after self._importDialog.set_default_response(Gtk.ResponseType.OK) self._importDialog.set_select_multiple(True) self._importDialog.set_modal(False) self._importDialog.set_current_folder(self.app.settings.lastImportFolder) self._importDialog.connect('response', self._dialogBoxResponseCb) self._importDialog.connect('close', self._dialogBoxCloseCb) pw = PreviewWidget(self.app) self._importDialog.set_preview_widget(pw) self._importDialog.set_use_preview_label(False) self._importDialog.connect('update-preview', pw.add_preview_request) # Filter for the "known good" formats by default filt_supported = Gtk.FileFilter() filt_known = Gtk.FileFilter() filt_supported.set_name(_("Supported file formats")) for category in SUPPORTED_FILE_FORMATS: # Category can be "video", "audio", "image", "application" for mime in SUPPORTED_FILE_FORMATS[category]: filt_supported.add_mime_type(category + "/" + mime) filt_known.add_mime_type(category + "/" + mime) # Also allow showing known but not reliable demuxers filt_known.set_name(_("All known file formats")) for fullmime in OTHER_KNOWN_FORMATS: filt_known.add_mime_type(fullmime) # ...and allow the user to override our whitelists default = Gtk.FileFilter() default.set_name(_("All files")) default.add_pattern("*") self._importDialog.add_filter(filt_supported) self._importDialog.add_filter(filt_known) self._importDialog.add_filter(default) self._importDialog.show() def _updateProgressbar(self): """ Update the _progressbar with the ratio of clips imported vs the total """ current_clip_iter = self.app.current.nb_imported_files total_clips = self.app.current.nb_remaining_file_to_import + current_clip_iter progressbar_text = _("Importing clip %(current_clip)d of %(total)d" % {"current_clip": current_clip_iter, "total": total_clips}) self._progressbar.set_text(progressbar_text) if current_clip_iter == 0: self._progressbar.set_fraction(0.0) elif total_clips != 0: self._progressbar.set_fraction((current_clip_iter - 1) / float(total_clips)) def _getThumbnailInDir(self, dir, hash): thumb_path_normal = dir + "normal/" + hash + ".png" try: thumbnail = GdkPixbuf.Pixbuf.new_from_file(thumb_path_normal) thumbnail_large = thumbnail thumbnail_height = int(thumbnail.get_height() / 2) thumbnail = thumbnail.scale_simple(64, thumbnail_height, GdkPixbuf.InterpType.BILINEAR) return thumbnail, thumbnail_large except GLib.GError: return None, None def _addAsset(self, asset): info = asset.get_info() # The code below tries to read existing thumbnails from the freedesktop # thumbnails directory (~/.thumbnails). The filenames are simply # the file URI hashed with md5, so we can retrieve them easily. video_streams = [i for i in info.get_stream_list() if isinstance(i, DiscovererVideoInfo)] if len(video_streams) > 0: # From the freedesktop spec: "if the environment variable # $XDG_CACHE_HOME is set and not blank then the directory # $XDG_CACHE_HOME/thumbnails will be used, otherwise # $HOME/.cache/thumbnails will be used." # Older version of the spec also mentioned $HOME/.thumbnails thumbnail_hash = md5(quote_uri(info.get_uri())).hexdigest() try: thumb_dir = os.environ['XDG_CACHE_HOME'] thumbnail, thumbnail_large = self._getThumbnailInDir(thumb_dir, thumbnail_hash) except KeyError: thumbnail, thumbnail_large = (None, None) if thumbnail is None: thumb_dir = os.path.expanduser("~/.cache/thumbnails/") thumbnail, thumbnail_large = self._getThumbnailInDir(thumb_dir, thumbnail_hash) if thumbnail is None: thumb_dir = os.path.expanduser("~/.thumbnails/") thumbnail, thumbnail_large = self._getThumbnailInDir(thumb_dir, thumbnail_hash) if thumbnail is None: thumbnail = self.videofilepixbuf # TODO gst discoverer should create missing thumbnails. thumbnail_large = thumbnail else: thumbnail = self.audiofilepixbuf thumbnail_large = self.audiofilepixbuf if info.get_duration() == Gst.CLOCK_TIME_NONE: duration = '' else: duration = beautify_length(info.get_duration()) name = info_name(info) self.pending_rows.append((thumbnail, thumbnail_large, beautify_info(info), asset, info.get_uri(), duration, name)) if len(self.pending_rows) > 50: self.flush_pending_rows() def flush_pending_rows(self): for row in self.pending_rows: self.storemodel.append(row) del self.pending_rows[:] # medialibrary callbacks def _assetAddedCb(self, unused_project, asset, current_clip_iter=None, total_clips=None): """ a file was added to the medialibrary """ if isinstance(asset, GES.UriClipAsset): self._updateProgressbar() self._addAsset(asset) def _assetRemovedCb(self, unsued_project, asset): """ the given uri was removed from the medialibrary """ # find the good line in the storemodel and remove it model = self.storemodel uri = asset.get_id() for row in model: if uri == row[COL_URI]: model.remove(row.iter) break if not len(model): self._welcome_infobar.show_all() self.debug("Removing %s", uri) def _errorCreatingAssetCb(self, unsued_project, error, id, type): """ The given uri isn't a media file """ if GObject.type_is_a(type, GES.UriClip): error = (id, str(error.domain), error) self._errors.append(error) self._updateProgressbar() def _sourcesStartedImportingCb(self, unsued_project): self.import_start_time = time.time() self._welcome_infobar.hide() self._progressbar.show() def _sourcesStoppedImportingCb(self, unsued_project): self.debug("Importing took %.3f seconds" % (time.time() - self.import_start_time)) self.flush_pending_rows() self._progressbar.hide() if self._errors: if len(self._errors) > 1: self._warning_label.set_text(_("Errors occurred while importing.")) self._view_error_button.set_label(_("View errors")) else: self._warning_label.set_text(_("An error occurred while importing.")) self._view_error_button.set_label(_("View error")) self._import_warning_infobar.show_all() ## Error Dialog Box callbacks def _errorDialogBoxCloseCb(self, unused_dialog): self._error_dialogbox.destroy() self._error_dialogbox = None def _errorDialogBoxResponseCb(self, unused_dialog, unused_response): self._error_dialogbox.destroy() self._error_dialogbox = None ## Import Sources Dialog Box callbacks def _dialogBoxResponseCb(self, dialogbox, response): self.debug("response:%r", response) if response == Gtk.ResponseType.OK: lastfolder = dialogbox.get_current_folder() self.app.settings.lastImportFolder = lastfolder self.app.settings.closeImportDialog = \ dialogbox.props.extra_widget.get_active() filenames = dialogbox.get_uris() self.app.current.addUris(filenames) if self.app.settings.closeImportDialog: dialogbox.destroy() self._importDialog = None else: dialogbox.destroy() self._importDialog = None def _dialogBoxCloseCb(self, unused_dialogbox): self.debug("closing") self._importDialog = None def _removeSources(self): """ Determine which clips are selected in the icon or list view, and ask MediaLibrary to remove them from the project. """ model = self.treeview.get_model() paths = self.getSelectedPaths() if paths is None or paths < 1: return # use row references so we don't have to care if a path has been removed rows = [] for path in paths: row = Gtk.TreeRowReference.new(model, path) rows.append(row) self.app.action_log.begin("remove clip from source list") for row in rows: asset = model[row.get_path()][COL_ASSET] self.app.current.remove_asset(asset) self.app.action_log.commit() def _sourceIsUsed(self, asset): """Check if a given URI is present in the timeline""" layers = self.app.current.timeline.get_layers() for layer in layers: for clip in layer.get_clips(): if clip.get_asset() == asset: return True return False def _selectUnusedSources(self): """ Select, in the media library, unused sources in the project. """ assets = self.app.current.list_assets(GES.UriClip) unused_sources_uris = [] model = self.treeview.get_model() selection = self.treeview.get_selection() for asset in assets: if not self._sourceIsUsed(asset): unused_sources_uris.append(asset.get_id()) # Hack around the fact that making selections (in a treeview/iconview) # deselects what was previously selected if self.clip_view == SHOW_TREEVIEW: self.treeview.get_selection().select_all() elif self.clip_view == SHOW_ICONVIEW: self.iconview.select_all() for row in model: if row[COL_URI] not in unused_sources_uris: if self.clip_view == SHOW_TREEVIEW: selection.unselect_iter(row.iter) else: self.iconview.unselect_path(row.path) ## UI callbacks def _removeClickedCb(self, unused_widget=None): """ Called when a user clicks on the remove button """ self._removeSources() def _clipPropertiesCb(self, unused_widget=None): """ Show the clip properties (resolution, framerate, audio channels...) and allow setting them as the new project settings. """ paths = self.getSelectedPaths()[0] # Only use the first item model = self.treeview.get_model() info = model[paths][COL_ASSET].get_info() d = clipmediapropsDialog(self.app.current, info.get_audio_streams(), info.get_video_streams()) d.run() def _warningInfoBarDismissedCb(self, unused_button): self._resetErrorList() def _resetErrorList(self): self._errors = [] self._import_warning_infobar.hide() def _viewErrorsButtonClickedCb(self, unused_button): """ Show a FileListErrorDialog to display import _errors. """ if len(self._errors) > 1: msgs = (_("Error while analyzing files"), _("The following files can not be used with PiTiVi.")) else: msgs = (_("Error while analyzing a file"), _("The following file can not be used with PiTiVi.")) self._error_dialogbox = FileListErrorDialog(*msgs) self._error_dialogbox.connect("close", self._errorDialogBoxCloseCb) self._error_dialogbox.connect("response", self._errorDialogBoxResponseCb) for uri, reason, extra in self._errors: self._error_dialogbox.addFailedFile(uri, reason, extra) self._error_dialogbox.window.show() # Reset the error list, since the user has read them. self._resetErrorList() def _toggleViewTypeCb(self, widget): if widget.get_active(): self._setClipView(SHOW_TREEVIEW) else: self._setClipView(SHOW_ICONVIEW) def _rowUnderMouseSelected(self, view, event): if isinstance(view, Gtk.TreeView): path = None tup = view.get_path_at_pos(int(event.x), int(event.y)) if tup: path, column, x, y = tup if path: selection = view.get_selection() return selection.path_is_selected(path) and selection.count_selected_rows() > 0 elif isinstance(view, Gtk.IconView): path = view.get_path_at_pos(int(event.x), int(event.y)) if path: selection = view.get_selected_items() return view.path_is_selected(path) and len(selection) else: assert False return False def _nothingUnderMouse(self, view, event): return not bool(view.get_path_at_pos(int(event.x), int(event.y))) def _viewGetFirstSelected(self): paths = self.getSelectedPaths() return paths[0] def _viewHasSelection(self): paths = self.getSelectedPaths() return bool(len(paths)) def _viewGetPathAtPos(self, event): if self.clip_view == SHOW_TREEVIEW: pathinfo = self.treeview.get_path_at_pos(int(event.x), int(event.y)) return pathinfo[0] elif self.clip_view == SHOW_ICONVIEW: return self.iconview.get_path_at_pos(int(event.x), int(event.y)) def _viewSelectPath(self, path): if self.clip_view == SHOW_TREEVIEW: selection = self.treeview.get_selection() selection.select_path(path) elif self.clip_view == SHOW_ICONVIEW: self.iconview.select_path(path) def _viewUnselectAll(self): if self.clip_view == SHOW_TREEVIEW: selection = self.treeview.get_selection() selection.unselect_all() elif self.clip_view == SHOW_ICONVIEW: self.iconview.unselect_all() def _treeViewButtonPressEventCb(self, treeview, event): chain_up = True if event.type == getattr(Gdk.EventType, '2BUTTON_PRESS'): # It is possible to double-click outside of clips: if self.getSelectedPaths() != []: # Here we used to emit "play", but # this is now handled by _itemOrRowActivatedCb instead. pass chain_up = False elif not event.get_state() & (Gdk.ModifierType.CONTROL_MASK | Gdk.ModifierType.SHIFT_MASK): chain_up = not self._rowUnderMouseSelected(treeview, event) if not chain_up: self._draggedPaths = self.getSelectedPaths() else: self._draggedPaths = None Gtk.TreeView.do_button_press_event(treeview, event) ts = self.treeview.get_selection() if self._draggedPaths: for path in self._draggedPaths: ts.select_path(path) return True def _treeViewButtonReleaseEventCb(self, treeview, event): ts = self.treeview.get_selection() state = event.get_state() & (Gdk.ModifierType.CONTROL_MASK | Gdk.ModifierType.SHIFT_MASK) path = self.treeview.get_path_at_pos(event.x, event.y) if not state and not self.dragged: ts.unselect_all() if path: ts.select_path(path[0]) def _viewSelectionChangedCb(self, unused): selected_items = len(self.getSelectedPaths()) if selected_items: self.selection_actions.set_sensitive(True) self._remove_button.set_sensitive(True) self._insert_button.set_sensitive(True) # Some actions can only be done on a single item at a time: self._clipprops_button.set_sensitive(False) if selected_items == 1: self._clipprops_button.set_sensitive(True) else: self.selection_actions.set_sensitive(False) self._remove_button.set_sensitive(False) self._insert_button.set_sensitive(False) self._clipprops_button.set_sensitive(False) def _itemOrRowActivatedCb(self, unused_view, path, *unused_column): """ When an item is double-clicked, or Space, Shift+Space, Return or Enter is pressed, preview the clip. This method is the same for both iconview and treeview. """ asset = self.modelFilter[path][COL_ASSET] self.emit('play', asset) def _iconViewButtonPressEventCb(self, iconview, event): chain_up = True if event.type == getattr(Gdk.EventType, '2BUTTON_PRESS'): # It is possible to double-click outside of clips: if self.getSelectedPaths() != []: # Here we used to emit "play", but # this is now handled by _itemOrRowActivatedCb instead. pass chain_up = False elif not event.get_state() & (Gdk.ModifierType.CONTROL_MASK | Gdk.ModifierType.SHIFT_MASK): chain_up = not self._rowUnderMouseSelected(iconview, event) if not chain_up: self._draggedPaths = self.getSelectedPaths() else: self._draggedPaths = None Gtk.IconView.do_button_press_event(iconview, event) if self._draggedPaths: for path in self._draggedPaths: self.iconview.select_path(path) self._ignoreRelease = chain_up self.iconview_cursor_pos = self.iconview.get_path_at_pos(event.x, event.y) return True def _iconViewButtonReleaseEventCb(self, iconview, event): control_mask = event.get_state() & Gdk.ModifierType.CONTROL_MASK shift_mask = event.get_state() & Gdk.ModifierType.SHIFT_MASK modifier_active = control_mask or shift_mask if not modifier_active and self.iconview_cursor_pos: current_cursor_pos = self.iconview.get_path_at_pos(event.x, event.y) if current_cursor_pos == self.iconview_cursor_pos: if iconview.path_is_selected(current_cursor_pos): iconview.unselect_all() iconview.select_path(current_cursor_pos) def _newProjectCreatedCb(self, app, project): if not self._project is project: self._project = project self._resetErrorList() self.storemodel.clear() self._connectToProject(project) def _newProjectLoadedCb(self, unused_pitivi, project): if not self._project is project: self._project = project self.storemodel.clear() self._connectToProject(project) # Make sure that the sources added to the project are added added self.flush_pending_rows() def _newProjectFailedCb(self, unused_pitivi, unused_reason, unused_uri): self.storemodel.clear() self.project_signals.disconnectAll() self._project = None def _addUris(self, uris): if self.app.current: self.app.current.addUris(uris) else: self.warning("Adding uris to project, but the project has changed in the meantime") return False ## Drag and Drop def _dndDataReceivedCb(self, unused_widget, unused_context, unused_x, unused_y, selection, targettype, unused_time): self.debug("targettype:%d, selection.data:%r", targettype, selection.get_data()) directories = [] remote_files = [] filenames = [] uris = selection.get_data().split("\r\n") uris = filter(lambda x: x != "", uris) for uri in uris: uri = unquote(uri.strip('\x00')) if os.path.isfile(uri[7:]): filenames.append(uri) elif os.path.isdir(uri[7:]): directories.append(uri) elif "://" in uri: #FIXME Very dubious check. remote_files.append(uri) if len(directories): # Recursively import from folders that were dragged into the library self.app.threads.addThread(PathWalker, directories, self._addUris) if len(remote_files): #TODO waiting for remote files downloader support to be implemented self.fixme("Importing remote files is not implemented") if len(filenames): self.app.current.addUris(filenames) #used with TreeView and IconView def _dndDragDataGetCb(self, unused_view, context, data, info, timestamp): paths = self.getSelectedPaths() uris = [self.modelFilter[path][COL_URI] for path in paths] data.set_uris(uris) def _dndDragBeginCb(self, unused_view, context): self.info("Drag operation begun") self.dragged = True paths = self.getSelectedPaths() if len(paths) < 1: context.drag_abort(int(time.time())) else: row = self.modelFilter[paths[0]] Gtk.drag_set_icon_pixbuf(context, row[COL_ICON], 0, 0) def _dndDragEndCb(self, unused_view, context): self.info("Drag operation ended") self.dragged = False def getSelectedPaths(self): """ Returns a list of selected treeview or iconview items """ if self.clip_view == SHOW_TREEVIEW: return self._getSelectedPathsTreeView() elif self.clip_view == SHOW_ICONVIEW: return self._getSelectedPathsIconView() def _getSelectedPathsTreeView(self): model, rows = self.treeview.get_selection().get_selected_rows() return rows def _getSelectedPathsIconView(self): paths = self.iconview.get_selected_items() paths.reverse() return paths def getSelectedItems(self): """ Returns a list of selected items URIs """ if self._draggedPaths: return [self.modelFilter[path][COL_URI] for path in self._draggedPaths] return [self.modelFilter[path][COL_URI] for path in self.getSelectedPaths()] def getSelectedAssets(self): """ Returns a list of selected items URIs """ if self._draggedPaths: return [self.modelFilter[path][COL_ASSET] for path in self._draggedPaths] return [self.modelFilter[path][COL_ASSET] for path in self.getSelectedPaths()]