OSDN Git Service

merge 0.9.4 to jp
[handbrake-jp/handbrake-jp.git] / win / C# / Functions / QueryGenerator.cs
diff --git a/win/C#/Functions/QueryGenerator.cs b/win/C#/Functions/QueryGenerator.cs
new file mode 100644 (file)
index 0000000..7491036
--- /dev/null
@@ -0,0 +1,594 @@
+/*  QueryGenerator.cs $\r
+       \r
+          This file is part of the HandBrake source code.\r
+          Homepage: <http://handbrake.fr/>.\r
+          It may be used under the terms of the GNU General Public License. */\r
+\r
+using System;\r
+using System.Windows.Forms;\r
+using System.Globalization;\r
+using System.IO;\r
+using System.Collections.Generic;\r
+\r
+namespace Handbrake.Functions\r
+{\r
+    class QueryGenerator\r
+    {\r
+        /// <summary>\r
+        /// Generates a full CLI query for either encoding or previe encoeds if duration and preview are defined.\r
+        /// </summary>\r
+        /// <param name="mainWindow"></param>\r
+        /// <param name="duration"></param>\r
+        /// <param name="preview"></param>\r
+        /// <returns></returns>\r
+        public string GenerateCLIQuery(frmMain mainWindow, int duration, string preview)\r
+        {\r
+            string query = "";\r
+\r
+            if (!string.IsNullOrEmpty(mainWindow.sourcePath))\r
+                if (mainWindow.sourcePath.Trim() != "Select \"Source\" to continue")\r
+                    query = " -i " + '"' + mainWindow.sourcePath + '"';\r
+\r
+            if (mainWindow.drp_dvdtitle.Text != "")\r
+            {\r
+                string[] titleInfo = mainWindow.drp_dvdtitle.Text.Split(' ');\r
+                query += " -t " + titleInfo[0];\r
+            }\r
+\r
+            if (!Properties.Settings.Default.noDvdNav)\r
+                if (mainWindow.drop_angle.Items.Count != 0)\r
+                    query += " --angle " + mainWindow.drop_angle.SelectedItem;\r
+\r
+\r
+            if (duration != 0 && preview != null) // Preivew Query\r
+            {\r
+                query += " --start-at-preview " + preview;\r
+                query += " --stop-at duration:" + duration + " ";\r
+\r
+                if (mainWindow.text_destination.Text != "")\r
+                    query += " -o " + '"' + mainWindow.text_destination.Text.Replace(".m", "_sample.m") + '"';\r
+            }\r
+            else // Non Preview Query\r
+            {\r
+                if (mainWindow.drop_chapterFinish.Text == mainWindow.drop_chapterStart.Text && mainWindow.drop_chapterStart.Text != "")\r
+                    query += " -c " + mainWindow.drop_chapterStart.Text;\r
+                else if (mainWindow.drop_chapterStart.Text == "Auto" && mainWindow.drop_chapterFinish.Text != "Auto")\r
+                    query += " -c " + "0-" + mainWindow.drop_chapterFinish.Text;\r
+                else if (mainWindow.drop_chapterStart.Text != "Auto" && mainWindow.drop_chapterFinish.Text != "Auto" && mainWindow.drop_chapterStart.Text != "")\r
+                    query += " -c " + mainWindow.drop_chapterStart.Text + "-" + mainWindow.drop_chapterFinish.Text;\r
+\r
+                if (mainWindow.text_destination.Text != "")\r
+                    query += " -o " + '"' + mainWindow.text_destination.Text + '"';\r
+            }\r
+\r
+            query += GenerateTabbedComponentsQuery(mainWindow);\r
+\r
+            return query;\r
+        }\r
+\r
+        /// <summary>\r
+        /// Generates part of the CLI query, for the tabbed components only.\r
+        /// </summary>\r
+        /// <param name="mainWindow"></param>\r
+        /// <returns></returns>\r
+        public static string GenerateTabbedComponentsQuery(frmMain mainWindow)\r
+        {\r
+            string query = "";\r
+\r
+            #region Output Settings Box\r
+            query += " -f " + mainWindow.drop_format.Text.ToLower().Replace(" file", "");\r
+\r
+            // These are output settings features\r
+            if (mainWindow.check_largeFile.Checked)\r
+                query += " -4 ";\r
+\r
+            if (mainWindow.check_iPodAtom.Checked)\r
+                query += " -I ";\r
+\r
+            if (mainWindow.check_optimiseMP4.Checked)\r
+                query += " -O ";\r
+            #endregion\r
+\r
+            #region Picture Settings Tab\r
+\r
+            // Use MaxWidth for built-in presets and width for user settings.\r
+            if (mainWindow.PictureSettings.PresetMaximumResolution.Width == 0)\r
+            {\r
+                if (mainWindow.PictureSettings.text_width.Value != 0)\r
+                    if (mainWindow.PictureSettings.drp_anamorphic.SelectedIndex != 1) // Prevent usage for strict anamorphic\r
+                        query += " -w " + mainWindow.PictureSettings.text_width.Text;\r
+            }\r
+            else\r
+            {\r
+                if (mainWindow.PictureSettings.text_width.Value != 0)\r
+                    if (mainWindow.PictureSettings.drp_anamorphic.SelectedIndex != 1)\r
+                        query += " -X " + mainWindow.PictureSettings.text_width.Text;\r
+            }\r
+\r
+            // Use MaxHeight for built-in presets and height for user settings.\r
+            if (mainWindow.PictureSettings.PresetMaximumResolution.Height == 0)\r
+            {\r
+                if (mainWindow.PictureSettings.text_height.Value != 0)\r
+                    if (mainWindow.PictureSettings.text_height.Text != "")\r
+                        if (mainWindow.PictureSettings.drp_anamorphic.SelectedIndex == 0 || mainWindow.PictureSettings.drp_anamorphic.SelectedIndex == 3) // Prevent usage for strict anamorphic\r
+                            query += " -l " + mainWindow.PictureSettings.text_height.Text;\r
+            }\r
+            else\r
+            {\r
+                if (mainWindow.PictureSettings.text_height.Value != 0)\r
+                    if (mainWindow.PictureSettings.drp_anamorphic.SelectedIndex == 0 || mainWindow.PictureSettings.drp_anamorphic.SelectedIndex == 3)\r
+                        query += " -Y " + mainWindow.PictureSettings.text_height.Text;\r
+            }\r
+\r
+            string cropTop = mainWindow.PictureSettings.crop_top.Text;\r
+            string cropBottom = mainWindow.PictureSettings.crop_bottom.Text;\r
+            string cropLeft = mainWindow.PictureSettings.crop_left.Text;\r
+            string cropRight = mainWindow.PictureSettings.crop_right.Text;\r
+\r
+            if (mainWindow.PictureSettings.check_customCrop.Checked)\r
+            {\r
+                if (mainWindow.PictureSettings.crop_top.Text == string.Empty)\r
+                    cropTop = "0";\r
+                if (mainWindow.PictureSettings.crop_bottom.Text == string.Empty)\r
+                    cropBottom = "0";\r
+                if (mainWindow.PictureSettings.crop_left.Text == string.Empty)\r
+                    cropLeft = "0";\r
+                if (mainWindow.PictureSettings.crop_right.Text == string.Empty)\r
+                    cropRight = "0";\r
+\r
+                query += " --crop " + cropTop + ":" + cropBottom + ":" + cropLeft + ":" + cropRight;\r
+            }\r
+\r
+            switch (mainWindow.PictureSettings.drp_anamorphic.SelectedIndex)\r
+            {\r
+                case 1:\r
+                    query += " --strict-anamorphic ";\r
+                    break;\r
+                case 2:\r
+                    query += " --loose-anamorphic ";\r
+                    break;\r
+                case 3:\r
+                    query += " --custom-anamorphic ";\r
+\r
+                    if (mainWindow.PictureSettings.drp_modulus.SelectedIndex != 0)\r
+                        query += " --modulus " + mainWindow.PictureSettings.drp_modulus.SelectedItem;\r
+\r
+                    if (mainWindow.PictureSettings.check_KeepAR.Checked)\r
+                        query += " --display-width " + mainWindow.PictureSettings.updownDisplayWidth.Text + " ";\r
+\r
+                    if (mainWindow.PictureSettings.check_KeepAR.Checked)\r
+                        query += " --keep-display-aspect ";\r
+\r
+                    if (!mainWindow.PictureSettings.check_KeepAR.Checked)\r
+                        if (mainWindow.PictureSettings.updownParWidth.Text != "" && mainWindow.PictureSettings.updownParHeight.Text != "")\r
+                            query += " --pixel-aspect " + mainWindow.PictureSettings.updownParWidth.Text + ":" + mainWindow.PictureSettings.updownParHeight.Text + " ";\r
+                    break;\r
+            }\r
+            #endregion\r
+\r
+            #region Filters\r
+            query += mainWindow.Filters.getCLIQuery;\r
+            #endregion\r
+\r
+            #region Video Settings Tab\r
+\r
+            switch (mainWindow.drp_videoEncoder.Text)\r
+            {\r
+                case "MPEG-4 (FFmpeg)":\r
+                    query += " -e ffmpeg";\r
+                    break;\r
+                case "H.264 (x264)":\r
+                    query += " -e x264";\r
+                    break;\r
+                case "VP3 (Theora)":\r
+                    query += " -e theora";\r
+                    break;\r
+                default:\r
+                    query += " -e x264";\r
+                    break;\r
+            }\r
+\r
+            // Video Settings\r
+            if (mainWindow.radio_avgBitrate.Checked)\r
+                query += " -b " + mainWindow.text_bitrate.Text;\r
+\r
+            if (mainWindow.radio_targetFilesize.Checked)\r
+                query += " -S " + mainWindow.text_filesize.Text;\r
+\r
+            // Video Quality Setting\r
+            if (mainWindow.radio_cq.Checked)\r
+            {\r
+                double cqStep = Properties.Settings.Default.x264cqstep;\r
+                double value;\r
+                switch (mainWindow.drp_videoEncoder.Text)\r
+                {\r
+                    case "MPEG-4 (FFmpeg)":\r
+                        value = 31 - (mainWindow.slider_videoQuality.Value - 1);\r
+                        query += " -q " + value.ToString(new CultureInfo("en-US"));\r
+                        break;\r
+                    case "H.264 (x264)":\r
+                        CultureInfo culture = CultureInfo.CreateSpecificCulture("en-US");\r
+                        value = 51 - mainWindow.slider_videoQuality.Value * cqStep;\r
+                        value = Math.Round(value, 2);\r
+                        query += " -q " + value.ToString(culture);\r
+                        break;\r
+                    case "VP3 (Theora)":\r
+                        value = mainWindow.slider_videoQuality.Value;\r
+                        query += " -q " + value.ToString(new CultureInfo("en-US"));\r
+                        break;\r
+                }\r
+            }\r
+\r
+            if (mainWindow.check_2PassEncode.Checked)\r
+                query += " -2 ";\r
+\r
+            if (mainWindow.check_turbo.Checked)\r
+                query += " -T ";\r
+\r
+            if (mainWindow.drp_videoFramerate.Text != "Same as source")\r
+                query += " -r " + mainWindow.drp_videoFramerate.Text;\r
+            #endregion\r
+\r
+            #region Audio Settings Tab\r
+\r
+            ListView audioTracks = mainWindow.AudioSettings.GetAudioPanel();\r
+            List<string> tracks = new List<string>();\r
+            List<string> codecs = new List<string>();\r
+            List<string> mixdowns = new List<string>();\r
+            List<string> samplerates = new List<string>();\r
+            List<string> bitrates = new List<string>();\r
+            List<string> drcs = new List<string>();\r
+\r
+            // No Audio\r
+            if (audioTracks.Items.Count == 0)\r
+                query += " -a none ";\r
+\r
+            // Gather information about each audio track and store them in the declared lists.\r
+            foreach (ListViewItem row in audioTracks.Items)\r
+            {\r
+                // Audio Track (-a)\r
+                if (row.SubItems[1].Text == "Automatic")\r
+                    tracks.Add("1");\r
+                else if (row.Text != "None")\r
+                {\r
+                    string[] tempSub = row.SubItems[1].Text.Split(' ');\r
+                    tracks.Add(tempSub[0]);\r
+                }\r
+\r
+                // Audio Codec (-E)\r
+                if (row.SubItems[2].Text != String.Empty)\r
+                    codecs.Add(getAudioEncoder(row.SubItems[2].Text));\r
+\r
+                // Audio Mixdown (-6)\r
+                if (row.SubItems[3].Text != String.Empty)\r
+                    mixdowns.Add(getMixDown(row.SubItems[3].Text));\r
+\r
+                // Sample Rate (-R)\r
+                if (row.SubItems[4].Text != String.Empty)\r
+                    samplerates.Add(row.SubItems[4].Text.Replace("Auto", "Auto"));\r
+\r
+                // Audio Bitrate (-B)\r
+                if (row.SubItems[5].Text != String.Empty)\r
+                    bitrates.Add(row.SubItems[5].Text.Replace("Auto", "auto"));\r
+\r
+                // DRC (-D)\r
+                if (row.SubItems[6].Text != String.Empty)\r
+                    drcs.Add(row.SubItems[6].Text);\r
+            }\r
+\r
+            // Audio Track (-a)\r
+            String audioItems = "";\r
+            Boolean firstLoop = true;\r
+\r
+            foreach (String item in tracks)\r
+            {\r
+                if (firstLoop)\r
+                {\r
+                    audioItems = item; firstLoop = false;\r
+                }\r
+                else\r
+                    audioItems += "," + item;\r
+            }\r
+            if (audioItems.Trim() != String.Empty)\r
+                query += " -a " + audioItems;\r
+            firstLoop = true; audioItems = ""; // Reset for another pass.\r
+\r
+            // Audio Codec (-E)\r
+            foreach (String item in codecs)\r
+            {\r
+\r
+                if (firstLoop)\r
+                {\r
+                    audioItems = item; firstLoop = false;\r
+                }\r
+                else\r
+                    audioItems += "," + item;\r
+            }\r
+            if (audioItems.Trim() != String.Empty)\r
+                query += " -E " + audioItems;\r
+            firstLoop = true; audioItems = ""; // Reset for another pass.\r
+\r
+            // Audio Mixdown (-6)\r
+            foreach (String item in mixdowns)\r
+            {\r
+                if (firstLoop)\r
+                {\r
+                    audioItems = item; firstLoop = false;\r
+                }\r
+                else\r
+                    audioItems += "," + item;\r
+            }\r
+            if (audioItems.Trim() != String.Empty)\r
+                query += " -6 " + audioItems;\r
+            firstLoop = true; audioItems = ""; // Reset for another pass.\r
+\r
+            // Sample Rate (-R)\r
+            foreach (String item in samplerates)\r
+            {\r
+                if (firstLoop)\r
+                {\r
+                    audioItems = item; firstLoop = false;\r
+                }\r
+                else\r
+                    audioItems += "," + item;\r
+            }\r
+            if (audioItems.Trim() != String.Empty)\r
+                query += " -R " + audioItems;\r
+            firstLoop = true; audioItems = ""; // Reset for another pass.\r
+\r
+            // Audio Bitrate (-B)\r
+            foreach (String item in bitrates)\r
+            {\r
+                if (firstLoop)\r
+                {\r
+                    audioItems = item; firstLoop = false;\r
+                }\r
+                else\r
+                    audioItems += "," + item;\r
+            }\r
+            if (audioItems.Trim() != String.Empty)\r
+                query += " -B " + audioItems;\r
+            firstLoop = true; audioItems = ""; // Reset for another pass.\r
+\r
+            // DRC (-D)\r
+            foreach (var itm in drcs)\r
+            {\r
+                string item = itm.ToString(new CultureInfo("en-US"));\r
+                if (firstLoop)\r
+                {\r
+                    audioItems = item; firstLoop = false;\r
+                }\r
+                else\r
+                    audioItems += "," + item;\r
+            }\r
+            if (audioItems.Trim() != String.Empty)\r
+                query += " -D " + audioItems;\r
+\r
+            #endregion\r
+\r
+            #region Subtitles Tab\r
+            if (mainWindow.Subtitles.lv_subList.Items.Count != 0) // If we have subtitle tracks\r
+            {\r
+                IDictionary<string, string> langMap = Main.mapLanguages();\r
+\r
+                // BitMap and CC's\r
+                string subtitleTracks = String.Empty;\r
+                string subtitleForced = String.Empty;\r
+                string subtitleBurn = String.Empty;\r
+                string subtitleDefault = String.Empty;\r
\r
+                // SRT\r
+                string srtFile = String.Empty;\r
+                string srtCodeset = String.Empty;\r
+                string srtOffset = String.Empty;\r
+                string srtLang = String.Empty;\r
+                string srtDefault = String.Empty;\r
+                int srtCount = 0;\r
+\r
+                List<Controls.SubtitleInfo> SubList = mainWindow.Subtitles.GetSubtitleInfoList();\r
+\r
+                foreach (var item in SubList)\r
+                {\r
+                    string itemToAdd, trackID;\r
+\r
+                    if (item.SrtPath != "-") // We have an SRT file\r
+                    {\r
+                        srtCount++; // SRT track id.\r
+\r
+                        srtLang += srtLang == "" ? langMap[item.SrtLang] : "," + langMap[item.SrtLang];\r
+                        srtCodeset += srtCodeset == "" ? item.SrtCharCode : "," + item.SrtCharCode;\r
+\r
+                        if (item.Default == "Yes") // default\r
+                            srtDefault = srtCount.ToString();\r
+\r
+                        itemToAdd = item.SrtPath;\r
+                        srtFile += srtFile == "" ? itemToAdd : "," + itemToAdd;\r
+\r
+                        itemToAdd = item.SrtOffset.ToString();\r
+                        srtOffset += srtOffset == "" ? itemToAdd : "," + itemToAdd;\r
+                    }\r
+                    else // We have Bitmap or CC\r
+                    {\r
+                        string[] tempSub;\r
+\r
+                        // Find --subtitle <string>\r
+                        if (item.Track.Contains("Foreign Audio Search"))\r
+                            itemToAdd = "scan";\r
+                        else\r
+                        {\r
+                            tempSub = item.Track.Split(' ');\r
+                            itemToAdd = tempSub[0];\r
+                        }\r
+\r
+                        subtitleTracks += subtitleTracks == "" ? itemToAdd : "," + itemToAdd;\r
+\r
+                        // Find --subtitle-forced\r
+                        itemToAdd = "";\r
+                        tempSub = item.Track.Split(' ');\r
+                        trackID = tempSub[0];\r
+\r
+                        if (item.Forced == "Yes")\r
+                            itemToAdd = trackID;\r
+\r
+                        if (itemToAdd != "")\r
+                            subtitleForced += subtitleForced == "" ? itemToAdd : "," + itemToAdd;\r
+\r
+                        // Find --subtitle-burn and --subtitle-default\r
+                        trackID = tempSub[0];\r
+\r
+                        if (trackID.Trim() == "Foreign")\r
+                            trackID = "scan";\r
+\r
+                        if (item.Burned == "Yes") // burn\r
+                            subtitleBurn = trackID;\r
+\r
+                        if (item.Default == "Yes") // default\r
+                            subtitleDefault = trackID;\r
+                    }\r
+                }\r
+\r
+                // Build The CLI Subtitles Query\r
+                if (subtitleTracks != "")\r
+                {\r
+                    query += " --subtitle " + subtitleTracks;\r
+\r
+                    if (subtitleForced != "")\r
+                        query += " --subtitle-forced=" + subtitleForced;\r
+                    if (subtitleBurn != "")\r
+                        query += " --subtitle-burn=" + subtitleBurn;\r
+                    if (subtitleDefault != "")\r
+                        query += " --subtitle-default=" + subtitleDefault;\r
+                }\r
+\r
+                if (srtFile != "") // SRTs\r
+                {\r
+                    query += " --srt-file " + "\"" + srtFile + "\"";\r
+\r
+                    if (srtCodeset != "")\r
+                        query += " --srt-codeset " + srtCodeset;\r
+                    if (srtOffset != "")\r
+                        query += " --srt-offset " + srtOffset;\r
+                    if (srtLang != "")\r
+                        query += " --srt-lang " + srtLang;\r
+                    if (srtDefault != "")\r
+                        query += " --srt-default=" + srtDefault;\r
+                }\r
+\r
+            }\r
+            #endregion\r
+\r
+            #region Chapter Markers\r
+\r
+            // Attach Source name and dvd title to the start of the chapters.csv filename.\r
+            // This is for the queue. It allows different chapter name files for each title.\r
+            string[] destName = mainWindow.text_destination.Text.Split('\\');\r
+            string dest_name = destName[destName.Length - 1];\r
+            dest_name = dest_name.Replace("\"", "");\r
+            dest_name = dest_name.Replace(".mp4", "").Replace(".m4v", "").Replace(".mkv", "");\r
+\r
+            string source_title = mainWindow.drp_dvdtitle.Text;\r
+            string[] titlesplit = source_title.Split(' ');\r
+            source_title = titlesplit[0];\r
+\r
+            if (mainWindow.Check_ChapterMarkers.Checked && mainWindow.Check_ChapterMarkers.Enabled)\r
+            {\r
+                if (dest_name.Trim() != String.Empty)\r
+                {\r
+                    string path = source_title != "Automatic"\r
+                                      ? Path.Combine(Path.GetTempPath(), dest_name + "-" + source_title + "-chapters.csv")\r
+                                      : Path.Combine(Path.GetTempPath(), dest_name + "-chapters.csv");\r
+\r
+                    if (chapterCSVSave(mainWindow, path) == false)\r
+                        query += " -m ";\r
+                    else\r
+                        query += " --markers=" + "\"" + path + "\"";\r
+                }\r
+                else\r
+                    query += " -m";\r
+            }\r
+            #endregion\r
+\r
+            #region  H264 Tab\r
+            if (mainWindow.x264Panel.x264Query != "")\r
+                query += " -x " + mainWindow.x264Panel.x264Query;\r
+            #endregion\r
+\r
+            #region Processors / Other\r
+            string processors = Properties.Settings.Default.Processors;\r
+            if (processors != "Automatic")\r
+                query += " -C " + processors + " ";\r
+\r
+            query += " -v " + Properties.Settings.Default.verboseLevel;\r
+\r
+            if (Properties.Settings.Default.noDvdNav)\r
+                query += " --no-dvdnav";\r
+            #endregion\r
+\r
+            return query;\r
+        }\r
+\r
+        private static string getMixDown(string selectedAudio)\r
+        {\r
+            switch (selectedAudio)\r
+            {\r
+                case "Automatic":\r
+                    return "auto";\r
+                case "Mono":\r
+                    return "mono";\r
+                case "Stereo":\r
+                    return "stereo";\r
+                case "Dolby Surround":\r
+                    return "dpl1";\r
+                case "Dolby Pro Logic II":\r
+                    return "dpl2";\r
+                case "6 Channel Discrete":\r
+                    return "6ch";\r
+                default:\r
+                    return "auto";\r
+            }\r
+        }\r
+        private static string getAudioEncoder(string selectedEncoder)\r
+        {\r
+            switch (selectedEncoder)\r
+            {\r
+                case "AAC (faac)":\r
+                    return "faac";\r
+                case "MP3 (lame)":\r
+                    return "lame";\r
+                case "Vorbis (vorbis)":\r
+                    return "vorbis";\r
+                case "AC3 Passthru":\r
+                    return "ac3";\r
+                case "DTS Passthru":\r
+                    return "dts";\r
+                default:\r
+                    return "";\r
+            }\r
+        }\r
+        private static Boolean chapterCSVSave(frmMain mainWindow, string filePathName)\r
+        {\r
+            try\r
+            {\r
+                string csv = "";\r
+\r
+                foreach (DataGridViewRow row in mainWindow.data_chpt.Rows)\r
+                {\r
+                    csv += row.Cells[0].Value.ToString();\r
+                    csv += ",";\r
+                    csv += row.Cells[1].Value.ToString();\r
+                    csv += Environment.NewLine;\r
+                }\r
+                StreamWriter file = new StreamWriter(filePathName);\r
+                file.Write(csv);\r
+                file.Close();\r
+                file.Dispose();\r
+                return true;\r
+            }\r
+            catch (Exception exc)\r
+            {\r
+                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
+                return false;\r
+            }\r
+        }\r
+    }\r
+}
\ No newline at end of file