win/C#/Functions/QueryGenerator.cs

   1 /*  QueryGenerator.cs $\r
   2     This file is part of the HandBrake source code.\r
   3     Homepage: <http://handbrake.fr/>.\r
   4     It may be used under the terms of the GNU General Public License. */\r
   5 \r
   6 namespace Handbrake.Functions\r
   7 {\r
   8     using System;\r
   9     using System.Collections.Generic;\r
  10     using System.Globalization;\r
  11     using System.IO;\r
  12     using System.Windows.Forms;\r
  13 \r
  14     /// <summary>\r
  15     /// Generate a CLI Query for HandBrakeCLI\r
  16     /// </summary>\r
  17     public class QueryGenerator\r
  18     {\r
  19         /// <summary>\r
  20         /// Generates a full CLI query for either encoding or previe encoeds if duration and preview are defined.\r
  21         /// </summary>\r
  22         /// <param name="mainWindow">\r
  23         /// The Main Window\r
  24         /// </param>\r
  25         /// <param name="mode">\r
  26         /// What Mode. (Point to Point Encoding)  Chapters, Seconds, Frames OR Preview Encode\r
  27         /// </param>\r
  28         /// <param name="duration">\r
  29         /// Time in seconds for preview mode\r
  30         /// </param>\r
  31         /// <param name="preview">\r
  32         /// Preview --start-at-preview (int) \r
  33         /// </param>\r
  34         /// <returns>\r
  35         /// CLI Query \r
  36         /// </returns>\r
  37         public static string GenerateCliQuery(frmMain mainWindow, int mode, int duration, string preview)\r
  38         {\r
  39             string query = string.Empty;\r
  40 \r
  41             if (!string.IsNullOrEmpty(mainWindow.sourcePath) &&\r
  42                 mainWindow.sourcePath.Trim() != "Select \"Source\" to continue")\r
  43                 query = " -i " + '"' + mainWindow.sourcePath + '"';\r
  44 \r
  45             if (mainWindow.drp_dvdtitle.Text != string.Empty)\r
  46             {\r
  47                 string[] titleInfo = mainWindow.drp_dvdtitle.Text.Split(' ');\r
  48                 query += " -t " + titleInfo[0];\r
  49             }\r
  50 \r
  51             if (!Properties.Settings.Default.noDvdNav && mainWindow.drop_angle.Items.Count != 0)\r
  52                 query += " --angle " + mainWindow.drop_angle.SelectedItem;\r
  53 \r
  54             // Decide what part of the video we want to encode.\r
  55             switch (mode)\r
  56             {\r
  57                 case 0: // Chapters\r
  58                     if (mainWindow.drop_chapterFinish.Text == mainWindow.drop_chapterStart.Text &&\r
  59                         mainWindow.drop_chapterStart.Text != string.Empty)\r
  60                         query += string.Format(" -c {0}", mainWindow.drop_chapterStart.Text);\r
  61                     else if (mainWindow.drop_chapterStart.Text != string.Empty &&\r
  62                              mainWindow.drop_chapterFinish.Text != string.Empty)\r
  63                         query += string.Format(" -c {0}-{1}", mainWindow.drop_chapterStart.Text,\r
  64                                                mainWindow.drop_chapterFinish.Text);\r
  65                     break;\r
  66                 case 1: // Seconds\r
  67                     int start, end;\r
  68                     int.TryParse(mainWindow.drop_chapterStart.Text, out start);\r
  69                     int.TryParse(mainWindow.drop_chapterFinish.Text, out end);\r
  70                     int calculatedDuration = end - start;\r
  71 \r
  72                     query += string.Format(" --start-at duration:{0} --stop-at duration:{1}",\r
  73                                            mainWindow.drop_chapterStart.Text, calculatedDuration);\r
  74                     break;\r
  75                 case 2: // Frames\r
  76                     int.TryParse(mainWindow.drop_chapterStart.Text, out start);\r
  77                     int.TryParse(mainWindow.drop_chapterFinish.Text, out end);\r
  78                     calculatedDuration = end - start;\r
  79 \r
  80                     query += string.Format(" --start-at frame:{0} --stop-at frame:{1}",\r
  81                                            mainWindow.drop_chapterStart.Text, calculatedDuration);\r
  82                     break;\r
  83                 case 3: // Preview\r
  84                     query += " --previews " + Properties.Settings.Default.previewScanCount + " ";\r
  85                     query += " --start-at-preview " + preview;\r
  86                     query += " --stop-at duration:" + duration + " ";\r
  87 \r
  88                     if (mainWindow.text_destination.Text != string.Empty)\r
  89                         query += string.Format(" -o \"{0}\" ",\r
  90                                                mainWindow.text_destination.Text.Replace(".m", "_sample.m"));\r
  91                     break;\r
  92                 default:\r
  93                     break;\r
  94             }\r
  95             if (mode != 3)\r
  96                 query += string.Format(" -o \"{0}\" ", mainWindow.text_destination.Text);\r
  97 \r
  98             query += GenerateTabbedComponentsQuery(mainWindow);\r
  99 \r
 100             return query;\r
 101         }\r
 102 \r
 103         /// <summary>\r
 104         /// Generates part of the CLI query, for the tabbed components only.\r
 105         /// </summary>\r
 106         /// <param name="mainWindow">frmMain the main window</param>\r
 107         /// <returns>The CLI Query for the Tab Screens on the main window</returns>\r
 108         public static string GenerateTabbedComponentsQuery(frmMain mainWindow)\r
 109         {\r
 110             string query = string.Empty;\r
 111 \r
 112             #region Output Settings Box\r
 113 \r
 114             query += " -f " + mainWindow.drop_format.Text.ToLower().Replace(" file", string.Empty);\r
 115 \r
 116             // These are output settings features\r
 117             if (mainWindow.check_largeFile.Checked)\r
 118                 query += " -4 ";\r
 119 \r
 120             if (mainWindow.check_iPodAtom.Checked)\r
 121                 query += " -I ";\r
 122 \r
 123             if (mainWindow.check_optimiseMP4.Checked)\r
 124                 query += " -O ";\r
 125 \r
 126             #endregion\r
 127 \r
 128             #region Picture Settings Tab\r
 129 \r
 130             if (mainWindow.PictureSettings.text_width.Value != 0)\r
 131                 if (mainWindow.PictureSettings.drp_anamorphic.SelectedIndex != 1) // Prevent usage for strict anamorphic\r
 132                     query += " -w " + mainWindow.PictureSettings.text_width.Text;\r
 133 \r
 134             if (mainWindow.PictureSettings.text_height.Value != 0 && mainWindow.PictureSettings.text_height.Text != string.Empty)\r
 135                     if (mainWindow.PictureSettings.drp_anamorphic.SelectedIndex == 0 || mainWindow.PictureSettings.drp_anamorphic.SelectedIndex == 3) // Prevent usage for strict anamorphic\r
 136                         query += " -l " + mainWindow.PictureSettings.text_height.Text;\r
 137 \r
 138             string cropTop = mainWindow.PictureSettings.crop_top.Text;\r
 139             string cropBottom = mainWindow.PictureSettings.crop_bottom.Text;\r
 140             string cropLeft = mainWindow.PictureSettings.crop_left.Text;\r
 141             string cropRight = mainWindow.PictureSettings.crop_right.Text;\r
 142 \r
 143             if (mainWindow.PictureSettings.check_customCrop.Checked)\r
 144             {\r
 145                 if (mainWindow.PictureSettings.crop_top.Text == string.Empty)\r
 146                     cropTop = "0";\r
 147                 if (mainWindow.PictureSettings.crop_bottom.Text == string.Empty)\r
 148                     cropBottom = "0";\r
 149                 if (mainWindow.PictureSettings.crop_left.Text == string.Empty)\r
 150                     cropLeft = "0";\r
 151                 if (mainWindow.PictureSettings.crop_right.Text == string.Empty)\r
 152                     cropRight = "0";\r
 153 \r
 154                 query += " --crop " + cropTop + ":" + cropBottom + ":" + cropLeft + ":" + cropRight;\r
 155             }\r
 156 \r
 157             switch (mainWindow.PictureSettings.drp_anamorphic.SelectedIndex)\r
 158             {\r
 159                 case 0:\r
 160                     if (mainWindow.PictureSettings.drp_modulus.SelectedIndex != 0)\r
 161                         query += " --modulus " + mainWindow.PictureSettings.drp_modulus.SelectedItem;\r
 162                     break;\r
 163                 case 1:\r
 164                     query += " --strict-anamorphic ";\r
 165                     break;\r
 166                 case 2:\r
 167                     query += " --loose-anamorphic ";\r
 168                     if (mainWindow.PictureSettings.drp_modulus.SelectedIndex != 0)\r
 169                         query += " --modulus " + mainWindow.PictureSettings.drp_modulus.SelectedItem;\r
 170                     break;\r
 171                 case 3:\r
 172                     query += " --custom-anamorphic ";\r
 173 \r
 174                     if (mainWindow.PictureSettings.drp_modulus.SelectedIndex != 0)\r
 175                         query += " --modulus " + mainWindow.PictureSettings.drp_modulus.SelectedItem;\r
 176 \r
 177                     if (mainWindow.PictureSettings.check_KeepAR.Checked)\r
 178                         query += " --display-width " + mainWindow.PictureSettings.updownDisplayWidth.Text + " ";\r
 179 \r
 180                     if (mainWindow.PictureSettings.check_KeepAR.Checked)\r
 181                         query += " --keep-display-aspect ";\r
 182 \r
 183                     if (!mainWindow.PictureSettings.check_KeepAR.Checked)\r
 184                         if (mainWindow.PictureSettings.updownParWidth.Text != string.Empty &&\r
 185                             mainWindow.PictureSettings.updownParHeight.Text != string.Empty)\r
 186                             query += " --pixel-aspect " + mainWindow.PictureSettings.updownParWidth.Text + ":" +\r
 187                                      mainWindow.PictureSettings.updownParHeight.Text + " ";\r
 188                     break;\r
 189             }\r
 190 \r
 191             #endregion\r
 192 \r
 193             // Filters Panel\r
 194             query += mainWindow.Filters.GetCliQuery;\r
 195 \r
 196             #region Video Settings Tab\r
 197 \r
 198             switch (mainWindow.drp_videoEncoder.Text)\r
 199             {\r
 200                 case "MPEG-4 (FFmpeg)":\r
 201                     query += " -e ffmpeg";\r
 202                     break;\r
 203                 case "H.264 (x264)":\r
 204                     query += " -e x264";\r
 205                     break;\r
 206                 case "VP3 (Theora)":\r
 207                     query += " -e theora";\r
 208                     break;\r
 209                 default:\r
 210                     query += " -e x264";\r
 211                     break;\r
 212             }\r
 213 \r
 214             // Video Settings\r
 215             if (mainWindow.radio_avgBitrate.Checked)\r
 216                 query += " -b " + mainWindow.text_bitrate.Text;\r
 217 \r
 218             if (mainWindow.radio_targetFilesize.Checked)\r
 219                 query += " -S " + mainWindow.text_filesize.Text;\r
 220 \r
 221             // Video Quality Setting\r
 222             if (mainWindow.radio_cq.Checked)\r
 223             {\r
 224                 double cqStep = Properties.Settings.Default.x264cqstep;\r
 225                 double value;\r
 226                 switch (mainWindow.drp_videoEncoder.Text)\r
 227                 {\r
 228                     case "MPEG-4 (FFmpeg)":\r
 229                         value = 31 - (mainWindow.slider_videoQuality.Value - 1);\r
 230                         query += " -q " + value.ToString(new CultureInfo("en-US"));\r
 231                         break;\r
 232                     case "H.264 (x264)":\r
 233                         CultureInfo culture = CultureInfo.CreateSpecificCulture("en-US");\r
 234                         value = 51 - mainWindow.slider_videoQuality.Value * cqStep;\r
 235                         value = Math.Round(value, 2);\r
 236                         query += " -q " + value.ToString(culture);\r
 237                         break;\r
 238                     case "VP3 (Theora)":\r
 239                         value = mainWindow.slider_videoQuality.Value;\r
 240                         query += " -q " + value.ToString(new CultureInfo("en-US"));\r
 241                         break;\r
 242                 }\r
 243             }\r
 244 \r
 245             if (mainWindow.check_2PassEncode.Checked)\r
 246                 query += " -2 ";\r
 247 \r
 248             if (mainWindow.check_turbo.Checked)\r
 249                 query += " -T ";\r
 250 \r
 251             if (mainWindow.drp_videoFramerate.Text != "Same as source")\r
 252                 query += " -r " + mainWindow.drp_videoFramerate.Text;\r
 253 \r
 254             #endregion\r
 255 \r
 256             #region Audio Settings Tab\r
 257 \r
 258             ListView audioTracks = mainWindow.AudioSettings.GetAudioPanel();\r
 259             List<string> tracks = new List<string>();\r
 260             List<string> codecs = new List<string>();\r
 261             List<string> mixdowns = new List<string>();\r
 262             List<string> samplerates = new List<string>();\r
 263             List<string> bitrates = new List<string>();\r
 264             List<string> drcs = new List<string>();\r
 265 \r
 266             // No Audio\r
 267             if (audioTracks.Items.Count == 0)\r
 268                 query += " -a none ";\r
 269 \r
 270             // Gather information about each audio track and store them in the declared lists.\r
 271             foreach (ListViewItem row in audioTracks.Items)\r
 272             {\r
 273                 // Audio Track (-a)\r
 274                 if (row.SubItems[1].Text == "Automatic")\r
 275                     tracks.Add("1");\r
 276                 else if (row.Text != "None")\r
 277                 {\r
 278                     string[] tempSub = row.SubItems[1].Text.Split(' ');\r
 279                     tracks.Add(tempSub[0]);\r
 280                 }\r
 281 \r
 282                 // Audio Codec (-E)\r
 283                 if (row.SubItems[2].Text != String.Empty)\r
 284                     codecs.Add(GetAudioEncoder(row.SubItems[2].Text));\r
 285 \r
 286                 // Audio Mixdown (-6)\r
 287                 if (row.SubItems[3].Text != String.Empty)\r
 288                     mixdowns.Add(GetMixDown(row.SubItems[3].Text));\r
 289 \r
 290                 // Sample Rate (-R)\r
 291                 if (row.SubItems[4].Text != String.Empty)\r
 292                     samplerates.Add(row.SubItems[4].Text);\r
 293 \r
 294                 // Audio Bitrate (-B)\r
 295                 if (row.SubItems[5].Text != String.Empty)\r
 296                     bitrates.Add(row.SubItems[5].Text.Replace("Auto", "auto"));\r
 297 \r
 298                 // DRC (-D)\r
 299                 if (row.SubItems[6].Text != String.Empty)\r
 300                     drcs.Add(row.SubItems[6].Text);\r
 301             }\r
 302 \r
 303             // Audio Track (-a)\r
 304             string audioItems = string.Empty;\r
 305             bool firstLoop = true;\r
 306 \r
 307             foreach (string item in tracks)\r
 308             {\r
 309                 if (firstLoop)\r
 310                 {\r
 311                     audioItems = item;\r
 312                     firstLoop = false;\r
 313                 }\r
 314                 else\r
 315                     audioItems += "," + item;\r
 316             }\r
 317             if (audioItems.Trim() != String.Empty)\r
 318                 query += " -a " + audioItems;\r
 319             firstLoop = true;\r
 320             audioItems = string.Empty; // Reset for another pass.\r
 321 \r
 322             // Audio Codec (-E)\r
 323             foreach (string item in codecs)\r
 324             {\r
 325                 if (firstLoop)\r
 326                 {\r
 327                     audioItems = item;\r
 328                     firstLoop = false;\r
 329                 }\r
 330                 else\r
 331                     audioItems += "," + item;\r
 332             }\r
 333             if (audioItems.Trim() != String.Empty)\r
 334                 query += " -E " + audioItems;\r
 335             firstLoop = true;\r
 336             audioItems = string.Empty; // Reset for another pass.\r
 337 \r
 338             // Audio Mixdown (-6)\r
 339             foreach (string item in mixdowns)\r
 340             {\r
 341                 if (firstLoop)\r
 342                 {\r
 343                     audioItems = item;\r
 344                     firstLoop = false;\r
 345                 }\r
 346                 else\r
 347                     audioItems += "," + item;\r
 348             }\r
 349             if (audioItems.Trim() != String.Empty)\r
 350                 query += " -6 " + audioItems;\r
 351             firstLoop = true;\r
 352             audioItems = string.Empty; // Reset for another pass.\r
 353 \r
 354             // Sample Rate (-R)\r
 355             foreach (string item in samplerates)\r
 356             {\r
 357                 if (firstLoop)\r
 358                 {\r
 359                     audioItems = item;\r
 360                     firstLoop = false;\r
 361                 }\r
 362                 else\r
 363                     audioItems += "," + item;\r
 364             }\r
 365             if (audioItems.Trim() != String.Empty)\r
 366                 query += " -R " + audioItems;\r
 367             firstLoop = true;\r
 368             audioItems = string.Empty; // Reset for another pass.\r
 369 \r
 370             // Audio Bitrate (-B)\r
 371             foreach (string item in bitrates)\r
 372             {\r
 373                 if (firstLoop)\r
 374                 {\r
 375                     audioItems = item;\r
 376                     firstLoop = false;\r
 377                 }\r
 378                 else\r
 379                     audioItems += "," + item;\r
 380             }\r
 381             if (audioItems.Trim() != String.Empty)\r
 382                 query += " -B " + audioItems;\r
 383             firstLoop = true;\r
 384             audioItems = string.Empty; // Reset for another pass.\r
 385 \r
 386             // DRC (-D)\r
 387             foreach (var itm in drcs)\r
 388             {\r
 389                 string item = itm.ToString(new CultureInfo("en-US"));\r
 390                 if (firstLoop)\r
 391                 {\r
 392                     audioItems = item;\r
 393                     firstLoop = false;\r
 394                 }\r
 395                 else\r
 396                     audioItems += "," + item;\r
 397             }\r
 398             if (audioItems.Trim() != String.Empty)\r
 399                 query += " -D " + audioItems;\r
 400 \r
 401             #endregion\r
 402 \r
 403             // Subtitles Panel\r
 404             query += mainWindow.Subtitles.GetCliQuery;\r
 405 \r
 406             #region Chapter Markers\r
 407 \r
 408             // Attach Source name and dvd title to the start of the chapters.csv filename.\r
 409             // This is for the queue. It allows different chapter name files for each title.\r
 410             string[] destName = mainWindow.text_destination.Text.Split('\\');\r
 411             string dest_name = destName[destName.Length - 1];\r
 412             dest_name = dest_name.Replace("\"", string.Empty);\r
 413             dest_name = dest_name.Replace(".mp4", string.Empty).Replace(".m4v", string.Empty).Replace(".mkv", string.Empty);\r
 414 \r
 415             string sourceTitle = mainWindow.drp_dvdtitle.Text;\r
 416             string[] titlesplit = sourceTitle.Split(' ');\r
 417             sourceTitle = titlesplit[0];\r
 418 \r
 419             if (mainWindow.Check_ChapterMarkers.Checked && mainWindow.Check_ChapterMarkers.Enabled)\r
 420             {\r
 421                 if (dest_name.Trim() != String.Empty)\r
 422                 {\r
 423                     string path = sourceTitle != "Automatic"\r
 424                                       ? Path.Combine(Path.GetTempPath(), dest_name + "-" + sourceTitle + "-chapters.csv")\r
 425                                       : Path.Combine(Path.GetTempPath(), dest_name + "-chapters.csv");\r
 426 \r
 427                     if (ChapterCsvSave(mainWindow, path) == false)\r
 428                         query += " -m ";\r
 429                     else\r
 430                         query += " --markers=" + "\"" + path + "\"";\r
 431                 }\r
 432                 else\r
 433                     query += " -m";\r
 434             }\r
 435 \r
 436             #endregion\r
 437 \r
 438             // X264 Panel\r
 439             query += " -x " + mainWindow.x264Panel.X264Query;\r
 440 \r
 441             #region Processors / Other\r
 442 \r
 443             string processors = Properties.Settings.Default.Processors;\r
 444             if (processors != "Automatic")\r
 445                 query += " -C " + processors + " ";\r
 446 \r
 447             query += " -v " + Properties.Settings.Default.verboseLevel;\r
 448 \r
 449             if (Properties.Settings.Default.noDvdNav)\r
 450                 query += " --no-dvdnav";\r
 451 \r
 452             #endregion\r
 453 \r
 454             return query;\r
 455         }\r
 456 \r
 457         /// <summary>\r
 458         /// Return the CLI Mixdown name\r
 459         /// </summary>\r
 460         /// <param name="selectedAudio">GUI mixdown name</param>\r
 461         /// <returns>CLI mixdown name</returns>\r
 462         private static string GetMixDown(string selectedAudio)\r
 463         {\r
 464             switch (selectedAudio)\r
 465             {\r
 466                 case "Automatic":\r
 467                     return "auto";\r
 468                 case "Mono":\r
 469                     return "mono";\r
 470                 case "Stereo":\r
 471                     return "stereo";\r
 472                 case "Dolby Surround":\r
 473                     return "dpl1";\r
 474                 case "Dolby Pro Logic II":\r
 475                     return "dpl2";\r
 476                 case "6 Channel Discrete":\r
 477                     return "6ch";\r
 478                 default:\r
 479                     return "auto";\r
 480             }\r
 481         }\r
 482 \r
 483         /// <summary>\r
 484         /// Get the CLI Audio Encoder name\r
 485         /// </summary>\r
 486         /// <param name="selectedEncoder">\r
 487         /// String The GUI Encode name\r
 488         /// </param>\r
 489         /// <returns>\r
 490         /// String CLI encoder name\r
 491         /// </returns>\r
 492         private static string GetAudioEncoder(string selectedEncoder)\r
 493         {\r
 494             switch (selectedEncoder)\r
 495             {\r
 496                 case "AAC (faac)":\r
 497                     return "faac";\r
 498                 case "MP3 (lame)":\r
 499                     return "lame";\r
 500                 case "Vorbis (vorbis)":\r
 501                     return "vorbis";\r
 502                 case "AC3 Passthru":\r
 503                     return "ac3";\r
 504                 case "DTS Passthru":\r
 505                     return "dts";\r
 506                 default:\r
 507                     return string.Empty;\r
 508             }\r
 509         }\r
 510 \r
 511         /// <summary>\r
 512         /// Create a CSV file with the data from the Main Window Chapters tab\r
 513         /// </summary>\r
 514         /// <param name="mainWindow">Main Window</param>\r
 515         /// <param name="filePathName">Path to save the csv file</param>\r
 516         /// <returns>True if successful </returns>\r
 517         private static bool ChapterCsvSave(frmMain mainWindow, string filePathName)\r
 518         {\r
 519             try\r
 520             {\r
 521                 string csv = string.Empty;\r
 522 \r
 523                 foreach (DataGridViewRow row in mainWindow.data_chpt.Rows)\r
 524                 {\r
 525                     csv += row.Cells[0].Value.ToString();\r
 526                     csv += ",";\r
 527                     csv += row.Cells[1].Value.ToString().Replace(",", "\\,");\r
 528                     csv += Environment.NewLine;\r
 529                 }\r
 530                 StreamWriter file = new StreamWriter(filePathName);\r
 531                 file.Write(csv);\r
 532                 file.Close();\r
 533                 file.Dispose();\r
 534                 return true;\r
 535             }\r
 536             catch (Exception exc)\r
 537             {\r
 538                 MessageBox.Show(\r
 539                     "Unable to save Chapter Makrers file! \nChapter marker names will NOT be saved in your encode \n\n" +\r
 540                     exc, "Warning", MessageBoxButtons.OK, MessageBoxIcon.Warning);\r
 541                 return false;\r
 542             }\r
 543         }\r
 544     }\r
 545 }