// --------------------------------------------------------------------------------------------------------------------
//
// This file is part of the HandBrake source code - It may be used under the terms of the GNU General Public License.
//
//
// Scan a Source
//
// --------------------------------------------------------------------------------------------------------------------
namespace HandBrake.ApplicationServices.Services
{
using System;
using System.Collections.Generic;
using System.IO;
using System.Text;
using System.Windows.Media.Imaging;
using HandBrake.ApplicationServices.EventArgs;
using HandBrake.ApplicationServices.Model;
using HandBrake.ApplicationServices.Model.Encoding;
using HandBrake.ApplicationServices.Parsing;
using HandBrake.ApplicationServices.Services.Interfaces;
using HandBrake.ApplicationServices.Utilities;
using HandBrake.Interop;
using HandBrake.Interop.EventArgs;
using HandBrake.Interop.Interfaces;
using HandBrake.Interop.Model;
using AudioTrack = HandBrake.ApplicationServices.Parsing.Audio;
using ScanProgressEventArgs = HandBrake.Interop.EventArgs.ScanProgressEventArgs;
using Size = System.Drawing.Size;
///
/// Scan a Source
///
public class LibScan : IScan
{
/*
* TODO
* 1. Expose the Previews code.
* 2. Cleanup old instances.
*
*/
#region Private Variables
///
/// Lock for the log file
///
static readonly object LogLock = new object();
///
/// Log data from HandBrakeInstance
///
private readonly StringBuilder logging;
///
/// The Log File Header
///
private readonly StringBuilder header;
///
/// The Current source scan path.
///
private string currentSourceScanPath;
///
/// The log dir.
///
private static string logDir = Environment.GetFolderPath(Environment.SpecialFolder.ApplicationData) + "\\HandBrake\\logs";
///
/// The dvd info path.
///
private string dvdInfoPath = Path.Combine(logDir, string.Format("last_scan_log{0}.txt", GeneralUtilities.ProcessId));
///
/// The scan log.
///
private StreamWriter scanLog;
///
/// LibHB Instance
///
private IHandBrakeInstance instance;
#endregion
///
/// Initializes a new instance of the class.
///
public LibScan()
{
logging = new StringBuilder();
header = GeneralUtilities.CreateCliLogHeader();
try
{
HandBrakeUtils.MessageLogged += this.HandBrakeInstanceMessageLogged;
HandBrakeUtils.ErrorLogged += this.HandBrakeInstanceErrorLogged;
}
catch (Exception)
{
// Do nothing.
}
}
#region Events
///
/// Scan has Started
///
public event EventHandler ScanStared;
///
/// Scan has completed
///
public event ScanCompletedStatus ScanCompleted;
///
/// Encode process has progressed
///
public event ScanProgessStatus ScanStatusChanged;
#endregion
#region Properties
///
/// Gets a value indicating whether IsScanning.
///
public bool IsScanning { get; private set; }
///
/// Gets the Souce Data.
///
public Source SouceData { get; private set; }
///
/// Gets ActivityLog.
///
public string ActivityLog
{
get
{
string noLog = "No log data available... Log data will show here after you scan a source. \n\nOpen the log file directory to get previous log files.";
return string.IsNullOrEmpty(this.logging.ToString()) ? this.header + noLog : this.header + this.logging.ToString();
}
}
#endregion
#region Public Methods
///
/// Scan a Source Path.
/// Title 0: scan all
///
///
/// Path to the file to scan
///
///
/// int title number. 0 for scan all
///
///
/// The post Action.
///
///
/// The configuraiton.
///
public void Scan(string sourcePath, int title, Action postAction, HBConfiguration configuraiton)
{
// Try to cleanup any previous scan instances.
if (instance != null)
{
try
{
this.scanLog.Close();
this.scanLog.Dispose();
instance.Dispose();
}
catch (Exception)
{
// Do Nothing
}
}
// Clear down the logging
this.logging.Clear();
try
{
// Make we don't pick up a stale last_scan_log_xyz.txt (and that we have rights to the file)
if (File.Exists(dvdInfoPath))
{
File.Delete(dvdInfoPath);
}
}
catch (Exception)
{
// Do nothing.
}
if (!Directory.Exists(Path.GetDirectoryName(dvdInfoPath)))
{
Directory.CreateDirectory(Path.GetDirectoryName(dvdInfoPath));
}
// Create a new scan log.
scanLog = new StreamWriter(dvdInfoPath);
// Create a new HandBrake Instance.
instance = new HandBrakeInstance();
instance.Initialize(1);
instance.ScanProgress += this.InstanceScanProgress;
instance.ScanCompleted += this.InstanceScanCompleted;
// Start the scan on a back
this.ScanSource(sourcePath, title, configuraiton.PreviewScanCount, configuraiton);
}
///
/// Kill the scan
///
public void Stop()
{
instance.StopScan();
try
{
if (this.scanLog != null)
{
this.scanLog.Close();
this.scanLog.Dispose();
}
}
catch (Exception)
{
// Do Nothing.
}
}
///
/// Get a Preview image for the current job and preview number.
///
///
/// The job.
///
///
/// The preview.
///
///
/// The .
///
public BitmapImage GetPreview(EncodeTask job, int preview)
{
if (this.instance == null)
{
return null;
}
EncodeJob encodeJob = InteropModelCreator.GetEncodeJob(job);
BitmapImage bitmapImage = this.instance.GetPreview(encodeJob, preview);
return bitmapImage;
}
#endregion
#region Private Methods
///
/// Start a scan for a given source path and title
///
///
/// Path to the source file
///
///
/// the title number to look at
///
///
/// The preview Count.
///
///
/// The configuraiton.
///
private void ScanSource(object sourcePath, int title, int previewCount, HBConfiguration configuraiton)
{
try
{
this.logging.Clear();
string source = sourcePath.ToString().EndsWith("\\") ? string.Format("\"{0}\\\\\"", sourcePath.ToString().TrimEnd('\\'))
: "\"" + sourcePath + "\"";
currentSourceScanPath = source;
IsScanning = true;
if (this.ScanStared != null)
this.ScanStared(this, new EventArgs());
TimeSpan minDuration =
TimeSpan.FromSeconds(
configuraiton.MinScanDuration);
HandBrakeUtils.SetDvdNav(!configuraiton.IsDvdNavDisabled);
this.instance.StartScan(sourcePath.ToString(), previewCount, minDuration, title != 0 ? title : 0);
}
catch (Exception exc)
{
this.Stop();
if (this.ScanCompleted != null)
this.ScanCompleted(this, new ScanCompletedEventArgs(false, exc, "An Error has occured in ScanService.ScanSource()"));
}
}
#endregion
#region HandBrakeInstance Event Handlers
///
/// Scan Completed Event Handler
///
///
/// The sender.
///
///
/// The EventArgs.
///
private void InstanceScanCompleted(object sender, EventArgs e)
{
// Write the log file out before we start processing incase we crash.
try
{
if (this.scanLog != null)
{
this.scanLog.Flush();
}
}
catch (Exception)
{
// Do Nothing.
}
// TODO -> Might be a better place to fix this.
string path = currentSourceScanPath;
if (currentSourceScanPath.Contains("\""))
{
path = currentSourceScanPath.Trim('\"');
}
// Process into internal structures.
this.SouceData = new Source { Titles = ConvertTitles(this.instance.Titles, this.instance.FeatureTitle), ScanPath = path };
IsScanning = false;
if (this.ScanCompleted != null)
this.ScanCompleted(this, new ScanCompletedEventArgs(false, null, string.Empty));
}
///
/// Scan Progress Event Handler
///
///
/// The sender.
///
///
/// The EventArgs.
///
private void InstanceScanProgress(object sender, ScanProgressEventArgs e)
{
if (this.ScanStatusChanged != null)
{
ApplicationServices.EventArgs.ScanProgressEventArgs eventArgs =
new ApplicationServices.EventArgs.ScanProgressEventArgs
{
CurrentTitle = e.CurrentTitle,
Titles = e.Titles,
Percentage = Math.Round((decimal)e.Progress * 100, 0)
};
this.ScanStatusChanged(this, eventArgs);
}
}
///
/// Log a message
///
///
/// The sender.
///
///
/// The MessageLoggedEventArgs.
///
private void HandBrakeInstanceErrorLogged(object sender, MessageLoggedEventArgs e)
{
lock (LogLock)
{
if (this.scanLog != null)
{
this.scanLog.WriteLine(e.Message);
}
this.logging.AppendLine(e.Message);
}
}
///
/// Log a message
///
///
/// The sender.
///
///
/// The MessageLoggedEventArgs.
///
private void HandBrakeInstanceMessageLogged(object sender, MessageLoggedEventArgs e)
{
lock (LogLock)
{
if (this.scanLog != null)
{
this.scanLog.WriteLine(e.Message);
}
this.logging.AppendLine(e.Message);
}
}
///
/// Convert Interop Title objects to App Services Title object
///
///
/// The titles.
///
///
/// The feature Title.
///
///
/// The convert titles.
///
private static List ConvertTitles(IEnumerable titles, int featureTitle)
{
List titleList = new List();
foreach (Interop.SourceData.Title title in titles)
{
Title converted = new Title
{
TitleNumber = title.TitleNumber,
Duration = title.Duration,
Resolution = new Size(title.Resolution.Width, title.Resolution.Height),
AspectRatio = title.AspectRatio,
AngleCount = title.AngleCount,
ParVal = new Size(title.ParVal.Width, title.ParVal.Height),
AutoCropDimensions = title.AutoCropDimensions,
Fps = title.Framerate,
SourceName = title.Path,
MainTitle = title.TitleNumber == featureTitle
};
foreach (Interop.SourceData.Chapter chapter in title.Chapters)
{
string chapterName = !string.IsNullOrEmpty(chapter.Name) ? chapter.Name : string.Empty;
converted.Chapters.Add(new Chapter(chapter.ChapterNumber, chapterName, chapter.Duration));
}
foreach (Interop.SourceData.AudioTrack track in title.AudioTracks)
{
converted.AudioTracks.Add(new AudioTrack(track.TrackNumber, track.Language, track.LanguageCode, track.Description, string.Empty, track.SampleRate, track.Bitrate));
}
foreach (Interop.SourceData.Subtitle track in title.Subtitles)
{
SubtitleType convertedType = new SubtitleType();
switch (track.SubtitleSource)
{
case Interop.SourceData.SubtitleSource.VobSub:
convertedType = SubtitleType.VobSub;
break;
case Interop.SourceData.SubtitleSource.UTF8:
convertedType = SubtitleType.UTF8Sub;
break;
case Interop.SourceData.SubtitleSource.TX3G:
convertedType = SubtitleType.TX3G;
break;
case Interop.SourceData.SubtitleSource.SSA:
convertedType = SubtitleType.SSA;
break;
case Interop.SourceData.SubtitleSource.SRT:
convertedType = SubtitleType.SRT;
break;
case Interop.SourceData.SubtitleSource.CC608:
convertedType = SubtitleType.CC;
break;
case Interop.SourceData.SubtitleSource.CC708:
convertedType = SubtitleType.CC;
break;
case Interop.SourceData.SubtitleSource.PGS:
convertedType = SubtitleType.PGS;
break;
}
converted.Subtitles.Add(new Subtitle(track.TrackNumber, track.Language, track.LanguageCode, convertedType));
}
titleList.Add(converted);
}
return titleList;
}
#endregion
}
}