Index: webrtc/modules/video_coding/protection_bitrate_calculator.h |
diff --git a/webrtc/modules/video_coding/protection_bitrate_calculator.h b/webrtc/modules/video_coding/protection_bitrate_calculator.h |
new file mode 100644 |
index 0000000000000000000000000000000000000000..df6c501365a854a08bc8a939a807da5e7b295ac6 |
--- /dev/null |
+++ b/webrtc/modules/video_coding/protection_bitrate_calculator.h |
@@ -0,0 +1,93 @@ |
+/* |
+ * Copyright (c) 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. |
+ */ |
+ |
+#ifndef WEBRTC_MODULES_VIDEO_CODING_PROTECTION_BITRATE_CALCULATOR_H_ |
+#define WEBRTC_MODULES_VIDEO_CODING_PROTECTION_BITRATE_CALCULATOR_H_ |
+ |
+#include <list> |
+#include <memory> |
+ |
+#include "webrtc/base/criticalsection.h" |
+#include "webrtc/modules/include/module_common_types.h" |
+#include "webrtc/modules/video_coding/include/video_coding.h" |
+#include "webrtc/modules/video_coding/media_opt_util.h" |
+#include "webrtc/system_wrappers/include/clock.h" |
+ |
+namespace webrtc { |
+ |
+// ProtectionBitrateCalculator calculates the calculates how much of the |
stefan-webrtc
2016/05/19 12:52:40
-"calculates the"
perkj_webrtc
2016/06/01 20:55:23
Done.
|
+// allocated network capacity that can be used by an encoder and how much that |
+// is needed for protection packets such as FEC and NACK. It uses an |
stefan-webrtc
2016/05/19 12:52:40
Maybe call it redundant packets instead?
perkj_webrtc
2016/06/01 20:55:23
Done.
|
+// implementation of |VCMProtectionCallback| to set new FEC parameters and get |
+// the bitrate currently used for FEC and NACK. |
+// Usage: |
+// Setup by calling SetProtectionMethod and SetEncodingData. |
+// For each encoded image, call UpdateWithEncodedData. |
+// Each time the bandwidth estimate change, call SetTargetRates. SetTargetRates |
+// will return the bitrate that can be used by an encoder. |
+class ProtectionBitrateCalculator { |
+ public: |
+ explicit ProtectionBitrateCalculator( |
stefan-webrtc
2016/05/19 12:52:40
Remove explicit
perkj_webrtc
2016/06/01 20:55:23
Done.
|
+ Clock* clock, |
+ VCMProtectionCallback* protection_callback); |
+ ~ProtectionBitrateCalculator(); |
+ |
+ void SetProtectionMethod(bool enable_fec, bool enable_nack); |
stefan-webrtc
2016/05/19 12:52:40
Would it make sense to somehow document the thread
perkj_webrtc
2016/06/01 20:55:23
Added a short note about thread safety.
|
+ |
+ // Informs media optimization of initial encoding state. |
+ void SetEncodingData(uint32_t estimated_bitrate_bps, |
+ uint16_t width, |
+ uint16_t height, |
+ uint32_t frame_rate, |
+ size_t num_temporal_layers, |
+ size_t mtu); |
+ |
+ // Returns target rate for the encoder given the channel parameters. |
+ // Inputs: estimated_bitrate_bps - the estimated network bitrate in bits/s. |
+ // fraction_lost - packet loss rate in % in the network. |
+ // round_trip_time_ms - round trip time in milliseconds. |
+ uint32_t SetTargetRates(uint32_t estimated_bitrate_bps, |
+ uint8_t fraction_lost, |
+ int64_t round_trip_time_ms); |
+ |
+ // Informs of encoded output. |
+ void UpdateWithEncodedData(const EncodedImage& encoded_image); |
+ |
+ uint32_t SentFrameRate(); |
+ uint32_t SentBitRate(); |
stefan-webrtc
2016/05/19 12:52:40
It's a bit weird that this class is named somethin
perkj_webrtc
2016/06/01 20:55:23
Removed. Now relies on fps as an input parameter f
|
+ |
+ private: |
+ enum { kBitrateAverageWinMs = 1000 }; |
+ |
+ void PurgeOldFrameSamples(int64_t now_ms) |
+ EXCLUSIVE_LOCKS_REQUIRED(crit_sect_); |
+ void UpdateSentBitrate(int64_t now_ms) EXCLUSIVE_LOCKS_REQUIRED(crit_sect_); |
+ void UpdateSentFramerate() EXCLUSIVE_LOCKS_REQUIRED(crit_sect_); |
+ |
+ uint32_t SentFrameRateInternal() EXCLUSIVE_LOCKS_REQUIRED(crit_sect_); |
+ |
+ // Protect all members. |
+ rtc::CriticalSection crit_sect_; |
+ Clock* const clock_; |
+ VCMProtectionCallback* const protection_callback_; |
+ |
+ std::unique_ptr<media_optimization::VCMLossProtectionLogic> loss_prot_logic_ |
+ GUARDED_BY(crit_sect_); |
+ size_t max_payload_size_ GUARDED_BY(crit_sect_); |
+ struct EncodedFrameSample; |
stefan-webrtc
2016/05/19 12:52:40
Move this to before all members instead to not mix
perkj_webrtc
2016/06/01 20:55:23
Done.
|
+ std::list<EncodedFrameSample> encoded_frame_samples_ GUARDED_BY(crit_sect_); |
+ uint32_t avg_sent_bit_rate_bps_ GUARDED_BY(crit_sect_); |
+ uint32_t avg_sent_framerate_ GUARDED_BY(crit_sect_); |
+ |
+ RTC_DISALLOW_COPY_AND_ASSIGN(ProtectionBitrateCalculator); |
+}; |
+ |
+} // namespace webrtc |
+#endif // WEBRTC_MODULES_VIDEO_CODING_PROTECTION_BITRATE_CALCULATOR_H_ |