OLD | NEW |
1 /* | 1 /* |
2 * Copyright (c) 2012 The WebRTC project authors. All Rights Reserved. | 2 * Copyright (c) 2012 The WebRTC project authors. All Rights Reserved. |
3 * | 3 * |
4 * Use of this source code is governed by a BSD-style license | 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 | 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 | 6 * tree. An additional intellectual property rights grant can be found |
7 * in the file PATENTS. All contributing project authors may | 7 * in the file PATENTS. All contributing project authors may |
8 * be found in the AUTHORS file in the root of the source tree. | 8 * be found in the AUTHORS file in the root of the source tree. |
9 */ | 9 */ |
10 | 10 |
11 #include "webrtc/video/encoder_state_feedback.h" | 11 #include "webrtc/video/encoder_state_feedback.h" |
12 | 12 |
13 #include <assert.h> | 13 #include <assert.h> |
14 | 14 |
15 #include "webrtc/base/checks.h" | 15 #include "webrtc/base/checks.h" |
16 #include "webrtc/modules/rtp_rtcp/include/rtp_rtcp_defines.h" | 16 #include "webrtc/modules/rtp_rtcp/include/rtp_rtcp_defines.h" |
17 #include "webrtc/system_wrappers/include/critical_section_wrapper.h" | |
18 #include "webrtc/video/vie_encoder.h" | 17 #include "webrtc/video/vie_encoder.h" |
19 | 18 |
20 namespace webrtc { | 19 namespace webrtc { |
21 | 20 |
22 // Helper class registered at the RTP module relaying callbacks to | 21 // Helper class registered at the RTP module relaying callbacks to |
23 // EncoderStatFeedback. | 22 // EncoderStatFeedback. |
24 class EncoderStateFeedbackObserver : public RtcpIntraFrameObserver { | 23 class EncoderStateFeedbackObserver : public RtcpIntraFrameObserver { |
25 public: | 24 public: |
26 explicit EncoderStateFeedbackObserver(EncoderStateFeedback* owner) | 25 explicit EncoderStateFeedbackObserver(EncoderStateFeedback* owner) |
27 : owner_(owner) {} | 26 : owner_(owner) {} |
(...skipping 12 matching lines...) Expand all Loading... |
40 | 39 |
41 virtual void OnLocalSsrcChanged(uint32_t old_ssrc, uint32_t new_ssrc) { | 40 virtual void OnLocalSsrcChanged(uint32_t old_ssrc, uint32_t new_ssrc) { |
42 owner_->OnLocalSsrcChanged(old_ssrc, new_ssrc); | 41 owner_->OnLocalSsrcChanged(old_ssrc, new_ssrc); |
43 } | 42 } |
44 | 43 |
45 private: | 44 private: |
46 EncoderStateFeedback* owner_; | 45 EncoderStateFeedback* owner_; |
47 }; | 46 }; |
48 | 47 |
49 EncoderStateFeedback::EncoderStateFeedback() | 48 EncoderStateFeedback::EncoderStateFeedback() |
50 : crit_(CriticalSectionWrapper::CreateCriticalSection()), | 49 : observer_(new EncoderStateFeedbackObserver(this)) {} |
51 observer_(new EncoderStateFeedbackObserver(this)) {} | |
52 | 50 |
53 EncoderStateFeedback::~EncoderStateFeedback() { | 51 EncoderStateFeedback::~EncoderStateFeedback() { |
54 assert(encoders_.empty()); | 52 assert(encoders_.empty()); |
55 } | 53 } |
56 | 54 |
57 void EncoderStateFeedback::AddEncoder(const std::vector<uint32_t>& ssrcs, | 55 void EncoderStateFeedback::AddEncoder(const std::vector<uint32_t>& ssrcs, |
58 ViEEncoder* encoder) { | 56 ViEEncoder* encoder) { |
59 RTC_DCHECK(!ssrcs.empty()); | 57 RTC_DCHECK(!ssrcs.empty()); |
60 CriticalSectionScoped lock(crit_.get()); | 58 rtc::CritScope lock(&crit_); |
61 for (uint32_t ssrc : ssrcs) { | 59 for (uint32_t ssrc : ssrcs) { |
62 RTC_DCHECK(encoders_.find(ssrc) == encoders_.end()); | 60 RTC_DCHECK(encoders_.find(ssrc) == encoders_.end()); |
63 encoders_[ssrc] = encoder; | 61 encoders_[ssrc] = encoder; |
64 } | 62 } |
65 } | 63 } |
66 | 64 |
67 void EncoderStateFeedback::RemoveEncoder(const ViEEncoder* encoder) { | 65 void EncoderStateFeedback::RemoveEncoder(const ViEEncoder* encoder) { |
68 CriticalSectionScoped lock(crit_.get()); | 66 rtc::CritScope lock(&crit_); |
69 SsrcEncoderMap::iterator it = encoders_.begin(); | 67 SsrcEncoderMap::iterator it = encoders_.begin(); |
70 while (it != encoders_.end()) { | 68 while (it != encoders_.end()) { |
71 if (it->second == encoder) { | 69 if (it->second == encoder) { |
72 encoders_.erase(it++); | 70 encoders_.erase(it++); |
73 } else { | 71 } else { |
74 ++it; | 72 ++it; |
75 } | 73 } |
76 } | 74 } |
77 } | 75 } |
78 | 76 |
79 RtcpIntraFrameObserver* EncoderStateFeedback::GetRtcpIntraFrameObserver() { | 77 RtcpIntraFrameObserver* EncoderStateFeedback::GetRtcpIntraFrameObserver() { |
80 return observer_.get(); | 78 return observer_.get(); |
81 } | 79 } |
82 | 80 |
83 void EncoderStateFeedback::OnReceivedIntraFrameRequest(uint32_t ssrc) { | 81 void EncoderStateFeedback::OnReceivedIntraFrameRequest(uint32_t ssrc) { |
84 CriticalSectionScoped lock(crit_.get()); | 82 rtc::CritScope lock(&crit_); |
85 SsrcEncoderMap::iterator it = encoders_.find(ssrc); | 83 SsrcEncoderMap::iterator it = encoders_.find(ssrc); |
86 if (it == encoders_.end()) | 84 if (it == encoders_.end()) |
87 return; | 85 return; |
88 | 86 |
89 it->second->OnReceivedIntraFrameRequest(ssrc); | 87 it->second->OnReceivedIntraFrameRequest(ssrc); |
90 } | 88 } |
91 | 89 |
92 void EncoderStateFeedback::OnReceivedSLI(uint32_t ssrc, uint8_t picture_id) { | 90 void EncoderStateFeedback::OnReceivedSLI(uint32_t ssrc, uint8_t picture_id) { |
93 CriticalSectionScoped lock(crit_.get()); | 91 rtc::CritScope lock(&crit_); |
94 SsrcEncoderMap::iterator it = encoders_.find(ssrc); | 92 SsrcEncoderMap::iterator it = encoders_.find(ssrc); |
95 if (it == encoders_.end()) | 93 if (it == encoders_.end()) |
96 return; | 94 return; |
97 | 95 |
98 it->second->OnReceivedSLI(ssrc, picture_id); | 96 it->second->OnReceivedSLI(ssrc, picture_id); |
99 } | 97 } |
100 | 98 |
101 void EncoderStateFeedback::OnReceivedRPSI(uint32_t ssrc, uint64_t picture_id) { | 99 void EncoderStateFeedback::OnReceivedRPSI(uint32_t ssrc, uint64_t picture_id) { |
102 CriticalSectionScoped lock(crit_.get()); | 100 rtc::CritScope lock(&crit_); |
103 SsrcEncoderMap::iterator it = encoders_.find(ssrc); | 101 SsrcEncoderMap::iterator it = encoders_.find(ssrc); |
104 if (it == encoders_.end()) | 102 if (it == encoders_.end()) |
105 return; | 103 return; |
106 | 104 |
107 it->second->OnReceivedRPSI(ssrc, picture_id); | 105 it->second->OnReceivedRPSI(ssrc, picture_id); |
108 } | 106 } |
109 | 107 |
110 void EncoderStateFeedback::OnLocalSsrcChanged(uint32_t old_ssrc, | 108 void EncoderStateFeedback::OnLocalSsrcChanged(uint32_t old_ssrc, |
111 uint32_t new_ssrc) { | 109 uint32_t new_ssrc) { |
112 CriticalSectionScoped lock(crit_.get()); | 110 rtc::CritScope lock(&crit_); |
113 SsrcEncoderMap::iterator it = encoders_.find(old_ssrc); | 111 SsrcEncoderMap::iterator it = encoders_.find(old_ssrc); |
114 if (it == encoders_.end() || encoders_.find(new_ssrc) != encoders_.end()) { | 112 if (it == encoders_.end() || encoders_.find(new_ssrc) != encoders_.end()) { |
115 return; | 113 return; |
116 } | 114 } |
117 | 115 |
118 ViEEncoder* encoder = it->second; | 116 ViEEncoder* encoder = it->second; |
119 encoders_.erase(it); | 117 encoders_.erase(it); |
120 encoders_[new_ssrc] = encoder; | 118 encoders_[new_ssrc] = encoder; |
121 encoder->OnLocalSsrcChanged(old_ssrc, new_ssrc); | 119 encoder->OnLocalSsrcChanged(old_ssrc, new_ssrc); |
122 } | 120 } |
123 | 121 |
124 } // namespace webrtc | 122 } // namespace webrtc |
OLD | NEW |