OSDN Git Service

WinGui:
[handbrake-jp/handbrake-jp-git.git] / win / C# / Functions / QueryGenerator.cs
1 /*  QueryGenerator.cs $\r
2     This file is part of the HandBrake source code.\r
3     Homepage: <http://handbrake.fr/>.\r
4     It may be used under the terms of the GNU General Public License. */\r
5 \r
6 namespace Handbrake.Functions\r
7 {\r
8     using System;\r
9     using System.Collections.Generic;\r
10     using System.Globalization;\r
11     using System.IO;\r
12     using System.Windows.Forms;\r
13 \r
14     /// <summary>\r
15     /// Generate a CLI Query for HandBrakeCLI\r
16     /// </summary>\r
17     public class QueryGenerator\r
18     {\r
19         /// <summary>\r
20         /// Generates a full CLI query for either encoding or previe encoeds if duration and preview are defined.\r
21         /// </summary>\r
22         /// <param name="mainWindow">\r
23         /// The Main Window\r
24         /// </param>\r
25         /// <param name="mode">\r
26         /// What Mode. (Point to Point Encoding)  Chapters, Seconds, Frames OR Preview Encode\r
27         /// </param>\r
28         /// <param name="duration">\r
29         /// Time in seconds for preview mode\r
30         /// </param>\r
31         /// <param name="preview">\r
32         /// Preview --start-at-preview (int) \r
33         /// </param>\r
34         /// <returns>\r
35         /// CLI Query \r
36         /// </returns>\r
37         public static string GenerateCliQuery(frmMain mainWindow, int mode, int duration, string preview)\r
38         {\r
39             string query = string.Empty;\r
40 \r
41             if (!string.IsNullOrEmpty(mainWindow.sourcePath) &&\r
42                 mainWindow.sourcePath.Trim() != "Select \"Source\" to continue")\r
43                 query = " -i " + '"' + mainWindow.sourcePath + '"';\r
44 \r
45             if (mainWindow.drp_dvdtitle.Text != string.Empty)\r
46             {\r
47                 string[] titleInfo = mainWindow.drp_dvdtitle.Text.Split(' ');\r
48                 query += " -t " + titleInfo[0];\r
49             }\r
50 \r
51             if (!Properties.Settings.Default.noDvdNav && mainWindow.drop_angle.Items.Count != 0)\r
52                 query += " --angle " + mainWindow.drop_angle.SelectedItem;\r
53 \r
54             // Decide what part of the video we want to encode.\r
55             switch (mode)\r
56             {\r
57                 case 0: // Chapters\r
58                     if (mainWindow.drop_chapterFinish.Text == mainWindow.drop_chapterStart.Text &&\r
59                         mainWindow.drop_chapterStart.Text != string.Empty)\r
60                         query += string.Format(" -c {0}", mainWindow.drop_chapterStart.Text);\r
61                     else if (mainWindow.drop_chapterStart.Text != string.Empty &&\r
62                              mainWindow.drop_chapterFinish.Text != string.Empty)\r
63                         query += string.Format(" -c {0}-{1}", mainWindow.drop_chapterStart.Text, \r
64                                                mainWindow.drop_chapterFinish.Text);\r
65                     break;\r
66                 case 1: // Seconds\r
67                     int start, end;\r
68                     int.TryParse(mainWindow.drop_chapterStart.Text, out start);\r
69                     int.TryParse(mainWindow.drop_chapterFinish.Text, out end);\r
70                     int calculatedDuration = end - start;\r
71 \r
72                     query += string.Format(" --start-at duration:{0} --stop-at duration:{1}", \r
73                                            mainWindow.drop_chapterStart.Text, calculatedDuration);\r
74                     break;\r
75                 case 2: // Frames\r
76                     int.TryParse(mainWindow.drop_chapterStart.Text, out start);\r
77                     int.TryParse(mainWindow.drop_chapterFinish.Text, out end);\r
78                     calculatedDuration = end - start;\r
79 \r
80                     query += string.Format(" --start-at frame:{0} --stop-at frame:{1}", \r
81                                            mainWindow.drop_chapterStart.Text, calculatedDuration);\r
82                     break;\r
83                 case 3: // Preview\r
84                     query += " --previews " + Properties.Settings.Default.previewScanCount + " ";\r
85                     query += " --start-at-preview " + preview;\r
86                     query += " --stop-at duration:" + duration + " ";\r
87 \r
88                     if (mainWindow.text_destination.Text != string.Empty)\r
89                         query += string.Format(" -o \"{0}\" ", \r
90                                                mainWindow.text_destination.Text.Replace(".m", "_sample.m"));\r
91                     break;\r
92                 default:\r
93                     break;\r
94             }\r
95             if (mode != 3)\r
96                 query += string.Format(" -o \"{0}\" ", mainWindow.text_destination.Text);\r
97 \r
98             query += GenerateTabbedComponentsQuery(mainWindow);\r
99 \r
100             return query;\r
101         }\r
102 \r
103         /// <summary>\r
104         /// Generates part of the CLI query, for the tabbed components only.\r
105         /// </summary>\r
106         /// <param name="mainWindow">frmMain the main window</param>\r
107         /// <returns>The CLI Query for the Tab Screens on the main window</returns>\r
108         public static string GenerateTabbedComponentsQuery(frmMain mainWindow)\r
109         {\r
110             string query = string.Empty;\r
111 \r
112             #region Output Settings Box\r
113 \r
114             query += " -f " + mainWindow.drop_format.Text.ToLower().Replace(" file", string.Empty);\r
115 \r
116             // These are output settings features\r
117             if (mainWindow.check_largeFile.Checked)\r
118                 query += " -4 ";\r
119 \r
120             if (mainWindow.check_iPodAtom.Checked)\r
121                 query += " -I ";\r
122 \r
123             if (mainWindow.check_optimiseMP4.Checked)\r
124                 query += " -O ";\r
125 \r
126             #endregion\r
127 \r
128             #region Picture Settings Tab\r
129 \r
130             // Use MaxWidth for built-in presets and width for user settings.\r
131             if (mainWindow.PictureSettings.PresetMaximumResolution.Width == 0)\r
132             {\r
133                 if (mainWindow.PictureSettings.text_width.Value != 0)\r
134                     if (mainWindow.PictureSettings.drp_anamorphic.SelectedIndex != 1)\r
135                         // Prevent usage for strict anamorphic\r
136                         query += " -w " + mainWindow.PictureSettings.text_width.Text;\r
137             }\r
138             else\r
139             {\r
140                 if (mainWindow.PictureSettings.text_width.Value != 0)\r
141                     if (mainWindow.PictureSettings.drp_anamorphic.SelectedIndex != 1)\r
142                         query += " -X " + mainWindow.PictureSettings.text_width.Text;\r
143             }\r
144 \r
145             // Use MaxHeight for built-in presets and height for user settings.\r
146             if (mainWindow.PictureSettings.PresetMaximumResolution.Height == 0)\r
147             {\r
148                 if (mainWindow.PictureSettings.text_height.Value != 0)\r
149                     if (mainWindow.PictureSettings.text_height.Text != string.Empty)\r
150                         if (mainWindow.PictureSettings.drp_anamorphic.SelectedIndex == 0 ||\r
151                             mainWindow.PictureSettings.drp_anamorphic.SelectedIndex == 3)\r
152                             // Prevent usage for strict anamorphic\r
153                             query += " -l " + mainWindow.PictureSettings.text_height.Text;\r
154             }\r
155             else\r
156             {\r
157                 if (mainWindow.PictureSettings.text_height.Value != 0)\r
158                     if (mainWindow.PictureSettings.drp_anamorphic.SelectedIndex == 0 ||\r
159                         mainWindow.PictureSettings.drp_anamorphic.SelectedIndex == 3)\r
160                         query += " -Y " + mainWindow.PictureSettings.text_height.Text;\r
161             }\r
162 \r
163             string cropTop = mainWindow.PictureSettings.crop_top.Text;\r
164             string cropBottom = mainWindow.PictureSettings.crop_bottom.Text;\r
165             string cropLeft = mainWindow.PictureSettings.crop_left.Text;\r
166             string cropRight = mainWindow.PictureSettings.crop_right.Text;\r
167 \r
168             if (mainWindow.PictureSettings.check_customCrop.Checked)\r
169             {\r
170                 if (mainWindow.PictureSettings.crop_top.Text == string.Empty)\r
171                     cropTop = "0";\r
172                 if (mainWindow.PictureSettings.crop_bottom.Text == string.Empty)\r
173                     cropBottom = "0";\r
174                 if (mainWindow.PictureSettings.crop_left.Text == string.Empty)\r
175                     cropLeft = "0";\r
176                 if (mainWindow.PictureSettings.crop_right.Text == string.Empty)\r
177                     cropRight = "0";\r
178 \r
179                 query += " --crop " + cropTop + ":" + cropBottom + ":" + cropLeft + ":" + cropRight;\r
180             }\r
181 \r
182             switch (mainWindow.PictureSettings.drp_anamorphic.SelectedIndex)\r
183             {\r
184                 case 0:\r
185                     if (mainWindow.PictureSettings.drp_modulus.SelectedIndex != 0)\r
186                         query += " --modulus " + mainWindow.PictureSettings.drp_modulus.SelectedItem;\r
187                     break;\r
188                 case 1:\r
189                     query += " --strict-anamorphic ";\r
190                     break;\r
191                 case 2:\r
192                     query += " --loose-anamorphic ";\r
193                     if (mainWindow.PictureSettings.drp_modulus.SelectedIndex != 0)\r
194                         query += " --modulus " + mainWindow.PictureSettings.drp_modulus.SelectedItem;\r
195                     break;\r
196                 case 3:\r
197                     query += " --custom-anamorphic ";\r
198 \r
199                     if (mainWindow.PictureSettings.drp_modulus.SelectedIndex != 0)\r
200                         query += " --modulus " + mainWindow.PictureSettings.drp_modulus.SelectedItem;\r
201 \r
202                     if (mainWindow.PictureSettings.check_KeepAR.Checked)\r
203                         query += " --display-width " + mainWindow.PictureSettings.updownDisplayWidth.Text + " ";\r
204 \r
205                     if (mainWindow.PictureSettings.check_KeepAR.Checked)\r
206                         query += " --keep-display-aspect ";\r
207 \r
208                     if (!mainWindow.PictureSettings.check_KeepAR.Checked)\r
209                         if (mainWindow.PictureSettings.updownParWidth.Text != string.Empty &&\r
210                             mainWindow.PictureSettings.updownParHeight.Text != string.Empty)\r
211                             query += " --pixel-aspect " + mainWindow.PictureSettings.updownParWidth.Text + ":" +\r
212                                      mainWindow.PictureSettings.updownParHeight.Text + " ";\r
213                     break;\r
214             }\r
215 \r
216             #endregion\r
217 \r
218             // Filters Panel\r
219             query += mainWindow.Filters.GetCliQuery;\r
220 \r
221             #region Video Settings Tab\r
222 \r
223             switch (mainWindow.drp_videoEncoder.Text)\r
224             {\r
225                 case "MPEG-4 (FFmpeg)":\r
226                     query += " -e ffmpeg";\r
227                     break;\r
228                 case "H.264 (x264)":\r
229                     query += " -e x264";\r
230                     break;\r
231                 case "VP3 (Theora)":\r
232                     query += " -e theora";\r
233                     break;\r
234                 default:\r
235                     query += " -e x264";\r
236                     break;\r
237             }\r
238 \r
239             // Video Settings\r
240             if (mainWindow.radio_avgBitrate.Checked)\r
241                 query += " -b " + mainWindow.text_bitrate.Text;\r
242 \r
243             if (mainWindow.radio_targetFilesize.Checked)\r
244                 query += " -S " + mainWindow.text_filesize.Text;\r
245 \r
246             // Video Quality Setting\r
247             if (mainWindow.radio_cq.Checked)\r
248             {\r
249                 double cqStep = Properties.Settings.Default.x264cqstep;\r
250                 double value;\r
251                 switch (mainWindow.drp_videoEncoder.Text)\r
252                 {\r
253                     case "MPEG-4 (FFmpeg)":\r
254                         value = 31 - (mainWindow.slider_videoQuality.Value - 1);\r
255                         query += " -q " + value.ToString(new CultureInfo("en-US"));\r
256                         break;\r
257                     case "H.264 (x264)":\r
258                         CultureInfo culture = CultureInfo.CreateSpecificCulture("en-US");\r
259                         value = 51 - mainWindow.slider_videoQuality.Value*cqStep;\r
260                         value = Math.Round(value, 2);\r
261                         query += " -q " + value.ToString(culture);\r
262                         break;\r
263                     case "VP3 (Theora)":\r
264                         value = mainWindow.slider_videoQuality.Value;\r
265                         query += " -q " + value.ToString(new CultureInfo("en-US"));\r
266                         break;\r
267                 }\r
268             }\r
269 \r
270             if (mainWindow.check_2PassEncode.Checked)\r
271                 query += " -2 ";\r
272 \r
273             if (mainWindow.check_turbo.Checked)\r
274                 query += " -T ";\r
275 \r
276             if (mainWindow.drp_videoFramerate.Text != "Same as source")\r
277                 query += " -r " + mainWindow.drp_videoFramerate.Text;\r
278 \r
279             #endregion\r
280 \r
281             #region Audio Settings Tab\r
282 \r
283             ListView audioTracks = mainWindow.AudioSettings.GetAudioPanel();\r
284             List<string> tracks = new List<string>();\r
285             List<string> codecs = new List<string>();\r
286             List<string> mixdowns = new List<string>();\r
287             List<string> samplerates = new List<string>();\r
288             List<string> bitrates = new List<string>();\r
289             List<string> drcs = new List<string>();\r
290 \r
291             // No Audio\r
292             if (audioTracks.Items.Count == 0)\r
293                 query += " -a none ";\r
294 \r
295             // Gather information about each audio track and store them in the declared lists.\r
296             foreach (ListViewItem row in audioTracks.Items)\r
297             {\r
298                 // Audio Track (-a)\r
299                 if (row.SubItems[1].Text == "Automatic")\r
300                     tracks.Add("1");\r
301                 else if (row.Text != "None")\r
302                 {\r
303                     string[] tempSub = row.SubItems[1].Text.Split(' ');\r
304                     tracks.Add(tempSub[0]);\r
305                 }\r
306 \r
307                 // Audio Codec (-E)\r
308                 if (row.SubItems[2].Text != String.Empty)\r
309                     codecs.Add(GetAudioEncoder(row.SubItems[2].Text));\r
310 \r
311                 // Audio Mixdown (-6)\r
312                 if (row.SubItems[3].Text != String.Empty)\r
313                     mixdowns.Add(GetMixDown(row.SubItems[3].Text));\r
314 \r
315                 // Sample Rate (-R)\r
316                 if (row.SubItems[4].Text != String.Empty)\r
317                     samplerates.Add(row.SubItems[4].Text);\r
318 \r
319                 // Audio Bitrate (-B)\r
320                 if (row.SubItems[5].Text != String.Empty)\r
321                     bitrates.Add(row.SubItems[5].Text.Replace("Auto", "auto"));\r
322 \r
323                 // DRC (-D)\r
324                 if (row.SubItems[6].Text != String.Empty)\r
325                     drcs.Add(row.SubItems[6].Text);\r
326             }\r
327 \r
328             // Audio Track (-a)\r
329             string audioItems = string.Empty;\r
330             bool firstLoop = true;\r
331 \r
332             foreach (string item in tracks)\r
333             {\r
334                 if (firstLoop)\r
335                 {\r
336                     audioItems = item;\r
337                     firstLoop = false;\r
338                 }\r
339                 else\r
340                     audioItems += "," + item;\r
341             }\r
342             if (audioItems.Trim() != String.Empty)\r
343                 query += " -a " + audioItems;\r
344             firstLoop = true;\r
345             audioItems = string.Empty; // Reset for another pass.\r
346 \r
347             // Audio Codec (-E)\r
348             foreach (string item in codecs)\r
349             {\r
350                 if (firstLoop)\r
351                 {\r
352                     audioItems = item;\r
353                     firstLoop = false;\r
354                 }\r
355                 else\r
356                     audioItems += "," + item;\r
357             }\r
358             if (audioItems.Trim() != String.Empty)\r
359                 query += " -E " + audioItems;\r
360             firstLoop = true;\r
361             audioItems = string.Empty; // Reset for another pass.\r
362 \r
363             // Audio Mixdown (-6)\r
364             foreach (string item in mixdowns)\r
365             {\r
366                 if (firstLoop)\r
367                 {\r
368                     audioItems = item;\r
369                     firstLoop = false;\r
370                 }\r
371                 else\r
372                     audioItems += "," + item;\r
373             }\r
374             if (audioItems.Trim() != String.Empty)\r
375                 query += " -6 " + audioItems;\r
376             firstLoop = true;\r
377             audioItems = string.Empty; // Reset for another pass.\r
378 \r
379             // Sample Rate (-R)\r
380             foreach (string item in samplerates)\r
381             {\r
382                 if (firstLoop)\r
383                 {\r
384                     audioItems = item;\r
385                     firstLoop = false;\r
386                 }\r
387                 else\r
388                     audioItems += "," + item;\r
389             }\r
390             if (audioItems.Trim() != String.Empty)\r
391                 query += " -R " + audioItems;\r
392             firstLoop = true;\r
393             audioItems = string.Empty; // Reset for another pass.\r
394 \r
395             // Audio Bitrate (-B)\r
396             foreach (string item in bitrates)\r
397             {\r
398                 if (firstLoop)\r
399                 {\r
400                     audioItems = item;\r
401                     firstLoop = false;\r
402                 }\r
403                 else\r
404                     audioItems += "," + item;\r
405             }\r
406             if (audioItems.Trim() != String.Empty)\r
407                 query += " -B " + audioItems;\r
408             firstLoop = true;\r
409             audioItems = string.Empty; // Reset for another pass.\r
410 \r
411             // DRC (-D)\r
412             foreach (var itm in drcs)\r
413             {\r
414                 string item = itm.ToString(new CultureInfo("en-US"));\r
415                 if (firstLoop)\r
416                 {\r
417                     audioItems = item;\r
418                     firstLoop = false;\r
419                 }\r
420                 else\r
421                     audioItems += "," + item;\r
422             }\r
423             if (audioItems.Trim() != String.Empty)\r
424                 query += " -D " + audioItems;\r
425 \r
426             #endregion\r
427 \r
428             // Subtitles Panel\r
429             query += mainWindow.Subtitles.GetCliQuery;\r
430 \r
431             #region Chapter Markers\r
432 \r
433             // Attach Source name and dvd title to the start of the chapters.csv filename.\r
434             // This is for the queue. It allows different chapter name files for each title.\r
435             string[] destName = mainWindow.text_destination.Text.Split('\\');\r
436             string dest_name = destName[destName.Length - 1];\r
437             dest_name = dest_name.Replace("\"", string.Empty);\r
438             dest_name = dest_name.Replace(".mp4", string.Empty).Replace(".m4v", string.Empty).Replace(".mkv", string.Empty);\r
439 \r
440             string sourceTitle = mainWindow.drp_dvdtitle.Text;\r
441             string[] titlesplit = sourceTitle.Split(' ');\r
442             sourceTitle = titlesplit[0];\r
443 \r
444             if (mainWindow.Check_ChapterMarkers.Checked && mainWindow.Check_ChapterMarkers.Enabled)\r
445             {\r
446                 if (dest_name.Trim() != String.Empty)\r
447                 {\r
448                     string path = sourceTitle != "Automatic"\r
449                                       ? Path.Combine(Path.GetTempPath(), dest_name + "-" + sourceTitle + "-chapters.csv")\r
450                                       : Path.Combine(Path.GetTempPath(), dest_name + "-chapters.csv");\r
451 \r
452                     if (ChapterCsvSave(mainWindow, path) == false)\r
453                         query += " -m ";\r
454                     else\r
455                         query += " --markers=" + "\"" + path + "\"";\r
456                 }\r
457                 else\r
458                     query += " -m";\r
459             }\r
460 \r
461             #endregion\r
462 \r
463             // X264 Panel\r
464             query += " -x " + mainWindow.x264Panel.X264Query;\r
465 \r
466             #region Processors / Other\r
467 \r
468             string processors = Properties.Settings.Default.Processors;\r
469             if (processors != "Automatic")\r
470                 query += " -C " + processors + " ";\r
471 \r
472             query += " -v " + Properties.Settings.Default.verboseLevel;\r
473 \r
474             if (Properties.Settings.Default.noDvdNav)\r
475                 query += " --no-dvdnav";\r
476 \r
477             #endregion\r
478 \r
479             return query;\r
480         }\r
481 \r
482         /// <summary>\r
483         /// Return the CLI Mixdown name\r
484         /// </summary>\r
485         /// <param name="selectedAudio">GUI mixdown name</param>\r
486         /// <returns>CLI mixdown name</returns>\r
487         private static string GetMixDown(string selectedAudio)\r
488         {\r
489             switch (selectedAudio)\r
490             {\r
491                 case "Automatic":\r
492                     return "auto";\r
493                 case "Mono":\r
494                     return "mono";\r
495                 case "Stereo":\r
496                     return "stereo";\r
497                 case "Dolby Surround":\r
498                     return "dpl1";\r
499                 case "Dolby Pro Logic II":\r
500                     return "dpl2";\r
501                 case "6 Channel Discrete":\r
502                     return "6ch";\r
503                 default:\r
504                     return "auto";\r
505             }\r
506         }\r
507 \r
508         /// <summary>\r
509         /// Get the CLI Audio Encoder name\r
510         /// </summary>\r
511         /// <param name="selectedEncoder">\r
512         /// String The GUI Encode name\r
513         /// </param>\r
514         /// <returns>\r
515         /// String CLI encoder name\r
516         /// </returns>\r
517         private static string GetAudioEncoder(string selectedEncoder)\r
518         {\r
519             switch (selectedEncoder)\r
520             {\r
521                 case "AAC (faac)":\r
522                     return "faac";\r
523                 case "MP3 (lame)":\r
524                     return "lame";\r
525                 case "Vorbis (vorbis)":\r
526                     return "vorbis";\r
527                 case "AC3 Passthru":\r
528                     return "ac3";\r
529                 case "DTS Passthru":\r
530                     return "dts";\r
531                 default:\r
532                     return string.Empty;\r
533             }\r
534         }\r
535 \r
536         /// <summary>\r
537         /// Create a CSV file with the data from the Main Window Chapters tab\r
538         /// </summary>\r
539         /// <param name="mainWindow">Main Window</param>\r
540         /// <param name="filePathName">Path to save the csv file</param>\r
541         /// <returns>True if successful </returns>\r
542         private static bool ChapterCsvSave(frmMain mainWindow, string filePathName)\r
543         {\r
544             try\r
545             {\r
546                 string csv = string.Empty;\r
547 \r
548                 foreach (DataGridViewRow row in mainWindow.data_chpt.Rows)\r
549                 {\r
550                     csv += row.Cells[0].Value.ToString();\r
551                     csv += ",";\r
552                     csv += row.Cells[1].Value.ToString().Replace(",", "\\,");\r
553                     csv += Environment.NewLine;\r
554                 }\r
555                 StreamWriter file = new StreamWriter(filePathName);\r
556                 file.Write(csv);\r
557                 file.Close();\r
558                 file.Dispose();\r
559                 return true;\r
560             }\r
561             catch (Exception exc)\r
562             {\r
563                 MessageBox.Show(\r
564                     "Unable to save Chapter Makrers file! \nChapter marker names will NOT be saved in your encode \n\n" +\r
565                     exc, "Warning", MessageBoxButtons.OK, MessageBoxIcon.Warning);\r
566                 return false;\r
567             }\r
568         }\r
569     }\r
570 }