1/*
2 * libjingle
3 * Copyright 2011, Google Inc.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions are met:
7 *
8 *  1. Redistributions of source code must retain the above copyright notice,
9 *     this list of conditions and the following disclaimer.
10 *  2. Redistributions in binary form must reproduce the above copyright notice,
11 *     this list of conditions and the following disclaimer in the documentation
12 *     and/or other materials provided with the distribution.
13 *  3. The name of the author may not be used to endorse or promote products
14 *     derived from this software without specific prior written permission.
15 *
16 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR IMPLIED
17 * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
18 * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO
19 * EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
20 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
21 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
22 * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
23 * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
24 * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
25 * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
26 */
27
28#include "talk/app/webrtc/mediastream.h"
29#include "webrtc/base/logging.h"
30
31namespace webrtc {
32
33template <class V>
34static typename V::iterator FindTrack(V* vector,
35                                      const std::string& track_id) {
36  typename V::iterator it = vector->begin();
37  for (; it != vector->end(); ++it) {
38    if ((*it)->id() == track_id) {
39      break;
40    }
41  }
42  return it;
43};
44
45rtc::scoped_refptr<MediaStream> MediaStream::Create(
46    const std::string& label) {
47  rtc::RefCountedObject<MediaStream>* stream =
48      new rtc::RefCountedObject<MediaStream>(label);
49  return stream;
50}
51
52MediaStream::MediaStream(const std::string& label)
53    : label_(label) {
54}
55
56bool MediaStream::AddTrack(AudioTrackInterface* track) {
57  return AddTrack<AudioTrackVector, AudioTrackInterface>(&audio_tracks_, track);
58}
59
60bool MediaStream::AddTrack(VideoTrackInterface* track) {
61  return AddTrack<VideoTrackVector, VideoTrackInterface>(&video_tracks_, track);
62}
63
64bool MediaStream::RemoveTrack(AudioTrackInterface* track) {
65  return RemoveTrack<AudioTrackVector>(&audio_tracks_, track);
66}
67
68bool MediaStream::RemoveTrack(VideoTrackInterface* track) {
69  return RemoveTrack<VideoTrackVector>(&video_tracks_, track);
70}
71
72rtc::scoped_refptr<AudioTrackInterface>
73MediaStream::FindAudioTrack(const std::string& track_id) {
74  AudioTrackVector::iterator it = FindTrack(&audio_tracks_, track_id);
75  if (it == audio_tracks_.end())
76    return NULL;
77  return *it;
78}
79
80rtc::scoped_refptr<VideoTrackInterface>
81MediaStream::FindVideoTrack(const std::string& track_id) {
82  VideoTrackVector::iterator it = FindTrack(&video_tracks_, track_id);
83  if (it == video_tracks_.end())
84    return NULL;
85  return *it;
86}
87
88template <typename TrackVector, typename Track>
89bool MediaStream::AddTrack(TrackVector* tracks, Track* track) {
90  typename TrackVector::iterator it = FindTrack(tracks, track->id());
91  if (it != tracks->end())
92    return false;
93  tracks->push_back(track);
94  FireOnChanged();
95  return true;
96}
97
98template <typename TrackVector>
99bool MediaStream::RemoveTrack(TrackVector* tracks,
100                              MediaStreamTrackInterface* track) {
101  ASSERT(tracks != NULL);
102  if (!track)
103    return false;
104  typename TrackVector::iterator it = FindTrack(tracks, track->id());
105  if (it == tracks->end())
106    return false;
107  tracks->erase(it);
108  FireOnChanged();
109  return true;
110}
111
112}  // namespace webrtc
113