OLD | NEW |
---|---|
1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 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 <stddef.h> | 5 #include <stddef.h> |
6 #include <stdint.h> | 6 #include <stdint.h> |
7 | 7 |
8 #include <algorithm> | 8 #include <algorithm> |
9 #include <sstream> | 9 #include <sstream> |
10 | 10 |
11 #include "base/macros.h" | 11 #include "base/macros.h" |
12 #include "media/base/stream_parser.h" | 12 #include "media/base/stream_parser.h" |
13 #include "media/base/stream_parser_buffer.h" | 13 #include "media/base/stream_parser_buffer.h" |
14 #include "testing/gtest/include/gtest/gtest.h" | 14 #include "testing/gtest/include/gtest/gtest.h" |
15 | 15 |
16 namespace media { | 16 namespace media { |
17 | 17 |
18 typedef StreamParser::TrackId TrackId; | 18 typedef StreamParser::TrackId TrackId; |
19 typedef StreamParser::BufferQueue BufferQueue; | 19 typedef StreamParser::BufferQueue BufferQueue; |
20 typedef StreamParser::TextBufferQueueMap TextBufferQueueMap; | |
21 | 20 |
22 const int kEnd = -1; | 21 const int kEnd = -1; |
23 const uint8_t kFakeData[] = {0xFF}; | 22 const uint8_t kFakeData[] = {0xFF}; |
24 const TrackId kAudioTrackId = 0; | 23 const TrackId kAudioTrackId = 0; |
25 const TrackId kVideoTrackId = 1; | 24 const TrackId kVideoTrackId = 1; |
26 const TrackId kTextTrackIdA = 2; | 25 const TrackId kTextTrackIdA = 2; |
27 const TrackId kTextTrackIdB = 3; | 26 const TrackId kTextTrackIdB = 3; |
28 | 27 |
29 static bool IsAudio(scoped_refptr<StreamParserBuffer> buffer) { | 28 static bool IsAudio(scoped_refptr<StreamParserBuffer> buffer) { |
30 return buffer->type() == DemuxerStream::AUDIO; | 29 return buffer->type() == DemuxerStream::AUDIO; |
(...skipping 40 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
71 bool (*predicate)(scoped_refptr<StreamParserBuffer> buffer)) { | 70 bool (*predicate)(scoped_refptr<StreamParserBuffer> buffer)) { |
72 return static_cast<size_t>(count_if(merged_buffers_.begin(), | 71 return static_cast<size_t>(count_if(merged_buffers_.begin(), |
73 merged_buffers_.end(), | 72 merged_buffers_.end(), |
74 predicate)); | 73 predicate)); |
75 } | 74 } |
76 | 75 |
77 // Appends test audio buffers in the sequence described by |decode_timestamps| | 76 // Appends test audio buffers in the sequence described by |decode_timestamps| |
78 // to |audio_buffers_|. See GenerateBuffers() for |decode_timestamps| format. | 77 // to |audio_buffers_|. See GenerateBuffers() for |decode_timestamps| format. |
79 void GenerateAudioBuffers(const int* decode_timestamps) { | 78 void GenerateAudioBuffers(const int* decode_timestamps) { |
80 GenerateBuffers(decode_timestamps, DemuxerStream::AUDIO, kAudioTrackId, | 79 GenerateBuffers(decode_timestamps, DemuxerStream::AUDIO, kAudioTrackId, |
81 &audio_buffers_); | 80 &buffer_queue_map_[kAudioTrackId]); |
82 } | 81 } |
83 | 82 |
84 // Appends test video buffers in the sequence described by |decode_timestamps| | 83 // Appends test video buffers in the sequence described by |decode_timestamps| |
85 // to |video_buffers_|. See GenerateBuffers() for |decode_timestamps| format. | 84 // to |video_buffers_|. See GenerateBuffers() for |decode_timestamps| format. |
86 void GenerateVideoBuffers(const int* decode_timestamps) { | 85 void GenerateVideoBuffers(const int* decode_timestamps) { |
87 GenerateBuffers(decode_timestamps, DemuxerStream::VIDEO, kVideoTrackId, | 86 GenerateBuffers(decode_timestamps, DemuxerStream::VIDEO, kVideoTrackId, |
88 &video_buffers_); | 87 &buffer_queue_map_[kVideoTrackId]); |
89 } | 88 } |
90 | 89 |
91 // Current tests only need up to two distinct text BufferQueues. This helper | 90 // Current tests only need up to two distinct text BufferQueues. This helper |
92 // conditionally appends buffers to the underlying |text_buffers_a_| and | 91 // conditionally appends buffers to the underlying |buffer_queue_map_| keyed |
93 // |text_buffers_b_| and conditionally inserts these BufferQueues into | 92 // by the respective track ID. If |decode_timestamps_{a,b}| |
94 // |text_map_| keyed by the respective track ID. If |decode_timestamps_{a,b}| | 93 // is NULL, then the corresponding BufferQueue is not changed at all. |
95 // is NULL, then the corresponding BufferQueue is neither appended to nor | |
96 // inserted into |text_map_| (though it may previously have been inserted). | |
97 // Note that key collision on map insertion does not replace the previous | 94 // Note that key collision on map insertion does not replace the previous |
98 // value. | 95 // value. |
99 void GenerateTextBuffers(const int* decode_timestamps_a, | 96 void GenerateTextBuffers(const int* decode_timestamps_a, |
100 const int* decode_timestamps_b) { | 97 const int* decode_timestamps_b) { |
101 if (decode_timestamps_a) { | 98 if (decode_timestamps_a) { |
102 GenerateBuffers(decode_timestamps_a, DemuxerStream::TEXT, kTextTrackIdA, | 99 GenerateBuffers(decode_timestamps_a, DemuxerStream::TEXT, kTextTrackIdA, |
103 &text_buffers_a_); | 100 &buffer_queue_map_[kTextTrackIdA]); |
104 text_map_.insert(std::make_pair(kTextTrackIdA, text_buffers_a_)); | |
105 } | 101 } |
106 | 102 |
107 if (decode_timestamps_b) { | 103 if (decode_timestamps_b) { |
108 GenerateBuffers(decode_timestamps_b, DemuxerStream::TEXT, kTextTrackIdB, | 104 GenerateBuffers(decode_timestamps_b, DemuxerStream::TEXT, kTextTrackIdB, |
109 &text_buffers_b_); | 105 &buffer_queue_map_[kTextTrackIdB]); |
110 text_map_.insert(std::make_pair(kTextTrackIdB, text_buffers_b_)); | |
111 } | 106 } |
112 } | 107 } |
113 | 108 |
114 // Returns a string that describes the sequence of buffers in | 109 // Returns a string that describes the sequence of buffers in |
115 // |merged_buffers_|. The string is a concatenation of space-delimited buffer | 110 // |merged_buffers_|. The string is a concatenation of space-delimited buffer |
116 // descriptors in the same sequence as |merged_buffers_|. Each descriptor is | 111 // descriptors in the same sequence as |merged_buffers_|. Each descriptor is |
117 // the concatenation of | 112 // the concatenation of |
118 // 1) a single character that describes the buffer's type(), e.g. A, V, or T | 113 // 1) a single character that describes the buffer's type(), e.g. A, V, or T |
119 // for audio, video, or text, respectively | 114 // for audio, video, or text, respectively |
120 // 2) the buffer's track_id() | 115 // 2) the buffer's track_id() |
(...skipping 48 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
169 // also verified to match the number of buffers in |audio_buffers_|, | 164 // also verified to match the number of buffers in |audio_buffers_|, |
170 // |video_buffers_|, and |text_map_|, respectively. | 165 // |video_buffers_|, and |text_map_|, respectively. |
171 void VerifyMergeSuccess(const std::string& expected, | 166 void VerifyMergeSuccess(const std::string& expected, |
172 bool verify_type_and_text_track_sequence) { | 167 bool verify_type_and_text_track_sequence) { |
173 // |merged_buffers| may already have some buffers. Count them by type for | 168 // |merged_buffers| may already have some buffers. Count them by type for |
174 // later inclusion in verification. | 169 // later inclusion in verification. |
175 size_t original_audio_in_merged = CountMatchingMergedBuffers(IsAudio); | 170 size_t original_audio_in_merged = CountMatchingMergedBuffers(IsAudio); |
176 size_t original_video_in_merged = CountMatchingMergedBuffers(IsVideo); | 171 size_t original_video_in_merged = CountMatchingMergedBuffers(IsVideo); |
177 size_t original_text_in_merged = CountMatchingMergedBuffers(IsText); | 172 size_t original_text_in_merged = CountMatchingMergedBuffers(IsText); |
178 | 173 |
179 EXPECT_TRUE(MergeBufferQueues(audio_buffers_, video_buffers_, text_map_, | 174 StreamParser::BufferQueueMap buffer_queue_map; |
180 &merged_buffers_)); | 175 for (const auto& it : buffer_queue_map_) |
DaleCurtis
2016/08/19 01:06:25
Needs {} -- again it seems like there should be an
servolk
2016/08/19 01:29:29
Sure I can add {}.
According to https://chromium-c
| |
176 if (!it.second.empty()) | |
177 buffer_queue_map.insert(std::make_pair(it.first, it.second)); | |
178 EXPECT_TRUE(MergeBufferQueues(buffer_queue_map, &merged_buffers_)); | |
181 | 179 |
182 // Verify resulting contents of |merged_buffers| matches |expected|. | 180 // Verify resulting contents of |merged_buffers| matches |expected|. |
183 EXPECT_EQ(expected, | 181 EXPECT_EQ(expected, |
184 MergedBufferQueueString(verify_type_and_text_track_sequence)); | 182 MergedBufferQueueString(verify_type_and_text_track_sequence)); |
185 | 183 |
186 // Verify that the correct number of each type of buffer is in the merge | 184 // Verify that the correct number of each type of buffer is in the merge |
187 // result. | 185 // result. |
188 size_t audio_in_merged = CountMatchingMergedBuffers(IsAudio); | 186 size_t audio_in_merged = CountMatchingMergedBuffers(IsAudio); |
189 size_t video_in_merged = CountMatchingMergedBuffers(IsVideo); | 187 size_t video_in_merged = CountMatchingMergedBuffers(IsVideo); |
190 size_t text_in_merged = CountMatchingMergedBuffers(IsText); | 188 size_t text_in_merged = CountMatchingMergedBuffers(IsText); |
191 | 189 |
192 EXPECT_GE(audio_in_merged, original_audio_in_merged); | 190 EXPECT_GE(audio_in_merged, original_audio_in_merged); |
193 EXPECT_GE(video_in_merged, original_video_in_merged); | 191 EXPECT_GE(video_in_merged, original_video_in_merged); |
194 EXPECT_GE(text_in_merged, original_text_in_merged); | 192 EXPECT_GE(text_in_merged, original_text_in_merged); |
195 | 193 |
196 EXPECT_EQ(audio_buffers_.size(), | 194 EXPECT_EQ(buffer_queue_map_[kAudioTrackId].size(), |
197 audio_in_merged - original_audio_in_merged); | 195 audio_in_merged - original_audio_in_merged); |
198 EXPECT_EQ(video_buffers_.size(), | 196 EXPECT_EQ(buffer_queue_map_[kVideoTrackId].size(), |
199 video_in_merged - original_video_in_merged); | 197 video_in_merged - original_video_in_merged); |
200 | 198 |
201 size_t expected_text_buffer_count = 0; | 199 size_t expected_text_buffer_count = 0; |
202 for (TextBufferQueueMap::const_iterator itr = text_map_.begin(); | 200 expected_text_buffer_count += buffer_queue_map_[kTextTrackIdA].size(); |
203 itr != text_map_.end(); | 201 expected_text_buffer_count += buffer_queue_map_[kTextTrackIdB].size(); |
204 ++itr) { | |
205 expected_text_buffer_count += itr->second.size(); | |
206 } | |
207 EXPECT_EQ(expected_text_buffer_count, | 202 EXPECT_EQ(expected_text_buffer_count, |
208 text_in_merged - original_text_in_merged); | 203 text_in_merged - original_text_in_merged); |
209 } | 204 } |
210 | 205 |
211 // Verifies that MergeBufferQueues() of the current |audio_buffers_|, | 206 // Verifies that MergeBufferQueues() of the current |audio_buffers_|, |
212 // |video_buffers_|, |text_map_|, and |merged_buffers_| returns false. | 207 // |video_buffers_|, |text_map_|, and |merged_buffers_| returns false. |
213 void VerifyMergeFailure() { | 208 void VerifyMergeFailure() { |
214 EXPECT_FALSE(MergeBufferQueues(audio_buffers_, video_buffers_, text_map_, | 209 StreamParser::BufferQueueMap buffer_queue_map; |
215 &merged_buffers_)); | 210 for (const auto& it : buffer_queue_map_) |
DaleCurtis
2016/08/19 01:06:25
needs {}, ditto.
servolk
2016/08/19 01:29:29
Acknowledged.
| |
211 if (!it.second.empty()) | |
212 buffer_queue_map.insert(std::make_pair(it.first, it.second)); | |
213 EXPECT_FALSE(MergeBufferQueues(buffer_queue_map, &merged_buffers_)); | |
216 } | 214 } |
217 | 215 |
218 // Helper to allow tests to clear all the input BufferQueues (except | 216 // Helper to allow tests to clear all the input BufferQueues (except |
219 // |merged_buffers_|) and the TextBufferQueueMap that are used in | 217 // |merged_buffers_|) and the BufferQueueMap that are used in |
220 // VerifyMerge{Success/Failure}(). | 218 // VerifyMerge{Success/Failure}(). |
221 void ClearQueuesAndTextMapButKeepAnyMergedBuffers() { | 219 void ClearQueuesAndTextMapButKeepAnyMergedBuffers() { |
222 audio_buffers_.clear(); | 220 buffer_queue_map_.clear(); |
223 video_buffers_.clear(); | |
224 text_buffers_a_.clear(); | |
225 text_buffers_b_.clear(); | |
226 text_map_.clear(); | |
227 } | 221 } |
228 | 222 |
229 private: | 223 private: |
230 BufferQueue audio_buffers_; | 224 std::map<StreamParser::TrackId, StreamParser::BufferQueue> buffer_queue_map_; |
231 BufferQueue video_buffers_; | |
232 BufferQueue text_buffers_a_; | |
233 BufferQueue text_buffers_b_; | |
234 BufferQueue merged_buffers_; | 225 BufferQueue merged_buffers_; |
235 TextBufferQueueMap text_map_; | |
236 | 226 |
237 DISALLOW_COPY_AND_ASSIGN(StreamParserTest); | 227 DISALLOW_COPY_AND_ASSIGN(StreamParserTest); |
238 }; | 228 }; |
239 | 229 |
240 TEST_F(StreamParserTest, MergeBufferQueues_AllEmpty) { | 230 TEST_F(StreamParserTest, MergeBufferQueues_AllEmpty) { |
241 std::string expected = ""; | 231 std::string expected = ""; |
242 VerifyMergeSuccess(expected, true); | 232 VerifyMergeSuccess(expected, true); |
243 } | 233 } |
244 | 234 |
245 TEST_F(StreamParserTest, MergeBufferQueues_SingleAudioBuffer) { | 235 TEST_F(StreamParserTest, MergeBufferQueues_SingleAudioBuffer) { |
(...skipping 129 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
375 VerifyMergeSuccess(expected, true); | 365 VerifyMergeSuccess(expected, true); |
376 | 366 |
377 // But appending something with a lower timestamp than the last timestamp | 367 // But appending something with a lower timestamp than the last timestamp |
378 // in the pre-existing merge result should fail. | 368 // in the pre-existing merge result should fail. |
379 int more_audio_timestamps[] = { 106, kEnd }; | 369 int more_audio_timestamps[] = { 106, kEnd }; |
380 GenerateAudioBuffers(more_audio_timestamps); | 370 GenerateAudioBuffers(more_audio_timestamps); |
381 VerifyMergeFailure(); | 371 VerifyMergeFailure(); |
382 } | 372 } |
383 | 373 |
384 } // namespace media | 374 } // namespace media |
OLD | NEW |