Fri, 16 Jan 2015 04:50:19 +0100
Replace accessor implementation with direct member state manipulation, by
request https://trac.torproject.org/projects/tor/ticket/9701#comment:32
1 /* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*-*/
2 /* This Source Code Form is subject to the terms of the Mozilla Public
3 * License, v. 2.0. If a copy of the MPL was not distributed with this file,
4 * You can obtain one at http://mozilla.org/MPL/2.0/. */
6 #include "WebMWriter.h"
7 #include "EbmlComposer.h"
9 namespace mozilla {
11 WebMWriter::WebMWriter(uint32_t aTrackTypes) : ContainerWriter()
12 {
13 mMetadataRequiredFlag = aTrackTypes;
14 mEbmlComposer = new EbmlComposer();
15 }
17 nsresult
18 WebMWriter::WriteEncodedTrack(const EncodedFrameContainer& aData,
19 uint32_t aFlags)
20 {
21 for (uint32_t i = 0 ; i < aData.GetEncodedFrames().Length(); i++) {
22 mEbmlComposer->WriteSimpleBlock(aData.GetEncodedFrames().ElementAt(i).get());
23 }
24 return NS_OK;
25 }
27 nsresult
28 WebMWriter::GetContainerData(nsTArray<nsTArray<uint8_t> >* aOutputBufs,
29 uint32_t aFlags)
30 {
31 mEbmlComposer->ExtractBuffer(aOutputBufs, aFlags);
32 if (aFlags & ContainerWriter::FLUSH_NEEDED) {
33 mIsWritingComplete = true;
34 }
35 return NS_OK;
36 }
38 nsresult
39 WebMWriter::SetMetadata(TrackMetadataBase* aMetadata)
40 {
41 MOZ_ASSERT(aMetadata);
42 if (aMetadata->GetKind() == TrackMetadataBase::METADATA_VP8) {
43 VP8Metadata* meta = static_cast<VP8Metadata*>(aMetadata);
44 MOZ_ASSERT(meta, "Cannot find vp8 encoder metadata");
45 mEbmlComposer->SetVideoConfig(meta->mWidth, meta->mHeight,
46 meta->mDisplayWidth, meta->mDisplayHeight,
47 meta->mEncodedFrameRate);
48 mMetadataRequiredFlag = mMetadataRequiredFlag & ~ContainerWriter::CREATE_VIDEO_TRACK;
49 }
51 if (aMetadata->GetKind() == TrackMetadataBase::METADATA_VORBIS) {
52 VorbisMetadata* meta = static_cast<VorbisMetadata*>(aMetadata);
53 MOZ_ASSERT(meta, "Cannot find vorbis encoder metadata");
54 mEbmlComposer->SetAudioConfig(meta->mSamplingFrequency, meta->mChannels, meta->mBitDepth);
55 mEbmlComposer->SetAudioCodecPrivateData(meta->mData);
56 mMetadataRequiredFlag = mMetadataRequiredFlag & ~ContainerWriter::CREATE_AUDIO_TRACK;
57 }
59 if (!mMetadataRequiredFlag) {
60 mEbmlComposer->GenerateHeader();
61 }
62 return NS_OK;
63 }
65 } // mozilla namespace