forked from amkovkov/GranuSightSoftware2
feat: SeparatorCalibration View and ViewModel
This commit is contained in:
@@ -1,11 +1,446 @@
|
||||
using System.Collections.ObjectModel;
|
||||
|
||||
using Microsoft.EntityFrameworkCore;
|
||||
using Microsoft.Extensions.Logging;
|
||||
|
||||
using Avalonia.Threading;
|
||||
|
||||
using CommunityToolkit.Mvvm.ComponentModel;
|
||||
using CommunityToolkit.Mvvm.Input;
|
||||
|
||||
using GSS2.Core.Analysis.AmineContent;
|
||||
using GSS2.Core.Analysis.AmineContent.Database.Calibration;
|
||||
using GSS2.Core.Hardware;
|
||||
using GSS2.UI.Core.ViewModels;
|
||||
|
||||
namespace GSS2.ViewModels.AmineContent;
|
||||
|
||||
public partial class SeparatorCalibrationViewModel : ViewModelBase
|
||||
{
|
||||
public SeparatorCalibrationViewModel()
|
||||
{
|
||||
private readonly ILogger<SeparatorCalibrationViewModel> _logger;
|
||||
private readonly AmineContentCalibrationContext _context;
|
||||
private readonly AmineContentImageCapturerService _imageCapturer;
|
||||
private readonly ImageStorageService _imageStorage;
|
||||
|
||||
private CancellationTokenSource? _captureImagesTaskCts = null;
|
||||
|
||||
[ObservableProperty] public partial bool MenuOpened { get; set; } = false;
|
||||
[ObservableProperty] public partial ObservableCollection<SeparatorRecord> Records { get; set; }
|
||||
|
||||
[ObservableProperty] public partial SeparatorRecord? SelectedRecord { get; set; }
|
||||
[ObservableProperty] public partial SeparatorRecord EditingRecord { get; set; }
|
||||
|
||||
[ObservableProperty] public partial int MainSectionsCurrentIndex { get; set; } = 0;
|
||||
[ObservableProperty] public partial bool MainSectionCanScrollBackward { get; set; } = false;
|
||||
[ObservableProperty] public partial bool MainSectionCanScrollForward { get; set; } = true;
|
||||
[ObservableProperty] public partial bool MainSectionIsScrolling { get; set; } = true;
|
||||
|
||||
[ObservableProperty] public partial Task? CaptureImagesTask { get; set; } = null;
|
||||
|
||||
[ObservableProperty] public partial ObservableCollection<AsyncImageRecordViewModel> Images { get; set; } = new();
|
||||
[ObservableProperty] public partial bool ImagesIsCapturing { get; set; } = false;
|
||||
[ObservableProperty] public partial string ImagesCapturingProgress { get; set; } = "";
|
||||
[ObservableProperty] public partial int ImagesSectionCurrentIndex { get; set; } = 0;
|
||||
[ObservableProperty] public partial bool ImagesSectionCanScrollForward { get; set; } = false;
|
||||
[ObservableProperty] public partial bool ImagesSectionCanScrollBackward { get; set; } = false;
|
||||
[ObservableProperty] public partial bool ImagesSectionIsScrolling { get; set; } = true;
|
||||
|
||||
public SeparatorCalibrationViewModel(
|
||||
ILogger<SeparatorCalibrationViewModel> logger,
|
||||
AmineContentCalibrationContext context,
|
||||
AmineContentImageCapturerService imageCapturer,
|
||||
ImageStorageService imageStorage
|
||||
)
|
||||
{
|
||||
_logger = logger;
|
||||
_context = context;
|
||||
_imageCapturer = imageCapturer;
|
||||
_imageStorage = imageStorage;
|
||||
_logger.LogInformation("Инициализация");
|
||||
|
||||
// Получаем записи из базы данных
|
||||
Records = new ObservableCollection<SeparatorRecord>(_context.SeparatorRecords.Include(r => r.Images));
|
||||
|
||||
// Берём первую запись
|
||||
SelectedRecord = Records.FirstOrDefault();
|
||||
// Если записей нет, то создаём пустую
|
||||
// на самом деле это излишне, потому что при изменении SelectedRecord
|
||||
// вызовется OnSelectedRecordChanged который сделает то же самое, но сделано это для избавления от предупреждений
|
||||
if (SelectedRecord is null)
|
||||
EditingRecord = new()
|
||||
{
|
||||
Type = "",
|
||||
Batch = "",
|
||||
Images = new()
|
||||
};
|
||||
else
|
||||
EditingRecord = CopyRecord(SelectedRecord);
|
||||
|
||||
_logger.LogInformation("Инициализировано");
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Вызывается при изменении <see cref="SelectedRecord"/>.<br/>
|
||||
/// Копирует <see cref="SelectedRecord"/> в <see cref="EditingRecord"/> или издаёт пустую <see cref="EditingRecord"/>.
|
||||
/// </summary>
|
||||
partial void OnSelectedRecordChanged(SeparatorRecord? value)
|
||||
{
|
||||
// Если выбранная запись отсутствует (в базе данных нет записей или создана новая запись)
|
||||
// то создаём пустую запись
|
||||
// TODO: Подумать над тем, чтобы сразу удалять несохранённые изображения из хранилища
|
||||
if (SelectedRecord is null)
|
||||
{
|
||||
EditingRecord = new()
|
||||
{
|
||||
Type = "",
|
||||
Batch = "",
|
||||
Images = new()
|
||||
};
|
||||
}
|
||||
else
|
||||
{
|
||||
// Иначе копируем запись, отвязывая её от базы данных
|
||||
EditingRecord = CopyRecord(SelectedRecord);
|
||||
}
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Вызывается при изменении <see cref="EditingRecord"/>.<br/>
|
||||
/// Заполняет <see cref="Images"/> изображениями из <see cref="EditingRecord"/>.
|
||||
/// </summary>
|
||||
partial void OnEditingRecordChanged(SeparatorRecord value)
|
||||
{
|
||||
var images = value.Images
|
||||
// Для каждого изображения
|
||||
.Select(i =>
|
||||
{
|
||||
// Находим путь через ImageStorageService
|
||||
var path = _imageStorage.GetFullPath(i.ImagePath, "separator");
|
||||
// Если изображения не найдено, возвращаем null
|
||||
if (path is null)
|
||||
return null;
|
||||
// Создаём AsyncImageRecordViewModel
|
||||
return new AsyncImageRecordViewModel(path, i);
|
||||
})
|
||||
// Оставляем только не null значения
|
||||
.OfType<AsyncImageRecordViewModel>();
|
||||
|
||||
// Создаём новую коллекцию изображений
|
||||
// TODO: Подумать над тем, чтобы чистить старую коллекцию перед созданием новой
|
||||
Images.Clear();
|
||||
Images = new(images);
|
||||
}
|
||||
|
||||
// Просто команды управления меню
|
||||
[RelayCommand] private void OpenMenu() => MenuOpened = true;
|
||||
[RelayCommand] private void CloseMenu() => MenuOpened = false;
|
||||
[RelayCommand] private void ToggleMenu() => MenuOpened = !MenuOpened;
|
||||
|
||||
// Команда создания новой записи
|
||||
// вызывает цепочку действий по изменению EditingRecord и Images
|
||||
[RelayCommand]
|
||||
private void CreateRecord()
|
||||
{
|
||||
// Из-за того что, если создана новая запись, то SelectedRecord == null
|
||||
// при простом присвоении SelectedRecord = null событие SelectedRecordChanged не вызовется,
|
||||
// потому что событие срабатывает только при изменении значения, а значение было null и присвоилось ему null,
|
||||
// поэтому если SelectedRecord == null, то нужно явно вызвать метод OnSelectedRecordChanged
|
||||
if (SelectedRecord is not null)
|
||||
SelectedRecord = null;
|
||||
else
|
||||
OnSelectedRecordChanged(null);
|
||||
}
|
||||
|
||||
// Команда удаления записи, создаёт новую запись и удаляет старую из списка и базы данных
|
||||
[RelayCommand]
|
||||
private void DeleteRecord()
|
||||
{
|
||||
// Сохраняем текущую запись
|
||||
var deletedRecord = SelectedRecord;
|
||||
// Создаём новую запись
|
||||
CreateRecord();
|
||||
|
||||
// Если запись на была сохранена в базу данных, то ничего не делаем
|
||||
if (deletedRecord is null)
|
||||
return;
|
||||
|
||||
// Если запись была в без данных, то
|
||||
// она была и в Records, удаляем её из Records
|
||||
Records.Remove(deletedRecord);
|
||||
// Удаляем запись из базы данных и сохраняем изменения
|
||||
// TODO: Подумать над тем, что возможно стоит не удалять запись полностью, а добавить флаг удаления,
|
||||
// но тогда нужно будет сохранять и изображения этой записи, как в базе данных, так и на диске,
|
||||
// тогда удалённые записи будут заполнять диск
|
||||
// TODO: Подумать над тем, чтобы сразу удалять изображения из хранилища
|
||||
_context.SeparatorRecords.Remove(deletedRecord);
|
||||
_context.SaveChanges();
|
||||
}
|
||||
|
||||
// Команда сохранения записи, сохраняет новую запись или изменения в отредактированной
|
||||
[RelayCommand]
|
||||
private void SaveRecord()
|
||||
{
|
||||
// Если это новая запись
|
||||
if (SelectedRecord is null)
|
||||
{
|
||||
// Если это новая запись, то копируем её
|
||||
var newRecord = CopyRecord(EditingRecord);
|
||||
|
||||
// Для каждого изображения в записи
|
||||
for (int i = 0; i < newRecord.Images.Count(); i++)
|
||||
{
|
||||
var image = newRecord.Images[i];
|
||||
// Если изображение есть в базе данных, то пропускаем его
|
||||
if (_context.Entry(image).State is not EntityState.Detached)
|
||||
continue;
|
||||
|
||||
// Если изображения нет в базе данных, то добавляем его в базу
|
||||
var attachedImage = _context.ImageRecords.Add(image);
|
||||
// Удаляем отвязанное от базы данных изображение из записи
|
||||
newRecord.Images.RemoveAt(i);
|
||||
// Добавляем привязанное изображение к записи
|
||||
newRecord.Images.Insert(i, attachedImage.Entity);
|
||||
}
|
||||
// Если записи нет в базе данных
|
||||
if (_context.Entry(newRecord).State is EntityState.Detached)
|
||||
{
|
||||
// То добавляем её в базу
|
||||
var attachedRecord = _context.SeparatorRecords.Add(newRecord);
|
||||
// Заменяем текущую запись на привязанную к базе данных
|
||||
newRecord = attachedRecord.Entity;
|
||||
}
|
||||
|
||||
// Сохраняем изменения в базе данных
|
||||
_context.SaveChanges();
|
||||
// Добавляем запись в список
|
||||
Records.Add(newRecord);
|
||||
// Выбираем запись
|
||||
SelectedRecord = newRecord;
|
||||
}
|
||||
// Если эта запись есть в базе
|
||||
else
|
||||
{
|
||||
// Переносим данные из редактируемой записи в выбранную
|
||||
SelectedRecord.Type = EditingRecord.Type;
|
||||
SelectedRecord.Batch = EditingRecord.Batch;
|
||||
|
||||
// Находим изображения которые нужно удалить -
|
||||
// те которые есть в выбранной записи, но нет в редактируемой
|
||||
var imagesToRemove = SelectedRecord.Images
|
||||
.Where(e => EditingRecord.Images.All(i => i.Id != e.Id))
|
||||
.ToList();
|
||||
|
||||
// Удаляем старые изображения
|
||||
// TODO: Подумать над тем, чтобы сразу удалять изображения из хранилища
|
||||
foreach (var image in imagesToRemove)
|
||||
SelectedRecord.Images.Remove(image);
|
||||
|
||||
// Для каждого изображения в редактируемой записи
|
||||
foreach (var editedImage in EditingRecord.Images)
|
||||
{
|
||||
// Находим изображение в выбранной записи (оно будет привязано к базе данных)
|
||||
var existing = SelectedRecord.Images
|
||||
.FirstOrDefault(i => i.Id == editedImage.Id);
|
||||
|
||||
// Если изображение не найдено
|
||||
if (existing is null)
|
||||
{
|
||||
// Копируем изображение в выбранную запись
|
||||
SelectedRecord.Images.Add(CopyRecord(editedImage));
|
||||
}
|
||||
// Если изображение найдено
|
||||
else
|
||||
{
|
||||
// Изменяем данные изображения на данные изображения из редактируемой записи
|
||||
existing.ImagePath = editedImage.ImagePath;
|
||||
existing.VisibleIntensity = editedImage.VisibleIntensity;
|
||||
existing.Uv365Intensity = editedImage.Uv365Intensity;
|
||||
existing.Uv254Intensity = editedImage.Uv254Intensity;
|
||||
}
|
||||
}
|
||||
|
||||
// Для каждого изображения в выбранной записи
|
||||
for (int i = 0; i < SelectedRecord.Images.Count(); i++)
|
||||
{
|
||||
var image = SelectedRecord.Images[i];
|
||||
// Если изображение есть в базе данных, то пропускаем его
|
||||
if (_context.Entry(image).State is not EntityState.Detached)
|
||||
continue;
|
||||
|
||||
// Если изображения нет в базе данных, то добавляем его в базу
|
||||
var attachedImage = _context.ImageRecords.Add(image);
|
||||
// Удаляем отвязанное от базы данных изображение из записи
|
||||
SelectedRecord.Images.RemoveAt(i);
|
||||
// Добавляем привязанное изображение к записи
|
||||
SelectedRecord.Images.Insert(i, attachedImage.Entity);
|
||||
}
|
||||
// Если записи нет в базе данных
|
||||
if (_context.Entry(SelectedRecord).State is EntityState.Detached)
|
||||
{
|
||||
// То добавляем её в базу
|
||||
var attachedRecord = _context.SeparatorRecords.Add(SelectedRecord);
|
||||
// Находим текущую запись в списке
|
||||
var index = Records.IndexOf(SelectedRecord);
|
||||
// Заменяем её в списке на привязанную к базе данных
|
||||
Records[index] = attachedRecord.Entity;
|
||||
// Заменяем выбранную запись на привязанную
|
||||
SelectedRecord = attachedRecord.Entity;
|
||||
}
|
||||
|
||||
// Сохраняем изменения в базе данных
|
||||
_context.SaveChanges();
|
||||
}
|
||||
}
|
||||
|
||||
// Команда изменения основной секции (информация <-> изображения)
|
||||
[RelayCommand] private void SwitchMainSection() => MainSectionsCurrentIndex = MainSectionsCurrentIndex == 0 ? 1 : 0;
|
||||
|
||||
// Команда навигации по изображениям
|
||||
[RelayCommand] private void ImagesSectionPrevious() => ImagesSectionCurrentIndex--;
|
||||
[RelayCommand] private void ImagesSectionNext() => ImagesSectionCurrentIndex++;
|
||||
|
||||
// Команда съёмки изображений
|
||||
[RelayCommand]
|
||||
private void CaptureImage()
|
||||
{
|
||||
// Если съёмка уже начата, то ничего не делаем
|
||||
if (CaptureImagesTask is not null)
|
||||
return;
|
||||
|
||||
// Создаём токен отмены задачи
|
||||
_captureImagesTaskCts = new CancellationTokenSource();
|
||||
var cancellationToken = _captureImagesTaskCts.Token;
|
||||
|
||||
// Запускаем получение изображений асинхронно
|
||||
CaptureImagesTask = Task.Run(async () =>
|
||||
{
|
||||
// Список полученных изображений
|
||||
var imageRecords = new List<ImageRecord>();
|
||||
|
||||
// Обновляем данные в интерфейсе
|
||||
// Dispatcher.UIThread.Invoke нужен потому-что этот метод может (и скорее всего будет) вызван в другом потоке,а
|
||||
// а изменения данных в интерфейсе могут быть вызваны только из потока в котором интерфейс был создан
|
||||
_ = Dispatcher.UIThread.InvokeAsync(() => ImagesIsCapturing = true);
|
||||
|
||||
// Получаем изображения
|
||||
var task = _imageCapturer.CaptureImages(async (data) =>
|
||||
{
|
||||
// Распаковываем данные
|
||||
var (i, visible, uv365, uv254, image) = data;
|
||||
// Выбрасываем исключение если задача отменена
|
||||
cancellationToken.ThrowIfCancellationRequested();
|
||||
|
||||
// Обновляем данные в интерфейсе
|
||||
// Dispatcher.UIThread.Invoke нужен потому-что этот метод может (и скорее всего будет) вызван в другом потоке,а
|
||||
// а изменения данных в интерфейсе могут быть вызваны только из потока в котором интерфейс был создан
|
||||
_ = Dispatcher.UIThread.InvokeAsync(() => ImagesCapturingProgress = $"Получение изображений: {i + 1} / {AmineContentImageCapturerService.IlluminatorIntensities.Count()}");
|
||||
|
||||
_ = Task.Run(async () =>
|
||||
{
|
||||
// Кодируем изображение в png
|
||||
var bytes = image.ImEncode(".png");
|
||||
// Сохраняем изображение в ImageStorageService
|
||||
string fileName;
|
||||
using (var stream = new MemoryStream(bytes))
|
||||
fileName = await _imageStorage.SaveAsync(stream, ".png", "separator", cancellationToken);
|
||||
|
||||
// Освобождаем память изображения
|
||||
image.Release();
|
||||
image = null;
|
||||
bytes = null;
|
||||
|
||||
// Добавляем изображение в список полученных
|
||||
imageRecords.Add(
|
||||
new ImageRecord()
|
||||
{
|
||||
ImagePath = fileName,
|
||||
VisibleIntensity = visible,
|
||||
Uv365Intensity = uv365,
|
||||
Uv254Intensity = uv254,
|
||||
}
|
||||
);
|
||||
});
|
||||
}, cancellationToken);
|
||||
|
||||
_ = task.ContinueWith(t =>
|
||||
{
|
||||
if (t.Exception is not null)
|
||||
{
|
||||
_logger.LogError(t.Exception, "Исключение во время получения изображений");
|
||||
foreach (var ex in t.Exception.InnerExceptions ?? [])
|
||||
_logger.LogError(ex, "Исключение во время получения изображений");
|
||||
}
|
||||
var images = imageRecords
|
||||
// Для каждого изображения
|
||||
.Select(i =>
|
||||
{
|
||||
// Находим путь через ImageStorageService
|
||||
var path = _imageStorage.GetFullPath(i.ImagePath, "separator");
|
||||
// Если изображения не найдено, возвращаем null
|
||||
if (path is null)
|
||||
return null;
|
||||
// Создаём AsyncImageRecordViewModel
|
||||
return new AsyncImageRecordViewModel(path, i);
|
||||
})
|
||||
// Оставляем только не null значения
|
||||
.OfType<AsyncImageRecordViewModel>();
|
||||
|
||||
// Обновляем данные в редактируемой коллекции
|
||||
// Так же Dispatcher.UIThread.Invoke нужен потому-что этот метод может (и скорее всего будет) вызван в другом потоке,а
|
||||
// а изменения данных в интерфейсе могут быть вызваны только из потока в котором интерфейс был создан
|
||||
_ = Dispatcher.UIThread.InvokeAsync(() =>
|
||||
{
|
||||
EditingRecord.Images = imageRecords;
|
||||
// Создаём новую коллекцию изображений
|
||||
// TODO: Подумать над тем, чтобы чистить старую коллекцию перед созданием новой
|
||||
Images = new(images);
|
||||
});
|
||||
|
||||
// Сбрасываем состояние задачи съёмки изображений
|
||||
// Так же Dispatcher.UIThread.Invoke нужен потому-что этот метод может (и скорее всего будет) вызван в другом потоке,а
|
||||
// а изменения данных в интерфейсе могут быть вызваны только из потока в котором интерфейс был создан
|
||||
_ = Dispatcher.UIThread.InvokeAsync(() =>
|
||||
{
|
||||
ImagesIsCapturing = false;
|
||||
ImagesCapturingProgress = "";
|
||||
CaptureImagesTask = null;
|
||||
});
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
// Команда отмены задачи съёмки изображений
|
||||
[RelayCommand] private void CancelImageCapturing() => _captureImagesTaskCts?.Cancel();
|
||||
|
||||
/// <summary>
|
||||
/// Копирует <see cref="SeparatorRecord"/>, отвязывая её от базы данных
|
||||
/// </summary>
|
||||
/// <returns>Скопированная <see cref="SeparatorRecord"/></returns>
|
||||
private static SeparatorRecord CopyRecord(SeparatorRecord source)
|
||||
{
|
||||
return new SeparatorRecord
|
||||
{
|
||||
Id = source.Id,
|
||||
Type = source.Type,
|
||||
Batch = source.Batch,
|
||||
Images = source.Images
|
||||
.Select(CopyRecord)
|
||||
.ToList()
|
||||
};
|
||||
}
|
||||
/// <summary>
|
||||
/// Копирует <see cref="ImageRecord"/>, отвязывая её от базы данных
|
||||
/// </summary>
|
||||
/// <returns>Скопированная <see cref="ImageRecord"/></returns>
|
||||
private static ImageRecord CopyRecord(ImageRecord source)
|
||||
{
|
||||
return new ImageRecord
|
||||
{
|
||||
Id = source.Id,
|
||||
ImagePath = source.ImagePath,
|
||||
VisibleIntensity = source.VisibleIntensity,
|
||||
Uv365Intensity = source.Uv365Intensity,
|
||||
Uv254Intensity = source.Uv254Intensity
|
||||
};
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user