OLD | NEW |
1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 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/files/memory_mapped_file.h" | 5 #include "base/files/memory_mapped_file.h" |
6 #include "base/logging.h" | 6 #include "base/logging.h" |
7 #include "media/base/test_data_util.h" | 7 #include "media/base/test_data_util.h" |
8 #include "media/filters/ivf_parser.h" | 8 #include "media/filters/ivf_parser.h" |
9 #include "media/filters/vp9_parser.h" | 9 #include "media/filters/vp9_parser.h" |
10 #include "testing/gtest/include/gtest/gtest.h" | 10 #include "testing/gtest/include/gtest/gtest.h" |
(...skipping 10 matching lines...) Expand all Loading... |
21 << file_path.MaybeAsASCII(); | 21 << file_path.MaybeAsASCII(); |
22 | 22 |
23 IvfFileHeader ivf_file_header; | 23 IvfFileHeader ivf_file_header; |
24 ASSERT_TRUE(ivf_parser_.Initialize(stream_->data(), stream_->length(), | 24 ASSERT_TRUE(ivf_parser_.Initialize(stream_->data(), stream_->length(), |
25 &ivf_file_header)); | 25 &ivf_file_header)); |
26 ASSERT_EQ(ivf_file_header.fourcc, 0x30395056u); // VP90 | 26 ASSERT_EQ(ivf_file_header.fourcc, 0x30395056u); // VP90 |
27 } | 27 } |
28 | 28 |
29 void TearDown() override { stream_.reset(); } | 29 void TearDown() override { stream_.reset(); } |
30 | 30 |
31 bool ParseNextFrame(struct Vp9FrameHeader* frame_hdr); | |
32 | |
33 const Vp9Segmentation& GetSegmentation() const { | |
34 return vp9_parser_.GetSegmentation(); | |
35 } | |
36 | |
37 const Vp9LoopFilter& GetLoopFilter() const { | |
38 return vp9_parser_.GetLoopFilter(); | |
39 } | |
40 | |
41 IvfParser ivf_parser_; | 31 IvfParser ivf_parser_; |
42 scoped_ptr<base::MemoryMappedFile> stream_; | 32 scoped_ptr<base::MemoryMappedFile> stream_; |
43 | |
44 Vp9Parser vp9_parser_; | |
45 }; | 33 }; |
46 | 34 |
47 bool Vp9ParserTest::ParseNextFrame(Vp9FrameHeader* fhdr) { | |
48 while (1) { | |
49 Vp9Parser::Result res = vp9_parser_.ParseNextFrame(fhdr); | |
50 if (res == Vp9Parser::kEOStream) { | |
51 IvfFrameHeader ivf_frame_header; | |
52 const uint8_t* ivf_payload; | |
53 | |
54 if (!ivf_parser_.ParseNextFrame(&ivf_frame_header, &ivf_payload)) | |
55 return false; | |
56 | |
57 vp9_parser_.SetStream(ivf_payload, ivf_frame_header.frame_size); | |
58 continue; | |
59 } | |
60 | |
61 return res == Vp9Parser::kOk; | |
62 } | |
63 } | |
64 | |
65 TEST_F(Vp9ParserTest, StreamFileParsing) { | 35 TEST_F(Vp9ParserTest, StreamFileParsing) { |
66 // Number of frames in the test stream to be parsed. | 36 // Number of frames in the test stream to be parsed. |
67 const int num_frames = 250; | 37 const int num_frames = 250; |
| 38 |
| 39 Vp9Parser vp9_parser; |
| 40 IvfFrameHeader ivf_frame_header; |
68 int num_parsed_frames = 0; | 41 int num_parsed_frames = 0; |
| 42 const uint8_t* ivf_payload; |
69 | 43 |
70 while (num_parsed_frames < num_frames) { | 44 // Parse until the end of stream/unsupported stream/error in stream is found. |
| 45 while (ivf_parser_.ParseNextFrame(&ivf_frame_header, &ivf_payload)) { |
71 Vp9FrameHeader fhdr; | 46 Vp9FrameHeader fhdr; |
72 if (!ParseNextFrame(&fhdr)) | |
73 break; | |
74 | 47 |
| 48 ASSERT_TRUE( |
| 49 vp9_parser.ParseFrame(ivf_payload, ivf_frame_header.frame_size, &fhdr)); |
75 ++num_parsed_frames; | 50 ++num_parsed_frames; |
76 } | 51 } |
77 | 52 |
78 DVLOG(1) << "Number of successfully parsed frames before EOS: " | 53 DVLOG(1) << "Number of successfully parsed frames before EOS: " |
79 << num_parsed_frames; | 54 << num_parsed_frames; |
80 | 55 |
81 EXPECT_EQ(num_frames, num_parsed_frames); | 56 EXPECT_EQ(num_frames, num_parsed_frames); |
82 } | 57 } |
83 | 58 |
84 TEST_F(Vp9ParserTest, VerifyFirstFrame) { | 59 TEST_F(Vp9ParserTest, VerifyFirstFrame) { |
| 60 Vp9Parser vp9_parser; |
| 61 IvfFrameHeader ivf_frame_header; |
| 62 const uint8_t* ivf_payload; |
| 63 |
| 64 ASSERT_TRUE(ivf_parser_.ParseNextFrame(&ivf_frame_header, &ivf_payload)); |
85 Vp9FrameHeader fhdr; | 65 Vp9FrameHeader fhdr; |
86 | 66 |
87 ASSERT_TRUE(ParseNextFrame(&fhdr)); | 67 ASSERT_TRUE( |
| 68 vp9_parser.ParseFrame(ivf_payload, ivf_frame_header.frame_size, &fhdr)); |
88 | 69 |
89 EXPECT_EQ(0, fhdr.profile); | 70 EXPECT_EQ(0, fhdr.profile); |
90 EXPECT_FALSE(fhdr.show_existing_frame); | 71 EXPECT_FALSE(fhdr.show_existing_frame); |
91 EXPECT_EQ(Vp9FrameHeader::KEYFRAME, fhdr.frame_type); | 72 EXPECT_EQ(Vp9FrameHeader::KEYFRAME, fhdr.frame_type); |
92 EXPECT_TRUE(fhdr.show_frame); | 73 EXPECT_TRUE(fhdr.show_frame); |
93 EXPECT_FALSE(fhdr.error_resilient_mode); | 74 EXPECT_FALSE(fhdr.error_resilient_mode); |
94 | 75 |
95 EXPECT_EQ(8, fhdr.bit_depth); | 76 EXPECT_EQ(8, fhdr.bit_depth); |
96 EXPECT_EQ(Vp9ColorSpace::UNKNOWN, fhdr.color_space); | 77 EXPECT_EQ(Vp9ColorSpace::UNKNOWN, fhdr.color_space); |
97 EXPECT_FALSE(fhdr.yuv_range); | 78 EXPECT_FALSE(fhdr.yuv_range); |
98 EXPECT_EQ(1, fhdr.subsampling_x); | 79 EXPECT_EQ(1, fhdr.subsampling_x); |
99 EXPECT_EQ(1, fhdr.subsampling_y); | 80 EXPECT_EQ(1, fhdr.subsampling_y); |
100 | 81 |
101 EXPECT_EQ(320u, fhdr.width); | 82 EXPECT_EQ(320u, fhdr.width); |
102 EXPECT_EQ(240u, fhdr.height); | 83 EXPECT_EQ(240u, fhdr.height); |
103 EXPECT_EQ(320u, fhdr.display_width); | 84 EXPECT_EQ(320u, fhdr.display_width); |
104 EXPECT_EQ(240u, fhdr.display_height); | 85 EXPECT_EQ(240u, fhdr.display_height); |
105 | 86 |
106 EXPECT_TRUE(fhdr.refresh_frame_context); | 87 EXPECT_TRUE(fhdr.refresh_frame_context); |
107 EXPECT_TRUE(fhdr.frame_parallel_decoding_mode); | 88 EXPECT_TRUE(fhdr.frame_parallel_decoding_mode); |
108 EXPECT_EQ(0, fhdr.frame_context_idx); | 89 EXPECT_EQ(0, fhdr.frame_context_idx); |
109 | 90 |
110 const Vp9LoopFilter& lf = GetLoopFilter(); | 91 EXPECT_EQ(9, fhdr.loop_filter.filter_level); |
111 EXPECT_EQ(9, lf.filter_level); | 92 EXPECT_EQ(0, fhdr.loop_filter.sharpness_level); |
112 EXPECT_EQ(0, lf.sharpness_level); | 93 |
| 94 const Vp9LoopFilter& lf = fhdr.loop_filter; |
113 EXPECT_TRUE(lf.mode_ref_delta_enabled); | 95 EXPECT_TRUE(lf.mode_ref_delta_enabled); |
114 EXPECT_TRUE(lf.mode_ref_delta_update); | 96 EXPECT_TRUE(lf.mode_ref_delta_update); |
115 EXPECT_TRUE(lf.update_ref_deltas[0]); | 97 EXPECT_TRUE(lf.update_ref_deltas[0]); |
116 EXPECT_EQ(1, lf.ref_deltas[0]); | 98 EXPECT_EQ(1, lf.ref_deltas[0]); |
117 EXPECT_EQ(-1, lf.ref_deltas[2]); | 99 EXPECT_EQ(-1, lf.ref_deltas[2]); |
118 EXPECT_EQ(-1, lf.ref_deltas[3]); | 100 EXPECT_EQ(-1, lf.ref_deltas[3]); |
119 | 101 |
120 const Vp9QuantizationParams& qp = fhdr.quant_params; | 102 const Vp9QuantizationParams& qp = fhdr.quant_params; |
121 EXPECT_EQ(65, qp.base_qindex); | 103 EXPECT_EQ(65, qp.base_qindex); |
122 EXPECT_FALSE(qp.y_dc_delta); | 104 EXPECT_FALSE(qp.y_dc_delta); |
123 EXPECT_FALSE(qp.uv_dc_delta); | 105 EXPECT_FALSE(qp.uv_dc_delta); |
124 EXPECT_FALSE(qp.uv_ac_delta); | 106 EXPECT_FALSE(qp.uv_ac_delta); |
125 EXPECT_FALSE(qp.IsLossless()); | 107 EXPECT_FALSE(qp.IsLossless()); |
126 | 108 |
127 const Vp9Segmentation& seg = GetSegmentation(); | 109 const Vp9Segmentation& seg = fhdr.segment; |
128 EXPECT_FALSE(seg.enabled); | 110 EXPECT_FALSE(seg.enabled); |
129 | 111 |
130 EXPECT_EQ(0, fhdr.log2_tile_cols); | 112 EXPECT_EQ(0, fhdr.log2_tile_cols); |
131 EXPECT_EQ(0, fhdr.log2_tile_rows); | 113 EXPECT_EQ(0, fhdr.log2_tile_rows); |
132 | 114 |
133 EXPECT_EQ(120u, fhdr.first_partition_size); | 115 EXPECT_EQ(120u, fhdr.first_partition_size); |
134 EXPECT_EQ(18u, fhdr.uncompressed_header_size); | 116 EXPECT_EQ(18u, fhdr.uncompressed_header_size); |
135 } | 117 } |
136 | 118 |
137 TEST_F(Vp9ParserTest, VerifyInterFrame) { | 119 TEST_F(Vp9ParserTest, VerifyInterFrame) { |
| 120 Vp9Parser vp9_parser; |
138 Vp9FrameHeader fhdr; | 121 Vp9FrameHeader fhdr; |
139 | 122 |
140 // To verify the second frame. | 123 // To verify the second frame. |
141 for (int i = 0; i < 2; i++) | 124 for (int i = 0; i < 2; i++) { |
142 ASSERT_TRUE(ParseNextFrame(&fhdr)); | 125 IvfFrameHeader ivf_frame_header; |
| 126 const uint8_t* ivf_payload; |
| 127 ASSERT_TRUE(ivf_parser_.ParseNextFrame(&ivf_frame_header, &ivf_payload)); |
| 128 ASSERT_TRUE( |
| 129 vp9_parser.ParseFrame(ivf_payload, ivf_frame_header.frame_size, &fhdr)); |
| 130 } |
143 | 131 |
144 EXPECT_EQ(Vp9FrameHeader::INTERFRAME, fhdr.frame_type); | 132 EXPECT_EQ(Vp9FrameHeader::INTERFRAME, fhdr.frame_type); |
145 EXPECT_FALSE(fhdr.show_frame); | 133 EXPECT_FALSE(fhdr.show_frame); |
146 EXPECT_FALSE(fhdr.intra_only); | 134 EXPECT_FALSE(fhdr.intra_only); |
147 EXPECT_FALSE(fhdr.reset_context); | 135 EXPECT_FALSE(fhdr.reset_context); |
148 EXPECT_TRUE(fhdr.RefreshFlag(2)); | 136 EXPECT_TRUE(fhdr.refresh_flag[5]); |
149 EXPECT_EQ(0, fhdr.frame_refs[0]); | 137 EXPECT_EQ(0, fhdr.frame_refs[0]); |
150 EXPECT_EQ(1, fhdr.frame_refs[1]); | 138 EXPECT_EQ(1, fhdr.frame_refs[1]); |
151 EXPECT_EQ(2, fhdr.frame_refs[2]); | 139 EXPECT_EQ(2, fhdr.frame_refs[2]); |
152 EXPECT_TRUE(fhdr.allow_high_precision_mv); | 140 EXPECT_TRUE(fhdr.allow_high_precision_mv); |
153 EXPECT_EQ(Vp9InterpFilter::EIGHTTAP, fhdr.interp_filter); | 141 EXPECT_EQ(Vp9InterpFilter::EIGHTTAP, fhdr.interp_filter); |
154 | 142 |
155 EXPECT_EQ(48u, fhdr.first_partition_size); | 143 EXPECT_EQ(48u, fhdr.first_partition_size); |
156 EXPECT_EQ(11u, fhdr.uncompressed_header_size); | 144 EXPECT_EQ(11u, fhdr.uncompressed_header_size); |
157 } | 145 } |
158 | 146 |
159 } // namespace media | 147 } // namespace media |
OLD | NEW |