win/C#/Functions/QueryGenerator.cs

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