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
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
|
/*
* 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 __SOFT_AVC_ENC_H__
#define __SOFT_AVC_ENC_H__
#include <media/stagefright/foundation/ABase.h>
#include <utils/Vector.h>
#include "SoftVideoEncoderOMXComponent.h"
namespace android {
#define MAX_INPUT_BUFFER_HEADERS 4
#define MAX_CONVERSION_BUFFERS 4
#define CODEC_MAX_CORES 4
#define LEN_STATUS_BUFFER (10 * 1024)
#define MAX_VBV_BUFF_SIZE (120 * 16384)
#define MAX_NUM_IO_BUFS 3
#define DEFAULT_MAX_REF_FRM 2
#define DEFAULT_MAX_REORDER_FRM 0
#define DEFAULT_QP_MIN 10
#define DEFAULT_QP_MAX 40
#define DEFAULT_MAX_BITRATE 20000000
#define DEFAULT_MAX_SRCH_RANGE_X 256
#define DEFAULT_MAX_SRCH_RANGE_Y 256
#define DEFAULT_MAX_FRAMERATE 120000
#define DEFAULT_NUM_CORES 1
#define DEFAULT_NUM_CORES_PRE_ENC 0
#define DEFAULT_FPS 30
#define DEFAULT_ENC_SPEED IVE_NORMAL
#define DEFAULT_MEM_REC_CNT 0
#define DEFAULT_RECON_ENABLE 0
#define DEFAULT_CHKSUM_ENABLE 0
#define DEFAULT_START_FRM 0
#define DEFAULT_NUM_FRMS 0xFFFFFFFF
#define DEFAULT_INP_COLOR_FORMAT IV_YUV_420SP_VU
#define DEFAULT_RECON_COLOR_FORMAT IV_YUV_420P
#define DEFAULT_LOOPBACK 0
#define DEFAULT_SRC_FRAME_RATE 30
#define DEFAULT_TGT_FRAME_RATE 30
#define DEFAULT_MAX_WD 1920
#define DEFAULT_MAX_HT 1920
#define DEFAULT_MAX_LEVEL 41
#define DEFAULT_STRIDE 0
#define DEFAULT_WD 1280
#define DEFAULT_HT 720
#define DEFAULT_PSNR_ENABLE 0
#define DEFAULT_ME_SPEED 100
#define DEFAULT_ENABLE_FAST_SAD 0
#define DEFAULT_ENABLE_ALT_REF 0
#define DEFAULT_RC_MODE IVE_RC_STORAGE
#define DEFAULT_BITRATE 6000000
#define DEFAULT_I_QP 22
#define DEFAULT_I_QP_MAX DEFAULT_QP_MAX
#define DEFAULT_I_QP_MIN DEFAULT_QP_MIN
#define DEFAULT_P_QP 28
#define DEFAULT_P_QP_MAX DEFAULT_QP_MAX
#define DEFAULT_P_QP_MIN DEFAULT_QP_MIN
#define DEFAULT_B_QP 22
#define DEFAULT_B_QP_MAX DEFAULT_QP_MAX
#define DEFAULT_B_QP_MIN DEFAULT_QP_MIN
#define DEFAULT_AIR IVE_AIR_MODE_NONE
#define DEFAULT_AIR_REFRESH_PERIOD 30
#define DEFAULT_SRCH_RNG_X 64
#define DEFAULT_SRCH_RNG_Y 48
#define DEFAULT_I_INTERVAL 30
#define DEFAULT_IDR_INTERVAL 1000
#define DEFAULT_B_FRAMES 0
#define DEFAULT_DISABLE_DEBLK_LEVEL 0
#define DEFAULT_HPEL 1
#define DEFAULT_QPEL 1
#define DEFAULT_I4 1
#define DEFAULT_EPROFILE IV_PROFILE_BASE
#define DEFAULT_ENTROPY_MODE 0
#define DEFAULT_SLICE_MODE IVE_SLICE_MODE_NONE
#define DEFAULT_SLICE_PARAM 256
#define DEFAULT_ARCH ARCH_ARM_A9Q
#define DEFAULT_SOC SOC_GENERIC
#define DEFAULT_INTRA4x4 0
#define STRLENGTH 500
#define MIN(a, b) ((a) < (b))? (a) : (b)
#define MAX(a, b) ((a) > (b))? (a) : (b)
#define ALIGN16(x) ((((x) + 15) >> 4) << 4)
#define ALIGN128(x) ((((x) + 127) >> 7) << 7)
#define ALIGN4096(x) ((((x) + 4095) >> 12) << 12)
/** Used to remove warnings about unused parameters */
#define UNUSED(x) ((void)(x))
/** Get time */
#define GETTIME(a, b) gettimeofday(a, b);
/** Compute difference between start and end */
#define TIME_DIFF(start, end, diff) \
diff = ((end.tv_sec - start.tv_sec) * 1000000) + \
(end.tv_usec - start.tv_usec);
#define ive_aligned_malloc(alignment, size) memalign(alignment, size)
#define ive_aligned_free(buf) free(buf)
struct SoftAVC : public SoftVideoEncoderOMXComponent {
SoftAVC(
const char *name,
const OMX_CALLBACKTYPE *callbacks,
OMX_PTR appData,
OMX_COMPONENTTYPE **component);
// Override SimpleSoftOMXComponent methods
virtual OMX_ERRORTYPE internalGetParameter(
OMX_INDEXTYPE index, OMX_PTR params);
virtual OMX_ERRORTYPE internalSetParameter(
OMX_INDEXTYPE index, const OMX_PTR params);
virtual void onQueueFilled(OMX_U32 portIndex);
protected:
virtual ~SoftAVC();
private:
enum {
kNumBuffers = 2,
};
// OMX input buffer's timestamp and flags
typedef struct {
int64_t mTimeUs;
int32_t mFlags;
} InputBufferInfo;
int32_t mStride;
struct timeval mTimeStart; // Time at the start of decode()
struct timeval mTimeEnd; // Time at the end of decode()
// If a request for a change it bitrate has been received.
bool mBitrateUpdated;
bool mKeyFrameRequested;
#ifdef FILE_DUMP_ENABLE
char mInFile[200];
char mOutFile[200];
#endif /* FILE_DUMP_ENABLE */
IV_COLOR_FORMAT_T mIvVideoColorFormat;
IV_PROFILE_T mAVCEncProfile;
WORD32 mAVCEncLevel;
bool mStarted;
bool mSpsPpsHeaderReceived;
bool mSawInputEOS;
bool mSawOutputEOS;
bool mSignalledError;
bool mIntra4x4;
bool mEnableFastSad;
bool mEnableAltRef;
bool mReconEnable;
bool mPSNREnable;
bool mEntropyMode;
IVE_SPEED_CONFIG mEncSpeed;
uint8_t *mConversionBuffers[MAX_CONVERSION_BUFFERS];
bool mConversionBuffersFree[MAX_CONVERSION_BUFFERS];
BufferInfo *mInputBufferInfo[MAX_INPUT_BUFFER_HEADERS];
iv_obj_t *mCodecCtx; // Codec context
iv_mem_rec_t *mMemRecords; // Memory records requested by the codec
size_t mNumMemRecords; // Number of memory records requested by codec
size_t mNumCores; // Number of cores used by the codec
UWORD32 mHeaderGenerated;
UWORD32 mBframes;
IV_ARCH_T mArch;
IVE_SLICE_MODE_T mSliceMode;
UWORD32 mSliceParam;
bool mHalfPelEnable;
UWORD32 mIInterval;
UWORD32 mIDRInterval;
UWORD32 mDisableDeblkLevel;
IVE_AIR_MODE_T mAIRMode;
UWORD32 mAIRRefreshPeriod;
void initEncParams();
OMX_ERRORTYPE initEncoder();
OMX_ERRORTYPE releaseEncoder();
// Verifies the component role tried to be set to this OMX component is
// strictly video_encoder.avc
OMX_ERRORTYPE internalSetRoleParams(
const OMX_PARAM_COMPONENTROLETYPE *role);
// Updates bitrate to reflect port settings.
OMX_ERRORTYPE internalSetBitrateParams(
const OMX_VIDEO_PARAM_BITRATETYPE *bitrate);
OMX_ERRORTYPE setConfig(
OMX_INDEXTYPE index, const OMX_PTR _params);
// Handles port definition changes.
OMX_ERRORTYPE internalSetPortParams(
const OMX_PARAM_PORTDEFINITIONTYPE *port);
OMX_ERRORTYPE internalSetFormatParams(
const OMX_VIDEO_PARAM_PORTFORMATTYPE *format);
OMX_ERRORTYPE setFrameType(IV_PICTURE_CODING_TYPE_T e_frame_type);
OMX_ERRORTYPE setQp();
OMX_ERRORTYPE setEncMode(IVE_ENC_MODE_T e_enc_mode);
OMX_ERRORTYPE setDimensions();
OMX_ERRORTYPE setNumCores();
OMX_ERRORTYPE setFrameRate();
OMX_ERRORTYPE setIpeParams();
OMX_ERRORTYPE setBitRate();
OMX_ERRORTYPE setAirParams();
OMX_ERRORTYPE setMeParams();
OMX_ERRORTYPE setGopParams();
OMX_ERRORTYPE setProfileParams();
OMX_ERRORTYPE setDeblockParams();
OMX_ERRORTYPE setVbvParams();
void logVersion();
OMX_ERRORTYPE setEncodeArgs(
ive_video_encode_ip_t *ps_encode_ip,
ive_video_encode_op_t *ps_encode_op,
OMX_BUFFERHEADERTYPE *inputBufferHeader,
OMX_BUFFERHEADERTYPE *outputBufferHeader);
DISALLOW_EVIL_CONSTRUCTORS(SoftAVC);
};
#ifdef FILE_DUMP_ENABLE
#define INPUT_DUMP_PATH "/sdcard/media/avce_input"
#define INPUT_DUMP_EXT "yuv"
#define OUTPUT_DUMP_PATH "/sdcard/media/avce_output"
#define OUTPUT_DUMP_EXT "h264"
#define GENERATE_FILE_NAMES() { \
GETTIME(&mTimeStart, NULL); \
strcpy(mInFile, ""); \
sprintf(mInFile, "%s_%ld.%ld.%s", INPUT_DUMP_PATH, \
mTimeStart.tv_sec, mTimeStart.tv_usec, \
INPUT_DUMP_EXT); \
strcpy(mOutFile, ""); \
sprintf(mOutFile, "%s_%ld.%ld.%s", OUTPUT_DUMP_PATH,\
mTimeStart.tv_sec, mTimeStart.tv_usec, \
OUTPUT_DUMP_EXT); \
}
#define CREATE_DUMP_FILE(m_filename) { \
FILE *fp = fopen(m_filename, "wb"); \
if (fp != NULL) { \
ALOGD("Opened file %s", m_filename); \
fclose(fp); \
} else { \
ALOGD("Could not open file %s", m_filename); \
} \
}
#define DUMP_TO_FILE(m_filename, m_buf, m_size) \
{ \
FILE *fp = fopen(m_filename, "ab"); \
if (fp != NULL && m_buf != NULL) { \
int i; \
i = fwrite(m_buf, 1, m_size, fp); \
ALOGD("fwrite ret %d to write %d", i, m_size); \
if (i != (int)m_size) { \
ALOGD("Error in fwrite, returned %d", i); \
perror("Error in write to file"); \
} \
fclose(fp); \
} else { \
ALOGD("Could not write to file %s", m_filename);\
if (fp != NULL) \
fclose(fp); \
} \
}
#else /* FILE_DUMP_ENABLE */
#define INPUT_DUMP_PATH
#define INPUT_DUMP_EXT
#define OUTPUT_DUMP_PATH
#define OUTPUT_DUMP_EXT
#define GENERATE_FILE_NAMES()
#define CREATE_DUMP_FILE(m_filename)
#define DUMP_TO_FILE(m_filename, m_buf, m_size)
#endif /* FILE_DUMP_ENABLE */
} // namespace android
#endif // __SOFT_AVC_ENC_H__
|
