RepoMirrors
/
hydrus
mirror of https://github.com/hydrusnetwork/hydrus.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
hydrus/hydrus/client/gui/canvas/ClientGUICanvas.py

4441 lines
148 KiB
Python
Raw Blame History

import typing
from qtpy import QtCore as QC
from qtpy import QtWidgets as QW
from qtpy import QtGui as QG
from hydrus.core import HydrusConstants as HC
from hydrus.core import HydrusData
from hydrus.core import HydrusExceptions
from hydrus.core import HydrusGlobals as HG
from hydrus.core import HydrusPaths
from hydrus.core import HydrusTags
from hydrus.client import ClientApplicationCommand as CAC
from hydrus.client import ClientConstants as CC
from hydrus.client import ClientData
from hydrus.client import ClientDuplicates
from hydrus.client import ClientLocation
from hydrus.client import ClientPaths
from hydrus.client import ClientSearch
from hydrus.client.gui import ClientGUICore as CGC
from hydrus.client.gui import ClientGUIDialogs
from hydrus.client.gui import ClientGUIDialogsManage
from hydrus.client.gui import ClientGUIDialogsQuick
from hydrus.client.gui import ClientGUIDuplicates
from hydrus.client.gui import ClientGUIFunctions
from hydrus.client.gui import ClientGUIMedia
from hydrus.client.gui import ClientGUIMediaActions
from hydrus.client.gui import ClientGUIMediaControls
from hydrus.client.gui import ClientGUIMediaMenus
from hydrus.client.gui import ClientGUIMenus
from hydrus.client.gui import ClientGUIRatings
from hydrus.client.gui import ClientGUIScrolledPanelsEdit
from hydrus.client.gui import ClientGUIScrolledPanelsManagement
from hydrus.client.gui import ClientGUIShortcuts
from hydrus.client.gui import ClientGUITags
from hydrus.client.gui import ClientGUITopLevelWindowsPanels
from hydrus.client.gui import QtPorting as QP
from hydrus.client.gui.canvas import ClientGUICanvasHoverFrames
from hydrus.client.gui.canvas import ClientGUICanvasMedia
from hydrus.client.media import ClientMedia
from hydrus.client.metadata import ClientRatings
from hydrus.client.metadata import ClientTags
from hydrus.client.metadata import ClientTagSorting
def AddAudioVolumeMenu( menu, canvas_type ):
mute_volume_type = None
volume_volume_type = ClientGUIMediaControls.AUDIO_GLOBAL
if canvas_type == CC.CANVAS_MEDIA_VIEWER:
mute_volume_type = ClientGUIMediaControls.AUDIO_MEDIA_VIEWER
if HG.client_controller.new_options.GetBoolean( 'media_viewer_uses_its_own_audio_volume' ):
volume_volume_type = ClientGUIMediaControls.AUDIO_MEDIA_VIEWER
elif canvas_type == CC.CANVAS_PREVIEW:
mute_volume_type = ClientGUIMediaControls.AUDIO_PREVIEW
if HG.client_controller.new_options.GetBoolean( 'preview_uses_its_own_audio_volume' ):
volume_volume_type = ClientGUIMediaControls.AUDIO_PREVIEW
volume_menu = QW.QMenu( menu )
( global_mute_option_name, global_volume_option_name ) = ClientGUIMediaControls.volume_types_to_option_names[ ClientGUIMediaControls.AUDIO_GLOBAL ]
if HG.client_controller.new_options.GetBoolean( global_mute_option_name ):
label = 'unmute global'
else:
label = 'mute global'
ClientGUIMenus.AppendMenuItem( volume_menu, label, 'Mute/unmute audio.', ClientGUIMediaControls.FlipMute, ClientGUIMediaControls.AUDIO_GLOBAL )
#
if mute_volume_type is not None:
ClientGUIMenus.AppendSeparator( volume_menu )
( mute_option_name, volume_option_name ) = ClientGUIMediaControls.volume_types_to_option_names[ mute_volume_type ]
if HG.client_controller.new_options.GetBoolean( mute_option_name ):
label = 'unmute {}'.format( ClientGUIMediaControls.volume_types_str_lookup[ mute_volume_type ] )
else:
label = 'mute {}'.format( ClientGUIMediaControls.volume_types_str_lookup[ mute_volume_type ] )
ClientGUIMenus.AppendMenuItem( volume_menu, label, 'Mute/unmute audio.', ClientGUIMediaControls.FlipMute, mute_volume_type )
#
ClientGUIMenus.AppendSeparator( volume_menu )
( mute_option_name, volume_option_name ) = ClientGUIMediaControls.volume_types_to_option_names[ volume_volume_type ]
# 0-100 inclusive
volumes = list( range( 0, 110, 10 ) )
current_volume = HG.client_controller.new_options.GetInteger( volume_option_name )
if current_volume not in volumes:
volumes.append( current_volume )
volumes.sort()
for volume in volumes:
label = 'volume: {}'.format( volume )
if volume == current_volume:
ClientGUIMenus.AppendMenuCheckItem( volume_menu, label, 'Set the volume.', True, ClientGUIMediaControls.ChangeVolume, volume_volume_type, volume )
else:
ClientGUIMenus.AppendMenuItem( volume_menu, label, 'Set the volume.', ClientGUIMediaControls.ChangeVolume, volume_volume_type, volume )
ClientGUIMenus.AppendMenu( menu, volume_menu, 'volume' )
# cribbing from here https://doc.qt.io/qt-5/layout.html#how-to-write-a-custom-layout-manager
# not finished, but a start as I continue to refactor. might want to rename to 'draggable layout' or something too, since it doesn't actually care about media container that much, and instead subclass vboxlayout?
class CanvasLayout( QW.QLayout ):
def __init__( self ):
QW.QLayout.__init__( self )
self._current_drag_delta = QC.QPoint( 0, 0 )
self._layout_items = []
def addItem( self, layout_item: QW.QLayoutItem ) -> None:
self._layout_items.append( layout_item )
def itemAt( self, index: int ):
try:
return self._layout_items[ index ]
except IndexError:
return None
def minimumSize(self) -> QC.QSize:
return self.sizeHint()
def resetDragDelta( self ):
self._current_drag_delta = QC.QPoint( 0, 0 )
def setGeometry( self, rect: QC.QRect ) -> None:
if len( self._layout_items ) == 0:
return
layout_item = self._layout_items[0]
size = self.sizeHint()
# the given rect is the whole canvas?
natural_x = ( rect.width() - size.width() ) // 2
natural_y = ( rect.height() - size.height() ) // 2
topleft = QC.QPoint( natural_x, natural_y ) + self._current_drag_delta
media_container_rect = QC.QRect( topleft, size )
layout_item.setGeometry( media_container_rect )
def sizeHint(self) -> QC.QSize:
if len( self._layout_items ) == 0:
return QC.QSize( 0, 0 )
else:
return self._layout_items[0].sizeHint()
def takeAt( self, index: int ):
layout_item = self.itemAt( index )
if layout_item is None:
return 0
del self._layout_items[ index ]
return layout_item
def updateDragDelta( self, delta: QC.QPoint ):
self._current_drag_delta += delta
class LayoutEventSilencer( QC.QObject ):
def eventFilter( self, watched, event ):
if watched == self.parent() and event.type() == QC.QEvent.LayoutRequest:
return True
return False
class Canvas( QW.QWidget, CAC.ApplicationCommandProcessorMixin ):
CANVAS_TYPE = CC.CANVAS_MEDIA_VIEWER
def __init__( self, parent, location_context: ClientLocation.LocationContext ):
CAC.ApplicationCommandProcessorMixin.__init__( self )
QW.QWidget.__init__( self, parent )
self.setSizePolicy( QW.QSizePolicy.Expanding, QW.QSizePolicy.Expanding )
self._location_context = location_context
self._current_media_start_time = HydrusData.GetNow()
self._new_options = HG.client_controller.new_options
self._canvas_key = HydrusData.GenerateKey()
self._maintain_pan_and_zoom = False
self._service_keys_to_services = {}
self._current_media = None
catch_mouse = True
# once we have catch_mouse full shortcut support for canvases, swap out this out for an option to swallow activating clicks
ignore_activating_mouse_click = catch_mouse and self.CANVAS_TYPE != CC.CANVAS_PREVIEW
self._my_shortcuts_handler = ClientGUIShortcuts.ShortcutsHandler( self, [ 'media', 'media_viewer' ], catch_mouse = catch_mouse, ignore_activating_mouse_click = ignore_activating_mouse_click )
self._layout_silencer = LayoutEventSilencer( self )
self.installEventFilter( self._layout_silencer )
self._click_drag_reporting_filter = MediaContainerDragClickReportingFilter( self )
self.installEventFilter( self._click_drag_reporting_filter )
self._media_container = ClientGUICanvasMedia.MediaContainer( self, self.CANVAS_TYPE, self._click_drag_reporting_filter )
self._last_drag_pos = None
self._current_drag_is_touch = False
self._last_motion_pos = QC.QPoint( 0, 0 )
self._widget_event_filter = QP.WidgetEventFilter( self )
self._media_container.readyForNeighbourPrefetch.connect( self._PrefetchNeighbours )
self._media_container.zoomChanged.connect( self.ZoomChanged )
HG.client_controller.sub( self, 'ZoomIn', 'canvas_zoom_in' )
HG.client_controller.sub( self, 'ZoomOut', 'canvas_zoom_out' )
HG.client_controller.sub( self, 'ZoomSwitch', 'canvas_zoom_switch' )
HG.client_controller.sub( self, 'OpenExternally', 'canvas_open_externally' )
HG.client_controller.sub( self, 'ManageTags', 'canvas_manage_tags' )
HG.client_controller.sub( self, 'update', 'notify_new_colourset' )
def _Archive( self ):
if self._current_media is not None:
HG.client_controller.Write( 'content_updates', { CC.COMBINED_LOCAL_FILE_SERVICE_KEY : [ HydrusData.ContentUpdate( HC.CONTENT_TYPE_FILES, HC.CONTENT_UPDATE_ARCHIVE, ( self._current_media.GetHash(), ) ) ] } )
def _CopyBMPToClipboard( self ):
copied = False
if self._current_media is not None:
if self._current_media.GetMime() in HC.IMAGES:
HG.client_controller.pub( 'clipboard', 'bmp', self._current_media )
copied = True
return copied
def _CopyHashToClipboard( self, hash_type ):
if self._current_media is None:
return
ClientGUIMedia.CopyHashesToClipboard( self, hash_type, [ self._current_media ] )
def _CopyFileToClipboard( self ):
if self._current_media is not None:
client_files_manager = HG.client_controller.client_files_manager
paths = [ client_files_manager.GetFilePath( self._current_media.GetHash(), self._current_media.GetMime() ) ]
HG.client_controller.pub( 'clipboard', 'paths', paths )
def _CopyPathToClipboard( self ):
if self._current_media is not None:
client_files_manager = HG.client_controller.client_files_manager
path = client_files_manager.GetFilePath( self._current_media.GetHash(), self._current_media.GetMime() )
HG.client_controller.pub( 'clipboard', 'text', path )
def _Delete( self, media = None, default_reason = None, file_service_key = None, just_get_jobs = False ):
if media is None:
if self._current_media is None:
return False
media = [ self._current_media ]
if default_reason is None:
default_reason = 'Deleted from Preview or Media Viewer.'
if file_service_key is None:
if len( self._location_context.current_service_keys ) == 1:
( possible_suggested_file_service_key, ) = self._location_context.current_service_keys
if HG.client_controller.services_manager.GetServiceType( possible_suggested_file_service_key ) in HC.SPECIFIC_LOCAL_FILE_SERVICES + ( HC.FILE_REPOSITORY, ):
file_service_key = possible_suggested_file_service_key
try:
( involves_physical_delete, jobs ) = ClientGUIDialogsQuick.GetDeleteFilesJobs( self, media, default_reason, suggested_file_service_key = file_service_key )
except HydrusExceptions.CancelledException:
return False
def do_it( jobs ):
for service_keys_to_content_updates in jobs:
HG.client_controller.WriteSynchronous( 'content_updates', service_keys_to_content_updates )
if just_get_jobs:
return jobs
else:
HG.client_controller.CallToThread( do_it, jobs )
return True
def _DrawBackgroundBitmap( self, painter: QG.QPainter ):
background_colour = self._GetBackgroundColour()
painter.setBackground( QG.QBrush( background_colour ) )
painter.eraseRect( painter.viewport() )
self._DrawBackgroundDetails( painter )
def _DrawBackgroundDetails( self, painter ):
pass
def _GetBackgroundColour( self ):
return self._new_options.GetColour( CC.COLOUR_MEDIA_BACKGROUND )
def _GetIndexString( self ):
return ''
def _Inbox( self ):
if self._current_media is None:
return
HG.client_controller.Write( 'content_updates', { CC.COMBINED_LOCAL_FILE_SERVICE_KEY : [ HydrusData.ContentUpdate( HC.CONTENT_TYPE_FILES, HC.CONTENT_UPDATE_INBOX, ( self._current_media.GetHash(), ) ) ] } )
def _ManageNotes( self, name_to_start_on = None ):
if self._current_media is None:
return
ClientGUIMediaActions.EditFileNotes( self, self._current_media, name_to_start_on = name_to_start_on )
def _ManageRatings( self ):
if self._current_media is None:
return
if len( HG.client_controller.services_manager.GetServices( HC.RATINGS_SERVICES ) ) > 0:
with ClientGUIDialogsManage.DialogManageRatings( self, ( self._current_media, ) ) as dlg:
dlg.exec()
def _ManageTags( self ):
if self._current_media is None:
return
for child in self.children():
if isinstance( child, ClientGUITopLevelWindowsPanels.FrameThatTakesScrollablePanel ):
panel = child.GetPanel()
if isinstance( panel, ClientGUITags.ManageTagsPanel ):
child.activateWindow()
command = CAC.ApplicationCommand.STATICCreateSimpleCommand( CAC.SIMPLE_SET_SEARCH_FOCUS )
panel.ProcessApplicationCommand( command )
return
# take any focus away from hover window, which will mess up window order when it hides due to the new frame
self.setFocus( QC.Qt.OtherFocusReason )
title = 'manage tags'
frame_key = 'manage_tags_frame'
manage_tags = ClientGUITopLevelWindowsPanels.FrameThatTakesScrollablePanel( self, title, frame_key )
panel = ClientGUITags.ManageTagsPanel( manage_tags, self._location_context, ( self._current_media, ), immediate_commit = True, canvas_key = self._canvas_key )
manage_tags.SetPanel( panel )
def _ManageURLs( self ):
if self._current_media is None:
return
title = 'manage known urls'
with ClientGUITopLevelWindowsPanels.DialogManage( self, title ) as dlg:
panel = ClientGUIScrolledPanelsManagement.ManageURLsPanel( dlg, ( self._current_media, ) )
dlg.SetPanel( panel )
dlg.exec()
def _MediaFocusWentToExternalProgram( self ):
if self._current_media is None:
return
mime = self._current_media.GetMime()
if self._current_media.HasDuration():
self._media_container.Pause()
def _OpenExternally( self ):
if self._current_media is None:
return
hash = self._current_media.GetHash()
mime = self._current_media.GetMime()
client_files_manager = HG.client_controller.client_files_manager
path = client_files_manager.GetFilePath( hash, mime )
launch_path = self._new_options.GetMimeLaunch( mime )
HydrusPaths.LaunchFile( path, launch_path )
self._MediaFocusWentToExternalProgram()
def _OpenFileInWebBrowser( self ):
if self._current_media is not None:
hash = self._current_media.GetHash()
mime = self._current_media.GetMime()
client_files_manager = HG.client_controller.client_files_manager
path = client_files_manager.GetFilePath( hash, mime )
ClientPaths.LaunchPathInWebBrowser( path )
self._MediaFocusWentToExternalProgram()
def _OpenFileLocation( self ):
if self._current_media is not None:
hash = self._current_media.GetHash()
mime = self._current_media.GetMime()
client_files_manager = HG.client_controller.client_files_manager
path = client_files_manager.GetFilePath( hash, mime )
HydrusPaths.OpenFileLocation( path )
self._MediaFocusWentToExternalProgram()
def _OpenKnownURL( self ):
if self._current_media is not None:
ClientGUIMedia.DoOpenKnownURLFromShortcut( self, self._current_media )
def _PrefetchNeighbours( self ):
pass
def _SaveCurrentMediaViewTime( self ):
now = HydrusData.GetNow()
view_timestamp = self._current_media_start_time
viewtime_delta = now - self._current_media_start_time
self._current_media_start_time = now
if self._current_media is None:
return
hash = self._current_media.GetHash()
HG.client_controller.file_viewing_stats_manager.FinishViewing( hash, self.CANVAS_TYPE, view_timestamp, viewtime_delta )
def _SeekDeltaCurrentMedia( self, direction, duration_ms ):
if self._current_media is None:
return
self._media_container.SeekDelta( direction, duration_ms )
def _ShowMediaInNewPage( self ):
if self._current_media is None:
return
hash = self._current_media.GetHash()
hashes = { hash }
HG.client_controller.pub( 'new_page_query', self._location_context, initial_hashes = hashes )
def _Undelete( self ):
if self._current_media is None:
return
ClientGUIMediaActions.UndeleteMedia( self, ( self._current_media, ) )
def CleanBeforeDestroy( self ):
self.ClearMedia()
def ClearMedia( self ):
self.SetMedia( None )
def BeginDrag( self ):
point = self.mapFromGlobal( QG.QCursor.pos() )
self._last_drag_pos = point
self._current_drag_is_touch = False
def resizeEvent( self, event ):
my_size = self.size()
if self._current_media is not None:
media_container_size = self._media_container.size()
if my_size != media_container_size:
self._media_container.ZoomReinit()
self._media_container.ResetCenterPosition()
self.EndDrag()
self.update()
def EndDrag( self ):
self._last_drag_pos = None
def FlipActiveCustomShortcutName( self, name ):
self._my_shortcuts_handler.FlipShortcuts( name )
def GetActiveCustomShortcutNames( self ):
return self._my_shortcuts_handler.GetCustomShortcutNames()
def ManageNotes( self, canvas_key, name_to_start_on = None ):
if canvas_key == self._canvas_key:
self._ManageNotes( name_to_start_on = name_to_start_on )
def ManageTags( self, canvas_key ):
if canvas_key == self._canvas_key:
self._ManageTags()
def MouseIsNearAnimationBar( self ):
if self._current_media is None:
return False
else:
return self._media_container.MouseIsNearAnimationBar()
def MouseIsOverMedia( self ):
if self._current_media is None:
return False
else:
media_mouse_pos = self._media_container.mapFromGlobal( QG.QCursor.pos() )
media_rect = self._media_container.rect()
return media_rect.contains( media_mouse_pos )
def OpenExternally( self, canvas_key ):
if self._canvas_key == canvas_key:
self._OpenExternally()
def paintEvent( self, event ):
painter = QG.QPainter( self )
self._DrawBackgroundBitmap( painter )
def PauseMedia( self ):
self._media_container.Pause()
def ProcessApplicationCommand( self, command: CAC.ApplicationCommand ):
command_processed = True
if command.IsSimpleCommand():
action = command.GetSimpleAction()
if action == CAC.SIMPLE_MANAGE_FILE_RATINGS:
self._ManageRatings()
elif action == CAC.SIMPLE_MANAGE_FILE_TAGS:
self._ManageTags()
elif action == CAC.SIMPLE_MANAGE_FILE_URLS:
self._ManageURLs()
elif action == CAC.SIMPLE_MANAGE_FILE_NOTES:
self._ManageNotes()
elif action == CAC.SIMPLE_OPEN_KNOWN_URL:
self._OpenKnownURL()
elif action == CAC.SIMPLE_ARCHIVE_FILE:
self._Archive()
elif action == CAC.SIMPLE_COPY_BMP:
self._CopyBMPToClipboard()
elif action == CAC.SIMPLE_COPY_BMP_OR_FILE_IF_NOT_BMPABLE:
copied = self._CopyBMPToClipboard()
if not copied:
self._CopyFileToClipboard()
elif action == CAC.SIMPLE_COPY_FILE:
self._CopyFileToClipboard()
elif action == CAC.SIMPLE_COPY_PATH:
self._CopyPathToClipboard()
elif action == CAC.SIMPLE_COPY_SHA256_HASH:
self._CopyHashToClipboard( 'sha256' )
elif action == CAC.SIMPLE_COPY_MD5_HASH:
self._CopyHashToClipboard( 'md5' )
elif action == CAC.SIMPLE_COPY_SHA1_HASH:
self._CopyHashToClipboard( 'sha1' )
elif action == CAC.SIMPLE_COPY_SHA512_HASH:
self._CopyHashToClipboard( 'sha512' )
elif action == CAC.SIMPLE_DELETE_FILE:
self._Delete()
elif action == CAC.SIMPLE_UNDELETE_FILE:
self._Undelete()
elif action == CAC.SIMPLE_INBOX_FILE:
self._Inbox()
elif action == CAC.SIMPLE_OPEN_FILE_IN_EXTERNAL_PROGRAM:
self._OpenExternally()
elif action == CAC.SIMPLE_PAN_UP:
self._media_container.DoManualPan( 0, -1 )
elif action == CAC.SIMPLE_PAN_DOWN:
self._media_container.DoManualPan( 0, 1 )
elif action == CAC.SIMPLE_PAN_LEFT:
self._media_container.DoManualPan( -1, 0 )
elif action == CAC.SIMPLE_PAN_RIGHT:
self._media_container.DoManualPan( 1, 0 )
elif action in ( CAC.SIMPLE_PAN_TOP_EDGE, CAC.SIMPLE_PAN_BOTTOM_EDGE, CAC.SIMPLE_PAN_LEFT_EDGE, CAC.SIMPLE_PAN_RIGHT_EDGE, CAC.SIMPLE_PAN_VERTICAL_CENTER, CAC.SIMPLE_PAN_HORIZONTAL_CENTER ):
self._media_container.DoEdgePan( action )
elif action == CAC.SIMPLE_PAUSE_MEDIA:
self._media_container.Pause()
elif action == CAC.SIMPLE_PAUSE_PLAY_MEDIA:
self._media_container.PausePlay()
elif action == CAC.SIMPLE_SHOW_DUPLICATES:
if self._current_media is not None:
hash = self._current_media.GetHash()
duplicate_type = command.GetSimpleData()
ClientGUIMedia.ShowDuplicatesInNewPage( self._location_context, hash, duplicate_type )
elif action == CAC.SIMPLE_DUPLICATE_MEDIA_CLEAR_FOCUSED_FALSE_POSITIVES:
# TODO: when media knows dupe relationships, all these lads here need a media scan for the existence of alternate groups or whatever
# no duplicate group->don't start the process
if self._current_media is not None:
hash = self._current_media.GetHash()
ClientGUIDuplicates.ClearFalsePositives( self, ( hash, ) )
elif action == CAC.SIMPLE_DUPLICATE_MEDIA_CLEAR_FALSE_POSITIVES:
if self._current_media is not None:
hash = self._current_media.GetHash()
ClientGUIDuplicates.ClearFalsePositives( self, ( hash, ) )
elif action == CAC.SIMPLE_DUPLICATE_MEDIA_DISSOLVE_FOCUSED_ALTERNATE_GROUP:
if self._current_media is not None:
hash = self._current_media.GetHash()
ClientGUIDuplicates.DissolveAlternateGroup( self, ( hash, ) )
elif action == CAC.SIMPLE_DUPLICATE_MEDIA_DISSOLVE_ALTERNATE_GROUP:
if self._current_media is not None:
hash = self._current_media.GetHash()
ClientGUIDuplicates.DissolveAlternateGroup( self, ( hash, ) )
elif action == CAC.SIMPLE_DUPLICATE_MEDIA_DISSOLVE_FOCUSED_DUPLICATE_GROUP:
if self._current_media is not None:
hash = self._current_media.GetHash()
ClientGUIDuplicates.DissolveDuplicateGroup( self, ( hash, ) )
elif action == CAC.SIMPLE_DUPLICATE_MEDIA_DISSOLVE_DUPLICATE_GROUP:
if self._current_media is not None:
hash = self._current_media.GetHash()
ClientGUIDuplicates.DissolveDuplicateGroup( self, ( hash, ) )
elif action == CAC.SIMPLE_DUPLICATE_MEDIA_REMOVE_FOCUSED_FROM_ALTERNATE_GROUP:
if self._current_media is not None:
hash = self._current_media.GetHash()
ClientGUIDuplicates.RemoveFromAlternateGroup( self, ( hash, ) )
elif action == CAC.SIMPLE_DUPLICATE_MEDIA_REMOVE_FOCUSED_FROM_DUPLICATE_GROUP:
if self._current_media is not None:
hash = self._current_media.GetHash()
ClientGUIDuplicates.RemoveFromDuplicateGroup( self, ( hash, ) )
elif action == CAC.SIMPLE_DUPLICATE_MEDIA_RESET_FOCUSED_POTENTIAL_SEARCH:
if self._current_media is not None:
hash = self._current_media.GetHash()
ClientGUIDuplicates.ResetPotentialSearch( self, ( hash, ) )
elif action == CAC.SIMPLE_DUPLICATE_MEDIA_RESET_POTENTIAL_SEARCH:
if self._current_media is not None:
hash = self._current_media.GetHash()
ClientGUIDuplicates.ResetPotentialSearch( self, ( hash, ) )
elif action == CAC.SIMPLE_DUPLICATE_MEDIA_REMOVE_FOCUSED_POTENTIALS:
if self._current_media is not None:
hash = self._current_media.GetHash()
ClientGUIDuplicates.RemovePotentials( self, ( hash, ) )
elif action == CAC.SIMPLE_DUPLICATE_MEDIA_REMOVE_POTENTIALS:
if self._current_media is not None:
hash = self._current_media.GetHash()
ClientGUIDuplicates.RemovePotentials( self, ( hash, ) )
elif action == CAC.SIMPLE_MEDIA_SEEK_DELTA:
( direction, ms ) = command.GetSimpleData()
self._SeekDeltaCurrentMedia( direction, ms )
elif action == CAC.SIMPLE_MOVE_ANIMATION_TO_PREVIOUS_FRAME:
self._media_container.GotoPreviousOrNextFrame( -1 )
elif action == CAC.SIMPLE_MOVE_ANIMATION_TO_NEXT_FRAME:
self._media_container.GotoPreviousOrNextFrame( 1 )
elif action == CAC.SIMPLE_ZOOM_IN:
self._media_container.ZoomIn()
elif action == CAC.SIMPLE_ZOOM_IN_VIEWER_CENTER:
self._media_container.ZoomIn( zoom_center_type_override = ClientGUICanvasMedia.ZOOM_CENTERPOINT_VIEWER_CENTER )
elif action == CAC.SIMPLE_ZOOM_OUT:
self._media_container.ZoomOut()
elif action == CAC.SIMPLE_ZOOM_OUT_VIEWER_CENTER:
self._media_container.ZoomOut( zoom_center_type_override = ClientGUICanvasMedia.ZOOM_CENTERPOINT_VIEWER_CENTER )
elif action == CAC.SIMPLE_SWITCH_BETWEEN_100_PERCENT_AND_CANVAS_ZOOM:
self._media_container.ZoomSwitch()
elif action == CAC.SIMPLE_SWITCH_BETWEEN_100_PERCENT_AND_MAX_ZOOM:
self._media_container.ZoomSwitch100Max()
elif action == CAC.SIMPLE_SWITCH_BETWEEN_CANVAS_AND_MAX_ZOOM:
self._media_container.ZoomSwitchCanvasMax()
elif action == CAC.SIMPLE_ZOOM_100:
self._media_container.Zoom100()
elif action == CAC.SIMPLE_ZOOM_CANVAS:
self._media_container.ZoomCanvas()
elif action == CAC.SIMPLE_ZOOM_DEFAULT:
self._media_container.ZoomDefault()
elif action == CAC.SIMPLE_ZOOM_MAX:
self._media_container.ZoomMax()
elif action == CAC.SIMPLE_SWITCH_BETWEEN_100_PERCENT_AND_CANVAS_ZOOM_VIEWER_CENTER:
self._media_container.ZoomSwitch( zoom_center_type_override = ClientGUICanvasMedia.ZOOM_CENTERPOINT_VIEWER_CENTER )
else:
command_processed = False
elif command.IsContentCommand():
if self._current_media is None:
return
command_processed = ClientGUIMediaActions.ApplyContentApplicationCommandToMedia( self, command, ( self._current_media, ) )
else:
command_processed = False
return command_processed
def ResetMediaWindowCenterPosition( self ):
self._media_container.ResetCenterPosition()
self.EndDrag()
def SetLocationContext( self, location_context: ClientLocation.LocationContext ):
self._location_context = location_context
def SetMedia( self, media: typing.Optional[ ClientMedia.MediaSingleton ] ):
if media is not None and not self.isVisible():
return
if media is not None:
media = media.GetDisplayMedia()
if not ClientGUICanvasMedia.CanDisplayMedia( media, self.CANVAS_TYPE ):
media = None
if media != self._current_media:
self.EndDrag()
HG.client_controller.ResetIdleTimer()
self._SaveCurrentMediaViewTime()
previous_media = self._current_media
self._current_media = media
if self._current_media is None:
self._media_container.ClearMedia()
else:
maintain_zoom = self._maintain_pan_and_zoom and previous_media is not None
maintain_pan = self._maintain_pan_and_zoom
( media_width, media_height ) = self._current_media.GetResolution()
size_is_ok = ( media_width is None or media_width > 0 ) and ( media_height is None or media_height > 0 )
if self._current_media.GetLocationsManager().IsLocal() and size_is_ok:
self._media_container.SetMedia( self._current_media, maintain_zoom, maintain_pan )
else:
self._current_media = None
HG.client_controller.pub( 'canvas_new_display_media', self._canvas_key, self._current_media )
HG.client_controller.pub( 'canvas_new_index_string', self._canvas_key, self._GetIndexString() )
self.update()
def minimumSizeHint( self ):
return QC.QSize( 120, 120 )
def ZoomChanged( self ):
self.update()
def ZoomIn( self, canvas_key ):
if canvas_key == self._canvas_key:
self._media_container.ZoomIn()
def ZoomOut( self, canvas_key ):
if canvas_key == self._canvas_key:
self._media_container.ZoomOut()
def ZoomSwitch( self, canvas_key ):
if canvas_key == self._canvas_key:
self._media_container.ZoomSwitch()
class MediaContainerDragClickReportingFilter( QC.QObject ):
def __init__( self, parent: Canvas ):
QC.QObject.__init__( self, parent )
self._canvas = parent
def eventFilter( self, watched, event ):
if event.type() == QC.QEvent.MouseButtonPress and event.button() == QC.Qt.LeftButton:
self._canvas.BeginDrag()
elif event.type() == QC.QEvent.MouseButtonRelease and event.button() == QC.Qt.LeftButton:
self._canvas.EndDrag()
return False
class CanvasPanel( Canvas ):
CANVAS_TYPE = CC.CANVAS_PREVIEW
def __init__( self, parent, page_key, location_context: ClientLocation.LocationContext ):
Canvas.__init__( self, parent, location_context )
self._page_key = page_key
self._hidden_page_current_media = None
self._hidden_page_paused_status = False
self._media_container.launchMediaViewer.connect( self.LaunchMediaViewer )
HG.client_controller.sub( self, 'ProcessContentUpdates', 'content_updates_gui' )
def mouseReleaseEvent( self, event ):
if event.button() != QC.Qt.RightButton:
Canvas.mouseReleaseEvent( self, event )
return
# contextmenu doesn't quite work here yet due to focus issues
self.ShowMenu()
def ClearMedia( self ):
self._hidden_page_current_media = None
Canvas.ClearMedia( self )
def PageHidden( self ):
if self._hidden_page_current_media is not None:
return
# TODO: ultimately, make an object for media/paused/position and have any media player able to give that and take it instead of setmedia
# then we'll be able to 'continue' playing state from preview to full view and other stuff like this, and simply
# also use that for all setmedia, and then if we have %-in-start options and paused/play-start options, we can initialise this object for that
hidden_page_current_media = self._current_media
hidden_page_pause_status = self._media_container.IsPaused()
self.ClearMedia()
self._hidden_page_current_media = hidden_page_current_media
self._hidden_page_paused_status = hidden_page_pause_status
def PageShown( self ):
self.SetMedia( self._hidden_page_current_media )
self._hidden_page_current_media = None
if self._media_container.IsPaused() != self._hidden_page_paused_status:
self._media_container.PausePlay()
def ShowMenu( self ):
menu = QW.QMenu()
new_options = HG.client_controller.new_options
advanced_mode = new_options.GetBoolean( 'advanced_mode' )
if self._current_media is not None:
services = HG.client_controller.services_manager.GetServices()
locations_manager = self._current_media.GetLocationsManager()
local_ratings_services = [ service for service in services if service.GetServiceType() in ( HC.LOCAL_RATING_LIKE, HC.LOCAL_RATING_NUMERICAL ) ]
i_can_post_ratings = len( local_ratings_services ) > 0
#
info_lines = list( self._current_media.GetPrettyInfoLines() )
top_line = info_lines.pop( 0 )
info_menu = QW.QMenu( menu )
ClientGUIMediaMenus.AddPrettyInfoLines( info_menu, info_lines )
ClientGUIMediaMenus.AddFileViewingStatsMenu( info_menu, ( self._current_media, ) )
ClientGUIMenus.AppendMenu( menu, info_menu, top_line )
ClientGUIMenus.AppendSeparator( menu )
AddAudioVolumeMenu( menu, self.CANVAS_TYPE )
if self._current_media is not None:
#
ClientGUIMenus.AppendSeparator( menu )
if self._current_media.HasInbox():
ClientGUIMenus.AppendMenuItem( menu, 'archive', 'Archive this file.', self._Archive )
if self._current_media.HasArchive():
ClientGUIMenus.AppendMenuItem( menu, 'inbox', 'Send this files back to the inbox.', self._Inbox )
ClientGUIMenus.AppendSeparator( menu )
local_file_service_keys = HG.client_controller.services_manager.GetServiceKeys( ( HC.LOCAL_FILE_DOMAIN, ) )
# brush this up to handle different service keys
# undelete do an optional service key too
local_file_service_keys_we_are_in = sorted( locations_manager.GetCurrent().intersection( local_file_service_keys ), key = HG.client_controller.services_manager.GetName )
for file_service_key in local_file_service_keys_we_are_in:
ClientGUIMenus.AppendMenuItem( menu, 'delete from {}'.format( HG.client_controller.services_manager.GetName( file_service_key ) ), 'Delete this file.', self._Delete, file_service_key = file_service_key )
if locations_manager.IsTrashed():
ClientGUIMenus.AppendMenuItem( menu, 'delete completely', 'Physically delete this file from disk.', self._Delete, file_service_key = CC.COMBINED_LOCAL_FILE_SERVICE_KEY )
ClientGUIMenus.AppendMenuItem( menu, 'undelete', 'Take this file out of the trash.', self._Undelete )
ClientGUIMenus.AppendSeparator( menu )
manage_menu = QW.QMenu( menu )
ClientGUIMenus.AppendMenuItem( manage_menu, 'tags', 'Manage this file\'s tags.', self._ManageTags )
if i_can_post_ratings:
ClientGUIMenus.AppendMenuItem( manage_menu, 'ratings', 'Manage this file\'s ratings.', self._ManageRatings )
ClientGUIMenus.AppendMenuItem( manage_menu, 'urls', 'Manage this file\'s known URLs.', self._ManageURLs )
num_notes = self._current_media.GetNotesManager().GetNumNotes()
notes_str = 'notes'
if num_notes > 0:
notes_str = '{} ({})'.format( notes_str, HydrusData.ToHumanInt( num_notes ) )
ClientGUIMenus.AppendMenuItem( manage_menu, notes_str, 'Manage this file\'s notes.', self._ManageNotes )
ClientGUIMediaMenus.AddManageFileViewingStatsMenu( self, manage_menu, [ self._current_media ] )
ClientGUIMenus.AppendMenu( menu, manage_menu, 'manage' )
ClientGUIMediaMenus.AddKnownURLsViewCopyMenu( self, menu, self._current_media )
open_menu = QW.QMenu( menu )
ClientGUIMenus.AppendMenuItem( open_menu, 'in external program', 'Open this file in your OS\'s default program.', self._OpenExternally )
ClientGUIMenus.AppendMenuItem( open_menu, 'in a new page', 'Show your current media in a simple new page.', self._ShowMediaInNewPage )
ClientGUIMenus.AppendMenuItem( open_menu, 'in web browser', 'Show this file in your OS\'s web browser.', self._OpenFileInWebBrowser )
show_open_in_explorer = advanced_mode and ( HC.PLATFORM_WINDOWS or HC.PLATFORM_MACOS )
if show_open_in_explorer:
ClientGUIMenus.AppendMenuItem( open_menu, 'in file browser', 'Show this file in your OS\'s file browser.', self._OpenFileLocation )
ClientGUIMenus.AppendMenu( menu, open_menu, 'open' )
share_menu = QW.QMenu( menu )
copy_menu = QW.QMenu( share_menu )
ClientGUIMenus.AppendMenuItem( copy_menu, 'file', 'Copy this file to your clipboard.', self._CopyFileToClipboard )
copy_hash_menu = QW.QMenu( copy_menu )
ClientGUIMenus.AppendMenuItem( copy_hash_menu, 'sha256 ({})'.format( self._current_media.GetHash().hex() ), 'Copy this file\'s SHA256 hash.', self._CopyHashToClipboard, 'sha256' )
ClientGUIMenus.AppendMenuItem( copy_hash_menu, 'md5', 'Copy this file\'s MD5 hash.', self._CopyHashToClipboard, 'md5' )
ClientGUIMenus.AppendMenuItem( copy_hash_menu, 'sha1', 'Copy this file\'s SHA1 hash.', self._CopyHashToClipboard, 'sha1' )
ClientGUIMenus.AppendMenuItem( copy_hash_menu, 'sha512', 'Copy this file\'s SHA512 hash.', self._CopyHashToClipboard, 'sha512' )
ClientGUIMenus.AppendMenu( copy_menu, copy_hash_menu, 'hash' )
if advanced_mode:
hash_id_str = str( self._current_media.GetHashId() )
ClientGUIMenus.AppendMenuItem( copy_menu, 'file_id ({})'.format( hash_id_str ), 'Copy this file\'s internal file/hash_id.', HG.client_controller.pub, 'clipboard', 'text', hash_id_str )
if self._current_media.GetMime() in HC.IMAGES:
ClientGUIMenus.AppendMenuItem( copy_menu, 'image (bitmap)', 'Copy this file to your clipboard as a bmp.', self._CopyBMPToClipboard )
ClientGUIMenus.AppendMenuItem( copy_menu, 'path', 'Copy this file\'s path to your clipboard.', self._CopyPathToClipboard )
ClientGUIMenus.AppendMenu( share_menu, copy_menu, 'copy' )
ClientGUIMenus.AppendMenu( menu, share_menu, 'share' )
CGC.core().PopupMenu( self, menu )
def LaunchMediaViewer( self ):
HG.client_controller.pub( 'launch_media_viewer', self._page_key )
def MediaFocusWentToExternalProgram( self, page_key ):
if page_key == self._page_key:
self._MediaFocusWentToExternalProgram()
def ProcessContentUpdates( self, service_keys_to_content_updates ):
if self._current_media is not None:
my_hash = self._current_media.GetHash()
do_redraw = False
for ( service_key, content_updates ) in service_keys_to_content_updates.items():
if True in ( my_hash in content_update.GetHashes() for content_update in content_updates ):
do_redraw = True
break
if do_redraw:
self.update()
def SetMedia( self, media ):
if HC.options[ 'hide_preview' ]:
return
Canvas.SetMedia( self, media )
class CanvasWithDetails( Canvas ):
def __init__( self, parent, location_context ):
Canvas.__init__( self, parent, location_context )
HG.client_controller.sub( self, 'RedrawDetails', 'refresh_all_tag_presentation_gui' )
def _DrawAdditionalTopMiddleInfo( self, painter: QG.QPainter, current_y ):
pass
def _DrawBackgroundDetails( self, painter: QG.QPainter ):
my_size = self.size()
my_width = my_size.width()
my_height = my_size.height()
if self._current_media is None:
text = self._GetNoMediaText()
( text_size, text ) = ClientGUIFunctions.GetTextSizeFromPainter( painter, text )
x = ( my_width - text_size.width() ) // 2
y = ( my_height - text_size.height() ) // 2
ClientGUIFunctions.DrawText( painter, x, y, text )
else:
self._DrawTags( painter )
self._DrawTopMiddle( painter )
current_y = self._DrawTopRight( painter )
self._DrawNotes( painter, current_y )
self._DrawIndexAndZoom( painter )
def _DrawIndexAndZoom( self, painter: QG.QPainter ):
my_size = self.size()
my_width = my_size.width()
my_height = my_size.height()
# bottom-right index
bottom_right_string = ClientData.ConvertZoomToPercentage( self._media_container.GetCurrentZoom() )
index_string = self._GetIndexString()
if len( index_string ) > 0:
bottom_right_string = '{} - {}'.format( bottom_right_string, index_string )
( text_size, bottom_right_string ) = ClientGUIFunctions.GetTextSizeFromPainter( painter, bottom_right_string )
ClientGUIFunctions.DrawText( painter, my_width - text_size.width() - 3, my_height - text_size.height() - 3, bottom_right_string )
def _DrawNotes( self, painter: QG.QPainter, current_y: int ):
notes_manager = self._current_media.GetNotesManager()
names_to_notes = notes_manager.GetNamesToNotes()
if len( names_to_notes ) == 0:
return
my_size = self.size()
my_width = my_size.width()
my_height = my_size.height()
max_notes_width_percentage = 20
PADDING = 4
max_notes_width = int( my_width * ( max_notes_width_percentage / 100 ) ) - ( PADDING * 2 )
notes_width = 0
original_font = painter.font()
name_font = QG.QFont( original_font )
name_font.setBold( True )
notes_font = QG.QFont( original_font )
notes_font.setBold( False )
for ( name, note ) in names_to_notes.items():
# without wrapping, let's see if we fit into a smaller box than the max possible
painter.setFont( name_font )
name_text_size = painter.fontMetrics().size( 0, name )
painter.setFont( notes_font )
note_text_size = painter.fontMetrics().size( 0, note )
notes_width = max( notes_width, name_text_size.width(), note_text_size.width() )
if notes_width > max_notes_width:
notes_width = max_notes_width
break
left_x = my_width - ( notes_width + PADDING )
current_y += PADDING * 2
draw_a_test_rect = False
if draw_a_test_rect:
painter.setPen( QG.QPen( QG.QColor( 20, 20, 20 ) ) )
painter.setBrush( QC.Qt.NoBrush )
painter.drawRect( left_x, current_y, notes_width, 100 )
for name in sorted( names_to_notes.keys() ):
painter.setFont( name_font )
text_rect = painter.fontMetrics().boundingRect( left_x, current_y, notes_width, 100, QC.Qt.AlignHCenter | QC.Qt.TextWordWrap, name )
painter.drawText( text_rect, QC.Qt.AlignHCenter | QC.Qt.TextWordWrap, name )
current_y += text_rect.height() + PADDING
#
painter.setFont( notes_font )
note = notes_manager.GetNote( name )
text_rect = painter.fontMetrics().boundingRect( left_x, current_y, notes_width, 100, QC.Qt.AlignJustify | QC.Qt.TextWordWrap, note )
painter.drawText( text_rect, QC.Qt.AlignJustify | QC.Qt.TextWordWrap, note )
current_y += text_rect.height() + PADDING
if current_y >= my_height:
break
# draw a horizontal line
painter.setFont( original_font )
def _DrawTags( self, painter: QG.QPainter ):
# tags on the top left
original_pen = painter.pen()
tags_manager = self._current_media.GetTagsManager()
current = tags_manager.GetCurrent( CC.COMBINED_TAG_SERVICE_KEY, ClientTags.TAG_DISPLAY_SINGLE_MEDIA )
pending = tags_manager.GetPending( CC.COMBINED_TAG_SERVICE_KEY, ClientTags.TAG_DISPLAY_SINGLE_MEDIA )
petitioned = tags_manager.GetPetitioned( CC.COMBINED_TAG_SERVICE_KEY, ClientTags.TAG_DISPLAY_SINGLE_MEDIA )
tags_i_want_to_display = set()
tags_i_want_to_display.update( current )
tags_i_want_to_display.update( pending )
tags_i_want_to_display.update( petitioned )
tags_i_want_to_display = list( tags_i_want_to_display )
tag_sort = HG.client_controller.new_options.GetDefaultTagSort()
ClientTagSorting.SortTags( tag_sort, tags_i_want_to_display )
current_y = 3
namespace_colours = HC.options[ 'namespace_colours' ]
for tag in tags_i_want_to_display:
display_string = ClientTags.RenderTag( tag, True )
if tag in pending:
display_string += ' (+)'
if tag in petitioned:
display_string += ' (-)'
( namespace, subtag ) = HydrusTags.SplitTag( tag )
if namespace in namespace_colours:
( r, g, b ) = namespace_colours[ namespace ]
else:
( r, g, b ) = namespace_colours[ None ]
painter.setPen( QG.QPen( QG.QColor( r, g, b ) ) )
( text_size, display_string ) = ClientGUIFunctions.GetTextSizeFromPainter( painter, display_string )
ClientGUIFunctions.DrawText( painter, 5, current_y, display_string )
current_y += text_size.height()
painter.setPen( original_pen )
def _DrawTopMiddle( self, painter: QG.QPainter ):
my_size = self.size()
my_width = my_size.width()
my_height = my_size.height()
# top-middle
painter.setPen( QG.QPen( self._new_options.GetColour( CC.COLOUR_MEDIA_TEXT ) ) )
current_y = 3
title_string = self._current_media.GetTitleString()
if len( title_string ) > 0:
( text_size, title_string ) = ClientGUIFunctions.GetTextSizeFromPainter( painter, title_string )
ClientGUIFunctions.DrawText( painter, ( my_width - text_size.width() ) // 2, current_y, title_string )
current_y += text_size.height() + 3
info_string = self._GetInfoString()
( text_size, info_string ) = ClientGUIFunctions.GetTextSizeFromPainter( painter, info_string )
ClientGUIFunctions.DrawText( painter, ( my_width - text_size.width() ) // 2, current_y, info_string )
current_y += text_size.height() + 3
self._DrawAdditionalTopMiddleInfo( painter, current_y )
def _DrawTopRight( self, painter: QG.QPainter ) -> int:
my_size = self.size()
my_width = my_size.width()
my_height = my_size.height()
current_y = 2
# ratings
services_manager = HG.client_controller.services_manager
like_services = services_manager.GetServices( ( HC.LOCAL_RATING_LIKE, ) )
like_services.reverse()
like_rating_current_x = my_width - 16 - 2 # -2 to line up exactly with the floating panel
for like_service in like_services:
service_key = like_service.GetServiceKey()
rating_state = ClientRatings.GetLikeStateFromMedia( ( self._current_media, ), service_key )
ClientGUIRatings.DrawLike( painter, like_rating_current_x, current_y, service_key, rating_state )
like_rating_current_x -= 16
if len( like_services ) > 0:
current_y += 18
numerical_services = services_manager.GetServices( ( HC.LOCAL_RATING_NUMERICAL, ) )
for numerical_service in numerical_services:
service_key = numerical_service.GetServiceKey()
( rating_state, rating ) = ClientRatings.GetNumericalStateFromMedia( ( self._current_media, ), service_key )
numerical_width = ClientGUIRatings.GetNumericalWidth( service_key )
ClientGUIRatings.DrawNumerical( painter, my_width - numerical_width - 2, current_y, service_key, rating_state, rating ) # -2 to line up exactly with the floating panel
current_y += 18
# icons
icons_to_show = []
if self._current_media.GetLocationsManager().IsTrashed():
icons_to_show.append( CC.global_pixmaps().trash )
if self._current_media.HasInbox():
icons_to_show.append( CC.global_pixmaps().inbox )
if len( icons_to_show ) > 0:
icon_x = 0
for icon in icons_to_show:
painter.drawPixmap( my_width + icon_x - 18, current_y, icon )
icon_x -= 18
current_y += 18
painter.setPen( QG.QPen( self._new_options.GetColour( CC.COLOUR_MEDIA_TEXT ) ) )
# repo strings
remote_strings = self._current_media.GetLocationsManager().GetRemoteLocationStrings()
for remote_string in remote_strings:
( text_size, remote_string ) = ClientGUIFunctions.GetTextSizeFromPainter( painter, remote_string )
ClientGUIFunctions.DrawText( painter, my_width - text_size.width() - 3, current_y, remote_string )
current_y += text_size.height()
# urls
urls = self._current_media.GetLocationsManager().GetURLs()
url_tuples = HG.client_controller.network_engine.domain_manager.ConvertURLsToMediaViewerTuples( urls )
for ( display_string, url ) in url_tuples:
( text_size, display_string ) = ClientGUIFunctions.GetTextSizeFromPainter( painter, display_string )
ClientGUIFunctions.DrawText( painter, my_width - text_size.width() - 3, current_y, display_string )
current_y += text_size.height() + 2
return current_y
def _GetInfoString( self ):
lines = [ line for line in self._current_media.GetPrettyInfoLines( only_interesting_lines = True ) if isinstance( line, str ) ]
lines.insert( 1, ClientData.ConvertZoomToPercentage( self._media_container.GetCurrentZoom() ) )
info_string = ' | '.join( lines )
return info_string
def _GetNoMediaText( self ):
return 'No media to display'
def RedrawDetails( self ):
self.update()
def TryToDoPreClose( self ):
can_close = True
return can_close
class CanvasWithHovers( CanvasWithDetails ):
def __init__( self, parent, location_context ):
CanvasWithDetails.__init__( self, parent, location_context )
self._hovers = []
top_hover = self._GenerateHoverTopFrame()
top_hover.sendApplicationCommand.connect( self.ProcessApplicationCommand )
self._media_container.zoomChanged.connect( top_hover.SetCurrentZoom )
self._hovers.append( top_hover )
self._my_shortcuts_handler.AddWindowToFilter( top_hover )
tags_hover = ClientGUICanvasHoverFrames.CanvasHoverFrameTags( self, self, top_hover, self._canvas_key )
tags_hover.sendApplicationCommand.connect( self.ProcessApplicationCommand )
self._hovers.append( tags_hover )
self._my_shortcuts_handler.AddWindowToFilter( tags_hover )
top_right_hover = ClientGUICanvasHoverFrames.CanvasHoverFrameTopRight( self, self, top_hover, self._canvas_key )
top_right_hover.sendApplicationCommand.connect( self.ProcessApplicationCommand )
self._hovers.append( top_right_hover )
self._my_shortcuts_handler.AddWindowToFilter( top_right_hover )
self._right_notes_hover = ClientGUICanvasHoverFrames.CanvasHoverFrameRightNotes( self, self, top_right_hover, self._canvas_key )
self._right_notes_hover.sendApplicationCommand.connect( self.ProcessApplicationCommand )
self._hovers.append( self._right_notes_hover )
self._my_shortcuts_handler.AddWindowToFilter( self._right_notes_hover )
for name in self._new_options.GetStringList( 'default_media_viewer_custom_shortcuts' ):
self._my_shortcuts_handler.AddShortcuts( name )
#
self._timer_cursor_hide_job = None
self._last_cursor_autohide_touch_time = HydrusData.GetNowFloat()
# need this as we need un-button-pressed move events for cursor hide
self.setMouseTracking( True )
self._RestartCursorHideWait()
HG.client_controller.sub( self, 'CloseFromHover', 'canvas_close' )
HG.client_controller.sub( self, 'FullscreenSwitch', 'canvas_fullscreen_switch' )
HG.client_controller.gui.RegisterUIUpdateWindow( self )
def _GenerateHoverTopFrame( self ):
raise NotImplementedError()
def _HideCursorCheck( self ):
hide_time_ms = HG.client_controller.new_options.GetNoneableInteger( 'media_viewer_cursor_autohide_time_ms' )
if hide_time_ms is None:
return
hide_time = hide_time_ms / 1000
can_hide = HydrusData.TimeHasPassedFloat( self._last_cursor_autohide_touch_time + hide_time )
can_check_again = ClientGUIFunctions.MouseIsOverWidget( self )
if not CC.CAN_HIDE_MOUSE:
can_hide = False
if CGC.core().MenuIsOpen():
can_hide = False
if ClientGUIFunctions.DialogIsOpen():
can_hide = False
can_check_again = False
if can_hide:
self.setCursor( QG.QCursor( QC.Qt.BlankCursor ) )
elif can_check_again:
self._RestartCursorHideCheckJob()
def _RestartCursorHideWait( self ):
self._last_cursor_autohide_touch_time = HydrusData.GetNowFloat()
self._RestartCursorHideCheckJob()
def _RestartCursorHideCheckJob( self ):
if self._timer_cursor_hide_job is not None:
timer_is_running_or_finished = self._timer_cursor_hide_job.CurrentlyWorking() or self._timer_cursor_hide_job.IsWorkComplete()
if not timer_is_running_or_finished:
return
self._timer_cursor_hide_job = HG.client_controller.CallLaterQtSafe( self, 0.1, 'hide cursor check', self._HideCursorCheck )
def _TryToCloseWindow( self ):
self.window().close()
def CloseFromHover( self, canvas_key ):
if canvas_key == self._canvas_key:
self._TryToCloseWindow()
def FullscreenSwitch( self, canvas_key ):
if canvas_key == self._canvas_key:
self.parentWidget().FullscreenSwitch()
def mouseMoveEvent( self, event ):
current_focus_tlw = QW.QApplication.activeWindow()
my_tlw = self.window()
if isinstance( current_focus_tlw, ClientGUICanvasHoverFrames.CanvasHoverFrame ) and ClientGUIFunctions.IsQtAncestor( current_focus_tlw, my_tlw, through_tlws = True ):
my_tlw.activateWindow()
#
CC.CAN_HIDE_MOUSE = True
# due to the mouse setPos below, the event pos can get funky I think due to out of order coordinate setting events, so we'll poll current value directly
event_pos = self.mapFromGlobal( QG.QCursor.pos() )
mouse_currently_shown = self.cursor() == QG.QCursor( QC.Qt.ArrowCursor )
show_mouse = mouse_currently_shown
is_dragging = event.buttons() & QC.Qt.LeftButton and self._last_drag_pos is not None
has_moved = event_pos != self._last_motion_pos
if is_dragging:
delta = event_pos - self._last_drag_pos
approx_distance = delta.manhattanLength()
if approx_distance > 0:
touchscreen_canvas_drags_unanchor = HG.client_controller.new_options.GetBoolean( 'touchscreen_canvas_drags_unanchor' )
if not self._current_drag_is_touch and approx_distance > 50:
# if user is able to generate such a large distance, they are almost certainly touching
self._current_drag_is_touch = True
# touch events obviously don't mix with warping well. the touch just warps it back and again and we get a massive delta!
touch_anchor_override = touchscreen_canvas_drags_unanchor and self._current_drag_is_touch
anchor_and_hide_canvas_drags = HG.client_controller.new_options.GetBoolean( 'anchor_and_hide_canvas_drags' )
if anchor_and_hide_canvas_drags and not touch_anchor_override:
show_mouse = False
global_mouse_pos = self.mapToGlobal( self._last_drag_pos )
QG.QCursor.setPos( global_mouse_pos )
ClientGUIShortcuts.CUMULATIVE_MOUSEWARP_MANHATTAN_LENGTH += approx_distance
else:
show_mouse = True
self._last_drag_pos = QC.QPoint( event_pos )
self._media_container.MoveDelta( delta )
else:
if has_moved:
self._last_motion_pos = QC.QPoint( event_pos )
show_mouse = True
if show_mouse:
if not mouse_currently_shown:
self.setCursor( QG.QCursor( QC.Qt.ArrowCursor ) )
self._RestartCursorHideWait()
else:
if mouse_currently_shown:
self.setCursor( QG.QCursor( QC.Qt.BlankCursor ) )
CanvasWithDetails.mouseMoveEvent( self, event )
def ProcessApplicationCommand( self, command: CAC.ApplicationCommand ):
command_processed = True
if command.IsSimpleCommand():
action = command.GetSimpleAction()
if action == CAC.SIMPLE_CLOSE_MEDIA_VIEWER:
self._TryToCloseWindow()
else:
command_processed = False
else:
command_processed = False
if not command_processed:
command_processed = CanvasWithDetails.ProcessApplicationCommand( self, command )
return command_processed
def TIMERUIUpdate( self ):
for hover in self._hovers:
hover.DoRegularHideShow()
class CanvasFilterDuplicates( CanvasWithHovers ):
CANVAS_TYPE = CC.CANVAS_MEDIA_VIEWER_DUPLICATES
showPairInPage = QC.Signal( list )
def __init__( self, parent, file_search_context_1: ClientSearch.FileSearchContext, file_search_context_2: ClientSearch.FileSearchContext, dupe_search_type, pixel_dupes_preference, max_hamming_distance ):
location_context = file_search_context_1.GetLocationContext()
CanvasWithHovers.__init__( self, parent, location_context )
hover = ClientGUICanvasHoverFrames.CanvasHoverFrameRightDuplicates( self, self, self._right_notes_hover, self._canvas_key )
hover.showPairInPage.connect( self._ShowPairInPage )
hover.sendApplicationCommand.connect( self.ProcessApplicationCommand )
self._hovers.append( hover )
self._my_shortcuts_handler.AddWindowToFilter( hover )
self._file_search_context_1 = file_search_context_1
self._file_search_context_2 = file_search_context_2
self._dupe_search_type = dupe_search_type
self._pixel_dupes_preference = pixel_dupes_preference
self._max_hamming_distance = max_hamming_distance
self._maintain_pan_and_zoom = True
self._currently_fetching_pairs = False
self._batch_of_pairs_to_process = []
self._current_pair_index = 0
self._processed_pairs = []
self._hashes_due_to_be_deleted_in_this_batch = set()
# ok we started excluding pairs if they had been deleted, now I am extending it to any files that have been processed.
# main thing is if you have AB, AC, that's neat and a bunch of people want it, but current processing system doesn't do B->A->C merge if it happens in a single batch
# I need to store dupe merge options rather than content updates apply them in db transaction or do the retroactive sync or similar to get this done properly
# so regrettably I turn it off for now
self._hashes_processed_in_this_batch = set()
self._media_list = ClientMedia.ListeningMediaList( location_context, [] )
self._my_shortcuts_handler.AddShortcuts( 'media_viewer_browser' )
self._my_shortcuts_handler.AddShortcuts( 'duplicate_filter' )
HG.client_controller.sub( self, 'ProcessContentUpdates', 'content_updates_gui' )
HG.client_controller.sub( self, 'Delete', 'canvas_delete' )
HG.client_controller.sub( self, 'Undelete', 'canvas_undelete' )
HG.client_controller.sub( self, 'SwitchMedia', 'canvas_show_next' )
HG.client_controller.sub( self, 'SwitchMedia', 'canvas_show_previous' )
QP.CallAfter( self._LoadNextBatchOfPairs )
def _CommitProcessed( self, blocking = True ):
pair_info = []
for ( duplicate_type, first_media, second_media, list_of_service_keys_to_content_updates, was_auto_skipped ) in self._processed_pairs:
if duplicate_type is None:
if len( list_of_service_keys_to_content_updates ) > 0:
for service_keys_to_content_updates in list_of_service_keys_to_content_updates:
if blocking:
HG.client_controller.WriteSynchronous( 'content_updates', service_keys_to_content_updates )
else:
HG.client_controller.Write( 'content_updates', service_keys_to_content_updates )
continue
if was_auto_skipped:
continue # it was a 'skip' decision
first_hash = first_media.GetHash()
second_hash = second_media.GetHash()
pair_info.append( ( duplicate_type, first_hash, second_hash, list_of_service_keys_to_content_updates ) )
if len( pair_info ) > 0:
if blocking:
HG.client_controller.WriteSynchronous( 'duplicate_pair_status', pair_info )
else:
HG.client_controller.Write( 'duplicate_pair_status', pair_info )
self._processed_pairs = []
self._hashes_due_to_be_deleted_in_this_batch = set()
self._hashes_processed_in_this_batch = set()
def _CurrentMediaIsBetter( self, delete_second = True ):
self._ProcessPair( HC.DUPLICATE_BETTER, delete_second = delete_second )
def _Delete( self, media = None, reason = None, file_service_key = None ):
if self._current_media is None:
return False
first_media = self._current_media
second_media = self._media_list.GetNext( self._current_media )
message = 'Delete just this file, or both?'
yes_tuples = []
yes_tuples.append( ( 'delete just this one', 'current' ) )
yes_tuples.append( ( 'delete both', 'both' ) )
try:
result = ClientGUIDialogsQuick.GetYesYesNo( self, message, yes_tuples = yes_tuples, no_label = 'forget it' )
except HydrusExceptions.CancelledException:
return False
if result == 'current':
media = [ first_media ]
default_reason = 'Deleted manually in Duplicate Filter.'
elif result == 'both':
media = [ first_media, second_media ]
default_reason = 'Deleted manually in Duplicate Filter, along with its potential duplicate.'
jobs = CanvasWithHovers._Delete( self, media = media, default_reason = default_reason, file_service_key = file_service_key, just_get_jobs = True )
deleted = isinstance( jobs, list ) and len( jobs ) > 0
if deleted:
for m in media:
self._hashes_due_to_be_deleted_in_this_batch.update( m.GetHashes() )
was_auto_skipped = False
( first_media_result, second_media_result ) = self._batch_of_pairs_to_process[ self._current_pair_index ]
first_media = ClientMedia.MediaSingleton( first_media_result )
second_media = ClientMedia.MediaSingleton( second_media_result )
process_tuple = ( None, first_media, second_media, jobs, was_auto_skipped )
self._ShowNextPair( process_tuple )
return deleted
def _DoCustomAction( self ):
if self._current_media is None:
return
duplicate_types = [ HC.DUPLICATE_BETTER, HC.DUPLICATE_SAME_QUALITY, HC.DUPLICATE_ALTERNATE, HC.DUPLICATE_FALSE_POSITIVE ]
choice_tuples = [ ( HC.duplicate_type_string_lookup[ duplicate_type ], duplicate_type ) for duplicate_type in duplicate_types ]
try:
duplicate_type = ClientGUIDialogsQuick.SelectFromList( self, 'select duplicate type', choice_tuples )
except HydrusExceptions.CancelledException:
return
new_options = HG.client_controller.new_options
if duplicate_type in [ HC.DUPLICATE_BETTER, HC.DUPLICATE_SAME_QUALITY ] or ( new_options.GetBoolean( 'advanced_mode' ) and duplicate_type == HC.DUPLICATE_ALTERNATE ):
duplicate_content_merge_options = new_options.GetDuplicateContentMergeOptions( duplicate_type )
with ClientGUITopLevelWindowsPanels.DialogEdit( self, 'edit duplicate merge options' ) as dlg_2:
panel = ClientGUIScrolledPanelsEdit.EditDuplicateContentMergeOptionsPanel( dlg_2, duplicate_type, duplicate_content_merge_options, for_custom_action = True )
dlg_2.SetPanel( panel )
if dlg_2.exec() == QW.QDialog.Accepted:
duplicate_content_merge_options = panel.GetValue()
else:
return
else:
duplicate_content_merge_options = None
message = 'Delete any of the files?'
yes_tuples = []
yes_tuples.append( ( 'delete neither', 'delete_neither' ) )
yes_tuples.append( ( 'delete this one', 'delete_first' ) )
yes_tuples.append( ( 'delete the other', 'delete_second' ) )
yes_tuples.append( ( 'delete both', 'delete_both' ) )
try:
result = ClientGUIDialogsQuick.GetYesYesNo( self, message, yes_tuples = yes_tuples, no_label = 'forget it' )
except HydrusExceptions.CancelledException:
return
delete_first = False
delete_second = False
if result == 'delete_first':
delete_first = True
elif result == 'delete_second':
delete_second = True
elif result == 'delete_both':
delete_first = True
delete_second = True
self._ProcessPair( duplicate_type, delete_first = delete_first, delete_second = delete_second, duplicate_content_merge_options = duplicate_content_merge_options )
def _DrawBackgroundDetails( self, painter ):
if self._currently_fetching_pairs:
text = 'Loading pairs\u2026'
( text_size, text ) = ClientGUIFunctions.GetTextSizeFromPainter( painter, text )
my_size = self.size()
x = ( my_size.width() - text_size.width() ) // 2
y = ( my_size.height() - text_size.height() ) // 2
ClientGUIFunctions.DrawText( painter, x, y, text )
else:
CanvasWithHovers._DrawBackgroundDetails( self, painter )
def _GenerateHoverTopFrame( self ):
return ClientGUICanvasHoverFrames.CanvasHoverFrameTopDuplicatesFilter( self, self, self._canvas_key )
def _GetBackgroundColour( self ):
normal_colour = self._new_options.GetColour( CC.COLOUR_MEDIA_BACKGROUND )
if self._current_media is None or len( self._media_list ) == 0:
return normal_colour
else:
if self._current_media == self._media_list.GetFirst():
return normal_colour
else:
new_options = HG.client_controller.new_options
duplicate_intensity = new_options.GetNoneableInteger( 'duplicate_background_switch_intensity' )
return ClientGUIFunctions.GetLighterDarkerColour( normal_colour, duplicate_intensity )
def _GetIndexString( self ):
if self._current_media is None or len( self._media_list ) == 0:
return '-'
else:
current_media_label = 'A' if self._current_media == self._media_list.GetFirst() else 'B'
progress = self._current_pair_index + 1
total = len( self._batch_of_pairs_to_process )
index_string = HydrusData.ConvertValueRangeToPrettyString( progress, total )
num_committable = self._GetNumCommittableDecisions()
num_deletable = self._GetNumCommittableDeletes()
components = []
if num_committable > 0:
components.append( '{} decisions'.format( HydrusData.ToHumanInt( num_committable ) ) )
if num_deletable > 0:
components.append( '{} deletes'.format( HydrusData.ToHumanInt( num_deletable ) ) )
if len( components ) == 0:
num_decisions_string = 'no decisions yet'
else:
num_decisions_string = ', '.join( components )
return '{} - {} - {}'.format( current_media_label, index_string, num_decisions_string )
def _GetNoMediaText( self ):
return 'Looking for pairs to compare--please wait.'
def _GetNumCommittableDecisions( self ):
return len( [ 1 for ( duplicate_type, first_media, second_media, list_of_service_keys_to_content_updates, was_auto_skipped ) in self._processed_pairs if duplicate_type is not None ] )
def _GetNumCommittableDeletes( self ):
return len( [ 1 for ( duplicate_type, first_media, second_media, list_of_service_keys_to_content_updates, was_auto_skipped ) in self._processed_pairs if duplicate_type is None and len( list_of_service_keys_to_content_updates ) > 0 ] )
def _GetNumRemainingDecisions( self ):
# this looks a little weird, but I want to be clear that we make a decision on the final index
last_decision_index = len( self._batch_of_pairs_to_process ) - 1
number_of_decisions_after_the_current = last_decision_index - self._current_pair_index
return max( 0, 1 + number_of_decisions_after_the_current )
def _GoBack( self ):
if self._current_pair_index > 0:
it_went_ok = self._RewindProcessing()
if it_went_ok:
self._ShowCurrentPair()
def _LoadNextBatchOfPairs( self ):
self._hashes_due_to_be_deleted_in_this_batch = set()
self._hashes_processed_in_this_batch = set()
self._processed_pairs = [] # just in case someone 'skip'ed everything in the last batch, so this never got cleared above in the commit
self.ClearMedia()
self._media_list = ClientMedia.ListeningMediaList( self._location_context, [] )
self._currently_fetching_pairs = True
HG.client_controller.CallToThread( self.THREADFetchPairs, self._file_search_context_1, self._file_search_context_2, self._dupe_search_type, self._pixel_dupes_preference, self._max_hamming_distance )
self.update()
def _MediaAreAlternates( self ):
self._ProcessPair( HC.DUPLICATE_ALTERNATE )
def _MediaAreFalsePositive( self ):
self._ProcessPair( HC.DUPLICATE_FALSE_POSITIVE )
def _MediaAreTheSame( self ):
self._ProcessPair( HC.DUPLICATE_SAME_QUALITY )
def _PrefetchNeighbours( self ):
if self._current_media is None:
return
other_media = self._media_list.GetNext( self._current_media )
media_to_prefetch = [ other_media ]
# this doesn't handle big skip events, but that's a job for later
if self._GetNumRemainingDecisions() > 1: # i.e. more than the current one we are looking at
media_to_prefetch.extend( self._batch_of_pairs_to_process[ self._current_pair_index + 1 ] )
image_cache = HG.client_controller.GetCache( 'images' )
for media in media_to_prefetch:
hash = media.GetHash()
mime = media.GetMime()
if media.IsStaticImage():
if not image_cache.HasImageRenderer( hash ):
# we do qt safe to make sure the job is cancelled if we are destroyed
HG.client_controller.CallAfterQtSafe( self, 'image pre-fetch', image_cache.PrefetchImageRenderer, media )
def _ProcessPair( self, duplicate_type, delete_first = False, delete_second = False, duplicate_content_merge_options = None ):
if self._current_media is None:
return
if duplicate_content_merge_options is None:
if duplicate_type in [ HC.DUPLICATE_BETTER, HC.DUPLICATE_SAME_QUALITY ] or ( HG.client_controller.new_options.GetBoolean( 'advanced_mode' ) and duplicate_type == HC.DUPLICATE_ALTERNATE ):
new_options = HG.client_controller.new_options
duplicate_content_merge_options = new_options.GetDuplicateContentMergeOptions( duplicate_type )
else:
duplicate_content_merge_options = ClientDuplicates.DuplicateContentMergeOptions()
first_media = self._current_media
second_media = self._media_list.GetNext( first_media )
was_auto_skipped = False
self._hashes_processed_in_this_batch.update( first_media.GetHashes() )
self._hashes_processed_in_this_batch.update( second_media.GetHashes() )
if delete_first or delete_second:
if delete_first:
self._hashes_due_to_be_deleted_in_this_batch.update( first_media.GetHashes() )
if delete_second:
self._hashes_due_to_be_deleted_in_this_batch.update( second_media.GetHashes() )
if duplicate_type in ( HC.DUPLICATE_BETTER, HC.DUPLICATE_WORSE ):
file_deletion_reason = 'better/worse'
if delete_second:
file_deletion_reason += ', worse file deleted'
else:
file_deletion_reason = HC.duplicate_type_string_lookup[ duplicate_type ]
if delete_first and delete_second:
file_deletion_reason += ', both files deleted'
file_deletion_reason = 'Deleted in Duplicate Filter ({}).'.format( file_deletion_reason )
else:
file_deletion_reason = None
list_of_service_keys_to_content_updates = [ duplicate_content_merge_options.ProcessPairIntoContentUpdates( first_media, second_media, delete_first = delete_first, delete_second = delete_second, file_deletion_reason = file_deletion_reason ) ]
process_tuple = ( duplicate_type, first_media, second_media, list_of_service_keys_to_content_updates, was_auto_skipped )
self._ShowNextPair( process_tuple )
def _RewindProcessing( self ) -> bool:
def test_we_can_pop():
if len( self._processed_pairs ) == 0:
# the first one shouldn't be auto-skipped, so if it was and now we can't pop, something weird happened
HG.client_controller.pub( 'new_similar_files_potentials_search_numbers' )
QW.QMessageBox.critical( self, 'Error', 'Due to an unexpected series of events, the duplicate filter has no valid pair to back up to. It could be some files were deleted during processing. The filter will now close.' )
self.window().deleteLater()
return False
return True
if self._current_pair_index > 0:
while True:
if not test_we_can_pop():
return False
( duplicate_type, first_media, second_media, list_of_service_keys_to_content_updates, was_auto_skipped ) = self._processed_pairs.pop()
self._current_pair_index -= 1
if not was_auto_skipped:
break
# only want this for the one that wasn't auto-skipped
for m in ( first_media, second_media ):
hash = m.GetHash()
self._hashes_due_to_be_deleted_in_this_batch.discard( hash )
self._hashes_processed_in_this_batch.discard( hash )
return True
return False
def _ShowCurrentPair( self ):
if self._currently_fetching_pairs:
return
( first_media_result, second_media_result ) = self._batch_of_pairs_to_process[ self._current_pair_index ]
first_media = ClientMedia.MediaSingleton( first_media_result )
second_media = ClientMedia.MediaSingleton( second_media_result )
score = ClientDuplicates.GetDuplicateComparisonScore( first_media, second_media )
if score > 0:
media_results_with_better_first = ( first_media_result, second_media_result )
else:
media_results_with_better_first = ( second_media_result, first_media_result )
self._media_list = ClientMedia.ListeningMediaList( self._location_context, media_results_with_better_first )
# reset zoom gubbins
self.SetMedia( None )
self.SetMedia( self._media_list.GetFirst() )
self._media_container.hide()
self._media_container.ZoomReinit()
self._media_container.ResetCenterPosition()
self.EndDrag()
self._media_container.show()
def _ShowNextPair( self, process_tuple: tuple ):
if self._currently_fetching_pairs:
return
# hackery dackery doo to quick solve something that is calling this a bunch of times while the 'and continue?' dialog is open, making like 16 of them
# a full rewrite is needed on this awful workflow
tlws = QW.QApplication.topLevelWidgets()
for tlw in tlws:
if isinstance( tlw, ClientGUITopLevelWindowsPanels.DialogCustomButtonQuestion ) and tlw.isModal():
return
#
def pair_is_good( pair ):
( first_media_result, second_media_result ) = pair
first_hash = first_media_result.GetHash()
second_hash = second_media_result.GetHash()
if first_hash in self._hashes_processed_in_this_batch or second_hash in self._hashes_processed_in_this_batch:
return False
if first_hash in self._hashes_due_to_be_deleted_in_this_batch or second_hash in self._hashes_due_to_be_deleted_in_this_batch:
return False
first_media = ClientMedia.MediaSingleton( first_media_result )
second_media = ClientMedia.MediaSingleton( second_media_result )
if not ClientGUICanvasMedia.CanDisplayMedia( first_media, self.CANVAS_TYPE ) or not ClientGUICanvasMedia.CanDisplayMedia( second_media, self.CANVAS_TYPE ):
return False
return True
#
self._processed_pairs.append( process_tuple )
self._current_pair_index += 1
while True:
num_remaining = self._GetNumRemainingDecisions()
if num_remaining == 0:
num_committable = self._GetNumCommittableDecisions()
num_deletable = self._GetNumCommittableDeletes()
if num_committable + num_deletable > 0:
components = []
if num_committable > 0:
components.append( '{} decisions'.format( HydrusData.ToHumanInt( num_committable ) ) )
if num_deletable > 0:
components.append( '{} deletes'.format( HydrusData.ToHumanInt( num_deletable ) ) )
label = 'commit {} and continue?'.format( ' and '.join( components ) )
result = ClientGUIDialogsQuick.GetInterstitialFilteringAnswer( self, label )
if result == QW.QDialog.Accepted:
self._CommitProcessed( blocking = True )
else:
it_went_ok = self._RewindProcessing()
if it_went_ok:
self._ShowCurrentPair()
return
else:
# nothing to commit, so let's see if we have a big problem here or if user just skipped all
we_saw_a_non_auto_skip = False
for ( duplicate_type, first_media, second_media, list_of_service_keys_to_content_updates, was_auto_skipped ) in self._processed_pairs:
if not was_auto_skipped:
we_saw_a_non_auto_skip = True
break
if not we_saw_a_non_auto_skip:
HG.client_controller.pub( 'new_similar_files_potentials_search_numbers' )
QW.QMessageBox.critical( self, 'Error', 'It seems an entire batch of pairs were unable to be displayed. The duplicate filter will now close.' )
self.window().deleteLater()
return
self._LoadNextBatchOfPairs()
return
current_pair = self._batch_of_pairs_to_process[ self._current_pair_index ]
if pair_is_good( current_pair ):
self._ShowCurrentPair()
return
else:
was_auto_skipped = True
self._processed_pairs.append( ( None, None, None, [], was_auto_skipped ) )
self._current_pair_index += 1
def _ShowPairInPage( self ):
if self._current_media is None:
return
self.showPairInPage.emit( [ self._current_media, self._media_list.GetNext( self._current_media ) ] )
def _SkipPair( self ):
if self._current_media is None:
return
was_auto_skipped = False
( first_media_result, second_media_result ) = self._batch_of_pairs_to_process[ self._current_pair_index ]
first_media = ClientMedia.MediaSingleton( first_media_result )
second_media = ClientMedia.MediaSingleton( second_media_result )
process_tuple = ( None, first_media, second_media, [], was_auto_skipped )
self._ShowNextPair( process_tuple )
def _SwitchMedia( self ):
if self._current_media is not None:
try:
other_media = self._media_list.GetNext( self._current_media )
self.SetMedia( other_media )
except HydrusExceptions.DataMissing:
return
def Archive( self, canvas_key ):
if self._canvas_key == canvas_key:
self._Archive()
def CleanBeforeDestroy( self ):
HG.client_controller.pub( 'new_similar_files_potentials_search_numbers' )
ClientDuplicates.hashes_to_jpeg_quality = {} # clear the cache
ClientDuplicates.hashes_to_pixel_hashes = {} # clear the cache
CanvasWithHovers.CleanBeforeDestroy( self )
def Delete( self, canvas_key ):
if self._canvas_key == canvas_key:
self._Delete()
def Inbox( self, canvas_key ):
if self._canvas_key == canvas_key:
self._Inbox()
def ProcessApplicationCommand( self, command: CAC.ApplicationCommand ):
command_processed = True
if command.IsSimpleCommand():
action = command.GetSimpleAction()
if action == CAC.SIMPLE_DUPLICATE_FILTER_THIS_IS_BETTER_AND_DELETE_OTHER:
self._CurrentMediaIsBetter( delete_second = True )
elif action == CAC.SIMPLE_DUPLICATE_FILTER_THIS_IS_BETTER_BUT_KEEP_BOTH:
self._CurrentMediaIsBetter( delete_second = False )
elif action == CAC.SIMPLE_DUPLICATE_FILTER_EXACTLY_THE_SAME:
self._MediaAreTheSame()
elif action == CAC.SIMPLE_DUPLICATE_FILTER_ALTERNATES:
self._MediaAreAlternates()
elif action == CAC.SIMPLE_DUPLICATE_FILTER_FALSE_POSITIVE:
self._MediaAreFalsePositive()
elif action == CAC.SIMPLE_DUPLICATE_FILTER_CUSTOM_ACTION:
self._DoCustomAction()
elif action == CAC.SIMPLE_DUPLICATE_FILTER_SKIP:
self._SkipPair()
elif action == CAC.SIMPLE_DUPLICATE_FILTER_BACK:
self._GoBack()
elif action in ( CAC.SIMPLE_VIEW_FIRST, CAC.SIMPLE_VIEW_LAST, CAC.SIMPLE_VIEW_PREVIOUS, CAC.SIMPLE_VIEW_NEXT ):
self._SwitchMedia()
else:
command_processed = False
else:
command_processed = False
if not command_processed:
command_processed = CanvasWithHovers.ProcessApplicationCommand( self, command )
return command_processed
def ProcessContentUpdates( self, service_keys_to_content_updates ):
def catch_up():
# ugly, but it will do for now
if len( self._media_list ) < 2 and len( self._batch_of_pairs_to_process ) > self._current_pair_index:
was_auto_skipped = True
( first_media_result, second_media_result ) = self._batch_of_pairs_to_process[ self._current_pair_index ]
first_media = ClientMedia.MediaSingleton( first_media_result )
second_media = ClientMedia.MediaSingleton( second_media_result )
process_tuple = ( None, first_media, second_media, [], was_auto_skipped )
self._ShowNextPair( process_tuple )
else:
self.update()
HG.client_controller.CallLaterQtSafe( self, 0.01, 'duplicates filter post-processing wait', catch_up )
def SetMedia( self, media ):
CanvasWithHovers.SetMedia( self, media )
if media is not None:
shown_media = self._current_media
comparison_media = self._media_list.GetNext( shown_media )
if shown_media != comparison_media:
HG.client_controller.pub( 'canvas_new_duplicate_pair', self._canvas_key, shown_media, comparison_media )
def SwitchMedia( self, canvas_key ):
if canvas_key == self._canvas_key:
self._SwitchMedia()
def TryToDoPreClose( self ):
num_committable = self._GetNumCommittableDecisions()
num_deletable = self._GetNumCommittableDeletes()
if num_committable + num_deletable > 0:
components = []
if num_committable > 0:
components.append( '{} decisions'.format( HydrusData.ToHumanInt( num_committable ) ) )
if num_deletable > 0:
components.append( '{} deletes'.format( HydrusData.ToHumanInt( num_deletable ) ) )
label = 'commit {}?'.format( ' and '.join( components ) )
( result, cancelled ) = ClientGUIDialogsQuick.GetFinishFilteringAnswer( self, label )
if cancelled:
close_was_triggered_by_everything_being_processed = self._GetNumRemainingDecisions() == 0
if close_was_triggered_by_everything_being_processed:
self._GoBack()
return False
elif result == QW.QDialog.Accepted:
self._CommitProcessed( blocking = False )
return CanvasWithHovers.TryToDoPreClose( self )
def Undelete( self, canvas_key ):
if canvas_key == self._canvas_key:
self._Undelete()
def THREADFetchPairs( self, file_search_context_1, file_search_context_2, dupe_search_type, pixel_dupes_preference, max_hamming_distance ):
def qt_close():
if not self or not QP.isValid( self ):
return
QW.QMessageBox.information( self, 'Information', 'All pairs have been filtered!' )
self._TryToCloseWindow()
def qt_continue( unprocessed_pairs ):
if not self or not QP.isValid( self ):
return
self._batch_of_pairs_to_process = unprocessed_pairs
self._current_pair_index = 0
self._currently_fetching_pairs = False
self._ShowCurrentPair()
result = HG.client_controller.Read( 'duplicate_pairs_for_filtering', file_search_context_1, file_search_context_2, dupe_search_type, pixel_dupes_preference, max_hamming_distance )
if len( result ) == 0:
QP.CallAfter( qt_close )
else:
QP.CallAfter( qt_continue, result )
class CanvasMediaList( ClientMedia.ListeningMediaList, CanvasWithHovers ):
exitFocusMedia = QC.Signal( ClientMedia.Media )
def __init__( self, parent, page_key, location_context: ClientLocation.LocationContext, media_results ):
CanvasWithHovers.__init__( self, parent, location_context )
ClientMedia.ListeningMediaList.__init__( self, location_context, media_results )
self._page_key = page_key
self._just_started = True
def TryToDoPreClose( self ):
if self._current_media is not None:
self.exitFocusMedia.emit( self._current_media )
return CanvasWithHovers.TryToDoPreClose( self )
def _GenerateHoverTopFrame( self ):
raise NotImplementedError()
def _GetIndexString( self ):
if self._current_media is None:
index_string = '-/' + HydrusData.ToHumanInt( len( self._sorted_media ) )
else:
index_string = HydrusData.ConvertValueRangeToPrettyString( self._sorted_media.index( self._current_media ) + 1, len( self._sorted_media ) )
return index_string
def _PrefetchNeighbours( self ):
media_looked_at = set()
to_render = []
previous = self._current_media
next = self._current_media
delay_base = HG.client_controller.new_options.GetInteger( 'media_viewer_prefetch_delay_base_ms' ) / 1000
num_to_go_back = HG.client_controller.new_options.GetInteger( 'media_viewer_prefetch_num_previous' )
num_to_go_forward = HG.client_controller.new_options.GetInteger( 'media_viewer_prefetch_num_next' )
# if media_looked_at nukes the list, we want shorter delays, so do next first
for i in range( num_to_go_forward ):
next = self._GetNext( next )
if next in media_looked_at:
break
else:
media_looked_at.add( next )
delay = delay_base * ( i + 1 )
to_render.append( ( next, delay ) )
for i in range( num_to_go_back ):
previous = self._GetPrevious( previous )
if previous in media_looked_at:
break
else:
media_looked_at.add( previous )
delay = delay_base * 2 * ( i + 1 )
to_render.append( ( previous, delay ) )
image_cache = HG.client_controller.GetCache( 'images' )
for ( media, delay ) in to_render:
hash = media.GetHash()
mime = media.GetMime()
if media.IsStaticImage():
if not image_cache.HasImageRenderer( hash ):
# we do qt safe to make sure the job is cancelled if we are destroyed
HG.client_controller.CallLaterQtSafe( self, delay, 'image pre-fetch', image_cache.PrefetchImageRenderer, media )
def _Remove( self ):
next_media = self._GetNext( self._current_media )
if next_media == self._current_media:
next_media = None
hashes = { self._current_media.GetHash() }
HG.client_controller.pub( 'remove_media', self._page_key, hashes )
singleton_media = { self._current_media }
ClientMedia.ListeningMediaList._RemoveMediaDirectly( self, singleton_media, {} )
if self.HasNoMedia():
self._TryToCloseWindow()
elif self.HasMedia( self._current_media ):
HG.client_controller.pub( 'canvas_new_index_string', self._canvas_key, self._GetIndexString() )
self.update()
else:
self.SetMedia( next_media )
def _ShowFirst( self ):
self.SetMedia( self._GetFirst() )
def _ShowLast( self ):
self.SetMedia( self._GetLast() )
def _ShowNext( self ):
self.SetMedia( self._GetNext( self._current_media ) )
def _ShowPrevious( self ):
self.SetMedia( self._GetPrevious( self._current_media ) )
def _StartSlideshow( self, interval: float ):
pass
def AddMediaResults( self, page_key, media_results ):
if page_key == self._page_key:
ClientMedia.ListeningMediaList.AddMediaResults( self, media_results )
HG.client_controller.pub( 'canvas_new_index_string', self._canvas_key, self._GetIndexString() )
self.update()
def EventFullscreenSwitch( self, event ):
self.parentWidget().FullscreenSwitch()
def ProcessContentUpdates( self, service_keys_to_content_updates ):
if self._current_media is None:
# probably a file view stats update as we close down--ignore it
return
if self.HasMedia( self._current_media ):
next_media = self._GetNext( self._current_media )
if next_media == self._current_media:
next_media = None
else:
next_media = None
ClientMedia.ListeningMediaList.ProcessContentUpdates( self, service_keys_to_content_updates )
if self.HasNoMedia():
self._TryToCloseWindow()
elif self.HasMedia( self._current_media ):
HG.client_controller.pub( 'canvas_new_index_string', self._canvas_key, self._GetIndexString() )
self.update()
elif self.HasMedia( next_media ):
self.SetMedia( next_media )
else:
self.SetMedia( self._GetFirst() )
def CommitArchiveDelete( page_key: bytes, location_context: ClientLocation.LocationContext, kept: typing.Collection[ ClientMedia.MediaSingleton ], deleted: typing.Collection[ ClientMedia.MediaSingleton ] ):
kept = list( kept )
deleted = list( deleted )
kept_hashes = [ m.GetHash() for m in kept ]
deleted_hashes = [ m.GetHash() for m in deleted ]
if HC.options[ 'remove_filtered_files' ]:
all_hashes = set()
all_hashes.update( kept_hashes )
all_hashes.update( deleted_hashes )
HG.client_controller.pub( 'remove_media', page_key, all_hashes )
location_context = location_context.Duplicate()
location_context.FixMissingServices( ClientLocation.ValidLocalDomainsFilter )
if location_context.IncludesCurrent():
deletee_file_service_keys = location_context.current_service_keys
else:
# if we are in a weird search domain, then just say 'delete from all local'
deletee_file_service_keys = [ CC.COMBINED_LOCAL_MEDIA_SERVICE_KEY ]
for block_of_deleted in HydrusData.SplitListIntoChunks( deleted, 64 ):
service_keys_to_content_updates = {}
reason = 'Deleted in Archive/Delete filter.'
for deletee_file_service_key in deletee_file_service_keys:
block_of_deleted_hashes = [ m.GetHash() for m in block_of_deleted if deletee_file_service_key in m.GetLocationsManager().GetCurrent() ]
service_keys_to_content_updates[ deletee_file_service_key ] = [ HydrusData.ContentUpdate( HC.CONTENT_TYPE_FILES, HC.CONTENT_UPDATE_DELETE, block_of_deleted_hashes, reason = reason ) ]
HG.client_controller.WriteSynchronous( 'content_updates', service_keys_to_content_updates )
# we do a second set of removes to deal with late processing and a quick F5ing user
if HC.options[ 'remove_filtered_files' ]:
block_of_deleted_hashes = [ m.GetHash() for m in block_of_deleted ]
HG.client_controller.pub( 'remove_media', page_key, block_of_deleted_hashes )
HG.client_controller.WaitUntilViewFree()
for block_of_kept_hashes in HydrusData.SplitListIntoChunks( kept_hashes, 64 ):
service_keys_to_content_updates = {}
service_keys_to_content_updates[ CC.COMBINED_LOCAL_FILE_SERVICE_KEY ] = [ HydrusData.ContentUpdate( HC.CONTENT_TYPE_FILES, HC.CONTENT_UPDATE_ARCHIVE, block_of_kept_hashes ) ]
HG.client_controller.WriteSynchronous( 'content_updates', service_keys_to_content_updates )
if HC.options[ 'remove_filtered_files' ]:
HG.client_controller.pub( 'remove_media', page_key, block_of_kept_hashes )
HG.client_controller.WaitUntilViewFree()
class CanvasMediaListFilterArchiveDelete( CanvasMediaList ):
def __init__( self, parent, page_key, location_context: ClientLocation.LocationContext, media_results ):
CanvasMediaList.__init__( self, parent, page_key, location_context, media_results )
self._my_shortcuts_handler.AddShortcuts( 'archive_delete_filter' )
self._kept = set()
self._deleted = set()
HG.client_controller.sub( self, 'Delete', 'canvas_delete' )
HG.client_controller.sub( self, 'Undelete', 'canvas_undelete' )
first_media = self._GetFirst()
if first_media is not None:
QP.CallAfter( self.SetMedia, first_media ) # don't set this until we have a size > (20, 20)!
def _Back( self ):
if self._current_media == self._GetFirst():
return
else:
self._ShowPrevious()
self._kept.discard( self._current_media )
self._deleted.discard( self._current_media )
def TryToDoPreClose( self ):
kept = list( self._kept )
deleted = ClientMedia.FilterAndReportDeleteLockFailures( self._deleted )
if len( kept ) > 0 or len( deleted ) > 0:
if len( kept ) > 0:
kept_label = 'keep {}'.format( HydrusData.ToHumanInt( len( kept ) ) )
else:
kept_label = None
deletion_options = []
if len( deleted ) > 0:
location_contexts_to_present_options_for = []
if not self._location_context.IsAllLocalFiles():
location_contexts_to_present_options_for.append( self._location_context )
current_local_service_keys = HydrusData.MassUnion( [ m.GetLocationsManager().GetCurrent() for m in deleted ] )
local_file_domain_service_keys = [ service_key for service_key in current_local_service_keys if HG.client_controller.services_manager.GetServiceType( service_key ) == HC.LOCAL_FILE_DOMAIN ]
location_contexts_to_present_options_for.extend( [ ClientLocation.LocationContext.STATICCreateSimple( service_key ) for service_key in local_file_domain_service_keys ] )
all_my_files_location_context = ClientLocation.LocationContext.STATICCreateSimple( CC.COMBINED_LOCAL_MEDIA_SERVICE_KEY )
if len( local_file_domain_service_keys ) > 1:
location_contexts_to_present_options_for.append( all_my_files_location_context )
elif len( local_file_domain_service_keys ) == 1:
if all_my_files_location_context in location_contexts_to_present_options_for:
location_contexts_to_present_options_for.remove( all_my_files_location_context )
if CC.TRASH_SERVICE_KEY in current_local_service_keys or CC.LOCAL_UPDATE_SERVICE_KEY in current_local_service_keys:
location_contexts_to_present_options_for.append( ClientLocation.LocationContext.STATICCreateSimple( CC.COMBINED_LOCAL_FILE_SERVICE_KEY ) )
location_contexts_to_present_options_for = HydrusData.DedupeList( location_contexts_to_present_options_for )
for location_context in location_contexts_to_present_options_for:
file_service_keys = location_context.current_service_keys
num_deletable = len( [ m for m in deleted if len( m.GetLocationsManager().GetCurrent().intersection( file_service_keys ) ) > 0 ] )
if num_deletable > 0:
if location_context == ClientLocation.LocationContext.STATICCreateSimple( CC.COMBINED_LOCAL_MEDIA_SERVICE_KEY ):
location_label = 'all local file domains'
elif location_context == ClientLocation.LocationContext.STATICCreateSimple( CC.COMBINED_LOCAL_FILE_SERVICE_KEY ):
location_label = 'my hard disk'
else:
location_label = location_context.ToString( HG.client_controller.services_manager.GetName )
delete_label = 'delete {} from {}'.format( HydrusData.ToHumanInt( num_deletable ), location_label )
deletion_options.append( ( location_context, delete_label ) )
( result, deletee_location_context, cancelled ) = ClientGUIDialogsQuick.GetFinishArchiveDeleteFilteringAnswer( self, kept_label, deletion_options )
if cancelled:
if self._current_media in self._kept:
self._kept.remove( self._current_media )
if self._current_media in self._deleted:
self._deleted.remove( self._current_media )
return False
elif result == QW.QDialog.Accepted:
self._kept = set()
self._deleted = set()
self._current_media = self._GetFirst() # so the pubsub on close is better
HG.client_controller.CallToThread( CommitArchiveDelete, self._page_key, deletee_location_context, kept, deleted )
return CanvasMediaList.TryToDoPreClose( self )
def _Delete( self, media = None, reason = None, file_service_key = None ):
if self._current_media is None:
return False
if self._current_media.HasDeleteLocked():
message = 'This file is delete-locked! Send it back to the inbox to delete it!'
QW.QMessageBox.information( self, 'Locked!', message )
return False
self._deleted.add( self._current_media )
if self._current_media == self._GetLast():
self._TryToCloseWindow()
else:
self._ShowNext()
return True
def _GenerateHoverTopFrame( self ):
return ClientGUICanvasHoverFrames.CanvasHoverFrameTopArchiveDeleteFilter( self, self, self._canvas_key )
def _Keep( self ):
self._kept.add( self._current_media )
if self._current_media == self._GetLast():
self._TryToCloseWindow()
else:
self._ShowNext()
def _Skip( self ):
if self._current_media == self._GetLast():
self._TryToCloseWindow()
else:
self._ShowNext()
def Keep( self, canvas_key ):
if canvas_key == self._canvas_key:
self._Keep()
def Back( self, canvas_key ):
if canvas_key == self._canvas_key:
self._Back()
def Delete( self, canvas_key ):
if canvas_key == self._canvas_key:
self._Delete()
def ProcessApplicationCommand( self, command: CAC.ApplicationCommand ):
command_processed = True
if command.IsSimpleCommand():
action = command.GetSimpleAction()
if action in ( CAC.SIMPLE_ARCHIVE_DELETE_FILTER_KEEP, CAC.SIMPLE_ARCHIVE_FILE ):
self._Keep()
elif action in ( CAC.SIMPLE_ARCHIVE_DELETE_FILTER_DELETE, CAC.SIMPLE_DELETE_FILE ):
self._Delete()
elif action == CAC.SIMPLE_ARCHIVE_DELETE_FILTER_SKIP:
self._Skip()
elif action == CAC.SIMPLE_ARCHIVE_DELETE_FILTER_BACK:
self._Back()
elif action == CAC.SIMPLE_LAUNCH_THE_ARCHIVE_DELETE_FILTER:
self._TryToCloseWindow()
else:
command_processed = False
else:
command_processed = False
if not command_processed:
command_processed = CanvasMediaList.ProcessApplicationCommand( self, command )
return command_processed
def Skip( self, canvas_key ):
if canvas_key == self._canvas_key:
self._Skip()
def Undelete( self, canvas_key ):
if canvas_key == self._canvas_key:
self._Undelete()
class CanvasMediaListNavigable( CanvasMediaList ):
userChangedMedia = QC.Signal()
def __init__( self, parent, page_key, location_context: ClientLocation.LocationContext, media_results ):
CanvasMediaList.__init__( self, parent, page_key, location_context, media_results )
self._my_shortcuts_handler.AddShortcuts( 'media_viewer_browser' )
HG.client_controller.sub( self, 'Delete', 'canvas_delete' )
HG.client_controller.sub( self, 'ShowNext', 'canvas_show_next' )
HG.client_controller.sub( self, 'ShowPrevious', 'canvas_show_previous' )
HG.client_controller.sub( self, 'Undelete', 'canvas_undelete' )
def _GenerateHoverTopFrame( self ):
return ClientGUICanvasHoverFrames.CanvasHoverFrameTopNavigableList( self, self, self._canvas_key )
def Archive( self, canvas_key ):
if self._canvas_key == canvas_key:
self._Archive()
def Delete( self, canvas_key ):
if self._canvas_key == canvas_key:
self._Delete()
def Inbox( self, canvas_key ):
if self._canvas_key == canvas_key:
self._Inbox()
def ProcessApplicationCommand( self, command: CAC.ApplicationCommand ):
command_processed = True
if command.IsSimpleCommand():
action = command.GetSimpleAction()
if action == CAC.SIMPLE_REMOVE_FILE_FROM_VIEW:
self._Remove()
elif action == CAC.SIMPLE_VIEW_FIRST:
self._ShowFirst()
self.userChangedMedia.emit()
elif action == CAC.SIMPLE_VIEW_LAST:
self._ShowLast()
self.userChangedMedia.emit()
elif action == CAC.SIMPLE_VIEW_PREVIOUS:
self._ShowPrevious()
self.userChangedMedia.emit()
elif action == CAC.SIMPLE_VIEW_NEXT:
self._ShowNext()
self.userChangedMedia.emit()
else:
command_processed = False
else:
command_processed = False
if not command_processed:
command_processed = CanvasMediaList.ProcessApplicationCommand( self, command )
return command_processed
def ShowFirst( self, canvas_key ):
if canvas_key == self._canvas_key:
self._ShowFirst()
self.userChangedMedia.emit()
def ShowLast( self, canvas_key ):
if canvas_key == self._canvas_key:
self._ShowLast()
self.userChangedMedia.emit()
def ShowNext( self, canvas_key ):
if canvas_key == self._canvas_key:
self._ShowNext()
self.userChangedMedia.emit()
def ShowPrevious( self, canvas_key ):
if canvas_key == self._canvas_key:
self._ShowPrevious()
self.userChangedMedia.emit()
def Undelete( self, canvas_key ):
if canvas_key == self._canvas_key:
self._Undelete()
class CanvasMediaListBrowser( CanvasMediaListNavigable ):
def __init__( self, parent, page_key, location_context: ClientLocation.LocationContext, media_results, first_hash ):
CanvasMediaListNavigable.__init__( self, parent, page_key, location_context, media_results )
self._timer_slideshow_job = None
self._timer_slideshow_interval = 0.0
if first_hash is None:
first_media = self._GetFirst()
else:
try:
first_media = self._GetMedia( { first_hash } )[0]
except:
first_media = self._GetFirst()
if first_media is not None:
QP.CallAfter( self.SetMedia, first_media ) # don't set this until we have a size > (20, 20)!
HG.client_controller.sub( self, 'AddMediaResults', 'add_media_results' )
self.userChangedMedia.connect( self.NotifyUserChangedMedia )
def _PausePlaySlideshow( self ):
if self._SlideshowIsRunning():
self._StopSlideshow()
elif self._timer_slideshow_interval > 0:
self._StartSlideshow( interval = self._timer_slideshow_interval )
def _SlideshowIsRunning( self ):
return self._timer_slideshow_job is not None
def _StartSlideshow( self, interval: float ):
self._StopSlideshow()
if interval > 0:
self._timer_slideshow_interval = interval
self._timer_slideshow_job = HG.client_controller.CallLaterQtSafe( self, self._timer_slideshow_interval, 'slideshow', self.DoSlideshow )
def _StartSlideshowCustomInterval( self ):
with ClientGUIDialogs.DialogTextEntry( self, 'Enter the interval, in seconds.', default = '15', min_char_width = 12 ) as dlg:
if dlg.exec() == QW.QDialog.Accepted:
try:
interval = float( dlg.GetValue() )
self._StartSlideshow( interval )
except:
pass
def _StopSlideshow( self ):
if self._SlideshowIsRunning():
self._timer_slideshow_job.Cancel()
self._timer_slideshow_job = None
self._media_container.StopForSlideshow( False )
def DoSlideshow( self ):
try:
# we are due for a slideshow change, so tell movie to stop for it once it has played once through
# if short movie, it has prob played through a lot and will shift immediately
# if longer movie, it has prob not played through but will now stop when it has and wait for us to change it then
self._media_container.StopForSlideshow( True )
if self._current_media is not None and self._SlideshowIsRunning():
if self._media_container.ReadyToSlideshow() and not CGC.core().MenuIsOpen():
self._media_container.StopForSlideshow( False )
self._ShowNext()
self._timer_slideshow_job = HG.client_controller.CallLaterQtSafe( self, self._timer_slideshow_interval, 'slideshow', self.DoSlideshow )
else:
self._timer_slideshow_job = HG.client_controller.CallLaterQtSafe( self, 0.1, 'slideshow', self.DoSlideshow )
except:
self._timer_slideshow_job = None
raise
def contextMenuEvent( self, event ):
if event.reason() == QG.QContextMenuEvent.Keyboard:
self.ShowMenu()
def NotifyUserChangedMedia( self ):
# reset the timer if user overrode
if self._SlideshowIsRunning():
self._StartSlideshow( interval = self._timer_slideshow_interval )
def ProcessApplicationCommand( self, command: CAC.ApplicationCommand ):
command_processed = True
if command.IsSimpleCommand():
action = command.GetSimpleAction()
if action == CAC.SIMPLE_PAUSE_PLAY_SLIDESHOW:
self._PausePlaySlideshow()
elif action == CAC.SIMPLE_SHOW_MENU:
self.ShowMenu()
else:
command_processed = False
else:
command_processed = False
if not command_processed:
command_processed = CanvasMediaListNavigable.ProcessApplicationCommand( self, command )
return command_processed
def ShowMenu( self ):
if self._current_media is not None:
new_options = HG.client_controller.new_options
advanced_mode = new_options.GetBoolean( 'advanced_mode' )
services = HG.client_controller.services_manager.GetServices()
local_ratings_services = [ service for service in services if service.GetServiceType() in ( HC.LOCAL_RATING_LIKE, HC.LOCAL_RATING_NUMERICAL ) ]
i_can_post_ratings = len( local_ratings_services ) > 0
self.EndDrag() # to stop successive right-click drag warp bug
locations_manager = self._current_media.GetLocationsManager()
menu = QW.QMenu()
#
info_lines = self._current_media.GetPrettyInfoLines()
top_line = info_lines.pop( 0 )
info_menu = QW.QMenu( menu )
ClientGUIMediaMenus.AddPrettyInfoLines( info_menu, info_lines )
ClientGUIMediaMenus.AddFileViewingStatsMenu( info_menu, ( self._current_media, ) )
ClientGUIMenus.AppendMenu( menu, info_menu, top_line )
#
ClientGUIMenus.AppendSeparator( menu )
if self._media_container.IsZoomable():
zoom_menu = QW.QMenu( menu )
ClientGUIMenus.AppendMenuItem( zoom_menu, 'zoom in', 'Zoom the media in.', self._media_container.ZoomIn )
ClientGUIMenus.AppendMenuItem( zoom_menu, 'zoom out', 'Zoom the media out.', self._media_container.ZoomOut )
current_zoom = self._media_container.GetCurrentZoom()
if current_zoom != 1.0:
ClientGUIMenus.AppendMenuItem( zoom_menu, 'zoom to 100%', 'Set the zoom to 100%.', self._media_container.ZoomSwitch )
elif current_zoom != self._media_container.GetCanvasZoom():
ClientGUIMenus.AppendMenuItem( zoom_menu, 'zoom fit', 'Set the zoom so the media fits the canvas.', self._media_container.ZoomSwitch )
if not self._media_container.IsAtMaxZoom():
ClientGUIMenus.AppendMenuItem( zoom_menu, 'zoom to max', 'Set the zoom to the maximum possible.', self._media_container.ZoomMax )
ClientGUIMenus.AppendMenu( menu, zoom_menu, 'current zoom: {}'.format( ClientData.ConvertZoomToPercentage( self._media_container.GetCurrentZoom() ) ) )
AddAudioVolumeMenu( menu, self.CANVAS_TYPE )
if self.parentWidget().isFullScreen():
ClientGUIMenus.AppendMenuItem( menu, 'exit fullscreen', 'Make this media viewer a regular window with borders.', self.parentWidget().FullscreenSwitch )
else:
ClientGUIMenus.AppendMenuItem( menu, 'go fullscreen', 'Make this media viewer a fullscreen window without borders.', self.parentWidget().FullscreenSwitch )
slideshow = QW.QMenu( menu )
ClientGUIMenus.AppendMenuItem( slideshow, '1 second', 'Start a slideshow with a one second interval.', self._StartSlideshow, 1.0 )
ClientGUIMenus.AppendMenuItem( slideshow, '5 second', 'Start a slideshow with a five second interval.', self._StartSlideshow, 5.0 )
ClientGUIMenus.AppendMenuItem( slideshow, '10 second', 'Start a slideshow with a ten second interval.', self._StartSlideshow, 10.0 )
ClientGUIMenus.AppendMenuItem( slideshow, '30 second', 'Start a slideshow with a thirty second interval.', self._StartSlideshow, 30.0 )
ClientGUIMenus.AppendMenuItem( slideshow, '60 second', 'Start a slideshow with a one minute interval.', self._StartSlideshow, 60.0 )
ClientGUIMenus.AppendMenuItem( slideshow, 'very fast', 'Start a very fast slideshow.', self._StartSlideshow, 0.08 )
ClientGUIMenus.AppendMenuItem( slideshow, 'custom interval', 'Start a slideshow with a custom interval.', self._StartSlideshowCustomInterval )
ClientGUIMenus.AppendMenu( menu, slideshow, 'start slideshow' )
if self._SlideshowIsRunning():
ClientGUIMenus.AppendMenuItem( menu, 'stop slideshow', 'Stop the current slideshow.', self._PausePlaySlideshow )
ClientGUIMenus.AppendSeparator( menu )
ClientGUIMenus.AppendMenuItem( menu, 'remove from view', 'Remove this file from the list you are viewing.', self._Remove )
ClientGUIMenus.AppendSeparator( menu )
if self._current_media.HasInbox():
ClientGUIMenus.AppendMenuItem( menu, 'archive', 'Archive this file, taking it out of the inbox.', self._Archive )
elif self._current_media.HasArchive() and self._current_media.GetLocationsManager().IsLocal():
ClientGUIMenus.AppendMenuItem( menu, 'return to inbox', 'Put this file back in the inbox.', self._Inbox )
ClientGUIMenus.AppendSeparator( menu )
#
local_file_service_keys = HG.client_controller.services_manager.GetServiceKeys( ( HC.LOCAL_FILE_DOMAIN, ) )
# brush this up to handle different service keys
# undelete do an optional service key too
local_file_service_keys_we_are_in = sorted( locations_manager.GetCurrent().intersection( local_file_service_keys ), key = HG.client_controller.services_manager.GetName )
for file_service_key in local_file_service_keys_we_are_in:
ClientGUIMenus.AppendMenuItem( menu, 'delete from {}'.format( HG.client_controller.services_manager.GetName( file_service_key ) ), 'Delete this file.', self._Delete, file_service_key = file_service_key )
#
if locations_manager.IsTrashed():
ClientGUIMenus.AppendMenuItem( menu, 'delete physically now', 'Delete this file immediately. This cannot be undone.', self._Delete, file_service_key = CC.COMBINED_LOCAL_FILE_SERVICE_KEY )
ClientGUIMenus.AppendMenuItem( menu, 'undelete', 'Take this file out of the trash, returning it to its original file service.', self._Undelete )
ClientGUIMenus.AppendSeparator( menu )
manage_menu = QW.QMenu( menu )
ClientGUIMenus.AppendMenuItem( manage_menu, 'tags', 'Manage this file\'s tags.', self._ManageTags )
if i_can_post_ratings:
ClientGUIMenus.AppendMenuItem( manage_menu, 'ratings', 'Manage this file\'s ratings.', self._ManageRatings )
ClientGUIMenus.AppendMenuItem( manage_menu, 'urls', 'Manage this file\'s known urls.', self._ManageURLs )
num_notes = self._current_media.GetNotesManager().GetNumNotes()
notes_str = 'notes'
if num_notes > 0:
notes_str = '{} ({})'.format( notes_str, HydrusData.ToHumanInt( num_notes ) )
ClientGUIMenus.AppendMenuItem( manage_menu, notes_str, 'Manage this file\'s notes.', self._ManageNotes )
ClientGUIMediaMenus.AddManageFileViewingStatsMenu( self, manage_menu, [ self._current_media ] )
ClientGUIMenus.AppendMenu( menu, manage_menu, 'manage' )
( local_duplicable_to_file_service_keys, local_moveable_from_and_to_file_service_keys ) = ClientGUIMediaActions.GetLocalFileActionServiceKeys( ( self._current_media, ) )
multiple_selected = False
ClientGUIMediaMenus.AddLocalFilesMoveAddToMenu( self, menu, local_duplicable_to_file_service_keys, local_moveable_from_and_to_file_service_keys, multiple_selected, self.ProcessApplicationCommand )
ClientGUIMediaMenus.AddKnownURLsViewCopyMenu( self, menu, self._current_media )
open_menu = QW.QMenu( menu )
ClientGUIMenus.AppendMenuItem( open_menu, 'in external program', 'Open this file in the default external program.', self._OpenExternally )
ClientGUIMenus.AppendMenuItem( open_menu, 'in a new page', 'Show your current media in a simple new page.', self._ShowMediaInNewPage )
ClientGUIMenus.AppendMenuItem( open_menu, 'in web browser', 'Show this file in your OS\'s web browser.', self._OpenFileInWebBrowser )
show_open_in_explorer = advanced_mode and ( HC.PLATFORM_WINDOWS or HC.PLATFORM_MACOS )
if show_open_in_explorer:
ClientGUIMenus.AppendMenuItem( open_menu, 'in file browser', 'Show this file in your OS\'s file browser.', self._OpenFileLocation )
ClientGUIMenus.AppendMenu( menu, open_menu, 'open' )
share_menu = QW.QMenu( menu )
copy_menu = QW.QMenu( share_menu )
ClientGUIMenus.AppendMenuItem( copy_menu, 'file', 'Copy this file to your clipboard.', self._CopyFileToClipboard )
copy_hash_menu = QW.QMenu( copy_menu )
ClientGUIMenus.AppendMenuItem( copy_hash_menu, 'sha256 ({})'.format( self._current_media.GetHash().hex() ), 'Copy this file\'s SHA256 hash to your clipboard.', self._CopyHashToClipboard, 'sha256' )
ClientGUIMenus.AppendMenuItem( copy_hash_menu, 'md5', 'Copy this file\'s MD5 hash to your clipboard.', self._CopyHashToClipboard, 'md5' )
ClientGUIMenus.AppendMenuItem( copy_hash_menu, 'sha1', 'Copy this file\'s SHA1 hash to your clipboard.', self._CopyHashToClipboard, 'sha1' )
ClientGUIMenus.AppendMenuItem( copy_hash_menu, 'sha512', 'Copy this file\'s SHA512 hash to your clipboard.', self._CopyHashToClipboard, 'sha512' )
ClientGUIMenus.AppendMenu( copy_menu, copy_hash_menu, 'hash' )
if advanced_mode:
hash_id_str = str( self._current_media.GetHashId() )
ClientGUIMenus.AppendMenuItem( copy_menu, 'file_id ({})'.format( hash_id_str ), 'Copy this file\'s internal file/hash_id.', HG.client_controller.pub, 'clipboard', 'text', hash_id_str )
if self._current_media.GetMime() in HC.IMAGES:
ClientGUIMenus.AppendMenuItem( copy_menu, 'image (bitmap)', 'Copy this file to your clipboard as a BMP image.', self._CopyBMPToClipboard )
ClientGUIMenus.AppendMenuItem( copy_menu, 'path', 'Copy this file\'s path to your clipboard.', self._CopyPathToClipboard )
ClientGUIMenus.AppendMenu( share_menu, copy_menu, 'copy' )
ClientGUIMenus.AppendMenu( menu, share_menu, 'share' )
CGC.core().PopupMenu( self, menu )
Reference in New Issue View Git Blame Copy Permalink