98 lines
2.9 KiB
C++
98 lines
2.9 KiB
C++
/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
|
|
/* vim:set ts=2 sw=2 sts=2 et cindent: */
|
|
/* 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 WMFMediaDataEncoder_h_
|
|
#define WMFMediaDataEncoder_h_
|
|
|
|
#include "MFTEncoder.h"
|
|
#include "PlatformEncoderModule.h"
|
|
#include "WMFDataEncoderUtils.h"
|
|
#include "WMFUtils.h"
|
|
#include <comdef.h>
|
|
#include "mozilla/WindowsProcessMitigations.h"
|
|
|
|
namespace mozilla {
|
|
|
|
class WMFMediaDataEncoder final : public MediaDataEncoder {
|
|
public:
|
|
NS_INLINE_DECL_THREADSAFE_REFCOUNTING(WMFMediaDataEncoder, final);
|
|
|
|
WMFMediaDataEncoder(const EncoderConfig& aConfig,
|
|
const RefPtr<TaskQueue>& aTaskQueue);
|
|
|
|
RefPtr<InitPromise> Init() override;
|
|
RefPtr<EncodePromise> Encode(const MediaData* aSample) override;
|
|
RefPtr<EncodePromise> Drain() override;
|
|
RefPtr<ShutdownPromise> Shutdown() override;
|
|
RefPtr<GenericPromise> SetBitrate(uint32_t aBitsPerSec) override;
|
|
|
|
RefPtr<ReconfigurationPromise> Reconfigure(
|
|
const RefPtr<const EncoderConfigurationChangeList>& aConfigurationChanges)
|
|
override;
|
|
nsCString GetDescriptionName() const override;
|
|
|
|
private:
|
|
~WMFMediaDataEncoder() = default;
|
|
|
|
// Automatically lock/unlock IMFMediaBuffer.
|
|
class LockBuffer final {
|
|
public:
|
|
explicit LockBuffer(RefPtr<IMFMediaBuffer>& aBuffer) : mBuffer(aBuffer) {
|
|
mResult = mBuffer->Lock(&mBytes, &mCapacity, &mLength);
|
|
}
|
|
|
|
~LockBuffer() {
|
|
if (SUCCEEDED(mResult)) {
|
|
mBuffer->Unlock();
|
|
}
|
|
}
|
|
|
|
BYTE* Data() { return mBytes; }
|
|
DWORD Capacity() { return mCapacity; }
|
|
DWORD Length() { return mLength; }
|
|
HRESULT Result() { return mResult; }
|
|
|
|
private:
|
|
RefPtr<IMFMediaBuffer> mBuffer;
|
|
BYTE* mBytes{};
|
|
DWORD mCapacity{};
|
|
DWORD mLength{};
|
|
HRESULT mResult{};
|
|
};
|
|
|
|
RefPtr<InitPromise> ProcessInit();
|
|
|
|
HRESULT InitMFTEncoder(RefPtr<MFTEncoder>& aEncoder);
|
|
void FillConfigData();
|
|
|
|
RefPtr<EncodePromise> ProcessEncode(RefPtr<const VideoData>&& aSample);
|
|
|
|
already_AddRefed<IMFSample> ConvertToNV12InputSample(
|
|
RefPtr<const VideoData>&& aData);
|
|
|
|
RefPtr<EncodePromise> ProcessOutputSamples(
|
|
nsTArray<RefPtr<IMFSample>>& aSamples);
|
|
already_AddRefed<MediaRawData> IMFSampleToMediaData(
|
|
RefPtr<IMFSample>& aSample);
|
|
|
|
bool WriteFrameData(RefPtr<MediaRawData>& aDest, LockBuffer& aSrc,
|
|
bool aIsKeyframe);
|
|
|
|
bool IsAnnexB() const;
|
|
|
|
void AssertOnTaskQueue() { MOZ_ASSERT(mTaskQueue->IsCurrentThreadIn()); }
|
|
|
|
EncoderConfig mConfig;
|
|
const RefPtr<TaskQueue> mTaskQueue;
|
|
const bool mHardwareNotAllowed;
|
|
RefPtr<MFTEncoder> mEncoder;
|
|
// SPS/PPS NALUs when encoding in AnnexB usage, avcC otherwise.
|
|
RefPtr<MediaByteBuffer> mConfigData;
|
|
};
|
|
|
|
} // namespace mozilla
|
|
|
|
#endif
|