mirror of
https://github.com/aaru-dps/Aaru.git
synced 2025-12-16 19:24:25 +00:00
Migrate image verify from Eto.Forms to Avalonia.
This commit is contained in:
636
Aaru.Gui/ViewModels/ImageVerifyViewModel.cs
Normal file
636
Aaru.Gui/ViewModels/ImageVerifyViewModel.cs
Normal file
@@ -0,0 +1,636 @@
|
||||
using System;
|
||||
using System.Collections.Generic;
|
||||
using System.Collections.ObjectModel;
|
||||
using System.Reactive;
|
||||
using System.Threading;
|
||||
using Aaru.CommonTypes.Interfaces;
|
||||
using Aaru.CommonTypes.Structs;
|
||||
using Aaru.Console;
|
||||
using Aaru.Core;
|
||||
using Aaru.Gui.Models;
|
||||
using Avalonia.Controls;
|
||||
using Avalonia.Threading;
|
||||
using ReactiveUI;
|
||||
|
||||
namespace Aaru.Gui.ViewModels
|
||||
{
|
||||
public class ImageVerifyViewModel : ViewModelBase
|
||||
{
|
||||
readonly IMediaImage _inputFormat;
|
||||
bool _cancel;
|
||||
bool _closeVisible;
|
||||
string _imageResultText;
|
||||
bool _imageResultVisible;
|
||||
bool _optionsVisible;
|
||||
bool _progress2Indeterminate;
|
||||
double _progress2MaxValue;
|
||||
string _progress2Text;
|
||||
double _progress2Value;
|
||||
bool _progress2Visible;
|
||||
bool _progressIndeterminate;
|
||||
double _progressMaxValue;
|
||||
string _progressText;
|
||||
double _progressValue;
|
||||
bool _progressVisible;
|
||||
bool _resultsVisible;
|
||||
string _sectorErrorsText;
|
||||
bool _sectorErrorsVisible;
|
||||
string _sectorsErrorsAllText;
|
||||
bool _sectorsErrorsAllVisible;
|
||||
bool _sectorSummaryVisible;
|
||||
string _sectorsUnknownAllText;
|
||||
bool _sectorsUnknownAllVisible;
|
||||
string _sectorsUnknownsText;
|
||||
bool _sectorsUnknownsVisible;
|
||||
bool _startVisible;
|
||||
bool _stopEnabled;
|
||||
bool _stopVisible;
|
||||
string _totalSectorErrorsText;
|
||||
string _totalSectorErrorsUnknownsText;
|
||||
string _totalSectorsText;
|
||||
string _totalSectorUnknownsText;
|
||||
bool _verifyImageChecked;
|
||||
bool _verifyImageEnabled;
|
||||
bool _verifySectorsChecked;
|
||||
bool _verifySectorsEnabled;
|
||||
bool _verifySectorsVisible;
|
||||
readonly Window _view;
|
||||
|
||||
public ImageVerifyViewModel(IMediaImage inputFormat, Window view)
|
||||
{
|
||||
_view = view;
|
||||
StartCommand = ReactiveCommand.Create(ExecuteStartCommand);
|
||||
CloseCommand = ReactiveCommand.Create(ExecuteCloseCommand);
|
||||
StopCommand = ReactiveCommand.Create(ExecuteStopCommand);
|
||||
_inputFormat = inputFormat;
|
||||
_cancel = false;
|
||||
ErrorList = new ObservableCollection<LbaModel>();
|
||||
UnknownList = new ObservableCollection<LbaModel>();
|
||||
VerifyImageEnabled = true;
|
||||
VerifySectorsEnabled = true;
|
||||
CloseVisible = true;
|
||||
StartVisible = true;
|
||||
OptionsVisible = true;
|
||||
}
|
||||
|
||||
public ObservableCollection<LbaModel> ErrorList { get; }
|
||||
public ObservableCollection<LbaModel> UnknownList { get; }
|
||||
public ReactiveCommand<Unit, Unit> StartCommand { get; }
|
||||
public ReactiveCommand<Unit, Unit> CloseCommand { get; }
|
||||
public ReactiveCommand<Unit, Unit> StopCommand { get; }
|
||||
|
||||
public bool VerifyImageEnabled
|
||||
{
|
||||
get => _verifyImageEnabled;
|
||||
set => this.RaiseAndSetIfChanged(ref _verifyImageEnabled, value);
|
||||
}
|
||||
|
||||
public bool VerifySectorsEnabled
|
||||
{
|
||||
get => _verifySectorsEnabled;
|
||||
set => this.RaiseAndSetIfChanged(ref _verifySectorsEnabled, value);
|
||||
}
|
||||
|
||||
public bool VerifySectorsVisible
|
||||
{
|
||||
get => _verifySectorsVisible;
|
||||
set => this.RaiseAndSetIfChanged(ref _verifySectorsVisible, value);
|
||||
}
|
||||
|
||||
public double ProgressMaxValue
|
||||
{
|
||||
get => _progressMaxValue;
|
||||
set => this.RaiseAndSetIfChanged(ref _progressMaxValue, value);
|
||||
}
|
||||
|
||||
public bool VerifyImageChecked
|
||||
{
|
||||
get => _verifyImageChecked;
|
||||
set => this.RaiseAndSetIfChanged(ref _verifyImageChecked, value);
|
||||
}
|
||||
|
||||
public bool ProgressIndeterminate
|
||||
{
|
||||
get => _progressIndeterminate;
|
||||
set => this.RaiseAndSetIfChanged(ref _progressIndeterminate, value);
|
||||
}
|
||||
|
||||
public bool ImageResultVisible
|
||||
{
|
||||
get => _imageResultVisible;
|
||||
set => this.RaiseAndSetIfChanged(ref _imageResultVisible, value);
|
||||
}
|
||||
|
||||
public string ImageResultText
|
||||
{
|
||||
get => _imageResultText;
|
||||
set => this.RaiseAndSetIfChanged(ref _imageResultText, value);
|
||||
}
|
||||
|
||||
public bool VerifySectorsChecked
|
||||
{
|
||||
get => _verifySectorsChecked;
|
||||
set => this.RaiseAndSetIfChanged(ref _verifySectorsChecked, value);
|
||||
}
|
||||
|
||||
public bool Progress2Visible
|
||||
{
|
||||
get => _progress2Visible;
|
||||
set => this.RaiseAndSetIfChanged(ref _progress2Visible, value);
|
||||
}
|
||||
|
||||
public bool Progress2Indeterminate
|
||||
{
|
||||
get => _progress2Indeterminate;
|
||||
set => this.RaiseAndSetIfChanged(ref _progress2Indeterminate, value);
|
||||
}
|
||||
|
||||
public double Progress2MaxValue
|
||||
{
|
||||
get => _progress2MaxValue;
|
||||
set => this.RaiseAndSetIfChanged(ref _progress2MaxValue, value);
|
||||
}
|
||||
|
||||
public string ProgressText
|
||||
{
|
||||
get => _progressText;
|
||||
set => this.RaiseAndSetIfChanged(ref _progressText, value);
|
||||
}
|
||||
|
||||
public double ProgressValue
|
||||
{
|
||||
get => _progressValue;
|
||||
set => this.RaiseAndSetIfChanged(ref _progressValue, value);
|
||||
}
|
||||
|
||||
public double Progress2Value
|
||||
{
|
||||
get => _progress2Value;
|
||||
set => this.RaiseAndSetIfChanged(ref _progress2Value, value);
|
||||
}
|
||||
|
||||
public string Progress2Text
|
||||
{
|
||||
get => _progress2Text;
|
||||
set => this.RaiseAndSetIfChanged(ref _progress2Text, value);
|
||||
}
|
||||
|
||||
public bool SectorsErrorsAllVisible
|
||||
{
|
||||
get => _sectorsErrorsAllVisible;
|
||||
set => this.RaiseAndSetIfChanged(ref _sectorsErrorsAllVisible, value);
|
||||
}
|
||||
|
||||
public string SectorsErrorsAllText
|
||||
{
|
||||
get => _sectorsErrorsAllText;
|
||||
set => this.RaiseAndSetIfChanged(ref _sectorsErrorsAllText, value);
|
||||
}
|
||||
|
||||
public bool SectorsUnknownAllVisible
|
||||
{
|
||||
get => _sectorsUnknownAllVisible;
|
||||
set => this.RaiseAndSetIfChanged(ref _sectorsUnknownAllVisible, value);
|
||||
}
|
||||
|
||||
public string SectorsUnknownAllText
|
||||
{
|
||||
get => _sectorsUnknownAllText;
|
||||
set => this.RaiseAndSetIfChanged(ref _sectorsUnknownAllText, value);
|
||||
}
|
||||
|
||||
public string SectorErrorsText
|
||||
{
|
||||
get => _sectorErrorsText;
|
||||
set => this.RaiseAndSetIfChanged(ref _sectorErrorsText, value);
|
||||
}
|
||||
|
||||
public bool SectorErrorsVisible
|
||||
{
|
||||
get => _sectorErrorsVisible;
|
||||
set => this.RaiseAndSetIfChanged(ref _sectorErrorsVisible, value);
|
||||
}
|
||||
|
||||
public bool SectorsUnknownsVisible
|
||||
{
|
||||
get => _sectorsUnknownsVisible;
|
||||
set => this.RaiseAndSetIfChanged(ref _sectorsUnknownsVisible, value);
|
||||
}
|
||||
|
||||
public string SectorsUnknownsText
|
||||
{
|
||||
get => _sectorsUnknownsText;
|
||||
set => this.RaiseAndSetIfChanged(ref _sectorsUnknownsText, value);
|
||||
}
|
||||
|
||||
public bool SectorSummaryVisible
|
||||
{
|
||||
get => _sectorSummaryVisible;
|
||||
set => this.RaiseAndSetIfChanged(ref _sectorSummaryVisible, value);
|
||||
}
|
||||
|
||||
public string TotalSectorsText
|
||||
{
|
||||
get => _totalSectorsText;
|
||||
set => this.RaiseAndSetIfChanged(ref _totalSectorsText, value);
|
||||
}
|
||||
|
||||
public string TotalSectorErrorsText
|
||||
{
|
||||
get => _totalSectorErrorsText;
|
||||
set => this.RaiseAndSetIfChanged(ref _totalSectorErrorsText, value);
|
||||
}
|
||||
|
||||
public string TotalSectorUnknownsText
|
||||
{
|
||||
get => _totalSectorUnknownsText;
|
||||
set => this.RaiseAndSetIfChanged(ref _totalSectorUnknownsText, value);
|
||||
}
|
||||
|
||||
public string TotalSectorErrorsUnknownsText
|
||||
{
|
||||
get => _totalSectorErrorsUnknownsText;
|
||||
set => this.RaiseAndSetIfChanged(ref _totalSectorErrorsUnknownsText, value);
|
||||
}
|
||||
|
||||
public bool OptionsVisible
|
||||
{
|
||||
get => _optionsVisible;
|
||||
set => this.RaiseAndSetIfChanged(ref _optionsVisible, value);
|
||||
}
|
||||
|
||||
public bool ResultsVisible
|
||||
{
|
||||
get => _resultsVisible;
|
||||
set => this.RaiseAndSetIfChanged(ref _resultsVisible, value);
|
||||
}
|
||||
|
||||
public bool ProgressVisible
|
||||
{
|
||||
get => _progressVisible;
|
||||
set => this.RaiseAndSetIfChanged(ref _progressVisible, value);
|
||||
}
|
||||
|
||||
public bool StartVisible
|
||||
{
|
||||
get => _startVisible;
|
||||
set => this.RaiseAndSetIfChanged(ref _startVisible, value);
|
||||
}
|
||||
|
||||
public bool StopVisible
|
||||
{
|
||||
get => _stopVisible;
|
||||
set => this.RaiseAndSetIfChanged(ref _stopVisible, value);
|
||||
}
|
||||
|
||||
public bool CloseVisible
|
||||
{
|
||||
get => _closeVisible;
|
||||
set => this.RaiseAndSetIfChanged(ref _closeVisible, value);
|
||||
}
|
||||
|
||||
public bool StopEnabled
|
||||
{
|
||||
get => _stopEnabled;
|
||||
set => this.RaiseAndSetIfChanged(ref _stopEnabled, value);
|
||||
}
|
||||
|
||||
void ExecuteStartCommand()
|
||||
{
|
||||
VerifyImageEnabled = false;
|
||||
VerifySectorsEnabled = false;
|
||||
CloseVisible = false;
|
||||
StartVisible = false;
|
||||
StopVisible = true;
|
||||
ProgressVisible = true;
|
||||
Progress2Visible = false;
|
||||
|
||||
VerifySectorsVisible = _inputFormat is IOpticalMediaImage ||
|
||||
_inputFormat is IVerifiableSectorsImage;
|
||||
|
||||
// TODO: Do not offer the option to use this form if the image does not support any kind of verification
|
||||
new Thread(DoWork).Start();
|
||||
}
|
||||
|
||||
async void DoWork()
|
||||
{
|
||||
bool formatHasTracks;
|
||||
var inputOptical = _inputFormat as IOpticalMediaImage;
|
||||
var verifiableSectorsImage = _inputFormat as IVerifiableSectorsImage;
|
||||
|
||||
try
|
||||
{
|
||||
formatHasTracks = inputOptical?.Tracks?.Count > 0;
|
||||
}
|
||||
catch
|
||||
{
|
||||
formatHasTracks = false;
|
||||
}
|
||||
|
||||
// Setup progress bars
|
||||
await Dispatcher.UIThread.InvokeAsync(() =>
|
||||
{
|
||||
ProgressVisible = true;
|
||||
ProgressMaxValue = 0;
|
||||
|
||||
if(VerifyImageChecked || VerifySectorsChecked)
|
||||
ProgressMaxValue = 1;
|
||||
|
||||
if(formatHasTracks && inputOptical != null)
|
||||
ProgressMaxValue += inputOptical.Tracks.Count;
|
||||
else
|
||||
{
|
||||
if(VerifySectorsChecked)
|
||||
{
|
||||
ProgressMaxValue = 2;
|
||||
Progress2Visible = false;
|
||||
Progress2Visible = false;
|
||||
}
|
||||
else
|
||||
{
|
||||
Progress2Visible = true;
|
||||
Progress2Visible = true;
|
||||
}
|
||||
}
|
||||
|
||||
ProgressMaxValue++;
|
||||
});
|
||||
|
||||
if(VerifyImageChecked)
|
||||
{
|
||||
if(!(_inputFormat is IVerifiableImage verifiableImage))
|
||||
await Dispatcher.UIThread.InvokeAsync(() =>
|
||||
{
|
||||
ImageResultVisible = true;
|
||||
ImageResultText = "Disc image does not support verification.";
|
||||
});
|
||||
else
|
||||
{
|
||||
await Dispatcher.UIThread.InvokeAsync(() =>
|
||||
{
|
||||
ProgressText = "Checking media image...";
|
||||
|
||||
if(VerifySectorsChecked)
|
||||
ProgressValue = 1;
|
||||
else
|
||||
ProgressIndeterminate = true;
|
||||
|
||||
Progress2Indeterminate = true;
|
||||
});
|
||||
|
||||
DateTime startCheck = DateTime.UtcNow;
|
||||
bool? discCheckStatus = verifiableImage.VerifyMediaImage();
|
||||
DateTime endCheck = DateTime.UtcNow;
|
||||
|
||||
TimeSpan checkTime = endCheck - startCheck;
|
||||
|
||||
await Dispatcher.UIThread.InvokeAsync(() =>
|
||||
{
|
||||
ImageResultVisible = true;
|
||||
|
||||
switch(discCheckStatus)
|
||||
{
|
||||
case true:
|
||||
ImageResultText = "Disc image checksums are correct";
|
||||
|
||||
break;
|
||||
case false:
|
||||
ImageResultText = "Disc image checksums are incorrect";
|
||||
|
||||
break;
|
||||
case null:
|
||||
ImageResultText = "Disc image does not contain checksums";
|
||||
|
||||
break;
|
||||
}
|
||||
});
|
||||
|
||||
AaruConsole.VerboseWriteLine("Checking disc image checksums took {0} seconds",
|
||||
checkTime.TotalSeconds);
|
||||
}
|
||||
}
|
||||
|
||||
if(VerifySectorsChecked)
|
||||
{
|
||||
DateTime startCheck = DateTime.Now;
|
||||
DateTime endCheck = startCheck;
|
||||
List<ulong> failingLbas = new List<ulong>();
|
||||
List<ulong> unknownLbas = new List<ulong>();
|
||||
|
||||
await Dispatcher.UIThread.InvokeAsync(() =>
|
||||
{
|
||||
Progress2Visible = true;
|
||||
Progress2Indeterminate = false;
|
||||
Progress2MaxValue = _inputFormat.Info.Sectors / 512d;
|
||||
StopEnabled = true;
|
||||
});
|
||||
|
||||
if(formatHasTracks)
|
||||
{
|
||||
ulong currentSectorAll = 0;
|
||||
|
||||
startCheck = DateTime.UtcNow;
|
||||
|
||||
foreach(Track currentTrack in inputOptical.Tracks)
|
||||
{
|
||||
await Dispatcher.UIThread.InvokeAsync(() =>
|
||||
{
|
||||
ProgressText =
|
||||
$"Verifying track {currentTrack.TrackSequence} of {inputOptical.Tracks.Count}";
|
||||
|
||||
ProgressValue++;
|
||||
});
|
||||
|
||||
ulong remainingSectors = currentTrack.TrackEndSector - currentTrack.TrackStartSector;
|
||||
ulong currentSector = 0;
|
||||
|
||||
while(remainingSectors > 0)
|
||||
{
|
||||
if(_cancel)
|
||||
{
|
||||
await Dispatcher.UIThread.InvokeAsync(() =>
|
||||
{
|
||||
CloseVisible = true;
|
||||
StartVisible = false;
|
||||
StopVisible = false;
|
||||
});
|
||||
|
||||
return;
|
||||
}
|
||||
|
||||
ulong all = currentSectorAll;
|
||||
|
||||
await Dispatcher.UIThread.InvokeAsync(() =>
|
||||
{
|
||||
Progress2Value = all / 512d;
|
||||
|
||||
Progress2Text =
|
||||
$"Checking sector {all} of {_inputFormat.Info.Sectors}, on track {currentTrack.TrackSequence}";
|
||||
});
|
||||
|
||||
List<ulong> tempFailingLbas;
|
||||
List<ulong> tempUnknownLbas;
|
||||
|
||||
if(remainingSectors < 512)
|
||||
inputOptical.VerifySectors(currentSector, (uint)remainingSectors,
|
||||
currentTrack.TrackSequence, out tempFailingLbas,
|
||||
out tempUnknownLbas);
|
||||
else
|
||||
inputOptical.VerifySectors(currentSector, 512, currentTrack.TrackSequence,
|
||||
out tempFailingLbas, out tempUnknownLbas);
|
||||
|
||||
failingLbas.AddRange(tempFailingLbas);
|
||||
|
||||
unknownLbas.AddRange(tempUnknownLbas);
|
||||
|
||||
if(remainingSectors < 512)
|
||||
{
|
||||
currentSector += remainingSectors;
|
||||
currentSectorAll += remainingSectors;
|
||||
remainingSectors = 0;
|
||||
}
|
||||
else
|
||||
{
|
||||
currentSector += 512;
|
||||
currentSectorAll += 512;
|
||||
remainingSectors -= 512;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
endCheck = DateTime.UtcNow;
|
||||
}
|
||||
else if(!(verifiableSectorsImage is null))
|
||||
{
|
||||
ulong remainingSectors = _inputFormat.Info.Sectors;
|
||||
ulong currentSector = 0;
|
||||
|
||||
startCheck = DateTime.UtcNow;
|
||||
|
||||
while(remainingSectors > 0)
|
||||
{
|
||||
if(_cancel)
|
||||
{
|
||||
await Dispatcher.UIThread.InvokeAsync(() =>
|
||||
{
|
||||
CloseVisible = true;
|
||||
StartVisible = false;
|
||||
StopVisible = false;
|
||||
});
|
||||
|
||||
return;
|
||||
}
|
||||
|
||||
ulong sector = currentSector;
|
||||
|
||||
await Dispatcher.UIThread.InvokeAsync(() =>
|
||||
{
|
||||
Progress2Value = (int)(sector / 512);
|
||||
Progress2Text = $"Checking sector {sector} of {_inputFormat.Info.Sectors}";
|
||||
});
|
||||
|
||||
List<ulong> tempFailingLbas;
|
||||
List<ulong> tempUnknownLbas;
|
||||
|
||||
if(remainingSectors < 512)
|
||||
verifiableSectorsImage.VerifySectors(currentSector, (uint)remainingSectors,
|
||||
out tempFailingLbas, out tempUnknownLbas);
|
||||
else
|
||||
verifiableSectorsImage.VerifySectors(currentSector, 512, out tempFailingLbas,
|
||||
out tempUnknownLbas);
|
||||
|
||||
failingLbas.AddRange(tempFailingLbas);
|
||||
|
||||
unknownLbas.AddRange(tempUnknownLbas);
|
||||
|
||||
if(remainingSectors < 512)
|
||||
{
|
||||
currentSector += remainingSectors;
|
||||
remainingSectors = 0;
|
||||
}
|
||||
else
|
||||
{
|
||||
currentSector += 512;
|
||||
remainingSectors -= 512;
|
||||
}
|
||||
}
|
||||
|
||||
endCheck = DateTime.UtcNow;
|
||||
}
|
||||
|
||||
TimeSpan checkTime = endCheck - startCheck;
|
||||
AaruConsole.VerboseWriteLine("Checking sector checksums took {0} seconds", checkTime.TotalSeconds);
|
||||
|
||||
await Dispatcher.UIThread.InvokeAsync(() =>
|
||||
{
|
||||
if(failingLbas.Count > 0)
|
||||
{
|
||||
if(failingLbas.Count == (int)_inputFormat.Info.Sectors)
|
||||
{
|
||||
SectorsErrorsAllVisible = true;
|
||||
SectorsErrorsAllText = "All sectors contain errors";
|
||||
}
|
||||
else
|
||||
{
|
||||
SectorErrorsText = "LBAs with error:";
|
||||
SectorErrorsVisible = true;
|
||||
|
||||
foreach(ulong t in failingLbas)
|
||||
ErrorList.Add(new LbaModel
|
||||
{
|
||||
Lba = t.ToString()
|
||||
});
|
||||
}
|
||||
}
|
||||
|
||||
if(unknownLbas.Count > 0)
|
||||
{
|
||||
if(unknownLbas.Count == (int)_inputFormat.Info.Sectors)
|
||||
{
|
||||
SectorsUnknownAllVisible = true;
|
||||
SectorsUnknownAllText = "All sectors are unknown";
|
||||
}
|
||||
else
|
||||
{
|
||||
SectorsUnknownsText = "Unknown LBAs:";
|
||||
SectorsUnknownsVisible = true;
|
||||
|
||||
foreach(ulong t in unknownLbas)
|
||||
UnknownList.Add(new LbaModel
|
||||
{
|
||||
Lba = t.ToString()
|
||||
});
|
||||
}
|
||||
}
|
||||
|
||||
SectorSummaryVisible = true;
|
||||
TotalSectorsText = $"Total sectors........... {_inputFormat.Info.Sectors}";
|
||||
TotalSectorErrorsText = $"Total errors............ {failingLbas.Count}";
|
||||
TotalSectorUnknownsText = $"Total unknowns.......... {unknownLbas.Count}";
|
||||
|
||||
TotalSectorErrorsUnknownsText = $"Total errors+unknowns... {failingLbas.Count + unknownLbas.Count}";
|
||||
});
|
||||
}
|
||||
|
||||
Statistics.AddCommand("verify");
|
||||
|
||||
await Dispatcher.UIThread.InvokeAsync(() =>
|
||||
{
|
||||
OptionsVisible = false;
|
||||
ResultsVisible = true;
|
||||
ProgressVisible = false;
|
||||
StartVisible = false;
|
||||
StopVisible = false;
|
||||
CloseVisible = true;
|
||||
});
|
||||
}
|
||||
|
||||
void ExecuteCloseCommand() => _view.Close();
|
||||
|
||||
protected internal void ExecuteStopCommand()
|
||||
{
|
||||
_cancel = true;
|
||||
StopEnabled = false;
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user