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
|
/* -*- 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 __FFmpegDataDecoder_h__
#define __FFmpegDataDecoder_h__
#include "FFmpegLibWrapper.h"
#include "PlatformDecoderModule.h"
#include "mozilla/StaticMutex.h"
// This must be the last header included
#include "FFmpegLibs.h"
namespace mozilla {
template <int V>
class FFmpegDataDecoder : public MediaDataDecoder {};
template <>
class FFmpegDataDecoder<LIBAV_VER>;
DDLoggedTypeNameAndBase(FFmpegDataDecoder<LIBAV_VER>, MediaDataDecoder);
template <>
class FFmpegDataDecoder<LIBAV_VER>
: public MediaDataDecoder,
public DecoderDoctorLifeLogger<FFmpegDataDecoder<LIBAV_VER>> {
public:
NS_INLINE_DECL_THREADSAFE_REFCOUNTING(FFmpegDataDecoder, final);
FFmpegDataDecoder(FFmpegLibWrapper* aLib, AVCodecID aCodecID);
static bool Link();
RefPtr<InitPromise> Init() override = 0;
RefPtr<DecodePromise> Decode(MediaRawData* aSample) override;
RefPtr<DecodePromise> Drain() override;
RefPtr<FlushPromise> Flush() override;
RefPtr<ShutdownPromise> Shutdown() override;
static AVCodec* FindAVCodec(FFmpegLibWrapper* aLib, AVCodecID aCodec);
#ifdef MOZ_WAYLAND
static AVCodec* FindHardwareAVCodec(FFmpegLibWrapper* aLib, AVCodecID aCodec);
#endif
protected:
// Flush and Drain operation, always run
virtual RefPtr<FlushPromise> ProcessFlush();
virtual void ProcessShutdown();
virtual void InitCodecContext() MOZ_REQUIRES(sMutex) {}
AVFrame* PrepareFrame();
MediaResult InitDecoder();
MediaResult AllocateExtraData();
MediaResult DoDecode(MediaRawData* aSample, bool* aGotFrame,
DecodedData& aResults);
FFmpegLibWrapper* mLib; // set in constructor
// mCodecContext is accessed on taskqueue only, no locking needed
AVCodecContext* mCodecContext;
AVCodecParserContext* mCodecParser;
AVFrame* mFrame;
RefPtr<MediaByteBuffer> mExtraData;
AVCodecID mCodecID; // set in constructor
protected:
virtual ~FFmpegDataDecoder();
static StaticMutex sMutex; // used to provide critical-section locking
// for calls into ffmpeg
const RefPtr<TaskQueue> mTaskQueue; // set in constructor
private:
RefPtr<DecodePromise> ProcessDecode(MediaRawData* aSample);
RefPtr<DecodePromise> ProcessDrain();
virtual MediaResult DoDecode(MediaRawData* aSample, uint8_t* aData, int aSize,
bool* aGotFrame,
MediaDataDecoder::DecodedData& aOutResults) = 0;
virtual bool NeedParser() const { return false; }
virtual int ParserFlags() const { return PARSER_FLAG_COMPLETE_FRAMES; }
MozPromiseHolder<DecodePromise> mPromise;
media::TimeUnit mLastInputDts; // used on Taskqueue
};
} // namespace mozilla
#endif // __FFmpegDataDecoder_h__
|
