blob: 491098dcb10c40daa96a85c75733de3050ff3df1 [file] [log] [blame]
Pawin Vongmasa36653902018-11-15 00:10:25 -08001/*
2 * Copyright (C) 2018 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 "C2SoftMpeg2Dec"
19#include <log/log.h>
20
21#include <media/stagefright/foundation/MediaDefs.h>
22
23#include <C2Debug.h>
24#include <C2PlatformSupport.h>
25#include <Codec2Mapper.h>
26#include <SimpleC2Interface.h>
27
28#include "C2SoftMpeg2Dec.h"
29#include "impeg2d.h"
30
31namespace android {
Harish Mahendrakare55c4712019-07-26 12:32:13 -070032constexpr size_t kMinInputBufferSize = 2 * 1024 * 1024;
Harish Mahendrakarfb145812021-01-25 16:11:37 -080033constexpr size_t kMaxDimension = 1920;
Pawin Vongmasa36653902018-11-15 00:10:25 -080034constexpr char COMPONENT_NAME[] = "c2.android.mpeg2.decoder";
35
36class C2SoftMpeg2Dec::IntfImpl : public SimpleInterface<void>::BaseParams {
37public:
38 explicit IntfImpl(const std::shared_ptr<C2ReflectorHelper> &helper)
39 : SimpleInterface<void>::BaseParams(
40 helper,
41 COMPONENT_NAME,
42 C2Component::KIND_DECODER,
43 C2Component::DOMAIN_VIDEO,
44 MEDIA_MIMETYPE_VIDEO_MPEG2) {
45 noPrivateBuffers(); // TODO: account for our buffers here
46 noInputReferences();
47 noOutputReferences();
48 noInputLatency();
49 noTimeStretch();
50
Harish Mahendrakar7be1bc92019-08-01 14:59:35 -070051 // TODO: Proper support for reorder depth.
52 addParameter(
53 DefineParam(mActualOutputDelay, C2_PARAMKEY_OUTPUT_DELAY)
54 .withConstValue(new C2PortActualDelayTuning::output(3u))
55 .build());
56
Pawin Vongmasa36653902018-11-15 00:10:25 -080057 // TODO: output latency and reordering
58
59 addParameter(
60 DefineParam(mAttrib, C2_PARAMKEY_COMPONENT_ATTRIBUTES)
61 .withConstValue(new C2ComponentAttributesSetting(C2Component::ATTRIB_IS_TEMPORAL))
62 .build());
63
64 addParameter(
65 DefineParam(mSize, C2_PARAMKEY_PICTURE_SIZE)
66 .withDefault(new C2StreamPictureSizeInfo::output(0u, 320, 240))
67 .withFields({
Harish Mahendrakarfb145812021-01-25 16:11:37 -080068 C2F(mSize, width).inRange(16, kMaxDimension, 2),
69 C2F(mSize, height).inRange(16, kMaxDimension, 2),
Pawin Vongmasa36653902018-11-15 00:10:25 -080070 })
71 .withSetter(SizeSetter)
72 .build());
73
74 addParameter(
75 DefineParam(mProfileLevel, C2_PARAMKEY_PROFILE_LEVEL)
76 .withDefault(new C2StreamProfileLevelInfo::input(0u,
77 C2Config::PROFILE_MP2V_SIMPLE, C2Config::LEVEL_MP2V_HIGH))
78 .withFields({
79 C2F(mProfileLevel, profile).oneOf({
80 C2Config::PROFILE_MP2V_SIMPLE,
81 C2Config::PROFILE_MP2V_MAIN}),
82 C2F(mProfileLevel, level).oneOf({
83 C2Config::LEVEL_MP2V_LOW,
84 C2Config::LEVEL_MP2V_MAIN,
85 C2Config::LEVEL_MP2V_HIGH_1440,
86 C2Config::LEVEL_MP2V_HIGH})
87 })
88 .withSetter(ProfileLevelSetter, mSize)
89 .build());
90
91 addParameter(
92 DefineParam(mMaxSize, C2_PARAMKEY_MAX_PICTURE_SIZE)
93 .withDefault(new C2StreamMaxPictureSizeTuning::output(0u, 320, 240))
94 .withFields({
Harish Mahendrakarfb145812021-01-25 16:11:37 -080095 C2F(mSize, width).inRange(2, kMaxDimension, 2),
96 C2F(mSize, height).inRange(2, kMaxDimension, 2),
Pawin Vongmasa36653902018-11-15 00:10:25 -080097 })
98 .withSetter(MaxPictureSizeSetter, mSize)
99 .build());
100
101 addParameter(
102 DefineParam(mMaxInputSize, C2_PARAMKEY_INPUT_MAX_BUFFER_SIZE)
Harish Mahendrakare55c4712019-07-26 12:32:13 -0700103 .withDefault(new C2StreamMaxBufferSizeInfo::input(0u, kMinInputBufferSize))
Pawin Vongmasa36653902018-11-15 00:10:25 -0800104 .withFields({
105 C2F(mMaxInputSize, value).any(),
106 })
107 .calculatedAs(MaxInputSizeSetter, mMaxSize)
108 .build());
109
110 C2ChromaOffsetStruct locations[1] = { C2ChromaOffsetStruct::ITU_YUV_420_0() };
111 std::shared_ptr<C2StreamColorInfo::output> defaultColorInfo =
112 C2StreamColorInfo::output::AllocShared(
113 1u, 0u, 8u /* bitDepth */, C2Color::YUV_420);
114 memcpy(defaultColorInfo->m.locations, locations, sizeof(locations));
115
116 defaultColorInfo =
117 C2StreamColorInfo::output::AllocShared(
118 { C2ChromaOffsetStruct::ITU_YUV_420_0() },
119 0u, 8u /* bitDepth */, C2Color::YUV_420);
120 helper->addStructDescriptors<C2ChromaOffsetStruct>();
121
122 addParameter(
123 DefineParam(mColorInfo, C2_PARAMKEY_CODED_COLOR_INFO)
124 .withConstValue(defaultColorInfo)
125 .build());
126
127 addParameter(
128 DefineParam(mDefaultColorAspects, C2_PARAMKEY_DEFAULT_COLOR_ASPECTS)
129 .withDefault(new C2StreamColorAspectsTuning::output(
130 0u, C2Color::RANGE_UNSPECIFIED, C2Color::PRIMARIES_UNSPECIFIED,
131 C2Color::TRANSFER_UNSPECIFIED, C2Color::MATRIX_UNSPECIFIED))
132 .withFields({
133 C2F(mDefaultColorAspects, range).inRange(
134 C2Color::RANGE_UNSPECIFIED, C2Color::RANGE_OTHER),
135 C2F(mDefaultColorAspects, primaries).inRange(
136 C2Color::PRIMARIES_UNSPECIFIED, C2Color::PRIMARIES_OTHER),
137 C2F(mDefaultColorAspects, transfer).inRange(
138 C2Color::TRANSFER_UNSPECIFIED, C2Color::TRANSFER_OTHER),
139 C2F(mDefaultColorAspects, matrix).inRange(
140 C2Color::MATRIX_UNSPECIFIED, C2Color::MATRIX_OTHER)
141 })
142 .withSetter(DefaultColorAspectsSetter)
143 .build());
144
145 addParameter(
146 DefineParam(mCodedColorAspects, C2_PARAMKEY_VUI_COLOR_ASPECTS)
147 .withDefault(new C2StreamColorAspectsInfo::input(
148 0u, C2Color::RANGE_LIMITED, C2Color::PRIMARIES_UNSPECIFIED,
149 C2Color::TRANSFER_UNSPECIFIED, C2Color::MATRIX_UNSPECIFIED))
150 .withFields({
151 C2F(mCodedColorAspects, range).inRange(
152 C2Color::RANGE_UNSPECIFIED, C2Color::RANGE_OTHER),
153 C2F(mCodedColorAspects, primaries).inRange(
154 C2Color::PRIMARIES_UNSPECIFIED, C2Color::PRIMARIES_OTHER),
155 C2F(mCodedColorAspects, transfer).inRange(
156 C2Color::TRANSFER_UNSPECIFIED, C2Color::TRANSFER_OTHER),
157 C2F(mCodedColorAspects, matrix).inRange(
158 C2Color::MATRIX_UNSPECIFIED, C2Color::MATRIX_OTHER)
159 })
160 .withSetter(CodedColorAspectsSetter)
161 .build());
162
163 addParameter(
164 DefineParam(mColorAspects, C2_PARAMKEY_COLOR_ASPECTS)
165 .withDefault(new C2StreamColorAspectsInfo::output(
166 0u, C2Color::RANGE_UNSPECIFIED, C2Color::PRIMARIES_UNSPECIFIED,
167 C2Color::TRANSFER_UNSPECIFIED, C2Color::MATRIX_UNSPECIFIED))
168 .withFields({
169 C2F(mColorAspects, range).inRange(
170 C2Color::RANGE_UNSPECIFIED, C2Color::RANGE_OTHER),
171 C2F(mColorAspects, primaries).inRange(
172 C2Color::PRIMARIES_UNSPECIFIED, C2Color::PRIMARIES_OTHER),
173 C2F(mColorAspects, transfer).inRange(
174 C2Color::TRANSFER_UNSPECIFIED, C2Color::TRANSFER_OTHER),
175 C2F(mColorAspects, matrix).inRange(
176 C2Color::MATRIX_UNSPECIFIED, C2Color::MATRIX_OTHER)
177 })
178 .withSetter(ColorAspectsSetter, mDefaultColorAspects, mCodedColorAspects)
179 .build());
180
181 // TODO: support more formats?
182 addParameter(
183 DefineParam(mPixelFormat, C2_PARAMKEY_PIXEL_FORMAT)
184 .withConstValue(new C2StreamPixelFormatInfo::output(
185 0u, HAL_PIXEL_FORMAT_YCBCR_420_888))
186 .build());
187 }
188
189 static C2R SizeSetter(bool mayBlock, const C2P<C2StreamPictureSizeInfo::output> &oldMe,
Lajos Molnar3bb81cd2019-02-20 15:10:30 -0800190 C2P<C2StreamPictureSizeInfo::output> &me) {
Pawin Vongmasa36653902018-11-15 00:10:25 -0800191 (void)mayBlock;
192 C2R res = C2R::Ok();
193 if (!me.F(me.v.width).supportsAtAll(me.v.width)) {
194 res = res.plus(C2SettingResultBuilder::BadValue(me.F(me.v.width)));
195 me.set().width = oldMe.v.width;
196 }
197 if (!me.F(me.v.height).supportsAtAll(me.v.height)) {
198 res = res.plus(C2SettingResultBuilder::BadValue(me.F(me.v.height)));
199 me.set().height = oldMe.v.height;
200 }
201 return res;
202 }
203
204 static C2R MaxPictureSizeSetter(bool mayBlock, C2P<C2StreamMaxPictureSizeTuning::output> &me,
205 const C2P<C2StreamPictureSizeInfo::output> &size) {
206 (void)mayBlock;
207 // TODO: get max width/height from the size's field helpers vs. hardcoding
Harish Mahendrakarfb145812021-01-25 16:11:37 -0800208 me.set().width = c2_min(c2_max(me.v.width, size.v.width), kMaxDimension);
209 me.set().height = c2_min(c2_max(me.v.height, size.v.height), kMaxDimension);
Pawin Vongmasa36653902018-11-15 00:10:25 -0800210 return C2R::Ok();
211 }
212
213 static C2R MaxInputSizeSetter(bool mayBlock, C2P<C2StreamMaxBufferSizeInfo::input> &me,
214 const C2P<C2StreamMaxPictureSizeTuning::output> &maxSize) {
215 (void)mayBlock;
216 // assume compression ratio of 1
Harish Mahendrakare55c4712019-07-26 12:32:13 -0700217 me.set().value = c2_max((((maxSize.v.width + 15) / 16)
218 * ((maxSize.v.height + 15) / 16) * 384), kMinInputBufferSize);
Pawin Vongmasa36653902018-11-15 00:10:25 -0800219 return C2R::Ok();
220 }
221
222 static C2R ProfileLevelSetter(bool mayBlock, C2P<C2StreamProfileLevelInfo::input> &me,
223 const C2P<C2StreamPictureSizeInfo::output> &size) {
224 (void)mayBlock;
225 (void)size;
226 (void)me; // TODO: validate
227 return C2R::Ok();
228 }
229
230 static C2R DefaultColorAspectsSetter(bool mayBlock, C2P<C2StreamColorAspectsTuning::output> &me) {
231 (void)mayBlock;
232 if (me.v.range > C2Color::RANGE_OTHER) {
233 me.set().range = C2Color::RANGE_OTHER;
234 }
235 if (me.v.primaries > C2Color::PRIMARIES_OTHER) {
236 me.set().primaries = C2Color::PRIMARIES_OTHER;
237 }
238 if (me.v.transfer > C2Color::TRANSFER_OTHER) {
239 me.set().transfer = C2Color::TRANSFER_OTHER;
240 }
241 if (me.v.matrix > C2Color::MATRIX_OTHER) {
242 me.set().matrix = C2Color::MATRIX_OTHER;
243 }
244 return C2R::Ok();
245 }
246
247 static C2R CodedColorAspectsSetter(bool mayBlock, C2P<C2StreamColorAspectsInfo::input> &me) {
248 (void)mayBlock;
249 if (me.v.range > C2Color::RANGE_OTHER) {
250 me.set().range = C2Color::RANGE_OTHER;
251 }
252 if (me.v.primaries > C2Color::PRIMARIES_OTHER) {
253 me.set().primaries = C2Color::PRIMARIES_OTHER;
254 }
255 if (me.v.transfer > C2Color::TRANSFER_OTHER) {
256 me.set().transfer = C2Color::TRANSFER_OTHER;
257 }
258 if (me.v.matrix > C2Color::MATRIX_OTHER) {
259 me.set().matrix = C2Color::MATRIX_OTHER;
260 }
261 return C2R::Ok();
262 }
263
264 static C2R ColorAspectsSetter(bool mayBlock, C2P<C2StreamColorAspectsInfo::output> &me,
265 const C2P<C2StreamColorAspectsTuning::output> &def,
266 const C2P<C2StreamColorAspectsInfo::input> &coded) {
267 (void)mayBlock;
268 // take default values for all unspecified fields, and coded values for specified ones
269 me.set().range = coded.v.range == RANGE_UNSPECIFIED ? def.v.range : coded.v.range;
270 me.set().primaries = coded.v.primaries == PRIMARIES_UNSPECIFIED
271 ? def.v.primaries : coded.v.primaries;
272 me.set().transfer = coded.v.transfer == TRANSFER_UNSPECIFIED
273 ? def.v.transfer : coded.v.transfer;
274 me.set().matrix = coded.v.matrix == MATRIX_UNSPECIFIED ? def.v.matrix : coded.v.matrix;
275 return C2R::Ok();
276 }
277
278 std::shared_ptr<C2StreamColorAspectsInfo::output> getColorAspects_l() {
279 return mColorAspects;
280 }
281
282private:
283 std::shared_ptr<C2StreamProfileLevelInfo::input> mProfileLevel;
284 std::shared_ptr<C2StreamPictureSizeInfo::output> mSize;
285 std::shared_ptr<C2StreamMaxPictureSizeTuning::output> mMaxSize;
286 std::shared_ptr<C2StreamMaxBufferSizeInfo::input> mMaxInputSize;
287 std::shared_ptr<C2StreamColorInfo::output> mColorInfo;
288 std::shared_ptr<C2StreamColorAspectsInfo::input> mCodedColorAspects;
289 std::shared_ptr<C2StreamColorAspectsTuning::output> mDefaultColorAspects;
290 std::shared_ptr<C2StreamColorAspectsInfo::output> mColorAspects;
291 std::shared_ptr<C2StreamPixelFormatInfo::output> mPixelFormat;
292};
293
294static size_t getCpuCoreCount() {
295 long cpuCoreCount = 1;
296#if defined(_SC_NPROCESSORS_ONLN)
297 cpuCoreCount = sysconf(_SC_NPROCESSORS_ONLN);
298#else
299 // _SC_NPROC_ONLN must be defined...
300 cpuCoreCount = sysconf(_SC_NPROC_ONLN);
301#endif
302 CHECK(cpuCoreCount >= 1);
303 ALOGV("Number of CPU cores: %ld", cpuCoreCount);
304 return (size_t)cpuCoreCount;
305}
306
307static void *ivd_aligned_malloc(WORD32 alignment, WORD32 size) {
308 return memalign(alignment, size);
309}
310
311static void ivd_aligned_free(void *mem) {
312 free(mem);
313}
314
315C2SoftMpeg2Dec::C2SoftMpeg2Dec(
316 const char *name,
317 c2_node_id_t id,
318 const std::shared_ptr<IntfImpl> &intfImpl)
319 : SimpleC2Component(std::make_shared<SimpleInterface<IntfImpl>>(name, id, intfImpl)),
320 mIntf(intfImpl),
321 mDecHandle(nullptr),
322 mMemRecords(nullptr),
323 mOutBufferDrain(nullptr),
324 mIvColorformat(IV_YUV_420P),
325 mWidth(320),
Rakesh Kumard91bc482019-02-18 10:42:41 +0530326 mHeight(240),
327 mOutIndex(0u) {
Pawin Vongmasa36653902018-11-15 00:10:25 -0800328 // If input dump is enabled, then open create an empty file
329 GENERATE_FILE_NAMES();
330 CREATE_DUMP_FILE(mInFile);
331}
332
333C2SoftMpeg2Dec::~C2SoftMpeg2Dec() {
334 onRelease();
335}
336
337c2_status_t C2SoftMpeg2Dec::onInit() {
338 status_t err = initDecoder();
339 return err == OK ? C2_OK : C2_CORRUPTED;
340}
341
342c2_status_t C2SoftMpeg2Dec::onStop() {
343 if (OK != resetDecoder()) return C2_CORRUPTED;
344 resetPlugin();
345 return C2_OK;
346}
347
348void C2SoftMpeg2Dec::onReset() {
349 (void) onStop();
350}
351
352void C2SoftMpeg2Dec::onRelease() {
353 (void) deleteDecoder();
354 if (mOutBufferDrain) {
355 ivd_aligned_free(mOutBufferDrain);
356 mOutBufferDrain = nullptr;
357 }
358 if (mOutBlock) {
359 mOutBlock.reset();
360 }
361 if (mMemRecords) {
362 ivd_aligned_free(mMemRecords);
363 mMemRecords = nullptr;
364 }
365}
366
367c2_status_t C2SoftMpeg2Dec::onFlush_sm() {
368 if (OK != setFlushMode()) return C2_CORRUPTED;
369
370 uint32_t displayStride = mStride;
371 uint32_t displayHeight = mHeight;
372 uint32_t bufferSize = displayStride * displayHeight * 3 / 2;
373 mOutBufferDrain = (uint8_t *)ivd_aligned_malloc(128, bufferSize);
374 if (!mOutBufferDrain) {
375 ALOGE("could not allocate tmp output buffer (for flush) of size %u ", bufferSize);
376 return C2_NO_MEMORY;
377 }
378
379 while (true) {
380 ivd_video_decode_ip_t s_decode_ip;
381 ivd_video_decode_op_t s_decode_op;
382
383 setDecodeArgs(&s_decode_ip, &s_decode_op, nullptr, nullptr, 0, 0, 0);
384 (void) ivdec_api_function(mDecHandle, &s_decode_ip, &s_decode_op);
385 if (0 == s_decode_op.u4_output_present) {
386 resetPlugin();
387 break;
388 }
389 }
390
391 if (mOutBufferDrain) {
392 ivd_aligned_free(mOutBufferDrain);
393 mOutBufferDrain = nullptr;
394 }
395
396 return C2_OK;
397}
398
399status_t C2SoftMpeg2Dec::getNumMemRecords() {
400 iv_num_mem_rec_ip_t s_num_mem_rec_ip;
401 iv_num_mem_rec_op_t s_num_mem_rec_op;
402
403 s_num_mem_rec_ip.u4_size = sizeof(s_num_mem_rec_ip);
404 s_num_mem_rec_ip.e_cmd = IV_CMD_GET_NUM_MEM_REC;
405 s_num_mem_rec_op.u4_size = sizeof(s_num_mem_rec_op);
406
407 IV_API_CALL_STATUS_T status = ivdec_api_function(mDecHandle,
408 &s_num_mem_rec_ip,
409 &s_num_mem_rec_op);
410 if (IV_SUCCESS != status) {
411 ALOGE("Error in getting mem records: 0x%x", s_num_mem_rec_op.u4_error_code);
412 return UNKNOWN_ERROR;
413 }
414 mNumMemRecords = s_num_mem_rec_op.u4_num_mem_rec;
415
416 return OK;
417}
418
419status_t C2SoftMpeg2Dec::fillMemRecords() {
420 iv_mem_rec_t *ps_mem_rec = (iv_mem_rec_t *) ivd_aligned_malloc(
421 128, mNumMemRecords * sizeof(iv_mem_rec_t));
422 if (!ps_mem_rec) {
423 ALOGE("Allocation failure");
424 return NO_MEMORY;
425 }
426 memset(ps_mem_rec, 0, mNumMemRecords * sizeof(iv_mem_rec_t));
427 for (size_t i = 0; i < mNumMemRecords; i++)
428 ps_mem_rec[i].u4_size = sizeof(iv_mem_rec_t);
429 mMemRecords = ps_mem_rec;
430
431 ivdext_fill_mem_rec_ip_t s_fill_mem_ip;
432 ivdext_fill_mem_rec_op_t s_fill_mem_op;
433
434 s_fill_mem_ip.s_ivd_fill_mem_rec_ip_t.u4_size = sizeof(ivdext_fill_mem_rec_ip_t);
435 s_fill_mem_ip.u4_share_disp_buf = 0;
srujan vandrangi28b54452023-09-25 20:05:39 +0530436 s_fill_mem_ip.u4_keep_threads_active = 1;
Pawin Vongmasa36653902018-11-15 00:10:25 -0800437 s_fill_mem_ip.e_output_format = mIvColorformat;
438 s_fill_mem_ip.u4_deinterlace = 1;
439 s_fill_mem_ip.s_ivd_fill_mem_rec_ip_t.e_cmd = IV_CMD_FILL_NUM_MEM_REC;
440 s_fill_mem_ip.s_ivd_fill_mem_rec_ip_t.pv_mem_rec_location = mMemRecords;
441 s_fill_mem_ip.s_ivd_fill_mem_rec_ip_t.u4_max_frm_wd = mWidth;
442 s_fill_mem_ip.s_ivd_fill_mem_rec_ip_t.u4_max_frm_ht = mHeight;
443 s_fill_mem_op.s_ivd_fill_mem_rec_op_t.u4_size = sizeof(ivdext_fill_mem_rec_op_t);
444 IV_API_CALL_STATUS_T status = ivdec_api_function(mDecHandle,
445 &s_fill_mem_ip,
446 &s_fill_mem_op);
447 if (IV_SUCCESS != status) {
448 ALOGE("Error in filling mem records: 0x%x",
449 s_fill_mem_op.s_ivd_fill_mem_rec_op_t.u4_error_code);
450 return UNKNOWN_ERROR;
451 }
452
453 CHECK_EQ(mNumMemRecords, s_fill_mem_op.s_ivd_fill_mem_rec_op_t.u4_num_mem_rec_filled);
454 for (size_t i = 0; i < mNumMemRecords; i++, ps_mem_rec++) {
455 ps_mem_rec->pv_base = ivd_aligned_malloc(
456 ps_mem_rec->u4_mem_alignment, ps_mem_rec->u4_mem_size);
457 if (!ps_mem_rec->pv_base) {
458 ALOGE("Allocation failure for memory record #%zu of size %u",
459 i, ps_mem_rec->u4_mem_size);
460 return NO_MEMORY;
461 }
462 }
463
464 return OK;
465}
466
467status_t C2SoftMpeg2Dec::createDecoder() {
468 ivdext_init_ip_t s_init_ip;
469 ivdext_init_op_t s_init_op;
470
471 s_init_ip.s_ivd_init_ip_t.u4_size = sizeof(ivdext_init_ip_t);
472 s_init_ip.s_ivd_init_ip_t.e_cmd = (IVD_API_COMMAND_TYPE_T)IV_CMD_INIT;
473 s_init_ip.s_ivd_init_ip_t.pv_mem_rec_location = mMemRecords;
474 s_init_ip.s_ivd_init_ip_t.u4_frm_max_wd = mWidth;
475 s_init_ip.s_ivd_init_ip_t.u4_frm_max_ht = mHeight;
476 s_init_ip.u4_share_disp_buf = 0;
477 s_init_ip.u4_deinterlace = 1;
srujan vandrangi28b54452023-09-25 20:05:39 +0530478 s_init_ip.u4_keep_threads_active = 1;
Pawin Vongmasa36653902018-11-15 00:10:25 -0800479 s_init_ip.s_ivd_init_ip_t.u4_num_mem_rec = mNumMemRecords;
480 s_init_ip.s_ivd_init_ip_t.e_output_format = mIvColorformat;
481 s_init_op.s_ivd_init_op_t.u4_size = sizeof(ivdext_init_op_t);
482
483 mDecHandle = (iv_obj_t *)mMemRecords[0].pv_base;
484 mDecHandle->pv_fxns = (void *)ivdec_api_function;
485 mDecHandle->u4_size = sizeof(iv_obj_t);
486
487 IV_API_CALL_STATUS_T status = ivdec_api_function(mDecHandle,
488 &s_init_ip,
489 &s_init_op);
490 if (status != IV_SUCCESS) {
491 ALOGE("error in %s: 0x%x", __func__,
492 s_init_op.s_ivd_init_op_t.u4_error_code);
493 return UNKNOWN_ERROR;
494 }
495
496 return OK;
497}
498
499status_t C2SoftMpeg2Dec::setNumCores() {
500 ivdext_ctl_set_num_cores_ip_t s_set_num_cores_ip;
501 ivdext_ctl_set_num_cores_op_t s_set_num_cores_op;
502
503 s_set_num_cores_ip.u4_size = sizeof(ivdext_ctl_set_num_cores_ip_t);
504 s_set_num_cores_ip.e_cmd = IVD_CMD_VIDEO_CTL;
505 s_set_num_cores_ip.e_sub_cmd = IVDEXT_CMD_CTL_SET_NUM_CORES;
506 s_set_num_cores_ip.u4_num_cores = mNumCores;
507 s_set_num_cores_op.u4_size = sizeof(ivdext_ctl_set_num_cores_op_t);
508 IV_API_CALL_STATUS_T status = ivdec_api_function(mDecHandle,
509 &s_set_num_cores_ip,
510 &s_set_num_cores_op);
511 if (status != IV_SUCCESS) {
512 ALOGD("error in %s: 0x%x", __func__, s_set_num_cores_op.u4_error_code);
513 return UNKNOWN_ERROR;
514 }
515
516 return OK;
517}
518
519status_t C2SoftMpeg2Dec::setParams(size_t stride) {
520 ivd_ctl_set_config_ip_t s_set_dyn_params_ip;
521 ivd_ctl_set_config_op_t s_set_dyn_params_op;
522
523 s_set_dyn_params_ip.u4_size = sizeof(ivd_ctl_set_config_ip_t);
524 s_set_dyn_params_ip.e_cmd = IVD_CMD_VIDEO_CTL;
525 s_set_dyn_params_ip.e_sub_cmd = IVD_CMD_CTL_SETPARAMS;
526 s_set_dyn_params_ip.u4_disp_wd = (UWORD32) stride;
527 s_set_dyn_params_ip.e_frm_skip_mode = IVD_SKIP_NONE;
528 s_set_dyn_params_ip.e_frm_out_mode = IVD_DISPLAY_FRAME_OUT;
529 s_set_dyn_params_ip.e_vid_dec_mode = IVD_DECODE_FRAME;
530 s_set_dyn_params_op.u4_size = sizeof(ivd_ctl_set_config_op_t);
531 IV_API_CALL_STATUS_T status = ivdec_api_function(mDecHandle,
532 &s_set_dyn_params_ip,
533 &s_set_dyn_params_op);
534 if (status != IV_SUCCESS) {
535 ALOGE("error in %s: 0x%x", __func__, s_set_dyn_params_op.u4_error_code);
536 return UNKNOWN_ERROR;
537 }
538
539 return OK;
540}
541
542status_t C2SoftMpeg2Dec::getVersion() {
543 ivd_ctl_getversioninfo_ip_t s_get_versioninfo_ip;
544 ivd_ctl_getversioninfo_op_t s_get_versioninfo_op;
545 UWORD8 au1_buf[512];
546
547 s_get_versioninfo_ip.u4_size = sizeof(ivd_ctl_getversioninfo_ip_t);
548 s_get_versioninfo_ip.e_cmd = IVD_CMD_VIDEO_CTL;
549 s_get_versioninfo_ip.e_sub_cmd = IVD_CMD_CTL_GETVERSION;
550 s_get_versioninfo_ip.pv_version_buffer = au1_buf;
551 s_get_versioninfo_ip.u4_version_buffer_size = sizeof(au1_buf);
552 s_get_versioninfo_op.u4_size = sizeof(ivd_ctl_getversioninfo_op_t);
553 IV_API_CALL_STATUS_T status = ivdec_api_function(mDecHandle,
554 &s_get_versioninfo_ip,
555 &s_get_versioninfo_op);
556 if (status != IV_SUCCESS) {
557 ALOGD("error in %s: 0x%x", __func__,
558 s_get_versioninfo_op.u4_error_code);
559 } else {
560 ALOGV("ittiam decoder version number: %s",
561 (char *) s_get_versioninfo_ip.pv_version_buffer);
562 }
563
564 return OK;
565}
566
567status_t C2SoftMpeg2Dec::initDecoder() {
568 status_t ret = getNumMemRecords();
569 if (OK != ret) return ret;
570
571 ret = fillMemRecords();
572 if (OK != ret) return ret;
573
574 if (OK != createDecoder()) return UNKNOWN_ERROR;
575
576 mNumCores = MIN(getCpuCoreCount(), MAX_NUM_CORES);
Harish Mahendrakare403bf72021-10-21 10:30:35 -0700577 mStride = ALIGN128(mWidth);
Pawin Vongmasa36653902018-11-15 00:10:25 -0800578 mSignalledError = false;
579 resetPlugin();
580 (void) setNumCores();
581 if (OK != setParams(mStride)) return UNKNOWN_ERROR;
582 (void) getVersion();
583
584 return OK;
585}
586
587bool C2SoftMpeg2Dec::setDecodeArgs(ivd_video_decode_ip_t *ps_decode_ip,
588 ivd_video_decode_op_t *ps_decode_op,
589 C2ReadView *inBuffer,
590 C2GraphicView *outBuffer,
591 size_t inOffset,
592 size_t inSize,
593 uint32_t tsMarker) {
594 uint32_t displayStride = mStride;
Harish Mahendrakardc1d2d92019-10-18 16:27:38 -0700595 if (outBuffer) {
596 C2PlanarLayout layout;
597 layout = outBuffer->layout();
598 displayStride = layout.planes[C2PlanarLayout::PLANE_Y].rowInc;
599 }
Pawin Vongmasa36653902018-11-15 00:10:25 -0800600 uint32_t displayHeight = mHeight;
601 size_t lumaSize = displayStride * displayHeight;
602 size_t chromaSize = lumaSize >> 2;
603
Harish Mahendrakardc1d2d92019-10-18 16:27:38 -0700604 if (mStride != displayStride) {
605 mStride = displayStride;
606 if (OK != setParams(mStride)) return false;
607 }
608
Pawin Vongmasa36653902018-11-15 00:10:25 -0800609 ps_decode_ip->u4_size = sizeof(ivd_video_decode_ip_t);
610 ps_decode_ip->e_cmd = IVD_CMD_VIDEO_DECODE;
611 if (inBuffer) {
612 ps_decode_ip->u4_ts = tsMarker;
613 ps_decode_ip->pv_stream_buffer = const_cast<uint8_t *>(inBuffer->data() + inOffset);
614 ps_decode_ip->u4_num_Bytes = inSize;
615 } else {
616 ps_decode_ip->u4_ts = 0;
617 ps_decode_ip->pv_stream_buffer = nullptr;
618 ps_decode_ip->u4_num_Bytes = 0;
619 }
620 ps_decode_ip->s_out_buffer.u4_min_out_buf_size[0] = lumaSize;
621 ps_decode_ip->s_out_buffer.u4_min_out_buf_size[1] = chromaSize;
622 ps_decode_ip->s_out_buffer.u4_min_out_buf_size[2] = chromaSize;
623 if (outBuffer) {
Harish Mahendrakareb767792020-05-02 00:14:38 +0530624 if (outBuffer->height() < displayHeight) {
Pawin Vongmasa36653902018-11-15 00:10:25 -0800625 ALOGE("Output buffer too small: provided (%dx%d) required (%ux%u)",
626 outBuffer->width(), outBuffer->height(), displayStride, displayHeight);
627 return false;
628 }
629 ps_decode_ip->s_out_buffer.pu1_bufs[0] = outBuffer->data()[C2PlanarLayout::PLANE_Y];
630 ps_decode_ip->s_out_buffer.pu1_bufs[1] = outBuffer->data()[C2PlanarLayout::PLANE_U];
631 ps_decode_ip->s_out_buffer.pu1_bufs[2] = outBuffer->data()[C2PlanarLayout::PLANE_V];
632 } else {
633 ps_decode_ip->s_out_buffer.pu1_bufs[0] = mOutBufferDrain;
634 ps_decode_ip->s_out_buffer.pu1_bufs[1] = mOutBufferDrain + lumaSize;
635 ps_decode_ip->s_out_buffer.pu1_bufs[2] = mOutBufferDrain + lumaSize + chromaSize;
636 }
637 ps_decode_ip->s_out_buffer.u4_num_bufs = 3;
638 ps_decode_op->u4_size = sizeof(ivd_video_decode_op_t);
639
640 return true;
641}
642
643
644bool C2SoftMpeg2Dec::getSeqInfo() {
645 ivdext_ctl_get_seq_info_ip_t s_ctl_get_seq_info_ip;
646 ivdext_ctl_get_seq_info_op_t s_ctl_get_seq_info_op;
647
648 s_ctl_get_seq_info_ip.u4_size = sizeof(ivdext_ctl_get_seq_info_ip_t);
649 s_ctl_get_seq_info_ip.e_cmd = IVD_CMD_VIDEO_CTL;
650 s_ctl_get_seq_info_ip.e_sub_cmd =
651 (IVD_CONTROL_API_COMMAND_TYPE_T)IMPEG2D_CMD_CTL_GET_SEQ_INFO;
652 s_ctl_get_seq_info_op.u4_size = sizeof(ivdext_ctl_get_seq_info_op_t);
653 IV_API_CALL_STATUS_T status = ivdec_api_function(mDecHandle,
654 &s_ctl_get_seq_info_ip,
655 &s_ctl_get_seq_info_op);
656 if (status != IV_SUCCESS) {
657 ALOGW("Error in getting Sequence info: 0x%x", s_ctl_get_seq_info_op.u4_error_code);
658 return false;
659 }
660
661 VuiColorAspects vuiColorAspects;
662 vuiColorAspects.primaries = s_ctl_get_seq_info_op.u1_colour_primaries;
663 vuiColorAspects.transfer = s_ctl_get_seq_info_op.u1_transfer_characteristics;
664 vuiColorAspects.coeffs = s_ctl_get_seq_info_op.u1_matrix_coefficients;
665 vuiColorAspects.fullRange = false; // mpeg2 video has limited range.
666
667 // convert vui aspects to C2 values if changed
668 if (!(vuiColorAspects == mBitstreamColorAspects)) {
669 mBitstreamColorAspects = vuiColorAspects;
670 ColorAspects sfAspects;
671 C2StreamColorAspectsInfo::input codedAspects = { 0u };
672 ColorUtils::convertIsoColorAspectsToCodecAspects(
673 vuiColorAspects.primaries, vuiColorAspects.transfer, vuiColorAspects.coeffs,
674 vuiColorAspects.fullRange, sfAspects);
675 if (!C2Mapper::map(sfAspects.mPrimaries, &codedAspects.primaries)) {
676 codedAspects.primaries = C2Color::PRIMARIES_UNSPECIFIED;
677 }
678 if (!C2Mapper::map(sfAspects.mRange, &codedAspects.range)) {
679 codedAspects.range = C2Color::RANGE_UNSPECIFIED;
680 }
681 if (!C2Mapper::map(sfAspects.mMatrixCoeffs, &codedAspects.matrix)) {
682 codedAspects.matrix = C2Color::MATRIX_UNSPECIFIED;
683 }
684 if (!C2Mapper::map(sfAspects.mTransfer, &codedAspects.transfer)) {
685 codedAspects.transfer = C2Color::TRANSFER_UNSPECIFIED;
686 }
687 std::vector<std::unique_ptr<C2SettingResult>> failures;
688 (void)mIntf->config({&codedAspects}, C2_MAY_BLOCK, &failures);
689 }
690 return true;
691}
692
693status_t C2SoftMpeg2Dec::setFlushMode() {
694 ivd_ctl_flush_ip_t s_set_flush_ip;
695 ivd_ctl_flush_op_t s_set_flush_op;
696
697 s_set_flush_ip.u4_size = sizeof(ivd_ctl_flush_ip_t);
698 s_set_flush_ip.e_cmd = IVD_CMD_VIDEO_CTL;
699 s_set_flush_ip.e_sub_cmd = IVD_CMD_CTL_FLUSH;
700 s_set_flush_op.u4_size = sizeof(ivd_ctl_flush_op_t);
701 IV_API_CALL_STATUS_T status = ivdec_api_function(mDecHandle,
702 &s_set_flush_ip,
703 &s_set_flush_op);
704 if (status != IV_SUCCESS) {
705 ALOGE("error in %s: 0x%x", __func__, s_set_flush_op.u4_error_code);
706 return UNKNOWN_ERROR;
707 }
708
709 return OK;
710}
711
712status_t C2SoftMpeg2Dec::resetDecoder() {
713 ivd_ctl_reset_ip_t s_reset_ip;
714 ivd_ctl_reset_op_t s_reset_op;
715
716 s_reset_ip.u4_size = sizeof(ivd_ctl_reset_ip_t);
717 s_reset_ip.e_cmd = IVD_CMD_VIDEO_CTL;
718 s_reset_ip.e_sub_cmd = IVD_CMD_CTL_RESET;
719 s_reset_op.u4_size = sizeof(ivd_ctl_reset_op_t);
720 IV_API_CALL_STATUS_T status = ivdec_api_function(mDecHandle,
721 &s_reset_ip,
722 &s_reset_op);
723 if (IV_SUCCESS != status) {
724 ALOGE("error in %s: 0x%x", __func__, s_reset_op.u4_error_code);
725 return UNKNOWN_ERROR;
726 }
727 (void) setNumCores();
728 mStride = 0;
729 mSignalledError = false;
730
731 return OK;
732}
733
734void C2SoftMpeg2Dec::resetPlugin() {
735 mSignalledOutputEos = false;
Ray Essick24754942022-04-16 09:50:35 -0700736 mTimeStart = mTimeEnd = systemTime();
Harish Mahendrakar05575b12022-10-19 18:17:07 -0700737 if (mOutBlock) {
738 mOutBlock.reset();
739 }
Pawin Vongmasa36653902018-11-15 00:10:25 -0800740}
741
742status_t C2SoftMpeg2Dec::deleteDecoder() {
743 if (mMemRecords) {
744 iv_mem_rec_t *ps_mem_rec = mMemRecords;
745
746 for (size_t i = 0; i < mNumMemRecords; i++, ps_mem_rec++) {
747 if (ps_mem_rec->pv_base) {
748 ivd_aligned_free(ps_mem_rec->pv_base);
749 }
750 }
751 ivd_aligned_free(mMemRecords);
752 mMemRecords = nullptr;
753 }
754 mDecHandle = nullptr;
755
756 return OK;
757}
758
759status_t C2SoftMpeg2Dec::reInitDecoder() {
760 deleteDecoder();
761
762 status_t ret = initDecoder();
763 if (OK != ret) {
764 ALOGE("Failed to initialize decoder");
765 deleteDecoder();
766 return ret;
767 }
768 return OK;
769}
770
771void fillEmptyWork(const std::unique_ptr<C2Work> &work) {
772 uint32_t flags = 0;
773 if (work->input.flags & C2FrameData::FLAG_END_OF_STREAM) {
774 flags |= C2FrameData::FLAG_END_OF_STREAM;
775 ALOGV("signalling eos");
776 }
777 work->worklets.front()->output.flags = (C2FrameData::flags_t)flags;
778 work->worklets.front()->output.buffers.clear();
779 work->worklets.front()->output.ordinal = work->input.ordinal;
780 work->workletsProcessed = 1u;
781}
782
783void C2SoftMpeg2Dec::finishWork(uint64_t index, const std::unique_ptr<C2Work> &work) {
784 std::shared_ptr<C2Buffer> buffer = createGraphicBuffer(std::move(mOutBlock),
785 C2Rect(mWidth, mHeight));
786 mOutBlock = nullptr;
787 {
788 IntfImpl::Lock lock = mIntf->lock();
789 buffer->setInfo(mIntf->getColorAspects_l());
790 }
791
Rakesh Kumard91bc482019-02-18 10:42:41 +0530792 class FillWork {
793 public:
794 FillWork(uint32_t flags, C2WorkOrdinalStruct ordinal,
795 const std::shared_ptr<C2Buffer>& buffer)
796 : mFlags(flags), mOrdinal(ordinal), mBuffer(buffer) {}
797 ~FillWork() = default;
798
799 void operator()(const std::unique_ptr<C2Work>& work) {
800 work->worklets.front()->output.flags = (C2FrameData::flags_t)mFlags;
801 work->worklets.front()->output.buffers.clear();
802 work->worklets.front()->output.ordinal = mOrdinal;
803 work->workletsProcessed = 1u;
804 work->result = C2_OK;
805 if (mBuffer) {
806 work->worklets.front()->output.buffers.push_back(mBuffer);
807 }
808 ALOGV("timestamp = %lld, index = %lld, w/%s buffer",
809 mOrdinal.timestamp.peekll(), mOrdinal.frameIndex.peekll(),
810 mBuffer ? "" : "o");
811 }
812
813 private:
814 const uint32_t mFlags;
815 const C2WorkOrdinalStruct mOrdinal;
816 const std::shared_ptr<C2Buffer> mBuffer;
817 };
818
Pawin Vongmasa36653902018-11-15 00:10:25 -0800819 auto fillWork = [buffer](const std::unique_ptr<C2Work> &work) {
820 work->worklets.front()->output.flags = (C2FrameData::flags_t)0;
821 work->worklets.front()->output.buffers.clear();
822 work->worklets.front()->output.buffers.push_back(buffer);
823 work->worklets.front()->output.ordinal = work->input.ordinal;
824 work->workletsProcessed = 1u;
825 };
826 if (work && c2_cntr64_t(index) == work->input.ordinal.frameIndex) {
Rakesh Kumard91bc482019-02-18 10:42:41 +0530827 bool eos = ((work->input.flags & C2FrameData::FLAG_END_OF_STREAM) != 0);
828 // TODO: Check if cloneAndSend can be avoided by tracking number of frames remaining
829 if (eos) {
830 if (buffer) {
831 mOutIndex = index;
832 C2WorkOrdinalStruct outOrdinal = work->input.ordinal;
833 cloneAndSend(
834 mOutIndex, work,
835 FillWork(C2FrameData::FLAG_INCOMPLETE, outOrdinal, buffer));
836 buffer.reset();
837 }
838 } else {
839 fillWork(work);
840 }
Pawin Vongmasa36653902018-11-15 00:10:25 -0800841 } else {
842 finish(index, fillWork);
843 }
844}
845
846c2_status_t C2SoftMpeg2Dec::ensureDecoderState(const std::shared_ptr<C2BlockPool> &pool) {
847 if (!mDecHandle) {
848 ALOGE("not supposed to be here, invalid decoder context");
849 return C2_CORRUPTED;
850 }
Pawin Vongmasa36653902018-11-15 00:10:25 -0800851 if (mOutBlock &&
Harish Mahendrakare403bf72021-10-21 10:30:35 -0700852 (mOutBlock->width() != ALIGN128(mWidth) || mOutBlock->height() != mHeight)) {
Pawin Vongmasa36653902018-11-15 00:10:25 -0800853 mOutBlock.reset();
854 }
855 if (!mOutBlock) {
856 uint32_t format = HAL_PIXEL_FORMAT_YV12;
857 C2MemoryUsage usage = { C2MemoryUsage::CPU_READ, C2MemoryUsage::CPU_WRITE };
Harish Mahendrakardc1d2d92019-10-18 16:27:38 -0700858 c2_status_t err =
Harish Mahendrakare403bf72021-10-21 10:30:35 -0700859 pool->fetchGraphicBlock(ALIGN128(mWidth), mHeight, format, usage, &mOutBlock);
Pawin Vongmasa36653902018-11-15 00:10:25 -0800860 if (err != C2_OK) {
861 ALOGE("fetchGraphicBlock for Output failed with status %d", err);
862 return err;
863 }
864 ALOGV("provided (%dx%d) required (%dx%d)",
Harish Mahendrakare403bf72021-10-21 10:30:35 -0700865 mOutBlock->width(), mOutBlock->height(), ALIGN128(mWidth), mHeight);
Pawin Vongmasa36653902018-11-15 00:10:25 -0800866 }
867
868 return C2_OK;
869}
870
871// TODO: can overall error checking be improved?
872// TODO: allow configuration of color format and usage for graphic buffers instead
873// of hard coding them to HAL_PIXEL_FORMAT_YV12
874// TODO: pass coloraspects information to surface
875// TODO: test support for dynamic change in resolution
876// TODO: verify if the decoder sent back all frames
877void C2SoftMpeg2Dec::process(
878 const std::unique_ptr<C2Work> &work,
879 const std::shared_ptr<C2BlockPool> &pool) {
880 // Initialize output work
881 work->result = C2_OK;
882 work->workletsProcessed = 0u;
883 work->worklets.front()->output.configUpdate.clear();
884 work->worklets.front()->output.flags = work->input.flags;
885
886 if (mSignalledError || mSignalledOutputEos) {
887 work->result = C2_BAD_VALUE;
888 return;
889 }
890
891 size_t inOffset = 0u;
892 size_t inSize = 0u;
893 uint32_t workIndex = work->input.ordinal.frameIndex.peeku() & 0xFFFFFFFF;
894 C2ReadView rView = mDummyReadView;
895 if (!work->input.buffers.empty()) {
896 rView = work->input.buffers[0]->data().linearBlocks().front().map().get();
897 inSize = rView.capacity();
898 if (inSize && rView.error()) {
899 ALOGE("read view map failed %d", rView.error());
900 work->result = C2_CORRUPTED;
901 return;
902 }
903 }
904 bool eos = ((work->input.flags & C2FrameData::FLAG_END_OF_STREAM) != 0);
905 bool hasPicture = false;
906
907 ALOGV("in buffer attr. size %zu timestamp %d frameindex %d, flags %x",
908 inSize, (int)work->input.ordinal.timestamp.peeku(),
909 (int)work->input.ordinal.frameIndex.peeku(), work->input.flags);
910 size_t inPos = 0;
911 while (inPos < inSize) {
912 if (C2_OK != ensureDecoderState(pool)) {
913 mSignalledError = true;
914 work->workletsProcessed = 1u;
915 work->result = C2_CORRUPTED;
916 return;
917 }
918 C2GraphicView wView = mOutBlock->map().get();
919 if (wView.error()) {
920 ALOGE("graphic view map failed %d", wView.error());
921 work->result = C2_CORRUPTED;
922 return;
923 }
924
925 ivd_video_decode_ip_t s_decode_ip;
926 ivd_video_decode_op_t s_decode_op;
927 if (!setDecodeArgs(&s_decode_ip, &s_decode_op, &rView, &wView,
928 inOffset + inPos, inSize - inPos, workIndex)) {
929 mSignalledError = true;
930 work->workletsProcessed = 1u;
931 work->result = C2_CORRUPTED;
932 return;
933 }
934 // If input dump is enabled, then write to file
935 DUMP_TO_FILE(mInFile, s_decode_ip.pv_stream_buffer, s_decode_ip.u4_num_Bytes);
Ray Essick24754942022-04-16 09:50:35 -0700936 nsecs_t delay = mTimeStart - mTimeEnd;
Pawin Vongmasa36653902018-11-15 00:10:25 -0800937 (void) ivdec_api_function(mDecHandle, &s_decode_ip, &s_decode_op);
Ray Essick24754942022-04-16 09:50:35 -0700938
939 mTimeEnd = systemTime();
940 nsecs_t decodeTime = mTimeEnd - mTimeStart;
941 ALOGV("decodeTime=%" PRId64 " delay=%" PRId64 " numBytes=%6d ", decodeTime, delay,
Pawin Vongmasa36653902018-11-15 00:10:25 -0800942 s_decode_op.u4_num_bytes_consumed);
943 if (IMPEG2D_UNSUPPORTED_DIMENSIONS == s_decode_op.u4_error_code) {
944 ALOGV("unsupported resolution : %dx%d", s_decode_op.u4_pic_wd, s_decode_op.u4_pic_ht);
945 drainInternal(DRAIN_COMPONENT_NO_EOS, pool, work);
946 resetPlugin();
947 work->workletsProcessed = 0u;
948 mWidth = s_decode_op.u4_pic_wd;
949 mHeight = s_decode_op.u4_pic_ht;
950
951 ALOGI("Configuring decoder: mWidth %d , mHeight %d ",
952 mWidth, mHeight);
Lajos Molnar3bb81cd2019-02-20 15:10:30 -0800953 C2StreamPictureSizeInfo::output size(0u, mWidth, mHeight);
Pawin Vongmasa36653902018-11-15 00:10:25 -0800954 std::vector<std::unique_ptr<C2SettingResult>> failures;
955 c2_status_t err =
956 mIntf->config({&size}, C2_MAY_BLOCK, &failures);
957 if (err == OK) {
958 work->worklets.front()->output.configUpdate.push_back(
959 C2Param::Copy(size));
960 } else {
961 ALOGE("Cannot set width and height");
962 mSignalledError = true;
963 work->workletsProcessed = 1u;
964 work->result = C2_CORRUPTED;
965 return;
966 }
967
968 if (OK != reInitDecoder()) {
969 ALOGE("Failed to reinitialize decoder");
970 mSignalledError = true;
971 work->workletsProcessed = 1u;
972 work->result = C2_CORRUPTED;
973 return;
974 }
975 continue;
976 } else if (IVD_RES_CHANGED == (s_decode_op.u4_error_code & 0xFF)) {
977 ALOGV("resolution changed");
978 drainInternal(DRAIN_COMPONENT_NO_EOS, pool, work);
979 resetDecoder();
980 resetPlugin();
981 work->workletsProcessed = 0u;
982 continue;
983 }
984 if (0 < s_decode_op.u4_pic_wd && 0 < s_decode_op.u4_pic_ht) {
985 if (s_decode_op.u4_pic_wd != mWidth || s_decode_op.u4_pic_ht != mHeight) {
986 mWidth = s_decode_op.u4_pic_wd;
987 mHeight = s_decode_op.u4_pic_ht;
988 CHECK_EQ(0u, s_decode_op.u4_output_present);
989
990 ALOGI("Configuring decoder out: mWidth %d , mHeight %d ",
991 mWidth, mHeight);
Lajos Molnar3bb81cd2019-02-20 15:10:30 -0800992 C2StreamPictureSizeInfo::output size(0u, mWidth, mHeight);
Pawin Vongmasa36653902018-11-15 00:10:25 -0800993 std::vector<std::unique_ptr<C2SettingResult>> failures;
994 c2_status_t err =
995 mIntf->config({&size}, C2_MAY_BLOCK, &failures);
996 if (err == OK) {
997 work->worklets.front()->output.configUpdate.push_back(
998 C2Param::Copy(size));
999 } else {
1000 ALOGE("Cannot set width and height");
1001 mSignalledError = true;
1002 work->workletsProcessed = 1u;
1003 work->result = C2_CORRUPTED;
1004 return;
1005 }
1006 }
1007 }
1008
1009 (void) getSeqInfo();
1010 hasPicture |= (1 == s_decode_op.u4_frame_decoded_flag);
1011 if (s_decode_op.u4_output_present) {
1012 finishWork(s_decode_op.u4_ts, work);
1013 }
1014
1015 inPos += s_decode_op.u4_num_bytes_consumed;
1016 if (hasPicture && (inSize - inPos) != 0) {
1017 ALOGD("decoded frame in current access nal, ignoring further trailing bytes %d",
1018 (int)inSize - (int)inPos);
1019 break;
1020 }
1021 }
1022
1023 if (eos) {
1024 drainInternal(DRAIN_COMPONENT_WITH_EOS, pool, work);
1025 mSignalledOutputEos = true;
1026 } else if (!hasPicture) {
1027 fillEmptyWork(work);
1028 }
1029}
1030
1031c2_status_t C2SoftMpeg2Dec::drainInternal(
1032 uint32_t drainMode,
1033 const std::shared_ptr<C2BlockPool> &pool,
1034 const std::unique_ptr<C2Work> &work) {
1035 if (drainMode == NO_DRAIN) {
1036 ALOGW("drain with NO_DRAIN: no-op");
1037 return C2_OK;
1038 }
1039 if (drainMode == DRAIN_CHAIN) {
1040 ALOGW("DRAIN_CHAIN not supported");
1041 return C2_OMITTED;
1042 }
1043
1044 if (OK != setFlushMode()) return C2_CORRUPTED;
1045 while (true) {
1046 if (C2_OK != ensureDecoderState(pool)) {
1047 mSignalledError = true;
1048 work->workletsProcessed = 1u;
1049 work->result = C2_CORRUPTED;
1050 return C2_CORRUPTED;
1051 }
1052 C2GraphicView wView = mOutBlock->map().get();
1053 if (wView.error()) {
1054 ALOGE("graphic view map failed %d", wView.error());
1055 return C2_CORRUPTED;
1056 }
1057 ivd_video_decode_ip_t s_decode_ip;
1058 ivd_video_decode_op_t s_decode_op;
1059 if (!setDecodeArgs(&s_decode_ip, &s_decode_op, nullptr, &wView, 0, 0, 0)) {
1060 mSignalledError = true;
1061 work->workletsProcessed = 1u;
1062 return C2_CORRUPTED;
1063 }
1064 (void) ivdec_api_function(mDecHandle, &s_decode_ip, &s_decode_op);
1065 if (s_decode_op.u4_output_present) {
1066 finishWork(s_decode_op.u4_ts, work);
1067 } else {
1068 fillEmptyWork(work);
1069 break;
1070 }
1071 }
1072
1073 return C2_OK;
1074}
1075
1076c2_status_t C2SoftMpeg2Dec::drain(
1077 uint32_t drainMode,
1078 const std::shared_ptr<C2BlockPool> &pool) {
1079 return drainInternal(drainMode, pool, nullptr);
1080}
1081
1082class C2SoftMpeg2DecFactory : public C2ComponentFactory {
1083public:
1084 C2SoftMpeg2DecFactory() : mHelper(std::static_pointer_cast<C2ReflectorHelper>(
1085 GetCodec2PlatformComponentStore()->getParamReflector())) {
1086 }
1087
1088 virtual c2_status_t createComponent(
1089 c2_node_id_t id,
1090 std::shared_ptr<C2Component>* const component,
1091 std::function<void(C2Component*)> deleter) override {
1092 *component = std::shared_ptr<C2Component>(
1093 new C2SoftMpeg2Dec(COMPONENT_NAME,
1094 id,
1095 std::make_shared<C2SoftMpeg2Dec::IntfImpl>(mHelper)),
1096 deleter);
1097 return C2_OK;
1098 }
1099
1100 virtual c2_status_t createInterface(
1101 c2_node_id_t id,
1102 std::shared_ptr<C2ComponentInterface>* const interface,
1103 std::function<void(C2ComponentInterface*)> deleter) override {
1104 *interface = std::shared_ptr<C2ComponentInterface>(
1105 new SimpleInterface<C2SoftMpeg2Dec::IntfImpl>(
1106 COMPONENT_NAME, id, std::make_shared<C2SoftMpeg2Dec::IntfImpl>(mHelper)),
1107 deleter);
1108 return C2_OK;
1109 }
1110
1111 virtual ~C2SoftMpeg2DecFactory() override = default;
1112
1113private:
1114 std::shared_ptr<C2ReflectorHelper> mHelper;
1115};
1116
1117} // namespace android
1118
Cindy Zhouf6c0c3c2020-12-02 10:53:40 -08001119__attribute__((cfi_canonical_jump_table))
Pawin Vongmasa36653902018-11-15 00:10:25 -08001120extern "C" ::C2ComponentFactory* CreateCodec2Factory() {
1121 ALOGV("in %s", __func__);
1122 return new ::android::C2SoftMpeg2DecFactory();
1123}
1124
Cindy Zhouf6c0c3c2020-12-02 10:53:40 -08001125__attribute__((cfi_canonical_jump_table))
Pawin Vongmasa36653902018-11-15 00:10:25 -08001126extern "C" void DestroyCodec2Factory(::C2ComponentFactory* factory) {
1127 ALOGV("in %s", __func__);
1128 delete factory;
1129}