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

Side by Side Diff: webrtc/modules/rtp_rtcp/source/producer_fec_unittest.cc

Issue 2305793003: Simplify public interface of ProducerFec. (Closed)
Patch Set: Fix fuzzer. Created 4 years, 2 months 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) 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
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(&params, 0); // Expecting one FEC packet. 114 producer_.SetFecParameters(&params, 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
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
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
OLDNEW
« no previous file with comments | « webrtc/modules/rtp_rtcp/source/producer_fec.cc ('k') | webrtc/modules/rtp_rtcp/source/rtp_sender_video.cc » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698