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
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
|
/*
* Copyright 2012, The Android Open Source Project
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#ifndef CONVERTER_H_
#define CONVERTER_H_
#include <media/stagefright/foundation/AHandler.h>
namespace android {
struct ABuffer;
struct IGraphicBufferProducer;
struct MediaCodec;
#define ENABLE_SILENCE_DETECTION 0
// Utility class that receives media access units and converts them into
// media access unit of a different format.
// Right now this'll convert raw video into H.264 and raw audio into AAC.
struct Converter : public AHandler {
enum {
kWhatAccessUnit,
kWhatEOS,
kWhatError,
kWhatShutdownCompleted,
};
enum FlagBits {
FLAG_USE_SURFACE_INPUT = 1,
FLAG_PREPEND_CSD_IF_NECESSARY = 2,
};
Converter(const sp<AMessage> ¬ify,
const sp<ALooper> &codecLooper,
const sp<AMessage> &outputFormat,
uint32_t flags = 0);
status_t init();
sp<IGraphicBufferProducer> getGraphicBufferProducer();
size_t getInputBufferCount() const;
sp<AMessage> getOutputFormat() const;
bool needToManuallyPrependSPSPPS() const;
void feedAccessUnit(const sp<ABuffer> &accessUnit);
void signalEOS();
void requestIDRFrame();
void dropAFrame();
void suspendEncoding(bool suspend);
void shutdownAsync();
int32_t getVideoBitrate() const;
void setVideoBitrate(int32_t bitrate);
static int32_t GetInt32Property(const char *propName, int32_t defaultValue);
enum {
// MUST not conflict with private enums below.
kWhatMediaPullerNotify = 'pulN',
};
protected:
virtual ~Converter();
virtual void onMessageReceived(const sp<AMessage> &msg);
private:
enum {
kWhatDoMoreWork,
kWhatRequestIDRFrame,
kWhatSuspendEncoding,
kWhatShutdown,
kWhatEncoderActivity,
kWhatDropAFrame,
kWhatReleaseOutputBuffer,
};
sp<AMessage> mNotify;
sp<ALooper> mCodecLooper;
sp<AMessage> mOutputFormat;
uint32_t mFlags;
bool mIsVideo;
bool mIsH264;
bool mIsPCMAudio;
bool mNeedToManuallyPrependSPSPPS;
sp<MediaCodec> mEncoder;
sp<AMessage> mEncoderActivityNotify;
sp<IGraphicBufferProducer> mGraphicBufferProducer;
Vector<sp<ABuffer> > mEncoderInputBuffers;
Vector<sp<ABuffer> > mEncoderOutputBuffers;
List<size_t> mAvailEncoderInputIndices;
List<sp<ABuffer> > mInputBufferQueue;
sp<ABuffer> mCSD0;
bool mDoMoreWorkPending;
#if ENABLE_SILENCE_DETECTION
int64_t mFirstSilentFrameUs;
bool mInSilentMode;
#endif
sp<ABuffer> mPartialAudioAU;
int32_t mPrevVideoBitrate;
int32_t mNumFramesToDrop;
bool mEncodingSuspended;
status_t initEncoder();
void releaseEncoder();
status_t feedEncoderInputBuffers();
void scheduleDoMoreWork();
status_t doMoreWork();
void notifyError(status_t err);
// Packetizes raw PCM audio data available in mInputBufferQueue
// into a format suitable for transport stream inclusion and
// notifies the observer.
status_t feedRawAudioInputBuffers();
static bool IsSilence(const sp<ABuffer> &accessUnit);
sp<ABuffer> prependCSD(const sp<ABuffer> &accessUnit) const;
DISALLOW_EVIL_CONSTRUCTORS(Converter);
};
} // namespace android
#endif // CONVERTER_H_
|
