Chromium Code Reviews| Index: webrtc/modules/video_coding/main/source/decoding_state_unittest.cc |
| diff --git a/webrtc/modules/video_coding/main/source/decoding_state_unittest.cc b/webrtc/modules/video_coding/main/source/decoding_state_unittest.cc |
| index 10f1d6e4dd7039de1a3752e7a986d13e07e56e61..49d1482f8e4289d2374d1cf17eb9243a9474185d 100644 |
| --- a/webrtc/modules/video_coding/main/source/decoding_state_unittest.cc |
| +++ b/webrtc/modules/video_coding/main/source/decoding_state_unittest.cc |
| @@ -446,4 +446,138 @@ TEST(TestDecodingState, PictureIdRepeat) { |
| EXPECT_FALSE(dec_state.ContinuousFrame(&frame)); |
| } |
| +TEST(TestDecodingState, FrameContinuityFlexibleMode) { |
| + VCMDecodingState dec_state; |
| + VCMFrameBuffer frame; |
| + VCMPacket packet; |
| + packet.isFirstPacket = true; |
| + packet.timestamp = 1; |
| + packet.seqNum = 0xffff; |
| + uint8_t data[] = "I need a data pointer for this test!"; |
| + packet.sizeBytes = sizeof(data); |
| + packet.dataPtr = data; |
| + packet.codecSpecificHeader.codec = kRtpVideoVp9; |
| + |
| + RTPVideoHeaderVP9& Vp9Hdr = packet.codecSpecificHeader.codecHeader.VP9; |
| + Vp9Hdr.picture_id = 10; |
| + Vp9Hdr.flexible_mode = true; |
| + |
| + FrameData frame_data; |
| + frame_data.rtt_ms = 0; |
| + frame_data.rolling_average_packets_per_frame = -1; |
| + |
| + // Key frame as first frame |
| + packet.frameType = kVideoFrameKey; |
| + EXPECT_LE(0, frame.InsertPacket(packet, 0, kNoErrors, frame_data)); |
| + EXPECT_TRUE(dec_state.ContinuousFrame(&frame)); |
| + |
| + // Delta frame as first frame |
| + frame.Reset(); |
| + packet.frameType = kVideoFrameDelta; |
| + EXPECT_LE(0, frame.InsertPacket(packet, 0, kNoErrors, frame_data)); |
| + EXPECT_FALSE(dec_state.ContinuousFrame(&frame)); |
| + |
| + // Key frame then delta frame |
| + frame.Reset(); |
| + packet.frameType = kVideoFrameKey; |
| + EXPECT_LE(0, frame.InsertPacket(packet, 0, kNoErrors, frame_data)); |
| + dec_state.SetState(&frame); |
| + frame.Reset(); |
| + packet.frameType = kVideoFrameDelta; |
| + Vp9Hdr.num_ref_pics = 1; |
| + Vp9Hdr.picture_id = 15; |
| + Vp9Hdr.pid_diff[0] = 5; |
| + EXPECT_LE(0, frame.InsertPacket(packet, 0, kNoErrors, frame_data)); |
| + EXPECT_TRUE(dec_state.ContinuousFrame(&frame)); |
| + dec_state.SetState(&frame); |
| + |
| + // Ref to 11, not continuous |
| + frame.Reset(); |
| + Vp9Hdr.picture_id = 16; |
| + EXPECT_LE(0, frame.InsertPacket(packet, 0, kNoErrors, frame_data)); |
| + EXPECT_FALSE(dec_state.ContinuousFrame(&frame)); |
| + |
| + // Ref to 15, continuous |
| + frame.Reset(); |
| + Vp9Hdr.picture_id = 16; |
| + Vp9Hdr.pid_diff[0] = 1; |
| + EXPECT_LE(0, frame.InsertPacket(packet, 0, kNoErrors, frame_data)); |
| + EXPECT_TRUE(dec_state.ContinuousFrame(&frame)); |
| + dec_state.SetState(&frame); |
| + |
| + // Ref to 11 and 15, not continuous |
| + frame.Reset(); |
| + Vp9Hdr.picture_id = 20; |
| + Vp9Hdr.num_ref_pics = 2; |
| + Vp9Hdr.pid_diff[0] = 9; |
| + Vp9Hdr.pid_diff[1] = 5; |
| + EXPECT_LE(0, frame.InsertPacket(packet, 0, kNoErrors, frame_data)); |
| + EXPECT_FALSE(dec_state.ContinuousFrame(&frame)); |
| + |
| + // Ref to 10, 15 and 16, continuous |
| + frame.Reset(); |
| + Vp9Hdr.picture_id = 22; |
| + Vp9Hdr.num_ref_pics = 3; |
| + Vp9Hdr.pid_diff[0] = 12; |
| + Vp9Hdr.pid_diff[1] = 7; |
| + Vp9Hdr.pid_diff[2] = 6; |
| + EXPECT_LE(0, frame.InsertPacket(packet, 0, kNoErrors, frame_data)); |
| + EXPECT_TRUE(dec_state.ContinuousFrame(&frame)); |
| + dec_state.SetState(&frame); |
| + |
| + // Frame at last index, ref to 10, continuous |
| + frame.Reset(); |
| + Vp9Hdr.picture_id = VCMDecodingState::kFrameDecodedLength - 1; |
| + Vp9Hdr.num_ref_pics = 1; |
| + Vp9Hdr.pid_diff[0] = VCMDecodingState::kFrameDecodedLength - 11; |
| + EXPECT_LE(0, frame.InsertPacket(packet, 0, kNoErrors, frame_data)); |
| + EXPECT_TRUE(dec_state.ContinuousFrame(&frame)); |
| + dec_state.SetState(&frame); |
| + |
| + // Frame after wrapping buffer length, ref to 10 and last index, continuous |
| + frame.Reset(); |
| + Vp9Hdr.picture_id = 0; |
| + Vp9Hdr.num_ref_pics = 2; |
| + Vp9Hdr.pid_diff[0] = 1; |
| + Vp9Hdr.pid_diff[1] = VCMDecodingState::kFrameDecodedLength - 10; |
| + EXPECT_LE(0, frame.InsertPacket(packet, 0, kNoErrors, frame_data)); |
| + EXPECT_TRUE(dec_state.ContinuousFrame(&frame)); |
| + dec_state.SetState(&frame); |
| + |
| + // Frame after wrapping start frame, ref to 0, continuous |
| + frame.Reset(); |
| + Vp9Hdr.picture_id = 20; |
| + Vp9Hdr.num_ref_pics = 1; |
| + Vp9Hdr.pid_diff[0] = 20; |
| + EXPECT_LE(0, frame.InsertPacket(packet, 0, kNoErrors, frame_data)); |
| + EXPECT_TRUE(dec_state.ContinuousFrame(&frame)); |
| + dec_state.SetState(&frame); |
| + |
| + // Frame after wrapping start frame, ref to 10, not continuous |
| + frame.Reset(); |
| + Vp9Hdr.picture_id = 23; |
| + Vp9Hdr.num_ref_pics = 1; |
| + Vp9Hdr.pid_diff[0] = 13; |
| + EXPECT_LE(0, frame.InsertPacket(packet, 0, kNoErrors, frame_data)); |
| + EXPECT_FALSE(dec_state.ContinuousFrame(&frame)); |
| + |
| + // Key frame, continuous |
| + frame.Reset(); |
| + packet.frameType = kVideoFrameKey; |
| + Vp9Hdr.picture_id = 25; |
| + Vp9Hdr.num_ref_pics = 0; |
| + EXPECT_LE(0, frame.InsertPacket(packet, 0, kNoErrors, frame_data)); |
| + EXPECT_TRUE(dec_state.ContinuousFrame(&frame)); |
| + dec_state.SetState(&frame); |
| + |
| + // Ref to frame previous to KF, not continuous |
| + frame.Reset(); |
| + packet.frameType = kVideoFrameDelta; |
| + Vp9Hdr.picture_id = 30; |
| + Vp9Hdr.num_ref_pics = 1; |
| + Vp9Hdr.pid_diff[0] = 30; |
| + EXPECT_LE(0, frame.InsertPacket(packet, 0, kNoErrors, frame_data)); |
| + EXPECT_FALSE(dec_state.ContinuousFrame(&frame)); |
| +} |
| + |
|
sprang_webrtc
2015/09/15 15:41:22
Any chance we can break this out into several smal
|
| } // namespace webrtc |