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