added -vq and -aq flags to select quality

Milek7 · Jun 20, 2014 · 32635fc · 32635fc
1 parent fd17a09
commit 32635fc
Show file tree

Hide file tree

Showing 2 changed files with 40 additions and 2 deletions.
diff --git a/src/Program.cs b/src/Program.cs
@@ -40,6 +40,9 @@ public void Process(IList<string> inputUrls, IList<string> outputUrls)
 
     internal class MainClass
     {
+        public static int audioQuality = 0; //TODO: Move audioQuality and videoQuality somewhere else.
+        public static int videoQuality = 0;
+
         private static void Main(string[] args)
         {
             Logo();
@@ -60,6 +63,22 @@ private static void Main(string[] args)
                 {
                     isDeterministic = true;
                 }
+                else if (args[j] == "--vq")
+                {
+                    ++j;
+                    if (j <= args.Length)
+                    {
+                        videoQuality = Convert.ToInt32(args[j]);
+                    }
+                }
+                else if (args[j] == "--aq")
+                {
+                    ++j;
+                    if (j <= args.Length)
+                    {
+                        audioQuality = Convert.ToInt32(args[j]);
+                    }
+                }
             }
             if (args.Length < j + 2)
             {

diff --git a/src/Smooth.cs b/src/Smooth.cs
@@ -200,9 +200,28 @@ public StreamInfo(XmlNode element, Uri manifestUri)
             this.pureUrl = this.pureUrl.Substring(0, this.pureUrl.LastIndexOf('/'));
             this.SelectedTracks = new List<TrackInfo>();
 
-            if (this.AvailableTracks.Count > 0)
+            if (this.AvailableTracks[0] is AudioTrackInfo)
             {
-                this.SelectedTracks.Add(this.AvailableTracks[0]);
+                if (this.AvailableTracks.Count >= MainClass.audioQuality)
+                {
+                    this.SelectedTracks.Add(this.AvailableTracks[MainClass.audioQuality]);
+                }
+                else
+                {
+                    this.SelectedTracks.Add(this.AvailableTracks[0]);
+                }
+            }
+
+            if (this.AvailableTracks[0] is VideoTrackInfo)
+            {
+                if (this.AvailableTracks.Count >= MainClass.videoQuality)
+                {
+                    this.SelectedTracks.Add(this.AvailableTracks[MainClass.videoQuality]);
+                }
+                else
+                {
+                    this.SelectedTracks.Add(this.AvailableTracks[0]);
+                }
             }
         }
         private void CheckUrlAttribute()