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