mirror of
https://github.com/aaru-dps/Aaru.git
synced 2025-12-16 19:24:25 +00:00
Migrate image entropy window from Eto.Forms to Avalonia.
This commit is contained in:
385
Aaru.Gui/ViewModels/ImageEntropyViewModel.cs
Normal file
385
Aaru.Gui/ViewModels/ImageEntropyViewModel.cs
Normal file
@@ -0,0 +1,385 @@
|
||||
using System;
|
||||
using System.Collections.ObjectModel;
|
||||
using System.Globalization;
|
||||
using System.Reactive;
|
||||
using System.Threading;
|
||||
using Aaru.CommonTypes.Interfaces;
|
||||
using Aaru.Console;
|
||||
using Aaru.Core;
|
||||
using Aaru.Gui.Models;
|
||||
using Avalonia.Controls;
|
||||
using Avalonia.Threading;
|
||||
using ReactiveUI;
|
||||
|
||||
namespace Aaru.Gui.ViewModels
|
||||
{
|
||||
public class ImageEntropyViewModel : ViewModelBase
|
||||
{
|
||||
readonly IMediaImage _inputFormat;
|
||||
bool _closeVisible;
|
||||
bool _duplicatedSectorsChecked;
|
||||
bool _duplicatedSectorsEnabled;
|
||||
EntropyResults _entropy;
|
||||
string _mediaEntropyText;
|
||||
bool _mediaEntropyVisible;
|
||||
string _mediaUniqueSectorsText;
|
||||
bool _mediaUniqueSectorsVisible;
|
||||
bool _optionsVisible;
|
||||
bool _progress1Visible;
|
||||
bool _progress2Indeterminate;
|
||||
double _progress2Max;
|
||||
string _progress2Text;
|
||||
double _progress2Value;
|
||||
bool _progress2Visible;
|
||||
bool _progressIndeterminate;
|
||||
double _progressMax;
|
||||
string _progressText;
|
||||
double _progressValue;
|
||||
bool _progressVisible;
|
||||
bool _resultsVisible;
|
||||
bool _separatedTracksChecked;
|
||||
bool _separatedTracksEnabled;
|
||||
bool _separatedTracksVisible;
|
||||
bool _startVisible;
|
||||
bool _stopVisible;
|
||||
EntropyResults[] _tracksEntropy;
|
||||
readonly Window _view;
|
||||
bool _wholeDiscChecked;
|
||||
bool _wholeDiscEnabled;
|
||||
bool _wholeDiscVisible;
|
||||
|
||||
public ImageEntropyViewModel(IMediaImage inputFormat, Window view)
|
||||
{
|
||||
_inputFormat = inputFormat;
|
||||
_view = view;
|
||||
TrackEntropy = new ObservableCollection<TrackEntropyModel>();
|
||||
StartCommand = ReactiveCommand.Create(ExecuteStartCommand);
|
||||
CloseCommand = ReactiveCommand.Create(ExecuteCloseCommand);
|
||||
StopCommand = ReactiveCommand.Create(ExecuteStopCommand);
|
||||
OptionsVisible = true;
|
||||
DuplicatedSectorsChecked = true;
|
||||
SeparatedTracksChecked = true;
|
||||
WholeDiscChecked = true;
|
||||
StartVisible = true;
|
||||
|
||||
var inputOptical = inputFormat as IOpticalMediaImage;
|
||||
|
||||
if(inputOptical?.Tracks.Count > 0)
|
||||
{
|
||||
SeparatedTracksVisible = true;
|
||||
WholeDiscVisible = true;
|
||||
}
|
||||
else
|
||||
{
|
||||
SeparatedTracksChecked = false;
|
||||
WholeDiscChecked = true;
|
||||
}
|
||||
}
|
||||
|
||||
public bool SeparatedTracksVisible
|
||||
{
|
||||
get => _separatedTracksVisible;
|
||||
set => this.RaiseAndSetIfChanged(ref _separatedTracksVisible, value);
|
||||
}
|
||||
|
||||
public bool WholeDiscVisible
|
||||
{
|
||||
get => _wholeDiscVisible;
|
||||
set => this.RaiseAndSetIfChanged(ref _wholeDiscVisible, value);
|
||||
}
|
||||
|
||||
public bool SeparatedTracksChecked
|
||||
{
|
||||
get => _separatedTracksChecked;
|
||||
set => this.RaiseAndSetIfChanged(ref _separatedTracksChecked, value);
|
||||
}
|
||||
|
||||
public bool WholeDiscChecked
|
||||
{
|
||||
get => _wholeDiscChecked;
|
||||
set => this.RaiseAndSetIfChanged(ref _wholeDiscChecked, value);
|
||||
}
|
||||
|
||||
public bool DuplicatedSectorsEnabled
|
||||
{
|
||||
get => _duplicatedSectorsEnabled;
|
||||
set => this.RaiseAndSetIfChanged(ref _duplicatedSectorsEnabled, value);
|
||||
}
|
||||
|
||||
public bool SeparatedTracksEnabled
|
||||
{
|
||||
get => _separatedTracksEnabled;
|
||||
set => this.RaiseAndSetIfChanged(ref _separatedTracksEnabled, value);
|
||||
}
|
||||
|
||||
public bool WholeDiscEnabled
|
||||
{
|
||||
get => _wholeDiscEnabled;
|
||||
set => this.RaiseAndSetIfChanged(ref _wholeDiscEnabled, value);
|
||||
}
|
||||
|
||||
public bool CloseVisible
|
||||
{
|
||||
get => _closeVisible;
|
||||
set => this.RaiseAndSetIfChanged(ref _closeVisible, 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 ProgressVisible
|
||||
{
|
||||
get => _progressVisible;
|
||||
set => this.RaiseAndSetIfChanged(ref _progressVisible, value);
|
||||
}
|
||||
|
||||
public bool DuplicatedSectorsChecked
|
||||
{
|
||||
get => _duplicatedSectorsChecked;
|
||||
set => this.RaiseAndSetIfChanged(ref _duplicatedSectorsChecked, value);
|
||||
}
|
||||
|
||||
public bool OptionsVisible
|
||||
{
|
||||
get => _optionsVisible;
|
||||
set => this.RaiseAndSetIfChanged(ref _optionsVisible, value);
|
||||
}
|
||||
|
||||
public bool ResultsVisible
|
||||
{
|
||||
get => _resultsVisible;
|
||||
set => this.RaiseAndSetIfChanged(ref _resultsVisible, value);
|
||||
}
|
||||
|
||||
public string MediaEntropyText
|
||||
{
|
||||
get => _mediaEntropyText;
|
||||
set => this.RaiseAndSetIfChanged(ref _mediaEntropyText, value);
|
||||
}
|
||||
|
||||
public bool MediaEntropyVisible
|
||||
{
|
||||
get => _mediaEntropyVisible;
|
||||
set => this.RaiseAndSetIfChanged(ref _mediaEntropyVisible, value);
|
||||
}
|
||||
|
||||
public string MediaUniqueSectorsText
|
||||
{
|
||||
get => _mediaUniqueSectorsText;
|
||||
set => this.RaiseAndSetIfChanged(ref _mediaUniqueSectorsText, value);
|
||||
}
|
||||
|
||||
public bool MediaUniqueSectorsVisible
|
||||
{
|
||||
get => _mediaUniqueSectorsVisible;
|
||||
set => this.RaiseAndSetIfChanged(ref _mediaUniqueSectorsVisible, value);
|
||||
}
|
||||
|
||||
public bool Progress1Visible
|
||||
{
|
||||
get => _progress1Visible;
|
||||
set => this.RaiseAndSetIfChanged(ref _progress1Visible, value);
|
||||
}
|
||||
|
||||
public bool Progress2Visible
|
||||
{
|
||||
get => _progress2Visible;
|
||||
set => this.RaiseAndSetIfChanged(ref _progress2Visible, value);
|
||||
}
|
||||
|
||||
public string ProgressText
|
||||
{
|
||||
get => _progressText;
|
||||
set => this.RaiseAndSetIfChanged(ref _progressText, value);
|
||||
}
|
||||
|
||||
public bool ProgressIndeterminate
|
||||
{
|
||||
get => _progressIndeterminate;
|
||||
set => this.RaiseAndSetIfChanged(ref _progressIndeterminate, value);
|
||||
}
|
||||
|
||||
public double ProgressMax
|
||||
{
|
||||
get => _progressMax;
|
||||
set => this.RaiseAndSetIfChanged(ref _progressMax, value);
|
||||
}
|
||||
|
||||
public double ProgressValue
|
||||
{
|
||||
get => _progressValue;
|
||||
set => this.RaiseAndSetIfChanged(ref _progressValue, value);
|
||||
}
|
||||
|
||||
public string Progress2Text
|
||||
{
|
||||
get => _progress2Text;
|
||||
set => this.RaiseAndSetIfChanged(ref _progress2Text, value);
|
||||
}
|
||||
|
||||
public bool Progress2Indeterminate
|
||||
{
|
||||
get => _progress2Indeterminate;
|
||||
set => this.RaiseAndSetIfChanged(ref _progress2Indeterminate, value);
|
||||
}
|
||||
|
||||
public double Progress2Max
|
||||
{
|
||||
get => _progress2Max;
|
||||
set => this.RaiseAndSetIfChanged(ref _progress2Max, value);
|
||||
}
|
||||
|
||||
public double Progress2Value
|
||||
{
|
||||
get => _progress2Value;
|
||||
set => this.RaiseAndSetIfChanged(ref _progress2Value, value);
|
||||
}
|
||||
|
||||
public string Title => "Calculating entropy";
|
||||
public ObservableCollection<TrackEntropyModel> TrackEntropy { get; }
|
||||
public ReactiveCommand<Unit, Unit> StartCommand { get; }
|
||||
public ReactiveCommand<Unit, Unit> CloseCommand { get; }
|
||||
public ReactiveCommand<Unit, Unit> StopCommand { get; }
|
||||
|
||||
void ExecuteStartCommand()
|
||||
{
|
||||
var entropyCalculator = new Entropy(false, false, _inputFormat);
|
||||
entropyCalculator.InitProgressEvent += InitProgress;
|
||||
entropyCalculator.InitProgress2Event += InitProgress2;
|
||||
entropyCalculator.UpdateProgressEvent += UpdateProgress;
|
||||
entropyCalculator.UpdateProgress2Event += UpdateProgress2;
|
||||
entropyCalculator.EndProgressEvent += EndProgress;
|
||||
entropyCalculator.EndProgress2Event += EndProgress2;
|
||||
DuplicatedSectorsEnabled = false;
|
||||
SeparatedTracksEnabled = false;
|
||||
WholeDiscEnabled = false;
|
||||
CloseVisible = false;
|
||||
StartVisible = false;
|
||||
StopVisible = false;
|
||||
ProgressVisible = true;
|
||||
|
||||
var thread = new Thread(async () =>
|
||||
{
|
||||
if(SeparatedTracksChecked)
|
||||
{
|
||||
_tracksEntropy = entropyCalculator.CalculateTracksEntropy(DuplicatedSectorsChecked);
|
||||
|
||||
foreach(EntropyResults trackEntropy in _tracksEntropy)
|
||||
{
|
||||
AaruConsole.WriteLine("Entropy for track {0} is {1:F4}.", trackEntropy.Track,
|
||||
trackEntropy.Entropy);
|
||||
|
||||
if(trackEntropy.UniqueSectors != null)
|
||||
AaruConsole.WriteLine("Track {0} has {1} unique sectors ({2:P3})", trackEntropy.Track,
|
||||
trackEntropy.UniqueSectors,
|
||||
(double)trackEntropy.UniqueSectors / (double)trackEntropy.Sectors);
|
||||
}
|
||||
}
|
||||
|
||||
if(WholeDiscChecked != true)
|
||||
return;
|
||||
|
||||
_entropy = entropyCalculator.CalculateMediaEntropy(DuplicatedSectorsChecked);
|
||||
|
||||
await Dispatcher.UIThread.InvokeAsync(Finish);
|
||||
});
|
||||
|
||||
Statistics.AddCommand("entropy");
|
||||
|
||||
thread.Start();
|
||||
}
|
||||
|
||||
void Finish()
|
||||
{
|
||||
OptionsVisible = false;
|
||||
CloseVisible = true;
|
||||
ProgressVisible = false;
|
||||
ResultsVisible = true;
|
||||
|
||||
if(SeparatedTracksChecked)
|
||||
{
|
||||
foreach(EntropyResults trackEntropy in _tracksEntropy)
|
||||
TrackEntropy.Add(new TrackEntropyModel
|
||||
{
|
||||
Track = trackEntropy.Track.ToString(),
|
||||
Entropy = trackEntropy.Entropy.ToString(CultureInfo.CurrentUICulture),
|
||||
UniqueSectors =
|
||||
$"{trackEntropy.UniqueSectors} ({(double)(trackEntropy.UniqueSectors ?? 0) / (double)trackEntropy.Sectors:P3})"
|
||||
});
|
||||
}
|
||||
|
||||
if(WholeDiscChecked != true)
|
||||
return;
|
||||
|
||||
MediaEntropyText = $"Entropy for disk is {_entropy.Entropy:F4}.";
|
||||
MediaEntropyVisible = true;
|
||||
|
||||
if(_entropy.UniqueSectors == null)
|
||||
return;
|
||||
|
||||
MediaUniqueSectorsText =
|
||||
$"Disk has {_entropy.UniqueSectors} unique sectors ({(double)_entropy.UniqueSectors / (double)_entropy.Sectors:P3})";
|
||||
|
||||
MediaUniqueSectorsVisible = true;
|
||||
}
|
||||
|
||||
void ExecuteCloseCommand() => _view.Close();
|
||||
|
||||
void ExecuteStopCommand() => throw new NotImplementedException();
|
||||
|
||||
void InitProgress() => Progress1Visible = true;
|
||||
|
||||
void EndProgress() => Progress1Visible = false;
|
||||
|
||||
void InitProgress2() => Progress2Visible = true;
|
||||
|
||||
void EndProgress2() => Progress2Visible = false;
|
||||
|
||||
async void UpdateProgress(string text, long current, long maximum) =>
|
||||
await Dispatcher.UIThread.InvokeAsync(() =>
|
||||
{
|
||||
ProgressText = text;
|
||||
|
||||
if(maximum == 0)
|
||||
{
|
||||
ProgressIndeterminate = true;
|
||||
|
||||
return;
|
||||
}
|
||||
|
||||
if(ProgressIndeterminate)
|
||||
ProgressIndeterminate = false;
|
||||
|
||||
ProgressMax = maximum;
|
||||
ProgressValue = current;
|
||||
});
|
||||
|
||||
async void UpdateProgress2(string text, long current, long maximum) =>
|
||||
await Dispatcher.UIThread.InvokeAsync(() =>
|
||||
{
|
||||
Progress2Text = text;
|
||||
|
||||
if(maximum == 0)
|
||||
{
|
||||
Progress2Indeterminate = true;
|
||||
|
||||
return;
|
||||
}
|
||||
|
||||
if(Progress2Indeterminate)
|
||||
Progress2Indeterminate = false;
|
||||
|
||||
Progress2Max = maximum;
|
||||
Progress2Value = current;
|
||||
});
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user