michael@0: /* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*-*/
michael@0: /* This Source Code Form is subject to the terms of the Mozilla Public
michael@0:  * License, v. 2.0. If a copy of the MPL was not distributed with this file,
michael@0:  * You can obtain one at http://mozilla.org/MPL/2.0/. */
michael@0: #include "OggWriter.h"
michael@0: #include "prtime.h"
michael@0: 
michael@0: #undef LOG
michael@0: #ifdef MOZ_WIDGET_GONK
michael@0: #include <android/log.h>
michael@0: #define LOG(args...) __android_log_print(ANDROID_LOG_INFO, "MediaEncoder", ## args);
michael@0: #else
michael@0: #define LOG(args, ...)
michael@0: #endif
michael@0: 
michael@0: namespace mozilla {
michael@0: 
michael@0: OggWriter::OggWriter() : ContainerWriter()
michael@0: {
michael@0:   if (NS_FAILED(Init())) {
michael@0:     LOG("ERROR! Fail to initialize the OggWriter.");
michael@0:   }
michael@0: }
michael@0: 
michael@0: OggWriter::~OggWriter()
michael@0: {
michael@0:   if (mInitialized) {
michael@0:     ogg_stream_clear(&mOggStreamState);
michael@0:   }
michael@0:   // mPacket's data was always owned by us, no need to ogg_packet_clear.
michael@0: }
michael@0: 
michael@0: nsresult
michael@0: OggWriter::Init()
michael@0: {
michael@0:   MOZ_ASSERT(!mInitialized);
michael@0: 
michael@0:   // The serial number (serialno) should be a random number, for the current
michael@0:   // implementation where the output file contains only a single stream, this
michael@0:   // serialno is used to differentiate between files.
michael@0:   srand(static_cast<unsigned>(PR_Now()));
michael@0:   int rc = ogg_stream_init(&mOggStreamState, rand());
michael@0: 
michael@0:   mPacket.b_o_s = 1;
michael@0:   mPacket.e_o_s = 0;
michael@0:   mPacket.granulepos = 0;
michael@0:   mPacket.packet = nullptr;
michael@0:   mPacket.packetno = 0;
michael@0:   mPacket.bytes = 0;
michael@0: 
michael@0:   mInitialized = (rc == 0);
michael@0: 
michael@0:   return (rc == 0) ? NS_OK : NS_ERROR_NOT_INITIALIZED;
michael@0: }
michael@0: 
michael@0: nsresult
michael@0: OggWriter::WriteEncodedTrack(const EncodedFrameContainer& aData,
michael@0:                              uint32_t aFlags)
michael@0: {
michael@0:   for (uint32_t i = 0; i < aData.GetEncodedFrames().Length(); i++) {
michael@0:     if (aData.GetEncodedFrames()[i]->GetFrameType() != EncodedFrame::OPUS_AUDIO_FRAME) {
michael@0:       LOG("[OggWriter] wrong encoded data type!");
michael@0:       return NS_ERROR_FAILURE;
michael@0:     }
michael@0: 
michael@0:     nsresult rv = WriteEncodedData(aData.GetEncodedFrames()[i]->GetFrameData(),
michael@0:                                    aData.GetEncodedFrames()[i]->GetDuration(),
michael@0:                                    aFlags);
michael@0:     if (NS_FAILED(rv)) {
michael@0:       LOG("%p Failed to WriteEncodedTrack!", this);
michael@0:       return rv;
michael@0:     }
michael@0:   }
michael@0:   return NS_OK;
michael@0: }
michael@0: 
michael@0: nsresult
michael@0: OggWriter::WriteEncodedData(const nsTArray<uint8_t>& aBuffer, int aDuration,
michael@0:                             uint32_t aFlags)
michael@0: {
michael@0:   if (!mInitialized) {
michael@0:     LOG("[OggWriter] OggWriter has not initialized!");
michael@0:     return NS_ERROR_FAILURE;
michael@0:   }
michael@0: 
michael@0:   MOZ_ASSERT(!ogg_stream_eos(&mOggStreamState),
michael@0:              "No data can be written after eos has marked.");
michael@0: 
michael@0:   // Set eos flag to true, and once the eos is written to a packet, there must
michael@0:   // not be anymore pages after a page has marked as eos.
michael@0:   if (aFlags & ContainerWriter::END_OF_STREAM) {
michael@0:     LOG("[OggWriter] Set e_o_s flag to true.");
michael@0:     mPacket.e_o_s = 1;
michael@0:   }
michael@0: 
michael@0:   mPacket.packet = const_cast<uint8_t*>(aBuffer.Elements());
michael@0:   mPacket.bytes = aBuffer.Length();
michael@0:   mPacket.granulepos += aDuration;
michael@0: 
michael@0:   // 0 returned on success. -1 returned in the event of internal error.
michael@0:   // The data in the packet is copied into the internal storage managed by the
michael@0:   // mOggStreamState, so we are free to alter the contents of mPacket after
michael@0:   // this call has returned.
michael@0:   int rc = ogg_stream_packetin(&mOggStreamState, &mPacket);
michael@0:   if (rc < 0) {
michael@0:     LOG("[OggWriter] Failed in ogg_stream_packetin! (%d).", rc);
michael@0:     return NS_ERROR_FAILURE;
michael@0:   }
michael@0: 
michael@0:   if (mPacket.b_o_s) {
michael@0:     mPacket.b_o_s = 0;
michael@0:   }
michael@0:   mPacket.packetno++;
michael@0:   mPacket.packet = nullptr;
michael@0: 
michael@0:   return NS_OK;
michael@0: }
michael@0: 
michael@0: void
michael@0: OggWriter::ProduceOggPage(nsTArray<nsTArray<uint8_t> >* aOutputBufs)
michael@0: {
michael@0:   aOutputBufs->AppendElement();
michael@0:   aOutputBufs->LastElement().SetLength(mOggPage.header_len +
michael@0:                                        mOggPage.body_len);
michael@0:   memcpy(aOutputBufs->LastElement().Elements(), mOggPage.header,
michael@0:          mOggPage.header_len);
michael@0:   memcpy(aOutputBufs->LastElement().Elements() + mOggPage.header_len,
michael@0:          mOggPage.body, mOggPage.body_len);
michael@0: }
michael@0: 
michael@0: nsresult
michael@0: OggWriter::GetContainerData(nsTArray<nsTArray<uint8_t> >* aOutputBufs,
michael@0:                             uint32_t aFlags)
michael@0: {
michael@0:   int rc = -1;
michael@0:   // Generate the oggOpus Header
michael@0:   if (aFlags & ContainerWriter::GET_HEADER) {
michael@0:     OpusMetadata* meta = static_cast<OpusMetadata*>(mMetadata.get());
michael@0:     NS_ASSERTION(meta, "should have meta data");
michael@0:     NS_ASSERTION(meta->GetKind() == TrackMetadataBase::METADATA_OPUS,
michael@0:                  "should have Opus meta data");
michael@0: 
michael@0:     nsresult rv = WriteEncodedData(meta->mIdHeader, 0);
michael@0:     NS_ENSURE_SUCCESS(rv, rv);
michael@0: 
michael@0:     rc = ogg_stream_flush(&mOggStreamState, &mOggPage);
michael@0:     NS_ENSURE_TRUE(rc > 0, NS_ERROR_FAILURE);
michael@0:     ProduceOggPage(aOutputBufs);
michael@0: 
michael@0:     rv = WriteEncodedData(meta->mCommentHeader, 0);
michael@0:     NS_ENSURE_SUCCESS(rv, rv);
michael@0: 
michael@0:     rc = ogg_stream_flush(&mOggStreamState, &mOggPage);
michael@0:     NS_ENSURE_TRUE(rc > 0, NS_ERROR_FAILURE);
michael@0: 
michael@0:     ProduceOggPage(aOutputBufs);
michael@0:     return NS_OK;
michael@0: 
michael@0:   // Force generate a page even if the amount of packet data is not enough.
michael@0:   // Usually do so after a header packet.
michael@0:   } else if (aFlags & ContainerWriter::FLUSH_NEEDED) {
michael@0:     // rc = 0 means no packet to put into a page, or an internal error.
michael@0:     rc = ogg_stream_flush(&mOggStreamState, &mOggPage);
michael@0:   } else {
michael@0:     // rc = 0 means insufficient data has accumulated to fill a page, or an
michael@0:     // internal error has occurred.
michael@0:     rc = ogg_stream_pageout(&mOggStreamState, &mOggPage);
michael@0:   }
michael@0: 
michael@0:   if (rc) {
michael@0:     ProduceOggPage(aOutputBufs);
michael@0:   }
michael@0:   if (aFlags & ContainerWriter::FLUSH_NEEDED) {
michael@0:     mIsWritingComplete = true;
michael@0:   }
michael@0:   return (rc > 0) ? NS_OK : NS_ERROR_FAILURE;
michael@0: }
michael@0: 
michael@0: nsresult
michael@0: OggWriter::SetMetadata(TrackMetadataBase* aMetadata)
michael@0: {
michael@0:   MOZ_ASSERT(aMetadata);
michael@0:   if (aMetadata->GetKind() != TrackMetadataBase::METADATA_OPUS) {
michael@0:     LOG("wrong meta data type!");
michael@0:     return NS_ERROR_FAILURE;
michael@0:   }
michael@0:   // Validate each field of METADATA
michael@0:   mMetadata = static_cast<OpusMetadata*>(aMetadata);
michael@0:   if (mMetadata->mIdHeader.Length() == 0) {
michael@0:     LOG("miss mIdHeader!");
michael@0:     return NS_ERROR_FAILURE;
michael@0:   }
michael@0:   if (mMetadata->mCommentHeader.Length() == 0) {
michael@0:     LOG("miss mCommentHeader!");
michael@0:     return NS_ERROR_FAILURE;
michael@0:   }
michael@0: 
michael@0:   return NS_OK;
michael@0: }
michael@0: 
michael@0: }