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 cqStep = Properties.Settings.Default.x264cqstep;\r
196                 double value;\r
197                 switch (mainWindow.drp_videoEncoder.Text)\r
198                 {\r
199                     case "MPEG-4 (FFmpeg)":\r
200                         value = 31 - (mainWindow.slider_videoQuality.Value - 1);\r
201                         query += " -q " + value.ToString(new CultureInfo("en-US"));\r
202                         break;\r
203                     case "H.264 (x264)":\r
204                         CultureInfo culture = CultureInfo.CreateSpecificCulture("en-US");\r
205                         value = 51 - mainWindow.slider_videoQuality.Value * cqStep;\r
206                         value = Math.Round(value, 2);\r
207                         query += " -q " + value.ToString(culture);\r
208                         break;\r
209                     case "VP3 (Theora)":\r
210                         value = mainWindow.slider_videoQuality.Value;\r
211                         query += " -q " + value.ToString(new CultureInfo("en-US"));\r
212                         break;\r
213                 }\r
214             }\r
215 \r
216             if (mainWindow.check_2PassEncode.Checked)\r
217                 query += " -2 ";\r
218 \r
219             if (mainWindow.check_turbo.Checked)\r
220                 query += " -T ";\r
221 \r
222             if (mainWindow.drp_videoFramerate.Text != "Same as source")\r
223                 query += " -r " + mainWindow.drp_videoFramerate.Text;\r
224             #endregion\r
225 \r
226             #region Audio Settings Tab\r
227 \r
228             ListView audioTracks = mainWindow.AudioSettings.GetAudioPanel();\r
229             List<string> tracks = new List<string>();\r
230             List<string> codecs = new List<string>();\r
231             List<string> mixdowns = new List<string>();\r
232             List<string> samplerates = new List<string>();\r
233             List<string> bitrates = new List<string>();\r
234             List<string> drcs = new List<string>();\r
235 \r
236             // No Audio\r
237             if (audioTracks.Items.Count == 0)\r
238                 query += " -a none ";\r
239 \r
240             // Gather information about each audio track and store them in the declared lists.\r
241             foreach (ListViewItem row in audioTracks.Items)\r
242             {\r
243                 // Audio Track (-a)\r
244                 if (row.SubItems[1].Text == "Automatic")\r
245                     tracks.Add("1");\r
246                 else if (row.Text != "None")\r
247                 {\r
248                     string[] tempSub = row.SubItems[1].Text.Split(' ');\r
249                     tracks.Add(tempSub[0]);\r
250                 }\r
251 \r
252                 // Audio Codec (-E)\r
253                 if (row.SubItems[2].Text != String.Empty)\r
254                     codecs.Add(getAudioEncoder(row.SubItems[2].Text));\r
255 \r
256                 // Audio Mixdown (-6)\r
257                 if (row.SubItems[3].Text != String.Empty)\r
258                     mixdowns.Add(getMixDown(row.SubItems[3].Text));\r
259 \r
260                 // Sample Rate (-R)\r
261                 if (row.SubItems[4].Text != String.Empty)\r
262                     samplerates.Add(row.SubItems[4].Text.Replace("Auto", "Auto"));\r
263 \r
264                 // Audio Bitrate (-B)\r
265                 if (row.SubItems[5].Text != String.Empty)\r
266                     bitrates.Add(row.SubItems[5].Text.Replace("Auto", "auto"));\r
267 \r
268                 // DRC (-D)\r
269                 if (row.SubItems[6].Text != String.Empty)\r
270                     drcs.Add(row.SubItems[6].Text);\r
271             }\r
272 \r
273             // Audio Track (-a)\r
274             String audioItems = "";\r
275             Boolean firstLoop = true;\r
276 \r
277             foreach (String item in tracks)\r
278             {\r
279                 if (firstLoop)\r
280                 {\r
281                     audioItems = item; firstLoop = false;\r
282                 }\r
283                 else\r
284                     audioItems += "," + item;\r
285             }\r
286             if (audioItems.Trim() != String.Empty)\r
287                 query += " -a " + audioItems;\r
288             firstLoop = true; audioItems = ""; // Reset for another pass.\r
289 \r
290             // Audio Codec (-E)\r
291             foreach (String item in codecs)\r
292             {\r
293 \r
294                 if (firstLoop)\r
295                 {\r
296                     audioItems = item; firstLoop = false;\r
297                 }\r
298                 else\r
299                     audioItems += "," + item;\r
300             }\r
301             if (audioItems.Trim() != String.Empty)\r
302                 query += " -E " + audioItems;\r
303             firstLoop = true; audioItems = ""; // Reset for another pass.\r
304 \r
305             // Audio Mixdown (-6)\r
306             foreach (String item in mixdowns)\r
307             {\r
308                 if (firstLoop)\r
309                 {\r
310                     audioItems = item; firstLoop = false;\r
311                 }\r
312                 else\r
313                     audioItems += "," + item;\r
314             }\r
315             if (audioItems.Trim() != String.Empty)\r
316                 query += " -6 " + audioItems;\r
317             firstLoop = true; audioItems = ""; // Reset for another pass.\r
318 \r
319             // Sample Rate (-R)\r
320             foreach (String item in samplerates)\r
321             {\r
322                 if (firstLoop)\r
323                 {\r
324                     audioItems = item; firstLoop = false;\r
325                 }\r
326                 else\r
327                     audioItems += "," + item;\r
328             }\r
329             if (audioItems.Trim() != String.Empty)\r
330                 query += " -R " + audioItems;\r
331             firstLoop = true; audioItems = ""; // Reset for another pass.\r
332 \r
333             // Audio Bitrate (-B)\r
334             foreach (String item in bitrates)\r
335             {\r
336                 if (firstLoop)\r
337                 {\r
338                     audioItems = item; firstLoop = false;\r
339                 }\r
340                 else\r
341                     audioItems += "," + item;\r
342             }\r
343             if (audioItems.Trim() != String.Empty)\r
344                 query += " -B " + audioItems;\r
345             firstLoop = true; audioItems = ""; // Reset for another pass.\r
346 \r
347             // DRC (-D)\r
348             foreach (String item in drcs)\r
349             {\r
350                 if (firstLoop)\r
351                 {\r
352                     audioItems = item; firstLoop = false;\r
353                 }\r
354                 else\r
355                     audioItems += "," + item;\r
356             }\r
357             if (audioItems.Trim() != String.Empty)\r
358                 query += " -D " + audioItems;\r
359 \r
360             #endregion\r
361 \r
362             #region Subtitles Tab\r
363             if (mainWindow.Subtitles.lv_subList.Items.Count != 0) // If we have subtitle tracks\r
364             {\r
365                 IDictionary<string, string> langMap = Main.mapLanguages();\r
366 \r
367                 // BitMap and CC's\r
368                 string subtitleTracks = String.Empty;\r
369                 string subtitleForced = String.Empty;\r
370                 string subtitleBurn = String.Empty;\r
371                 string subtitleDefault = String.Empty;\r
372 \r
373                 // SRT\r
374                 string srtFile = String.Empty;\r
375                 string srtCodeset = String.Empty;\r
376                 string srtOffset = String.Empty;\r
377                 string srtLang = String.Empty;\r
378                 string srtDefault = String.Empty;\r
379                 int srtCount = 0;\r
380 \r
381                 foreach (ListViewItem item in mainWindow.Subtitles.lv_subList.Items)\r
382                 {\r
383                     string itemToAdd, trackID;\r
384 \r
385                     if (item.SubItems.Count != 5) // We have an SRT file\r
386                     {\r
387                         srtCount++; // SRT track id.\r
388                         string[] trackData = item.SubItems[1].Text.Split(',');\r
389                         if (trackData != null)\r
390                         {\r
391                             string charCode = trackData[1].Replace("(", "").Replace(")", "").Trim();\r
392                             string realLangCode = langMap[trackData[0].Trim()];\r
393 \r
394                             srtLang += srtLang == "" ? realLangCode : "," + realLangCode;\r
395                             srtCodeset += srtCodeset == "" ? charCode : "," + charCode;\r
396                         }\r
397                         if (item.SubItems[4].Text == "Yes") // default\r
398                             srtDefault = srtCount.ToString();\r
399 \r
400                         itemToAdd = item.SubItems[5].Text;\r
401                         srtFile += srtFile == "" ? itemToAdd : "," + itemToAdd;\r
402 \r
403                         itemToAdd = item.SubItems[6].Text;\r
404                         srtOffset += srtOffset == "" ? itemToAdd : "," + itemToAdd;\r
405                     }\r
406                     else // We have Bitmap or CC\r
407                     {\r
408                         string[] tempSub;\r
409 \r
410                         // Find --subtitle <string>\r
411                         if (item.SubItems[1].Text.Contains("Foreign Audio Search"))\r
412                             itemToAdd = "scan";\r
413                         else\r
414                         {\r
415                             tempSub = item.SubItems[1].Text.Split(' ');\r
416                             itemToAdd = tempSub[0];\r
417                         }\r
418 \r
419                         subtitleTracks += subtitleTracks == "" ? itemToAdd : "," + itemToAdd;\r
420 \r
421                         // Find --subtitle-forced\r
422                         itemToAdd = "";\r
423                         tempSub = item.SubItems[1].Text.Split(' ');\r
424                         trackID = tempSub[0];\r
425 \r
426                         if (item.SubItems[2].Text == "Yes")\r
427                             itemToAdd = trackID;\r
428 \r
429                         if (itemToAdd != "")\r
430                             subtitleForced += subtitleForced == "" ? itemToAdd : "," + itemToAdd;\r
431 \r
432                         // Find --subtitle-burn and --subtitle-default\r
433                         tempSub = item.SubItems[1].Text.Split(' ');\r
434                         trackID = tempSub[0];\r
435 \r
436                         if (trackID.Trim() == "Foreign")\r
437                             trackID = "scan";\r
438 \r
439                         if (item.SubItems[3].Text == "Yes") // burn\r
440                             subtitleBurn = trackID;\r
441 \r
442                         if (item.SubItems[4].Text == "Yes") // default\r
443                             subtitleDefault = trackID;\r
444                     }\r
445                 }\r
446 \r
447                 // Build The CLI Subtitles Query\r
448                 if (subtitleTracks != "")\r
449                 {\r
450                     query += " --subtitle " + subtitleTracks;\r
451 \r
452                     if (subtitleForced != "")\r
453                         query += " --subtitle-forced " + subtitleForced;\r
454                     if (subtitleBurn != "")\r
455                         query += " --subtitle-burn " + subtitleBurn;\r
456                     if (subtitleDefault != "")\r
457                         query += " --subtitle-default " + subtitleDefault;\r
458                 }\r
459 \r
460                 if (srtFile != "") // SRTs\r
461                 {\r
462                     query += " --srt-file " + "\"" + srtFile + "\"";\r
463 \r
464                     if (srtCodeset != "")\r
465                         query += " --srt-codeset " + srtCodeset;\r
466                     if (srtOffset != "")\r
467                         query += " --srt-offset " + srtOffset;\r
468                     if (srtLang != "")\r
469                         query += " --srt-lang " + srtLang;\r
470                     if (srtDefault != "")\r
471                         query += " --subtitle-default " + srtDefault;\r
472                 }\r
473 \r
474             }\r
475             #endregion\r
476 \r
477             #region Chapter Markers\r
478 \r
479             // Attach Source name and dvd title to the start of the chapters.csv filename.\r
480             // This is for the queue. It allows different chapter name files for each title.\r
481             string[] destName = mainWindow.text_destination.Text.Split('\\');\r
482             string dest_name = destName[destName.Length - 1];\r
483             dest_name = dest_name.Replace("\"", "");\r
484             dest_name = dest_name.Replace(".mp4", "").Replace(".m4v", "").Replace(".mkv", "");\r
485 \r
486             string source_title = mainWindow.drp_dvdtitle.Text;\r
487             string[] titlesplit = source_title.Split(' ');\r
488             source_title = titlesplit[0];\r
489 \r
490             if (mainWindow.Check_ChapterMarkers.Checked && mainWindow.Check_ChapterMarkers.Enabled)\r
491             {\r
492                 if (dest_name.Trim() != String.Empty)\r
493                 {\r
494                     string path = source_title != "Automatic"\r
495                                       ? Path.Combine(Path.GetTempPath(), dest_name + "-" + source_title + "-chapters.csv")\r
496                                       : Path.Combine(Path.GetTempPath(), dest_name + "-chapters.csv");\r
497 \r
498                     if (chapterCSVSave(mainWindow, path) == false)\r
499                         query += " -m ";\r
500                     else\r
501                         query += " --markers=" + "\"" + path + "\"";\r
502                 }\r
503                 else\r
504                     query += " -m";\r
505             }\r
506             #endregion\r
507 \r
508             #region  H264 Tab\r
509             if (mainWindow.x264Panel.x264Query != "")\r
510                 query += " -x " + mainWindow.x264Panel.x264Query;\r
511             #endregion\r
512 \r
513             #region Processors / Other\r
514             string processors = Properties.Settings.Default.Processors;\r
515             if (processors != "Automatic")\r
516                 query += " -C " + processors + " ";\r
517 \r
518             query += " -v " + Properties.Settings.Default.verboseLevel;\r
519 \r
520             if (Properties.Settings.Default.dvdnav)\r
521                 query += " --dvdnav";\r
522             #endregion\r
523 \r
524             return query;\r
525         }\r
526 \r
527         private static string getMixDown(string selectedAudio)\r
528         {\r
529             switch (selectedAudio)\r
530             {\r
531                 case "Automatic":\r
532                     return "auto";\r
533                 case "Mono":\r
534                     return "mono";\r
535                 case "Stereo":\r
536                     return "stereo";\r
537                 case "Dolby Surround":\r
538                     return "dpl1";\r
539                 case "Dolby Pro Logic II":\r
540                     return "dpl2";\r
541                 case "6 Channel Discrete":\r
542                     return "6ch";\r
543                 default:\r
544                     return "auto";\r
545             }\r
546         }\r
547         private static string getAudioEncoder(string selectedEncoder)\r
548         {\r
549             switch (selectedEncoder)\r
550             {\r
551                 case "AAC (faac)":\r
552                     return "faac";\r
553                 case "MP3 (lame)":\r
554                     return "lame";\r
555                 case "Vorbis (vorbis)":\r
556                     return "vorbis";\r
557                 case "AC3 Passthru":\r
558                     return "ac3";\r
559                 case "DTS Passthru":\r
560                     return "dts";\r
561                 default:\r
562                     return "";\r
563             }\r
564         }\r
565         private static Boolean chapterCSVSave(frmMain mainWindow, string filePathName)\r
566         {\r
567             try\r
568             {\r
569                 string csv = "";\r
570 \r
571                 foreach (DataGridViewRow row in mainWindow.data_chpt.Rows)\r
572                 {\r
573                     csv += row.Cells[0].Value.ToString();\r
574                     csv += ",";\r
575                     csv += row.Cells[1].Value.ToString();\r
576                     csv += Environment.NewLine;\r
577                 }\r
578                 StreamWriter file = new StreamWriter(filePathName);\r
579                 file.Write(csv);\r
580                 file.Close();\r
581                 file.Dispose();\r
582                 return true;\r
583             }\r
584             catch (Exception exc)\r
585             {\r
586                 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
587                 return false;\r
588             }\r
589         }\r
590     }\r
591 }