// --------------------------------------------------------------------------------------------------------------------
//
// This file is part of the HandBrake source code - It may be used under the terms of the GNU General Public License.
//
//
// HandBrakes Main Window
//
// --------------------------------------------------------------------------------------------------------------------
namespace HandBrakeWPF.ViewModels
{
using System;
using System.Collections.Generic;
using System.Diagnostics;
using System.Globalization;
using System.IO;
using System.Linq;
using System.Threading;
using System.Windows;
using System.Windows.Media.Imaging;
using Caliburn.Micro;
using HandBrake.ApplicationServices;
using HandBrake.ApplicationServices.Factories;
using HandBrake.ApplicationServices.Model;
using HandBrake.ApplicationServices.Model.Encoding;
using HandBrake.ApplicationServices.Parsing;
using HandBrake.ApplicationServices.Services.Interfaces;
using HandBrake.ApplicationServices.Utilities;
using HandBrakeWPF.Commands;
using HandBrakeWPF.Helpers;
using HandBrakeWPF.Model;
using HandBrakeWPF.Services.Interfaces;
using HandBrakeWPF.ViewModels.Interfaces;
using HandBrakeWPF.Views;
using Microsoft.Win32;
using Ookii.Dialogs.Wpf;
using Image = System.Windows.Controls.Image;
///
/// HandBrakes Main Window
///
public class MainViewModel : ViewModelBase, IMainViewModel
{
#region Private Variables and Services
///
/// The Encode Service
///
private readonly IQueueProcessor queueProcessor;
///
/// The preset service
///
private readonly IPresetService presetService;
///
/// The Error Service Backing field.
///
private readonly IErrorService errorService;
///
/// The Shell View Model
///
private readonly IShellViewModel shellViewModel;
///
/// Backing field for the update serivce.
///
private readonly IUpdateService updateService;
///
/// The drive detect service.
///
private readonly IDriveDetectService driveDetectService;
///
/// Backing field for the user setting service.
///
private readonly IUserSettingService userSettingService;
///
/// The Source Scan Service.
///
private readonly IScanServiceWrapper scanService;
///
/// The Encode Service
///
private readonly IEncodeServiceWrapper encodeService;
///
/// HandBrakes Main Window Title
///
private string windowName;
///
/// The Source Label
///
private string sourceLabel;
///
/// The Selected Output Format Backing Field
///
private OutputFormat selectedOutputFormat;
///
/// Is a MKV file backing field
///
private bool isMkv;
///
/// The Toolbar Status Label
///
private string statusLabel;
///
/// Program Status Label
///
private string programStatusLabel;
///
/// Backing field for the scanned source.
///
private Source scannedSource;
///
/// Backing field for the selected title.
///
private Title selectedTitle;
///
/// Backing field for duration
///
private string duration;
///
/// Is Encoding Backing Field
///
private bool isEncoding;
///
/// An Indicated to show the status window
///
private bool showStatusWindow;
///
/// Backing field for the selected preset.
///
private Preset selectedPreset;
///
/// Queue Edit Task
///
private EncodeTask queueEditTask;
///
/// The Source Menu Backing Field
///
private IEnumerable sourceMenu;
#endregion
///
/// Initializes a new instance of the class.
/// The viewmodel for HandBrakes main window.
///
///
/// The User Setting Service
///
///
/// The scan Service.
///
///
/// The encode Service.
///
///
/// The preset Service.
///
///
/// The Error Service
///
///
/// The shell View Model.
///
///
/// The update Service.
///
///
/// The drive Detect Service.
///
///
/// The notification Service.
/// *** Leave in Constructor. *** TODO find out why?
///
public MainViewModel(IUserSettingService userSettingService, IScanServiceWrapper scanService, IEncodeServiceWrapper encodeService, IPresetService presetService,
IErrorService errorService, IShellViewModel shellViewModel, IUpdateService updateService, IDriveDetectService driveDetectService, INotificationService notificationService)
{
this.scanService = scanService;
this.encodeService = encodeService;
this.presetService = presetService;
this.errorService = errorService;
this.shellViewModel = shellViewModel;
this.updateService = updateService;
this.driveDetectService = driveDetectService;
this.userSettingService = userSettingService;
this.queueProcessor = IoC.Get();
// Setup Properties
this.WindowTitle = "HandBrake";
this.CurrentTask = new EncodeTask();
this.CurrentTask.PropertyChanged += this.CurrentTask_PropertyChanged;
this.ScannedSource = new Source();
// Setup Events
this.scanService.ScanStared += this.ScanStared;
this.scanService.ScanCompleted += this.ScanCompleted;
this.scanService.ScanStatusChanged += this.ScanStatusChanged;
this.queueProcessor.JobProcessingStarted += this.QueueProcessorJobProcessingStarted;
this.queueProcessor.QueueCompleted += this.QueueCompleted;
this.queueProcessor.QueueChanged += this.QueueChanged;
this.queueProcessor.EncodeService.EncodeStatusChanged += this.EncodeStatusChanged;
this.userSettingService.SettingChanged += this.UserSettingServiceSettingChanged;
this.Presets = this.presetService.Presets;
this.CancelScanCommand = new CancelScanCommand(this.scanService);
}
#region View Model Properties
///
/// Gets or sets PictureSettingsViewModel.
///
public IPictureSettingsViewModel PictureSettingsViewModel { get; set; }
///
/// Gets or sets AudioViewModel.
///
public IAudioViewModel AudioViewModel { get; set; }
///
/// Gets or sets SubtitleViewModel.
///
public ISubtitlesViewModel SubtitleViewModel { get; set; }
///
/// Gets or sets ChaptersViewModel.
///
public IChaptersViewModel ChaptersViewModel { get; set; }
///
/// Gets or sets AdvancedViewModel.
///
public IAdvancedViewModel AdvancedViewModel { get; set; }
///
/// Gets or sets VideoViewModel.
///
public IVideoViewModel VideoViewModel { get; set; }
///
/// Gets or sets FiltersViewModel.
///
public IFiltersViewModel FiltersViewModel { get; set; }
#endregion
#region Properties
///
/// Gets or sets TestProperty.
///
public string WindowTitle
{
get
{
return this.windowName;
}
set
{
if (!Equals(this.windowName, value))
{
this.windowName = value;
}
}
}
///
/// Gets or sets the Program Status Toolbar Label
/// This indicates the status of HandBrake
///
public string ProgramStatusLabel
{
get
{
return string.IsNullOrEmpty(this.programStatusLabel) ? "Ready" : this.programStatusLabel;
}
set
{
if (!Equals(this.programStatusLabel, value))
{
this.programStatusLabel = value;
this.NotifyOfPropertyChange(() => this.ProgramStatusLabel);
}
}
}
///
/// Gets or sets the Program Status Toolbar Label
/// This indicates the status of HandBrake
///
public string StatusLabel
{
get
{
return string.IsNullOrEmpty(this.statusLabel) ? "Ready" : this.statusLabel;
}
set
{
if (!Equals(this.statusLabel, value))
{
this.statusLabel = value;
this.NotifyOfPropertyChange(() => this.StatusLabel);
}
}
}
///
/// Gets or sets the source menu.
///
public IEnumerable SourceMenu
{
get
{
return this.sourceMenu;
}
set
{
this.sourceMenu = value;
this.NotifyOfPropertyChange(() => SourceMenu);
}
}
///
/// Gets or sets Presets.
///
public IEnumerable Presets { get; set; }
///
/// Gets or sets SelectedPreset.
///
public Preset SelectedPreset
{
get
{
return this.selectedPreset;
}
set
{
this.selectedPreset = value;
if (this.SelectedPreset != null)
{
// Main Window Settings
this.CurrentTask.LargeFile = selectedPreset.Task.LargeFile;
this.CurrentTask.OptimizeMP4 = selectedPreset.Task.OptimizeMP4;
this.CurrentTask.IPod5GSupport = selectedPreset.Task.IPod5GSupport;
this.SelectedOutputFormat = selectedPreset.Task.OutputFormat;
// Tab Settings
this.PictureSettingsViewModel.SetPreset(this.SelectedPreset, this.CurrentTask);
this.VideoViewModel.SetPreset(this.SelectedPreset, this.CurrentTask);
this.FiltersViewModel.SetPreset(this.SelectedPreset, this.CurrentTask);
this.AudioViewModel.SetPreset(this.SelectedPreset, this.CurrentTask);
this.SubtitleViewModel.SetPreset(this.SelectedPreset, this.CurrentTask);
this.ChaptersViewModel.SetPreset(this.SelectedPreset, this.CurrentTask);
this.AdvancedViewModel.SetPreset(this.SelectedPreset, this.CurrentTask);
// Do this again to force an update for m4v/mp4 selection
this.SelectedOutputFormat = selectedPreset.Task.OutputFormat;
}
this.NotifyOfPropertyChange(() => this.SelectedPreset);
}
}
///
/// Gets or sets The Current Encode Task that the user is building
///
public EncodeTask CurrentTask { get; set; }
///
/// Gets or sets the Last Scanned Source
/// This object contains information about the scanned source.
///
public Source ScannedSource
{
get
{
return this.scannedSource;
}
set
{
this.scannedSource = value;
this.NotifyOfPropertyChange("ScannedSource");
}
}
///
/// Gets or sets the Source Label
/// This indicates the status of scans.
///
public string SourceLabel
{
get
{
return string.IsNullOrEmpty(this.sourceLabel) ? "Select 'Source' to continue" : this.sourceLabel;
}
set
{
if (!Equals(this.sourceLabel, value))
{
this.sourceLabel = value;
this.NotifyOfPropertyChange("SourceLabel");
}
}
}
///
/// Gets SourceName.
///
public string SourceName
{
get
{
// The title that is selected has a source name. This means it's part of a batch scan.
if (selectedTitle != null && !string.IsNullOrEmpty(selectedTitle.SourceName))
{
return Path.GetFileNameWithoutExtension(selectedTitle.SourceName);
}
// Check if we have a Folder, if so, check if it's a DVD / Bluray drive and get the label.
if (ScannedSource.ScanPath.EndsWith("\\"))
{
foreach (DriveInformation item in GeneralUtilities.GetDrives())
{
if (item.RootDirectory.Contains(this.ScannedSource.ScanPath))
{
return item.VolumeLabel;
}
}
}
if (Path.GetFileNameWithoutExtension(this.ScannedSource.ScanPath) != "VIDEO_TS")
return Path.GetFileNameWithoutExtension(this.ScannedSource.ScanPath);
return Path.GetFileNameWithoutExtension(Path.GetDirectoryName(this.ScannedSource.ScanPath));
}
}
///
/// Gets RangeMode.
///
public IEnumerable RangeMode
{
get
{
return new List
{
PointToPointMode.Chapters, PointToPointMode.Seconds, PointToPointMode.Frames
};
}
}
///
/// Gets a value indicating whether ShowTextEntryForPointToPointMode.
///
public bool ShowTextEntryForPointToPointMode
{
get
{
return this.SelectedPointToPoint != PointToPointMode.Chapters;
}
}
///
/// Gets StartEndRangeItems.
///
public IEnumerable StartEndRangeItems
{
get
{
if (this.SelectedTitle == null)
{
return null;
}
return this.SelectedTitle.Chapters.Select(item => item.ChapterNumber).Select(dummy => dummy).ToList();
}
}
///
/// Gets Angles.
///
public IEnumerable Angles
{
get
{
if (this.SelectedTitle == null)
{
return null;
}
List items = new List();
for (int i = 1; i <= this.selectedTitle.AngleCount + 1; i++)
{
items.Add(i);
}
return items;
}
}
///
/// Gets or sets Duration.
///
public string Duration
{
get
{
return string.IsNullOrEmpty(duration) ? "--:--:--" : duration;
}
set
{
duration = value;
this.NotifyOfPropertyChange("Duration");
}
}
///
/// Gets or sets a value indicating whether IsEncoding.
///
public bool IsEncoding
{
get
{
return this.isEncoding;
}
set
{
this.isEncoding = value;
this.NotifyOfPropertyChange(() => this.IsEncoding);
}
}
///
/// Gets or sets a value indicating whether ShowStatusWindow.
///
public bool ShowStatusWindow
{
get
{
return this.showStatusWindow;
}
set
{
this.showStatusWindow = value;
this.NotifyOfPropertyChange(() => this.ShowStatusWindow);
}
}
///
/// Gets or sets a value indicating whether IsMkv.
///
public bool IsMkv
{
get
{
return this.isMkv;
}
set
{
this.isMkv = value;
this.NotifyOfPropertyChange("IsMkv");
}
}
///
/// Gets RangeMode.
///
public IEnumerable OutputFormats
{
get
{
return new List
{
OutputFormat.Mp4, OutputFormat.Mkv
};
}
}
///
/// Gets a value indicating whether show debug menu.
///
public bool ShowDebugMenu
{
get
{
return this.userSettingService.GetUserSetting(UserSettingConstants.EnableDebugFeatures);
}
}
///
/// Gets or sets the cancel scan command.
///
public CancelScanCommand CancelScanCommand { get; set; }
///
/// Gets or sets Destination.
///
public string Destination
{
get
{
return this.CurrentTask.Destination;
}
set
{
this.CurrentTask.Destination = value;
this.NotifyOfPropertyChange(() => this.Destination);
}
}
///
/// Gets or sets SelectedTitle.
///
public Title SelectedTitle
{
get
{
return this.selectedTitle;
}
set
{
if (!object.Equals(this.selectedTitle, value))
{
this.selectedTitle = value;
if (this.selectedTitle == null)
{
return;
}
// Use the Path on the Title, or the Source Scan path if one doesn't exist.
this.CurrentTask.Source = !string.IsNullOrEmpty(this.selectedTitle.SourceName) ? this.selectedTitle.SourceName : this.ScannedSource.ScanPath;
this.CurrentTask.Title = value.TitleNumber;
this.NotifyOfPropertyChange(() => this.StartEndRangeItems);
this.NotifyOfPropertyChange(() => this.SelectedTitle);
this.NotifyOfPropertyChange(() => this.Angles);
// Default the Start and End Point dropdowns
this.SelectedStartPoint = 1;
this.SelectedEndPoint = this.selectedTitle.Chapters != null &&
this.selectedTitle.Chapters.Count != 0
? this.selectedTitle.Chapters.Last().ChapterNumber
: 1;
this.SelectedPointToPoint = PointToPointMode.Chapters;
this.SelectedAngle = 1;
if (this.UserSettingService.GetUserSetting(UserSettingConstants.AutoNaming))
{
this.Destination = AutoNameHelper.AutoName(this.CurrentTask, this.SourceName);
}
this.NotifyOfPropertyChange(() => this.CurrentTask);
this.Duration = this.selectedTitle.Duration.ToString();
// Setup the tab controls
this.SetupTabs();
}
}
}
///
/// Gets or sets SelectedAngle.
///
public int SelectedAngle
{
get
{
return this.CurrentTask.StartPoint;
}
set
{
this.CurrentTask.Angle = value;
this.NotifyOfPropertyChange(() => this.SelectedAngle);
}
}
///
/// Gets or sets SelectedStartPoint.
///
public int SelectedStartPoint
{
get
{
return this.CurrentTask.StartPoint;
}
set
{
this.CurrentTask.StartPoint = value;
this.NotifyOfPropertyChange(() => this.SelectedStartPoint);
this.Duration = this.DurationCalculation();
if (this.UserSettingService.GetUserSetting(UserSettingConstants.AutoNaming))
{
this.Destination = AutoNameHelper.AutoName(this.CurrentTask, this.SourceName);
}
}
}
///
/// Gets or sets SelectedEndPoint.
///
public int SelectedEndPoint
{
get
{
return this.CurrentTask.EndPoint;
}
set
{
this.CurrentTask.EndPoint = value;
this.NotifyOfPropertyChange(() => this.SelectedEndPoint);
this.Duration = this.DurationCalculation();
if (this.UserSettingService.GetUserSetting(UserSettingConstants.AutoNaming))
{
this.Destination = AutoNameHelper.AutoName(this.CurrentTask, this.SourceName);
}
}
}
///
/// Gets or sets SelectedPointToPoint.
///
public PointToPointMode SelectedPointToPoint
{
get
{
return this.CurrentTask.PointToPointMode;
}
set
{
this.CurrentTask.PointToPointMode = value;
this.NotifyOfPropertyChange(() => SelectedPointToPoint);
this.NotifyOfPropertyChange(() => ShowTextEntryForPointToPointMode);
if (value == PointToPointMode.Chapters && this.SelectedTitle != null)
{
this.SelectedStartPoint = 1;
this.SelectedEndPoint = selectedTitle.Chapters.Last().ChapterNumber;
}
else if (value == PointToPointMode.Seconds)
{
this.SelectedStartPoint = 0;
int timeInSeconds;
if (int.TryParse(selectedTitle.Duration.TotalSeconds.ToString(CultureInfo.InvariantCulture), out timeInSeconds))
{
this.SelectedEndPoint = timeInSeconds;
}
}
else
{
// Note this does not account for VFR. It's only a guesstimate.
double estimatedTotalFrames = selectedTitle.Fps * selectedTitle.Duration.TotalSeconds;
this.SelectedStartPoint = 0;
int totalFrames;
if (int.TryParse(estimatedTotalFrames.ToString(CultureInfo.InvariantCulture), out totalFrames))
{
this.SelectedEndPoint = totalFrames;
}
}
}
}
///
/// Gets or sets SelectedOutputFormat.
///
public OutputFormat SelectedOutputFormat
{
get
{
return this.selectedOutputFormat;
}
set
{
this.selectedOutputFormat = value;
this.CurrentTask.OutputFormat = value;
this.NotifyOfPropertyChange(() => SelectedOutputFormat);
this.NotifyOfPropertyChange(() => this.CurrentTask.OutputFormat);
this.NotifyOfPropertyChange(() => IsMkv);
this.SetExtension(string.Format(".{0}", this.selectedOutputFormat.ToString().ToLower())); // TODO, tidy up
this.VideoViewModel.RefreshTask();
this.AudioViewModel.RefreshTask();
}
}
///
/// Gets a value indicating whether show advanced tab.
///
public bool ShowAdvancedTab
{
get
{
return this.userSettingService.GetUserSetting(UserSettingConstants.ShowAdvancedTab);
}
}
#endregion
#region Load and Shutdown Handling
///
/// Initialise this view model.
///
public override void OnLoad()
{
// Check the CLI Executable.
CliCheckHelper.CheckCLIVersion();
// Perform an update check if required
this.updateService.PerformStartupUpdateCheck(this.HandleUpdateCheckResults);
// Setup the presets.
if (this.presetService.CheckIfPresetsAreOutOfDate())
if (!this.userSettingService.GetUserSetting(UserSettingConstants.PresetNotification))
this.errorService.ShowMessageBox("HandBrake has determined your built-in presets are out of date... These presets will now be updated.",
"Preset Update", MessageBoxButton.OK, MessageBoxImage.Information);
// Queue Recovery
QueueRecoveryHelper.RecoverQueue(this.queueProcessor, this.errorService);
this.SelectedPreset = this.presetService.DefaultPreset;
// Populate the Source menu with drives.
this.SourceMenu = this.GenerateSourceMenu();
this.driveDetectService.StartDetection(this.DriveTrayChanged);
// Log Cleaning
if (userSettingService.GetUserSetting(UserSettingConstants.ClearOldLogs))
{
Thread clearLog = new Thread(() => GeneralUtilities.ClearLogFiles(30));
clearLog.Start();
}
}
///
/// Shutdown this View
///
public void Shutdown()
{
// Shutdown Service
this.driveDetectService.Close();
this.scanService.Shutdown();
this.encodeService.Shutdown();
// Unsubscribe from Events.
this.scanService.ScanStared -= this.ScanStared;
this.scanService.ScanCompleted -= this.ScanCompleted;
this.scanService.ScanStatusChanged -= this.ScanStatusChanged;
this.queueProcessor.QueueCompleted -= this.QueueCompleted;
this.queueProcessor.QueueChanged -= this.QueueChanged;
this.queueProcessor.JobProcessingStarted -= this.QueueProcessorJobProcessingStarted;
this.queueProcessor.EncodeService.EncodeStatusChanged -= this.EncodeStatusChanged;
this.userSettingService.SettingChanged -= this.UserSettingServiceSettingChanged;
}
#endregion
#region Menu and Taskbar
///
/// Open the About Window
///
public void OpenAboutApplication()
{
OpenOptionsScreenCommand command = new OpenOptionsScreenCommand();
command.Execute(OptionsTab.About);
}
///
/// Open the Options Window
///
public void OpenOptionsWindow()
{
this.shellViewModel.DisplayWindow(ShellWindow.OptionsWindow);
}
///
/// Open the Log Window
///
public void OpenLogWindow()
{
Window window = Application.Current.Windows.Cast().FirstOrDefault(x => x.GetType() == typeof(LogView));
if (window != null)
{
window.Activate();
}
else
{
this.WindowManager.ShowWindow(IoC.Get());
}
}
///
/// Open the Queue Window.
///
public void OpenQueueWindow()
{
Window window = Application.Current.Windows.Cast().FirstOrDefault(x => x.GetType() == typeof(QueueView));
if (window != null)
{
window.Activate();
}
else
{
this.WindowManager.ShowWindow(IoC.Get());
}
}
///
/// Open the Queue Window.
///
public void OpenPreviewWindow()
{
Window window = Application.Current.Windows.Cast().FirstOrDefault(x => x.GetType() == typeof(PreviewView));
IPreviewViewModel viewModel = IoC.Get();
if (window != null)
{
viewModel.Task = this.CurrentTask;
window.Activate();
}
else
{
viewModel.Task = this.CurrentTask;
this.WindowManager.ShowWindow(viewModel);
}
}
///
/// Launch the Help pages.
///
public void LaunchHelp()
{
Process.Start("https://trac.handbrake.fr/wiki/HandBrakeGuide");
}
///
/// Check for Updates.
///
public void CheckForUpdates()
{
OpenOptionsScreenCommand command = new OpenOptionsScreenCommand();
command.Execute(OptionsTab.Updates);
}
///
/// Add the current task to the queue.
///
public void AddToQueue()
{
if (this.ScannedSource == null || string.IsNullOrEmpty(this.ScannedSource.ScanPath) || this.ScannedSource.Titles.Count == 0)
{
this.errorService.ShowMessageBox("You must first scan a source and setup your job before adding to the queue.", "Error", MessageBoxButton.OK, MessageBoxImage.Error);
return;
}
QueueTask task = new QueueTask { Task = new EncodeTask(this.CurrentTask) };
if (!this.queueProcessor.CheckForDestinationPathDuplicates(task.Task.Destination))
{
this.queueProcessor.Add(task);
}
else
{
this.errorService.ShowMessageBox("There are jobs on the queue with the same destination path. Please choose a different path for this job.", "Error", MessageBoxButton.OK, MessageBoxImage.Warning);
}
if (!this.IsEncoding)
{
this.ProgramStatusLabel = string.Format("{0} Encodes Pending", this.queueProcessor.Count);
}
}
///
/// Add all Items to the queue
///
public void AddAllToQueue()
{
if (this.ScannedSource == null || this.ScannedSource.Titles == null || this.ScannedSource.Titles.Count == 0)
{
this.errorService.ShowMessageBox("You must first scan a source and setup your job before adding to the queue.", "Error", MessageBoxButton.OK, MessageBoxImage.Error);
return;
}
if (!AutoNameHelper.IsAutonamingEnabled())
{
this.errorService.ShowMessageBox("You must turn on automatic file naming in preferences before you can add to the queue.", "Error", MessageBoxButton.OK, MessageBoxImage.Error);
return;
}
foreach (Title title in this.ScannedSource.Titles)
{
this.SelectedTitle = title;
this.AddToQueue();
}
}
///
/// Folder Scan
///
public void FolderScan()
{
VistaFolderBrowserDialog dialog = new VistaFolderBrowserDialog { Description = "Please select a folder.", UseDescriptionForTitle = true };
dialog.ShowDialog();
this.StartScan(dialog.SelectedPath, 0);
}
///
/// File Scan
///
public void FileScan()
{
VistaOpenFileDialog dialog = new VistaOpenFileDialog { Filter = "All files (*.*)|*.*" };
dialog.ShowDialog();
this.StartScan(dialog.FileName, 0);
}
///
/// Folder Scan
///
public void FolderScanTitleSpecific()
{
VistaFolderBrowserDialog dialog = new VistaFolderBrowserDialog { Description = "Please select a folder.", UseDescriptionForTitle = true };
dialog.ShowDialog();
if (string.IsNullOrEmpty(dialog.SelectedPath))
{
return;
}
ITitleSpecificViewModel titleSpecificView = IoC.Get();
this.WindowManager.ShowDialog(titleSpecificView);
if (titleSpecificView.SelectedTitle.HasValue)
{
this.StartScan(dialog.SelectedPath, titleSpecificView.SelectedTitle.Value);
}
}
///
/// File Scan
///
public void FileScanTitleSpecific()
{
VistaOpenFileDialog dialog = new VistaOpenFileDialog { Filter = "All files (*.*)|*.*" };
dialog.ShowDialog();
if (string.IsNullOrEmpty(dialog.FileName))
{
return;
}
ITitleSpecificViewModel titleSpecificView = IoC.Get();
this.WindowManager.ShowDialog(titleSpecificView);
if (titleSpecificView.SelectedTitle.HasValue)
{
this.StartScan(dialog.FileName, titleSpecificView.SelectedTitle.Value);
}
}
///
/// Cancel a Scan
///
public void CancelScan()
{
this.scanService.Stop();
}
///
/// Start an Encode
///
public void StartEncode()
{
if (this.queueProcessor.IsProcessing)
{
this.errorService.ShowMessageBox("HandBrake is already encoding.", "Error", MessageBoxButton.OK, MessageBoxImage.Error);
return;
}
// Check if we already have jobs, and if we do, just start the queue.
if (this.queueProcessor.Count != 0)
{
this.queueProcessor.Start();
return;
}
// Otherwise, perform Santiy Checking then add to the queue and start if everything is ok.
if (this.SelectedTitle == null)
{
this.errorService.ShowMessageBox("You must first scan a source.", "Error", MessageBoxButton.OK, MessageBoxImage.Error);
return;
}
if (string.IsNullOrEmpty(this.Destination))
{
this.errorService.ShowMessageBox("The Destination field was empty.", "Error", MessageBoxButton.OK, MessageBoxImage.Error);
return;
}
if (File.Exists(this.Destination))
{
MessageBoxResult result = this.errorService.ShowMessageBox("The current file already exists, do you wish to overwrite it?", "Question", MessageBoxButton.YesNo, MessageBoxImage.Question);
if (result == MessageBoxResult.No)
{
return;
}
}
// Create the Queue Task and Start Processing
QueueTask task = new QueueTask
{
Task = new EncodeTask(this.CurrentTask),
CustomQuery = false
};
this.queueProcessor.Add(task);
this.queueProcessor.Start();
this.IsEncoding = true;
}
///
/// Edit a Queue Task
///
///
/// The task.
///
public void EditQueueJob(EncodeTask task)
{
// Rescan the source to make sure it's still valid
this.queueEditTask = task;
this.scanService.Scan(task.Source, task.Title, this.UserSettingService.GetUserSetting(ASUserSettingConstants.PreviewScanCount), QueueEditAction);
}
///
/// Pause an Encode
///
public void PauseEncode()
{
this.queueProcessor.Pause();
}
///
/// Stop an Encode.
///
public void StopEncode()
{
this.queueProcessor.Pause();
this.encodeService.Stop();
}
///
/// Shutdown the Application
///
public void ExitApplication()
{
Application.Current.Shutdown();
}
///
/// DEBUG: Show CLI Query for settings+6
///
public void ShowCliQuery()
{
this.errorService.ShowMessageBox(
QueryGeneratorUtility.GenerateQuery(this.CurrentTask,
userSettingService.GetUserSetting(ASUserSettingConstants.PreviewScanCount),
userSettingService.GetUserSetting(ASUserSettingConstants.Verbosity),
userSettingService.GetUserSetting(ASUserSettingConstants.DisableLibDvdNav)),
"CLI Query",
MessageBoxButton.OK,
MessageBoxImage.Information);
}
///
/// The debug scan log.
///
public void DebugScanLog()
{
VistaOpenFileDialog dialog = new VistaOpenFileDialog();
dialog.ShowDialog();
if (File.Exists(dialog.FileName))
{
this.scanService.DebugScanLog(dialog.FileName);
}
}
#endregion
#region Main Window Public Methods
///
/// Support dropping a file onto the main window to scan.
///
///
/// The DragEventArgs.
///
public void FilesDroppedOnWindow(DragEventArgs e)
{
if (e.Data.GetDataPresent(DataFormats.FileDrop))
{
string[] fileNames = e.Data.GetData(DataFormats.FileDrop, true) as string[];
if (fileNames != null && fileNames.Any() && (File.Exists(fileNames[0]) || Directory.Exists(fileNames[0])))
{
this.StartScan(fileNames[0], 0);
}
}
e.Handled = true;
}
///
/// The Destination Path
///
public void BrowseDestination()
{
SaveFileDialog saveFileDialog = new SaveFileDialog
{
Filter = "mp4|*.mp4;*.m4v|mkv|*.mkv",
CheckPathExists = true,
AddExtension = true,
DefaultExt = ".mp4",
OverwritePrompt = true,
FilterIndex = this.CurrentTask.OutputFormat == OutputFormat.Mkv ? 1 : 0,
};
if (this.CurrentTask != null && !string.IsNullOrEmpty(this.CurrentTask.Destination))
{
if (Directory.Exists(Path.GetDirectoryName(this.CurrentTask.Destination)))
{
saveFileDialog.InitialDirectory = Path.GetDirectoryName(this.CurrentTask.Destination) + "\\";
}
saveFileDialog.FileName = Path.GetFileName(this.CurrentTask.Destination);
}
saveFileDialog.ShowDialog();
this.Destination = saveFileDialog.FileName;
// Set the Extension Dropdown. This will also set Mp4/m4v correctly.
if (!string.IsNullOrEmpty(saveFileDialog.FileName))
{
switch (Path.GetExtension(saveFileDialog.FileName))
{
case ".mkv":
this.SelectedOutputFormat = OutputFormat.Mkv;
break;
case ".mp4":
this.SelectedOutputFormat = OutputFormat.Mp4;
break;
case ".m4v":
this.SelectedOutputFormat = OutputFormat.M4V;
break;
}
this.NotifyOfPropertyChange(() => this.CurrentTask);
}
}
///
/// Add a Preset
///
public void PresetAdd()
{
IAddPresetViewModel presetViewModel = IoC.Get();
presetViewModel.Setup(this.CurrentTask);
this.WindowManager.ShowWindow(presetViewModel);
}
///
/// Update a selected preset.
///
public void PresetUpdate()
{
if (this.SelectedPreset == null)
{
this.errorService.ShowMessageBox(
"Please select a preset to update.", "No Preset selected", MessageBoxButton.OK, MessageBoxImage.Warning);
return;
}
if (this.SelectedPreset.IsBuildIn)
{
this.errorService.ShowMessageBox(
"You can not modify built in presets. Please select one of your own presets.", "No Preset selected", MessageBoxButton.OK, MessageBoxImage.Warning);
return;
}
if (this.errorService.ShowMessageBox("Are you sure you wish to update the selected preset?", "Are you sure?", MessageBoxButton.YesNo, MessageBoxImage.Question) == MessageBoxResult.Yes)
{
this.SelectedPreset.Task = new EncodeTask(this.CurrentTask);
this.presetService.Update(this.SelectedPreset);
this.errorService.ShowMessageBox(
"The Preset has now been updated with your current settings.", "Preset Updated", MessageBoxButton.OK, MessageBoxImage.Information);
}
}
///
/// Remove a Preset
///
public void PresetRemove()
{
if (this.selectedPreset != null)
{
this.presetService.Remove(this.selectedPreset);
}
else
{
MessageBox.Show("Please select a preset.", "Presets", MessageBoxButton.OK, MessageBoxImage.Warning);
}
}
///
/// Set a default preset
///
public void PresetSetDefault()
{
if (this.selectedPreset != null)
{
this.presetService.SetDefault(this.selectedPreset);
MessageBox.Show(string.Format("New Default Preset Set: {0}", this.selectedPreset.Name), "Presets", MessageBoxButton.OK, MessageBoxImage.Warning);
}
else
{
MessageBox.Show("Please select a preset.", "Presets", MessageBoxButton.OK, MessageBoxImage.Warning);
}
}
///
/// Import a Preset
///
public void PresetImport()
{
VistaOpenFileDialog dialog = new VistaOpenFileDialog { Filter = "Plist (*.plist)|*.plist", CheckFileExists = true };
dialog.ShowDialog();
string filename = dialog.FileName;
if (!string.IsNullOrEmpty(filename))
{
PList plist = new PList(filename);
Preset preset = PlistPresetFactory.CreatePreset(plist);
if (this.presetService.CheckIfPresetExists(preset.Name))
{
if (!presetService.CanUpdatePreset(preset.Name))
{
MessageBox.Show(
"You can not import a preset with the same name as a built-in preset.",
"Error",
MessageBoxButton.OK,
MessageBoxImage.Error);
return;
}
MessageBoxResult result =
MessageBox.Show(
"This preset appears to already exist. Would you like to overwrite it?",
"Overwrite preset?",
MessageBoxButton.YesNo,
MessageBoxImage.Warning);
if (result == MessageBoxResult.Yes)
{
presetService.Update(preset);
}
}
else
{
presetService.Add(preset);
}
this.NotifyOfPropertyChange(() => this.Presets);
}
}
///
/// Export a Preset
///
public void PresetExport()
{
VistaSaveFileDialog savefiledialog = new VistaSaveFileDialog { Filter = "plist|*.plist", CheckPathExists = true, AddExtension = true };
if (this.selectedPreset != null)
{
savefiledialog.ShowDialog();
string filename = savefiledialog.FileName;
if (filename != null)
{
PlistUtility.Export(savefiledialog.FileName, this.selectedPreset, userSettingService.GetUserSetting(ASUserSettingConstants.HandBrakeBuild).ToString(CultureInfo.InvariantCulture));
}
}
else
{
MessageBox.Show("Please select a preset.", "Presets", MessageBoxButton.OK, MessageBoxImage.Warning);
}
}
///
/// Reset built-in presets
///
public void PresetReset()
{
this.presetService.UpdateBuiltInPresets();
this.NotifyOfPropertyChange("Presets");
this.SelectedPreset = this.presetService.DefaultPreset;
}
///
/// Start a Scan
///
///
/// The filename.
///
///
/// The title.
///
public void StartScan(string filename, int title)
{
if (!string.IsNullOrEmpty(filename))
{
this.scanService.Scan(filename, title, this.UserSettingService.GetUserSetting(ASUserSettingConstants.PreviewScanCount), null);
}
}
#endregion
#region Private Methods
///
/// Update all the UI Components to allow the user to edit their previous settings.
///
///
/// The successful.
///
private void QueueEditAction(bool successful)
{
Execute.OnUIThread(() =>
{
// Copy all the Scan data into the UI
this.scanService.SouceData.CopyTo(this.ScannedSource);
this.NotifyOfPropertyChange(() => this.ScannedSource);
this.NotifyOfPropertyChange(() => this.ScannedSource.Titles);
// Select the Users Title
this.CurrentTask = new EncodeTask(queueEditTask);
this.NotifyOfPropertyChange(() => this.CurrentTask);
this.SelectedTitle = this.ScannedSource.Titles.FirstOrDefault(t => t.TitleNumber == this.CurrentTask.Title);
// Update the Main UI control Area (TODO)
this.CurrentTask = new EncodeTask(queueEditTask);
this.NotifyOfPropertyChange(() => this.CurrentTask);
// Update the Tab Controls (TODO)
this.PictureSettingsViewModel.UpdateTask(this.CurrentTask);
this.VideoViewModel.UpdateTask(this.CurrentTask);
this.FiltersViewModel.UpdateTask(this.CurrentTask);
this.AudioViewModel.UpdateTask(this.CurrentTask);
this.SubtitleViewModel.UpdateTask(this.CurrentTask);
this.ChaptersViewModel.UpdateTask(this.CurrentTask);
this.AdvancedViewModel.UpdateTask(this.CurrentTask);
// Cleanup
this.ShowStatusWindow = false;
});
}
///
/// Make sure the correct file extension is set based on user preferences and setup the GUI for the file container selected.
///
///
/// The new extension.
///
private void SetExtension(string newExtension)
{
// Make sure the output extension is set correctly based on the users preferences and selection.
if (newExtension == ".mp4" || newExtension == ".m4v")
{
switch (this.UserSettingService.GetUserSetting(UserSettingConstants.UseM4v))
{
case 0: // Auto
newExtension = this.CurrentTask.RequiresM4v ? ".m4v" : ".mp4";
break;
case 1: // MP4
newExtension = ".mp4";
break;
case 2: // M4v
newExtension = ".m4v";
break;
}
this.IsMkv = false;
}
// Now disable controls that are not required. The Following are for MP4 only!
if (newExtension == ".mkv")
{
this.IsMkv = true;
this.CurrentTask.LargeFile = false;
this.CurrentTask.OptimizeMP4 = false;
this.CurrentTask.IPod5GSupport = false;
}
// Update The browse file extension display
if (Path.HasExtension(newExtension))
{
this.Destination = Path.ChangeExtension(this.Destination, newExtension);
}
// Update the UI Display
this.NotifyOfPropertyChange(() => this.CurrentTask);
}
///
/// Setup the UI tabs. Passes in any relevant models for setup.
///
private void SetupTabs()
{
// Setup the Tabs
if (this.selectedTitle != null)
{
this.PictureSettingsViewModel.SetSource(this.SelectedTitle, this.SelectedPreset, this.CurrentTask);
this.VideoViewModel.SetSource(this.SelectedTitle, this.SelectedPreset, this.CurrentTask);
this.FiltersViewModel.SetSource(this.SelectedTitle, this.SelectedPreset, this.CurrentTask);
this.AudioViewModel.SetSource(this.SelectedTitle, this.SelectedPreset, this.CurrentTask);
this.SubtitleViewModel.SetSource(this.SelectedTitle, this.SelectedPreset, this.CurrentTask);
this.ChaptersViewModel.SetSource(this.SelectedTitle, this.SelectedPreset, this.CurrentTask);
this.AdvancedViewModel.SetSource(this.SelectedTitle, this.SelectedPreset, this.CurrentTask);
}
}
///
/// Calculate the duration between the end and start point
///
///
/// The duration calculation.
///
private string DurationCalculation()
{
if (this.selectedTitle == null)
{
return "--:--:--";
}
double startEndDuration = this.SelectedEndPoint - this.SelectedStartPoint;
switch (this.SelectedPointToPoint)
{
case PointToPointMode.Chapters:
return this.SelectedTitle.CalculateDuration(this.SelectedStartPoint, this.SelectedEndPoint).ToString();
case PointToPointMode.Seconds:
return TimeSpan.FromSeconds(startEndDuration).ToString();
case PointToPointMode.Frames:
startEndDuration = startEndDuration / selectedTitle.Fps;
return TimeSpan.FromSeconds(startEndDuration).ToString();
}
return "--:--:--";
}
///
/// Handle Update Check Results
///
///
/// The information.
///
private void HandleUpdateCheckResults(UpdateCheckInformation information)
{
if (information.NewVersionAvailable)
{
this.ProgramStatusLabel = "A New Update is Available. Goto Tools Menu > Options to Install";
}
}
#endregion
#region Event Handlers
///
/// Handle the Scan Status Changed Event.
///
///
/// The Sender
///
///
/// The EventArgs
///
private void ScanStatusChanged(object sender, HandBrake.ApplicationServices.EventArgs.ScanProgressEventArgs e)
{
this.SourceLabel = "Scanning Title " + e.CurrentTitle + " of " + e.Titles;
this.StatusLabel = "Scanning Title " + e.CurrentTitle + " of " + e.Titles;
}
///
/// Handle the Scan Completed Event
///
///
/// The Sender
///
///
/// The EventArgs
///
private void ScanCompleted(object sender, HandBrake.ApplicationServices.EventArgs.ScanCompletedEventArgs e)
{
this.scanService.SouceData.CopyTo(this.ScannedSource);
Execute.OnUIThread(() =>
{
if (e.Successful)
{
this.NotifyOfPropertyChange(() => this.ScannedSource);
this.NotifyOfPropertyChange(() => this.ScannedSource.Titles);
this.SelectedTitle = this.ScannedSource.Titles.FirstOrDefault(t => t.MainTitle)
?? this.ScannedSource.Titles.FirstOrDefault();
}
this.ShowStatusWindow = false;
if (e.Successful)
{
if (this.SelectedTitle != null && !string.IsNullOrEmpty(this.SelectedTitle.SourceName))
{
this.SourceLabel = this.SelectedTitle.SourceName;
}
else
{
this.SourceLabel = this.SourceName;
}
this.StatusLabel = "Scan Completed";
}
else if (!e.Successful && e.Exception == null)
{
this.SourceLabel = "Scan Cancelled.";
this.StatusLabel = "Scan Cancelled.";
}
else
{
this.SourceLabel = "Scan Failed... See Activity Log for details.";
this.StatusLabel = "Scan Failed... See Activity Log for details.";
}
});
}
///
/// Handle the Scan Started Event
///
///
/// The Sender
///
///
/// The EventArgs
///
private void ScanStared(object sender, EventArgs e)
{
Execute.OnUIThread(
() =>
{
this.StatusLabel = "Scanning source, please wait...";
this.ShowStatusWindow = true;
});
}
///
/// The Encode Status has changed Handler
///
///
/// The Sender
///
///
/// The Encode Progress Event Args
///
private void EncodeStatusChanged(object sender, HandBrake.ApplicationServices.EventArgs.EncodeProgressEventArgs e)
{
Execute.OnUIThread(
() =>
{
if (this.IsEncoding)
{
this.ProgramStatusLabel =
string.Format(
"{0:00.00}%, FPS: {1:000.0}, Avg FPS: {2:000.0}, Time Remaining: {3}, Elapsed: {4:hh\\:mm\\:ss}, Pending Jobs {5}",
e.PercentComplete,
e.CurrentFrameRate,
e.AverageFrameRate,
e.EstimatedTimeLeft,
e.ElapsedTime,
this.queueProcessor.Count);
}
});
}
///
/// Handle the Queue Starting Event
///
///
/// The sender.
///
///
/// The e.
///
void QueueProcessorJobProcessingStarted(object sender, HandBrake.ApplicationServices.EventArgs.QueueProgressEventArgs e)
{
Execute.OnUIThread(
() =>
{
this.ProgramStatusLabel = "Preparing to encode ...";
this.IsEncoding = true;
});
}
///
/// The Queue has completed handler
///
///
/// The Sender
///
///
/// The EventArgs
///
private void QueueCompleted(object sender, EventArgs e)
{
this.IsEncoding = false;
Execute.OnUIThread(
() =>
{
this.ProgramStatusLabel = "Queue Finished";
this.IsEncoding = false;
});
}
///
/// The queue changed.
///
///
/// The sender.
///
///
/// The EventArgs.
///
private void QueueChanged(object sender, EventArgs e)
{
Execute.OnUIThread(
() =>
{
this.ProgramStatusLabel = string.Format("{0} Encodes Pending", this.queueProcessor.Count);
});
}
///
/// The process drive.
///
///
/// The item.
///
private void ProcessDrive(object item)
{
if (item != null)
{
this.StartScan(((DriveInformation)item).RootDirectory, 0);
}
}
///
/// The generate source menu.
///
///
/// The System.Collections.Generic.IEnumerable`1[T -> HandBrakeWPF.Model.SourceMenuItem].
///
private IEnumerable GenerateSourceMenu()
{
List menuItems = new List();
SourceMenuItem folderScan = new SourceMenuItem
{
Image = new Image { Source = new BitmapImage(new Uri("pack://application:,,,/HandBrake;component/Views/Images/folder.png")), Width = 16, Height = 16 },
Text = "Open Folder",
Command = new SourceMenuCommand(this.FolderScan)
};
SourceMenuItem fileScan = new SourceMenuItem
{
Image = new Image { Source = new BitmapImage(new Uri("pack://application:,,,/HandBrake;component/Views/Images/Movies.png")), Width = 16, Height = 16 },
Text = "Open File",
Command = new SourceMenuCommand(this.FileScan)
};
SourceMenuItem titleSpecific = new SourceMenuItem { Text = "Title Specific Scan" };
SourceMenuItem folderScanTitle = new SourceMenuItem
{
Image = new Image { Source = new BitmapImage(new Uri("pack://application:,,,/HandBrake;component/Views/Images/folder.png")), Width = 16, Height = 16 },
Text = "Open Folder",
Command = new SourceMenuCommand(this.FolderScanTitleSpecific)
};
SourceMenuItem fileScanTitle = new SourceMenuItem
{
Image = new Image { Source = new BitmapImage(new Uri("pack://application:,,,/HandBrake;component/Views/Images/Movies.png")), Width = 16, Height = 16 },
Text = "Open File",
Command = new SourceMenuCommand(this.FileScanTitleSpecific)
};
titleSpecific.Children.Add(folderScanTitle);
titleSpecific.Children.Add(fileScanTitle);
menuItems.Add(folderScan);
menuItems.Add(fileScan);
menuItems.Add(titleSpecific);
// Drives
menuItems.AddRange(
from item in GeneralUtilities.GetDrives()
let driveInformation = item
select
new SourceMenuItem
{
Image = new Image { Source = new BitmapImage(new Uri("pack://application:,,,/HandBrake;component/Views/Images/disc_small.png")), Width = 16, Height = 16 },
Text = string.Format("{0} ({1})", item.RootDirectory, item.VolumeLabel),
Command = new SourceMenuCommand(() => this.ProcessDrive(driveInformation)),
Tag = item
});
return menuItems;
}
///
/// The drive tray changed.
///
private void DriveTrayChanged()
{
Caliburn.Micro.Execute.OnUIThread(() => this.SourceMenu = this.GenerateSourceMenu());
}
///
/// Allows the main window to respond to setting changes.
///
///
/// The sender.
///
///
/// The e.
///
private void UserSettingServiceSettingChanged(object sender, HandBrake.ApplicationServices.EventArgs.SettingChangedEventArgs e)
{
if (e.Key == UserSettingConstants.ShowAdvancedTab)
{
this.NotifyOfPropertyChange(() => this.ShowAdvancedTab);
}
}
///
/// Handle the property changed event of the encode task.
/// Allows the main window to respond to changes.
///
///
/// The sender.
///
///
/// The e.
///
private void CurrentTask_PropertyChanged(object sender, System.ComponentModel.PropertyChangedEventArgs e)
{
if (e.PropertyName == UserSettingConstants.ShowAdvancedTab)
{
this.NotifyOfPropertyChange(() => this.ShowAdvancedTab);
}
}
#endregion
}
}