blob: 7aa85d5b7d1fb9b58035bd15751d417d52914a51 [file] [log] [blame]
// Copyright (c) 2012 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "media/webm/webm_tracks_parser.h"
#include "base/logging.h"
#include "base/string_util.h"
#include "media/webm/webm_constants.h"
#include "media/webm/webm_content_encodings.h"
namespace media {
// Values for TrackType element.
static const int kWebMTrackTypeVideo = 1;
static const int kWebMTrackTypeAudio = 2;
WebMTracksParser::WebMTracksParser(int64 timecode_scale)
: timecode_scale_(timecode_scale),
track_type_(-1),
track_num_(-1),
track_default_duration_(-1),
audio_track_num_(-1),
video_track_num_(-1) {
}
WebMTracksParser::~WebMTracksParser() {}
const uint8* WebMTracksParser::video_encryption_key_id() const {
if (!video_content_encodings_client_.get())
return NULL;
DCHECK(!video_content_encodings_client_->content_encodings().empty());
return video_content_encodings_client_->content_encodings()[0]->
encryption_key_id();
}
int WebMTracksParser::video_encryption_key_id_size() const {
if (!video_content_encodings_client_.get())
return 0;
DCHECK(!video_content_encodings_client_->content_encodings().empty());
return video_content_encodings_client_->content_encodings()[0]->
encryption_key_id_size();
}
int WebMTracksParser::Parse(const uint8* buf, int size) {
track_type_ =-1;
track_num_ = -1;
track_default_duration_ = -1;
audio_track_num_ = -1;
audio_default_duration_ = base::TimeDelta();
video_track_num_ = -1;
video_default_duration_ = base::TimeDelta();
WebMListParser parser(kWebMIdTracks, this);
int result = parser.Parse(buf, size);
if (result <= 0)
return result;
// For now we do all or nothing parsing.
return parser.IsParsingComplete() ? result : 0;
}
WebMParserClient* WebMTracksParser::OnListStart(int id) {
if (id == kWebMIdContentEncodings) {
DCHECK(!track_content_encodings_client_.get());
track_content_encodings_client_.reset(new WebMContentEncodingsClient);
return track_content_encodings_client_->OnListStart(id);
}
if (id == kWebMIdTrackEntry) {
track_type_ = -1;
track_num_ = -1;
track_default_duration_ = -1;
return this;
}
return this;
}
bool WebMTracksParser::OnListEnd(int id) {
if (id == kWebMIdContentEncodings) {
DCHECK(track_content_encodings_client_.get());
return track_content_encodings_client_->OnListEnd(id);
}
if (id == kWebMIdTrackEntry) {
if (track_type_ == -1 || track_num_ == -1) {
DVLOG(1) << "Missing TrackEntry data"
<< " TrackType " << track_type_
<< " TrackNum " << track_num_;
return false;
}
base::TimeDelta default_duration;
if (track_default_duration_ > 0) {
// Convert nanoseconds to base::TimeDelta.
default_duration = base::TimeDelta::FromMicroseconds(
track_default_duration_ / 1000.0);
}
if (track_type_ == kWebMTrackTypeVideo) {
video_track_num_ = track_num_;
video_default_duration_ = default_duration;
if (track_content_encodings_client_.get()) {
video_content_encodings_client_ =
track_content_encodings_client_.Pass();
}
} else if (track_type_ == kWebMTrackTypeAudio) {
audio_track_num_ = track_num_;
audio_default_duration_ = default_duration;
if (track_content_encodings_client_.get()) {
audio_content_encodings_client_ =
track_content_encodings_client_.Pass();
}
} else {
DVLOG(1) << "Unexpected TrackType " << track_type_;
return false;
}
track_type_ = -1;
track_num_ = -1;
track_content_encodings_client_.reset();
return true;
}
return true;
}
bool WebMTracksParser::OnUInt(int id, int64 val) {
int64* dst = NULL;
switch (id) {
case kWebMIdTrackNumber:
dst = &track_num_;
break;
case kWebMIdTrackType:
dst = &track_type_;
break;
case kWebMIdDefaultDuration:
dst = &track_default_duration_;
break;
default:
return true;
}
if (*dst != -1) {
DVLOG(1) << "Multiple values for id " << std::hex << id << " specified";
return false;
}
*dst = val;
return true;
}
bool WebMTracksParser::OnFloat(int id, double val) {
return true;
}
bool WebMTracksParser::OnBinary(int id, const uint8* data, int size) {
return true;
}
bool WebMTracksParser::OnString(int id, const std::string& str) {
if (id == kWebMIdCodecID && str != "A_VORBIS" && str != "V_VP8") {
DVLOG(1) << "Unexpected CodecID " << str;
return false;
}
return true;
}
} // namespace media