| /* |
| * Copyright 2016 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 "webrtc/api/trackmediainfomap.h" |
| |
| #include <utility> |
| |
| namespace webrtc { |
| |
| namespace { |
| |
| template<typename K, typename V> |
| V FindValueOrNull(const std::map<K, V>& map, const K& key) { |
| auto it = map.find(key); |
| return (it != map.end()) ? it->second : nullptr; |
| } |
| |
| template<typename K, typename V> |
| const V* FindAddressOrNull(const std::map<K, V>& map, const K& key) { |
| auto it = map.find(key); |
| return (it != map.end()) ? &it->second : nullptr; |
| } |
| |
| void GetAudioAndVideoTrackBySsrc( |
| const std::vector<rtc::scoped_refptr<RtpSenderInterface>>& rtp_senders, |
| const std::vector<rtc::scoped_refptr<RtpReceiverInterface>>& rtp_receivers, |
| std::map<uint32_t, AudioTrackInterface*>* audio_track_by_ssrc, |
| std::map<uint32_t, VideoTrackInterface*>* video_track_by_ssrc) { |
| RTC_DCHECK(audio_track_by_ssrc->empty()); |
| RTC_DCHECK(video_track_by_ssrc->empty()); |
| // TODO(hbos): RTP senders/receivers uses a proxy to the signaling thread, and |
| // our sender/receiver implementations invokes on the worker thread. (This |
| // means one thread jump if on signaling thread and two thread jumps if on any |
| // other threads). Is there a way to avoid thread jump(s) on a per |
| // sender/receiver, per method basis? |
| for (const rtc::scoped_refptr<RtpSenderInterface>& rtp_sender : rtp_senders) { |
| cricket::MediaType media_type = rtp_sender->media_type(); |
| MediaStreamTrackInterface* track = rtp_sender->track(); |
| if (!track) { |
| continue; |
| } |
| RTC_DCHECK_EQ(track->kind(), |
| media_type == cricket::MEDIA_TYPE_AUDIO |
| ? MediaStreamTrackInterface::kAudioKind |
| : MediaStreamTrackInterface::kVideoKind); |
| // TODO(deadbeef): |ssrc| should be removed in favor of |GetParameters|. |
| uint32_t ssrc = rtp_sender->ssrc(); |
| if (ssrc != 0) { |
| if (media_type == cricket::MEDIA_TYPE_AUDIO) { |
| RTC_DCHECK(audio_track_by_ssrc->find(ssrc) == |
| audio_track_by_ssrc->end()); |
| (*audio_track_by_ssrc)[ssrc] = static_cast<AudioTrackInterface*>(track); |
| } else { |
| RTC_DCHECK(video_track_by_ssrc->find(ssrc) == |
| video_track_by_ssrc->end()); |
| (*video_track_by_ssrc)[ssrc] = static_cast<VideoTrackInterface*>(track); |
| } |
| } |
| } |
| for (const rtc::scoped_refptr<RtpReceiverInterface>& rtp_receiver : |
| rtp_receivers) { |
| cricket::MediaType media_type = rtp_receiver->media_type(); |
| MediaStreamTrackInterface* track = rtp_receiver->track(); |
| RTC_DCHECK(track); |
| RTC_DCHECK_EQ(track->kind(), |
| media_type == cricket::MEDIA_TYPE_AUDIO |
| ? MediaStreamTrackInterface::kAudioKind |
| : MediaStreamTrackInterface::kVideoKind); |
| RtpParameters params = rtp_receiver->GetParameters(); |
| for (const RtpEncodingParameters& encoding : params.encodings) { |
| if (!encoding.ssrc) { |
| continue; |
| } |
| if (media_type == cricket::MEDIA_TYPE_AUDIO) { |
| RTC_DCHECK(audio_track_by_ssrc->find(*encoding.ssrc) == |
| audio_track_by_ssrc->end()); |
| (*audio_track_by_ssrc)[*encoding.ssrc] = |
| static_cast<AudioTrackInterface*>(track); |
| } else { |
| RTC_DCHECK(video_track_by_ssrc->find(*encoding.ssrc) == |
| video_track_by_ssrc->end()); |
| (*video_track_by_ssrc)[*encoding.ssrc] = |
| static_cast<VideoTrackInterface*>(track); |
| } |
| } |
| } |
| } |
| |
| } // namespace |
| |
| TrackMediaInfoMap::TrackMediaInfoMap( |
| std::unique_ptr<cricket::VoiceMediaInfo> voice_media_info, |
| std::unique_ptr<cricket::VideoMediaInfo> video_media_info, |
| const std::vector<rtc::scoped_refptr<RtpSenderInterface>>& rtp_senders, |
| const std::vector<rtc::scoped_refptr<RtpReceiverInterface>>& rtp_receivers) |
| : voice_media_info_(std::move(voice_media_info)), |
| video_media_info_(std::move(video_media_info)) { |
| std::map<uint32_t, AudioTrackInterface*> audio_track_by_ssrc; |
| std::map<uint32_t, VideoTrackInterface*> video_track_by_ssrc; |
| GetAudioAndVideoTrackBySsrc( |
| rtp_senders, rtp_receivers, &audio_track_by_ssrc, &video_track_by_ssrc); |
| if (voice_media_info_) { |
| for (auto& sender_info : voice_media_info_->senders) { |
| AudioTrackInterface* associated_track = |
| FindValueOrNull(audio_track_by_ssrc, sender_info.ssrc()); |
| if (associated_track) { |
| // One sender is associated with at most one track. |
| // One track may be associated with multiple senders. |
| audio_track_by_sender_info_[&sender_info] = associated_track; |
| voice_infos_by_local_track_[associated_track].push_back(&sender_info); |
| } |
| } |
| for (auto& receiver_info : voice_media_info_->receivers) { |
| AudioTrackInterface* associated_track = |
| FindValueOrNull(audio_track_by_ssrc, receiver_info.ssrc()); |
| if (associated_track) { |
| // One receiver is associated with at most one track, which is uniquely |
| // associated with that receiver. |
| audio_track_by_receiver_info_[&receiver_info] = associated_track; |
| RTC_DCHECK(voice_info_by_remote_track_.find(associated_track) == |
| voice_info_by_remote_track_.end()); |
| voice_info_by_remote_track_[associated_track] = &receiver_info; |
| } |
| } |
| } |
| if (video_media_info_) { |
| for (auto& sender_info : video_media_info_->senders) { |
| VideoTrackInterface* associated_track = |
| FindValueOrNull(video_track_by_ssrc, sender_info.ssrc()); |
| if (associated_track) { |
| // One sender is associated with at most one track. |
| // One track may be associated with multiple senders. |
| video_track_by_sender_info_[&sender_info] = associated_track; |
| video_infos_by_local_track_[associated_track].push_back(&sender_info); |
| } |
| } |
| for (auto& receiver_info : video_media_info_->receivers) { |
| VideoTrackInterface* associated_track = |
| FindValueOrNull(video_track_by_ssrc, receiver_info.ssrc()); |
| if (associated_track) { |
| // One receiver is associated with at most one track, which is uniquely |
| // associated with that receiver. |
| video_track_by_receiver_info_[&receiver_info] = associated_track; |
| RTC_DCHECK(video_info_by_remote_track_.find(associated_track) == |
| video_info_by_remote_track_.end()); |
| video_info_by_remote_track_[associated_track] = &receiver_info; |
| } |
| } |
| } |
| } |
| |
| const std::vector<cricket::VoiceSenderInfo*>* |
| TrackMediaInfoMap::GetVoiceSenderInfos( |
| const AudioTrackInterface& local_audio_track) const { |
| return FindAddressOrNull(voice_infos_by_local_track_, &local_audio_track); |
| } |
| |
| const cricket::VoiceReceiverInfo* TrackMediaInfoMap::GetVoiceReceiverInfo( |
| const AudioTrackInterface& remote_audio_track) const { |
| return FindValueOrNull(voice_info_by_remote_track_, &remote_audio_track); |
| } |
| |
| const std::vector<cricket::VideoSenderInfo*>* |
| TrackMediaInfoMap::GetVideoSenderInfos( |
| const VideoTrackInterface& local_video_track) const { |
| return FindAddressOrNull(video_infos_by_local_track_, &local_video_track); |
| } |
| |
| const cricket::VideoReceiverInfo* TrackMediaInfoMap::GetVideoReceiverInfo( |
| const VideoTrackInterface& remote_video_track) const { |
| return FindValueOrNull(video_info_by_remote_track_, &remote_video_track); |
| } |
| |
| rtc::scoped_refptr<AudioTrackInterface> TrackMediaInfoMap::GetAudioTrack( |
| const cricket::VoiceSenderInfo& voice_sender_info) const { |
| return FindValueOrNull(audio_track_by_sender_info_, &voice_sender_info); |
| } |
| |
| rtc::scoped_refptr<AudioTrackInterface> TrackMediaInfoMap::GetAudioTrack( |
| const cricket::VoiceReceiverInfo& voice_receiver_info) const { |
| return FindValueOrNull(audio_track_by_receiver_info_, &voice_receiver_info); |
| } |
| |
| rtc::scoped_refptr<VideoTrackInterface> TrackMediaInfoMap::GetVideoTrack( |
| const cricket::VideoSenderInfo& video_sender_info) const { |
| return FindValueOrNull(video_track_by_sender_info_, &video_sender_info); |
| } |
| |
| rtc::scoped_refptr<VideoTrackInterface> TrackMediaInfoMap::GetVideoTrack( |
| const cricket::VideoReceiverInfo& video_receiver_info) const { |
| return FindValueOrNull(video_track_by_receiver_info_, &video_receiver_info); |
| } |
| |
| } // namespace webrtc |
