|
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/. */ |
|
5 |
|
6 #include "WebMWriter.h" |
|
7 #include "EbmlComposer.h" |
|
8 |
|
9 namespace mozilla { |
|
10 |
|
11 WebMWriter::WebMWriter(uint32_t aTrackTypes) : ContainerWriter() |
|
12 { |
|
13 mMetadataRequiredFlag = aTrackTypes; |
|
14 mEbmlComposer = new EbmlComposer(); |
|
15 } |
|
16 |
|
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 } |
|
26 |
|
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 } |
|
37 |
|
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 } |
|
50 |
|
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 } |
|
58 |
|
59 if (!mMetadataRequiredFlag) { |
|
60 mEbmlComposer->GenerateHeader(); |
|
61 } |
|
62 return NS_OK; |
|
63 } |
|
64 |
|
65 } // mozilla namespace |