SGTrack: Add get functions.

nickarora · Nov 13, 2019 · c3a752b · c3a752b
1 parent e1cb725
commit c3a752b
Show file tree

Hide file tree

Showing 2 changed files with 51 additions and 0 deletions.
diff --git a/SGPlayer/Classes/Core/SGAsset/SGTrack.h b/SGPlayer/Classes/Core/SGAsset/SGTrack.h
@@ -9,6 +9,26 @@
 #import <Foundation/Foundation.h>
 #import "SGDefines.h"
 
+@class SGTrack;
+
+/*!
+ @abstract
+   Get track with index.
+*/
+SGTrack *SGTrackWithIndex(NSArray<SGTrack *> *tracks, NSInteger index);
+
+/*!
+ @abstract
+   Get track with media type.
+*/
+SGTrack *SGTrackWithType(NSArray<SGTrack *> *tracks, SGMediaType type);
+
+/*!
+ @abstract
+   Get tracks with media types.
+*/
+NSArray<SGTrack *> *SGTracksWithType(NSArray<SGTrack *> *tracks, SGMediaType type);
+
 @interface SGTrack : NSObject <NSCopying>
 
 + (instancetype)new NS_UNAVAILABLE;

diff --git a/SGPlayer/Classes/Core/SGAsset/SGTrack.m b/SGPlayer/Classes/Core/SGAsset/SGTrack.m
@@ -9,6 +9,37 @@
 #import "SGTrack.h"
 #import "SGTrack+Internal.h"
 
+SGTrack *SGTrackWithIndex(NSArray<SGTrack *> *tracks, NSInteger index)
+{
+    for (SGTrack *obj in tracks) {
+        if (obj.index == index) {
+            return obj;
+        }
+    }
+    return nil;
+}
+
+SGTrack *SGTrackWithType(NSArray<SGTrack *> *tracks, SGMediaType type)
+{
+    for (SGTrack *obj in tracks) {
+        if (obj.type == type) {
+            return obj;
+        }
+    }
+    return nil;
+}
+
+NSArray<SGTrack *> *SGTracksWithType(NSArray<SGTrack *> *tracks, SGMediaType type)
+{
+    NSMutableArray *array = [NSMutableArray array];
+    for (SGTrack *obj in tracks) {
+        if (obj.type == type) {
+            [array addObject:obj];
+        }
+    }
+    return array.count ? [array copy] : nil;
+}
+
 @implementation SGTrack
 
 - (id)copyWithZone:(NSZone *)zone