OLD | NEW |
1 /* | 1 /* |
2 * Copyright (c) 2012 The WebRTC project authors. All Rights Reserved. | 2 * Copyright (c) 2012 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 |
(...skipping 76 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
87 if (p.marker_bit) { | 87 if (p.marker_bit) { |
88 packet[1] |= 0x80; | 88 packet[1] |= 0x80; |
89 } else { | 89 } else { |
90 packet[1] &= ~0x80; | 90 packet[1] &= ~0x80; |
91 } | 91 } |
92 ByteWriter<uint16_t>::WriteBigEndian(&packet[2], p.seq_num); | 92 ByteWriter<uint16_t>::WriteBigEndian(&packet[2], p.seq_num); |
93 producer_.AddRtpPacketAndGenerateFec(packet, p.payload_size, p.header_size); | 93 producer_.AddRtpPacketAndGenerateFec(packet, p.payload_size, p.header_size); |
94 size_t num_fec_packets = producer_.NumAvailableFecPackets(); | 94 size_t num_fec_packets = producer_.NumAvailableFecPackets(); |
95 if (num_fec_packets > 0) { | 95 if (num_fec_packets > 0) { |
96 std::vector<std::unique_ptr<RedPacket>> fec_packets = | 96 std::vector<std::unique_ptr<RedPacket>> fec_packets = |
97 producer_.GetFecPacketsAsRed(kRedPayloadType, kFecPayloadType, 100, | 97 producer_.GetUlpfecPacketsAsRed(kRedPayloadType, kFecPayloadType, 100, |
98 p.header_size); | 98 p.header_size); |
99 EXPECT_EQ(num_fec_packets, fec_packets.size()); | 99 EXPECT_EQ(num_fec_packets, fec_packets.size()); |
100 } | 100 } |
101 } | 101 } |
102 } | 102 } |
103 | 103 |
104 TEST_F(ProducerFecTest, OneFrameFec) { | 104 TEST_F(ProducerFecTest, OneFrameFec) { |
105 // The number of media packets (|kNumPackets|), number of frames (one for | 105 // The number of media packets (|kNumPackets|), number of frames (one for |
106 // this test), and the protection factor (|params->fec_rate|) are set to make | 106 // this test), and the protection factor (|params->fec_rate|) are set to make |
107 // sure the conditions for generating FEC are satisfied. This means: | 107 // sure the conditions for generating FEC are satisfied. This means: |
108 // (1) protection factor is high enough so that actual overhead over 1 frame | 108 // (1) protection factor is high enough so that actual overhead over 1 frame |
109 // of packets is within |kMaxExcessOverhead|, and (2) the total number of | 109 // of packets is within |kMaxExcessOverhead|, and (2) the total number of |
110 // media packets for 1 frame is at least |minimum_media_packets_fec_|. | 110 // media packets for 1 frame is at least |minimum_media_packets_fec_|. |
111 constexpr size_t kNumPackets = 4; | 111 constexpr size_t kNumPackets = 4; |
112 FecProtectionParams params = {15, 3, kFecMaskRandom}; | 112 FecProtectionParams params = {15, 3, kFecMaskRandom}; |
113 packet_generator_.NewFrame(kNumPackets); | 113 packet_generator_.NewFrame(kNumPackets); |
114 producer_.SetFecParameters(¶ms, 0); // Expecting one FEC packet. | 114 producer_.SetFecParameters(¶ms, 0); // Expecting one FEC packet. |
115 uint32_t last_timestamp = 0; | 115 uint32_t last_timestamp = 0; |
116 for (size_t i = 0; i < kNumPackets; ++i) { | 116 for (size_t i = 0; i < kNumPackets; ++i) { |
117 std::unique_ptr<AugmentedPacket> packet = | 117 std::unique_ptr<AugmentedPacket> packet = |
118 packet_generator_.NextPacket(i, 10); | 118 packet_generator_.NextPacket(i, 10); |
119 EXPECT_EQ(0, producer_.AddRtpPacketAndGenerateFec( | 119 EXPECT_EQ(0, producer_.AddRtpPacketAndGenerateFec( |
120 packet->data, packet->length, kRtpHeaderSize)); | 120 packet->data, packet->length, kRtpHeaderSize)); |
121 last_timestamp = packet->header.header.timestamp; | 121 last_timestamp = packet->header.header.timestamp; |
122 } | 122 } |
123 EXPECT_TRUE(producer_.FecAvailable()); | 123 EXPECT_TRUE(producer_.FecAvailable()); |
124 uint16_t seq_num = packet_generator_.NextPacketSeqNum(); | 124 uint16_t seq_num = packet_generator_.NextPacketSeqNum(); |
125 std::vector<std::unique_ptr<RedPacket>> red_packets = | 125 std::vector<std::unique_ptr<RedPacket>> red_packets = |
126 producer_.GetFecPacketsAsRed(kRedPayloadType, kFecPayloadType, seq_num, | 126 producer_.GetUlpfecPacketsAsRed(kRedPayloadType, kFecPayloadType, seq_num, |
127 kRtpHeaderSize); | 127 kRtpHeaderSize); |
128 EXPECT_FALSE(producer_.FecAvailable()); | 128 EXPECT_FALSE(producer_.FecAvailable()); |
129 ASSERT_EQ(1u, red_packets.size()); | 129 ASSERT_EQ(1u, red_packets.size()); |
130 VerifyHeader(seq_num, last_timestamp, kRedPayloadType, kFecPayloadType, | 130 VerifyHeader(seq_num, last_timestamp, kRedPayloadType, kFecPayloadType, |
131 red_packets.front().get(), false); | 131 red_packets.front().get(), false); |
132 } | 132 } |
133 | 133 |
134 TEST_F(ProducerFecTest, TwoFrameFec) { | 134 TEST_F(ProducerFecTest, TwoFrameFec) { |
135 // The number of media packets/frame (|kNumPackets|), the number of frames | 135 // The number of media packets/frame (|kNumPackets|), the number of frames |
136 // (|kNumFrames|), and the protection factor (|params->fec_rate|) are set to | 136 // (|kNumFrames|), and the protection factor (|params->fec_rate|) are set to |
137 // make sure the conditions for generating FEC are satisfied. This means: | 137 // make sure the conditions for generating FEC are satisfied. This means: |
(...skipping 13 matching lines...) Expand all Loading... |
151 std::unique_ptr<AugmentedPacket> packet = | 151 std::unique_ptr<AugmentedPacket> packet = |
152 packet_generator_.NextPacket(i * kNumPackets + j, 10); | 152 packet_generator_.NextPacket(i * kNumPackets + j, 10); |
153 EXPECT_EQ(0, producer_.AddRtpPacketAndGenerateFec( | 153 EXPECT_EQ(0, producer_.AddRtpPacketAndGenerateFec( |
154 packet->data, packet->length, kRtpHeaderSize)); | 154 packet->data, packet->length, kRtpHeaderSize)); |
155 last_timestamp = packet->header.header.timestamp; | 155 last_timestamp = packet->header.header.timestamp; |
156 } | 156 } |
157 } | 157 } |
158 EXPECT_TRUE(producer_.FecAvailable()); | 158 EXPECT_TRUE(producer_.FecAvailable()); |
159 uint16_t seq_num = packet_generator_.NextPacketSeqNum(); | 159 uint16_t seq_num = packet_generator_.NextPacketSeqNum(); |
160 std::vector<std::unique_ptr<RedPacket>> red_packets = | 160 std::vector<std::unique_ptr<RedPacket>> red_packets = |
161 producer_.GetFecPacketsAsRed(kRedPayloadType, kFecPayloadType, seq_num, | 161 producer_.GetUlpfecPacketsAsRed(kRedPayloadType, kFecPayloadType, seq_num, |
162 kRtpHeaderSize); | 162 kRtpHeaderSize); |
163 EXPECT_FALSE(producer_.FecAvailable()); | 163 EXPECT_FALSE(producer_.FecAvailable()); |
164 ASSERT_EQ(1u, red_packets.size()); | 164 ASSERT_EQ(1u, red_packets.size()); |
165 VerifyHeader(seq_num, last_timestamp, kRedPayloadType, kFecPayloadType, | 165 VerifyHeader(seq_num, last_timestamp, kRedPayloadType, kFecPayloadType, |
166 red_packets.front().get(), false); | 166 red_packets.front().get(), false); |
167 } | 167 } |
168 | 168 |
169 TEST_F(ProducerFecTest, BuildRedPacket) { | 169 TEST_F(ProducerFecTest, BuildRedPacket) { |
170 packet_generator_.NewFrame(1); | 170 packet_generator_.NewFrame(1); |
171 std::unique_ptr<AugmentedPacket> packet = packet_generator_.NextPacket(0, 10); | 171 std::unique_ptr<AugmentedPacket> packet = packet_generator_.NextPacket(0, 10); |
172 std::unique_ptr<RedPacket> red_packet = | 172 std::unique_ptr<RedPacket> red_packet = |
(...skipping 21 matching lines...) Expand all Loading... |
194 ProducerFec::BuildRedPacket(packet->data, packet->length - kRtpHeaderSize, | 194 ProducerFec::BuildRedPacket(packet->data, packet->length - kRtpHeaderSize, |
195 kRtpHeaderSize, kRedPayloadType); | 195 kRtpHeaderSize, kRedPayloadType); |
196 EXPECT_EQ(packet->length + kRedForFecHeaderLength, red_packet->length()); | 196 EXPECT_EQ(packet->length + kRedForFecHeaderLength, red_packet->length()); |
197 VerifyHeader(packet->header.header.sequenceNumber, | 197 VerifyHeader(packet->header.header.sequenceNumber, |
198 packet->header.header.timestamp, kRedPayloadType, | 198 packet->header.header.timestamp, kRedPayloadType, |
199 packet->header.header.payloadType, red_packet.get(), | 199 packet->header.header.payloadType, red_packet.get(), |
200 true); // Marker bit set. | 200 true); // Marker bit set. |
201 } | 201 } |
202 | 202 |
203 } // namespace webrtc | 203 } // namespace webrtc |
OLD | NEW |