98 lines
2.7 KiB
C++
98 lines
2.7 KiB
C++
/*
|
|
* Copyright 2011 The WebRTC project authors. All Rights Reserved.
|
|
*
|
|
* Use of this source code is governed by a BSD-style license
|
|
* that can be found in the LICENSE file in the root of the source
|
|
* tree. An additional intellectual property rights grant can be found
|
|
* in the file PATENTS. All contributing project authors may
|
|
* be found in the AUTHORS file in the root of the source tree.
|
|
*/
|
|
|
|
#include "pc/media_stream.h"
|
|
|
|
#include <stddef.h>
|
|
|
|
#include <vector>
|
|
|
|
#include "rtc_base/checks.h"
|
|
#include "rtc_base/ref_counted_object.h"
|
|
|
|
namespace webrtc {
|
|
|
|
template <class V>
|
|
static typename V::iterator FindTrack(V* vector, const std::string& track_id) {
|
|
typename V::iterator it = vector->begin();
|
|
for (; it != vector->end(); ++it) {
|
|
if ((*it)->id() == track_id) {
|
|
break;
|
|
}
|
|
}
|
|
return it;
|
|
}
|
|
|
|
rtc::scoped_refptr<MediaStream> MediaStream::Create(const std::string& id) {
|
|
rtc::RefCountedObject<MediaStream>* stream =
|
|
new rtc::RefCountedObject<MediaStream>(id);
|
|
return stream;
|
|
}
|
|
|
|
MediaStream::MediaStream(const std::string& id) : id_(id) {}
|
|
|
|
bool MediaStream::AddTrack(AudioTrackInterface* track) {
|
|
return AddTrack<AudioTrackVector, AudioTrackInterface>(&audio_tracks_, track);
|
|
}
|
|
|
|
bool MediaStream::AddTrack(VideoTrackInterface* track) {
|
|
return AddTrack<VideoTrackVector, VideoTrackInterface>(&video_tracks_, track);
|
|
}
|
|
|
|
bool MediaStream::RemoveTrack(AudioTrackInterface* track) {
|
|
return RemoveTrack<AudioTrackVector>(&audio_tracks_, track);
|
|
}
|
|
|
|
bool MediaStream::RemoveTrack(VideoTrackInterface* track) {
|
|
return RemoveTrack<VideoTrackVector>(&video_tracks_, track);
|
|
}
|
|
|
|
rtc::scoped_refptr<AudioTrackInterface> MediaStream::FindAudioTrack(
|
|
const std::string& track_id) {
|
|
AudioTrackVector::iterator it = FindTrack(&audio_tracks_, track_id);
|
|
if (it == audio_tracks_.end())
|
|
return NULL;
|
|
return *it;
|
|
}
|
|
|
|
rtc::scoped_refptr<VideoTrackInterface> MediaStream::FindVideoTrack(
|
|
const std::string& track_id) {
|
|
VideoTrackVector::iterator it = FindTrack(&video_tracks_, track_id);
|
|
if (it == video_tracks_.end())
|
|
return NULL;
|
|
return *it;
|
|
}
|
|
|
|
template <typename TrackVector, typename Track>
|
|
bool MediaStream::AddTrack(TrackVector* tracks, Track* track) {
|
|
typename TrackVector::iterator it = FindTrack(tracks, track->id());
|
|
if (it != tracks->end())
|
|
return false;
|
|
tracks->push_back(track);
|
|
FireOnChanged();
|
|
return true;
|
|
}
|
|
|
|
template <typename TrackVector>
|
|
bool MediaStream::RemoveTrack(TrackVector* tracks,
|
|
MediaStreamTrackInterface* track) {
|
|
RTC_DCHECK(tracks != NULL);
|
|
if (!track)
|
|
return false;
|
|
typename TrackVector::iterator it = FindTrack(tracks, track->id());
|
|
if (it == tracks->end())
|
|
return false;
|
|
tracks->erase(it);
|
|
FireOnChanged();
|
|
return true;
|
|
}
|
|
|
|
} // namespace webrtc
|