Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(143)

Side by Side Diff: webrtc/modules/video_coding/utility/quality_scaler.h

Issue 2398963003: Move usage of QualityScaler to ViEEncoder. (Closed)
Patch Set: Code review Created 4 years, 1 month ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
1 /* 1 /*
2 * Copyright (c) 2014 The WebRTC project authors. All Rights Reserved. 2 * Copyright (c) 2014 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 #ifndef WEBRTC_MODULES_VIDEO_CODING_UTILITY_QUALITY_SCALER_H_ 11 #ifndef WEBRTC_MODULES_VIDEO_CODING_UTILITY_QUALITY_SCALER_H_
12 #define WEBRTC_MODULES_VIDEO_CODING_UTILITY_QUALITY_SCALER_H_ 12 #define WEBRTC_MODULES_VIDEO_CODING_UTILITY_QUALITY_SCALER_H_
13 13
14 #include <utility>
15
14 #include "webrtc/common_types.h" 16 #include "webrtc/common_types.h"
17 #include "webrtc/video_frame.h"
magjed_webrtc 2016/10/27 11:45:57 Remove this unused include as well as the i420_buf
kthelgason 2016/10/28 13:20:15 Done.
18 #include "webrtc/base/optional.h"
19 #include "webrtc/base/sequenced_task_checker.h"
15 #include "webrtc/common_video/include/i420_buffer_pool.h" 20 #include "webrtc/common_video/include/i420_buffer_pool.h"
16 #include "webrtc/modules/video_coding/utility/moving_average.h" 21 #include "webrtc/modules/video_coding/utility/moving_average.h"
17 22
18 namespace webrtc { 23 namespace webrtc {
24
25 // An interface for a class that receives scale up/down requests.
26 class ScalingObserverInterface {
27 public:
28 enum ScaleReason { kQuality, kCpu };
sprang_webrtc 2016/10/27 13:43:48 prefer enum class
kthelgason 2016/10/28 13:20:15 I get that strongly-typed enums have several benef
sprang_webrtc 2016/10/31 10:24:31 I'd still go with an enum class, and be more expli
sprang_webrtc 2016/11/14 10:25:38 No?
29 static const size_t ScaleReasonSize = 2;
sprang_webrtc 2016/10/27 13:43:48 nit: kScaleReasonSize or SCALE_REASON_SIZE
kthelgason 2016/10/28 13:20:15 Done.
30 // Called to signal that we can handle larger frames.
31 virtual void ScaleUp(ScaleReason reason) = 0;
32 // Called to signal that encoder to scale down.
33 virtual void ScaleDown(ScaleReason reason) = 0;
34
35 protected:
36 virtual ~ScalingObserverInterface() {}
37 };
38
39 // QualityScaler runs asynchronously and monitors QP values of encoded frames.
40 // It holds a reference to a ScalingObserverInterface implementation to signal
41 // an intent to scale up or down.
19 class QualityScaler { 42 class QualityScaler {
20 public: 43 public:
21 struct Resolution { 44 typedef std::pair<int, int> QPThresholds;
sprang_webrtc 2016/10/27 13:43:48 I'd prefer a simple struct, just because threshold
kthelgason 2016/10/28 13:20:15 Done.
22 int width; 45 struct Settings {
23 int height; 46 Settings(bool on, int low, int high)
47 : enabled(on),
48 thresholds(rtc::Optional<QPThresholds>(QPThresholds(low, high))) {}
49 explicit Settings(bool on) : enabled(on) {}
50 const bool enabled;
51 const rtc::Optional<QPThresholds> thresholds;
24 }; 52 };
25 53
26 QualityScaler(); 54 // Construct a QualityScaler with a given |observer|
27 void Init(VideoCodecType codec_type, 55 explicit QualityScaler(ScalingObserverInterface* observer);
28 int initial_bitrate_kbps, 56 virtual ~QualityScaler();
29 int width, 57 // Init starts the quality scaler periodically checking what the average QP
30 int height, 58 // has been recently.
31 int fps); 59 void Init(VideoCodecType codec_type);
32 void Init(int low_qp_threshold, 60 // If specific thresholds are desired these can be supplied as |thresholds|.
33 int high_qp_threshold, 61 void Init(QPThresholds thresholds);
34 int initial_bitrate_kbps, 62 // Stop must be called to stop the periodic task before QualityScaler is
35 int width, 63 // destroyed.
36 int height, 64 void Stop();
37 int fps); 65 // Should be called each time the encoder drops a frame
38 void ReportFramerate(int framerate); 66 void ReportDroppedFrame();
67 // Inform the QualityScaler of the last seen QP.
39 void ReportQP(int qp); 68 void ReportQP(int qp);
40 void ReportDroppedFrame(); 69
41 void OnEncodeFrame(int width, int height); 70 // This method declared virtual to help with testing.
42 Resolution GetScaledResolution() const; 71 virtual int64_t GetTimeoutMs();
magjed_webrtc 2016/10/27 11:45:57 nit: I don't think the name 'Timeout' is appropria
kthelgason 2016/10/28 13:20:15 I agree that is a better approach. I didn't see a
43 rtc::scoped_refptr<VideoFrameBuffer> GetScaledBuffer(
44 const rtc::scoped_refptr<VideoFrameBuffer>& frame);
45 int downscale_shift() const { return downscale_shift_; }
46 72
47 private: 73 private:
74 class CheckQPTask;
75 void CheckQP();
48 void ClearSamples(); 76 void ClearSamples();
49 void ScaleUp(); 77 void ReportQPLow();
50 void ScaleDown(); 78 void ReportQPHigh();
51 void UpdateTargetResolution(int width, int height);
52 79
53 I420BufferPool pool_; 80 ScalingObserverInterface* observer_ GUARDED_BY(&task_checker_);
sprang_webrtc 2016/10/27 13:43:48 ScalingObserverInterface* const observer_
kthelgason 2016/10/28 13:20:15 Done.
81 CheckQPTask* check_qp_task_ GUARDED_BY(&task_checker_);
82 rtc::SequencedTaskChecker task_checker_;
54 83
55 size_t num_samples_downscale_; 84 bool fast_rampup_ GUARDED_BY(&task_checker_);
56 size_t num_samples_upscale_; 85 int64_t measure_interval_ GUARDED_BY(&task_checker_);
magjed_webrtc 2016/10/27 11:45:57 nit: I would like to add a 'sec' suffix to the var
kthelgason 2016/10/28 13:20:15 Done.
57 bool fast_rampup_; 86 MovingAverage average_qp_ GUARDED_BY(&task_checker_);
58 MovingAverage average_qp_; 87 MovingAverage framedrop_percent_ GUARDED_BY(&task_checker_);
59 MovingAverage framedrop_percent_;
60 88
61 int low_qp_threshold_; 89 int low_qp_threshold_ GUARDED_BY(&task_checker_);
62 int high_qp_threshold_; 90 int high_qp_threshold_ GUARDED_BY(&task_checker_);
sprang_webrtc 2016/10/27 13:43:48 Maybe a single QPThresholds member?
kthelgason 2016/10/28 13:20:15 Done.
63 Resolution target_res_;
64 91
65 int downscale_shift_; 92 static const auto scale_reason_ = ScalingObserverInterface::kQuality;
sprang_webrtc 2016/10/27 13:43:48 Prefer not to use auto here.
kthelgason 2016/10/28 13:20:15 Why? the type should be completely obvious from th
sprang_webrtc 2016/10/31 10:24:31 If you change to an enum class, maybe. Otherwise s
magjed_webrtc 2016/11/10 13:06:59 It's not allowed to use auto for non-local variabl
66 int maximum_shift_;
67 }; 93 };
68 94
69 } // namespace webrtc 95 } // namespace webrtc
70 96
71 #endif // WEBRTC_MODULES_VIDEO_CODING_UTILITY_QUALITY_SCALER_H_ 97 #endif // WEBRTC_MODULES_VIDEO_CODING_UTILITY_QUALITY_SCALER_H_
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698