win/C#/frmMain/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.Text;\r
   9 using System.Windows.Forms;\r
  10 using System.Globalization;\r
  11 using System.IO;\r
  12 using System.Collections.Generic;\r
  13 \r
  14 namespace Handbrake\r
  15 {\r
  16     class QueryGenerator\r
  17     {\r
  18         /// <summary>\r
  19         /// Generates a CLI query based on the GUI widgets.\r
  20         /// </summary>\r
  21         /// <param name="mainWindow"></param>\r
  22         /// <returns>The CLI String</returns>\r
  23         public string GenerateTheQuery(frmMain mainWindow)\r
  24         {\r
  25             // Source tab\r
  26             string query = "";\r
  27 \r
  28             if ((mainWindow.text_source.Text != "") && (mainWindow.text_source.Text.Trim() != "Click 'Source' to continue"))\r
  29                 query = " -i " + '"' + mainWindow.text_source.Text + '"';\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 (mainWindow.drop_chapterFinish.Text == mainWindow.drop_chapterStart.Text && mainWindow.drop_chapterStart.Text != "Auto")\r
  38                 query += " -c " + mainWindow.drop_chapterStart.Text;\r
  39             else if (mainWindow.drop_chapterStart.Text == "Auto" && mainWindow.drop_chapterFinish.Text != "Auto")\r
  40                 query += " -c " + "0-" + mainWindow.drop_chapterFinish.Text;\r
  41             else if (mainWindow.drop_chapterStart.Text != "Auto" && mainWindow.drop_chapterFinish.Text != "Auto")\r
  42                 query += " -c " + mainWindow.drop_chapterStart.Text + "-" + mainWindow.drop_chapterFinish.Text;\r
  43 \r
  44             // Destination tab\r
  45             if (mainWindow.text_destination.Text != "")\r
  46                 query += " -o " + '"' + mainWindow.text_destination.Text + '"';\r
  47 \r
  48             query += generateTabbedComponentsQuery(mainWindow);\r
  49             return query;\r
  50         }\r
  51 \r
  52         /// <summary>\r
  53         /// Generates a CLI query for the preview function.\r
  54         /// This basically forces a shortened version of the encdode.\r
  55         /// </summary>\r
  56         /// <param name="mainWindow"></param>\r
  57         /// <param name="duration">Duration</param>\r
  58         /// <param name="preview">Start at preview</param>\r
  59         /// <returns>Returns a CLI query String.</returns>\r
  60         public string GeneratePreviewQuery(frmMain mainWindow, string duration, string preview)\r
  61         {\r
  62             int seconds;\r
  63             int.TryParse(duration, out seconds);\r
  64 \r
  65             // Source tab\r
  66             string query = "";\r
  67 \r
  68             if ((mainWindow.text_source.Text != "") && (mainWindow.text_source.Text.Trim() != "Click 'Source' to continue"))\r
  69                 query = " -i " + '"' + mainWindow.text_source.Text + '"';\r
  70 \r
  71             if (mainWindow.drp_dvdtitle.Text != "Automatic")\r
  72             {\r
  73                 string[] titleInfo = mainWindow.drp_dvdtitle.Text.Split(' ');\r
  74                 query += " -t " + titleInfo[0];\r
  75             }\r
  76 \r
  77             query += " --start-at-preview " + preview;\r
  78             query += " --stop-at-duration " + duration + " ";\r
  79 \r
  80             // Destination tab\r
  81             if (mainWindow.text_destination.Text != "")\r
  82                 query += " -o " + '"' + mainWindow.text_destination.Text.Replace(".m", "_sample.m").Replace(".avi", "_sample.avi").Replace(".ogm", "_sample.ogm") + '"';\r
  83 \r
  84             query += generateTabbedComponentsQuery(mainWindow);\r
  85             return query;\r
  86         }\r
  87 \r
  88         /// <summary>\r
  89         /// Generates part of the CLI query, for the tabbed components only.\r
  90         /// </summary>\r
  91         /// <param name="mainWindow"></param>\r
  92         /// <returns></returns>\r
  93         public string generateTabbedComponentsQuery(frmMain mainWindow)\r
  94         {\r
  95             string query = "";\r
  96 \r
  97             #region Output Settings Box\r
  98             query += " -f " + mainWindow.drop_format.Text.ToLower().Replace(" file", "");\r
  99 \r
 100             // These are output settings features\r
 101             if (mainWindow.check_largeFile.Checked)\r
 102                 query += " -4 ";\r
 103 \r
 104             if (mainWindow.check_iPodAtom.Checked)\r
 105                 query += " -I ";\r
 106 \r
 107             if (mainWindow.check_optimiseMP4.Checked)\r
 108                 query += " -O ";\r
 109             #endregion\r
 110 \r
 111             #region Picture Settings Tab\r
 112 \r
 113             // Use MaxWidth for built-in presets and width for user settings.\r
 114             if (mainWindow.maxWidth == 0)\r
 115             {\r
 116 \r
 117                 if (mainWindow.text_width.Text != "")\r
 118                     query += " -w " + mainWindow.text_width.Text;\r
 119             }\r
 120             else\r
 121             {\r
 122                 if (mainWindow.text_width.Text != "")\r
 123                     query += " -X " + mainWindow.text_width.Text;\r
 124             }\r
 125 \r
 126             // Use MaxHeight for built-in presets and height for user settings.\r
 127             if (mainWindow.maxHeight == 0)\r
 128             {\r
 129                 if (mainWindow.text_height.Text != "")\r
 130                     query += " -l " + mainWindow.text_height.Text;\r
 131             }\r
 132             else\r
 133             {\r
 134                 if (mainWindow.text_height.Text != "")\r
 135                     query += " -Y " + mainWindow.text_height.Text;\r
 136             }\r
 137 \r
 138             string cropTop = mainWindow.text_top.Text;\r
 139             string cropBottom = mainWindow.text_bottom.Text;\r
 140             string cropLeft = mainWindow.text_left.Text;\r
 141             string cropRight = mainWindow.text_right.Text;\r
 142 \r
 143             if (mainWindow.check_customCrop.Checked)\r
 144             {\r
 145                 if (mainWindow.text_top.Text == string.Empty)\r
 146                     cropTop = "0";\r
 147                 if (mainWindow.text_bottom.Text == string.Empty)\r
 148                     cropBottom = "0";\r
 149                 if (mainWindow.text_left.Text == string.Empty)\r
 150                     cropLeft = "0";\r
 151                 if (mainWindow.text_right.Text == string.Empty)\r
 152                     cropRight = "0";\r
 153 \r
 154                 query += " --crop " + cropTop + ":" + cropBottom + ":" + cropLeft + ":" + cropRight;\r
 155             }\r
 156 \r
 157             if (mainWindow.drp_anamorphic.SelectedIndex == 1)\r
 158                 query += " -p ";\r
 159             else if (mainWindow.drp_anamorphic.SelectedIndex == 2)\r
 160                 query += " -P ";\r
 161 \r
 162             if (mainWindow.slider_deblock.Value != 4)\r
 163                 query += " --deblock=" + mainWindow.slider_deblock.Value;\r
 164 \r
 165             \r
 166             #endregion\r
 167 \r
 168             #region Filters\r
 169             query += mainWindow.ctl_detelecine.getCLIQuery;\r
 170             query += mainWindow.ctl_decomb.getCLIQuery;\r
 171             query += mainWindow.ctl_deinterlace.getCLIQuery;\r
 172             query += mainWindow.ctl_denoise.getCLIQuery;\r
 173             #endregion\r
 174 \r
 175             #region Video Settings Tab\r
 176 \r
 177             switch (mainWindow.drp_videoEncoder.Text)\r
 178             {\r
 179                 case "MPEG-4 (FFmpeg)":\r
 180                     query += " -e ffmpeg";\r
 181                     break;\r
 182                 case "MPEG-4 (XviD)":\r
 183                     query += " -e xvid";\r
 184                     break;\r
 185                 case "H.264 (x264)":\r
 186                     query += " -e x264";\r
 187                     break;\r
 188                 case "VP3 (Theora)":\r
 189                     query += " -e theora";\r
 190                     break;\r
 191                 default:\r
 192                     query += " -e x264";\r
 193                     break;\r
 194             }\r
 195 \r
 196             if (mainWindow.check_grayscale.Checked)\r
 197                 query += " -g ";\r
 198 \r
 199             // Video Settings\r
 200             if (mainWindow.radio_avgBitrate.Checked)\r
 201                 query += " -b " + mainWindow.text_bitrate.Text;\r
 202 \r
 203             if (mainWindow.radio_targetFilesize.Checked)\r
 204                 query += " -S " + mainWindow.text_filesize.Text;\r
 205 \r
 206             // Video Quality Setting\r
 207             if (mainWindow.radio_cq.Checked)\r
 208             {\r
 209                 float value;\r
 210                 switch (mainWindow.drp_videoEncoder.Text)\r
 211                 {\r
 212                     case "MPEG-4 (FFmpeg)":\r
 213                         value = 31 - (mainWindow.slider_videoQuality.Value -1);\r
 214                         query += " -q " + value.ToString(new CultureInfo("en-US"));\r
 215                         break;\r
 216                     case "MPEG-4 (XviD)":\r
 217                         value = 31 - (mainWindow.slider_videoQuality.Value - 1);\r
 218                         query += " -q " + value.ToString(new CultureInfo("en-US"));\r
 219                         break;\r
 220                     case "H.264 (x264)":\r
 221                         float divided;\r
 222                         float.TryParse(Properties.Settings.Default.x264cqstep, out divided);\r
 223                         value = 51 - mainWindow.slider_videoQuality.Value * divided;\r
 224                         query += " -q " + value.ToString(new CultureInfo("en-US"));\r
 225                         break;\r
 226                     case "VP3 (Theora)":\r
 227                         value = mainWindow.slider_videoQuality.Value;\r
 228                         query += " -q " + value.ToString(new CultureInfo("en-US"));\r
 229                         break;\r
 230                 } \r
 231             }     \r
 232 \r
 233             if (mainWindow.check_2PassEncode.Checked)\r
 234                 query += " -2 ";\r
 235 \r
 236             if (mainWindow.check_turbo.Checked)\r
 237                 query += " -T ";\r
 238 \r
 239             if (mainWindow.drp_videoFramerate.Text != "Same as source")\r
 240                 query += " -r " + mainWindow.drp_videoFramerate.Text;\r
 241             #endregion\r
 242 \r
 243             #region Audio Settings Tab\r
 244 \r
 245             ListView audioTracks = mainWindow.lv_audioList;\r
 246             List<string> tracks = new List<string>();\r
 247             List<string> codecs = new List<string>();\r
 248             List<string> mixdowns = new List<string>();\r
 249             List<string> samplerates = new List<string>();\r
 250             List<string> bitrates = new List<string>();\r
 251             List<string> drcs = new List<string>();\r
 252 \r
 253             // No Audio\r
 254             if (audioTracks.Items.Count == 0)\r
 255                 query += " -a none ";\r
 256 \r
 257             // Gather information about each audio track and store them in the declared lists.\r
 258             foreach (ListViewItem row in audioTracks.Items)\r
 259             {\r
 260                 // Audio Track (-a)\r
 261                 if (row.Text == "Automatic")\r
 262                     tracks.Add("1");\r
 263                 else if (row.Text != "None")\r
 264                 {\r
 265                     string[] tempSub = row.Text.Split(' ');\r
 266                     tracks.Add(tempSub[0]);\r
 267                 }\r
 268 \r
 269                 // Audio Codec (-E)\r
 270                 if (row.SubItems[1].Text != String.Empty)\r
 271                     codecs.Add(getAudioEncoder(row.SubItems[1].Text));\r
 272 \r
 273                 // Audio Mixdown (-6)\r
 274                 if (row.SubItems[2].Text != String.Empty)\r
 275                     mixdowns.Add(getMixDown(row.SubItems[2].Text));\r
 276 \r
 277                 // Sample Rate (-R)\r
 278                 if (row.SubItems[3].Text != String.Empty)\r
 279                     samplerates.Add(row.SubItems[3].Text.Replace("Auto", "0"));\r
 280 \r
 281                 // Audio Bitrate (-B)\r
 282                 if (row.SubItems[4].Text != String.Empty)\r
 283                     bitrates.Add(row.SubItems[4].Text.Replace("Auto", "0"));\r
 284 \r
 285                 // DRC (-D)\r
 286                 if (row.SubItems[5].Text != String.Empty)\r
 287                     drcs.Add(row.SubItems[5].Text);\r
 288             }\r
 289 \r
 290             // Audio Track (-a)\r
 291             String audioItems = "";\r
 292             Boolean firstLoop = true;\r
 293 \r
 294             foreach (String item in tracks)\r
 295             {\r
 296                 if (firstLoop)\r
 297                 {\r
 298                     audioItems = item; firstLoop = false;\r
 299                 }\r
 300                 else\r
 301                     audioItems += "," + item;\r
 302             }\r
 303             if (audioItems.Trim() != String.Empty)\r
 304                 query += " -a " + audioItems;\r
 305             firstLoop = true; audioItems = ""; // Reset for another pass.\r
 306 \r
 307             // Audio Codec (-E)\r
 308             foreach (String item in codecs)\r
 309             {\r
 310                 if (firstLoop)\r
 311                 {\r
 312                     audioItems = item; firstLoop = false;\r
 313                 }\r
 314                 else\r
 315                     audioItems += "," + item;\r
 316             }\r
 317             if (audioItems.Trim() != String.Empty)\r
 318                 query += " -E " + audioItems;\r
 319             firstLoop = true; audioItems = ""; // Reset for another pass.\r
 320 \r
 321             // Audio Mixdown (-6)\r
 322             foreach (String item in mixdowns)\r
 323             {\r
 324                 if (firstLoop)\r
 325                 {\r
 326                     audioItems = item; firstLoop = false;\r
 327                 }\r
 328                 else\r
 329                     audioItems += "," + item;\r
 330             }\r
 331             if (audioItems.Trim() != String.Empty)\r
 332                 query += " -6 " + audioItems;\r
 333             firstLoop = true; audioItems = ""; // Reset for another pass.\r
 334 \r
 335             // Sample Rate (-R)\r
 336             foreach (String item in samplerates)\r
 337             {\r
 338                 if (firstLoop)\r
 339                 {\r
 340                     audioItems = item; firstLoop = false;\r
 341                 }\r
 342                 else\r
 343                     audioItems += "," + item;\r
 344             }\r
 345             if (audioItems.Trim() != String.Empty)\r
 346                 query += " -R " + audioItems;\r
 347             firstLoop = true; audioItems = ""; // Reset for another pass.\r
 348 \r
 349             // Audio Bitrate (-B)\r
 350             foreach (String item in bitrates)\r
 351             {\r
 352                 if (firstLoop)\r
 353                 {\r
 354                     audioItems = item; firstLoop = false;\r
 355                 }\r
 356                 else\r
 357                     audioItems += "," + item;\r
 358             }\r
 359             if (audioItems.Trim() != String.Empty)\r
 360                 query += " -B " + audioItems;\r
 361             firstLoop = true; audioItems = ""; // Reset for another pass.\r
 362 \r
 363             // DRC (-D)\r
 364             foreach (String item in drcs)\r
 365             {\r
 366                 if (firstLoop)\r
 367                 {\r
 368                     audioItems = item; firstLoop = false;\r
 369                 }\r
 370                 else\r
 371                     audioItems += "," + item;\r
 372             }\r
 373             if (audioItems.Trim() != String.Empty)\r
 374                 query += " -D " + audioItems;\r
 375 \r
 376             // Subtitles\r
 377             string subtitles = mainWindow.drp_subtitle.Text;\r
 378             if (subtitles == "Autoselect")\r
 379                 query += " -U ";\r
 380             else if (subtitles != "" && subtitles != "None")\r
 381             {\r
 382                 string[] tempSub;\r
 383                 tempSub = subtitles.Split(' ');\r
 384                 query += " -s " + tempSub[0];\r
 385             }\r
 386 \r
 387             if (mainWindow.check_forced.Checked)\r
 388                 query += " -F ";\r
 389 \r
 390             #endregion\r
 391 \r
 392             #region Chapter Markers\r
 393 \r
 394             // Attach Source name and dvd title to the start of the chapters.csv filename.\r
 395             // This is for the queue. It allows different chapter name files for each title.\r
 396             string[] destName =  mainWindow.text_destination.Text.Split('\\');\r
 397             string dest_name = destName[destName.Length - 1];\r
 398             dest_name = dest_name.Replace("\"", "");\r
 399             dest_name = dest_name.Replace(".mp4", "").Replace(".m4v", "").Replace(".avi", "").Replace(".mkv", "").Replace(".ogm", "");\r
 400 \r
 401             string source_title = mainWindow.drp_dvdtitle.Text;\r
 402             string[] titlesplit = source_title.Split(' ');\r
 403             source_title = titlesplit[0];\r
 404 \r
 405             if (mainWindow.Check_ChapterMarkers.Checked)\r
 406             {\r
 407                 if (dest_name.Trim() != String.Empty)\r
 408                 {\r
 409                     string path;\r
 410                     if (source_title != "Automatic")\r
 411                         path = Path.Combine(Path.GetTempPath(), dest_name + "-" + source_title + "-chapters.csv");\r
 412                     else\r
 413                         path = Path.Combine(Path.GetTempPath(), dest_name + "-chapters.csv");\r
 414 \r
 415                     if (chapterCSVSave(mainWindow, path) == false)\r
 416                         query += " -m ";\r
 417                     else\r
 418                         query += " --markers=" + "\"" + path + "\"";\r
 419                 }\r
 420                 else\r
 421                     query += " -m";\r
 422             }\r
 423             #endregion\r
 424 \r
 425             #region  H264 Tab\r
 426             if (mainWindow.rtf_x264Query.Text != "")\r
 427                 query += " -x " + mainWindow.rtf_x264Query.Text;\r
 428             #endregion\r
 429 \r
 430             #region Processors / Other\r
 431             string processors = Properties.Settings.Default.Processors;\r
 432             if (processors != "Automatic")\r
 433                 query += " -C " + processors + " ";\r
 434 \r
 435             query += " -v ";\r
 436             #endregion\r
 437 \r
 438             return query;\r
 439         }\r
 440 \r
 441         /// <summary>\r
 442         /// Get the CLI equive of the audio mixdown from the widget name.\r
 443         /// </summary>\r
 444         /// <param name="selectedAudio"></param>\r
 445         /// <returns></returns>\r
 446         /// \r
 447         private string getMixDown(string selectedAudio)\r
 448         {\r
 449             switch (selectedAudio)\r
 450             {\r
 451                 case "Automatic":\r
 452                     return "auto";\r
 453                 case "Mono":\r
 454                     return "mono";\r
 455                 case "Stereo":\r
 456                     return "stereo";\r
 457                 case "Dolby Surround":\r
 458                     return "dpl1";\r
 459                 case "Dolby Pro Logic II":\r
 460                     return "dpl2";\r
 461                 case "6 Channel Discrete":\r
 462                     return "6ch";\r
 463                 default:\r
 464                     return "auto";\r
 465             }\r
 466         }\r
 467 \r
 468         /// <summary>\r
 469         /// Get the CLI equiv of the audio encoder from the widget name.\r
 470         /// </summary>\r
 471         /// <param name="selectedEncoder"></param>\r
 472         /// <returns></returns>\r
 473         /// \r
 474         private string getAudioEncoder(string selectedEncoder)\r
 475         {\r
 476             switch (selectedEncoder)\r
 477             {\r
 478                 case "AAC":\r
 479                     return "faac";\r
 480                 case "MP3":\r
 481                     return "lame";\r
 482                 case "Vorbis":\r
 483                     return "vorbis";\r
 484                 case "AC3":\r
 485                     return "ac3";\r
 486                 default:\r
 487                     return "";\r
 488             }\r
 489         }\r
 490 \r
 491         /// <summary>\r
 492         /// This function saves the data in the chapters tab, dataGridView into a CSV file called chapters.csv\r
 493         /// in a directory specified by file_path_name\r
 494         /// </summary>\r
 495         /// <param name="mainWindow"></param>\r
 496         /// <param name="file_path_name"></param>\r
 497         /// <returns></returns>\r
 498         private Boolean chapterCSVSave(frmMain mainWindow, string file_path_name)\r
 499         {\r
 500             try\r
 501             {\r
 502                 StringBuilder csv = new StringBuilder();\r
 503 \r
 504                 foreach (DataGridViewRow row in mainWindow.data_chpt.Rows)\r
 505                 {\r
 506                     csv.Append(row.Cells[0].Value.ToString());\r
 507                     csv.Append(",");\r
 508                     csv.Append(row.Cells[1].Value.ToString());\r
 509                     csv.Append(Environment.NewLine);\r
 510                 }\r
 511                 StreamWriter file = new StreamWriter(file_path_name);\r
 512                 file.Write(csv.ToString());\r
 513                 file.Close();\r
 514                 file.Dispose();\r
 515                 return true;\r
 516 \r
 517             }\r
 518             catch (Exception exc)\r
 519             {\r
 520                 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
 521                 return false;\r
 522             }\r
 523         }\r
 524     }\r
 525 }