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