| 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 94 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 105 | 105 |
| 106 TEST_F(ProducerFecTest, OneFrameFec) { | 106 TEST_F(ProducerFecTest, OneFrameFec) { |
| 107 // The number of media packets (|kNumPackets|), number of frames (one for | 107 // The number of media packets (|kNumPackets|), number of frames (one for |
| 108 // this test), and the protection factor (|params->fec_rate|) are set to make | 108 // this test), and the protection factor (|params->fec_rate|) are set to make |
| 109 // sure the conditions for generating FEC are satisfied. This means: | 109 // sure the conditions for generating FEC are satisfied. This means: |
| 110 // (1) protection factor is high enough so that actual overhead over 1 frame | 110 // (1) protection factor is high enough so that actual overhead over 1 frame |
| 111 // of packets is within |kMaxExcessOverhead|, and (2) the total number of | 111 // of packets is within |kMaxExcessOverhead|, and (2) the total number of |
| 112 // media packets for 1 frame is at least |minimum_media_packets_fec_|. | 112 // media packets for 1 frame is at least |minimum_media_packets_fec_|. |
| 113 const int kNumPackets = 4; | 113 const int kNumPackets = 4; |
| 114 FecProtectionParams params = {15, false, 3}; | 114 FecProtectionParams params = {15, false, 3}; |
| 115 std::list<RtpPacket*> rtp_packets; | 115 std::list<test::RawRtpPacket*> rtp_packets; |
| 116 generator_->NewFrame(kNumPackets); | 116 generator_->NewFrame(kNumPackets); |
| 117 producer_->SetFecParameters(¶ms, 0); // Expecting one FEC packet. | 117 producer_->SetFecParameters(¶ms, 0); // Expecting one FEC packet. |
| 118 uint32_t last_timestamp = 0; | 118 uint32_t last_timestamp = 0; |
| 119 for (int i = 0; i < kNumPackets; ++i) { | 119 for (int i = 0; i < kNumPackets; ++i) { |
| 120 RtpPacket* rtp_packet = generator_->NextPacket(i, 10); | 120 test::RawRtpPacket* rtp_packet = generator_->NextPacket(i, 10); |
| 121 rtp_packets.push_back(rtp_packet); | 121 rtp_packets.push_back(rtp_packet); |
| 122 EXPECT_EQ(0, producer_->AddRtpPacketAndGenerateFec(rtp_packet->data, | 122 EXPECT_EQ(0, producer_->AddRtpPacketAndGenerateFec(rtp_packet->data, |
| 123 rtp_packet->length, | 123 rtp_packet->length, |
| 124 kRtpHeaderSize)); | 124 kRtpHeaderSize)); |
| 125 last_timestamp = rtp_packet->header.header.timestamp; | 125 last_timestamp = rtp_packet->header.header.timestamp; |
| 126 } | 126 } |
| 127 EXPECT_TRUE(producer_->FecAvailable()); | 127 EXPECT_TRUE(producer_->FecAvailable()); |
| 128 uint16_t seq_num = generator_->NextSeqNum(); | 128 uint16_t seq_num = generator_->NextSeqNum(); |
| 129 std::vector<RedPacket*> packets = producer_->GetFecPackets(kRedPayloadType, | 129 std::vector<RedPacket*> packets = producer_->GetFecPackets(kRedPayloadType, |
| 130 kFecPayloadType, | 130 kFecPayloadType, |
| (...skipping 15 matching lines...) Expand all Loading... |
| 146 // (|kNumFrames|), and the protection factor (|params->fec_rate|) are set to | 146 // (|kNumFrames|), and the protection factor (|params->fec_rate|) are set to |
| 147 // make sure the conditions for generating FEC are satisfied. This means: | 147 // make sure the conditions for generating FEC are satisfied. This means: |
| 148 // (1) protection factor is high enough so that actual overhead over | 148 // (1) protection factor is high enough so that actual overhead over |
| 149 // |kNumFrames| is within |kMaxExcessOverhead|, and (2) the total number of | 149 // |kNumFrames| is within |kMaxExcessOverhead|, and (2) the total number of |
| 150 // media packets for |kNumFrames| frames is at least | 150 // media packets for |kNumFrames| frames is at least |
| 151 // |minimum_media_packets_fec_|. | 151 // |minimum_media_packets_fec_|. |
| 152 const int kNumPackets = 2; | 152 const int kNumPackets = 2; |
| 153 const int kNumFrames = 2; | 153 const int kNumFrames = 2; |
| 154 | 154 |
| 155 FecProtectionParams params = {15, 0, 3}; | 155 FecProtectionParams params = {15, 0, 3}; |
| 156 std::list<RtpPacket*> rtp_packets; | 156 std::list<test::RawRtpPacket*> rtp_packets; |
| 157 producer_->SetFecParameters(¶ms, 0); // Expecting one FEC packet. | 157 producer_->SetFecParameters(¶ms, 0); // Expecting one FEC packet. |
| 158 uint32_t last_timestamp = 0; | 158 uint32_t last_timestamp = 0; |
| 159 for (int i = 0; i < kNumFrames; ++i) { | 159 for (int i = 0; i < kNumFrames; ++i) { |
| 160 generator_->NewFrame(kNumPackets); | 160 generator_->NewFrame(kNumPackets); |
| 161 for (int j = 0; j < kNumPackets; ++j) { | 161 for (int j = 0; j < kNumPackets; ++j) { |
| 162 RtpPacket* rtp_packet = generator_->NextPacket(i * kNumPackets + j, 10); | 162 test::RawRtpPacket* rtp_packet = |
| 163 generator_->NextPacket(i * kNumPackets + j, 10); |
| 163 rtp_packets.push_back(rtp_packet); | 164 rtp_packets.push_back(rtp_packet); |
| 164 EXPECT_EQ(0, producer_->AddRtpPacketAndGenerateFec(rtp_packet->data, | 165 EXPECT_EQ(0, producer_->AddRtpPacketAndGenerateFec(rtp_packet->data, |
| 165 rtp_packet->length, | 166 rtp_packet->length, |
| 166 kRtpHeaderSize)); | 167 kRtpHeaderSize)); |
| 167 last_timestamp = rtp_packet->header.header.timestamp; | 168 last_timestamp = rtp_packet->header.header.timestamp; |
| 168 } | 169 } |
| 169 } | 170 } |
| 170 EXPECT_TRUE(producer_->FecAvailable()); | 171 EXPECT_TRUE(producer_->FecAvailable()); |
| 171 uint16_t seq_num = generator_->NextSeqNum(); | 172 uint16_t seq_num = generator_->NextSeqNum(); |
| 172 std::vector<RedPacket*> packets = producer_->GetFecPackets(kRedPayloadType, | 173 std::vector<RedPacket*> packets = producer_->GetFecPackets(kRedPayloadType, |
| 173 kFecPayloadType, | 174 kFecPayloadType, |
| 174 seq_num, | 175 seq_num, |
| 175 kRtpHeaderSize); | 176 kRtpHeaderSize); |
| 176 EXPECT_FALSE(producer_->FecAvailable()); | 177 EXPECT_FALSE(producer_->FecAvailable()); |
| 177 ASSERT_EQ(1u, packets.size()); | 178 ASSERT_EQ(1u, packets.size()); |
| 178 VerifyHeader(seq_num, last_timestamp, kRedPayloadType, kFecPayloadType, | 179 VerifyHeader(seq_num, last_timestamp, kRedPayloadType, kFecPayloadType, |
| 179 packets.front(), false); | 180 packets.front(), false); |
| 180 while (!rtp_packets.empty()) { | 181 while (!rtp_packets.empty()) { |
| 181 delete rtp_packets.front(); | 182 delete rtp_packets.front(); |
| 182 rtp_packets.pop_front(); | 183 rtp_packets.pop_front(); |
| 183 } | 184 } |
| 184 delete packets.front(); | 185 delete packets.front(); |
| 185 } | 186 } |
| 186 | 187 |
| 187 TEST_F(ProducerFecTest, BuildRedPacket) { | 188 TEST_F(ProducerFecTest, BuildRedPacket) { |
| 188 generator_->NewFrame(1); | 189 generator_->NewFrame(1); |
| 189 RtpPacket* packet = generator_->NextPacket(0, 10); | 190 test::RawRtpPacket* packet = generator_->NextPacket(0, 10); |
| 190 rtc::scoped_ptr<RedPacket> red_packet(producer_->BuildRedPacket( | 191 rtc::scoped_ptr<RedPacket> red_packet(producer_->BuildRedPacket( |
| 191 packet->data, packet->length - kRtpHeaderSize, kRtpHeaderSize, | 192 packet->data, packet->length - kRtpHeaderSize, kRtpHeaderSize, |
| 192 kRedPayloadType)); | 193 kRedPayloadType)); |
| 193 EXPECT_EQ(packet->length + 1, red_packet->length()); | 194 EXPECT_EQ(packet->length + 1, red_packet->length()); |
| 194 VerifyHeader(packet->header.header.sequenceNumber, | 195 VerifyHeader(packet->header.header.sequenceNumber, |
| 195 packet->header.header.timestamp, | 196 packet->header.header.timestamp, |
| 196 kRedPayloadType, | 197 kRedPayloadType, |
| 197 packet->header.header.payloadType, | 198 packet->header.header.payloadType, |
| 198 red_packet.get(), | 199 red_packet.get(), |
| 199 true); // Marker bit set. | 200 true); // Marker bit set. |
| 200 for (int i = 0; i < 10; ++i) | 201 for (int i = 0; i < 10; ++i) |
| 201 EXPECT_EQ(i, red_packet->data()[kRtpHeaderSize + 1 + i]); | 202 EXPECT_EQ(i, red_packet->data()[kRtpHeaderSize + 1 + i]); |
| 202 delete packet; | 203 delete packet; |
| 203 } | 204 } |
| 204 | 205 |
| 205 } // namespace webrtc | 206 } // namespace webrtc |
| OLD | NEW |