henrike@webrtc.org | 28e2075 | 2013-07-10 00:45:36 | [diff] [blame] | 1 | /* |
kjellander | b24317b | 2016-02-10 15:54:43 | [diff] [blame] | 2 | * Copyright 2011 The WebRTC project authors. All Rights Reserved. |
henrike@webrtc.org | 28e2075 | 2013-07-10 00:45:36 | [diff] [blame] | 3 | * |
kjellander | b24317b | 2016-02-10 15:54:43 | [diff] [blame] | 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. |
henrike@webrtc.org | 28e2075 | 2013-07-10 00:45:36 | [diff] [blame] | 9 | */ |
| 10 | |
Mirko Bonadei | 92ea95e | 2017-09-15 04:47:31 | [diff] [blame] | 11 | #include "pc/mediastream.h" |
Yves Gerey | 3e70781 | 2018-11-28 15:47:49 | [diff] [blame^] | 12 | |
| 13 | #include <stddef.h> |
| 14 | #include <vector> |
| 15 | |
Mirko Bonadei | 92ea95e | 2017-09-15 04:47:31 | [diff] [blame] | 16 | #include "rtc_base/checks.h" |
Niels Möller | 84255bb | 2017-10-06 11:43:23 | [diff] [blame] | 17 | #include "rtc_base/refcountedobject.h" |
henrike@webrtc.org | 28e2075 | 2013-07-10 00:45:36 | [diff] [blame] | 18 | |
| 19 | namespace webrtc { |
| 20 | |
| 21 | template <class V> |
Yves Gerey | 665174f | 2018-06-19 13:03:05 | [diff] [blame] | 22 | static typename V::iterator FindTrack(V* vector, const std::string& track_id) { |
henrike@webrtc.org | 28e2075 | 2013-07-10 00:45:36 | [diff] [blame] | 23 | typename V::iterator it = vector->begin(); |
| 24 | for (; it != vector->end(); ++it) { |
| 25 | if ((*it)->id() == track_id) { |
| 26 | break; |
| 27 | } |
| 28 | } |
| 29 | return it; |
Steve Anton | 36b29d1 | 2017-10-30 16:57:42 | [diff] [blame] | 30 | } |
henrike@webrtc.org | 28e2075 | 2013-07-10 00:45:36 | [diff] [blame] | 31 | |
Yves Gerey | 665174f | 2018-06-19 13:03:05 | [diff] [blame] | 32 | rtc::scoped_refptr<MediaStream> MediaStream::Create(const std::string& id) { |
buildbot@webrtc.org | d4e598d | 2014-07-29 17:36:52 | [diff] [blame] | 33 | rtc::RefCountedObject<MediaStream>* stream = |
Seth Hampson | 845e878 | 2018-03-02 19:34:10 | [diff] [blame] | 34 | new rtc::RefCountedObject<MediaStream>(id); |
henrike@webrtc.org | 28e2075 | 2013-07-10 00:45:36 | [diff] [blame] | 35 | return stream; |
| 36 | } |
| 37 | |
Yves Gerey | 665174f | 2018-06-19 13:03:05 | [diff] [blame] | 38 | MediaStream::MediaStream(const std::string& id) : id_(id) {} |
henrike@webrtc.org | 28e2075 | 2013-07-10 00:45:36 | [diff] [blame] | 39 | |
| 40 | bool MediaStream::AddTrack(AudioTrackInterface* track) { |
| 41 | return AddTrack<AudioTrackVector, AudioTrackInterface>(&audio_tracks_, track); |
| 42 | } |
| 43 | |
| 44 | bool MediaStream::AddTrack(VideoTrackInterface* track) { |
| 45 | return AddTrack<VideoTrackVector, VideoTrackInterface>(&video_tracks_, track); |
| 46 | } |
| 47 | |
| 48 | bool MediaStream::RemoveTrack(AudioTrackInterface* track) { |
| 49 | return RemoveTrack<AudioTrackVector>(&audio_tracks_, track); |
| 50 | } |
| 51 | |
| 52 | bool MediaStream::RemoveTrack(VideoTrackInterface* track) { |
| 53 | return RemoveTrack<VideoTrackVector>(&video_tracks_, track); |
| 54 | } |
| 55 | |
Yves Gerey | 665174f | 2018-06-19 13:03:05 | [diff] [blame] | 56 | rtc::scoped_refptr<AudioTrackInterface> MediaStream::FindAudioTrack( |
| 57 | const std::string& track_id) { |
henrike@webrtc.org | 28e2075 | 2013-07-10 00:45:36 | [diff] [blame] | 58 | AudioTrackVector::iterator it = FindTrack(&audio_tracks_, track_id); |
| 59 | if (it == audio_tracks_.end()) |
| 60 | return NULL; |
| 61 | return *it; |
| 62 | } |
| 63 | |
Yves Gerey | 665174f | 2018-06-19 13:03:05 | [diff] [blame] | 64 | rtc::scoped_refptr<VideoTrackInterface> MediaStream::FindVideoTrack( |
| 65 | const std::string& track_id) { |
henrike@webrtc.org | 28e2075 | 2013-07-10 00:45:36 | [diff] [blame] | 66 | VideoTrackVector::iterator it = FindTrack(&video_tracks_, track_id); |
| 67 | if (it == video_tracks_.end()) |
| 68 | return NULL; |
| 69 | return *it; |
| 70 | } |
| 71 | |
| 72 | template <typename TrackVector, typename Track> |
| 73 | bool MediaStream::AddTrack(TrackVector* tracks, Track* track) { |
| 74 | typename TrackVector::iterator it = FindTrack(tracks, track->id()); |
| 75 | if (it != tracks->end()) |
| 76 | return false; |
| 77 | tracks->push_back(track); |
| 78 | FireOnChanged(); |
| 79 | return true; |
| 80 | } |
| 81 | |
| 82 | template <typename TrackVector> |
| 83 | bool MediaStream::RemoveTrack(TrackVector* tracks, |
| 84 | MediaStreamTrackInterface* track) { |
nisse | ede5da4 | 2017-01-12 13:15:36 | [diff] [blame] | 85 | RTC_DCHECK(tracks != NULL); |
henrike@webrtc.org | 28e2075 | 2013-07-10 00:45:36 | [diff] [blame] | 86 | if (!track) |
| 87 | return false; |
| 88 | typename TrackVector::iterator it = FindTrack(tracks, track->id()); |
| 89 | if (it == tracks->end()) |
| 90 | return false; |
| 91 | tracks->erase(it); |
| 92 | FireOnChanged(); |
| 93 | return true; |
| 94 | } |
| 95 | |
| 96 | } // namespace webrtc |