blob: e38735dd3863db672b79b492ffdcfd27c2ec76ac [file] [log] [blame]
Fyodor Kyslovadc71142022-09-15 16:47:03 +00001/*
2 * Copyright (C) 2022 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17//#define LOG_NDEBUG 0
18#define LOG_TAG "C2SoftAomEnc"
19#include <log/log.h>
20
21#include <media/stagefright/foundation/AUtils.h>
22#include <media/stagefright/foundation/MediaDefs.h>
23
24#include <C2Debug.h>
Harish Mahendrakar7de78d42023-02-06 17:12:18 -080025#include <Codec2CommonUtils.h>
Aayush Soni643b9182023-01-18 20:23:56 +053026#include <Codec2Mapper.h>
Fyodor Kyslovadc71142022-09-15 16:47:03 +000027#include <C2PlatformSupport.h>
28#include <SimpleC2Interface.h>
29
30#include "C2SoftAomEnc.h"
31
32namespace android {
33
34constexpr char COMPONENT_NAME[] = "c2.android.av1.encoder";
35
36#define DEFAULT_SPEED 10
37
38C2SoftAomEnc::IntfImpl::IntfImpl(const std::shared_ptr<C2ReflectorHelper>& helper)
39 : SimpleInterface<void>::BaseParams(helper, COMPONENT_NAME, C2Component::KIND_ENCODER,
40 C2Component::DOMAIN_VIDEO, MEDIA_MIMETYPE_VIDEO_AV1) {
41 noPrivateBuffers(); // TODO: account for our buffers here
42 noInputReferences();
43 noOutputReferences();
44 noInputLatency();
45 noTimeStretch();
46 setDerivedInstance(this);
47
48 addParameter(DefineParam(mUsage, C2_PARAMKEY_INPUT_STREAM_USAGE)
49 .withConstValue(new C2StreamUsageTuning::input(
50 0u, (uint64_t)C2MemoryUsage::CPU_READ))
51 .build());
52
53 addParameter(DefineParam(mSize, C2_PARAMKEY_PICTURE_SIZE)
54 .withDefault(new C2StreamPictureSizeInfo::input(0u, 320, 240))
55 .withFields({
56 C2F(mSize, width).inRange(2, 2048, 2),
57 C2F(mSize, height).inRange(2, 2048, 2),
58 })
59 .withSetter(SizeSetter)
60 .build());
61
62 addParameter(DefineParam(mBitrateMode, C2_PARAMKEY_BITRATE_MODE)
63 .withDefault(new C2StreamBitrateModeTuning::output(
64 0u, C2Config::BITRATE_VARIABLE))
65 .withFields({C2F(mBitrateMode, value)
66 .oneOf({C2Config::BITRATE_CONST,
Fyodor Kyslov58d8dea2023-02-10 02:15:54 +000067 C2Config::BITRATE_VARIABLE,
68 C2Config::BITRATE_IGNORE})})
Fyodor Kyslovadc71142022-09-15 16:47:03 +000069 .withSetter(Setter<decltype(*mBitrateMode)>::StrictValueWithNoDeps)
70 .build());
71
72 addParameter(DefineParam(mFrameRate, C2_PARAMKEY_FRAME_RATE)
73 .withDefault(new C2StreamFrameRateInfo::output(0u, 30.))
74 // TODO: More restriction?
75 .withFields({C2F(mFrameRate, value).greaterThan(0.)})
76 .withSetter(Setter<decltype(*mFrameRate)>::StrictValueWithNoDeps)
77 .build());
78
79 addParameter(DefineParam(mSyncFramePeriod, C2_PARAMKEY_SYNC_FRAME_INTERVAL)
80 .withDefault(new C2StreamSyncFrameIntervalTuning::output(0u, 1000000))
81 .withFields({C2F(mSyncFramePeriod, value).any()})
82 .withSetter(Setter<decltype(*mSyncFramePeriod)>::StrictValueWithNoDeps)
83 .build());
84
85 addParameter(DefineParam(mBitrate, C2_PARAMKEY_BITRATE)
86 .withDefault(new C2StreamBitrateInfo::output(0u, 64000))
87 .withFields({C2F(mBitrate, value).inRange(4096, 40000000)})
88 .withSetter(BitrateSetter)
89 .build());
90
Fyodor Kyslov3e3a92b2023-03-15 00:39:36 +000091 addParameter(DefineParam(mComplexity, C2_PARAMKEY_COMPLEXITY)
92 .withDefault(new C2StreamComplexityTuning::output(0u, 0))
93 .withFields({C2F(mComplexity, value).inRange(0, 5)})
94 .withSetter(Setter<decltype(*mComplexity)>::NonStrictValueWithNoDeps)
95 .build());
96
Fyodor Kyslov58d8dea2023-02-10 02:15:54 +000097 addParameter(DefineParam(mQuality, C2_PARAMKEY_QUALITY)
98 .withDefault(new C2StreamQualityTuning::output(0u, 80))
99 .withFields({C2F(mQuality, value).inRange(0, 100)})
100 .withSetter(Setter<decltype(*mQuality)>::NonStrictValueWithNoDeps)
101 .build());
102
Fyodor Kyslovadc71142022-09-15 16:47:03 +0000103 addParameter(DefineParam(mIntraRefresh, C2_PARAMKEY_INTRA_REFRESH)
104 .withConstValue(new C2StreamIntraRefreshTuning::output(
105 0u, C2Config::INTRA_REFRESH_DISABLED, 0.))
106 .build());
107
108 addParameter(DefineParam(mProfileLevel, C2_PARAMKEY_PROFILE_LEVEL)
109 .withDefault(new C2StreamProfileLevelInfo::output(0u, PROFILE_AV1_0,
110 LEVEL_AV1_4_1))
111 .withFields({
112 C2F(mProfileLevel, profile).equalTo(PROFILE_AV1_0),
Fyodor Kyslova8092322022-10-09 02:47:33 +0000113 C2F(mProfileLevel, level)
114 .oneOf({LEVEL_AV1_2, LEVEL_AV1_2_1, LEVEL_AV1_2_2,
115 LEVEL_AV1_2_3, LEVEL_AV1_3, LEVEL_AV1_3_1,
116 LEVEL_AV1_3_2, LEVEL_AV1_3_3, LEVEL_AV1_4,
117 LEVEL_AV1_4_1}),
Fyodor Kyslovadc71142022-09-15 16:47:03 +0000118 })
mtk12101c1670e92023-02-10 09:34:02 +0800119 .withSetter(ProfileLevelSetter, mSize, mFrameRate, mBitrate)
Fyodor Kyslovadc71142022-09-15 16:47:03 +0000120 .build());
121
Harish Mahendrakar7de78d42023-02-06 17:12:18 -0800122 std::vector<uint32_t> pixelFormats = {HAL_PIXEL_FORMAT_IMPLEMENTATION_DEFINED,
123 HAL_PIXEL_FORMAT_YCBCR_420_888};
124 if (isHalPixelFormatSupported((AHardwareBuffer_Format)HAL_PIXEL_FORMAT_YCBCR_P010)) {
125 pixelFormats.push_back(HAL_PIXEL_FORMAT_YCBCR_P010);
126 }
Fyodor Kyslovd97e9912022-11-07 19:23:21 +0000127 addParameter(DefineParam(mPixelFormat, C2_PARAMKEY_PIXEL_FORMAT)
Harish Mahendrakar7de78d42023-02-06 17:12:18 -0800128 .withDefault(new C2StreamPixelFormatInfo::input(
Fyodor Kyslovd97e9912022-11-07 19:23:21 +0000129 0u, HAL_PIXEL_FORMAT_YCBCR_420_888))
Harish Mahendrakar7de78d42023-02-06 17:12:18 -0800130 .withFields({C2F(mPixelFormat, value).oneOf({pixelFormats})})
Fyodor Kyslovd97e9912022-11-07 19:23:21 +0000131 .withSetter((Setter<decltype(*mPixelFormat)>::StrictValueWithNoDeps))
132 .build());
133
Fyodor Kyslovadc71142022-09-15 16:47:03 +0000134 addParameter(DefineParam(mRequestSync, C2_PARAMKEY_REQUEST_SYNC_FRAME)
135 .withDefault(new C2StreamRequestSyncFrameTuning::output(0u, C2_FALSE))
136 .withFields({C2F(mRequestSync, value).oneOf({C2_FALSE, C2_TRUE})})
137 .withSetter(Setter<decltype(*mRequestSync)>::NonStrictValueWithNoDeps)
138 .build());
139 addParameter(
140 DefineParam(mColorAspects, C2_PARAMKEY_COLOR_ASPECTS)
141 .withDefault(new C2StreamColorAspectsInfo::input(
142 0u, C2Color::RANGE_UNSPECIFIED, C2Color::PRIMARIES_UNSPECIFIED,
143 C2Color::TRANSFER_UNSPECIFIED, C2Color::MATRIX_UNSPECIFIED))
144 .withFields(
145 {C2F(mColorAspects, range)
146 .inRange(C2Color::RANGE_UNSPECIFIED, C2Color::RANGE_OTHER),
147 C2F(mColorAspects, primaries)
148 .inRange(C2Color::PRIMARIES_UNSPECIFIED,
149 C2Color::PRIMARIES_OTHER),
150 C2F(mColorAspects, transfer)
151 .inRange(C2Color::TRANSFER_UNSPECIFIED,
152 C2Color::TRANSFER_OTHER),
153 C2F(mColorAspects, matrix)
154 .inRange(C2Color::MATRIX_UNSPECIFIED, C2Color::MATRIX_OTHER)})
155 .withSetter(ColorAspectsSetter)
156 .build());
157
158 addParameter(
159 DefineParam(mCodedColorAspects, C2_PARAMKEY_VUI_COLOR_ASPECTS)
160 .withDefault(new C2StreamColorAspectsInfo::output(
161 0u, C2Color::RANGE_LIMITED, C2Color::PRIMARIES_UNSPECIFIED,
162 C2Color::TRANSFER_UNSPECIFIED, C2Color::MATRIX_UNSPECIFIED))
163 .withFields(
164 {C2F(mCodedColorAspects, range)
165 .inRange(C2Color::RANGE_UNSPECIFIED, C2Color::RANGE_OTHER),
166 C2F(mCodedColorAspects, primaries)
167 .inRange(C2Color::PRIMARIES_UNSPECIFIED,
168 C2Color::PRIMARIES_OTHER),
169 C2F(mCodedColorAspects, transfer)
170 .inRange(C2Color::TRANSFER_UNSPECIFIED,
171 C2Color::TRANSFER_OTHER),
172 C2F(mCodedColorAspects, matrix)
173 .inRange(C2Color::MATRIX_UNSPECIFIED, C2Color::MATRIX_OTHER)})
174 .withSetter(CodedColorAspectsSetter, mColorAspects)
175 .build());
176}
177
178C2R C2SoftAomEnc::IntfImpl::BitrateSetter(bool mayBlock, C2P<C2StreamBitrateInfo::output>& me) {
179 (void)mayBlock;
180 C2R res = C2R::Ok();
181 if (me.v.value < 4096) {
182 me.set().value = 4096;
183 }
184 return res;
185}
186
187C2R C2SoftAomEnc::IntfImpl::SizeSetter(bool mayBlock,
188 const C2P<C2StreamPictureSizeInfo::input>& oldMe,
189 C2P<C2StreamPictureSizeInfo::input>& me) {
190 (void)mayBlock;
191 C2R res = C2R::Ok();
192 if (!me.F(me.v.width).supportsAtAll(me.v.width)) {
193 res = res.plus(C2SettingResultBuilder::BadValue(me.F(me.v.width)));
194 me.set().width = oldMe.v.width;
195 }
196 if (!me.F(me.v.height).supportsAtAll(me.v.height)) {
197 res = res.plus(C2SettingResultBuilder::BadValue(me.F(me.v.height)));
198 me.set().height = oldMe.v.height;
199 }
200 return res;
201}
202
203C2R C2SoftAomEnc::IntfImpl::ProfileLevelSetter(bool mayBlock,
mtk12101c1670e92023-02-10 09:34:02 +0800204 C2P<C2StreamProfileLevelInfo::output>& me,
205 const C2P<C2StreamPictureSizeInfo::input>& size,
206 const C2P<C2StreamFrameRateInfo::output>& frameRate,
207 const C2P<C2StreamBitrateInfo::output>& bitrate) {
Fyodor Kyslovadc71142022-09-15 16:47:03 +0000208 (void)mayBlock;
209 if (!me.F(me.v.profile).supportsAtAll(me.v.profile)) {
210 me.set().profile = PROFILE_AV1_0;
211 }
mtk12101c1670e92023-02-10 09:34:02 +0800212 struct LevelLimits {
213 C2Config::level_t level;
214 float samplesPerSec;
215 uint64_t samples;
216 uint32_t bitrate;
217 size_t maxHSize;
218 size_t maxVSize;
219 };
220 constexpr LevelLimits kLimits[] = {
221 {LEVEL_AV1_2, 4423680, 147456, 1500000, 2048, 1152},
222 {LEVEL_AV1_2_1, 8363520, 278784, 3000000, 2816, 1584},
223 {LEVEL_AV1_3, 19975680, 665856, 6000000, 4352, 2448},
224 {LEVEL_AV1_3_1, 37950720, 1065024, 10000000, 5504, 3096},
225 {LEVEL_AV1_4, 70778880, 2359296, 12000000, 6144, 3456},
226 {LEVEL_AV1_4_1, 141557760, 2359296, 20000000, 6144, 3456},
227 };
228
229 uint64_t samples = size.v.width * size.v.height;
230 float samplesPerSec = float(samples) * frameRate.v.value;
231
232 // Check if the supplied level meets the samples / bitrate requirements.
233 // If not, update the level with the lowest level meeting the requirements.
234 bool found = false;
235
236 // By default needsUpdate = false in case the supplied level does meet
237 // the requirements.
238 bool needsUpdate = false;
Fyodor Kyslovadc71142022-09-15 16:47:03 +0000239 if (!me.F(me.v.level).supportsAtAll(me.v.level)) {
mtk12101c1670e92023-02-10 09:34:02 +0800240 needsUpdate = true;
241 }
242 for (const LevelLimits& limit : kLimits) {
243 if (samples <= limit.samples && samplesPerSec <= limit.samplesPerSec &&
244 bitrate.v.value <= limit.bitrate && size.v.width <= limit.maxHSize &&
245 size.v.height <= limit.maxVSize) {
246 // This is the lowest level that meets the requirements, and if
247 // we haven't seen the supplied level yet, that means we don't
248 // need the update.
249 if (needsUpdate) {
250 ALOGD("Given level %x does not cover current configuration: "
251 "adjusting to %x",
252 me.v.level, limit.level);
253 me.set().level = limit.level;
254 }
255 found = true;
256 break;
257 }
258 if (me.v.level == limit.level) {
259 // We break out of the loop when the lowest feasible level is
260 // found. The fact that we're here means that our level doesn't
261 // meet the requirement and needs to be updated.
262 needsUpdate = true;
263 }
264 }
265 if (!found) {
266 // We set to the highest supported level.
Fyodor Kyslovadc71142022-09-15 16:47:03 +0000267 me.set().level = LEVEL_AV1_4_1;
268 }
269 return C2R::Ok();
270}
271
272uint32_t C2SoftAomEnc::IntfImpl::getSyncFramePeriod() const {
273 if (mSyncFramePeriod->value < 0 || mSyncFramePeriod->value == INT64_MAX) {
274 return 0;
275 }
276 double period = mSyncFramePeriod->value / 1e6 * mFrameRate->value;
277 return (uint32_t)c2_max(c2_min(period + 0.5, double(UINT32_MAX)), 1.);
278}
279
280C2R C2SoftAomEnc::IntfImpl::ColorAspectsSetter(bool mayBlock,
281 C2P<C2StreamColorAspectsInfo::input>& me) {
282 (void)mayBlock;
283 if (me.v.range > C2Color::RANGE_OTHER) {
284 me.set().range = C2Color::RANGE_OTHER;
285 }
286 if (me.v.primaries > C2Color::PRIMARIES_OTHER) {
287 me.set().primaries = C2Color::PRIMARIES_OTHER;
288 }
289 if (me.v.transfer > C2Color::TRANSFER_OTHER) {
290 me.set().transfer = C2Color::TRANSFER_OTHER;
291 }
292 if (me.v.matrix > C2Color::MATRIX_OTHER) {
293 me.set().matrix = C2Color::MATRIX_OTHER;
294 }
295 return C2R::Ok();
296}
297C2R C2SoftAomEnc::IntfImpl::CodedColorAspectsSetter(
298 bool mayBlock, C2P<C2StreamColorAspectsInfo::output>& me,
299 const C2P<C2StreamColorAspectsInfo::input>& coded) {
300 (void)mayBlock;
301 me.set().range = coded.v.range;
302 me.set().primaries = coded.v.primaries;
303 me.set().transfer = coded.v.transfer;
304 me.set().matrix = coded.v.matrix;
305 return C2R::Ok();
306}
307
308C2SoftAomEnc::C2SoftAomEnc(const char* name, c2_node_id_t id,
309 const std::shared_ptr<IntfImpl>& intfImpl)
310 : SimpleC2Component(std::make_shared<SimpleInterface<IntfImpl>>(name, id, intfImpl)),
311 mIntf(intfImpl),
312 mCodecContext(nullptr),
313 mCodecConfiguration(nullptr),
314 mCodecInterface(nullptr),
315 mStrideAlign(2),
316 mBitrateControlMode(AOM_VBR),
317 mMinQuantizer(0),
318 mMaxQuantizer(0),
Fyodor Kyslovd97e9912022-11-07 19:23:21 +0000319 mLastTimestamp(INT64_MAX),
Fyodor Kyslovadc71142022-09-15 16:47:03 +0000320 mSignalledOutputEos(false),
321 mSignalledError(false),
Fyodor Kyslovd97e9912022-11-07 19:23:21 +0000322 mHeadersReceived(false),
323 mIs10Bit(false) {
Fyodor Kyslovadc71142022-09-15 16:47:03 +0000324 ALOGV("Constructor");
325}
326
327C2SoftAomEnc::~C2SoftAomEnc() {
328 ALOGV("Destructor");
329 onRelease();
330}
331
332c2_status_t C2SoftAomEnc::onInit() {
Fyodor Kyslovd97e9912022-11-07 19:23:21 +0000333 return C2_OK;
Fyodor Kyslovadc71142022-09-15 16:47:03 +0000334}
335
336c2_status_t C2SoftAomEnc::onStop() {
337 onRelease();
338 return C2_OK;
339}
340
341void C2SoftAomEnc::onReset() {
342 (void)onStop();
343}
344
345void C2SoftAomEnc::onRelease() {
346 if (mCodecContext) {
347 aom_codec_destroy(mCodecContext);
348 delete mCodecContext;
349 mCodecContext = nullptr;
350 }
351
352 if (mCodecConfiguration) {
353 delete mCodecConfiguration;
354 mCodecConfiguration = nullptr;
355 }
356
357 // this one is not allocated by us
358 mCodecInterface = nullptr;
Fyodor Kyslovd97e9912022-11-07 19:23:21 +0000359 mHeadersReceived = false;
Fyodor Kyslovadc71142022-09-15 16:47:03 +0000360}
361
362c2_status_t C2SoftAomEnc::onFlush_sm() {
363 return onStop();
364}
365
Fyodor Kyslov58d8dea2023-02-10 02:15:54 +0000366// c2Quality is in range of 0-100 (the more - the better),
367// for AOM quality we are using a range of 15-50 (the less - the better)
368static int MapC2QualityToAOMQuality (int c2Quality) {
369 return 15 + 35 * (100 - c2Quality) / 100;
370}
371
Fyodor Kyslov3e3a92b2023-03-15 00:39:36 +0000372static int MapC2ComplexityToAOMSpeed (int c2Complexity) {
373 int mapping[6] = {10, 9, 8, 7, 6, 6};
374 if (c2Complexity > 5 || c2Complexity < 0) {
375 ALOGW("Wrong complexity setting. Falling back to speed 10");
376 return 10;
377 }
378 return mapping[c2Complexity];
379}
380
Fyodor Kyslovadc71142022-09-15 16:47:03 +0000381aom_codec_err_t C2SoftAomEnc::setupCodecParameters() {
382 aom_codec_err_t codec_return = AOM_CODEC_OK;
383
Fyodor Kyslov3e3a92b2023-03-15 00:39:36 +0000384 codec_return = aom_codec_control(mCodecContext, AOME_SET_CPUUSED,
385 MapC2ComplexityToAOMSpeed(mComplexity->value));
Fyodor Kyslovadc71142022-09-15 16:47:03 +0000386 if (codec_return != AOM_CODEC_OK) goto BailOut;
387
388 codec_return = aom_codec_control(mCodecContext, AV1E_SET_ROW_MT, 1);
389 if (codec_return != AOM_CODEC_OK) goto BailOut;
390
391 codec_return = aom_codec_control(mCodecContext, AV1E_SET_ENABLE_CDEF, 1);
392 if (codec_return != AOM_CODEC_OK) goto BailOut;
393
394 codec_return = aom_codec_control(mCodecContext, AV1E_SET_ENABLE_TPL_MODEL, 0);
395 if (codec_return != AOM_CODEC_OK) goto BailOut;
396
397 codec_return = aom_codec_control(mCodecContext, AV1E_SET_DELTAQ_MODE, 0);
398 if (codec_return != AOM_CODEC_OK) goto BailOut;
399
400 codec_return = aom_codec_control(mCodecContext, AV1E_SET_ENABLE_ORDER_HINT, 0);
401 if (codec_return != AOM_CODEC_OK) goto BailOut;
402
403 codec_return = aom_codec_control(mCodecContext, AV1E_SET_AQ_MODE, 3);
404 if (codec_return != AOM_CODEC_OK) goto BailOut;
405
406 codec_return = aom_codec_control(mCodecContext, AV1E_SET_COEFF_COST_UPD_FREQ, 3);
407 if (codec_return != AOM_CODEC_OK) goto BailOut;
408
409 codec_return = aom_codec_control(mCodecContext, AV1E_SET_MODE_COST_UPD_FREQ, 3);
410 if (codec_return != AOM_CODEC_OK) goto BailOut;
411
412 codec_return = aom_codec_control(mCodecContext, AV1E_SET_MV_COST_UPD_FREQ, 3);
413 if (codec_return != AOM_CODEC_OK) goto BailOut;
414
415 codec_return = aom_codec_control(mCodecContext, AV1E_SET_ENABLE_PALETTE, 0);
416 if (codec_return != AOM_CODEC_OK) goto BailOut;
417
418 codec_return = aom_codec_control(mCodecContext, AV1E_SET_ENABLE_OBMC, 0);
419 if (codec_return != AOM_CODEC_OK) goto BailOut;
420
421 codec_return = aom_codec_control(mCodecContext, AV1E_SET_NOISE_SENSITIVITY, 0);
422 if (codec_return != AOM_CODEC_OK) goto BailOut;
423
424 codec_return = aom_codec_control(mCodecContext, AV1E_SET_ENABLE_WARPED_MOTION, 0);
425 if (codec_return != AOM_CODEC_OK) goto BailOut;
426
427 codec_return = aom_codec_control(mCodecContext, AV1E_SET_ENABLE_GLOBAL_MOTION, 0);
428 if (codec_return != AOM_CODEC_OK) goto BailOut;
429
430 codec_return = aom_codec_control(mCodecContext, AV1E_SET_ENABLE_REF_FRAME_MVS, 0);
431 if (codec_return != AOM_CODEC_OK) goto BailOut;
432
433 codec_return = aom_codec_control(mCodecContext, AV1E_SET_ENABLE_CFL_INTRA, 0);
434 if (codec_return != AOM_CODEC_OK) goto BailOut;
435
436 codec_return = aom_codec_control(mCodecContext, AV1E_SET_ENABLE_SMOOTH_INTRA, 0);
437 if (codec_return != AOM_CODEC_OK) goto BailOut;
438 codec_return = aom_codec_control(mCodecContext, AV1E_SET_ENABLE_ANGLE_DELTA, 0);
439 if (codec_return != AOM_CODEC_OK) goto BailOut;
440 codec_return = aom_codec_control(mCodecContext, AV1E_SET_ENABLE_FILTER_INTRA, 0);
441 if (codec_return != AOM_CODEC_OK) goto BailOut;
442
443 codec_return = aom_codec_control(mCodecContext, AV1E_SET_INTRA_DEFAULT_TX_ONLY, 1);
444 if (codec_return != AOM_CODEC_OK) goto BailOut;
445 codec_return = aom_codec_control(mCodecContext, AV1E_SET_DISABLE_TRELLIS_QUANT, 1);
446 if (codec_return != AOM_CODEC_OK) goto BailOut;
447
448 codec_return = aom_codec_control(mCodecContext, AV1E_SET_ENABLE_DIST_WTD_COMP, 0);
449 if (codec_return != AOM_CODEC_OK) goto BailOut;
450 codec_return = aom_codec_control(mCodecContext, AV1E_SET_ENABLE_DIFF_WTD_COMP, 0);
451 if (codec_return != AOM_CODEC_OK) goto BailOut;
452 codec_return = aom_codec_control(mCodecContext, AV1E_SET_ENABLE_DUAL_FILTER, 0);
453 if (codec_return != AOM_CODEC_OK) goto BailOut;
454 codec_return = aom_codec_control(mCodecContext, AV1E_SET_ENABLE_INTERINTRA_COMP, 0);
455 if (codec_return != AOM_CODEC_OK) goto BailOut;
456 codec_return = aom_codec_control(mCodecContext, AV1E_SET_ENABLE_INTERINTRA_WEDGE, 0);
457 if (codec_return != AOM_CODEC_OK) goto BailOut;
458 codec_return = aom_codec_control(mCodecContext, AV1E_SET_ENABLE_INTRA_EDGE_FILTER, 0);
459 if (codec_return != AOM_CODEC_OK) goto BailOut;
460 codec_return = aom_codec_control(mCodecContext, AV1E_SET_ENABLE_INTRABC, 0);
461 if (codec_return != AOM_CODEC_OK) goto BailOut;
462 codec_return = aom_codec_control(mCodecContext, AV1E_SET_ENABLE_MASKED_COMP, 0);
463 if (codec_return != AOM_CODEC_OK) goto BailOut;
464 codec_return = aom_codec_control(mCodecContext, AV1E_SET_ENABLE_PAETH_INTRA, 0);
465 if (codec_return != AOM_CODEC_OK) goto BailOut;
466 codec_return = aom_codec_control(mCodecContext, AV1E_SET_ENABLE_QM, 0);
467 if (codec_return != AOM_CODEC_OK) goto BailOut;
468 codec_return = aom_codec_control(mCodecContext, AV1E_SET_ENABLE_RECT_PARTITIONS, 0);
469 if (codec_return != AOM_CODEC_OK) goto BailOut;
470 codec_return = aom_codec_control(mCodecContext, AV1E_SET_ENABLE_RESTORATION, 0);
471 if (codec_return != AOM_CODEC_OK) goto BailOut;
472 codec_return = aom_codec_control(mCodecContext, AV1E_SET_ENABLE_SMOOTH_INTERINTRA, 0);
473 if (codec_return != AOM_CODEC_OK) goto BailOut;
474 codec_return = aom_codec_control(mCodecContext, AV1E_SET_ENABLE_TX64, 0);
475 if (codec_return != AOM_CODEC_OK) goto BailOut;
476
477 codec_return = aom_codec_control(mCodecContext, AV1E_SET_MAX_REFERENCE_FRAMES, 3);
478 if (codec_return != AOM_CODEC_OK) goto BailOut;
479
Fyodor Kyslov58d8dea2023-02-10 02:15:54 +0000480 if (mBitrateControlMode == AOM_Q) {
481 const int aomCQLevel = MapC2QualityToAOMQuality(mQuality->value);
482 ALOGV("Set Q from %d to CQL %d",
483 mQuality->value, aomCQLevel);
484
485 codec_return = aom_codec_control(mCodecContext, AOME_SET_CQ_LEVEL, aomCQLevel);
486 if (codec_return != AOM_CODEC_OK) goto BailOut;
487 }
488
Aayush Soni643b9182023-01-18 20:23:56 +0530489 ColorAspects sfAspects;
490 if (!C2Mapper::map(mColorAspects->primaries, &sfAspects.mPrimaries)) {
491 sfAspects.mPrimaries = android::ColorAspects::PrimariesUnspecified;
492 }
493 if (!C2Mapper::map(mColorAspects->range, &sfAspects.mRange)) {
494 sfAspects.mRange = android::ColorAspects::RangeUnspecified;
495 }
496 if (!C2Mapper::map(mColorAspects->matrix, &sfAspects.mMatrixCoeffs)) {
497 sfAspects.mMatrixCoeffs = android::ColorAspects::MatrixUnspecified;
498 }
499 if (!C2Mapper::map(mColorAspects->transfer, &sfAspects.mTransfer)) {
500 sfAspects.mTransfer = android::ColorAspects::TransferUnspecified;
501 }
502 int32_t primaries, transfer, matrixCoeffs;
503 bool range;
504 ColorUtils::convertCodecColorAspectsToIsoAspects(sfAspects,
505 &primaries,
506 &transfer,
507 &matrixCoeffs,
508 &range);
509
510 codec_return = aom_codec_control(mCodecContext, AV1E_SET_COLOR_RANGE, range);
511 if (codec_return != AOM_CODEC_OK) goto BailOut;
512 codec_return = aom_codec_control(mCodecContext, AV1E_SET_COLOR_PRIMARIES, primaries);
513 if (codec_return != AOM_CODEC_OK) goto BailOut;
514 codec_return = aom_codec_control(mCodecContext, AV1E_SET_TRANSFER_CHARACTERISTICS, transfer);
515 if (codec_return != AOM_CODEC_OK) goto BailOut;
516 codec_return = aom_codec_control(mCodecContext, AV1E_SET_MATRIX_COEFFICIENTS, matrixCoeffs);
517 if (codec_return != AOM_CODEC_OK) goto BailOut;
518
Fyodor Kyslovadc71142022-09-15 16:47:03 +0000519BailOut:
520 return codec_return;
521}
522
523status_t C2SoftAomEnc::initEncoder() {
524 aom_codec_err_t codec_return;
525 status_t result = UNKNOWN_ERROR;
526 {
527 IntfImpl::Lock lock = mIntf->lock();
528 // Fetch config
529 mSize = mIntf->getSize_l();
530 mBitrate = mIntf->getBitrate_l();
531 mBitrateMode = mIntf->getBitrateMode_l();
532 mFrameRate = mIntf->getFrameRate_l();
533 mIntraRefresh = mIntf->getIntraRefresh_l();
534 mRequestSync = mIntf->getRequestSync_l();
Aayush Soni643b9182023-01-18 20:23:56 +0530535 mColorAspects = mIntf->getCodedColorAspects_l();
Fyodor Kyslov58d8dea2023-02-10 02:15:54 +0000536 mQuality = mIntf->getQuality_l();
Fyodor Kyslov3e3a92b2023-03-15 00:39:36 +0000537 mComplexity = mIntf->getComplexity_l();
Fyodor Kyslovadc71142022-09-15 16:47:03 +0000538 }
539
Fyodor Kyslovd97e9912022-11-07 19:23:21 +0000540
Fyodor Kyslovadc71142022-09-15 16:47:03 +0000541 switch (mBitrateMode->value) {
542 case C2Config::BITRATE_CONST:
543 mBitrateControlMode = AOM_CBR;
544 break;
Fyodor Kyslov58d8dea2023-02-10 02:15:54 +0000545 case C2Config::BITRATE_IGNORE:
546 mBitrateControlMode = AOM_Q;
547 break;
Fyodor Kyslovadc71142022-09-15 16:47:03 +0000548 case C2Config::BITRATE_VARIABLE:
549 [[fallthrough]];
550 default:
551 mBitrateControlMode = AOM_VBR;
552 break;
553 }
554
555 mCodecInterface = aom_codec_av1_cx();
556 if (!mCodecInterface) goto CleanUp;
557
Fyodor Kyslov3e3a92b2023-03-15 00:39:36 +0000558 ALOGD("AOM: initEncoder. BRMode: %u. KF: %u. QP: %u - %u, 10Bit: %d, comlexity %d",
Fyodor Kyslovd97e9912022-11-07 19:23:21 +0000559 (uint32_t)mBitrateControlMode,
Fyodor Kyslov3e3a92b2023-03-15 00:39:36 +0000560 mIntf->getSyncFramePeriod(), mMinQuantizer, mMaxQuantizer, mIs10Bit, mComplexity->value);
Fyodor Kyslovadc71142022-09-15 16:47:03 +0000561
562 mCodecConfiguration = new aom_codec_enc_cfg_t;
563 if (!mCodecConfiguration) goto CleanUp;
564
565 codec_return = aom_codec_enc_config_default(mCodecInterface, mCodecConfiguration,
566 AOM_USAGE_REALTIME); // RT mode
567 if (codec_return != AOM_CODEC_OK) {
568 ALOGE("Error populating default configuration for aom encoder.");
569 goto CleanUp;
570 }
571
572 mCodecConfiguration->g_w = mSize->width;
573 mCodecConfiguration->g_h = mSize->height;
Fyodor Kyslovd97e9912022-11-07 19:23:21 +0000574 mCodecConfiguration->g_bit_depth = mIs10Bit ? AOM_BITS_10 : AOM_BITS_8;
575 mCodecConfiguration->g_input_bit_depth = mIs10Bit ? 10 : 8;
576
Fyodor Kyslovadc71142022-09-15 16:47:03 +0000577
578 mCodecConfiguration->g_threads = 0;
579 mCodecConfiguration->g_error_resilient = 0;
580
581 // timebase unit is microsecond
582 // g_timebase is in seconds (i.e. 1/1000000 seconds)
583 mCodecConfiguration->g_timebase.num = 1;
584 mCodecConfiguration->g_timebase.den = 1000000;
585 // rc_target_bitrate is in kbps, mBitrate in bps
586 mCodecConfiguration->rc_target_bitrate = (mBitrate->value + 500) / 1000;
Fyodor Kyslov58d8dea2023-02-10 02:15:54 +0000587 mCodecConfiguration->rc_end_usage = mBitrateControlMode == AOM_Q ? AOM_Q : AOM_CBR;
Fyodor Kyslovadc71142022-09-15 16:47:03 +0000588 // Disable frame drop - not allowed in MediaCodec now.
589 mCodecConfiguration->rc_dropframe_thresh = 0;
590 // Disable lagged encoding.
591 mCodecConfiguration->g_lag_in_frames = 0;
592
593 // Disable spatial resizing.
594 mCodecConfiguration->rc_resize_mode = 0;
595 // Single-pass mode.
596 mCodecConfiguration->g_pass = AOM_RC_ONE_PASS;
597
598 // Maximum key frame interval - for CBR boost to 3000
599 mCodecConfiguration->kf_max_dist = 3000;
600 // Encoder determines optimal key frame placement automatically.
601 mCodecConfiguration->kf_mode = AOM_KF_AUTO;
Fyodor Kyslovadc71142022-09-15 16:47:03 +0000602 // The amount of data that may be buffered by the decoding
603 // application in ms.
604 mCodecConfiguration->rc_buf_sz = 1000;
605
606 if (mBitrateControlMode == AOM_CBR) {
Fyodor Kyslov90d99d62023-05-02 23:28:45 +0000607 // Initial value of the buffer level in ms.
608 mCodecConfiguration->rc_buf_initial_sz = 500;
609 // Amount of data that the encoder should try to maintain in ms.
610 mCodecConfiguration->rc_buf_optimal_sz = 600;
Fyodor Kyslovadc71142022-09-15 16:47:03 +0000611 // Maximum amount of bits that can be subtracted from the target
612 // bitrate - expressed as percentage of the target bitrate.
613 mCodecConfiguration->rc_undershoot_pct = 100;
614 // Maximum amount of bits that can be added to the target
615 // bitrate - expressed as percentage of the target bitrate.
616 mCodecConfiguration->rc_overshoot_pct = 10;
617 } else {
618 // Maximum amount of bits that can be subtracted from the target
619 // bitrate - expressed as percentage of the target bitrate.
620 mCodecConfiguration->rc_undershoot_pct = 100;
621 // Maximum amount of bits that can be added to the target
622 // bitrate - expressed as percentage of the target bitrate.
Fyodor Kyslov90d99d62023-05-02 23:28:45 +0000623 mCodecConfiguration->rc_overshoot_pct = 100;
Fyodor Kyslovadc71142022-09-15 16:47:03 +0000624 }
625
626 if (mIntf->getSyncFramePeriod() >= 0) {
627 mCodecConfiguration->kf_max_dist = mIntf->getSyncFramePeriod();
628 mCodecConfiguration->kf_min_dist = mIntf->getSyncFramePeriod();
629 mCodecConfiguration->kf_mode = AOM_KF_AUTO;
630 }
631 if (mMinQuantizer > 0) {
632 mCodecConfiguration->rc_min_quantizer = mMinQuantizer;
633 }
634 if (mMaxQuantizer > 0) {
635 mCodecConfiguration->rc_max_quantizer = mMaxQuantizer;
Fyodor Kyslov90d99d62023-05-02 23:28:45 +0000636 } else {
637 if (mBitrateControlMode == AOM_VBR) {
638 // For VBR we are limiting MaxQP to 52 (down 11 steps) to maintain quality
639 // 52 comes from experiments done on libaom standalone app
640 mCodecConfiguration->rc_max_quantizer = 52;
641 }
Fyodor Kyslovadc71142022-09-15 16:47:03 +0000642 }
643
644 mCodecContext = new aom_codec_ctx_t;
645 if (!mCodecContext) goto CleanUp;
646 codec_return = aom_codec_enc_init(mCodecContext, mCodecInterface, mCodecConfiguration,
Fyodor Kyslovd97e9912022-11-07 19:23:21 +0000647 mIs10Bit ? AOM_CODEC_USE_HIGHBITDEPTH : 0);
Fyodor Kyslovadc71142022-09-15 16:47:03 +0000648 if (codec_return != AOM_CODEC_OK) {
649 ALOGE("Error initializing aom encoder");
650 goto CleanUp;
651 }
652
653 codec_return = setupCodecParameters();
654 if (codec_return != AOM_CODEC_OK) {
655 ALOGE("Error setting up codec parameters");
656 goto CleanUp;
657 }
658
659 mHeadersReceived = false;
660
661 {
662 uint32_t width = mSize->width;
663 uint32_t height = mSize->height;
664 if (((uint64_t)width * height) > ((uint64_t)INT32_MAX / 3)) {
665 ALOGE("b/25812794, Buffer size is too big, width=%u, height=%u.", width, height);
666 } else {
667 uint32_t stride = (width + mStrideAlign - 1) & ~(mStrideAlign - 1);
668 uint32_t vstride = (height + mStrideAlign - 1) & ~(mStrideAlign - 1);
Fyodor Kyslovd97e9912022-11-07 19:23:21 +0000669 mConversionBuffer = MemoryBlock::Allocate(stride * vstride * 3 / (mIs10Bit? 1 : 2));
Fyodor Kyslovadc71142022-09-15 16:47:03 +0000670 if (!mConversionBuffer.size()) {
671 ALOGE("Allocating conversion buffer failed.");
672 } else {
673 mNumInputFrames = -1;
674 return OK;
675 }
676 }
677 }
678
679CleanUp:
680 onRelease();
681 return result;
682}
683
684void C2SoftAomEnc::process(const std::unique_ptr<C2Work>& work,
685 const std::shared_ptr<C2BlockPool>& pool) {
686 // Initialize output work
687 work->result = C2_OK;
688 work->workletsProcessed = 1u;
689 work->worklets.front()->output.flags = work->input.flags;
690
691 if (mSignalledError || mSignalledOutputEos) {
692 work->result = C2_BAD_VALUE;
693 return;
694 }
Fyodor Kyslovd97e9912022-11-07 19:23:21 +0000695
Suyog Pawar0ed520b2023-05-12 14:16:24 +0000696 std::shared_ptr<C2GraphicView> rView;
Fyodor Kyslovd97e9912022-11-07 19:23:21 +0000697 std::shared_ptr<C2Buffer> inputBuffer;
698 if (!work->input.buffers.empty()) {
699 inputBuffer = work->input.buffers[0];
Suyog Pawar0ed520b2023-05-12 14:16:24 +0000700 rView = std::make_shared<C2GraphicView>(
Fyodor Kyslovd97e9912022-11-07 19:23:21 +0000701 inputBuffer->data().graphicBlocks().front().map().get());
702 if (rView->error() != C2_OK) {
703 ALOGE("graphic view map err = %d", rView->error());
704 work->result = C2_CORRUPTED;
705 return;
706 }
707 } else {
708 ALOGV("Empty input Buffer");
709 uint32_t flags = 0;
710 if (work->input.flags & C2FrameData::FLAG_END_OF_STREAM) {
711 flags |= C2FrameData::FLAG_END_OF_STREAM;
712 }
713 work->worklets.front()->output.flags = (C2FrameData::flags_t)flags;
714 work->worklets.front()->output.buffers.clear();
715 work->worklets.front()->output.ordinal = work->input.ordinal;
716 work->workletsProcessed = 1u;
717 return;
718 }
719
720 bool end_of_stream = ((work->input.flags & C2FrameData::FLAG_END_OF_STREAM) != 0);
721 aom_image_t raw_frame;
722 const C2PlanarLayout& layout = rView->layout();
723 if (!mHeadersReceived) {
724 mIs10Bit = (layout.planes[layout.PLANE_Y].bitDepth == 10);
725
726 // Re-Initialize encoder
727 if (mCodecContext){
728 onRelease();
729 }
730 }
Fyodor Kyslovadc71142022-09-15 16:47:03 +0000731 if (!mCodecContext && OK != initEncoder()) {
732 ALOGE("Failed to initialize encoder");
733 mSignalledError = true;
734 work->result = C2_CORRUPTED;
735 return;
736 }
737
Suyog Pawar0ed520b2023-05-12 14:16:24 +0000738 //(b/279387842)
739 //workaround for incorrect crop size in view when using surface mode
740 rView->setCrop_be(C2Rect(mSize->width, mSize->height));
741
Fyodor Kyslovadc71142022-09-15 16:47:03 +0000742 if (!mHeadersReceived) {
743 Av1Config av1_config;
744 constexpr uint32_t header_length = 2048;
745 uint8_t header[header_length];
746 size_t header_bytes;
747 aom_fixed_buf_t* obu_sequence_header = aom_codec_get_global_headers(mCodecContext);
748 int ret = 1;
749 if (obu_sequence_header) {
750 if (get_av1config_from_obu(reinterpret_cast<const uint8_t*>(obu_sequence_header->buf),
751 obu_sequence_header->sz, false, &av1_config) == 0) {
752 ret = write_av1config(&av1_config, header_length, &header_bytes, header);
753
754 } else {
755 ALOGE("Can not get config");
756 }
757 free(obu_sequence_header->buf);
758 free(obu_sequence_header);
759 }
760
761 if (ret) {
762 ALOGE("Can not write config");
763 mSignalledError = true;
764 work->result = C2_NO_MEMORY;
765 work->workletsProcessed = 1u;
766 return;
767 }
768
769 mHeadersReceived = true;
770 std::unique_ptr<C2StreamInitDataInfo::output> csd =
771 C2StreamInitDataInfo::output::AllocUnique(header_bytes, 0u);
772 if (!csd) {
773 ALOGE("CSD allocation failed");
774 mSignalledError = true;
775 work->result = C2_NO_MEMORY;
776 work->workletsProcessed = 1u;
777 return;
778 }
779 memcpy(csd->m.value, header, header_bytes);
780 work->worklets.front()->output.configUpdate.push_back(std::move(csd));
781 ALOGV("CSD Produced of size %zu bytes", header_bytes);
782 }
783
Fyodor Kyslovadc71142022-09-15 16:47:03 +0000784 const C2ConstGraphicBlock inBuffer = inputBuffer->data().graphicBlocks().front();
785 if (inBuffer.width() < mSize->width || inBuffer.height() < mSize->height) {
786 ALOGE("unexpected Input buffer attributes %d(%d) x %d(%d)", inBuffer.width(), mSize->width,
787 inBuffer.height(), mSize->height);
788 mSignalledError = true;
789 work->result = C2_BAD_VALUE;
790 return;
791 }
Fyodor Kyslovd97e9912022-11-07 19:23:21 +0000792
793
Fyodor Kyslovadc71142022-09-15 16:47:03 +0000794 uint32_t width = mSize->width;
795 uint32_t height = mSize->height;
796 if (width > 0x8000 || height > 0x8000) {
797 ALOGE("Image too big: %u x %u", width, height);
798 work->result = C2_BAD_VALUE;
799 return;
800 }
801 uint32_t stride = (width + mStrideAlign - 1) & ~(mStrideAlign - 1);
802 uint32_t vstride = (height + mStrideAlign - 1) & ~(mStrideAlign - 1);
803 switch (layout.type) {
804 case C2PlanarLayout::TYPE_RGB:
805 case C2PlanarLayout::TYPE_RGBA: {
806 std::shared_ptr<C2StreamColorAspectsInfo::output> colorAspects;
807 {
808 IntfImpl::Lock lock = mIntf->lock();
809 colorAspects = mIntf->getCodedColorAspects_l();
810 }
811 ConvertRGBToPlanarYUV(mConversionBuffer.data(), stride, vstride,
812 mConversionBuffer.size(), *rView.get(), colorAspects->matrix,
813 colorAspects->range);
814 aom_img_wrap(&raw_frame, AOM_IMG_FMT_I420, width, height, mStrideAlign,
815 mConversionBuffer.data());
816 break;
817 }
818 case C2PlanarLayout::TYPE_YUV: {
Fyodor Kyslovd97e9912022-11-07 19:23:21 +0000819 const bool isYUV420_10bit = IsYUV420_10bit(*rView);
820 if (!IsYUV420(*rView) && !isYUV420_10bit) {
Fyodor Kyslovadc71142022-09-15 16:47:03 +0000821 ALOGE("input is not YUV420");
822 work->result = C2_BAD_VALUE;
823 return;
824 }
Fyodor Kyslovd97e9912022-11-07 19:23:21 +0000825 if (!isYUV420_10bit) {
826 if (IsI420(*rView)) {
827 // I420 compatible - though with custom offset and stride
828 aom_img_wrap(&raw_frame, AOM_IMG_FMT_I420, width, height, mStrideAlign,
829 (uint8_t*)rView->data()[0]);
830 raw_frame.planes[1] = (uint8_t*)rView->data()[1];
831 raw_frame.planes[2] = (uint8_t*)rView->data()[2];
832 raw_frame.stride[0] = layout.planes[layout.PLANE_Y].rowInc;
833 raw_frame.stride[1] = layout.planes[layout.PLANE_U].rowInc;
834 raw_frame.stride[2] = layout.planes[layout.PLANE_V].rowInc;
835 } else {
836 // TODO(kyslov): Add image wrap for NV12
837 // copy to I420
838 MediaImage2 img = CreateYUV420PlanarMediaImage2(width, height, stride, vstride);
839 if (mConversionBuffer.size() >= stride * vstride * 3 / 2) {
840 status_t err = ImageCopy(mConversionBuffer.data(), &img, *rView);
841 if (err != OK) {
842 ALOGE("Buffer conversion failed: %d", err);
843 work->result = C2_BAD_VALUE;
844 return;
845 }
846 aom_img_wrap(&raw_frame, AOM_IMG_FMT_I420, stride, vstride, mStrideAlign,
847 mConversionBuffer.data());
848 aom_img_set_rect(&raw_frame, 0, 0, width, height, 0);
849 } else {
850 ALOGE("Conversion buffer is too small: %u x %u for %zu", stride, vstride,
851 mConversionBuffer.size());
Fyodor Kyslovadc71142022-09-15 16:47:03 +0000852 work->result = C2_BAD_VALUE;
853 return;
854 }
Fyodor Kyslovd97e9912022-11-07 19:23:21 +0000855 }
856 } else { // 10 bits
857 if (IsP010(*rView)) {
858 if (mConversionBuffer.size() >= stride * vstride * 3) {
859 uint16_t *dstY, *dstU, *dstV;
860 dstY = (uint16_t*)mConversionBuffer.data();
Aayush Soni698e2f22023-02-18 01:10:53 +0530861 dstU = dstY + stride * vstride;
862 dstV = dstU + (stride * vstride) / 4;
Fyodor Kyslovd97e9912022-11-07 19:23:21 +0000863 convertP010ToYUV420Planar16(dstY, dstU, dstV, (uint16_t*)(rView->data()[0]),
Aayush Soni698e2f22023-02-18 01:10:53 +0530864 (uint16_t*)(rView->data()[1]),
865 layout.planes[layout.PLANE_Y].rowInc / 2,
866 layout.planes[layout.PLANE_U].rowInc / 2,
Fyodor Kyslovd97e9912022-11-07 19:23:21 +0000867 stride, stride / 2, stride / 2, stride,
868 vstride);
869 aom_img_wrap(&raw_frame, AOM_IMG_FMT_I42016, stride, vstride, mStrideAlign,
870 mConversionBuffer.data());
871 aom_img_set_rect(&raw_frame, 0, 0, width, height, 0);
872 } else {
873 ALOGE("Conversion buffer is too small: %u x %u for %zu", stride, vstride,
874 mConversionBuffer.size());
875 work->result = C2_BAD_VALUE;
876 return;
877 }
Fyodor Kyslovadc71142022-09-15 16:47:03 +0000878 } else {
Fyodor Kyslovd97e9912022-11-07 19:23:21 +0000879 ALOGE("Image format conversion is not supported.");
Fyodor Kyslovadc71142022-09-15 16:47:03 +0000880 work->result = C2_BAD_VALUE;
881 return;
882 }
883 }
884 break;
885 }
Aayush Soni5e82a6c2023-02-27 18:34:23 +0530886 case C2PlanarLayout::TYPE_YUVA: {
887 if (mConversionBuffer.size() >= stride * vstride * 3) {
888 uint16_t *dstY, *dstU, *dstV;
889 dstY = (uint16_t*)mConversionBuffer.data();
890 dstU = dstY + stride * vstride;
891 dstV = dstU + (stride * vstride) / 4;
892 convertRGBA1010102ToYUV420Planar16(dstY, dstU, dstV, (uint32_t*)(rView->data()[0]),
893 layout.planes[layout.PLANE_Y].rowInc / 4, stride,
894 vstride, mColorAspects->matrix,
895 mColorAspects->range);
896 aom_img_wrap(&raw_frame, AOM_IMG_FMT_I42016, stride, vstride, mStrideAlign,
897 mConversionBuffer.data());
898 aom_img_set_rect(&raw_frame, 0, 0, width, height, 0);
899 } else {
900 ALOGE("Conversion buffer is too small: %u x %u for %zu", stride, vstride,
901 mConversionBuffer.size());
902 work->result = C2_BAD_VALUE;
903 return;
904 }
905 break;
906 }
907
Fyodor Kyslovadc71142022-09-15 16:47:03 +0000908 default:
909 ALOGE("Unrecognized plane type: %d", layout.type);
910 work->result = C2_BAD_VALUE;
911 return;
912 }
913
914 aom_enc_frame_flags_t flags = 0;
915 // handle dynamic config parameters
916 {
917 IntfImpl::Lock lock = mIntf->lock();
918 std::shared_ptr<C2StreamIntraRefreshTuning::output> intraRefresh =
919 mIntf->getIntraRefresh_l();
920 std::shared_ptr<C2StreamBitrateInfo::output> bitrate = mIntf->getBitrate_l();
921 std::shared_ptr<C2StreamRequestSyncFrameTuning::output> requestSync =
922 mIntf->getRequestSync_l();
923 lock.unlock();
924
925 if (intraRefresh != mIntraRefresh) {
926 mIntraRefresh = intraRefresh;
927 ALOGV("Got mIntraRefresh request");
928 }
929
930 if (requestSync != mRequestSync) {
931 // we can handle IDR immediately
932 if (requestSync->value) {
933 // unset request
934 C2StreamRequestSyncFrameTuning::output clearSync(0u, C2_FALSE);
935 std::vector<std::unique_ptr<C2SettingResult>> failures;
936 mIntf->config({&clearSync}, C2_MAY_BLOCK, &failures);
937 ALOGV("Got sync request");
938 flags |= AOM_EFLAG_FORCE_KF;
939 }
940 mRequestSync = requestSync;
941 }
942
943 if (bitrate != mBitrate) {
944 mBitrate = bitrate;
945 mCodecConfiguration->rc_target_bitrate = (mBitrate->value + 500) / 1000;
946 aom_codec_err_t res = aom_codec_enc_config_set(mCodecContext, mCodecConfiguration);
947 if (res != AOM_CODEC_OK) {
948 ALOGE("aom encoder failed to update bitrate: %s", aom_codec_err_to_string(res));
949 mSignalledError = true;
950 work->result = C2_CORRUPTED;
951 return;
952 }
953 }
954 }
955
956 uint64_t input_timestamp = work->input.ordinal.timestamp.peekull();
957 uint32_t frame_duration;
958 if (input_timestamp > mLastTimestamp) {
959 frame_duration = (uint32_t)(input_timestamp - mLastTimestamp);
960 } else {
961 // Use default of 30 fps in case of 0 frame rate.
962 float frame_rate = mFrameRate->value;
963 if (frame_rate < 0.001) {
964 frame_rate = 30.0;
965 }
966 frame_duration = (uint32_t)(1000000 / frame_rate + 0.5);
967 }
968 mLastTimestamp = input_timestamp;
969
970 aom_codec_err_t codec_return =
971 aom_codec_encode(mCodecContext, &raw_frame, input_timestamp, frame_duration, flags);
972 if (codec_return != AOM_CODEC_OK) {
973 ALOGE("aom encoder failed to encode frame");
974 mSignalledError = true;
975 work->result = C2_CORRUPTED;
976 return;
977 }
978
979 bool populated = false;
980 aom_codec_iter_t encoded_packet_iterator = nullptr;
981 const aom_codec_cx_pkt_t* encoded_packet;
982 while ((encoded_packet = aom_codec_get_cx_data(mCodecContext, &encoded_packet_iterator))) {
983 if (encoded_packet->kind == AOM_CODEC_CX_FRAME_PKT) {
984 std::shared_ptr<C2LinearBlock> block;
985 C2MemoryUsage usage = {C2MemoryUsage::CPU_READ, C2MemoryUsage::CPU_WRITE};
986 c2_status_t err = pool->fetchLinearBlock(encoded_packet->data.frame.sz, usage, &block);
987 if (err != C2_OK) {
988 ALOGE("fetchLinearBlock for Output failed with status %d", err);
989 work->result = C2_NO_MEMORY;
990 return;
991 }
992 C2WriteView wView = block->map().get();
993 if (wView.error()) {
994 ALOGE("write view map failed %d", wView.error());
995 work->result = C2_CORRUPTED;
996 return;
997 }
998
999 memcpy(wView.data(), encoded_packet->data.frame.buf, encoded_packet->data.frame.sz);
1000 ++mNumInputFrames;
1001
1002 ALOGD("bytes generated %zu", encoded_packet->data.frame.sz);
1003 uint32_t flags = 0;
1004 if (end_of_stream) {
1005 flags |= C2FrameData::FLAG_END_OF_STREAM;
1006 }
1007
1008 work->worklets.front()->output.flags = (C2FrameData::flags_t)flags;
1009 work->worklets.front()->output.buffers.clear();
1010 std::shared_ptr<C2Buffer> buffer =
1011 createLinearBuffer(block, 0, encoded_packet->data.frame.sz);
1012 if (encoded_packet->data.frame.flags & AOM_FRAME_IS_KEY) {
1013 buffer->setInfo(std::make_shared<C2StreamPictureTypeMaskInfo::output>(
1014 0u /* stream id */, C2Config::SYNC_FRAME));
1015 }
1016 work->worklets.front()->output.buffers.push_back(buffer);
1017 work->worklets.front()->output.ordinal = work->input.ordinal;
1018 work->worklets.front()->output.ordinal.timestamp = encoded_packet->data.frame.pts;
1019 work->workletsProcessed = 1u;
1020 populated = true;
1021 if (end_of_stream) {
1022 mSignalledOutputEos = true;
1023 ALOGV("signalled End Of Stream");
1024 }
1025 }
1026 }
1027 if (!populated) {
1028 work->workletsProcessed = 0u;
1029 }
1030}
1031
1032c2_status_t C2SoftAomEnc::drain(uint32_t drainMode, const std::shared_ptr<C2BlockPool>& pool) {
1033 (void)pool;
1034 if (drainMode == NO_DRAIN) {
1035 ALOGW("drain with NO_DRAIN: no-op");
1036 return C2_OK;
1037 }
1038 if (drainMode == DRAIN_CHAIN) {
1039 ALOGW("DRAIN_CHAIN not supported");
1040 return C2_OMITTED;
1041 }
1042
1043 return C2_OK;
1044}
1045
1046class C2SoftAomEncFactory : public C2ComponentFactory {
1047 public:
1048 C2SoftAomEncFactory()
1049 : mHelper(std::static_pointer_cast<C2ReflectorHelper>(
1050 GetCodec2PlatformComponentStore()->getParamReflector())) {}
1051
1052 virtual c2_status_t createComponent(c2_node_id_t id,
1053 std::shared_ptr<C2Component>* const component,
1054 std::function<void(C2Component*)> deleter) override {
1055 *component = std::shared_ptr<C2Component>(
1056 new C2SoftAomEnc(COMPONENT_NAME, id,
1057 std::make_shared<C2SoftAomEnc::IntfImpl>(mHelper)),
1058 deleter);
1059 return C2_OK;
1060 }
1061
1062 virtual c2_status_t createInterface(
1063 c2_node_id_t id, std::shared_ptr<C2ComponentInterface>* const interface,
1064 std::function<void(C2ComponentInterface*)> deleter) override {
1065 *interface = std::shared_ptr<C2ComponentInterface>(
1066 new SimpleInterface<C2SoftAomEnc::IntfImpl>(
1067 COMPONENT_NAME, id, std::make_shared<C2SoftAomEnc::IntfImpl>(mHelper)),
1068 deleter);
1069 return C2_OK;
1070 }
1071
1072 virtual ~C2SoftAomEncFactory() override = default;
1073
1074 private:
1075 std::shared_ptr<C2ReflectorHelper> mHelper;
1076};
1077
1078} // namespace android
1079
1080__attribute__((cfi_canonical_jump_table)) extern "C" ::C2ComponentFactory* CreateCodec2Factory() {
1081 ALOGV("in %s", __func__);
1082 return new ::android::C2SoftAomEncFactory();
1083}
1084
1085__attribute__((cfi_canonical_jump_table)) extern "C" void DestroyCodec2Factory(
1086 ::C2ComponentFactory* factory) {
1087 ALOGV("in %s", __func__);
1088 delete factory;
1089}