/* QueryGeneratorUtility.cs $
This file is part of the HandBrake source code.
Homepage: .
It may be used under the terms of the GNU General Public License. */
namespace HandBrake.ApplicationServices.Utilities
{
using System;
using System.Collections.Generic;
using System.Collections.ObjectModel;
using System.Globalization;
using System.IO;
using HandBrake.ApplicationServices.Functions;
using HandBrake.ApplicationServices.Model;
using HandBrake.ApplicationServices.Model.Encoding;
using HandBrake.ApplicationServices.Services.Interfaces;
using HandBrake.Interop.Model.Encoding;
using HandBrake.Interop.Model.Encoding.x264;
///
/// Generate a CLI Query for HandBrakeCLI
///
public class QueryGeneratorUtility
{
///
/// Backing field for the user settings service.
///
private static readonly IUserSettingService UserSettingService = ServiceManager.UserSettingService;
///
/// Generate a CLI Query for an EncodeTask Model object
///
///
/// The task.
///
///
/// A Cli Query
///
public static string GenerateQuery(EncodeTask task)
{
string query = string.Empty;
query += SourceQuery(task, null, null);
query += DestinationQuery(task);
query += GenerateTabbedComponentsQuery(task, true);
return query;
}
///
/// Generate a Query for a Preview Encode
///
///
/// The task.
///
///
/// The duration.
///
///
/// The start At Preview.
///
///
/// A Cli query suitable for generating a preview video.
///
public static string GeneratePreviewQuery(EncodeTask task, int duration, string startAtPreview)
{
string query = string.Empty;
query += SourceQuery(task, duration, startAtPreview);
query += DestinationQuery(task);
query += GenerateTabbedComponentsQuery(task, true);
return query;
}
#region Individual Query Sections
///
/// Generate a Query from an Encode Task Object.
///
///
/// The task.
///
///
/// The enableFilters.
///
///
/// The CLI query for the Tabbed section of the main window UI
///
private static string GenerateTabbedComponentsQuery(EncodeTask task, bool enableFilters)
{
string query = string.Empty;
// Output Settings
query += OutputSettingsQuery(task);
// Filters Panel
if (enableFilters)
query += FiltersQuery(task);
// Picture Settings
query += PictureSettingsQuery(task);
// Video Settings
query += VideoSettingsQuery(task);
// Audio Settings
query += AudioSettingsQuery(task);
// Subtitles Panel
query += SubtitlesQuery(task);
// Chapter Markers
query += ChapterMarkersQuery(task);
// Advanced Panel
query += AdvancedQuery(task);
// Extra Settings
query += ExtraSettings();
return query;
}
///
/// Generate the Command Line Arguments for the Source
///
///
/// The encode task.
///
///
/// The duration.
///
///
/// The preview.
///
///
/// A Cli Query as a string
///
private static string SourceQuery(EncodeTask task, int? duration, string preview)
{
string query = string.Empty;
query += string.Format(" -i \"{0}\"", task.Source);
query += string.Format(" -t {0}", task.Title);
query += string.Format(" --angle {0}", task.Angle);
// Decide what part of the video we want to encode.
switch (task.PointToPointMode)
{
case PointToPointMode.Chapters: // Chapters
if (task.StartPoint == task.EndPoint)
query += string.Format(" -c {0}", task.StartPoint);
else
query += string.Format(" -c {0}-{1}", task.StartPoint, task.EndPoint);
break;
case PointToPointMode.Seconds: // Seconds
int calculatedDuration = task.EndPoint - task.StartPoint;
query += string.Format(" --start-at duration:{0} --stop-at duration:{1}", task.StartPoint, calculatedDuration);
break;
case PointToPointMode.Frames: // Frames
calculatedDuration = task.EndPoint - task.StartPoint;
query += string.Format(" --start-at frame:{0} --stop-at frame:{1}", task.StartPoint, calculatedDuration);
break;
case PointToPointMode.Preview: // Preview
query += " --previews " + UserSettingService.GetUserSetting(ASUserSettingConstants.PreviewScanCount) + " ";
query += " --start-at-preview " + preview;
query += " --stop-at duration:" + duration + " ";
break;
}
return query;
}
///
/// Generate the Command Line Arguments for the Destination File
///
///
/// The encode task.
///
///
/// A Cli Query as a string
///
private static string DestinationQuery(EncodeTask task)
{
string query = string.Empty;
if (task.PointToPointMode == PointToPointMode.Preview)
query += string.Format(" -o \"{0}\" ", task.Destination.Replace(".m", "_sample.m"));
else
query += string.Format(" -o \"{0}\" ", task.Destination);
return query;
}
///
/// Generate the Command Line Arguments for the Output Settings
///
///
/// The encode task.
///
///
/// A Cli Query as a string
///
private static string OutputSettingsQuery(EncodeTask task)
{
string query = string.Empty;
query += string.Format(" -f {0} ", EnumHelper.GetDescription(task.OutputFormat).ToLower());
// These are output settings features
if (task.LargeFile)
query += " -4 ";
if (task.IPod5GSupport)
query += " -I ";
if (task.OptimizeMP4)
query += " -O ";
return query;
}
///
/// Generate the Command Line Arguments for the Picture Settings tab
///
///
/// The encode task.
///
///
/// A Cli Query as a string
///
private static string PictureSettingsQuery(EncodeTask task)
{
string query = string.Empty;
if (task.Anamorphic != Anamorphic.Strict)
{
if (task.MaxWidth.HasValue) query += string.Format(" -X {0}", task.MaxWidth);
else if (task.Width.HasValue && task.Width != 0) query += string.Format(" -w {0}", task.Width);
if (task.MaxWidth.HasValue) query += string.Format(" -Y {0}", task.MaxHeight);
else if (task.Height.HasValue && task.Height != 0) query += string.Format(" -h {0}", task.Height);
}
if (task.HasCropping)
{
query += string.Format(" --crop {0}:{1}:{2}:{3}", task.Cropping.Top, task.Cropping.Bottom, task.Cropping.Left, task.Cropping.Right);
}
switch (task.Anamorphic)
{
case Anamorphic.Strict:
query += " --strict-anamorphic ";
break;
case Anamorphic.Loose:
query += " --loose-anamorphic ";
break;
case Anamorphic.Custom:
query += " --custom-anamorphic ";
if (task.DisplayWidth.HasValue)
query += " --display-width " + task.DisplayWidth + " ";
if (task.KeepDisplayAspect)
query += " --keep-display-aspect ";
if (!task.KeepDisplayAspect)
query += string.Format(" --pixel-aspect {0}:{1}", task.PixelAspectX, task.PixelAspectY);
break;
}
if (task.Modulus.HasValue)
{
query += " --modulus " + task.Modulus;
}
return query;
}
///
/// Generate the Command Line Arguments for the Filters Tab
///
///
/// The encode task.
///
///
/// A Cli Query as a string
///
private static string FiltersQuery(EncodeTask task)
{
string query = string.Empty;
switch (task.Detelecine) // DeTelecine
{
case Detelecine.Off:
query += string.Empty;
break;
case Detelecine.Default:
query += " --detelecine";
break;
case Detelecine.Custom:
query += string.Format(" --detelecine=\"{0}\"", task.CustomDetelecine);
break;
default:
query += string.Empty;
break;
}
switch (task.Decomb) // Decomb
{
case Decomb.Off:
query += string.Empty;
break;
case Decomb.Default:
query += " --decomb";
break;
case Decomb.Custom:
query += string.Format(" --decomb=\"{0}\"", task.CustomDecomb);
break;
default:
query += string.Empty;
break;
}
switch (task.Deinterlace) // DeInterlace
{
case Deinterlace.Fast:
query += " --deinterlace=\"fast\"";
break;
case Deinterlace.Slow:
query += " --deinterlace=\"slow\"";
break;
case Deinterlace.Slower:
query += " --deinterlace=\"slower\"";
break;
case Deinterlace.Custom:
query += string.Format(" --deinterlace=\"{0}\"", task.CustomDeinterlace);
break;
default:
query += string.Empty;
break;
}
switch (task.Denoise) // Denoise
{
case Denoise.Weak:
query += " --denoise=\"weak\"";
break;
case Denoise.Medium:
query += " --denoise=\"medium\"";
break;
case Denoise.Strong:
query += " --denoise=\"strong\"";
break;
case Denoise.Custom:
query += string.Format(" --denoise=\"{0}\"", task.CustomDenoise);
break;
default:
query += string.Empty;
break;
}
if (task.Deblock != 0)
query += string.Format(" --deblock={0}", task.Deblock);
if (task.Grayscale)
query += " -g ";
return query;
}
///
/// Generate the Command Line Arguments for the Video Settings Tab
///
///
/// The encode task.
///
///
/// A Cli Query as a string
///
private static string VideoSettingsQuery(EncodeTask task)
{
string query = string.Empty;
switch (task.VideoEncoder)
{
case VideoEncoder.FFMpeg:
query += " -e ffmpeg";
break;
case VideoEncoder.X264:
query += " -e x264";
break;
case VideoEncoder.Theora:
query += " -e theora";
break;
default:
query += " -e x264";
break;
}
double x264CqStep = UserSettingService.GetUserSetting(ASUserSettingConstants.X264Step);
switch (task.VideoEncodeRateType)
{
case VideoEncodeRateType.AverageBitrate:
if (task.VideoBitrate.HasValue)
query += string.Format(" -b {0}", task.VideoBitrate.Value);
break;
case VideoEncodeRateType.ConstantQuality:
double value;
switch (task.VideoEncoder)
{
case VideoEncoder.FFMpeg:
value = 31 - (task.Quality.Value - 1);
query += string.Format(" -q {0}", value.ToString(CultureInfo.InvariantCulture));
break;
case VideoEncoder.X264:
value = 51 - (task.Quality.Value * x264CqStep);
value = Math.Round(value, 2);
query += string.Format(" -q {0}", value.ToString(CultureInfo.InvariantCulture));
break;
case VideoEncoder.Theora:
value = task.Quality.Value;
query += string.Format(" -q {0}", value.ToString(CultureInfo.InvariantCulture));
break;
}
break;
}
if (task.TwoPass)
query += " -2 ";
if (task.TurboFirstPass)
query += " -T ";
if (task.Framerate.HasValue)
query += string.Format(" -r {0}", task.Framerate);
switch (task.FramerateMode)
{
case FramerateMode.CFR:
query += " --cfr";
break;
case FramerateMode.VFR:
query += " --vfr";
break;
case FramerateMode.PFR:
query += " --pfr";
break;
default:
query += " --vfr";
break;
}
return query;
}
///
/// Generate the Command Line Arguments for the Audio Settings Tab
///
///
/// The encode task.
///
///
/// A Cli Query as a string
///
private static string AudioSettingsQuery(EncodeTask task)
{
string query = string.Empty;
ObservableCollection audioTracks = task.AudioTracks;
List tracks = new List();
List codecs = new List();
List mixdowns = new List();
List samplerates = new List();
List bitrates = new List();
List drcs = new List();
// No Audio
if (audioTracks.Count == 0)
query += " -a none ";
// Gather information about each audio track and store them in the declared lists.
foreach (AudioTrack track in audioTracks)
{
if (track.Track == null)
{
continue;
}
tracks.Add(track.Track.Value);
// Audio Codec (-E)
codecs.Add(track.Encoder);
// Audio Mixdown (-6)
mixdowns.Add(track.MixDown);
// Sample Rate (-R)
samplerates.Add(track.SampleRate);
// Audio Bitrate (-B)
bitrates.Add(track.Bitrate);
// DRC (-D)
drcs.Add(track.DRC);
}
// Audio Track (-a)
string audioItems = string.Empty;
bool firstLoop = true;
foreach (int item in tracks)
{
if (firstLoop)
{
audioItems = item.ToString();
firstLoop = false;
}
else
audioItems += "," + item;
}
if (audioItems.Trim() != String.Empty)
query += " -a " + audioItems;
firstLoop = true;
audioItems = string.Empty; // Reset for another pass.
// Audio Codec (-E)
foreach (AudioEncoder item in codecs)
{
if (firstLoop)
{
audioItems = Converters.GetCliAudioEncoder(item);
firstLoop = false;
}
else
audioItems += "," + Converters.GetCliAudioEncoder(item);
}
if (audioItems.Trim() != String.Empty)
query += " -E " + audioItems;
firstLoop = true;
audioItems = string.Empty; // Reset for another pass.
// Audio Mixdown (-6)
foreach (Mixdown item in mixdowns)
{
if (firstLoop)
{
audioItems = Converters.GetCliMixDown(item);
firstLoop = false;
}
else
audioItems += "," + Converters.GetCliMixDown(item);
}
if (audioItems.Trim() != String.Empty)
query += " -6 " + audioItems;
firstLoop = true;
audioItems = string.Empty; // Reset for another pass.
// Sample Rate (-R)
foreach (double item in samplerates)
{
if (firstLoop)
{
audioItems = item.ToString();
firstLoop = false;
}
else
audioItems += "," + item;
}
if (audioItems.Trim() != String.Empty)
query += " -R " + audioItems;
firstLoop = true;
audioItems = string.Empty; // Reset for another pass.
// Audio Bitrate (-B)
foreach (int item in bitrates)
{
if (firstLoop)
{
audioItems = item.ToString();
firstLoop = false;
}
else
audioItems += "," + item;
}
if (audioItems.Trim() != String.Empty)
query += " -B " + audioItems;
firstLoop = true;
audioItems = string.Empty; // Reset for another pass.
// DRC (-D)
foreach (var itm in drcs)
{
string item = itm.ToString(new CultureInfo("en-US"));
if (firstLoop)
{
audioItems = item;
firstLoop = false;
}
else
audioItems += "," + item;
}
if (audioItems.Trim() != String.Empty)
query += " -D " + audioItems;
// Passthru Settings
if (task.AllowedPassthruOptions != null)
{
string fallbackEncoders = string.Empty;
if (task.AllowedPassthruOptions.AudioAllowAACPass)
{
fallbackEncoders += "aac";
}
if (task.AllowedPassthruOptions.AudioAllowAC3Pass)
{
fallbackEncoders += string.IsNullOrEmpty(fallbackEncoders) ? "ac3" : ",ac3";
}
if (task.AllowedPassthruOptions.AudioAllowDTSHDPass)
{
fallbackEncoders += string.IsNullOrEmpty(fallbackEncoders) ? "dtshd" : ",dtshd";
}
if (task.AllowedPassthruOptions.AudioAllowDTSPass)
{
fallbackEncoders += string.IsNullOrEmpty(fallbackEncoders) ? "dts" : ",dts";
}
if (task.AllowedPassthruOptions.AudioAllowMP3Pass)
{
fallbackEncoders += string.IsNullOrEmpty(fallbackEncoders) ? "mp3" : ",mp3";
}
if (!string.IsNullOrEmpty(fallbackEncoders))
{
query += string.Format(" --audio-copy-mask {0}", fallbackEncoders);
}
query += string.Format(" --audio-fallback {0}", Converters.GetCliAudioEncoder(task.AllowedPassthruOptions.AudioEncoderFallback));
}
return query;
}
///
/// Generate the Command Line Arguments for the Subtitles Tab
///
///
/// The encode task.
///
///
/// A Cli Query as a string
///
private static string SubtitlesQuery(EncodeTask task)
{
string query = string.Empty;
if (task.SubtitleTracks.Count != 0)
{
// BitMap and CC's
string subtitleTracks = String.Empty;
string subtitleForced = String.Empty;
string subtitleBurn = String.Empty;
string subtitleDefault = String.Empty;
// SRT
string srtFile = String.Empty;
string srtCodeset = String.Empty;
string srtOffset = String.Empty;
string srtLang = String.Empty;
string srtDefault = String.Empty;
int srtCount = 0;
int subCount = 0;
// Languages
IDictionary langMap = LanguageUtilities.MapLanguages();
foreach (SubtitleTrack item in task.SubtitleTracks)
{
string itemToAdd;
if (item.IsSrtSubtitle) // We have an SRT file
{
srtCount++; // SRT track id.
srtLang += srtLang == string.Empty ? langMap[item.SrtLang] : "," + langMap[item.SrtLang];
srtCodeset += srtCodeset == string.Empty ? item.SrtCharCode : "," + item.SrtCharCode;
if (item.Default)
srtDefault = srtCount.ToString();
itemToAdd = item.SrtPath;
srtFile += srtFile == string.Empty ? itemToAdd : "," + itemToAdd;
itemToAdd = item.SrtOffset.ToString();
srtOffset += srtOffset == string.Empty ? itemToAdd : "," + itemToAdd;
}
else // We have Bitmap or CC
{
subCount++;
// Find --subtitle
if (item.Track.Contains("Foreign Audio Search"))
itemToAdd = "scan";
else
{
string[] tempSub = item.Track.Split(' ');
itemToAdd = tempSub[0];
}
subtitleTracks += subtitleTracks == string.Empty ? itemToAdd : "," + itemToAdd;
// Find --subtitle-forced
if (item.Forced)
subtitleForced += subtitleForced == string.Empty ? subCount.ToString() : "," + subCount;
// Find --subtitle-burn
if (item.Burned)
subtitleBurn = subCount.ToString();
// Find --subtitle-default
if (item.Default)
subtitleDefault = subCount.ToString();
}
}
// Build The CLI Subtitles Query
if (subtitleTracks != string.Empty)
{
query += " --subtitle " + subtitleTracks;
if (subtitleForced != string.Empty)
query += " --subtitle-forced=" + subtitleForced;
if (subtitleBurn != string.Empty)
query += " --subtitle-burn=" + subtitleBurn;
if (subtitleDefault != string.Empty)
query += " --subtitle-default=" + subtitleDefault;
}
if (srtFile != string.Empty) // SRTs
{
query += " --srt-file " + "\"" + srtFile + "\"";
if (srtCodeset != string.Empty)
query += " --srt-codeset " + srtCodeset;
if (srtOffset != string.Empty)
query += " --srt-offset " + srtOffset;
if (srtLang != string.Empty)
query += " --srt-lang " + srtLang;
if (srtDefault != string.Empty)
query += " --srt-default=" + srtDefault;
}
}
return query;
}
///
/// Generate the Command Line Arguments for the Chapter markers tab
///
///
/// The encode task.
///
///
/// A Cli Query as a string
///
private static string ChapterMarkersQuery(EncodeTask task)
{
string query = string.Empty;
// Attach Source name and dvd title to the start of the chapters.csv filename.
// This is for the queue. It allows different chapter name files for each title.
string destName = Path.GetFileNameWithoutExtension(task.Destination);
string sourceTitle = task.Title.ToString();
if (task.IncludeChapterMarkers && destName != null)
{
if (destName.Trim() != String.Empty)
{
string path = sourceTitle != "Automatic"
? Path.Combine(Path.GetTempPath(), destName + "-" + sourceTitle + "-chapters.csv")
: Path.Combine(Path.GetTempPath(), destName + "-chapters.csv");
if (ChapterCsvSave(task.ChapterNames, path) == false)
query += " -m ";
else
query += " --markers=" + "\"" + path + "\"";
}
else
query += " -m";
}
return query;
}
///
/// Generate the Command Line Arguments for the Advanced Encoder Options
///
///
/// The encode task.
///
///
/// A Cli Query as a string
///
private static string AdvancedQuery(EncodeTask task)
{
if (task.VideoEncoder == VideoEncoder.X264)
{
string query = string.Empty;
if (task.x264Preset != x264Preset.None)
{
query += string.Format("--x264-preset={0} ", task.x264Preset.ToString().ToLower().Replace(" ", string.Empty));
}
if (task.x264Profile != x264Profile.None)
{
query += string.Format("--x264-profile={0} ", task.x264Profile.ToString().ToLower().Replace(" ", string.Empty));
}
if (task.X264Tune != x264Tune.None)
{
query += string.Format("--x264-tune={0} ", task.X264Tune.ToString().ToLower().Replace(" ", string.Empty));
}
if (!string.IsNullOrEmpty(task.AdvancedEncoderOptions))
{
query += string.Format(" -x {0}", task.AdvancedEncoderOptions);
}
return query;
}
return string.IsNullOrEmpty(task.AdvancedEncoderOptions) ? string.Empty : string.Format(" -x {0}", task.AdvancedEncoderOptions);
}
///
/// Generate the Command Line Arguments for any additional advanced options.
///
///
/// A Cli Query as a string
///
private static string ExtraSettings()
{
string query = string.Empty;
// Verbosity Level
int verbosity = UserSettingService.GetUserSetting(ASUserSettingConstants.Verbosity);
query += string.Format(" --verbose= {0}", verbosity);
// LibDVDNav
if (UserSettingService.GetUserSetting(ASUserSettingConstants.DisableLibDvdNav))
query += " --no-dvdnav";
return query;
}
#endregion
#region Helpers
///
/// Create a CSV file with the data from the Main Window Chapters tab
///
/// The List of chapters
/// Path to save the csv file
/// True if successful
private static bool ChapterCsvSave(IEnumerable chapters, string filePathName)
{
string csv = string.Empty;
int counter = 0;
foreach (ChapterMarker name in chapters)
{
csv += counter + "," + name.ChapterName.Replace(",", "\\,") + Environment.NewLine;
counter++;
}
StreamWriter file = new StreamWriter(filePathName);
file.Write(csv);
file.Close();
file.Dispose();
return true;
}
#endregion
}
}