/* * 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. */ #ifndef PC_STREAM_COLLECTION_H_ #define PC_STREAM_COLLECTION_H_ #include #include #include #include "api/peer_connection_interface.h" namespace webrtc { // Implementation of StreamCollection. class StreamCollection : public StreamCollectionInterface { public: static rtc::scoped_refptr Create() { return rtc::make_ref_counted(); } static rtc::scoped_refptr Create( StreamCollection* streams) { return rtc::make_ref_counted(streams); } virtual size_t count() { return media_streams_.size(); } virtual MediaStreamInterface* at(size_t index) { return media_streams_.at(index).get(); } virtual MediaStreamInterface* find(const std::string& id) { for (StreamVector::iterator it = media_streams_.begin(); it != media_streams_.end(); ++it) { if ((*it)->id().compare(id) == 0) { return (*it).get(); } } return NULL; } virtual MediaStreamTrackInterface* FindAudioTrack(const std::string& id) { for (size_t i = 0; i < media_streams_.size(); ++i) { MediaStreamTrackInterface* track = media_streams_[i]->FindAudioTrack(id).get(); if (track) { return track; } } return NULL; } virtual MediaStreamTrackInterface* FindVideoTrack(const std::string& id) { for (size_t i = 0; i < media_streams_.size(); ++i) { MediaStreamTrackInterface* track = media_streams_[i]->FindVideoTrack(id).get(); if (track) { return track; } } return NULL; } void AddStream(rtc::scoped_refptr stream) { for (StreamVector::iterator it = media_streams_.begin(); it != media_streams_.end(); ++it) { if ((*it)->id().compare(stream->id()) == 0) return; } media_streams_.push_back(std::move(stream)); } void RemoveStream(MediaStreamInterface* remove_stream) { for (StreamVector::iterator it = media_streams_.begin(); it != media_streams_.end(); ++it) { if ((*it)->id().compare(remove_stream->id()) == 0) { media_streams_.erase(it); break; } } } protected: StreamCollection() {} explicit StreamCollection(StreamCollection* original) : media_streams_(original->media_streams_) {} typedef std::vector > StreamVector; StreamVector media_streams_; }; } // namespace webrtc #endif // PC_STREAM_COLLECTION_H_