Jon Hjelle | ca91e38 | 2016-01-21 15:36:47 -0800 | [diff] [blame] | 1 | /* |
| 2 | * Copyright 2015 The WebRTC project authors. All Rights Reserved. |
| 3 | * |
| 4 | * Use of this source code is governed by a BSD-style license |
| 5 | * that can be found in the LICENSE file in the root of the source |
| 6 | * tree. An additional intellectual property rights grant can be found |
| 7 | * in the file PATENTS. All contributing project authors may |
| 8 | * be found in the AUTHORS file in the root of the source tree. |
| 9 | */ |
| 10 | |
tkchin | 9eeb624 | 2016-04-27 01:54:20 -0700 | [diff] [blame] | 11 | #import "RTCVideoTrack+Private.h" |
Jon Hjelle | ca91e38 | 2016-01-21 15:36:47 -0800 | [diff] [blame] | 12 | |
tkchin | 9eeb624 | 2016-04-27 01:54:20 -0700 | [diff] [blame] | 13 | #import "NSString+StdString.h" |
| 14 | #import "RTCMediaStreamTrack+Private.h" |
| 15 | #import "RTCPeerConnectionFactory+Private.h" |
| 16 | #import "RTCVideoRendererAdapter+Private.h" |
| 17 | #import "RTCVideoSource+Private.h" |
Jon Hjelle | ca91e38 | 2016-01-21 15:36:47 -0800 | [diff] [blame] | 18 | |
| 19 | @implementation RTCVideoTrack { |
| 20 | NSMutableArray *_adapters; |
| 21 | } |
| 22 | |
| 23 | @synthesize source = _source; |
| 24 | |
| 25 | - (instancetype)initWithFactory:(RTCPeerConnectionFactory *)factory |
| 26 | source:(RTCVideoSource *)source |
| 27 | trackId:(NSString *)trackId { |
| 28 | NSParameterAssert(factory); |
| 29 | NSParameterAssert(source); |
| 30 | NSParameterAssert(trackId.length); |
| 31 | std::string nativeId = [NSString stdStringForString:trackId]; |
| 32 | rtc::scoped_refptr<webrtc::VideoTrackInterface> track = |
| 33 | factory.nativeFactory->CreateVideoTrack(nativeId, |
| 34 | source.nativeVideoSource); |
Peter Hanspers | d9b64cd | 2018-01-12 16:16:18 +0100 | [diff] [blame] | 35 | if (self = [self initWithNativeTrack:track type:RTCMediaStreamTrackTypeVideo]) { |
hjon | a2f7798 | 2016-03-04 07:09:09 -0800 | [diff] [blame] | 36 | _source = source; |
| 37 | } |
| 38 | return self; |
Jon Hjelle | ca91e38 | 2016-01-21 15:36:47 -0800 | [diff] [blame] | 39 | } |
| 40 | |
hjon | a2f7798 | 2016-03-04 07:09:09 -0800 | [diff] [blame] | 41 | - (instancetype)initWithNativeTrack: |
Jon Hjelle | ca91e38 | 2016-01-21 15:36:47 -0800 | [diff] [blame] | 42 | (rtc::scoped_refptr<webrtc::MediaStreamTrackInterface>)nativeMediaTrack |
hjon | a2f7798 | 2016-03-04 07:09:09 -0800 | [diff] [blame] | 43 | type:(RTCMediaStreamTrackType)type { |
Jon Hjelle | ca91e38 | 2016-01-21 15:36:47 -0800 | [diff] [blame] | 44 | NSParameterAssert(nativeMediaTrack); |
| 45 | NSParameterAssert(type == RTCMediaStreamTrackTypeVideo); |
| 46 | if (self = [super initWithNativeTrack:nativeMediaTrack type:type]) { |
| 47 | _adapters = [NSMutableArray array]; |
Jon Hjelle | ca91e38 | 2016-01-21 15:36:47 -0800 | [diff] [blame] | 48 | } |
| 49 | return self; |
| 50 | } |
| 51 | |
| 52 | - (void)dealloc { |
| 53 | for (RTCVideoRendererAdapter *adapter in _adapters) { |
Niels Möller | 8f59762 | 2016-03-23 10:33:07 +0100 | [diff] [blame] | 54 | self.nativeVideoTrack->RemoveSink(adapter.nativeVideoRenderer); |
Jon Hjelle | ca91e38 | 2016-01-21 15:36:47 -0800 | [diff] [blame] | 55 | } |
| 56 | } |
| 57 | |
hjon | a2f7798 | 2016-03-04 07:09:09 -0800 | [diff] [blame] | 58 | - (RTCVideoSource *)source { |
| 59 | if (!_source) { |
perkj | a3ede6c | 2016-03-08 01:27:48 +0100 | [diff] [blame] | 60 | rtc::scoped_refptr<webrtc::VideoTrackSourceInterface> source = |
hjon | a2f7798 | 2016-03-04 07:09:09 -0800 | [diff] [blame] | 61 | self.nativeVideoTrack->GetSource(); |
| 62 | if (source) { |
| 63 | _source = [[RTCVideoSource alloc] initWithNativeVideoSource:source.get()]; |
| 64 | } |
| 65 | } |
| 66 | return _source; |
| 67 | } |
| 68 | |
Jon Hjelle | ca91e38 | 2016-01-21 15:36:47 -0800 | [diff] [blame] | 69 | - (void)addRenderer:(id<RTCVideoRenderer>)renderer { |
| 70 | // Make sure we don't have this renderer yet. |
| 71 | for (RTCVideoRendererAdapter *adapter in _adapters) { |
tkchin | 5fa51e2 | 2016-10-05 13:16:03 -0700 | [diff] [blame] | 72 | if (adapter.videoRenderer == renderer) { |
Jon Hjelle | ca91e38 | 2016-01-21 15:36:47 -0800 | [diff] [blame] | 73 | NSAssert(NO, @"|renderer| is already attached to this track"); |
tkchin | 5fa51e2 | 2016-10-05 13:16:03 -0700 | [diff] [blame] | 74 | return; |
Jon Hjelle | ca91e38 | 2016-01-21 15:36:47 -0800 | [diff] [blame] | 75 | } |
| 76 | } |
| 77 | // Create a wrapper that provides a native pointer for us. |
| 78 | RTCVideoRendererAdapter* adapter = |
| 79 | [[RTCVideoRendererAdapter alloc] initWithNativeRenderer:renderer]; |
| 80 | [_adapters addObject:adapter]; |
Niels Möller | 8f59762 | 2016-03-23 10:33:07 +0100 | [diff] [blame] | 81 | self.nativeVideoTrack->AddOrUpdateSink(adapter.nativeVideoRenderer, |
| 82 | rtc::VideoSinkWants()); |
Jon Hjelle | ca91e38 | 2016-01-21 15:36:47 -0800 | [diff] [blame] | 83 | } |
| 84 | |
| 85 | - (void)removeRenderer:(id<RTCVideoRenderer>)renderer { |
Jon Hjelle | ca91e38 | 2016-01-21 15:36:47 -0800 | [diff] [blame] | 86 | __block NSUInteger indexToRemove = NSNotFound; |
| 87 | [_adapters enumerateObjectsUsingBlock:^(RTCVideoRendererAdapter *adapter, |
| 88 | NSUInteger idx, |
| 89 | BOOL *stop) { |
| 90 | if (adapter.videoRenderer == renderer) { |
| 91 | indexToRemove = idx; |
| 92 | *stop = YES; |
| 93 | } |
| 94 | }]; |
| 95 | if (indexToRemove == NSNotFound) { |
| 96 | return; |
| 97 | } |
hjon | a2f7798 | 2016-03-04 07:09:09 -0800 | [diff] [blame] | 98 | RTCVideoRendererAdapter *adapterToRemove = |
| 99 | [_adapters objectAtIndex:indexToRemove]; |
Niels Möller | 8f59762 | 2016-03-23 10:33:07 +0100 | [diff] [blame] | 100 | self.nativeVideoTrack->RemoveSink(adapterToRemove.nativeVideoRenderer); |
Jon Hjelle | ca91e38 | 2016-01-21 15:36:47 -0800 | [diff] [blame] | 101 | [_adapters removeObjectAtIndex:indexToRemove]; |
| 102 | } |
| 103 | |
| 104 | #pragma mark - Private |
| 105 | |
| 106 | - (rtc::scoped_refptr<webrtc::VideoTrackInterface>)nativeVideoTrack { |
| 107 | return static_cast<webrtc::VideoTrackInterface *>(self.nativeTrack.get()); |
| 108 | } |
| 109 | |
| 110 | @end |