namespace HandBrake.Interop
{
using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Runtime.InteropServices;
using System.Text;
using System.Timers;
using System.Threading;
using System.Windows.Media.Imaging;
using HandBrake.SourceData;
using HandBrake.Interop;
///
/// A wrapper for a HandBrake instance.
///
public class HandBrakeInstance : IDisposable
{
///
/// The modulus for picture size when auto-sizing dimensions.
///
private const int PictureAutoSizeModulus = 2;
///
/// Modulus used for picture size when using non-anamorphic.
///
private const int NonAnamorphicAutoSizeModulus = 16;
///
/// The number of MS between status polls when scanning.
///
private const double ScanPollIntervalMs = 200;
///
/// The number of MS between status polls when encoding.
///
private const double EncodePollIntervalMs = 200;
///
/// X264 options to add for a turbo first pass.
///
private const string TurboX264Opts = "ref=1:subme=2:me=dia:analyse=none:trellis=0:no-fast-pskip=0:8x8dct=0:weightb=0";
///
/// Default value to give as a minimum duration when scanning.
///
private const ulong DefaultMinDuration = 900000;
///
/// The native handle to the HandBrake instance.
///
private IntPtr hbHandle;
///
/// The number of previews created during scan.
///
private int previewCount;
///
/// The timer to poll for scan status.
///
private System.Timers.Timer scanPollTimer;
///
/// The timer to poll for encode status.
///
private System.Timers.Timer encodePollTimer;
///
/// The list of original titles in native structure form.
///
private List originalTitles;
///
/// The list of titles on this instance.
///
private List titles;
///
/// The current encode job for this instance.
///
private EncodeJob currentJob;
///
/// True if the current job is scanning for subtitles.
///
private bool subtitleScan;
///
/// The index of the default title.
///
private int featureTitle;
///
/// A list of native memory locations allocated by this instance.
///
private List encodeAllocatedMemory;
///
/// Fires for progress updates when scanning.
///
public event EventHandler ScanProgress;
///
/// Fires when a scan has completed.
///
public event EventHandler ScanCompleted;
///
/// Fires for progress updates when encoding.
///
public event EventHandler EncodeProgress;
///
/// Fires when an encode has completed.
///
public event EventHandler EncodeCompleted;
///
/// Destructor.
///
~HandBrakeInstance()
{
this.Dispose(false);
}
///
/// The list of titles on this instance.
///
public List Titles
{
get
{
return this.titles;
}
}
///
/// The number of previews created during scan.
///
public int PreviewCount
{
get
{
return this.previewCount;
}
}
///
/// Gets the index of the default title.
///
public int FeatureTitle
{
get
{
return this.featureTitle;
}
}
///
/// Initializes this instance.
///
///
public void Initialize(int verbosity)
{
HandBrakeUtils.RegisterLogger();
this.hbHandle = HBFunctions.hb_init(verbosity, update_check: 0);
}
///
/// Starts scanning the given path.
///
/// The path to the video to scan.
/// The number of preview images to make.
public void StartScan(string path, int previewCount)
{
this.StartScan(path, previewCount, 0);
}
///
/// Starts a scan of the given path.
///
/// The path of the video to scan.
/// The number of previews to make on each title.
/// The title index to scan (1-based, 0 for all titles).
public void StartScan(string path, int previewCount, int titleIndex)
{
this.previewCount = previewCount;
HBFunctions.hb_scan(this.hbHandle, path, titleIndex, previewCount, 1, DefaultMinDuration);
this.scanPollTimer = new System.Timers.Timer();
this.scanPollTimer.Interval = ScanPollIntervalMs;
// Lambda notation used to make sure we can view any JIT exceptions the method throws
this.scanPollTimer.Elapsed += (o, e) =>
{
this.PollScanProgress();
};
this.scanPollTimer.Start();
}
///
/// Stops an ongoing scan.
///
public void StopScan()
{
HBFunctions.hb_scan_stop(this.hbHandle);
}
///
/// Gets an image for the given job and preview
///
///
/// Only incorporates sizing and aspect ratio into preview image.
///
/// The encode job to preview.
/// The index of the preview to get (0-based).
/// An image with the requested preview.
public BitmapImage GetPreview(EncodeJob job, int previewNumber)
{
hb_title_s title = this.GetOriginalTitle(job.Title);
hb_job_s nativeJob = InteropUtilities.ReadStructure(title.job);
List allocatedMemory = this.ApplyJob(ref nativeJob, job);
// There are some problems with getting previews with deinterlacing. Disabling for now.
nativeJob.deinterlace = 0;
// Create a new job pointer from our modified job object
IntPtr newJob = Marshal.AllocHGlobal(Marshal.SizeOf(typeof(hb_job_s)));
Marshal.StructureToPtr(nativeJob, newJob, false);
allocatedMemory.Add(newJob);
int outputWidth = nativeJob.width;
int outputHeight = nativeJob.height;
int imageBufferSize = outputWidth * outputHeight * 4;
IntPtr nativeBuffer = Marshal.AllocHGlobal(imageBufferSize);
allocatedMemory.Add(nativeBuffer);
HBFunctions.hb_set_job(this.hbHandle, job.Title, ref nativeJob);
HBFunctions.hb_get_preview_by_index(this.hbHandle, job.Title, previewNumber, nativeBuffer);
// Copy the filled image buffer to a managed array.
byte[] managedBuffer = new byte[imageBufferSize];
Marshal.Copy(nativeBuffer, managedBuffer, 0, imageBufferSize);
InteropUtilities.FreeMemory(allocatedMemory);
var bitmap = new System.Drawing.Bitmap(outputWidth, outputHeight);
System.Drawing.Imaging.BitmapData bitmapData = bitmap.LockBits(new System.Drawing.Rectangle(0, 0, outputWidth, outputHeight), System.Drawing.Imaging.ImageLockMode.WriteOnly, System.Drawing.Imaging.PixelFormat.Format32bppRgb);
IntPtr ptr = bitmapData.Scan0;
for (int i = 0; i < nativeJob.height; i++)
{
Marshal.Copy(managedBuffer, i * nativeJob.width * 4, ptr, nativeJob.width * 4);
ptr = IntPtr.Add(ptr, bitmapData.Stride);
}
bitmap.UnlockBits(bitmapData);
using (var memoryStream = new MemoryStream())
{
try
{
bitmap.Save(memoryStream, System.Drawing.Imaging.ImageFormat.Bmp);
}
finally
{
bitmap.Dispose();
}
var wpfBitmap = new BitmapImage();
wpfBitmap.BeginInit();
wpfBitmap.CacheOption = BitmapCacheOption.OnLoad;
wpfBitmap.StreamSource = memoryStream;
wpfBitmap.EndInit();
wpfBitmap.Freeze();
return wpfBitmap;
}
}
///
/// Calculates the video bitrate for the given job and target size.
///
/// The encode job.
/// The target size in MB.
/// The currently selected encode length. Used in preview
/// for calculating bitrate when the target size would be wrong.
/// The video bitrate in kbps.
public int CalculateBitrate(EncodeJob job, int sizeMB, double overallSelectedLengthSeconds = 0)
{
long availableBytes = ((long) sizeMB) * 1024 * 1024;
EncodingProfile profile = job.EncodingProfile;
Title title = this.GetTitle(job.Title);
double lengthSeconds = overallSelectedLengthSeconds > 0 ? overallSelectedLengthSeconds : HandBrakeUtils.GetJobLengthSeconds(job, title);
lengthSeconds += 1.5;
double outputFramerate;
if (profile.Framerate == 0)
{
outputFramerate = title.Framerate;
}
else
{
// Not sure what to do for VFR here hb_calc_bitrate never handled it...
// just use the peak for now.
outputFramerate = profile.Framerate;
}
long frames = (long)(lengthSeconds * outputFramerate);
availableBytes -= frames * HandBrakeUtils.ContainerOverheadPerFrame;
List> outputTrackList = this.GetOutputTracks(job, title);
availableBytes -= HandBrakeUtils.GetAudioSize(job, lengthSeconds, title, outputTrackList);
if (availableBytes < 0)
{
return 0;
}
// Video bitrate is in kilobits per second, or where 1 kbps is 1000 bits per second.
// So 1 kbps is 125 bytes per second.
return (int)(availableBytes / (125 * lengthSeconds));
}
///
/// Gives estimated file size (in MB) of the given job and video bitrate.
///
/// The encode job.
/// The video bitrate to be used (kbps).
/// The estimated file size (in MB) of the given job and video bitrate.
public double CalculateFileSize(EncodeJob job, int videoBitrate)
{
long totalBytes = 0;
EncodingProfile profile = job.EncodingProfile;
Title title = this.GetTitle(job.Title);
double lengthSeconds = HandBrakeUtils.GetJobLengthSeconds(job, title);
lengthSeconds += 1.5;
double outputFramerate;
if (profile.Framerate == 0)
{
outputFramerate = title.Framerate;
}
else
{
// Not sure what to do for VFR here hb_calc_bitrate never handled it...
// just use the peak for now.
outputFramerate = profile.Framerate;
}
long frames = (long)(lengthSeconds * outputFramerate);
totalBytes += (long)(lengthSeconds * videoBitrate * 125);
totalBytes += frames * HandBrakeUtils.ContainerOverheadPerFrame;
List> outputTrackList = this.GetOutputTracks(job, title);
totalBytes += HandBrakeUtils.GetAudioSize(job, lengthSeconds, title, outputTrackList);
return (double)totalBytes / 1024 / 1024;
}
///
/// Starts an encode with the given job.
///
/// The job to start.
public void StartEncode(EncodeJob jobToStart)
{
this.StartEncode(jobToStart, false, 0, 0, 0);
}
///
/// Starts an encode with the given job.
///
/// The job to start.
/// True if this is a preview encode.
/// The preview number to start the encode at (0-based).
/// The number of seconds in the preview.
/// The currently selected encode length. Used in preview
/// for calculating bitrate when the target size would be wrong.
public void StartEncode(EncodeJob job, bool preview, int previewNumber, int previewSeconds, double overallSelectedLengthSeconds)
{
this.currentJob = job;
hb_job_s nativeJob = InteropUtilities.ReadStructure(this.GetOriginalTitle(job.Title).job);
this.encodeAllocatedMemory = this.ApplyJob(ref nativeJob, job, preview, previewNumber, previewSeconds, overallSelectedLengthSeconds);
if (!preview && job.EncodingProfile.IncludeChapterMarkers)
{
Title title = this.GetTitle(job.Title);
int numChapters = title.Chapters.Count;
if (job.UseDefaultChapterNames)
{
for (int i = 0; i < numChapters; i++)
{
HBFunctions.hb_set_chapter_name(this.hbHandle, job.Title, i + 1, "Chapter " + (i + 1));
}
}
else
{
for (int i = 0; i < numChapters; i++)
{
HBFunctions.hb_set_chapter_name(this.hbHandle, job.Title, i + 1, job.CustomChapterNames[i]);
}
}
}
this.subtitleScan = false;
if (job.Subtitles.SourceSubtitles != null)
{
foreach (SourceSubtitle subtitle in job.Subtitles.SourceSubtitles)
{
if (subtitle.TrackNumber == 0)
{
this.subtitleScan = true;
break;
}
}
}
string x264Options = job.EncodingProfile.X264Options ?? string.Empty;
IntPtr originalX264Options = Marshal.StringToHGlobalAnsi(x264Options);
this.encodeAllocatedMemory.Add(originalX264Options);
if (this.subtitleScan)
{
// If we need to scan subtitles, enqueue a pre-processing job to do that.
nativeJob.pass = -1;
nativeJob.indepth_scan = 1;
nativeJob.advanced_opts = IntPtr.Zero;
HBFunctions.hb_add(this.hbHandle, ref nativeJob);
}
nativeJob.indepth_scan = 0;
if (job.EncodingProfile.TwoPass)
{
// First pass. Apply turbo options if needed.
nativeJob.pass = 1;
string firstPassAdvancedOptions = x264Options;
if (job.EncodingProfile.TurboFirstPass)
{
if (firstPassAdvancedOptions == string.Empty)
{
firstPassAdvancedOptions = TurboX264Opts;
}
else
{
firstPassAdvancedOptions += ":" + TurboX264Opts;
}
}
nativeJob.advanced_opts = Marshal.StringToHGlobalAnsi(firstPassAdvancedOptions);
this.encodeAllocatedMemory.Add(nativeJob.advanced_opts);
HBFunctions.hb_add(this.hbHandle, ref nativeJob);
// Second pass. Apply normal options.
nativeJob.pass = 2;
nativeJob.advanced_opts = originalX264Options;
HBFunctions.hb_add(this.hbHandle, ref nativeJob);
}
else
{
// One pass job.
nativeJob.pass = 0;
nativeJob.advanced_opts = originalX264Options;
HBFunctions.hb_add(this.hbHandle, ref nativeJob);
}
HBFunctions.hb_start(this.hbHandle);
this.encodePollTimer = new System.Timers.Timer();
this.encodePollTimer.Interval = EncodePollIntervalMs;
this.encodePollTimer.Elapsed += (o, e) =>
{
this.PollEncodeProgress();
};
this.encodePollTimer.Start();
}
///
/// Pauses the current encode.
///
public void PauseEncode()
{
HBFunctions.hb_pause(this.hbHandle);
}
///
/// Resumes a paused encode.
///
public void ResumeEncode()
{
HBFunctions.hb_resume(this.hbHandle);
}
///
/// Stops the current encode.
///
public void StopEncode()
{
HBFunctions.hb_stop(this.hbHandle);
// Also remove all jobs from the queue (in case we stopped a 2-pass encode)
var currentJobs = new List();
int jobs = HBFunctions.hb_count(this.hbHandle);
for (int i = 0; i < jobs; i++)
{
currentJobs.Add(HBFunctions.hb_job(this.hbHandle, 0));
}
foreach (IntPtr job in currentJobs)
{
HBFunctions.hb_rem(this.hbHandle, job);
}
}
///
/// Gets the final size for a given encode job.
///
/// The encode job to use.
/// The storage width.
/// The storage height.
/// The pixel aspect X number.
/// The pixel aspect Y number.
public void GetSize(EncodeJob job, out int width, out int height, out int parWidth, out int parHeight)
{
if (job.EncodingProfile.Anamorphic == Anamorphic.None)
{
Title title = this.GetTitle(job.Title);
Size storageDimensions = CalculateNonAnamorphicOutput(job.EncodingProfile, title);
width = storageDimensions.Width;
height = storageDimensions.Height;
parWidth = 1;
parHeight = 1;
return;
}
var nativeJob = InteropUtilities.ReadStructure(this.GetOriginalTitle(job.Title).job);
List allocatedMemory = this.ApplyJob(ref nativeJob, job);
int refWidth = 0;
int refHeight = 0;
int refParWidth = 0;
int refParHeight = 0;
HBFunctions.hb_set_job(this.hbHandle, job.Title, ref nativeJob);
HBFunctions.hb_set_anamorphic_size_by_index(this.hbHandle, job.Title, ref refWidth, ref refHeight, ref refParWidth, ref refParHeight);
InteropUtilities.FreeMemory(allocatedMemory);
width = refWidth;
height = refHeight;
parWidth = refParWidth;
parHeight = refParHeight;
}
///
/// Frees any resources associated with this object.
///
public void Dispose()
{
this.Dispose(true);
GC.SuppressFinalize(this);
}
///
/// Call before app shutdown. Performs global cleanup.
///
public static void DisposeGlobal()
{
HBFunctions.hb_global_close();
}
///
/// Frees any resources associated with this object.
///
/// True if managed objects as well as unmanaged should be disposed.
protected virtual void Dispose(bool disposing)
{
if (disposing)
{
// Free other state (managed objects).
}
// Free unmanaged objects.
IntPtr handlePtr = Marshal.AllocHGlobal(Marshal.SizeOf(typeof(IntPtr)));
Marshal.WriteIntPtr(handlePtr, this.hbHandle);
HBFunctions.hb_close(handlePtr);
Marshal.FreeHGlobal(handlePtr);
}
///
/// Checks the status of the ongoing scan.
///
private void PollScanProgress()
{
var state = new hb_state_s();
HBFunctions.hb_get_state(this.hbHandle, ref state);
if (state.state == NativeConstants.HB_STATE_SCANNING)
{
if (this.ScanProgress != null)
{
int currentTitle = state.param.scanning.title_cur;
int totalTitles = state.param.scanning.title_count;
this.ScanProgress(this, new ScanProgressEventArgs { CurrentTitle = currentTitle, Titles = totalTitles });
}
}
else if (state.state == NativeConstants.HB_STATE_SCANDONE)
{
this.titles = new List();
IntPtr listPtr = HBFunctions.hb_get_titles(this.hbHandle);
this.originalTitles = InteropUtilities.ConvertList(listPtr);
foreach (hb_title_s title in this.originalTitles)
{
var newTitle = this.ConvertTitle(title);
this.titles.Add(newTitle);
}
if (this.originalTitles.Count > 0)
{
var nativeJob = InteropUtilities.ReadStructure(this.originalTitles[0].job);
this.featureTitle = nativeJob.feature;
}
else
{
this.featureTitle = 0;
}
this.scanPollTimer.Stop();
if (this.ScanCompleted != null)
{
this.ScanCompleted(this, new EventArgs());
}
}
}
///
/// Checks the status of the ongoing encode.
///
private void PollEncodeProgress()
{
hb_state_s state = new hb_state_s();
HBFunctions.hb_get_state(this.hbHandle, ref state);
if (state.state == NativeConstants.HB_STATE_WORKING)
{
if (this.EncodeProgress != null)
{
int pass = 1;
int rawJobNumber = state.param.working.job_cur;
if (this.currentJob.EncodingProfile.TwoPass)
{
if (this.subtitleScan)
{
switch (rawJobNumber)
{
case 1:
pass = -1;
break;
case 2:
pass = 1;
break;
case 3:
pass = 2;
break;
default:
break;
}
}
else
{
switch (rawJobNumber)
{
case 1:
pass = 1;
break;
case 2:
pass = 2;
break;
default:
break;
}
}
}
else
{
if (this.subtitleScan)
{
switch (rawJobNumber)
{
case 1:
pass = -1;
break;
case 2:
pass = 1;
break;
default:
break;
}
}
else
{
pass = 1;
}
}
var progressEventArgs = new EncodeProgressEventArgs
{
FractionComplete = state.param.working.progress,
CurrentFrameRate = state.param.working.rate_cur,
AverageFrameRate = state.param.working.rate_avg,
EstimatedTimeLeft = new TimeSpan(state.param.working.hours, state.param.working.minutes, state.param.working.seconds),
Pass = pass
};
this.EncodeProgress(this, progressEventArgs);
}
}
else if (state.state == NativeConstants.HB_STATE_MUXING)
{
//System.Diagnostics.Debug.WriteLine("Muxing...");
}
else if (state.state == NativeConstants.HB_STATE_WORKDONE)
{
InteropUtilities.FreeMemory(this.encodeAllocatedMemory);
this.encodePollTimer.Stop();
if (this.EncodeCompleted != null)
{
this.EncodeCompleted(this, new EncodeCompletedEventArgs { Error = state.param.workdone.error > 0 });
}
}
}
///
/// Applies the encoding job to the native memory structure and returns a list of memory
/// locations allocated during this.
///
/// The native structure to apply to job info to.
/// The job info to apply.
/// The list of memory locations allocated for the job.
private List ApplyJob(ref hb_job_s nativeJob, EncodeJob job)
{
return this.ApplyJob(ref nativeJob, job, false, 0, 0, 0);
}
///
/// Applies the encoding job to the native memory structure and returns a list of memory
/// locations allocated during this.
///
/// The native structure to apply to job info to.
/// The job info to apply.
/// True if this is a preview encode.
/// The preview number (0-based) to encode.
/// The number of seconds in the preview.
/// The currently selected encode length. Used in preview
/// for calculating bitrate when the target size would be wrong.
/// The list of memory locations allocated for the job.
private List ApplyJob(ref hb_job_s nativeJob, EncodeJob job, bool preview, int previewNumber, int previewSeconds, double overallSelectedLengthSeconds)
{
var allocatedMemory = new List();
Title title = this.GetTitle(job.Title);
hb_title_s originalTitle = this.GetOriginalTitle(job.Title);
EncodingProfile profile = job.EncodingProfile;
if (preview)
{
nativeJob.start_at_preview = previewNumber + 1;
nativeJob.seek_points = this.previewCount;
// There are 90,000 PTS per second.
nativeJob.pts_to_stop = previewSeconds * 90000;
}
else
{
switch (job.RangeType)
{
case VideoRangeType.Chapters:
if (job.ChapterStart > 0 && job.ChapterEnd > 0)
{
nativeJob.chapter_start = job.ChapterStart;
nativeJob.chapter_end = job.ChapterEnd;
}
else
{
nativeJob.chapter_start = 1;
nativeJob.chapter_end = title.Chapters.Count;
}
break;
case VideoRangeType.Seconds:
if (job.SecondsStart < 0 || job.SecondsEnd < 0 || job.SecondsStart >= job.SecondsEnd)
{
throw new ArgumentException("Seconds range " + job.SecondsStart + "-" + job.SecondsEnd + " is invalid.", "job");
}
// For some reason "pts_to_stop" actually means the number of pts to stop AFTER the start point.
nativeJob.pts_to_start = (int)(job.SecondsStart * 90000);
nativeJob.pts_to_stop = (int)((job.SecondsEnd - job.SecondsStart) * 90000);
break;
case VideoRangeType.Frames:
if (job.FramesStart < 0 || job.FramesEnd < 0 || job.FramesStart >= job.FramesEnd)
{
throw new ArgumentException("Frames range " + job.FramesStart + "-" + job.FramesEnd + " is invalid.", "job");
}
// "frame_to_stop" actually means the number of frames total to encode AFTER the start point.
nativeJob.frame_to_start = job.FramesStart;
nativeJob.frame_to_stop = job.FramesEnd - job.FramesStart;
break;
}
}
nativeJob.chapter_markers = profile.IncludeChapterMarkers ? 1 : 0;
Cropping crop;
if (profile.CustomCropping)
{
crop = profile.Cropping;
}
else
{
crop = title.AutoCropDimensions;
}
nativeJob.crop[0] = crop.Top;
nativeJob.crop[1] = crop.Bottom;
nativeJob.crop[2] = crop.Left;
nativeJob.crop[3] = crop.Right;
var filterList = new List();
if (profile.Deinterlace != Deinterlace.Off)
{
nativeJob.deinterlace = 1;
string settings = null;
switch (profile.Deinterlace)
{
case Deinterlace.Fast:
settings = "-1";
break;
case Deinterlace.Slow:
settings = "2";
break;
case Deinterlace.Slower:
settings = "0";
break;
case Deinterlace.Custom:
settings = profile.CustomDeinterlace;
break;
default:
break;
}
this.AddFilter(filterList, NativeConstants.HB_FILTER_DEINTERLACE, settings, allocatedMemory);
}
else
{
nativeJob.deinterlace = 0;
}
if (profile.Detelecine != Detelecine.Off)
{
string settings = null;
if (profile.Detelecine == Detelecine.Custom)
{
settings = profile.CustomDetelecine;
}
this.AddFilter(filterList, NativeConstants.HB_FILTER_DETELECINE, settings, allocatedMemory);
}
if (profile.Decomb != Decomb.Off)
{
string settings = null;
if (profile.Decomb == Decomb.Custom)
{
settings = profile.CustomDecomb;
}
this.AddFilter(filterList, NativeConstants.HB_FILTER_DECOMB, settings, allocatedMemory);
}
if (profile.Deblock > 0)
{
this.AddFilter(filterList, NativeConstants.HB_FILTER_DEBLOCK, profile.Deblock.ToString(), allocatedMemory);
}
if (profile.Denoise != Denoise.Off)
{
string settings = null;
switch (profile.Denoise)
{
case Denoise.Weak:
settings = "2:1:2:3";
break;
case Denoise.Medium:
settings = "3:2:2:3";
break;
case Denoise.Strong:
settings = "7:7:5:5";
break;
case Denoise.Custom:
settings = profile.CustomDenoise;
break;
default:
break;
}
this.AddFilter(filterList, NativeConstants.HB_FILTER_DENOISE, settings, allocatedMemory);
}
NativeList filterListNative = InteropUtilities.CreateIntPtrList(filterList);
nativeJob.filters = filterListNative.ListPtr;
allocatedMemory.AddRange(filterListNative.AllocatedMemory);
int width = profile.Width;
int height = profile.Height;
int cropHorizontal = crop.Left + crop.Right;
int cropVertical = crop.Top + crop.Bottom;
if (width == 0)
{
width = title.Resolution.Width - cropHorizontal;
}
if (profile.MaxWidth > 0 && width > profile.MaxWidth)
{
width = profile.MaxWidth;
}
if (height == 0)
{
height = title.Resolution.Height - cropVertical;
}
if (profile.MaxHeight > 0 && height > profile.MaxHeight)
{
height = profile.MaxHeight;
}
nativeJob.grayscale = profile.Grayscale ? 1 : 0;
switch (profile.Anamorphic)
{
case Anamorphic.None:
nativeJob.anamorphic.mode = 0;
Size outputSize = CalculateNonAnamorphicOutput(profile, title);
width = outputSize.Width;
height = outputSize.Height;
nativeJob.anamorphic.keep_display_aspect = profile.KeepDisplayAspect ? 1 : 0;
break;
case Anamorphic.Strict:
nativeJob.anamorphic.mode = 1;
break;
case Anamorphic.Loose:
nativeJob.anamorphic.mode = 2;
break;
case Anamorphic.Custom:
nativeJob.anamorphic.mode = 3;
nativeJob.modulus = profile.Modulus;
if (profile.UseDisplayWidth)
{
if (profile.KeepDisplayAspect)
{
int cropWidth = title.Resolution.Width - cropHorizontal;
int cropHeight = title.Resolution.Height - cropVertical;
double displayAspect = ((double)(cropWidth * title.ParVal.Width)) / (cropHeight * title.ParVal.Height);
int displayWidth = profile.DisplayWidth;
if (profile.Height > 0)
{
displayWidth = (int)((double)profile.Height * displayAspect);
}
else if (displayWidth > 0)
{
height = (int)((double)displayWidth / displayAspect);
}
else
{
displayWidth = (int)((double)cropHeight * displayAspect);
}
nativeJob.anamorphic.dar_width = displayWidth;
nativeJob.anamorphic.dar_height = height;
nativeJob.anamorphic.keep_display_aspect = 1;
}
nativeJob.anamorphic.dar_width = profile.DisplayWidth;
nativeJob.anamorphic.dar_height = height;
nativeJob.anamorphic.keep_display_aspect = profile.KeepDisplayAspect ? 1 : 0;
}
else
{
nativeJob.anamorphic.par_width = profile.PixelAspectX;
nativeJob.anamorphic.par_height = profile.PixelAspectY;
nativeJob.anamorphic.keep_display_aspect = 0;
}
break;
default:
break;
}
nativeJob.width = width;
nativeJob.height = height;
nativeJob.maxWidth = profile.MaxWidth;
nativeJob.maxHeight = profile.MaxHeight;
switch (profile.VideoEncoder)
{
case VideoEncoder.X264:
nativeJob.vcodec = NativeConstants.HB_VCODEC_X264;
break;
case VideoEncoder.Theora:
nativeJob.vcodec = NativeConstants.HB_VCODEC_THEORA;
break;
case VideoEncoder.FFMpeg:
nativeJob.vcodec = NativeConstants.HB_VCODEC_FFMPEG;
break;
default:
break;
}
if (profile.Framerate == 0)
{
nativeJob.cfr = 0;
}
else
{
if (profile.PeakFramerate)
{
nativeJob.cfr = 2;
}
else
{
nativeJob.cfr = 1;
}
nativeJob.vrate = 27000000;
nativeJob.vrate_base = Converters.FramerateToVrate(profile.Framerate);
}
// vfr
// areBframes
// color_matrix
List titleAudio = InteropUtilities.ConvertList(originalTitle.list_audio);
var audioList = new List();
int numTracks = 0;
List> outputTrackList = this.GetOutputTracks(job, title);
foreach (Tuple outputTrack in outputTrackList)
{
audioList.Add(ConvertAudioBack(outputTrack.Item1, titleAudio[outputTrack.Item2 - 1], outputTrack.Item2, numTracks++, allocatedMemory));
}
NativeList nativeAudioList = InteropUtilities.ConvertListBack(audioList);
nativeJob.list_audio = nativeAudioList.ListPtr;
allocatedMemory.AddRange(nativeAudioList.AllocatedMemory);
// Create a new empty list
int totalSubtitles = 0;
if (job.Subtitles != null)
{
if (job.Subtitles.SourceSubtitles != null)
{
totalSubtitles += job.Subtitles.SourceSubtitles.Count;
}
if (job.Subtitles.SrtSubtitles != null)
{
totalSubtitles += job.Subtitles.SrtSubtitles.Count;
}
}
NativeList nativeSubtitleList = InteropUtilities.CreateNativeList(totalSubtitles + 2);
nativeJob.list_subtitle = nativeSubtitleList.ListPtr;
allocatedMemory.AddRange(nativeSubtitleList.AllocatedMemory);
if (job.Subtitles != null)
{
if (job.Subtitles.SourceSubtitles != null && job.Subtitles.SourceSubtitles.Count > 0)
{
List titleSubtitles = InteropUtilities.ConvertList(originalTitle.list_subtitle);
foreach (SourceSubtitle sourceSubtitle in job.Subtitles.SourceSubtitles)
{
if (sourceSubtitle.TrackNumber == 0)
{
// Use subtitle search.
nativeJob.select_subtitle_config.force = sourceSubtitle.Forced ? 1 : 0;
nativeJob.select_subtitle_config.default_track = sourceSubtitle.Default ? 1 : 0;
if (!sourceSubtitle.BurnedIn)
{
nativeJob.select_subtitle_config.dest = hb_subtitle_config_s_subdest.PASSTHRUSUB;
}
nativeJob.indepth_scan = 1;
}
else
{
// Use specified subtitle.
hb_subtitle_s nativeSubtitle = titleSubtitles[sourceSubtitle.TrackNumber - 1];
var subtitleConfig = new hb_subtitle_config_s();
subtitleConfig.force = sourceSubtitle.Forced ? 1 : 0;
subtitleConfig.default_track = sourceSubtitle.Default ? 1 : 0;
bool supportsBurn = nativeSubtitle.source == hb_subtitle_s_subsource.VOBSUB || nativeSubtitle.source == hb_subtitle_s_subsource.SSASUB;
if (supportsBurn && sourceSubtitle.BurnedIn)
{
subtitleConfig.dest = hb_subtitle_config_s_subdest.RENDERSUB;
}
else
{
subtitleConfig.dest = hb_subtitle_config_s_subdest.PASSTHRUSUB;
}
int subtitleAddSucceded = HBFunctions.hb_subtitle_add(ref nativeJob, ref subtitleConfig, sourceSubtitle.TrackNumber - 1);
if (subtitleAddSucceded == 0)
{
System.Diagnostics.Debug.WriteLine("Subtitle add failed");
}
}
}
}
if (job.Subtitles.SrtSubtitles != null)
{
foreach (SrtSubtitle srtSubtitle in job.Subtitles.SrtSubtitles)
{
var subtitleConfig = new hb_subtitle_config_s();
subtitleConfig.src_codeset = srtSubtitle.CharacterCode;
subtitleConfig.src_filename = srtSubtitle.FileName;
subtitleConfig.offset = srtSubtitle.Offset;
//subtitleConfig.dest = hb_subtitle_config_s_subdest.PASSTHRUSUB;
subtitleConfig.default_track = srtSubtitle.Default ? 1 : 0;
int srtAddSucceded = HBFunctions.hb_srt_add(ref nativeJob, ref subtitleConfig, srtSubtitle.LanguageCode);
if (srtAddSucceded == 0)
{
System.Diagnostics.Debug.WriteLine("SRT add failed");
}
}
}
}
if (profile.OutputFormat == OutputFormat.Mp4)
{
nativeJob.mux = NativeConstants.HB_MUX_MP4;
}
else
{
nativeJob.mux = NativeConstants.HB_MUX_MKV;
}
nativeJob.file = job.OutputPath;
nativeJob.largeFileSize = profile.LargeFile ? 1 : 0;
nativeJob.mp4_optimize = profile.Optimize ? 1 : 0;
nativeJob.ipod_atom = profile.IPod5GSupport ? 1 : 0;
if (title.AngleCount > 1)
{
nativeJob.angle = job.Angle;
}
switch (profile.VideoEncodeRateType)
{
case VideoEncodeRateType.ConstantQuality:
nativeJob.vquality = (float)profile.Quality;
nativeJob.vbitrate = 0;
break;
case VideoEncodeRateType.AverageBitrate:
nativeJob.vquality = -1;
nativeJob.vbitrate = profile.VideoBitrate;
break;
case VideoEncodeRateType.TargetSize:
nativeJob.vquality = -1;
nativeJob.vbitrate = this.CalculateBitrate(job, profile.TargetSize, overallSelectedLengthSeconds);
break;
default:
break;
}
// frames_to_skip
return allocatedMemory;
}
///
/// Gets a list of encodings and target track indices (1-based).
///
/// The encode job
/// The title the job is meant to encode.
/// A list of encodings and target track indices (1-based).
private List> GetOutputTracks(EncodeJob job, Title title)
{
var list = new List>();
foreach (AudioEncoding encoding in job.EncodingProfile.AudioEncodings)
{
if (encoding.InputNumber == 0)
{
// Add this encoding for all chosen tracks
foreach (int chosenTrack in job.ChosenAudioTracks)
{
list.Add(new Tuple(encoding, chosenTrack));
}
}
else if (encoding.InputNumber <= job.ChosenAudioTracks.Count)
{
// Add this encoding for the specified track, if it exists
int trackNumber = job.ChosenAudioTracks[encoding.InputNumber - 1];
// In normal cases we'll never have a chosen audio track that doesn't exist but when batch encoding
// we just choose the first audio track without checking if it exists.
if (trackNumber <= title.AudioTracks.Count)
{
list.Add(new Tuple(encoding, trackNumber));
}
}
}
return list;
}
///
/// Adds a filter to the given filter list.
///
/// The filter list to add to.
/// The type of filter.
/// Settings for the filter.
/// The list of allocated memory.
private void AddFilter(List filterList, int filterType, string settings, List allocatedMemory)
{
IntPtr settingsNativeString = Marshal.StringToHGlobalAnsi(settings);
filterList.Add(HBFunctions.hb_get_filter_object(filterType, settingsNativeString));
allocatedMemory.Add(settingsNativeString);
}
///
/// Gets the title, given the 1-based index.
///
/// The index of the title (1-based).
/// The requested Title.
private Title GetTitle(int titleIndex)
{
return this.Titles.SingleOrDefault(title => title.TitleNumber == titleIndex);
}
///
/// Gets the native title object from the title index.
///
/// The index of the title (1-based).
/// Gets the native title object for the given index.
private hb_title_s GetOriginalTitle(int titleIndex)
{
List matchingTitles = this.originalTitles.Where(title => title.index == titleIndex).ToList();
if (matchingTitles.Count == 0)
{
throw new ArgumentException("Could not find specified title.");
}
if (matchingTitles.Count > 1)
{
throw new ArgumentException("Multiple titles matched.");
}
return matchingTitles[0];
}
///
/// Applies an audio encoding to a native audio encoding base structure.
///
/// The encoding to apply.
/// The base native structure.
///
///
/// The collection of allocated memory.
/// The resulting native audio structure.
private hb_audio_s ConvertAudioBack(AudioEncoding encoding, hb_audio_s baseStruct, int track, int outputTrack, List allocatedMemory)
{
hb_audio_s nativeAudio = baseStruct;
//nativeAudio.config.input.track = track;
nativeAudio.config.output.track = outputTrack;
if (encoding.Encoder == AudioEncoder.Passthrough)
{
// If we've been given a general "Passthrough" codec, see if it's valid for this input track.
if ((baseStruct.config.input.codec & NativeConstants.HB_ACODEC_PASS_MASK) > 0)
{
// We can do passthrough for this input.
nativeAudio.config.output.codec = nativeAudio.config.input.codec | NativeConstants.HB_ACODEC_PASS_FLAG;
}
else
{
// We can't do passthrough for this input. Set it to a DTS passthrough, which will cause the track to be dropped.
nativeAudio.config.output.codec = NativeConstants.HB_ACODEC_DCA_PASS;
}
}
else
{
nativeAudio.config.output.codec = Converters.AudioEncoderToNative(encoding.Encoder);
}
if (encoding.Encoder != AudioEncoder.Passthrough && encoding.Encoder != AudioEncoder.Ac3Passthrough && encoding.Encoder != AudioEncoder.DtsPassthrough)
{
nativeAudio.config.output.bitrate = encoding.Bitrate;
nativeAudio.config.output.dynamic_range_compression = 0.0;
nativeAudio.config.output.gain = encoding.Gain;
if (encoding.Mixdown == Mixdown.Auto)
{
nativeAudio.config.output.mixdown = HBFunctions.hb_get_default_mixdown(nativeAudio.config.output.codec, nativeAudio.config.input.channel_layout);
}
else
{
nativeAudio.config.output.mixdown = Converters.MixdownToNative(encoding.Mixdown);
}
if (encoding.SampleRateRaw == 0)
{
nativeAudio.config.output.samplerate = nativeAudio.config.input.samplerate;
}
else
{
nativeAudio.config.output.samplerate = encoding.SampleRateRaw;
}
}
if (!string.IsNullOrEmpty(encoding.Name))
{
IntPtr encodingNamePtr = Marshal.StringToHGlobalAnsi(encoding.Name);
nativeAudio.config.output.name = encodingNamePtr;
allocatedMemory.Add(encodingNamePtr);
}
nativeAudio.padding = new byte[MarshalingConstants.AudioPaddingBytes];
return nativeAudio;
}
///
/// Converts a native title to a Title object.
///
/// The native title structure.
/// The managed Title object.
private Title ConvertTitle(hb_title_s title)
{
var newTitle = new Title
{
TitleNumber = title.index,
Resolution = new Size(title.width, title.height),
ParVal = new Size(title.pixel_aspect_width, title.pixel_aspect_height),
Duration = TimeSpan.FromSeconds(title.duration / 90000),
AutoCropDimensions = new Cropping
{
Top = title.crop[0],
Bottom = title.crop[1],
Left = title.crop[2],
Right = title.crop[3]
},
AspectRatio = title.aspect,
AngleCount = title.angle_count,
VideoCodecName = title.video_codec_name,
Framerate = ((double)title.rate) / title.rate_base
};
switch (title.type)
{
case hb_title_type_anon.HB_STREAM_TYPE:
newTitle.InputType = InputType.Stream;
break;
case hb_title_type_anon.HB_DVD_TYPE:
newTitle.InputType = InputType.Dvd;
break;
case hb_title_type_anon.HB_BD_TYPE:
newTitle.InputType = InputType.Bluray;
break;
}
int currentSubtitleTrack = 1;
List subtitleList = InteropUtilities.ConvertList(title.list_subtitle);
foreach (hb_subtitle_s subtitle in subtitleList)
{
var newSubtitle = new Subtitle
{
TrackNumber = currentSubtitleTrack,
Language = subtitle.lang,
LanguageCode = subtitle.iso639_2
};
if (subtitle.format == hb_subtitle_s_subtype.PICTURESUB)
{
newSubtitle.SubtitleType = SubtitleType.Picture;
}
else if (subtitle.format == hb_subtitle_s_subtype.TEXTSUB)
{
newSubtitle.SubtitleType = SubtitleType.Text;
}
switch (subtitle.source)
{
case hb_subtitle_s_subsource.CC608SUB:
newSubtitle.SubtitleSource = SubtitleSource.CC608;
break;
case hb_subtitle_s_subsource.CC708SUB:
newSubtitle.SubtitleSource = SubtitleSource.CC708;
break;
case hb_subtitle_s_subsource.SRTSUB:
newSubtitle.SubtitleSource = SubtitleSource.SRT;
break;
case hb_subtitle_s_subsource.SSASUB:
newSubtitle.SubtitleSource = SubtitleSource.SSA;
break;
case hb_subtitle_s_subsource.TX3GSUB:
newSubtitle.SubtitleSource = SubtitleSource.TX3G;
break;
case hb_subtitle_s_subsource.UTF8SUB:
newSubtitle.SubtitleSource = SubtitleSource.UTF8;
break;
case hb_subtitle_s_subsource.VOBSUB:
newSubtitle.SubtitleSource = SubtitleSource.VobSub;
break;
default:
break;
}
newTitle.Subtitles.Add(newSubtitle);
currentSubtitleTrack++;
}
int currentAudioTrack = 1;
List audioList = InteropUtilities.ConvertList(title.list_audio);
foreach (hb_audio_s audio in audioList)
{
var newAudio = new AudioTrack
{
TrackNumber = currentAudioTrack,
Codec = Converters.NativeToAudioCodec(audio.config.input.codec),
Language = audio.config.lang.simple,
LanguageCode = audio.config.lang.iso639_2,
Description = audio.config.lang.description,
ChannelLayout = audio.config.input.channel_layout,
SampleRate = audio.config.input.samplerate,
Bitrate = audio.config.input.bitrate
};
newTitle.AudioTracks.Add(newAudio);
currentAudioTrack++;
}
List chapterList = InteropUtilities.ConvertList(title.list_chapter);
foreach (hb_chapter_s chapter in chapterList)
{
var newChapter = new Chapter
{
ChapterNumber = chapter.index,
Duration = TimeSpan.FromSeconds(((double)chapter.duration) / 90000)
};
newTitle.Chapters.Add(newChapter);
}
return newTitle;
}
///
/// Calculates the output size for a non-anamorphic job.
///
/// The encoding profile for the job.
/// The title being encoded.
/// The dimensions of the final encode.
private static Size CalculateNonAnamorphicOutput(EncodingProfile profile, Title title)
{
int sourceWidth = title.Resolution.Width;
int sourceHeight = title.Resolution.Height;
int width = profile.Width;
int height = profile.Height;
Cropping crop;
if (profile.CustomCropping)
{
crop = profile.Cropping;
}
else
{
crop = title.AutoCropDimensions;
}
sourceWidth -= crop.Left;
sourceWidth -= crop.Right;
sourceHeight -= crop.Top;
sourceHeight -= crop.Bottom;
double croppedAspectRatio = ((double)sourceWidth * title.ParVal.Width) / (sourceHeight * title.ParVal.Height);
if (width == 0)
{
width = sourceWidth;
}
if (profile.MaxWidth > 0 && width > profile.MaxWidth)
{
width = profile.MaxWidth;
}
if (height == 0)
{
height = sourceHeight;
}
if (profile.MaxHeight > 0 && height > profile.MaxHeight)
{
height = profile.MaxHeight;
}
if (profile.KeepDisplayAspect)
{
if (profile.Width == 0 && profile.Height == 0 || profile.Width == 0)
{
width = (int)((double)height * croppedAspectRatio);
if (profile.MaxWidth > 0 && width > profile.MaxWidth)
{
width = profile.MaxWidth;
height = (int)((double)width / croppedAspectRatio);
height = GetNearestValue(height, PictureAutoSizeModulus);
}
width = GetNearestValue(width, PictureAutoSizeModulus);
}
else if (profile.Height == 0)
{
height = (int)((double)width / croppedAspectRatio);
if (profile.MaxHeight > 0 && height > profile.MaxHeight)
{
height = profile.MaxHeight;
width = (int)((double)height * croppedAspectRatio);
width = GetNearestValue(width, PictureAutoSizeModulus);
}
height = GetNearestValue(height, PictureAutoSizeModulus);
}
}
return new Size(width, height);
}
///
/// Gets the closest value to the given number divisible by the given modulus.
///
/// The number to approximate.
/// The modulus.
/// The closest value to the given number divisible by the given modulus.
private static int GetNearestValue(int number, int modulus)
{
return modulus * ((number + modulus / 2) / modulus);
}
}
}