117 lines
3.1 KiB
C++
117 lines
3.1 KiB
C++
/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*-*/
|
|
/* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this file,
|
|
* You can obtain one at http://mozilla.org/MPL/2.0/. */
|
|
|
|
#ifndef OpusTrackEncoder_h_
|
|
#define OpusTrackEncoder_h_
|
|
|
|
#include <stdint.h>
|
|
#include <speex/speex_resampler.h>
|
|
#include "TimeUnits.h"
|
|
#include "TrackEncoder.h"
|
|
|
|
struct OpusEncoder;
|
|
|
|
namespace mozilla {
|
|
|
|
// Opus meta data structure
|
|
class OpusMetadata : public TrackMetadataBase {
|
|
public:
|
|
// The ID Header of OggOpus. refer to http://wiki.xiph.org/OggOpus.
|
|
nsTArray<uint8_t> mIdHeader;
|
|
// The Comment Header of OggOpus.
|
|
nsTArray<uint8_t> mCommentHeader;
|
|
int32_t mChannels;
|
|
float mSamplingFrequency;
|
|
MetadataKind GetKind() const override { return METADATA_OPUS; }
|
|
};
|
|
|
|
class OpusTrackEncoder : public AudioTrackEncoder {
|
|
public:
|
|
OpusTrackEncoder(TrackRate aTrackRate,
|
|
MediaQueue<EncodedFrame>& aEncodedDataQueue);
|
|
virtual ~OpusTrackEncoder();
|
|
|
|
already_AddRefed<TrackMetadataBase> GetMetadata() override;
|
|
|
|
/**
|
|
* The encoder lookahead at 48k rate.
|
|
*/
|
|
int GetLookahead() const;
|
|
|
|
protected:
|
|
/**
|
|
* The number of frames, in the input rate mTrackRate, needed to fill an
|
|
* encoded opus packet. A frame is a sample per channel.
|
|
*/
|
|
int NumInputFramesPerPacket() const override;
|
|
|
|
nsresult Init(int aChannels) override;
|
|
|
|
/**
|
|
* Encodes buffered data and pushes it to mEncodedDataQueue.
|
|
*/
|
|
nsresult Encode(AudioSegment* aSegment) override;
|
|
|
|
/**
|
|
* The number of frames, in the output rate (see GetOutputSampleRate), needed
|
|
* to fill an encoded opus packet. A frame is a sample per channel.
|
|
*/
|
|
int NumOutputFramesPerPacket() const;
|
|
|
|
/**
|
|
* True if the input needs to be resampled to be fed to the underlying opus
|
|
* encoder.
|
|
*/
|
|
bool NeedsResampler() const;
|
|
|
|
public:
|
|
/**
|
|
* Get the samplerate of the data to be fed to the Opus encoder. This might be
|
|
* different from the input samplerate if resampling occurs.
|
|
*/
|
|
const TrackRate mOutputSampleRate;
|
|
|
|
private:
|
|
/**
|
|
* The Opus encoder from libopus.
|
|
*/
|
|
OpusEncoder* mEncoder;
|
|
|
|
/**
|
|
* Total samples of delay added by codec (in rate mOutputSampleRate), can
|
|
* be queried by the encoder. From the perspective of decoding, real data
|
|
* begins this many samples late, so the encoder needs to append this many
|
|
* null samples to the end of stream, in order to align the time of input and
|
|
* output.
|
|
*/
|
|
int mLookahead;
|
|
|
|
/**
|
|
* Number of mLookahead samples that has been written. When non-zero and equal
|
|
* to mLookahead, encoding is complete.
|
|
*/
|
|
int mLookaheadWritten;
|
|
|
|
/**
|
|
* If the input sample rate does not divide 48kHz evenly, the input data are
|
|
* resampled.
|
|
*/
|
|
SpeexResamplerState* mResampler;
|
|
|
|
/**
|
|
* Store the resampled frames that don't fit into an Opus packet duration.
|
|
* They will be prepended to the resampled frames next encoding cycle.
|
|
*/
|
|
nsTArray<AudioDataValue> mResampledLeftover;
|
|
|
|
/**
|
|
* Number of audio frames encoded, in kOpusSamplingRate.
|
|
*/
|
|
uint64_t mNumOutputFrames;
|
|
};
|
|
|
|
} // namespace mozilla
|
|
|
|
#endif
|