blob: 6b9e8eb8e5144f0107da7235cf631d198dd08e9c [file] [log] [blame]
Ying Wang3b790f32018-01-19 16:58:571/*
2 * Copyright (c) 2016 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#ifndef MODULES_VIDEO_CODING_FEC_CONTROLLER_DEFAULT_H_
12#define MODULES_VIDEO_CODING_FEC_CONTROLLER_DEFAULT_H_
13
Yves Gerey3e707812018-11-28 15:47:4914#include <stddef.h>
15#include <stdint.h>
Jonas Olssona4d87372019-07-05 17:08:3316
Ying Wang3b790f32018-01-19 16:58:5717#include <memory>
18#include <vector>
Yves Gerey3e707812018-11-28 15:47:4919
Ying Wang3b790f32018-01-19 16:58:5720#include "api/fec_controller.h"
Ying Wang3b790f32018-01-19 16:58:5721#include "modules/video_coding/media_opt_util.h"
Steve Anton10542f22019-01-11 17:11:0022#include "rtc_base/constructor_magic.h"
Markus Handell6deec382020-07-07 10:17:1223#include "rtc_base/synchronization/mutex.h"
Yves Gerey3e707812018-11-28 15:47:4924#include "rtc_base/thread_annotations.h"
Ying Wang3b790f32018-01-19 16:58:5725#include "system_wrappers/include/clock.h"
26
27namespace webrtc {
28
29class FecControllerDefault : public FecController {
30 public:
31 FecControllerDefault(Clock* clock,
32 VCMProtectionCallback* protection_callback);
33 explicit FecControllerDefault(Clock* clock);
Mirko Bonadei8fdcac32018-08-28 14:30:1834 ~FecControllerDefault() override;
Ying Wang3b790f32018-01-19 16:58:5735 void SetProtectionCallback(
36 VCMProtectionCallback* protection_callback) override;
37 void SetProtectionMethod(bool enable_fec, bool enable_nack) override;
38 void SetEncodingData(size_t width,
39 size_t height,
40 size_t num_temporal_layers,
41 size_t max_payload_size) override;
42 uint32_t UpdateFecRates(uint32_t estimated_bitrate_bps,
43 int actual_framerate_fps,
44 uint8_t fraction_lost,
45 std::vector<bool> loss_mask_vector,
46 int64_t round_trip_time_ms) override;
Niels Möller87e2d782019-03-07 09:18:2347 void UpdateWithEncodedData(
48 const size_t encoded_image_length,
49 const VideoFrameType encoded_image_frametype) override;
Mirko Bonadei8fdcac32018-08-28 14:30:1850 bool UseLossVectorMask() override;
Ying Wang68b2df72018-10-22 14:50:4351 float GetProtectionOverheadRateThreshold();
Ying Wang3b790f32018-01-19 16:58:5752
53 private:
54 enum { kBitrateAverageWinMs = 1000 };
55 Clock* const clock_;
56 VCMProtectionCallback* protection_callback_;
Markus Handell6deec382020-07-07 10:17:1257 Mutex mutex_;
Ying Wang3b790f32018-01-19 16:58:5758 std::unique_ptr<media_optimization::VCMLossProtectionLogic> loss_prot_logic_
Markus Handell6deec382020-07-07 10:17:1259 RTC_GUARDED_BY(mutex_);
60 size_t max_payload_size_ RTC_GUARDED_BY(mutex_);
Ying Wang3b790f32018-01-19 16:58:5761 RTC_DISALLOW_COPY_AND_ASSIGN(FecControllerDefault);
Ying Wang68b2df72018-10-22 14:50:4362 const float overhead_threshold_;
Ying Wang3b790f32018-01-19 16:58:5763};
64
65} // namespace webrtc
66#endif // MODULES_VIDEO_CODING_FEC_CONTROLLER_DEFAULT_H_