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