// --------------------------------------------------------------------------------------------------------------------
//
// This file is part of the HandBrake source code - It may be used under the terms of the GNU General Public License.
//
//
// The Preview View Model
//
// --------------------------------------------------------------------------------------------------------------------
namespace HandBrakeWPF.ViewModels
{
using System;
using System.Collections.Generic;
using System.ComponentModel;
using System.Diagnostics;
using System.IO;
using System.Linq;
using System.Windows;
using Caliburn.Micro;
using HandBrakeWPF.EventArgs;
using HandBrakeWPF.Properties;
using HandBrakeWPF.Services.Interfaces;
using HandBrakeWPF.Services.Queue.Interfaces;
using HandBrakeWPF.Services.Queue.Model;
using HandBrakeWPF.Utilities;
using HandBrakeWPF.ViewModels.Interfaces;
using Microsoft.Win32;
using EncodeCompletedEventArgs = Services.Encode.EventArgs.EncodeCompletedEventArgs;
using EncodeProgressEventArgs = Services.Encode.EventArgs.EncodeProgressEventArgs;
///
/// The Preview View Model
///
public class QueueViewModel : ViewModelBase, IQueueViewModel
{
#region Constants and Fields
private readonly IErrorService errorService;
private readonly IUserSettingService userSettingService;
private readonly IQueueProcessor queueProcessor;
private string jobStatus;
private string jobsPending;
private string whenDoneAction;
private QueueTask selectedTask;
private bool isQueueRunning;
private double progressValue;
private bool intermediateProgress;
private bool showEncodeProgress;
#endregion
#region Constructors and Destructors
///
/// Initializes a new instance of the class.
///
///
/// The user Setting Service.
///
///
/// The Queue Processor Service
///
///
/// The Error Service
///
public QueueViewModel(IUserSettingService userSettingService, IQueueProcessor queueProcessor, IErrorService errorService)
{
this.userSettingService = userSettingService;
this.queueProcessor = queueProcessor;
this.errorService = errorService;
this.Title = Resources.QueueViewModel_Queue;
this.JobsPending = Resources.QueueViewModel_NoEncodesPending;
this.JobStatus = Resources.QueueViewModel_NoJobsPending;
this.SelectedItems = new BindingList();
this.DisplayName = "Queue";
this.IsQueueRunning = false;
this.WhenDoneAction = this.userSettingService.GetUserSetting(UserSettingConstants.WhenCompleteAction);
}
#endregion
#region Properties
///
/// Gets or sets a value indicating whether the Queue is paused or not..
///
public bool IsQueueRunning
{
get
{
return this.isQueueRunning;
}
set
{
if (value == this.isQueueRunning) return;
this.isQueueRunning = value;
this.NotifyOfPropertyChange(() => this.IsQueueRunning);
}
}
///
/// Gets or sets JobStatus.
///
public string JobStatus
{
get
{
return this.jobStatus;
}
set
{
this.jobStatus = value;
this.NotifyOfPropertyChange(() => this.JobStatus);
}
}
///
/// Gets or sets JobsPending.
///
public string JobsPending
{
get
{
return this.jobsPending;
}
set
{
this.jobsPending = value;
this.NotifyOfPropertyChange(() => this.JobsPending);
}
}
///
/// Gets or sets WhenDoneAction.
///
public string WhenDoneAction
{
get
{
return this.whenDoneAction;
}
set
{
this.whenDoneAction = value;
this.NotifyOfPropertyChange(() => this.WhenDoneAction);
}
}
///
/// Gets the queue tasks.
///
public BindingList QueueTasks
{
get
{
return this.queueProcessor.Queue;
}
}
///
/// Gets or sets the selected items.
///
public BindingList SelectedItems { get; set; }
public QueueTask SelectedTask
{
get
{
return this.selectedTask;
}
set
{
if (Equals(value, this.selectedTask)) return;
this.selectedTask = value;
this.NotifyOfPropertyChange(() => this.SelectedTask);
this.HandleLogData();
this.NotifyOfPropertyChange(() => this.CanRetryJob);
this.NotifyOfPropertyChange(() => this.CanEditJob);
this.NotifyOfPropertyChange(() => this.CanRemoveJob);
this.NotifyOfPropertyChange(() => this.StatsVisible);
}
}
public string ActivityLog { get; private set; }
public bool CanRetryJob => this.SelectedTask != null && this.SelectedTask.Status != QueueItemStatus.Waiting;
public bool CanEditJob => this.SelectedTask != null;
public bool CanRemoveJob => this.SelectedTask != null;
public double ProgressValue
{
get => this.progressValue;
set
{
if (value == this.progressValue) return;
this.progressValue = value;
this.NotifyOfPropertyChange(() => this.ProgressValue);
}
}
public bool IntermediateProgress
{
get => this.intermediateProgress;
set
{
if (value == this.intermediateProgress) return;
this.intermediateProgress = value;
this.NotifyOfPropertyChange(() => this.IntermediateProgress);
}
}
public bool ShowEncodeProgress
{
get => this.showEncodeProgress;
set
{
if (value == this.showEncodeProgress) return;
this.showEncodeProgress = value;
this.NotifyOfPropertyChange(() => this.ShowEncodeProgress);
}
}
public bool IsNewQueueVisible { get; set; }
public bool IsInline { get; set; }
public bool StatsVisible
{
get
{
if (this.SelectedTask != null &&
(this.selectedTask.Status == QueueItemStatus.Completed || this.selectedTask.Status == QueueItemStatus.Error))
{
return true;
}
return false;
}
}
#endregion
#region Public Methods
///
/// Update the When Done Setting
///
///
/// The action.
///
public void WhenDone(string action)
{
this.WhenDone(action, true);
}
///
/// Update the When Done Setting
///
///
/// The action.
///
///
/// Save the change to the setting. Use false when updating UI.
///
public void WhenDone(string action, bool saveChange)
{
this.WhenDoneAction = action;
if (saveChange)
{
this.userSettingService.SetUserSetting(UserSettingConstants.WhenCompleteAction, action);
}
IOptionsViewModel ovm = IoC.Get();
ovm.UpdateSettings();
}
///
/// Clear the Queue
///
public void Clear()
{
MessageBoxResult result = this.errorService.ShowMessageBox(
Resources.QueueViewModel_ClearQueueConfrimation, Resources.Confirm, MessageBoxButton.YesNo, MessageBoxImage.Warning);
if (result == MessageBoxResult.Yes)
{
this.queueProcessor.Clear();
}
}
///
/// Clear Completed Items
///
public void ClearCompleted()
{
this.queueProcessor.ClearCompleted();
}
///
/// Close this window.
///
public void Close()
{
this.TryClose();
}
///
/// Handle the On Window Load
///
public override void OnLoad()
{
// Setup the window to the correct state.
this.IsQueueRunning = this.queueProcessor.EncodeService.IsEncoding;
this.JobsPending = string.Format(Resources.QueueViewModel_JobsPending, this.queueProcessor.Count);
base.OnLoad();
}
///
/// Can Pause the Queue.
/// Used by Caliburn Micro to enable/disable the context menu item.
///
///
/// True when we can pause the queue.
///
public bool CanPauseQueue()
{
return this.IsQueueRunning;
}
///
/// Pause the Queue
///
public void PauseQueue()
{
this.queueProcessor.Pause();
this.JobStatus = Resources.QueueViewModel_QueuePending;
this.JobsPending = string.Format(Resources.QueueViewModel_JobsPending, this.queueProcessor.Count);
this.IsQueueRunning = false;
MessageBox.Show(Resources.QueueViewModel_QueuePauseNotice, Resources.QueueViewModel_Queue,
MessageBoxButton.OK, MessageBoxImage.Information);
}
///
/// Pause the Queue
///
///
/// Prevents evaluation of CanPauseQueue
///
public void PauseQueueToolbar()
{
this.PauseQueue();
}
///
/// The remove selected jobs.
///
public void RemoveSelectedJobs()
{
MessageBoxResult result =
this.errorService.ShowMessageBox(
Resources.QueueViewModel_DelSelectedJobConfirmation,
Resources.Question,
MessageBoxButton.YesNo,
MessageBoxImage.Question);
if (result == MessageBoxResult.No)
{
return;
}
List tasksToRemove = this.SelectedItems.ToList();
foreach (QueueTask job in tasksToRemove)
{
this.RemoveJob(job);
}
}
///
/// Remove a Job from the queue
///
///
/// The Job to remove from the queue
///
public void RemoveJob(object queueTask)
{
QueueTask task = queueTask as QueueTask;
if (task == null)
{
return;
}
bool removed = false;
int index = this.QueueTasks.IndexOf(task);
if (task.Status == QueueItemStatus.InProgress)
{
MessageBoxResult result =
this.errorService.ShowMessageBox(
Resources.QueueViewModel_JobCurrentlyRunningWarning,
Resources.Warning,
MessageBoxButton.YesNo,
MessageBoxImage.Question);
if (result == MessageBoxResult.Yes)
{
this.queueProcessor.EncodeService.Stop();
this.queueProcessor.Remove(task);
removed = true;
}
}
else
{
this.queueProcessor.Remove(task);
removed = true;
}
if (this.QueueTasks.Any() && removed)
{
this.SelectedTask = index > 1 ? this.QueueTasks[index - 1] : this.QueueTasks.FirstOrDefault();
}
}
///
/// Reset the job state to waiting.
///
///
/// The task.
///
public void RetryJob(QueueTask task)
{
task.Status = QueueItemStatus.Waiting;
this.queueProcessor.BackupQueue(null);
this.JobsPending = string.Format(Resources.QueueViewModel_JobsPending, this.queueProcessor.Count);
this.NotifyOfPropertyChange(() => this.CanRetryJob);
}
///
/// Start Encode
///
public void StartQueue()
{
if (!this.QueueTasks.Any(a => a.Status == QueueItemStatus.Waiting || a.Status == QueueItemStatus.InProgress))
{
this.errorService.ShowMessageBox(
Resources.QueueViewModel_NoPendingJobs, Resources.Error, MessageBoxButton.OK, MessageBoxImage.Error);
return;
}
var firstOrDefault = this.QueueTasks.FirstOrDefault(s => s.Status == QueueItemStatus.Waiting);
if (firstOrDefault != null && !DriveUtilities.HasMinimumDiskSpace(firstOrDefault.Task.Destination,
this.userSettingService.GetUserSetting(UserSettingConstants.PauseOnLowDiskspaceLevel)))
{
this.errorService.ShowMessageBox(Resources.Main_LowDiskspace, Resources.Error, MessageBoxButton.OK, MessageBoxImage.Error);
return;
}
this.JobStatus = Resources.QueueViewModel_QueueStarted;
this.JobsPending = string.Format(Resources.QueueViewModel_JobsPending, this.queueProcessor.Count);
this.IsQueueRunning = true;
this.queueProcessor.Start(userSettingService.GetUserSetting(UserSettingConstants.ClearCompletedFromQueue));
}
///
/// Export the Queue to a file.
///
public void Export()
{
SaveFileDialog dialog = new SaveFileDialog
{
Filter = "Json (*.json)|*.json",
OverwritePrompt = true,
DefaultExt = ".json",
AddExtension = true
};
if (dialog.ShowDialog() == true)
{
this.queueProcessor.ExportJson(dialog.FileName);
}
}
///
/// Import a saved queue
///
public void Import()
{
OpenFileDialog dialog = new OpenFileDialog { Filter = "Json (*.json)|*.json", CheckFileExists = true };
if (dialog.ShowDialog() == true)
{
this.queueProcessor.ImportJson(dialog.FileName);
}
}
///
/// Edit this Job
///
///
/// The task.
///
public void EditJob(QueueTask task)
{
MessageBoxResult result = this.errorService.ShowMessageBox(
Resources.QueueViewModel_EditConfrimation,
"Modify Job?",
MessageBoxButton.YesNo,
MessageBoxImage.Question);
if (result != MessageBoxResult.Yes)
{
return;
}
// Remove the job if it is not already encoding. Let the user decide if they want to cancel or not.
this.RemoveJob(task);
// Pass a copy of the job back to the Main Screen
IMainViewModel mvm = IoC.Get();
mvm.EditQueueJob(task);
}
public void OpenSourceDirectory(QueueTask task)
{
if (task != null)
{
this.OpenDirectory(task.ScannedSourcePath);
}
}
public void OpenDestinationDirectory(QueueTask task)
{
if (task != null)
{
this.OpenDirectory(task.Task.Destination);
}
}
public void ResetSelectedJobs()
{
foreach (var task in this.SelectedItems)
{
if (task.Status == QueueItemStatus.Completed || task.Status == QueueItemStatus.Error)
{
this.RetryJob(task);
}
}
}
public void ResetAllJobs()
{
foreach (var task in this.QueueTasks)
{
if (task.Status == QueueItemStatus.Completed || task.Status == QueueItemStatus.Error)
{
this.RetryJob(task);
}
}
}
#endregion
#region Methods
public void Activate()
{
this.OnActivate();
}
public void Deactivate()
{
this.OnDeactivate(false);
}
///
/// Override the OnActive to run the Screen Loading code in the view model base.
///
protected override void OnActivate()
{
this.Load();
this.queueProcessor.QueueCompleted += this.queueProcessor_QueueCompleted;
this.queueProcessor.QueueChanged += this.QueueManager_QueueChanged;
this.queueProcessor.EncodeService.EncodeStatusChanged += this.EncodeService_EncodeStatusChanged;
this.queueProcessor.EncodeService.EncodeCompleted += this.EncodeService_EncodeCompleted;
this.queueProcessor.JobProcessingStarted += this.QueueProcessorJobProcessingStarted;
this.queueProcessor.QueuePaused += this.QueueProcessor_QueuePaused;
this.JobsPending = string.Format(Resources.QueueViewModel_JobsPending, this.queueProcessor.Count);
this.JobStatus = Resources.QueueViewModel_QueueReady;
base.OnActivate();
}
///
/// Override the Deactivate
///
///
/// The close.
///
protected override void OnDeactivate(bool close)
{
this.queueProcessor.QueueCompleted -= this.queueProcessor_QueueCompleted;
this.queueProcessor.QueueChanged -= this.QueueManager_QueueChanged;
this.queueProcessor.EncodeService.EncodeStatusChanged -= this.EncodeService_EncodeStatusChanged;
this.queueProcessor.EncodeService.EncodeCompleted -= this.EncodeService_EncodeCompleted;
this.queueProcessor.JobProcessingStarted -= this.QueueProcessorJobProcessingStarted;
this.queueProcessor.QueuePaused -= this.QueueProcessor_QueuePaused;
base.OnDeactivate(close);
}
private void OpenDirectory(string directory)
{
try
{
if (!string.IsNullOrEmpty(directory))
{
if (!File.Exists(directory) && !directory.EndsWith("\\"))
{
directory = directory + "\\";
}
directory = Path.GetDirectoryName(directory);
if (directory != null && Directory.Exists(directory))
{
Process.Start(directory);
}
}
}
catch (Exception exc)
{
this.errorService.ShowError(Resources.MainViewModel_UnableToLaunchDestDir, Resources.MainViewModel_UnableToLaunchDestDirSolution, exc);
}
}
///
/// Open the Log file directory
///
public void OpenLogDirectory()
{
string logDir = DirectoryUtilities.GetLogDirectory();
string windir = Environment.GetEnvironmentVariable("WINDIR");
Process prc = new Process { StartInfo = { FileName = windir + @"\explorer.exe", Arguments = logDir } };
prc.Start();
}
///
/// Copy the log file to the system clipboard
///
public void CopyLog()
{
try
{
Clipboard.SetDataObject(this.ActivityLog, true);
}
catch (Exception exc)
{
this.errorService.ShowError(Resources.Clipboard_Unavailable, Resources.Clipboard_Unavailable_Solution, exc);
}
}
private void HandleLogData()
{
if (this.SelectedTask == null || this.SelectedTask.Status == QueueItemStatus.InProgress || this.SelectedTask.Status == QueueItemStatus.Waiting)
{
this.ActivityLog = ResourcesUI.QueueView_LogNotAvailableYet;
}
else
{
try
{
// TODO full log path
if (!string.IsNullOrEmpty(this.SelectedTask.Statistics.CompletedActivityLogPath)
&& File.Exists(this.SelectedTask.Statistics.CompletedActivityLogPath))
{
using (StreamReader logReader = new StreamReader(this.SelectedTask.Statistics.CompletedActivityLogPath))
{
string logContent = logReader.ReadToEnd();
this.ActivityLog = logContent;
}
}
else
{
this.ActivityLog = string.Empty;
}
}
catch (Exception exc)
{
Debug.WriteLine(exc);
this.ActivityLog = exc.ToString();
}
}
this.NotifyOfPropertyChange(() => this.ActivityLog);
}
///
/// Handle the Encode Status Changed Event.
///
///
/// The sender.
///
///
/// The EncodeProgressEventArgs.
///
private void EncodeService_EncodeStatusChanged(object sender, EncodeProgressEventArgs e)
{
Execute.OnUIThread(() =>
{
string jobsPending = string.Format(Resources.Main_JobsPending_addon, this.queueProcessor.Count);
this.IntermediateProgress = false;
if (e.IsSubtitleScan)
{
this.JobStatus = string.Format(Resources.MainViewModel_EncodeStatusChanged_SubScan_StatusLabel,
e.Task,
e.TaskCount,
e.PercentComplete,
e.EstimatedTimeLeft,
e.ElapsedTime,
jobsPending);
this.ProgressValue = e.PercentComplete;
}
else if (e.IsMuxing)
{
this.JobStatus = ResourcesUI.MainView_Muxing;
this.IntermediateProgress = true;
}
else if (e.IsSearching)
{
this.JobStatus = string.Format(ResourcesUI.MainView_ProgressStatusWithTask, ResourcesUI.MainView_Searching, e.PercentComplete, e.EstimatedTimeLeft, jobsPending);
this.ProgressValue = e.PercentComplete;
}
else
{
this.JobStatus =
string.Format(Resources.MainViewModel_EncodeStatusChanged_StatusLabel,
e.Task,
e.TaskCount,
e.PercentComplete,
e.CurrentFrameRate,
e.AverageFrameRate,
e.EstimatedTimeLeft,
e.ElapsedTime,
jobsPending);
this.ProgressValue = e.PercentComplete;
}
});
}
///
/// Handle the Queue Changed Event.
///
///
/// The sender.
///
///
/// The e.
///
private void QueueManager_QueueChanged(object sender, EventArgs e)
{
this.JobsPending = string.Format(Resources.QueueViewModel_JobsPending, this.queueProcessor.Count);
if (!queueProcessor.IsProcessing)
{
this.JobStatus = Resources.QueueViewModel_QueueNotRunning;
this.IsQueueRunning = false;
}
this.NotifyOfPropertyChange(() => this.CanRetryJob);
this.NotifyOfPropertyChange(() => this.CanEditJob);
this.NotifyOfPropertyChange(() => this.CanRemoveJob);
this.NotifyOfPropertyChange(() => this.StatsVisible);
this.HandleLogData();
}
///
/// Handle the Queue Completed Event
///
///
/// The sender.
///
///
/// The EventArgs.
///
private void queueProcessor_QueueCompleted(object sender, EventArgs e)
{
this.JobStatus = Resources.QueueViewModel_QueueCompleted;
this.JobsPending = string.Format(Resources.QueueViewModel_JobsPending, this.queueProcessor.Count);
this.IsQueueRunning = false;
}
///
/// The encode service_ encode completed.
///
///
/// The sender.
///
///
/// The e.
///
private void EncodeService_EncodeCompleted(object sender, EncodeCompletedEventArgs e)
{
if (!this.queueProcessor.IsProcessing)
{
this.JobStatus = Resources.QueueViewModel_LastJobFinished;
}
}
///
/// The queue processor job processing started.
///
///
/// The sender.
///
///
/// The QueueProgressEventArgs.
///
private void QueueProcessorJobProcessingStarted(object sender, QueueProgressEventArgs e)
{
this.JobStatus = Resources.QueueViewModel_QueueStarted;
this.JobsPending = string.Format(Resources.QueueViewModel_JobsPending, this.queueProcessor.Count);
this.IsQueueRunning = true;
}
private void QueueProcessor_QueuePaused(object sender, EventArgs e)
{
this.JobStatus = Resources.QueueViewModel_QueuePaused;
this.JobsPending = string.Format(Resources.QueueViewModel_JobsPending, this.queueProcessor.Count);
this.IsQueueRunning = false;
}
#endregion
}
}