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 \r
  13 namespace Handbrake.Functions\r
  14 {\r
  15     class QueryGenerator\r
  16     {\r
  17         /// <summary>\r
  18         /// Generates a CLI query based on the GUI widgets.\r
  19         /// </summary>\r
  20         /// <param name="mainWindow"></param>\r
  21         /// <returns>The CLI String</returns>\r
  22         public string generateTheQuery(frmMain mainWindow)\r
  23         {\r
  24             // Source tab\r
  25             string query = "";\r
  26 \r
  27             if (!string.IsNullOrEmpty(mainWindow.sourcePath))\r
  28                 if (mainWindow.sourcePath.Trim() != "Select \"Source\" to continue")\r
  29                     query = " -i " + '"' + mainWindow.sourcePath + '"';\r
  30 \r
  31             if (mainWindow.drp_dvdtitle.Text != "Automatic")\r
  32             {\r
  33                 string[] titleInfo = mainWindow.drp_dvdtitle.Text.Split(' ');\r
  34                 query += " -t " + titleInfo[0];\r
  35             }\r
  36 \r
  37             if (Properties.Settings.Default.dvdnav)\r
  38                 if (mainWindow.drop_angle.Items.Count != 0)\r
  39                     query += " --angle " + mainWindow.drop_angle.SelectedItem;\r
  40 \r
  41             if (mainWindow.drop_chapterFinish.Text == mainWindow.drop_chapterStart.Text && mainWindow.drop_chapterStart.Text != "Auto")\r
  42                 query += " -c " + mainWindow.drop_chapterStart.Text;\r
  43             else if (mainWindow.drop_chapterStart.Text == "Auto" && mainWindow.drop_chapterFinish.Text != "Auto")\r
  44                 query += " -c " + "0-" + mainWindow.drop_chapterFinish.Text;\r
  45             else if (mainWindow.drop_chapterStart.Text != "Auto" && mainWindow.drop_chapterFinish.Text != "Auto")\r
  46                 query += " -c " + mainWindow.drop_chapterStart.Text + "-" + mainWindow.drop_chapterFinish.Text;\r
  47 \r
  48             // Destination tab\r
  49             if (mainWindow.text_destination.Text != "")\r
  50                 query += " -o " + '"' + mainWindow.text_destination.Text + '"';\r
  51 \r
  52             query += generateTabbedComponentsQuery(mainWindow);\r
  53             return query;\r
  54         }\r
  55 \r
  56         /// <summary>\r
  57         /// Generates a CLI query for the preview function.\r
  58         /// This basically forces a shortened version of the encdode.\r
  59         /// </summary>\r
  60         /// <param name="mainWindow"></param>\r
  61         /// <param name="duration">Duration</param>\r
  62         /// <param name="preview">Start at preview</param>\r
  63         /// <returns>Returns a CLI query String.</returns>\r
  64         public string generatePreviewQuery(frmMain mainWindow, string duration, string preview)\r
  65         {\r
  66             int seconds;\r
  67             int.TryParse(duration, out seconds);\r
  68 \r
  69             // Source tab\r
  70             string query = "";\r
  71 \r
  72             if (!string.IsNullOrEmpty(mainWindow.sourcePath))\r
  73                 if (mainWindow.sourcePath.Trim() != "Select \"Source\" to continue")\r
  74                     query = " -i " + '"' + mainWindow.sourcePath + '"';\r
  75 \r
  76             if (mainWindow.drp_dvdtitle.Text != "Automatic")\r
  77             {\r
  78                 string[] titleInfo = mainWindow.drp_dvdtitle.Text.Split(' ');\r
  79                 query += " -t " + titleInfo[0];\r
  80             }\r
  81 \r
  82             if (mainWindow.drop_angle.SelectedIndex != 0)\r
  83                 query += " --angle " + mainWindow.drop_angle.SelectedItem;\r
  84 \r
  85             query += " --start-at-preview " + preview;\r
  86             query += " --stop-at duration:" + duration + " ";\r
  87 \r
  88             // Destination tab\r
  89             if (mainWindow.text_destination.Text != "")\r
  90                 query += " -o " + '"' + mainWindow.text_destination.Text.Replace(".m", "_sample.m") + '"';\r
  91 \r
  92             query += generateTabbedComponentsQuery(mainWindow);\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"></param>\r
 100         /// <returns></returns>\r
 101         public static string generateTabbedComponentsQuery(frmMain mainWindow)\r
 102         {\r
 103             string query = "";\r
 104 \r
 105             #region Output Settings Box\r
 106             query += " -f " + mainWindow.drop_format.Text.ToLower().Replace(" file", "");\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.maxWidth == 0)\r
 123             {\r
 124 \r
 125                 if (mainWindow.PictureSettings.text_width.Text != "")\r
 126                     if (mainWindow.PictureSettings.drp_anamorphic.SelectedIndex != 1) // Prevent usage for strict anamorphic\r
 127                         query += " -w " + mainWindow.PictureSettings.text_width.Text;\r
 128             }\r
 129             else\r
 130             {\r
 131                 if (mainWindow.PictureSettings.text_width.Text != "")\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.maxHeight == 0)\r
 137             {\r
 138                 if (mainWindow.PictureSettings.text_height.Text != "0")\r
 139                     if (mainWindow.PictureSettings.text_height.Text != "")\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.Text != "")\r
 146                     query += " -Y " + mainWindow.PictureSettings.text_height.Text;\r
 147             }\r
 148 \r
 149             string cropTop = mainWindow.PictureSettings.crop_top.Text;\r
 150             string cropBottom = mainWindow.PictureSettings.crop_bottom.Text;\r
 151             string cropLeft = mainWindow.PictureSettings.crop_left.Text;\r
 152             string cropRight = mainWindow.PictureSettings.crop_right.Text;\r
 153 \r
 154             if (mainWindow.PictureSettings.check_customCrop.Checked)\r
 155             {\r
 156                 if (mainWindow.PictureSettings.crop_top.Text == string.Empty)\r
 157                     cropTop = "0";\r
 158                 if (mainWindow.PictureSettings.crop_bottom.Text == string.Empty)\r
 159                     cropBottom = "0";\r
 160                 if (mainWindow.PictureSettings.crop_left.Text == string.Empty)\r
 161                     cropLeft = "0";\r
 162                 if (mainWindow.PictureSettings.crop_right.Text == string.Empty)\r
 163                     cropRight = "0";\r
 164 \r
 165                 query += " --crop " + cropTop + ":" + cropBottom + ":" + cropLeft + ":" + cropRight;\r
 166             }\r
 167 \r
 168             switch (mainWindow.PictureSettings.drp_anamorphic.SelectedIndex)\r
 169             {\r
 170                 case 1:\r
 171                     query += " --strict-anamorphic ";\r
 172                     break;\r
 173                 case 2:\r
 174                     query += " --loose-anamorphic ";\r
 175                     break;\r
 176                 case 3:\r
 177                     query += " --custom-anamorphic ";\r
 178                     query += " --display-width " + mainWindow.PictureSettings.txt_displayWidth.Text + " ";\r
 179                     if (mainWindow.PictureSettings.check_KeepAR.Checked)\r
 180                         query += " --keep-display-aspect ";\r
 181                     if (mainWindow.PictureSettings.txt_parWidth.Text != "" && mainWindow.PictureSettings.txt_parHeight.Text != "")\r
 182                         query += " --pixel-aspect " + mainWindow.PictureSettings.txt_parWidth.Text + ":" + mainWindow.PictureSettings.txt_parHeight.Text + " ";\r
 183                     break;\r
 184             }\r
 185             #endregion\r
 186 \r
 187             #region Filters\r
 188             query += mainWindow.Filters.getCLIQuery;           \r
 189             #endregion\r
 190 \r
 191             #region Video Settings Tab\r
 192 \r
 193             switch (mainWindow.drp_videoEncoder.Text)\r
 194             {\r
 195                 case "MPEG-4 (FFmpeg)":\r
 196                     query += " -e ffmpeg";\r
 197                     break;\r
 198                 case "H.264 (x264)":\r
 199                     query += " -e x264";\r
 200                     break;\r
 201                 case "VP3 (Theora)":\r
 202                     query += " -e theora";\r
 203                     break;\r
 204                 default:\r
 205                     query += " -e x264";\r
 206                     break;\r
 207             }\r
 208 \r
 209             // Video Settings\r
 210             if (mainWindow.radio_avgBitrate.Checked)\r
 211                 query += " -b " + mainWindow.text_bitrate.Text;\r
 212 \r
 213             if (mainWindow.radio_targetFilesize.Checked)\r
 214                 query += " -S " + mainWindow.text_filesize.Text;\r
 215 \r
 216             // Video Quality Setting\r
 217             if (mainWindow.radio_cq.Checked)\r
 218             {\r
 219                 double value;\r
 220                 switch (mainWindow.drp_videoEncoder.Text)\r
 221                 {\r
 222                     case "MPEG-4 (FFmpeg)":\r
 223                         value = 31 - (mainWindow.slider_videoQuality.Value - 1);\r
 224                         query += " -q " + value.ToString(new CultureInfo("en-US"));\r
 225                         break;\r
 226                     case "H.264 (x264)":\r
 227                         CultureInfo culture = CultureInfo.CreateSpecificCulture("en-US");\r
 228                         value = 51 - mainWindow.slider_videoQuality.Value * Properties.Settings.Default.x264cqstep;\r
 229                         value = Math.Round(value, 2);\r
 230                         query += " -q " + value.ToString(culture);\r
 231                         break;\r
 232                     case "VP3 (Theora)":\r
 233                         value = mainWindow.slider_videoQuality.Value;\r
 234                         query += " -q " + value.ToString(new CultureInfo("en-US"));\r
 235                         break;\r
 236                 }\r
 237             }\r
 238 \r
 239             if (mainWindow.check_2PassEncode.Checked)\r
 240                 query += " -2 ";\r
 241 \r
 242             if (mainWindow.check_turbo.Checked)\r
 243                 query += " -T ";\r
 244 \r
 245             if (mainWindow.drp_videoFramerate.Text != "Same as source")\r
 246                 query += " -r " + mainWindow.drp_videoFramerate.Text;\r
 247             #endregion\r
 248 \r
 249             #region Audio Settings Tab\r
 250 \r
 251             ListView audioTracks = mainWindow.AudioSettings.getAudioPanel();\r
 252             List<string> tracks = new List<string>();\r
 253             List<string> codecs = new List<string>();\r
 254             List<string> mixdowns = new List<string>();\r
 255             List<string> samplerates = new List<string>();\r
 256             List<string> bitrates = new List<string>();\r
 257             List<string> drcs = new List<string>();\r
 258 \r
 259             // No Audio\r
 260             if (audioTracks.Items.Count == 0)\r
 261                 query += " -a none ";\r
 262 \r
 263             // Gather information about each audio track and store them in the declared lists.\r
 264             foreach (ListViewItem row in audioTracks.Items)\r
 265             {\r
 266                 // Audio Track (-a)\r
 267                 if (row.SubItems[1].Text == "Automatic")\r
 268                     tracks.Add("1");\r
 269                 else if (row.Text != "None")\r
 270                 {\r
 271                     string[] tempSub = row.SubItems[1].Text.Split(' ');\r
 272                     tracks.Add(tempSub[0]);\r
 273                 }\r
 274 \r
 275                 // Audio Codec (-E)\r
 276                 if (row.SubItems[2].Text != String.Empty)\r
 277                     codecs.Add(getAudioEncoder(row.SubItems[2].Text));\r
 278 \r
 279                 // Audio Mixdown (-6)\r
 280                 if (row.SubItems[3].Text != String.Empty)\r
 281                     mixdowns.Add(getMixDown(row.SubItems[3].Text));\r
 282 \r
 283                 // Sample Rate (-R)\r
 284                 if (row.SubItems[4].Text != String.Empty)\r
 285                     samplerates.Add(row.SubItems[4].Text.Replace("Auto", "Auto"));\r
 286 \r
 287                 // Audio Bitrate (-B)\r
 288                 if (row.SubItems[5].Text != String.Empty)\r
 289                     bitrates.Add(row.SubItems[5].Text.Replace("Auto", "auto"));\r
 290 \r
 291                 // DRC (-D)\r
 292                 if (row.SubItems[6].Text != String.Empty)\r
 293                     drcs.Add(row.SubItems[6].Text);\r
 294             }\r
 295 \r
 296             // Audio Track (-a)\r
 297             String audioItems = "";\r
 298             Boolean firstLoop = true;\r
 299 \r
 300             foreach (String item in tracks)\r
 301             {\r
 302                 if (firstLoop)\r
 303                 {\r
 304                     audioItems = item; firstLoop = false;\r
 305                 }\r
 306                 else\r
 307                     audioItems += "," + item;\r
 308             }\r
 309             if (audioItems.Trim() != String.Empty)\r
 310                 query += " -a " + audioItems;\r
 311             firstLoop = true; audioItems = ""; // Reset for another pass.\r
 312 \r
 313             // Audio Codec (-E)\r
 314             foreach (String item in codecs)\r
 315             {\r
 316 \r
 317                 if (firstLoop)\r
 318                 {\r
 319                     audioItems = item; firstLoop = false;\r
 320                 }\r
 321                 else\r
 322                     audioItems += "," + item;\r
 323             }\r
 324             if (audioItems.Trim() != String.Empty)\r
 325                 query += " -E " + audioItems;\r
 326             firstLoop = true; audioItems = ""; // Reset for another pass.\r
 327 \r
 328             // Audio Mixdown (-6)\r
 329             foreach (String item in mixdowns)\r
 330             {\r
 331                 if (firstLoop)\r
 332                 {\r
 333                     audioItems = item; firstLoop = false;\r
 334                 }\r
 335                 else\r
 336                     audioItems += "," + item;\r
 337             }\r
 338             if (audioItems.Trim() != String.Empty)\r
 339                 query += " -6 " + audioItems;\r
 340             firstLoop = true; audioItems = ""; // Reset for another pass.\r
 341 \r
 342             // Sample Rate (-R)\r
 343             foreach (String item in samplerates)\r
 344             {\r
 345                 if (firstLoop)\r
 346                 {\r
 347                     audioItems = item; firstLoop = false;\r
 348                 }\r
 349                 else\r
 350                     audioItems += "," + item;\r
 351             }\r
 352             if (audioItems.Trim() != String.Empty)\r
 353                 query += " -R " + audioItems;\r
 354             firstLoop = true; audioItems = ""; // Reset for another pass.\r
 355 \r
 356             // Audio Bitrate (-B)\r
 357             foreach (String item in bitrates)\r
 358             {\r
 359                 if (firstLoop)\r
 360                 {\r
 361                     audioItems = item; firstLoop = false;\r
 362                 }\r
 363                 else\r
 364                     audioItems += "," + item;\r
 365             }\r
 366             if (audioItems.Trim() != String.Empty)\r
 367                 query += " -B " + audioItems;\r
 368             firstLoop = true; audioItems = ""; // Reset for another pass.\r
 369 \r
 370             // DRC (-D)\r
 371             foreach (String item in drcs)\r
 372             {\r
 373                 if (firstLoop)\r
 374                 {\r
 375                     audioItems = item; firstLoop = false;\r
 376                 }\r
 377                 else\r
 378                     audioItems += "," + item;\r
 379             }\r
 380             if (audioItems.Trim() != String.Empty)\r
 381                 query += " -D " + audioItems;\r
 382 \r
 383             #endregion\r
 384 \r
 385             #region Subtitles Tab\r
 386 \r
 387             if (mainWindow.Subtitles.lv_subList.Items.Count != 0) // If we have subtitle tracks\r
 388             {\r
 389                 // Find --subtitle <string>\r
 390                 query += " --subtitle ";\r
 391                 String subtitleTracks = "";\r
 392                 String itemToAdd;\r
 393                 foreach (ListViewItem item in mainWindow.Subtitles.lv_subList.Items)\r
 394                 {\r
 395                     if (item.SubItems[1].Text.Contains("Foreign Audio Search"))\r
 396                         itemToAdd = "scan";\r
 397                     else\r
 398                     {\r
 399                         string[] tempSub = item.SubItems[1].Text.Split(' ');\r
 400                         itemToAdd = tempSub[0];\r
 401                     }\r
 402 \r
 403                     subtitleTracks += subtitleTracks == "" ? itemToAdd : "," + itemToAdd;\r
 404                 }\r
 405                 query += subtitleTracks;\r
 406 \r
 407 \r
 408                 // Find --subtitle-forced\r
 409                 String forcedTrack = "";\r
 410                 foreach (ListViewItem item in mainWindow.Subtitles.lv_subList.Items)\r
 411                 {\r
 412                     itemToAdd = "";\r
 413                     string[] tempSub = item.SubItems[1].Text.Split(' ');\r
 414                     string trackID = tempSub[0];\r
 415 \r
 416                     if (item.SubItems[2].Text == "Yes")\r
 417                         itemToAdd = trackID;\r
 418 \r
 419                     if (itemToAdd != "")\r
 420                         forcedTrack += forcedTrack == "" ? itemToAdd : "," + itemToAdd;\r
 421                 }\r
 422                 if (forcedTrack != "")\r
 423                     query += " --subtitle-forced " + forcedTrack;\r
 424 \r
 425 \r
 426                 // Find --subtitle-burn and --subtitle-default\r
 427                 String burned = "";\r
 428                 String defaultTrack = "";\r
 429                 foreach (ListViewItem item in mainWindow.Subtitles.lv_subList.Items)\r
 430                 {\r
 431                     string[] tempSub = item.SubItems[1].Text.Split(' ');\r
 432                     string trackID = tempSub[0];\r
 433 \r
 434                     if (trackID.Trim() == "Foreign")\r
 435                         trackID = "scan";\r
 436 \r
 437                     if (item.SubItems[3].Text == "Yes") // Burned\r
 438                         burned = trackID;\r
 439 \r
 440                     if (item.SubItems[4].Text == "Yes") // Burned\r
 441                         defaultTrack = trackID;\r
 442 \r
 443                 }\r
 444                 if (burned != "")\r
 445                     query += " --subtitle-burn " + burned;\r
 446 \r
 447                 if (defaultTrack != "")\r
 448                     query += " --subtitle-default " + defaultTrack;\r
 449 \r
 450             }\r
 451 \r
 452             #endregion\r
 453 \r
 454             #region Chapter Markers\r
 455 \r
 456             // Attach Source name and dvd title to the start of the chapters.csv filename.\r
 457             // This is for the queue. It allows different chapter name files for each title.\r
 458             string[] destName = mainWindow.text_destination.Text.Split('\\');\r
 459             string dest_name = destName[destName.Length - 1];\r
 460             dest_name = dest_name.Replace("\"", "");\r
 461             dest_name = dest_name.Replace(".mp4", "").Replace(".m4v", "").Replace(".mkv", "");\r
 462 \r
 463             string source_title = mainWindow.drp_dvdtitle.Text;\r
 464             string[] titlesplit = source_title.Split(' ');\r
 465             source_title = titlesplit[0];\r
 466 \r
 467             if (mainWindow.Check_ChapterMarkers.Checked)\r
 468             {\r
 469                 if (dest_name.Trim() != String.Empty)\r
 470                 {\r
 471                     string path = source_title != "Automatic"\r
 472                                       ? Path.Combine(Path.GetTempPath(), dest_name + "-" + source_title + "-chapters.csv")\r
 473                                       : Path.Combine(Path.GetTempPath(), dest_name + "-chapters.csv");\r
 474 \r
 475                     if (chapterCSVSave(mainWindow, path) == false)\r
 476                         query += " -m ";\r
 477                     else\r
 478                         query += " --markers=" + "\"" + path + "\"";\r
 479                 }\r
 480                 else\r
 481                     query += " -m";\r
 482             }\r
 483             #endregion\r
 484 \r
 485             #region  H264 Tab\r
 486             if (mainWindow.x264Panel.x264Query != "")\r
 487                 query += " -x " + mainWindow.x264Panel.x264Query;\r
 488             #endregion\r
 489 \r
 490             #region Processors / Other\r
 491             string processors = Properties.Settings.Default.Processors;\r
 492             if (processors != "Automatic")\r
 493                 query += " -C " + processors + " ";\r
 494 \r
 495             query += " -v " + Properties.Settings.Default.verboseLevel;\r
 496 \r
 497             if (Properties.Settings.Default.dvdnav)\r
 498                 query += " --dvdnav";\r
 499             #endregion\r
 500 \r
 501             return query;\r
 502         }\r
 503 \r
 504         private static string getMixDown(string selectedAudio)\r
 505         {\r
 506             switch (selectedAudio)\r
 507             {\r
 508                 case "Automatic":\r
 509                     return "auto";\r
 510                 case "Mono":\r
 511                     return "mono";\r
 512                 case "Stereo":\r
 513                     return "stereo";\r
 514                 case "Dolby Surround":\r
 515                     return "dpl1";\r
 516                 case "Dolby Pro Logic II":\r
 517                     return "dpl2";\r
 518                 case "6 Channel Discrete":\r
 519                     return "6ch";\r
 520                 default:\r
 521                     return "auto";\r
 522             }\r
 523         }\r
 524         private static string getAudioEncoder(string selectedEncoder)\r
 525         {\r
 526             switch (selectedEncoder)\r
 527             {\r
 528                 case "AAC (faac)":\r
 529                     return "faac";\r
 530                 case "MP3 (lame)":\r
 531                     return "lame";\r
 532                 case "Vorbis (vorbis)":\r
 533                     return "vorbis";\r
 534                 case "AC3 Passthru":\r
 535                     return "ac3";\r
 536                 case "DTS Passthru":\r
 537                     return "dts";\r
 538                 default:\r
 539                     return "";\r
 540             }\r
 541         }\r
 542         private static Boolean chapterCSVSave(frmMain mainWindow, string filePathName)\r
 543         {\r
 544             try\r
 545             {\r
 546                 string csv = "";\r
 547 \r
 548                 foreach (DataGridViewRow row in mainWindow.data_chpt.Rows)\r
 549                 {\r
 550                     csv += row.Cells[0].Value.ToString();\r
 551                     csv += ",";\r
 552                     csv += row.Cells[1].Value.ToString();\r
 553                     csv += Environment.NewLine;\r
 554                 }\r
 555                 StreamWriter file = new StreamWriter(filePathName);\r
 556                 file.Write(csv);\r
 557                 file.Close();\r
 558                 file.Dispose();\r
 559                 return true;\r
 560 \r
 561             }\r
 562             catch (Exception exc)\r
 563             {\r
 564                 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
 565                 return false;\r
 566             }\r
 567         }\r
 568     }\r
 569 }