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