1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
|
/* -*- 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 DOM_MEDIA_PLATFORMS_FFMPEG_FFMPEGVIDEOENCODER_H_
#define DOM_MEDIA_PLATFORMS_FFMPEG_FFMPEGVIDEOENCODER_H_
#include "FFmpegLibWrapper.h"
#include "PlatformEncoderModule.h"
#include "SimpleMap.h"
#include "mozilla/ThreadSafety.h"
// This must be the last header included
#include "FFmpegLibs.h"
namespace mozilla {
template <int V>
AVCodecID GetFFmpegEncoderCodecId(CodecType aCodec);
template <>
AVCodecID GetFFmpegEncoderCodecId<LIBAV_VER>(CodecType aCodec);
template <int V>
class FFmpegVideoEncoder : public MediaDataEncoder {};
// TODO: Bug 1860925: FFmpegDataEncoder
template <>
class FFmpegVideoEncoder<LIBAV_VER> final : public MediaDataEncoder {
using DurationMap = SimpleMap<int64_t>;
public:
FFmpegVideoEncoder(const FFmpegLibWrapper* aLib, AVCodecID aCodecID,
const RefPtr<TaskQueue>& aTaskQueue,
const EncoderConfig& aConfig);
/* MediaDataEncoder Methods */
// All methods run on the task queue, except for GetDescriptionName.
RefPtr<InitPromise> Init() override;
RefPtr<EncodePromise> Encode(const MediaData* aSample) override;
RefPtr<ReconfigurationPromise> Reconfigure(
const RefPtr<const EncoderConfigurationChangeList>& aConfigurationChanges)
override;
RefPtr<EncodePromise> Drain() override;
RefPtr<ShutdownPromise> Shutdown() override;
RefPtr<GenericPromise> SetBitrate(uint32_t aBitRate) override;
nsCString GetDescriptionName() const override;
private:
~FFmpegVideoEncoder() = default;
// Methods only called on mTaskQueue.
RefPtr<InitPromise> ProcessInit();
RefPtr<EncodePromise> ProcessEncode(RefPtr<const MediaData> aSample);
RefPtr<ReconfigurationPromise> ProcessReconfigure(
const RefPtr<const EncoderConfigurationChangeList> aConfigurationChanges);
RefPtr<EncodePromise> ProcessDrain();
RefPtr<ShutdownPromise> ProcessShutdown();
MediaResult InitInternal();
void ShutdownInternal();
// TODO: Share these with FFmpegDataDecoder.
int OpenCodecContext(const AVCodec* aCodec, AVDictionary** aOptions)
MOZ_EXCLUDES(sMutex);
void CloseCodecContext() MOZ_EXCLUDES(sMutex);
bool PrepareFrame();
void DestroyFrame();
bool ScaleInputFrame();
#if LIBAVCODEC_VERSION_MAJOR >= 58
RefPtr<EncodePromise> EncodeWithModernAPIs(RefPtr<const VideoData> aSample);
RefPtr<EncodePromise> DrainWithModernAPIs();
#endif
RefPtr<MediaRawData> ToMediaRawData(AVPacket* aPacket);
Result<already_AddRefed<MediaByteBuffer>, nsresult> GetExtraData(
AVPacket* aPacket);
void ForceEnablingFFmpegDebugLogs();
struct SVCSettings {
nsTArray<uint8_t> mTemporalLayerIds;
// A key-value pair for av_opt_set.
std::pair<nsCString, nsCString> mSettingKeyValue;
};
Maybe<SVCSettings> GetSVCSettings();
struct H264Settings {
int mProfile;
int mLevel;
// A list of key-value pairs for av_opt_set.
nsTArray<std::pair<nsCString, nsCString>> mSettingKeyValuePairs;
};
H264Settings GetH264Settings(const H264Specific& aH264Specific);
// This refers to a static FFmpegLibWrapper, so raw pointer is adequate.
const FFmpegLibWrapper* mLib;
const AVCodecID mCodecID;
const RefPtr<TaskQueue> mTaskQueue;
// set in constructor, modified when parameters change
EncoderConfig mConfig;
// mTaskQueue only.
nsCString mCodecName;
AVCodecContext* mCodecContext;
AVFrame* mFrame;
DurationMap mDurationMap;
struct SVCInfo {
explicit SVCInfo(nsTArray<uint8_t>&& aTemporalLayerIds)
: mTemporalLayerIds(std::move(aTemporalLayerIds)), mNextIndex(0) {}
const nsTArray<uint8_t> mTemporalLayerIds;
size_t mNextIndex;
// Return the current temporal layer id and update the next.
uint8_t UpdateTemporalLayerId();
};
Maybe<SVCInfo> mSVCInfo;
// Provide critical-section for open/close mCodecContext.
// TODO: Merge this with FFmpegDataDecoder's one.
static StaticMutex sMutex;
};
} // namespace mozilla
#endif /* DOM_MEDIA_PLATFORMS_FFMPEG_FFMPEGVIDEOENCODER_H_ */
|
