OLD | NEW |
(Empty) | |
| 1 /* |
| 2 * Copyright (c) 2015 The WebRTC project authors. All Rights Reserved. |
| 3 * |
| 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. |
| 9 */ |
| 10 |
| 11 #include "webrtc/voice_engine/test/auto_test/fakes/loudest_filter.h" |
| 12 |
| 13 #include "webrtc/base/checks.h" |
| 14 |
| 15 namespace voetest { |
| 16 |
| 17 void LoudestFilter::RemoveTimeoutStreams(uint32 time_ms) { |
| 18 auto it = stream_levels_.begin(); |
| 19 while (it != stream_levels_.end()) { |
| 20 if (rtc::TimeDiff(time_ms, it->second.last_time_ms) > |
| 21 kStreamTimeOutMs) { |
| 22 stream_levels_.erase(it++); |
| 23 } else { |
| 24 ++it; |
| 25 } |
| 26 } |
| 27 } |
| 28 |
| 29 unsigned int LoudestFilter::FindQuietestStream() { |
| 30 int quietest_level = kInvalidAudioLevel; |
| 31 unsigned int quietest_ssrc = 0; |
| 32 for (auto stream : stream_levels_) { |
| 33 // A smaller value if audio level corresponds to a louder sound. |
| 34 if (quietest_level == kInvalidAudioLevel || |
| 35 stream.second.audio_level > quietest_level) { |
| 36 quietest_level = stream.second.audio_level; |
| 37 quietest_ssrc = stream.first; |
| 38 } |
| 39 } |
| 40 return quietest_ssrc; |
| 41 } |
| 42 |
| 43 bool LoudestFilter::ForwardThisPacket(const webrtc::RTPHeader& rtp_header) { |
| 44 uint32 time_now_ms = rtc::Time(); |
| 45 RemoveTimeoutStreams(time_now_ms); |
| 46 |
| 47 int source_ssrc = rtp_header.ssrc; |
| 48 int audio_level = rtp_header.extension.hasAudioLevel ? |
| 49 rtp_header.extension.audioLevel : kInvalidAudioLevel; |
| 50 |
| 51 if (audio_level == kInvalidAudioLevel) { |
| 52 // Always forward streams with unknown audio level, and don't keep their |
| 53 // states. |
| 54 return true; |
| 55 } |
| 56 |
| 57 auto it = stream_levels_.find(source_ssrc); |
| 58 if (it != stream_levels_.end()) { |
| 59 // Stream has been forwarded. Update and continue to forward. |
| 60 it->second.audio_level = audio_level; |
| 61 it->second.last_time_ms = time_now_ms; |
| 62 return true; |
| 63 } |
| 64 |
| 65 if (stream_levels_.size() < kMaxMixSize) { |
| 66 stream_levels_[source_ssrc].Set(audio_level, time_now_ms); |
| 67 return true; |
| 68 } |
| 69 |
| 70 unsigned int quietest_ssrc = FindQuietestStream(); |
| 71 CHECK_NE(0u, quietest_ssrc); |
| 72 // A smaller value if audio level corresponds to a louder sound. |
| 73 if (audio_level < stream_levels_[quietest_ssrc].audio_level) { |
| 74 stream_levels_.erase(quietest_ssrc); |
| 75 stream_levels_[source_ssrc].Set(audio_level, time_now_ms); |
| 76 return true; |
| 77 } |
| 78 return false; |
| 79 } |
| 80 |
| 81 } // namespace voetest |
| 82 |
OLD | NEW |