niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 1 | /* |
| 2 | * Copyright (c) 2011 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 | |
stefan@webrtc.org | 1ea4b50 | 2013-01-07 08:49:41 +0000 | [diff] [blame] | 11 | #ifndef WEBRTC_MODULES_VIDEO_CODING_MAIN_SOURCE_RECEIVER_H_ |
| 12 | #define WEBRTC_MODULES_VIDEO_CODING_MAIN_SOURCE_RECEIVER_H_ |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 13 | |
stefan@webrtc.org | 1ea4b50 | 2013-01-07 08:49:41 +0000 | [diff] [blame] | 14 | #include "webrtc/modules/video_coding/main/source/jitter_buffer.h" |
| 15 | #include "webrtc/modules/video_coding/main/source/packet.h" |
stefan@webrtc.org | 1ea4b50 | 2013-01-07 08:49:41 +0000 | [diff] [blame] | 16 | #include "webrtc/modules/video_coding/main/source/timing.h" |
pbos@webrtc.org | a440732 | 2013-07-16 12:32:05 +0000 | [diff] [blame] | 17 | #include "webrtc/system_wrappers/interface/critical_section_wrapper.h" |
agalusza@google.com | a7e360e | 2013-08-01 03:15:08 +0000 | [diff] [blame] | 18 | #include "webrtc/modules/video_coding/main/interface/video_coding.h" |
| 19 | #include "webrtc/modules/video_coding/main/interface/video_coding_defines.h" |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 20 | |
stefan@webrtc.org | 1ea4b50 | 2013-01-07 08:49:41 +0000 | [diff] [blame] | 21 | namespace webrtc { |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 22 | |
stefan@webrtc.org | a678a3b | 2013-01-21 07:42:11 +0000 | [diff] [blame] | 23 | class Clock; |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 24 | class VCMEncodedFrame; |
| 25 | |
stefan@webrtc.org | 1ea4b50 | 2013-01-07 08:49:41 +0000 | [diff] [blame] | 26 | class VCMReceiver { |
| 27 | public: |
| 28 | VCMReceiver(VCMTiming* timing, |
stefan@webrtc.org | a678a3b | 2013-01-21 07:42:11 +0000 | [diff] [blame] | 29 | Clock* clock, |
Wan-Teh Chang | 92d9489 | 2015-05-28 13:36:06 -0700 | [diff] [blame] | 30 | EventFactory* event_factory); |
Qiang Chen | d4cec15 | 2015-06-19 09:17:00 -0700 | [diff] [blame] | 31 | |
| 32 | // Using this constructor, you can specify a different event factory for the |
| 33 | // jitter buffer. Useful for unit tests when you want to simulate incoming |
| 34 | // packets, in which case the jitter buffer's wait event is different from |
| 35 | // that of VCMReceiver itself. |
| 36 | VCMReceiver(VCMTiming* timing, |
| 37 | Clock* clock, |
| 38 | rtc::scoped_ptr<EventWrapper> receiver_event, |
| 39 | rtc::scoped_ptr<EventWrapper> jitter_buffer_event); |
| 40 | |
stefan@webrtc.org | 1ea4b50 | 2013-01-07 08:49:41 +0000 | [diff] [blame] | 41 | ~VCMReceiver(); |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 42 | |
stefan@webrtc.org | 1ea4b50 | 2013-01-07 08:49:41 +0000 | [diff] [blame] | 43 | void Reset(); |
pkasting@chromium.org | 16825b1 | 2015-01-12 21:51:21 +0000 | [diff] [blame] | 44 | void UpdateRtt(int64_t rtt); |
stefan@webrtc.org | 1ea4b50 | 2013-01-07 08:49:41 +0000 | [diff] [blame] | 45 | int32_t InsertPacket(const VCMPacket& packet, |
| 46 | uint16_t frame_width, |
| 47 | uint16_t frame_height); |
| 48 | VCMEncodedFrame* FrameForDecoding(uint16_t max_wait_time_ms, |
| 49 | int64_t& next_render_time_ms, |
pbos@webrtc.org | 4f16c87 | 2014-11-24 09:06:48 +0000 | [diff] [blame] | 50 | bool render_timing = true); |
stefan@webrtc.org | 1ea4b50 | 2013-01-07 08:49:41 +0000 | [diff] [blame] | 51 | void ReleaseFrame(VCMEncodedFrame* frame); |
| 52 | void ReceiveStatistics(uint32_t* bitrate, uint32_t* framerate); |
stefan@webrtc.org | 1ea4b50 | 2013-01-07 08:49:41 +0000 | [diff] [blame] | 53 | uint32_t DiscardedPackets() const; |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 54 | |
stefan@webrtc.org | 1ea4b50 | 2013-01-07 08:49:41 +0000 | [diff] [blame] | 55 | // NACK. |
stefan@webrtc.org | a64300a | 2013-03-04 15:24:40 +0000 | [diff] [blame] | 56 | void SetNackMode(VCMNackMode nackMode, |
pkasting@chromium.org | 16825b1 | 2015-01-12 21:51:21 +0000 | [diff] [blame] | 57 | int64_t low_rtt_nack_threshold_ms, |
| 58 | int64_t high_rtt_nack_threshold_ms); |
stefan@webrtc.org | becf9c8 | 2013-02-01 15:09:57 +0000 | [diff] [blame] | 59 | void SetNackSettings(size_t max_nack_list_size, |
stefan@webrtc.org | ef14488 | 2013-05-07 19:16:33 +0000 | [diff] [blame] | 60 | int max_packet_age_to_nack, |
| 61 | int max_incomplete_time_ms); |
stefan@webrtc.org | 1ea4b50 | 2013-01-07 08:49:41 +0000 | [diff] [blame] | 62 | VCMNackMode NackMode() const; |
Wan-Teh Chang | b1825a4 | 2015-06-03 15:03:35 -0700 | [diff] [blame] | 63 | std::vector<uint16_t> NackList(bool* request_key_frame); |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 64 | |
mikhal@webrtc.org | ef9f76a | 2013-02-15 23:22:18 +0000 | [diff] [blame] | 65 | // Receiver video delay. |
| 66 | int SetMinReceiverDelay(int desired_delay_ms); |
| 67 | |
mikhal@webrtc.org | dc3cd21 | 2013-04-25 20:27:04 +0000 | [diff] [blame] | 68 | // Decoding with errors. |
agalusza@google.com | a7e360e | 2013-08-01 03:15:08 +0000 | [diff] [blame] | 69 | void SetDecodeErrorMode(VCMDecodeErrorMode decode_error_mode); |
| 70 | VCMDecodeErrorMode DecodeErrorMode() const; |
mikhal@webrtc.org | dc3cd21 | 2013-04-25 20:27:04 +0000 | [diff] [blame] | 71 | |
mikhal@webrtc.org | 381da4b | 2013-04-25 21:45:29 +0000 | [diff] [blame] | 72 | // Returns size in time (milliseconds) of complete continuous frames in the |
mikhal@webrtc.org | 759b041 | 2013-05-07 16:36:00 +0000 | [diff] [blame] | 73 | // jitter buffer. The render time is estimated based on the render delay at |
| 74 | // the time this function is called. |
mikhal@webrtc.org | 381da4b | 2013-04-25 21:45:29 +0000 | [diff] [blame] | 75 | int RenderBufferSizeMs(); |
| 76 | |
pbos@webrtc.org | 5570769 | 2014-12-19 15:45:03 +0000 | [diff] [blame] | 77 | void RegisterStatsCallback(VCMReceiveStatisticsCallback* callback); |
pbos@webrtc.org | ce4e9a3 | 2014-12-18 13:50:16 +0000 | [diff] [blame] | 78 | |
pbos@webrtc.org | 4dd40d6 | 2015-02-17 13:22:43 +0000 | [diff] [blame] | 79 | void TriggerDecoderShutdown(); |
| 80 | |
stefan@webrtc.org | 1ea4b50 | 2013-01-07 08:49:41 +0000 | [diff] [blame] | 81 | private: |
stefan@webrtc.org | 1ea4b50 | 2013-01-07 08:49:41 +0000 | [diff] [blame] | 82 | CriticalSectionWrapper* crit_sect_; |
pbos@webrtc.org | 4dd40d6 | 2015-02-17 13:22:43 +0000 | [diff] [blame] | 83 | Clock* const clock_; |
stefan@webrtc.org | 1ea4b50 | 2013-01-07 08:49:41 +0000 | [diff] [blame] | 84 | VCMJitterBuffer jitter_buffer_; |
| 85 | VCMTiming* timing_; |
kwiberg@webrtc.org | 00b8f6b | 2015-02-26 14:34:55 +0000 | [diff] [blame] | 86 | rtc::scoped_ptr<EventWrapper> render_wait_event_; |
mikhal@webrtc.org | ef9f76a | 2013-02-15 23:22:18 +0000 | [diff] [blame] | 87 | int max_video_delay_ms_; |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 88 | }; |
| 89 | |
stefan@webrtc.org | 1ea4b50 | 2013-01-07 08:49:41 +0000 | [diff] [blame] | 90 | } // namespace webrtc |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 91 | |
stefan@webrtc.org | 1ea4b50 | 2013-01-07 08:49:41 +0000 | [diff] [blame] | 92 | #endif // WEBRTC_MODULES_VIDEO_CODING_MAIN_SOURCE_RECEIVER_H_ |