OLD | NEW |
1 /* | 1 /* |
2 * Copyright (c) 2016 The WebRTC project authors. All Rights Reserved. | 2 * Copyright (c) 2016 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_RTP_RTCP_INCLUDE_FLEXFEC_SENDER_H_ | 11 #ifndef WEBRTC_MODULES_RTP_RTCP_INCLUDE_FLEXFEC_SENDER_H_ |
12 #define WEBRTC_MODULES_RTP_RTCP_INCLUDE_FLEXFEC_SENDER_H_ | 12 #define WEBRTC_MODULES_RTP_RTCP_INCLUDE_FLEXFEC_SENDER_H_ |
13 | 13 |
14 #include <memory> | 14 #include <memory> |
15 #include <vector> | 15 #include <vector> |
16 | 16 |
17 #include "webrtc/base/array_view.h" | 17 #include "webrtc/base/array_view.h" |
18 #include "webrtc/base/basictypes.h" | 18 #include "webrtc/base/basictypes.h" |
19 #include "webrtc/base/random.h" | 19 #include "webrtc/base/random.h" |
20 #include "webrtc/base/sequenced_task_checker.h" | |
21 #include "webrtc/config.h" | 20 #include "webrtc/config.h" |
22 #include "webrtc/modules/include/module_common_types.h" | 21 #include "webrtc/modules/include/module_common_types.h" |
23 #include "webrtc/modules/rtp_rtcp/include/flexfec_sender.h" | 22 #include "webrtc/modules/rtp_rtcp/include/flexfec_sender.h" |
| 23 #include "webrtc/modules/rtp_rtcp/include/rtp_rtcp_defines.h" |
24 #include "webrtc/modules/rtp_rtcp/source/rtp_header_extension.h" | 24 #include "webrtc/modules/rtp_rtcp/source/rtp_header_extension.h" |
25 #include "webrtc/modules/rtp_rtcp/source/rtp_packet_to_send.h" | 25 #include "webrtc/modules/rtp_rtcp/source/rtp_packet_to_send.h" |
26 #include "webrtc/modules/rtp_rtcp/source/ulpfec_generator.h" | 26 #include "webrtc/modules/rtp_rtcp/source/ulpfec_generator.h" |
27 #include "webrtc/system_wrappers/include/clock.h" | 27 #include "webrtc/system_wrappers/include/clock.h" |
28 | 28 |
29 namespace webrtc { | 29 namespace webrtc { |
30 | 30 |
31 class RtpPacketToSend; | 31 class RtpPacketToSend; |
32 | 32 |
33 // Note that this class is not thread safe, and thus requires external | 33 // Note that this class is not thread safe, and thus requires external |
34 // synchronization. Currently, this is done using the lock in PayloadRouter. | 34 // synchronization. Currently, this is done using the lock in PayloadRouter. |
35 | 35 |
36 class FlexfecSender { | 36 class FlexfecSender { |
37 public: | 37 public: |
38 FlexfecSender(int payload_type, | 38 FlexfecSender(int payload_type, |
39 uint32_t ssrc, | 39 uint32_t ssrc, |
40 uint32_t protected_media_ssrc, | 40 uint32_t protected_media_ssrc, |
41 const std::vector<RtpExtension>& rtp_header_extensions, | 41 const std::vector<RtpExtension>& rtp_header_extensions, |
42 rtc::ArrayView<const RtpExtensionSize> extension_sizes, | 42 rtc::ArrayView<const RtpExtensionSize> extension_sizes, |
| 43 const RtpState* rtp_state, |
43 Clock* clock); | 44 Clock* clock); |
44 ~FlexfecSender(); | 45 ~FlexfecSender(); |
45 | 46 |
46 uint32_t ssrc() const { return ssrc_; } | 47 uint32_t ssrc() const { return ssrc_; } |
47 | 48 |
48 // Sets the FEC rate, max frames sent before FEC packets are sent, | 49 // Sets the FEC rate, max frames sent before FEC packets are sent, |
49 // and what type of generator matrices are used. | 50 // and what type of generator matrices are used. |
50 void SetFecParameters(const FecProtectionParams& params); | 51 void SetFecParameters(const FecProtectionParams& params); |
51 | 52 |
52 // Adds a media packet to the internal buffer. When enough media packets | 53 // Adds a media packet to the internal buffer. When enough media packets |
53 // have been added, the FEC packets are generated and stored internally. | 54 // have been added, the FEC packets are generated and stored internally. |
54 // These FEC packets are then obtained by calling GetFecPackets(). | 55 // These FEC packets are then obtained by calling GetFecPackets(). |
55 // Returns true if the media packet was successfully added. | 56 // Returns true if the media packet was successfully added. |
56 bool AddRtpPacketAndGenerateFec(const RtpPacketToSend& packet); | 57 bool AddRtpPacketAndGenerateFec(const RtpPacketToSend& packet); |
57 | 58 |
58 // Returns true if there are generated FEC packets available. | 59 // Returns true if there are generated FEC packets available. |
59 bool FecAvailable() const; | 60 bool FecAvailable() const; |
60 | 61 |
61 // Returns generated FlexFEC packets. | 62 // Returns generated FlexFEC packets. |
62 std::vector<std::unique_ptr<RtpPacketToSend>> GetFecPackets(); | 63 std::vector<std::unique_ptr<RtpPacketToSend>> GetFecPackets(); |
63 | 64 |
64 // Returns the overhead, per packet, for FlexFEC. | 65 // Returns the overhead, per packet, for FlexFEC. |
65 size_t MaxPacketOverhead() const; | 66 size_t MaxPacketOverhead() const; |
66 | 67 |
| 68 // Only called on the VideoSendStream queue, after operation has shut down. |
| 69 RtpState GetRtpState(); |
| 70 |
67 private: | 71 private: |
68 // Utility. | 72 // Utility. |
69 Clock* const clock_; | 73 Clock* const clock_; |
70 Random random_; | 74 Random random_; |
71 int64_t last_generated_packet_ms_; | 75 int64_t last_generated_packet_ms_; |
72 | 76 |
73 // Config. | 77 // Config. |
74 const int payload_type_; | 78 const int payload_type_; |
75 const uint32_t timestamp_offset_; | 79 const uint32_t timestamp_offset_; |
76 const uint32_t ssrc_; | 80 const uint32_t ssrc_; |
77 const uint32_t protected_media_ssrc_; | 81 const uint32_t protected_media_ssrc_; |
78 // Sequence number of next packet to generate. | 82 // Sequence number of next packet to generate. |
79 uint16_t seq_num_; | 83 uint16_t seq_num_; |
80 | 84 |
81 // Implementation. | 85 // Implementation. |
82 UlpfecGenerator ulpfec_generator_; | 86 UlpfecGenerator ulpfec_generator_; |
83 const RtpHeaderExtensionMap rtp_header_extension_map_; | 87 const RtpHeaderExtensionMap rtp_header_extension_map_; |
84 const size_t header_extensions_size_; | 88 const size_t header_extensions_size_; |
85 }; | 89 }; |
86 | 90 |
87 } // namespace webrtc | 91 } // namespace webrtc |
88 | 92 |
89 #endif // WEBRTC_MODULES_RTP_RTCP_INCLUDE_FLEXFEC_SENDER_H_ | 93 #endif // WEBRTC_MODULES_RTP_RTCP_INCLUDE_FLEXFEC_SENDER_H_ |
OLD | NEW |