Rebuild thumbnails

Author: David Khristepher Santos

Diffusion.Toolkit/Controls/ThumbnailView.xaml

@@ -601,6 +601,7 @@
                     <MenuItem Style="{StaticResource FolderMenuItem}" Header="{lex:Loc Thumbnail.ContextMenu.ScanNew}" Command="{Binding ScanFolderCommand}" />
                     <Separator Style="{StaticResource FolderSeparator}"/>
                     <MenuItem Style="{StaticResource FileMenuItem}" Header="{lex:Loc Thumbnail.ContextMenu.Rescan}" Command="{Binding RescanCommand}" />
+                    <MenuItem Style="{StaticResource FileMenuItem}" Header="{lex:Loc Thumbnail.ContextMenu.RebuildThumbnail}" Command="{Binding RebuildThumbnailCommand}" />
                 </ContextMenu>
             </ListView.ContextMenu>
             <ListView.Template>

Diffusion.Toolkit/Controls/ThumbnailView.xaml.cs

@@ -76,6 +76,8 @@ public ThumbnailView()
             Model.CopyCommand = new RelayCommand<object>(o => CopySelected());
             Model.MoveCommand = new RelayCommand<object>(o => MoveSelected());
             Model.RescanCommand = new AsyncCommand<object>(o => RescanSelected());
+            Model.RebuildThumbnailCommand = new RelayCommand<object>(o => RebuildThumbnails());
+
             Model.RescanFolderCommand = new AsyncCommand<object>(o => RescanFolder(true));
             Model.ScanFolderCommand = new AsyncCommand<object>(o => RescanFolder(false));
 
@@ -149,6 +151,16 @@ await ServiceLocator.MetadataScannerService.QueueBatchAsync(
         }
 
 
+        private void RebuildThumbnails()
+        {
+            var imageEntries = ThumbnailListView.SelectedItems.Cast<ImageEntry>().ToList();
+
+            foreach (var imageEntry in imageEntries)
+            {
+                ServiceLocator.ThumbnailService.RebuildThumbnail(imageEntry);
+            }
+        }
+
         private void CopySelected()
         {
             var imageEntries = ThumbnailListView.SelectedItems.Cast<ImageEntry>().ToList();

Diffusion.Toolkit/Controls/ThumbnailViewModel.cs

@@ -361,6 +361,7 @@ public int PageSize
     }
 
     public ICommand RescanCommand { get; set; }
+    public ICommand RebuildThumbnailCommand { get; set; }
     public ICommand PermanentlyDeleteCommand { get; set; }
 
     public int ThumbnailSpacing

Diffusion.Toolkit/Localization/default.json

@@ -283,6 +283,7 @@
   "Thumbnail.ContextMenu.ForDeletion": "For _Deletion",
   "Thumbnail.ContextMenu.OpenWith": "Open with...",
   "Thumbnail.ContextMenu.Restore": "_Restore",
+  "Thumbnail.ContextMenu.RebuildThumbnail": "Rebuild Thumbnail",
   "Thumbnail.ContextMenu.Rescan": "Rescan",
   "Thumbnail.ContextMenu.Remove": "Remove",
   "Thumbnail.ContextMenu.PermanentlyDelete": "Permanently Delete",

Diffusion.Toolkit/Thumbnails/ThumbnailJob.cs

@@ -11,4 +11,5 @@ public class ThumbnailJob
     public ImageType Type { get; set; }
     public int Width { get; set; }
     public int Height { get; set; }
+    public bool Rebuild { get; set; }
 }

Diffusion.Toolkit/Thumbnails/ThumbnailService.cs

@@ -70,6 +70,44 @@ public ThumbnailService()
     private Dispatcher _dispatcher => ServiceLocator.Dispatcher;
 
 
+    public void RebuildThumbnail(ImageEntry image)
+    {
+        image.LoadState = LoadState.Loading;
+
+        var job = new ThumbnailJob()
+        {
+            BatchId = image.BatchId,
+            EntryType = image.EntryType,
+            Path = image.Path,
+            Type = image.Type,
+            Height = image.Height,
+            Width = image.Width,
+            Rebuild = true
+        };
+
+        _ = QueueAsync(job, (d) =>
+        {
+            image.LoadState = LoadState.Loaded;
+
+            if (d.Success)
+            {
+                _dispatcher.Invoke(() =>
+                {
+                    image.Thumbnail = d.Image;
+                    image.ThumbnailHeight = d.Image.Height;
+                    image.ThumbnailWidth = d.Image.Width;
+                });
+            }
+            else
+            {
+                _dispatcher.Invoke(() => { image.Unavailable = true; });
+            }
+
+            //Debug.WriteLine($"Finished job {job.RequestId}");
+            //OnPropertyChanged(nameof(Thumbnail));
+        });
+    }
+
     public void QueueImage(ImageEntry image)
     {
         image.LoadState = LoadState.Loading;
@@ -201,7 +239,7 @@ private async Task ProcessTaskAsync(CancellationToken token)
 
                 if (_enableCache)
                 {
-                    if (!ThumbnailCache.Instance.TryGetThumbnail(job.Data.Path, Size,
+                    if (job.Data.Rebuild || !ThumbnailCache.Instance.TryGetThumbnail(job.Data.Path, Size,
                             out BitmapSource? thumbnail))
                     {
                         // Debug.WriteLine($"Loading from disk");