aboutsummaryrefslogtreecommitdiff
path: root/MediaBrowser.Model/MediaInfo/MediaStreamSelector.cs
blob: d498c3a32a810d55f9055e5c8796c601ea2d8154 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
using MediaBrowser.Model.Configuration;
using MediaBrowser.Model.Dto;
using MediaBrowser.Model.Entities;
using System;
using System.Collections.Generic;
using System.Linq;

namespace MediaBrowser.Model.MediaInfo
{
    public static class MediaStreamSelector
    {
        public static int? GetDefaultAudioStreamIndex(MediaSourceInfo mediaSource, IEnumerable<string> preferredLanguages, bool preferDefaultTrack)
        {
            var streams = GetSortedStreams(mediaSource.MediaStreams, MediaStreamType.Audio, preferredLanguages.FirstOrDefault())
                .ToList();

            if (preferDefaultTrack)
            {
                var defaultStream = streams.FirstOrDefault(i => i.IsDefault);

                if (defaultStream != null)
                {
                    return defaultStream.Index;
                }
            }

            var stream = streams.FirstOrDefault();

            if (stream != null)
            {
                return stream.Index;
            }

            return null;
        }

        public static int? GetDefaultSubtitleStreamIndex(MediaSourceInfo mediaSource, 
            IEnumerable<string> preferredLanguages, 
            SubtitlePlaybackMode mode,
            string audioTrackLanguage)
        {
            var streams = GetSortedStreams(mediaSource.MediaStreams, MediaStreamType.Subtitle, preferredLanguages.FirstOrDefault())
                .ToList();

            MediaStream stream = null;

            if (mode == SubtitlePlaybackMode.Default)
            {
                stream = streams.FirstOrDefault(i => i.IsDefault);
            }
            else if (mode == SubtitlePlaybackMode.Always)
            {
                stream = streams.FirstOrDefault(i => i.IsDefault) ??
                    streams.FirstOrDefault();
            }
            else if (mode == SubtitlePlaybackMode.OnlyForced)
            {
                stream = streams.FirstOrDefault(i => i.IsForced);
            }

            if (stream != null)
            {
                return stream.Index;
            }

            return null;
        }
        
        private static IEnumerable<MediaStream> GetSortedStreams(IEnumerable<MediaStream> streams, MediaStreamType type, string defaultLanguage)
        {
            var orderStreams = streams
                .Where(i => i.Type == type);

            // For subs give a preference to text for performance

            if (string.IsNullOrEmpty(defaultLanguage))
            {
                return orderStreams.OrderBy(i => i.IsDefault)
                    .ThenBy(i => !i.IsGraphicalSubtitleStream)
                    .ThenBy(i => i.Index)
                    .ToList();
            }

            return orderStreams.OrderBy(i => string.Equals(i.Language, defaultLanguage, StringComparison.OrdinalIgnoreCase))
                .ThenBy(i => i.IsDefault)
                .ThenBy(i => !i.IsGraphicalSubtitleStream)
                .ThenBy(i => i.Index)
                .ToList();
        }
    }
}