OLD | NEW |
| (Empty) |
1 /* | |
2 * Copyright (c) 2012 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 WEBRTC_MODULES_RTP_RTCP_SOURCE_PRODUCER_FEC_H_ | |
12 #define WEBRTC_MODULES_RTP_RTCP_SOURCE_PRODUCER_FEC_H_ | |
13 | |
14 #include <list> | |
15 #include <memory> | |
16 #include <vector> | |
17 | |
18 #include "webrtc/modules/rtp_rtcp/source/forward_error_correction.h" | |
19 | |
20 namespace webrtc { | |
21 | |
22 class RedPacket { | |
23 public: | |
24 explicit RedPacket(size_t length); | |
25 | |
26 void CreateHeader(const uint8_t* rtp_header, | |
27 size_t header_length, | |
28 int red_payload_type, | |
29 int payload_type); | |
30 void SetSeqNum(int seq_num); | |
31 void AssignPayload(const uint8_t* payload, size_t length); | |
32 void ClearMarkerBit(); | |
33 uint8_t* data() const; | |
34 size_t length() const; | |
35 | |
36 private: | |
37 std::unique_ptr<uint8_t[]> data_; | |
38 size_t length_; | |
39 size_t header_length_; | |
40 }; | |
41 | |
42 class ProducerFec { | |
43 public: | |
44 ProducerFec(); | |
45 ~ProducerFec(); | |
46 | |
47 static std::unique_ptr<RedPacket> BuildRedPacket(const uint8_t* data_buffer, | |
48 size_t payload_length, | |
49 size_t rtp_header_length, | |
50 int red_payload_type); | |
51 | |
52 void SetFecParameters(const FecProtectionParams* params); | |
53 | |
54 // Adds a media packet to the internal buffer. When enough media packets | |
55 // have been added, the FEC packets are generated and stored internally. | |
56 // These FEC packets are then obtained by calling GetFecPacketsAsRed(). | |
57 int AddRtpPacketAndGenerateFec(const uint8_t* data_buffer, | |
58 size_t payload_length, | |
59 size_t rtp_header_length); | |
60 | |
61 // Returns true if there are generated FEC packets available. | |
62 bool FecAvailable() const; | |
63 | |
64 size_t NumAvailableFecPackets() const; | |
65 | |
66 // Returns the overhead, per packet, for FEC (and possibly RED). | |
67 size_t MaxPacketOverhead() const; | |
68 | |
69 // Returns generated FEC packets with RED headers added. | |
70 std::vector<std::unique_ptr<RedPacket>> GetUlpfecPacketsAsRed( | |
71 int red_payload_type, | |
72 int ulpfec_payload_type, | |
73 uint16_t first_seq_num, | |
74 size_t rtp_header_length); | |
75 | |
76 private: | |
77 // Overhead is defined as relative to the number of media packets, and not | |
78 // relative to total number of packets. This definition is inherited from the | |
79 // protection factor produced by video_coding module and how the FEC | |
80 // generation is implemented. | |
81 int Overhead() const; | |
82 | |
83 // Returns true if the excess overhead (actual - target) for the FEC is below | |
84 // the amount |kMaxExcessOverhead|. This effects the lower protection level | |
85 // cases and low number of media packets/frame. The target overhead is given | |
86 // by |params_.fec_rate|, and is only achievable in the limit of large number | |
87 // of media packets. | |
88 bool ExcessOverheadBelowMax() const; | |
89 | |
90 // Returns true if the number of added media packets is at least | |
91 // |min_num_media_packets_|. This condition tries to capture the effect | |
92 // that, for the same amount of protection/overhead, longer codes | |
93 // (e.g. (2k,2m) vs (k,m)) are generally more effective at recovering losses. | |
94 bool MinimumMediaPacketsReached() const; | |
95 | |
96 void ResetState(); | |
97 | |
98 std::unique_ptr<ForwardErrorCorrection> fec_; | |
99 ForwardErrorCorrection::PacketList media_packets_; | |
100 std::list<ForwardErrorCorrection::Packet*> generated_fec_packets_; | |
101 int num_protected_frames_; | |
102 int min_num_media_packets_; | |
103 FecProtectionParams params_; | |
104 FecProtectionParams new_params_; | |
105 }; | |
106 | |
107 } // namespace webrtc | |
108 | |
109 #endif // WEBRTC_MODULES_RTP_RTCP_SOURCE_PRODUCER_FEC_H_ | |
OLD | NEW |