| OLD | NEW |
| 1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "base/test/simple_test_tick_clock.h" | 5 #include "base/test/simple_test_tick_clock.h" |
| 6 #include "media/cast/framer/framer.h" | 6 #include "media/cast/framer/framer.h" |
| 7 #include "media/cast/rtp_receiver/mock_rtp_payload_feedback.h" | 7 #include "media/cast/rtp_receiver/mock_rtp_payload_feedback.h" |
| 8 #include "testing/gtest/include/gtest/gtest.h" | 8 #include "testing/gtest/include/gtest/gtest.h" |
| 9 | 9 |
| 10 namespace media { | 10 namespace media { |
| (...skipping 15 matching lines...) Expand all Loading... |
| 26 std::vector<uint8> payload_; | 26 std::vector<uint8> payload_; |
| 27 RtpCastHeader rtp_header_; | 27 RtpCastHeader rtp_header_; |
| 28 MockRtpPayloadFeedback mock_rtp_payload_feedback_; | 28 MockRtpPayloadFeedback mock_rtp_payload_feedback_; |
| 29 Framer framer_; | 29 Framer framer_; |
| 30 base::SimpleTestTickClock testing_clock_; | 30 base::SimpleTestTickClock testing_clock_; |
| 31 | 31 |
| 32 DISALLOW_COPY_AND_ASSIGN(FramerTest); | 32 DISALLOW_COPY_AND_ASSIGN(FramerTest); |
| 33 }; | 33 }; |
| 34 | 34 |
| 35 TEST_F(FramerTest, EmptyState) { | 35 TEST_F(FramerTest, EmptyState) { |
| 36 transport::EncodedVideoFrame frame; | 36 transport::EncodedFrame frame; |
| 37 bool next_frame = false; | 37 bool next_frame = false; |
| 38 EXPECT_FALSE(framer_.GetEncodedVideoFrame(&frame, &next_frame)); | 38 EXPECT_FALSE(framer_.GetEncodedVideoFrame(&frame, &next_frame)); |
| 39 } | 39 } |
| 40 | 40 |
| 41 TEST_F(FramerTest, AlwaysStartWithKey) { | 41 TEST_F(FramerTest, AlwaysStartWithKey) { |
| 42 transport::EncodedVideoFrame frame; | 42 transport::EncodedFrame frame; |
| 43 bool next_frame = false; | 43 bool next_frame = false; |
| 44 bool complete = false; | 44 bool complete = false; |
| 45 bool duplicate = false; | 45 bool duplicate = false; |
| 46 | 46 |
| 47 // Insert non key first frame. | 47 // Insert non key first frame. |
| 48 complete = framer_.InsertPacket( | 48 complete = framer_.InsertPacket( |
| 49 payload_.data(), payload_.size(), rtp_header_, &duplicate); | 49 payload_.data(), payload_.size(), rtp_header_, &duplicate); |
| 50 EXPECT_TRUE(complete); | 50 EXPECT_TRUE(complete); |
| 51 EXPECT_FALSE(framer_.GetEncodedVideoFrame(&frame, &next_frame)); | 51 EXPECT_FALSE(framer_.GetEncodedVideoFrame(&frame, &next_frame)); |
| 52 rtp_header_.frame_id = 1; | 52 rtp_header_.frame_id = 1; |
| 53 rtp_header_.reference_frame_id = 1; | 53 rtp_header_.reference_frame_id = 1; |
| 54 rtp_header_.is_key_frame = true; | 54 rtp_header_.is_key_frame = true; |
| 55 complete = framer_.InsertPacket( | 55 complete = framer_.InsertPacket( |
| 56 payload_.data(), payload_.size(), rtp_header_, &duplicate); | 56 payload_.data(), payload_.size(), rtp_header_, &duplicate); |
| 57 EXPECT_TRUE(complete); | 57 EXPECT_TRUE(complete); |
| 58 EXPECT_TRUE(framer_.GetEncodedVideoFrame(&frame, &next_frame)); | 58 EXPECT_TRUE(framer_.GetEncodedVideoFrame(&frame, &next_frame)); |
| 59 EXPECT_TRUE(next_frame); | 59 EXPECT_TRUE(next_frame); |
| 60 EXPECT_EQ(transport::EncodedFrame::KEY, frame.dependency); |
| 60 EXPECT_EQ(1u, frame.frame_id); | 61 EXPECT_EQ(1u, frame.frame_id); |
| 61 EXPECT_TRUE(frame.key_frame); | 62 EXPECT_EQ(1u, frame.referenced_frame_id); |
| 62 framer_.ReleaseFrame(frame.frame_id); | 63 framer_.ReleaseFrame(frame.frame_id); |
| 63 } | 64 } |
| 64 | 65 |
| 65 TEST_F(FramerTest, CompleteFrame) { | 66 TEST_F(FramerTest, CompleteFrame) { |
| 66 transport::EncodedVideoFrame frame; | 67 transport::EncodedFrame frame; |
| 67 bool next_frame = false; | 68 bool next_frame = false; |
| 68 bool complete = false; | 69 bool complete = false; |
| 69 bool duplicate = false; | 70 bool duplicate = false; |
| 70 | 71 |
| 71 // Start with a complete key frame. | 72 // Start with a complete key frame. |
| 72 rtp_header_.is_key_frame = true; | 73 rtp_header_.is_key_frame = true; |
| 73 complete = framer_.InsertPacket( | 74 complete = framer_.InsertPacket( |
| 74 payload_.data(), payload_.size(), rtp_header_, &duplicate); | 75 payload_.data(), payload_.size(), rtp_header_, &duplicate); |
| 75 EXPECT_TRUE(complete); | 76 EXPECT_TRUE(complete); |
| 76 EXPECT_TRUE(framer_.GetEncodedVideoFrame(&frame, &next_frame)); | 77 EXPECT_TRUE(framer_.GetEncodedVideoFrame(&frame, &next_frame)); |
| 77 EXPECT_TRUE(next_frame); | 78 EXPECT_TRUE(next_frame); |
| 79 EXPECT_EQ(transport::EncodedFrame::KEY, frame.dependency); |
| 78 EXPECT_EQ(0u, frame.frame_id); | 80 EXPECT_EQ(0u, frame.frame_id); |
| 79 EXPECT_TRUE(frame.key_frame); | 81 EXPECT_EQ(0u, frame.referenced_frame_id); |
| 80 framer_.ReleaseFrame(frame.frame_id); | 82 framer_.ReleaseFrame(frame.frame_id); |
| 81 | 83 |
| 82 // Incomplete delta. | 84 // Incomplete delta. |
| 83 ++rtp_header_.frame_id; | 85 ++rtp_header_.frame_id; |
| 84 rtp_header_.reference_frame_id = rtp_header_.frame_id - 1; | 86 rtp_header_.reference_frame_id = rtp_header_.frame_id - 1; |
| 85 rtp_header_.is_key_frame = false; | 87 rtp_header_.is_key_frame = false; |
| 86 rtp_header_.max_packet_id = 2; | 88 rtp_header_.max_packet_id = 2; |
| 87 complete = framer_.InsertPacket( | 89 complete = framer_.InsertPacket( |
| 88 payload_.data(), payload_.size(), rtp_header_, &duplicate); | 90 payload_.data(), payload_.size(), rtp_header_, &duplicate); |
| 89 EXPECT_FALSE(complete); | 91 EXPECT_FALSE(complete); |
| 90 EXPECT_FALSE(framer_.GetEncodedVideoFrame(&frame, &next_frame)); | 92 EXPECT_FALSE(framer_.GetEncodedVideoFrame(&frame, &next_frame)); |
| 91 | 93 |
| 92 // Complete delta - can't skip, as incomplete sequence. | 94 // Complete delta - can't skip, as incomplete sequence. |
| 93 ++rtp_header_.frame_id; | 95 ++rtp_header_.frame_id; |
| 94 rtp_header_.reference_frame_id = rtp_header_.frame_id - 1; | 96 rtp_header_.reference_frame_id = rtp_header_.frame_id - 1; |
| 95 rtp_header_.max_packet_id = 0; | 97 rtp_header_.max_packet_id = 0; |
| 96 complete = framer_.InsertPacket( | 98 complete = framer_.InsertPacket( |
| 97 payload_.data(), payload_.size(), rtp_header_, &duplicate); | 99 payload_.data(), payload_.size(), rtp_header_, &duplicate); |
| 98 EXPECT_TRUE(complete); | 100 EXPECT_TRUE(complete); |
| 99 EXPECT_FALSE(framer_.GetEncodedVideoFrame(&frame, &next_frame)); | 101 EXPECT_FALSE(framer_.GetEncodedVideoFrame(&frame, &next_frame)); |
| 100 } | 102 } |
| 101 | 103 |
| 102 TEST_F(FramerTest, DuplicatePackets) { | 104 TEST_F(FramerTest, DuplicatePackets) { |
| 103 transport::EncodedVideoFrame frame; | 105 transport::EncodedFrame frame; |
| 104 bool next_frame = false; | 106 bool next_frame = false; |
| 105 bool complete = false; | 107 bool complete = false; |
| 106 bool duplicate = false; | 108 bool duplicate = false; |
| 107 | 109 |
| 108 // Start with an incomplete key frame. | 110 // Start with an incomplete key frame. |
| 109 rtp_header_.is_key_frame = true; | 111 rtp_header_.is_key_frame = true; |
| 110 rtp_header_.max_packet_id = 1; | 112 rtp_header_.max_packet_id = 1; |
| 111 duplicate = true; | 113 duplicate = true; |
| 112 complete = framer_.InsertPacket( | 114 complete = framer_.InsertPacket( |
| 113 payload_.data(), payload_.size(), rtp_header_, &duplicate); | 115 payload_.data(), payload_.size(), rtp_header_, &duplicate); |
| (...skipping 10 matching lines...) Expand all Loading... |
| 124 EXPECT_FALSE(framer_.GetEncodedVideoFrame(&frame, &next_frame)); | 126 EXPECT_FALSE(framer_.GetEncodedVideoFrame(&frame, &next_frame)); |
| 125 | 127 |
| 126 // Complete key frame. | 128 // Complete key frame. |
| 127 rtp_header_.packet_id = 1; | 129 rtp_header_.packet_id = 1; |
| 128 duplicate = true; | 130 duplicate = true; |
| 129 complete = framer_.InsertPacket( | 131 complete = framer_.InsertPacket( |
| 130 payload_.data(), payload_.size(), rtp_header_, &duplicate); | 132 payload_.data(), payload_.size(), rtp_header_, &duplicate); |
| 131 EXPECT_TRUE(complete); | 133 EXPECT_TRUE(complete); |
| 132 EXPECT_FALSE(duplicate); | 134 EXPECT_FALSE(duplicate); |
| 133 EXPECT_TRUE(framer_.GetEncodedVideoFrame(&frame, &next_frame)); | 135 EXPECT_TRUE(framer_.GetEncodedVideoFrame(&frame, &next_frame)); |
| 136 EXPECT_EQ(transport::EncodedFrame::KEY, frame.dependency); |
| 134 EXPECT_EQ(0u, frame.frame_id); | 137 EXPECT_EQ(0u, frame.frame_id); |
| 138 EXPECT_EQ(0u, frame.referenced_frame_id); |
| 135 | 139 |
| 136 // Add same packet again in complete key frame. | 140 // Add same packet again in complete key frame. |
| 137 duplicate = false; | 141 duplicate = false; |
| 138 complete = framer_.InsertPacket( | 142 complete = framer_.InsertPacket( |
| 139 payload_.data(), payload_.size(), rtp_header_, &duplicate); | 143 payload_.data(), payload_.size(), rtp_header_, &duplicate); |
| 140 EXPECT_FALSE(complete); | 144 EXPECT_FALSE(complete); |
| 141 EXPECT_TRUE(duplicate); | 145 EXPECT_TRUE(duplicate); |
| 142 EXPECT_TRUE(framer_.GetEncodedVideoFrame(&frame, &next_frame)); | 146 EXPECT_TRUE(framer_.GetEncodedVideoFrame(&frame, &next_frame)); |
| 147 EXPECT_EQ(transport::EncodedFrame::KEY, frame.dependency); |
| 143 EXPECT_EQ(0u, frame.frame_id); | 148 EXPECT_EQ(0u, frame.frame_id); |
| 149 EXPECT_EQ(0u, frame.referenced_frame_id); |
| 144 framer_.ReleaseFrame(frame.frame_id); | 150 framer_.ReleaseFrame(frame.frame_id); |
| 145 | 151 |
| 146 // Incomplete delta frame. | 152 // Incomplete delta frame. |
| 147 ++rtp_header_.frame_id; | 153 ++rtp_header_.frame_id; |
| 148 rtp_header_.reference_frame_id = rtp_header_.frame_id - 1; | 154 rtp_header_.reference_frame_id = rtp_header_.frame_id - 1; |
| 149 rtp_header_.packet_id = 0; | 155 rtp_header_.packet_id = 0; |
| 150 rtp_header_.is_key_frame = false; | 156 rtp_header_.is_key_frame = false; |
| 151 duplicate = true; | 157 duplicate = true; |
| 152 complete = framer_.InsertPacket( | 158 complete = framer_.InsertPacket( |
| 153 payload_.data(), payload_.size(), rtp_header_, &duplicate); | 159 payload_.data(), payload_.size(), rtp_header_, &duplicate); |
| (...skipping 10 matching lines...) Expand all Loading... |
| 164 EXPECT_FALSE(framer_.GetEncodedVideoFrame(&frame, &next_frame)); | 170 EXPECT_FALSE(framer_.GetEncodedVideoFrame(&frame, &next_frame)); |
| 165 | 171 |
| 166 // Complete delta frame. | 172 // Complete delta frame. |
| 167 rtp_header_.packet_id = 1; | 173 rtp_header_.packet_id = 1; |
| 168 duplicate = true; | 174 duplicate = true; |
| 169 complete = framer_.InsertPacket( | 175 complete = framer_.InsertPacket( |
| 170 payload_.data(), payload_.size(), rtp_header_, &duplicate); | 176 payload_.data(), payload_.size(), rtp_header_, &duplicate); |
| 171 EXPECT_TRUE(complete); | 177 EXPECT_TRUE(complete); |
| 172 EXPECT_FALSE(duplicate); | 178 EXPECT_FALSE(duplicate); |
| 173 EXPECT_TRUE(framer_.GetEncodedVideoFrame(&frame, &next_frame)); | 179 EXPECT_TRUE(framer_.GetEncodedVideoFrame(&frame, &next_frame)); |
| 180 EXPECT_EQ(transport::EncodedFrame::DEPENDENT, frame.dependency); |
| 174 EXPECT_EQ(1u, frame.frame_id); | 181 EXPECT_EQ(1u, frame.frame_id); |
| 182 EXPECT_EQ(0u, frame.referenced_frame_id); |
| 175 | 183 |
| 176 // Add same packet again in complete delta frame. | 184 // Add same packet again in complete delta frame. |
| 177 duplicate = false; | 185 duplicate = false; |
| 178 complete = framer_.InsertPacket( | 186 complete = framer_.InsertPacket( |
| 179 payload_.data(), payload_.size(), rtp_header_, &duplicate); | 187 payload_.data(), payload_.size(), rtp_header_, &duplicate); |
| 180 EXPECT_FALSE(complete); | 188 EXPECT_FALSE(complete); |
| 181 EXPECT_TRUE(duplicate); | 189 EXPECT_TRUE(duplicate); |
| 182 EXPECT_TRUE(framer_.GetEncodedVideoFrame(&frame, &next_frame)); | 190 EXPECT_TRUE(framer_.GetEncodedVideoFrame(&frame, &next_frame)); |
| 191 EXPECT_EQ(transport::EncodedFrame::DEPENDENT, frame.dependency); |
| 183 EXPECT_EQ(1u, frame.frame_id); | 192 EXPECT_EQ(1u, frame.frame_id); |
| 193 EXPECT_EQ(0u, frame.referenced_frame_id); |
| 184 } | 194 } |
| 185 | 195 |
| 186 TEST_F(FramerTest, ContinuousSequence) { | 196 TEST_F(FramerTest, ContinuousSequence) { |
| 187 transport::EncodedVideoFrame frame; | 197 transport::EncodedFrame frame; |
| 188 bool next_frame = false; | 198 bool next_frame = false; |
| 189 bool complete = false; | 199 bool complete = false; |
| 190 bool duplicate = false; | 200 bool duplicate = false; |
| 191 | 201 |
| 192 // Start with a complete key frame. | 202 // Start with a complete key frame. |
| 193 rtp_header_.is_key_frame = true; | 203 rtp_header_.is_key_frame = true; |
| 194 complete = framer_.InsertPacket( | 204 complete = framer_.InsertPacket( |
| 195 payload_.data(), payload_.size(), rtp_header_, &duplicate); | 205 payload_.data(), payload_.size(), rtp_header_, &duplicate); |
| 196 EXPECT_TRUE(complete); | 206 EXPECT_TRUE(complete); |
| 197 EXPECT_TRUE(framer_.GetEncodedVideoFrame(&frame, &next_frame)); | 207 EXPECT_TRUE(framer_.GetEncodedVideoFrame(&frame, &next_frame)); |
| 198 EXPECT_TRUE(next_frame); | 208 EXPECT_TRUE(next_frame); |
| 209 EXPECT_EQ(transport::EncodedFrame::KEY, frame.dependency); |
| 199 EXPECT_EQ(0u, frame.frame_id); | 210 EXPECT_EQ(0u, frame.frame_id); |
| 200 EXPECT_TRUE(frame.key_frame); | 211 EXPECT_EQ(0u, frame.referenced_frame_id); |
| 201 framer_.ReleaseFrame(frame.frame_id); | 212 framer_.ReleaseFrame(frame.frame_id); |
| 202 | 213 |
| 203 // Complete - not continuous. | 214 // Complete - not continuous. |
| 204 rtp_header_.frame_id = 2; | 215 rtp_header_.frame_id = 2; |
| 205 rtp_header_.reference_frame_id = rtp_header_.frame_id - 1; | 216 rtp_header_.reference_frame_id = rtp_header_.frame_id - 1; |
| 206 rtp_header_.is_key_frame = false; | 217 rtp_header_.is_key_frame = false; |
| 207 complete = framer_.InsertPacket( | 218 complete = framer_.InsertPacket( |
| 208 payload_.data(), payload_.size(), rtp_header_, &duplicate); | 219 payload_.data(), payload_.size(), rtp_header_, &duplicate); |
| 209 EXPECT_TRUE(complete); | 220 EXPECT_TRUE(complete); |
| 210 EXPECT_FALSE(framer_.GetEncodedVideoFrame(&frame, &next_frame)); | 221 EXPECT_FALSE(framer_.GetEncodedVideoFrame(&frame, &next_frame)); |
| 211 } | 222 } |
| 212 | 223 |
| 213 TEST_F(FramerTest, Wrap) { | 224 TEST_F(FramerTest, Wrap) { |
| 214 // Insert key frame, frame_id = 255 (will jump to that) | 225 // Insert key frame, frame_id = 255 (will jump to that) |
| 215 transport::EncodedVideoFrame frame; | 226 transport::EncodedFrame frame; |
| 216 bool next_frame = false; | 227 bool next_frame = false; |
| 217 bool duplicate = false; | 228 bool duplicate = false; |
| 218 | 229 |
| 219 // Start with a complete key frame. | 230 // Start with a complete key frame. |
| 220 rtp_header_.is_key_frame = true; | 231 rtp_header_.is_key_frame = true; |
| 221 rtp_header_.frame_id = 255; | 232 rtp_header_.frame_id = 255; |
| 222 rtp_header_.reference_frame_id = 255; | 233 rtp_header_.reference_frame_id = 255; |
| 223 framer_.InsertPacket( | 234 framer_.InsertPacket( |
| 224 payload_.data(), payload_.size(), rtp_header_, &duplicate); | 235 payload_.data(), payload_.size(), rtp_header_, &duplicate); |
| 225 EXPECT_TRUE(framer_.GetEncodedVideoFrame(&frame, &next_frame)); | 236 EXPECT_TRUE(framer_.GetEncodedVideoFrame(&frame, &next_frame)); |
| 226 EXPECT_TRUE(next_frame); | 237 EXPECT_TRUE(next_frame); |
| 238 EXPECT_EQ(transport::EncodedFrame::KEY, frame.dependency); |
| 227 EXPECT_EQ(255u, frame.frame_id); | 239 EXPECT_EQ(255u, frame.frame_id); |
| 240 EXPECT_EQ(255u, frame.referenced_frame_id); |
| 228 framer_.ReleaseFrame(frame.frame_id); | 241 framer_.ReleaseFrame(frame.frame_id); |
| 229 | 242 |
| 230 // Insert wrapped delta frame - should be continuous. | 243 // Insert wrapped delta frame - should be continuous. |
| 231 rtp_header_.is_key_frame = false; | 244 rtp_header_.is_key_frame = false; |
| 232 rtp_header_.frame_id = 256; | 245 rtp_header_.frame_id = 256; |
| 233 framer_.InsertPacket( | 246 framer_.InsertPacket( |
| 234 payload_.data(), payload_.size(), rtp_header_, &duplicate); | 247 payload_.data(), payload_.size(), rtp_header_, &duplicate); |
| 235 EXPECT_TRUE(framer_.GetEncodedVideoFrame(&frame, &next_frame)); | 248 EXPECT_TRUE(framer_.GetEncodedVideoFrame(&frame, &next_frame)); |
| 236 EXPECT_TRUE(next_frame); | 249 EXPECT_TRUE(next_frame); |
| 250 EXPECT_EQ(transport::EncodedFrame::DEPENDENT, frame.dependency); |
| 237 EXPECT_EQ(256u, frame.frame_id); | 251 EXPECT_EQ(256u, frame.frame_id); |
| 252 EXPECT_EQ(255u, frame.referenced_frame_id); |
| 238 framer_.ReleaseFrame(frame.frame_id); | 253 framer_.ReleaseFrame(frame.frame_id); |
| 239 } | 254 } |
| 240 | 255 |
| 241 TEST_F(FramerTest, Reset) { | 256 TEST_F(FramerTest, Reset) { |
| 242 transport::EncodedVideoFrame frame; | 257 transport::EncodedFrame frame; |
| 243 bool next_frame = false; | 258 bool next_frame = false; |
| 244 bool complete = false; | 259 bool complete = false; |
| 245 bool duplicate = false; | 260 bool duplicate = false; |
| 246 | 261 |
| 247 // Start with a complete key frame. | 262 // Start with a complete key frame. |
| 248 rtp_header_.is_key_frame = true; | 263 rtp_header_.is_key_frame = true; |
| 249 complete = framer_.InsertPacket( | 264 complete = framer_.InsertPacket( |
| 250 payload_.data(), payload_.size(), rtp_header_, &duplicate); | 265 payload_.data(), payload_.size(), rtp_header_, &duplicate); |
| 251 EXPECT_TRUE(complete); | 266 EXPECT_TRUE(complete); |
| 252 framer_.Reset(); | 267 framer_.Reset(); |
| 253 EXPECT_FALSE(framer_.GetEncodedVideoFrame(&frame, &next_frame)); | 268 EXPECT_FALSE(framer_.GetEncodedVideoFrame(&frame, &next_frame)); |
| 254 } | 269 } |
| 255 | 270 |
| 256 TEST_F(FramerTest, RequireKeyAfterReset) { | 271 TEST_F(FramerTest, RequireKeyAfterReset) { |
| 257 transport::EncodedVideoFrame frame; | 272 transport::EncodedFrame frame; |
| 258 bool next_frame = false; | 273 bool next_frame = false; |
| 259 bool duplicate = false; | 274 bool duplicate = false; |
| 260 | 275 |
| 261 framer_.Reset(); | 276 framer_.Reset(); |
| 262 | 277 |
| 263 // Start with a complete key frame. | 278 // Start with a complete key frame. |
| 264 rtp_header_.is_key_frame = false; | 279 rtp_header_.is_key_frame = false; |
| 265 rtp_header_.frame_id = 0; | 280 rtp_header_.frame_id = 0; |
| 266 framer_.InsertPacket( | 281 framer_.InsertPacket( |
| 267 payload_.data(), payload_.size(), rtp_header_, &duplicate); | 282 payload_.data(), payload_.size(), rtp_header_, &duplicate); |
| 268 EXPECT_FALSE(framer_.GetEncodedVideoFrame(&frame, &next_frame)); | 283 EXPECT_FALSE(framer_.GetEncodedVideoFrame(&frame, &next_frame)); |
| 269 rtp_header_.frame_id = 1; | 284 rtp_header_.frame_id = 1; |
| 270 rtp_header_.reference_frame_id = 1; | 285 rtp_header_.reference_frame_id = 1; |
| 271 rtp_header_.is_key_frame = true; | 286 rtp_header_.is_key_frame = true; |
| 272 framer_.InsertPacket( | 287 framer_.InsertPacket( |
| 273 payload_.data(), payload_.size(), rtp_header_, &duplicate); | 288 payload_.data(), payload_.size(), rtp_header_, &duplicate); |
| 274 EXPECT_TRUE(framer_.GetEncodedVideoFrame(&frame, &next_frame)); | 289 EXPECT_TRUE(framer_.GetEncodedVideoFrame(&frame, &next_frame)); |
| 275 EXPECT_TRUE(next_frame); | 290 EXPECT_TRUE(next_frame); |
| 276 } | 291 } |
| 277 | 292 |
| 278 TEST_F(FramerTest, BasicNonLastReferenceId) { | 293 TEST_F(FramerTest, BasicNonLastReferenceId) { |
| 279 transport::EncodedVideoFrame frame; | 294 transport::EncodedFrame frame; |
| 280 bool next_frame = false; | 295 bool next_frame = false; |
| 281 bool duplicate = false; | 296 bool duplicate = false; |
| 282 | 297 |
| 283 rtp_header_.is_key_frame = true; | 298 rtp_header_.is_key_frame = true; |
| 284 rtp_header_.frame_id = 0; | 299 rtp_header_.frame_id = 0; |
| 285 framer_.InsertPacket( | 300 framer_.InsertPacket( |
| 286 payload_.data(), payload_.size(), rtp_header_, &duplicate); | 301 payload_.data(), payload_.size(), rtp_header_, &duplicate); |
| 287 | 302 |
| 288 EXPECT_TRUE(framer_.GetEncodedVideoFrame(&frame, &next_frame)); | 303 EXPECT_TRUE(framer_.GetEncodedVideoFrame(&frame, &next_frame)); |
| 289 framer_.ReleaseFrame(frame.frame_id); | 304 framer_.ReleaseFrame(frame.frame_id); |
| 290 | 305 |
| 291 rtp_header_.is_key_frame = false; | 306 rtp_header_.is_key_frame = false; |
| 292 rtp_header_.reference_frame_id = 0; | 307 rtp_header_.reference_frame_id = 0; |
| 293 rtp_header_.frame_id = 5; | 308 rtp_header_.frame_id = 5; |
| 294 framer_.InsertPacket( | 309 framer_.InsertPacket( |
| 295 payload_.data(), payload_.size(), rtp_header_, &duplicate); | 310 payload_.data(), payload_.size(), rtp_header_, &duplicate); |
| 296 | 311 |
| 297 EXPECT_TRUE(framer_.GetEncodedVideoFrame(&frame, &next_frame)); | 312 EXPECT_TRUE(framer_.GetEncodedVideoFrame(&frame, &next_frame)); |
| 298 EXPECT_FALSE(next_frame); | 313 EXPECT_FALSE(next_frame); |
| 299 } | 314 } |
| 300 | 315 |
| 301 TEST_F(FramerTest, InOrderReferenceFrameSelection) { | 316 TEST_F(FramerTest, InOrderReferenceFrameSelection) { |
| 302 // Create pattern: 0, 1, 4, 5. | 317 // Create pattern: 0, 1, 4, 5. |
| 303 transport::EncodedVideoFrame frame; | 318 transport::EncodedFrame frame; |
| 304 bool next_frame = false; | 319 bool next_frame = false; |
| 305 bool duplicate = false; | 320 bool duplicate = false; |
| 306 | 321 |
| 307 rtp_header_.is_key_frame = true; | 322 rtp_header_.is_key_frame = true; |
| 308 rtp_header_.frame_id = 0; | 323 rtp_header_.frame_id = 0; |
| 309 framer_.InsertPacket( | 324 framer_.InsertPacket( |
| 310 payload_.data(), payload_.size(), rtp_header_, &duplicate); | 325 payload_.data(), payload_.size(), rtp_header_, &duplicate); |
| 311 rtp_header_.is_key_frame = false; | 326 rtp_header_.is_key_frame = false; |
| 312 rtp_header_.frame_id = 1; | 327 rtp_header_.frame_id = 1; |
| 313 framer_.InsertPacket( | 328 framer_.InsertPacket( |
| 314 payload_.data(), payload_.size(), rtp_header_, &duplicate); | 329 payload_.data(), payload_.size(), rtp_header_, &duplicate); |
| 315 | 330 |
| 316 // Insert frame #2 partially. | 331 // Insert frame #2 partially. |
| 317 rtp_header_.frame_id = 2; | 332 rtp_header_.frame_id = 2; |
| 318 rtp_header_.max_packet_id = 1; | 333 rtp_header_.max_packet_id = 1; |
| 319 framer_.InsertPacket( | 334 framer_.InsertPacket( |
| 320 payload_.data(), payload_.size(), rtp_header_, &duplicate); | 335 payload_.data(), payload_.size(), rtp_header_, &duplicate); |
| 321 rtp_header_.frame_id = 4; | 336 rtp_header_.frame_id = 4; |
| 322 rtp_header_.max_packet_id = 0; | 337 rtp_header_.max_packet_id = 0; |
| 323 rtp_header_.reference_frame_id = 0; | 338 rtp_header_.reference_frame_id = 0; |
| 324 framer_.InsertPacket( | 339 framer_.InsertPacket( |
| 325 payload_.data(), payload_.size(), rtp_header_, &duplicate); | 340 payload_.data(), payload_.size(), rtp_header_, &duplicate); |
| 326 EXPECT_TRUE(framer_.GetEncodedVideoFrame(&frame, &next_frame)); | 341 EXPECT_TRUE(framer_.GetEncodedVideoFrame(&frame, &next_frame)); |
| 342 EXPECT_EQ(transport::EncodedFrame::KEY, frame.dependency); |
| 327 EXPECT_EQ(0u, frame.frame_id); | 343 EXPECT_EQ(0u, frame.frame_id); |
| 344 EXPECT_EQ(0u, frame.referenced_frame_id); |
| 328 framer_.ReleaseFrame(frame.frame_id); | 345 framer_.ReleaseFrame(frame.frame_id); |
| 329 EXPECT_TRUE(framer_.GetEncodedVideoFrame(&frame, &next_frame)); | 346 EXPECT_TRUE(framer_.GetEncodedVideoFrame(&frame, &next_frame)); |
| 330 EXPECT_TRUE(next_frame); | 347 EXPECT_TRUE(next_frame); |
| 348 EXPECT_EQ(transport::EncodedFrame::DEPENDENT, frame.dependency); |
| 331 EXPECT_EQ(1u, frame.frame_id); | 349 EXPECT_EQ(1u, frame.frame_id); |
| 350 EXPECT_EQ(0u, frame.referenced_frame_id); |
| 332 framer_.ReleaseFrame(frame.frame_id); | 351 framer_.ReleaseFrame(frame.frame_id); |
| 333 EXPECT_TRUE(framer_.GetEncodedVideoFrame(&frame, &next_frame)); | 352 EXPECT_TRUE(framer_.GetEncodedVideoFrame(&frame, &next_frame)); |
| 334 EXPECT_FALSE(next_frame); | 353 EXPECT_FALSE(next_frame); |
| 354 EXPECT_EQ(transport::EncodedFrame::DEPENDENT, frame.dependency); |
| 335 EXPECT_EQ(4u, frame.frame_id); | 355 EXPECT_EQ(4u, frame.frame_id); |
| 356 EXPECT_EQ(0u, frame.referenced_frame_id); |
| 336 framer_.ReleaseFrame(frame.frame_id); | 357 framer_.ReleaseFrame(frame.frame_id); |
| 337 // Insert remaining packet of frame #2 - should no be continuous. | 358 // Insert remaining packet of frame #2 - should no be continuous. |
| 338 rtp_header_.frame_id = 2; | 359 rtp_header_.frame_id = 2; |
| 339 rtp_header_.packet_id = 1; | 360 rtp_header_.packet_id = 1; |
| 340 framer_.InsertPacket( | 361 framer_.InsertPacket( |
| 341 payload_.data(), payload_.size(), rtp_header_, &duplicate); | 362 payload_.data(), payload_.size(), rtp_header_, &duplicate); |
| 342 EXPECT_FALSE(framer_.GetEncodedVideoFrame(&frame, &next_frame)); | 363 EXPECT_FALSE(framer_.GetEncodedVideoFrame(&frame, &next_frame)); |
| 343 rtp_header_.frame_id = 5; | 364 rtp_header_.frame_id = 5; |
| 344 rtp_header_.reference_frame_id = rtp_header_.frame_id - 1; | 365 rtp_header_.reference_frame_id = rtp_header_.frame_id - 1; |
| 345 rtp_header_.packet_id = 0; | 366 rtp_header_.packet_id = 0; |
| 346 rtp_header_.max_packet_id = 0; | 367 rtp_header_.max_packet_id = 0; |
| 347 framer_.InsertPacket( | 368 framer_.InsertPacket( |
| 348 payload_.data(), payload_.size(), rtp_header_, &duplicate); | 369 payload_.data(), payload_.size(), rtp_header_, &duplicate); |
| 349 EXPECT_TRUE(framer_.GetEncodedVideoFrame(&frame, &next_frame)); | 370 EXPECT_TRUE(framer_.GetEncodedVideoFrame(&frame, &next_frame)); |
| 350 EXPECT_TRUE(next_frame); | 371 EXPECT_TRUE(next_frame); |
| 372 EXPECT_EQ(transport::EncodedFrame::DEPENDENT, frame.dependency); |
| 351 EXPECT_EQ(5u, frame.frame_id); | 373 EXPECT_EQ(5u, frame.frame_id); |
| 374 EXPECT_EQ(4u, frame.referenced_frame_id); |
| 352 } | 375 } |
| 353 | 376 |
| 354 TEST_F(FramerTest, AudioWrap) { | 377 TEST_F(FramerTest, AudioWrap) { |
| 355 // All audio frames are marked as key frames. | 378 // All audio frames are marked as key frames. |
| 356 transport::EncodedAudioFrame frame; | 379 transport::EncodedFrame frame; |
| 357 bool next_frame = false; | 380 bool next_frame = false; |
| 358 bool duplicate = false; | 381 bool duplicate = false; |
| 359 | 382 |
| 360 rtp_header_.is_key_frame = true; | 383 rtp_header_.is_key_frame = true; |
| 361 rtp_header_.frame_id = 254; | 384 rtp_header_.frame_id = 254; |
| 362 rtp_header_.reference_frame_id = 254; | 385 rtp_header_.reference_frame_id = 254; |
| 363 | 386 |
| 364 framer_.InsertPacket( | 387 framer_.InsertPacket( |
| 365 payload_.data(), payload_.size(), rtp_header_, &duplicate); | 388 payload_.data(), payload_.size(), rtp_header_, &duplicate); |
| 366 EXPECT_TRUE(framer_.GetEncodedAudioFrame(&frame, &next_frame)); | 389 EXPECT_TRUE(framer_.GetEncodedAudioFrame(&frame, &next_frame)); |
| 367 EXPECT_TRUE(next_frame); | 390 EXPECT_TRUE(next_frame); |
| 391 EXPECT_EQ(transport::EncodedFrame::KEY, frame.dependency); |
| 368 EXPECT_EQ(254u, frame.frame_id); | 392 EXPECT_EQ(254u, frame.frame_id); |
| 393 EXPECT_EQ(254u, frame.referenced_frame_id); |
| 369 framer_.ReleaseFrame(frame.frame_id); | 394 framer_.ReleaseFrame(frame.frame_id); |
| 370 | 395 |
| 371 rtp_header_.frame_id = 255; | 396 rtp_header_.frame_id = 255; |
| 372 rtp_header_.reference_frame_id = 255; | 397 rtp_header_.reference_frame_id = 255; |
| 373 framer_.InsertPacket( | 398 framer_.InsertPacket( |
| 374 payload_.data(), payload_.size(), rtp_header_, &duplicate); | 399 payload_.data(), payload_.size(), rtp_header_, &duplicate); |
| 375 | 400 |
| 376 // Insert wrapped frame - should be continuous. | 401 // Insert wrapped frame - should be continuous. |
| 377 rtp_header_.frame_id = 256; | 402 rtp_header_.frame_id = 256; |
| 378 rtp_header_.reference_frame_id = 256; | 403 rtp_header_.reference_frame_id = 256; |
| 379 framer_.InsertPacket( | 404 framer_.InsertPacket( |
| 380 payload_.data(), payload_.size(), rtp_header_, &duplicate); | 405 payload_.data(), payload_.size(), rtp_header_, &duplicate); |
| 381 | 406 |
| 382 EXPECT_TRUE(framer_.GetEncodedAudioFrame(&frame, &next_frame)); | 407 EXPECT_TRUE(framer_.GetEncodedAudioFrame(&frame, &next_frame)); |
| 383 EXPECT_TRUE(next_frame); | 408 EXPECT_TRUE(next_frame); |
| 409 EXPECT_EQ(transport::EncodedFrame::KEY, frame.dependency); |
| 384 EXPECT_EQ(255u, frame.frame_id); | 410 EXPECT_EQ(255u, frame.frame_id); |
| 411 EXPECT_EQ(255u, frame.referenced_frame_id); |
| 385 framer_.ReleaseFrame(frame.frame_id); | 412 framer_.ReleaseFrame(frame.frame_id); |
| 386 | 413 |
| 387 EXPECT_TRUE(framer_.GetEncodedAudioFrame(&frame, &next_frame)); | 414 EXPECT_TRUE(framer_.GetEncodedAudioFrame(&frame, &next_frame)); |
| 388 EXPECT_TRUE(next_frame); | 415 EXPECT_TRUE(next_frame); |
| 416 EXPECT_EQ(transport::EncodedFrame::KEY, frame.dependency); |
| 389 EXPECT_EQ(256u, frame.frame_id); | 417 EXPECT_EQ(256u, frame.frame_id); |
| 418 EXPECT_EQ(256u, frame.referenced_frame_id); |
| 390 framer_.ReleaseFrame(frame.frame_id); | 419 framer_.ReleaseFrame(frame.frame_id); |
| 391 } | 420 } |
| 392 | 421 |
| 393 TEST_F(FramerTest, AudioWrapWithMissingFrame) { | 422 TEST_F(FramerTest, AudioWrapWithMissingFrame) { |
| 394 // All audio frames are marked as key frames. | 423 // All audio frames are marked as key frames. |
| 395 transport::EncodedAudioFrame frame; | 424 transport::EncodedFrame frame; |
| 396 bool next_frame = false; | 425 bool next_frame = false; |
| 397 bool duplicate = false; | 426 bool duplicate = false; |
| 398 | 427 |
| 399 // Insert and get first packet. | 428 // Insert and get first packet. |
| 400 rtp_header_.is_key_frame = true; | 429 rtp_header_.is_key_frame = true; |
| 401 rtp_header_.frame_id = 253; | 430 rtp_header_.frame_id = 253; |
| 402 rtp_header_.reference_frame_id = 253; | 431 rtp_header_.reference_frame_id = 253; |
| 403 framer_.InsertPacket( | 432 framer_.InsertPacket( |
| 404 payload_.data(), payload_.size(), rtp_header_, &duplicate); | 433 payload_.data(), payload_.size(), rtp_header_, &duplicate); |
| 405 EXPECT_TRUE(framer_.GetEncodedAudioFrame(&frame, &next_frame)); | 434 EXPECT_TRUE(framer_.GetEncodedAudioFrame(&frame, &next_frame)); |
| 406 EXPECT_TRUE(next_frame); | 435 EXPECT_TRUE(next_frame); |
| 436 EXPECT_EQ(transport::EncodedFrame::KEY, frame.dependency); |
| 407 EXPECT_EQ(253u, frame.frame_id); | 437 EXPECT_EQ(253u, frame.frame_id); |
| 438 EXPECT_EQ(253u, frame.referenced_frame_id); |
| 408 framer_.ReleaseFrame(frame.frame_id); | 439 framer_.ReleaseFrame(frame.frame_id); |
| 409 | 440 |
| 410 // Insert third and fourth packets. | 441 // Insert third and fourth packets. |
| 411 rtp_header_.frame_id = 255; | 442 rtp_header_.frame_id = 255; |
| 412 rtp_header_.reference_frame_id = 255; | 443 rtp_header_.reference_frame_id = 255; |
| 413 framer_.InsertPacket( | 444 framer_.InsertPacket( |
| 414 payload_.data(), payload_.size(), rtp_header_, &duplicate); | 445 payload_.data(), payload_.size(), rtp_header_, &duplicate); |
| 415 rtp_header_.frame_id = 256; | 446 rtp_header_.frame_id = 256; |
| 416 rtp_header_.reference_frame_id = 256; | 447 rtp_header_.reference_frame_id = 256; |
| 417 framer_.InsertPacket( | 448 framer_.InsertPacket( |
| 418 payload_.data(), payload_.size(), rtp_header_, &duplicate); | 449 payload_.data(), payload_.size(), rtp_header_, &duplicate); |
| 419 | 450 |
| 420 // Get third and fourth packets. | 451 // Get third and fourth packets. |
| 421 EXPECT_TRUE(framer_.GetEncodedAudioFrame(&frame, &next_frame)); | 452 EXPECT_TRUE(framer_.GetEncodedAudioFrame(&frame, &next_frame)); |
| 422 EXPECT_FALSE(next_frame); | 453 EXPECT_FALSE(next_frame); |
| 454 EXPECT_EQ(transport::EncodedFrame::KEY, frame.dependency); |
| 423 EXPECT_EQ(255u, frame.frame_id); | 455 EXPECT_EQ(255u, frame.frame_id); |
| 456 EXPECT_EQ(255u, frame.referenced_frame_id); |
| 424 framer_.ReleaseFrame(frame.frame_id); | 457 framer_.ReleaseFrame(frame.frame_id); |
| 425 EXPECT_TRUE(framer_.GetEncodedAudioFrame(&frame, &next_frame)); | 458 EXPECT_TRUE(framer_.GetEncodedAudioFrame(&frame, &next_frame)); |
| 426 EXPECT_TRUE(next_frame); | 459 EXPECT_TRUE(next_frame); |
| 460 EXPECT_EQ(transport::EncodedFrame::KEY, frame.dependency); |
| 427 EXPECT_EQ(256u, frame.frame_id); | 461 EXPECT_EQ(256u, frame.frame_id); |
| 462 EXPECT_EQ(256u, frame.referenced_frame_id); |
| 428 framer_.ReleaseFrame(frame.frame_id); | 463 framer_.ReleaseFrame(frame.frame_id); |
| 429 } | 464 } |
| 430 | 465 |
| 431 } // namespace cast | 466 } // namespace cast |
| 432 } // namespace media | 467 } // namespace media |
| OLD | NEW |