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