1.1 --- /dev/null Thu Jan 01 00:00:00 1970 +0000
1.2 +++ b/content/media/webm/WebMWriter.cpp Wed Dec 31 06:09:35 2014 +0100
1.3 @@ -0,0 +1,65 @@
1.4 +/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*-*/
1.5 +/* This Source Code Form is subject to the terms of the Mozilla Public
1.6 + * License, v. 2.0. If a copy of the MPL was not distributed with this file,
1.7 + * You can obtain one at http://mozilla.org/MPL/2.0/. */
1.8 +
1.9 +#include "WebMWriter.h"
1.10 +#include "EbmlComposer.h"
1.11 +
1.12 +namespace mozilla {
1.13 +
1.14 +WebMWriter::WebMWriter(uint32_t aTrackTypes) : ContainerWriter()
1.15 +{
1.16 + mMetadataRequiredFlag = aTrackTypes;
1.17 + mEbmlComposer = new EbmlComposer();
1.18 +}
1.19 +
1.20 +nsresult
1.21 +WebMWriter::WriteEncodedTrack(const EncodedFrameContainer& aData,
1.22 + uint32_t aFlags)
1.23 +{
1.24 + for (uint32_t i = 0 ; i < aData.GetEncodedFrames().Length(); i++) {
1.25 + mEbmlComposer->WriteSimpleBlock(aData.GetEncodedFrames().ElementAt(i).get());
1.26 + }
1.27 + return NS_OK;
1.28 +}
1.29 +
1.30 +nsresult
1.31 +WebMWriter::GetContainerData(nsTArray<nsTArray<uint8_t> >* aOutputBufs,
1.32 + uint32_t aFlags)
1.33 +{
1.34 + mEbmlComposer->ExtractBuffer(aOutputBufs, aFlags);
1.35 + if (aFlags & ContainerWriter::FLUSH_NEEDED) {
1.36 + mIsWritingComplete = true;
1.37 + }
1.38 + return NS_OK;
1.39 +}
1.40 +
1.41 +nsresult
1.42 +WebMWriter::SetMetadata(TrackMetadataBase* aMetadata)
1.43 +{
1.44 + MOZ_ASSERT(aMetadata);
1.45 + if (aMetadata->GetKind() == TrackMetadataBase::METADATA_VP8) {
1.46 + VP8Metadata* meta = static_cast<VP8Metadata*>(aMetadata);
1.47 + MOZ_ASSERT(meta, "Cannot find vp8 encoder metadata");
1.48 + mEbmlComposer->SetVideoConfig(meta->mWidth, meta->mHeight,
1.49 + meta->mDisplayWidth, meta->mDisplayHeight,
1.50 + meta->mEncodedFrameRate);
1.51 + mMetadataRequiredFlag = mMetadataRequiredFlag & ~ContainerWriter::CREATE_VIDEO_TRACK;
1.52 + }
1.53 +
1.54 + if (aMetadata->GetKind() == TrackMetadataBase::METADATA_VORBIS) {
1.55 + VorbisMetadata* meta = static_cast<VorbisMetadata*>(aMetadata);
1.56 + MOZ_ASSERT(meta, "Cannot find vorbis encoder metadata");
1.57 + mEbmlComposer->SetAudioConfig(meta->mSamplingFrequency, meta->mChannels, meta->mBitDepth);
1.58 + mEbmlComposer->SetAudioCodecPrivateData(meta->mData);
1.59 + mMetadataRequiredFlag = mMetadataRequiredFlag & ~ContainerWriter::CREATE_AUDIO_TRACK;
1.60 + }
1.61 +
1.62 + if (!mMetadataRequiredFlag) {
1.63 + mEbmlComposer->GenerateHeader();
1.64 + }
1.65 + return NS_OK;
1.66 +}
1.67 +
1.68 +} // mozilla namespace
