OSDN Git Service

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