blob: 7f12f7ebb59f6c8738fff69ff21b9bbee726d65d (
plain)
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
|
/* -*- 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 mozilla_AppleVTEncoder_h_
#define mozilla_AppleVTEncoder_h_
#include <CoreMedia/CoreMedia.h>
#include <VideoToolbox/VideoToolbox.h>
#include "PlatformEncoderModule.h"
#include "TimeUnits.h"
namespace mozilla {
namespace layers {
class Image;
}
class AppleVTEncoder final : public MediaDataEncoder {
public:
using Config = H264Config;
AppleVTEncoder(const Config& aConfig, RefPtr<TaskQueue> aTaskQueue,
const bool aHwardwareNotAllowed)
: mConfig(aConfig),
mTaskQueue(aTaskQueue),
mHardwareNotAllowed(aHwardwareNotAllowed),
mFramesCompleted(false),
mError(NS_OK),
mSession(nullptr) {
MOZ_ASSERT(mConfig.mSize.width > 0 && mConfig.mSize.height > 0);
MOZ_ASSERT(mTaskQueue);
}
RefPtr<InitPromise> Init() override;
RefPtr<EncodePromise> Encode(const MediaData* aSample) override;
RefPtr<EncodePromise> Drain() override;
RefPtr<ShutdownPromise> Shutdown() override;
RefPtr<GenericPromise> SetBitrate(Rate aBitsPerSec) override;
nsCString GetDescriptionName() const override {
MOZ_ASSERT(mSession);
return mIsHardwareAccelerated ? "apple hardware VT encoder"_ns
: "apple software VT encoder"_ns;
}
void OutputFrame(CMSampleBufferRef aBuffer);
private:
virtual ~AppleVTEncoder() { MOZ_ASSERT(!mSession); }
RefPtr<EncodePromise> ProcessEncode(RefPtr<const VideoData> aSample);
void ProcessOutput(RefPtr<MediaRawData>&& aOutput);
void ResolvePromise();
RefPtr<EncodePromise> ProcessDrain();
RefPtr<ShutdownPromise> ProcessShutdown();
CFDictionaryRef BuildSourceImageBufferAttributes();
CVPixelBufferRef CreateCVPixelBuffer(const layers::Image* aSource);
bool WriteExtraData(MediaRawData* aDst, CMSampleBufferRef aSrc,
const bool aAsAnnexB);
void AssertOnTaskQueue() { MOZ_ASSERT(mTaskQueue->IsCurrentThreadIn()); }
const Config mConfig;
const RefPtr<TaskQueue> mTaskQueue;
const bool mHardwareNotAllowed;
// Access only in mTaskQueue.
EncodedData mEncodedData;
bool mFramesCompleted;
RefPtr<MediaByteBuffer> mAvcc; // Stores latest avcC data.
MediaResult mError;
// Written by Init() but used only in task queue.
VTCompressionSessionRef mSession;
// Can be accessed on any thread, but only written on during init.
Atomic<bool> mIsHardwareAccelerated;
// Written during init and shutdown.
Atomic<bool> mInited;
};
} // namespace mozilla
#endif // mozilla_AppleVTEncoder_h_
|
