OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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 "media/webm/webm_stream_parser.h" | 5 #include "media/webm/webm_stream_parser.h" |
6 | 6 |
7 #include <string> | 7 #include <string> |
8 | 8 |
9 #include "base/callback.h" | 9 #include "base/callback.h" |
10 #include "base/logging.h" | 10 #include "base/logging.h" |
11 #include "base/stl_util.h" | |
12 #include "media/webm/webm_cluster_parser.h" | 11 #include "media/webm/webm_cluster_parser.h" |
13 #include "media/webm/webm_constants.h" | 12 #include "media/webm/webm_constants.h" |
14 #include "media/webm/webm_content_encodings.h" | 13 #include "media/webm/webm_content_encodings.h" |
15 #include "media/webm/webm_crypto_helpers.h" | 14 #include "media/webm/webm_crypto_helpers.h" |
16 #include "media/webm/webm_info_parser.h" | 15 #include "media/webm/webm_info_parser.h" |
17 #include "media/webm/webm_tracks_parser.h" | 16 #include "media/webm/webm_tracks_parser.h" |
18 | 17 |
19 namespace media { | 18 namespace media { |
20 | 19 |
21 WebMStreamParser::WebMStreamParser() | 20 WebMStreamParser::WebMStreamParser() |
22 : state_(kWaitingForInit), | 21 : state_(kWaitingForInit), |
23 waiting_for_buffers_(false) { | 22 waiting_for_buffers_(false) { |
24 } | 23 } |
25 | 24 |
26 WebMStreamParser::~WebMStreamParser() { | 25 WebMStreamParser::~WebMStreamParser() { |
27 STLDeleteValues(&text_track_map_); | |
28 } | 26 } |
29 | 27 |
30 void WebMStreamParser::Init(const InitCB& init_cb, | 28 void WebMStreamParser::Init(const InitCB& init_cb, |
31 const NewConfigCB& config_cb, | 29 const NewConfigCB& config_cb, |
32 const NewBuffersCB& new_buffers_cb, | 30 const NewBuffersCB& new_buffers_cb, |
33 const NewTextBuffersCB& text_cb, | 31 const NewTextBuffersCB& text_cb, |
34 const NeedKeyCB& need_key_cb, | 32 const NeedKeyCB& need_key_cb, |
35 const AddTextTrackCB& add_text_track_cb, | 33 const NewTextTrackCB& new_text_track_cb, |
36 const NewMediaSegmentCB& new_segment_cb, | 34 const NewMediaSegmentCB& new_segment_cb, |
37 const base::Closure& end_of_segment_cb, | 35 const base::Closure& end_of_segment_cb, |
38 const LogCB& log_cb) { | 36 const LogCB& log_cb) { |
39 DCHECK_EQ(state_, kWaitingForInit); | 37 DCHECK_EQ(state_, kWaitingForInit); |
40 DCHECK(init_cb_.is_null()); | 38 DCHECK(init_cb_.is_null()); |
41 DCHECK(!init_cb.is_null()); | 39 DCHECK(!init_cb.is_null()); |
42 DCHECK(!config_cb.is_null()); | 40 DCHECK(!config_cb.is_null()); |
43 DCHECK(!new_buffers_cb.is_null()); | 41 DCHECK(!new_buffers_cb.is_null()); |
44 DCHECK(!text_cb.is_null()); | 42 DCHECK(!text_cb.is_null()); |
45 DCHECK(!need_key_cb.is_null()); | 43 DCHECK(!need_key_cb.is_null()); |
46 DCHECK(!new_segment_cb.is_null()); | 44 DCHECK(!new_segment_cb.is_null()); |
47 DCHECK(!end_of_segment_cb.is_null()); | 45 DCHECK(!end_of_segment_cb.is_null()); |
48 | 46 |
49 ChangeState(kParsingHeaders); | 47 ChangeState(kParsingHeaders); |
50 init_cb_ = init_cb; | 48 init_cb_ = init_cb; |
51 config_cb_ = config_cb; | 49 config_cb_ = config_cb; |
52 new_buffers_cb_ = new_buffers_cb; | 50 new_buffers_cb_ = new_buffers_cb; |
53 text_cb_ = text_cb; | 51 text_cb_ = text_cb; |
54 need_key_cb_ = need_key_cb; | 52 need_key_cb_ = need_key_cb; |
55 add_text_track_cb_ = add_text_track_cb; | 53 new_text_track_cb_ = new_text_track_cb; |
56 new_segment_cb_ = new_segment_cb; | 54 new_segment_cb_ = new_segment_cb; |
57 end_of_segment_cb_ = end_of_segment_cb; | 55 end_of_segment_cb_ = end_of_segment_cb; |
58 log_cb_ = log_cb; | 56 log_cb_ = log_cb; |
59 } | 57 } |
60 | 58 |
61 void WebMStreamParser::Flush() { | 59 void WebMStreamParser::Flush() { |
62 DCHECK_NE(state_, kWaitingForInit); | 60 DCHECK_NE(state_, kWaitingForInit); |
63 | 61 |
64 byte_queue_.Reset(); | 62 byte_queue_.Reset(); |
65 | 63 |
(...skipping 102 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
168 WebMInfoParser info_parser; | 166 WebMInfoParser info_parser; |
169 result = info_parser.Parse(cur, cur_size); | 167 result = info_parser.Parse(cur, cur_size); |
170 | 168 |
171 if (result <= 0) | 169 if (result <= 0) |
172 return result; | 170 return result; |
173 | 171 |
174 cur += result; | 172 cur += result; |
175 cur_size -= result; | 173 cur_size -= result; |
176 bytes_parsed += result; | 174 bytes_parsed += result; |
177 | 175 |
178 WebMTracksParser tracks_parser(log_cb_, add_text_track_cb_.is_null()); | 176 WebMTracksParser tracks_parser(log_cb_, new_text_track_cb_.is_null()); |
179 result = tracks_parser.Parse(cur, cur_size); | 177 result = tracks_parser.Parse(cur, cur_size); |
180 | 178 |
181 if (result <= 0) | 179 if (result <= 0) |
182 return result; | 180 return result; |
183 | 181 |
184 bytes_parsed += result; | 182 bytes_parsed += result; |
185 | 183 |
186 base::TimeDelta duration = kInfiniteDuration(); | 184 base::TimeDelta duration = kInfiniteDuration(); |
187 | 185 |
188 if (info_parser.duration() > 0) { | 186 if (info_parser.duration() > 0) { |
(...skipping 16 matching lines...) Expand all Loading... |
205 } | 203 } |
206 | 204 |
207 typedef WebMTracksParser::TextTracks TextTracks; | 205 typedef WebMTracksParser::TextTracks TextTracks; |
208 const TextTracks& text_tracks = tracks_parser.text_tracks(); | 206 const TextTracks& text_tracks = tracks_parser.text_tracks(); |
209 | 207 |
210 for (TextTracks::const_iterator itr = text_tracks.begin(); | 208 for (TextTracks::const_iterator itr = text_tracks.begin(); |
211 itr != text_tracks.end(); ++itr) { | 209 itr != text_tracks.end(); ++itr) { |
212 const WebMTracksParser::TextTrackInfo& text_track_info = itr->second; | 210 const WebMTracksParser::TextTrackInfo& text_track_info = itr->second; |
213 | 211 |
214 // TODO(matthewjheaney): verify that WebVTT uses ISO 639-2 for lang | 212 // TODO(matthewjheaney): verify that WebVTT uses ISO 639-2 for lang |
215 scoped_ptr<TextTrack> text_track = | 213 new_text_track_cb_.Run(itr->first, |
216 add_text_track_cb_.Run(text_track_info.kind, | 214 text_track_info.kind, |
217 text_track_info.name, | 215 text_track_info.name, |
218 text_track_info.language); | 216 text_track_info.language); |
219 | |
220 // Assume ownership of pointer, and cache the text track object, for use | |
221 // later when we have text track buffers. (The text track objects are | |
222 // deallocated in the dtor for this class.) | |
223 | |
224 if (text_track) | |
225 text_track_map_.insert(std::make_pair(itr->first, text_track.release())); | |
226 } | 217 } |
227 | 218 |
228 cluster_parser_.reset(new WebMClusterParser( | 219 cluster_parser_.reset(new WebMClusterParser( |
229 info_parser.timecode_scale(), | 220 info_parser.timecode_scale(), |
230 tracks_parser.audio_track_num(), | 221 tracks_parser.audio_track_num(), |
231 tracks_parser.video_track_num(), | 222 tracks_parser.video_track_num(), |
232 text_tracks, | 223 text_tracks, |
233 tracks_parser.ignored_tracks(), | 224 tracks_parser.ignored_tracks(), |
234 tracks_parser.audio_encryption_key_id(), | 225 tracks_parser.audio_encryption_key_id(), |
235 tracks_parser.video_encryption_key_id(), | 226 tracks_parser.video_encryption_key_id(), |
(...skipping 58 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
294 return -1; | 285 return -1; |
295 } | 286 } |
296 | 287 |
297 WebMClusterParser::TextTrackIterator text_track_iter = | 288 WebMClusterParser::TextTrackIterator text_track_iter = |
298 cluster_parser_->CreateTextTrackIterator(); | 289 cluster_parser_->CreateTextTrackIterator(); |
299 | 290 |
300 int text_track_num; | 291 int text_track_num; |
301 const BufferQueue* text_buffers; | 292 const BufferQueue* text_buffers; |
302 | 293 |
303 while (text_track_iter(&text_track_num, &text_buffers)) { | 294 while (text_track_iter(&text_track_num, &text_buffers)) { |
304 TextTrackMap::iterator find_result = text_track_map_.find(text_track_num); | 295 if (!text_buffers->empty() && !text_cb_.Run(text_track_num, *text_buffers)) |
305 | |
306 if (find_result == text_track_map_.end()) | |
307 continue; | |
308 | |
309 TextTrack* const text_track = find_result->second; | |
310 | |
311 if (!text_buffers->empty() && !text_cb_.Run(text_track, *text_buffers)) | |
312 return -1; | 296 return -1; |
313 } | 297 } |
314 | 298 |
315 if (cluster_ended) | 299 if (cluster_ended) |
316 end_of_segment_cb_.Run(); | 300 end_of_segment_cb_.Run(); |
317 | 301 |
318 return bytes_parsed; | 302 return bytes_parsed; |
319 } | 303 } |
320 | 304 |
321 void WebMStreamParser::FireNeedKey(const std::string& key_id) { | 305 void WebMStreamParser::FireNeedKey(const std::string& key_id) { |
322 std::vector<uint8> key_id_vector(key_id.begin(), key_id.end()); | 306 std::vector<uint8> key_id_vector(key_id.begin(), key_id.end()); |
323 need_key_cb_.Run(kWebMEncryptInitDataType, key_id_vector); | 307 need_key_cb_.Run(kWebMEncryptInitDataType, key_id_vector); |
324 } | 308 } |
325 | 309 |
326 } // namespace media | 310 } // namespace media |
OLD | NEW |