Android 13 - Media框架（31）- ACodec（七）

之前的章节中我们解了 input buffer 是如何传递给 OMX 的，以及Output buffer 是如何分配并且注册给 OMX 的。这一节我们就来看ACodec是如何处理OMX的Callback的。

1、OMXNodeInstance Callback

这一节我们只大致记录Callback是如何传递给ACodec的。在之前的学习中我们了解到OMXNodeInstance中会有一个专门的线程来处理OMX的callback，这个线程的作用是把Callback按照时间顺序回传给ACodec。

CallbackDispatcher中维护了一个list，将消息回传给ACodec时并不是将list中的消息一条一条回传的，而是将list中所有的消息一次性回传，这也就是为什么ACodec处理OMXNodeInstance的消息时会有循环遍历。

在调用CodecObserver做消息上抛之前，会调用OMXNodeInstance::handleMessage 对消息做预处理，这里的预处里包括是否要将buffer做拷贝等等。

2、onOMXEmptyBufferDone

OMX使用完input buffer后，消息上抛到ACodec层，ACodec 会调用onOMXEmptyBufferDone再处理input buffer。

bool ACodec::BaseState::onOMXEmptyBufferDone(IOMX::buffer_id bufferID, int fenceFd) {
    ALOGV("[%s] onOMXEmptyBufferDone %u",
         mCodec->mComponentName.c_str(), bufferID);

    BufferInfo *info = mCodec->findBufferByID(kPortIndexInput, bufferID);
    BufferInfo::Status status = BufferInfo::getSafeStatus(info);
    // 检查 Buffer 状态
    if (status != BufferInfo::OWNED_BY_COMPONENT) {
        ALOGE("Wrong ownership in EBD: %s(%d) buffer #%u", _asString(status), status, bufferID);
        mCodec->dumpBuffers(kPortIndexInput);
        if (fenceFd >= 0) {
            ::close(fenceFd);
        }
        return false;
    }
    // input buffer 回到 ACodec
    info->mStatus = BufferInfo::OWNED_BY_US;

    // input buffers cannot take fences, so wait for any fence now
    (void)mCodec->waitForFence(fenceFd, "onOMXEmptyBufferDone");
    fenceFd = -1;

    // still save fence for completeness
    info->setWriteFence(fenceFd, "onOMXEmptyBufferDone");

    // We're in "store-metadata-in-buffers" mode, the underlying
    // OMX component had access to data that's implicitly refcounted
    // by this "MediaBuffer" object. Now that the OMX component has
    // told us that it's done with the input buffer, we can decrement
    // the mediaBuffer's reference count.
    info->mData->meta()->setObject("mediaBufferHolder", sp<MediaBufferHolder>(nullptr));
	// 获取当前的 PortMode
    PortMode mode = getPortMode(kPortIndexInput);

    switch (mode) {
        case KEEP_BUFFERS:
            break;

        case RESUBMIT_BUFFERS:
            postFillThisBuffer(info);
            break;

        case FREE_BUFFERS:
        default:
            ALOGE("SHOULD NOT REACH HERE: cannot free empty output buffers");
            return false;
    }

    return true;
}

对input buffer的处理很简单，检查当前ACodec处在的状态并作出反应，如果处在 ExecutingState 则调用 postFillThisBuffer 将 Buffer 提交给 MediaCodec，同时清除 ACodec 存储的 mData。其他状态下则持有 input buffer 不会将其回传给 MediaCodec。

2、onOMXFillBufferDone

ACodec 处理 output buffer 的代码比较长，但是也不难，接下来就做分解学习：

首先有个 debug log，我们可以打开宏TRACK_BUFFER_TIMING来使用这部分内容，把input buffer写给 OMX 时会将pts以及调用时间做记录，在output buffer回传回来时，检查pts，打印出解码该帧消耗的时间。

#if TRACK_BUFFER_TIMING
    index = mCodec->mBufferStats.indexOfKey(timeUs);
    if (index >= 0) {
        ACodec::BufferStats *stats = &mCodec->mBufferStats.editValueAt(index);
        stats->mFillBufferDoneTimeUs = ALooper::GetNowUs();

        ALOGI("frame PTS %lld: %lld",
                timeUs,
                stats->mFillBufferDoneTimeUs - stats->mEmptyBufferTimeUs);

        mCodec->mBufferStats.removeItemsAt(index);
        stats = NULL;
    }
#endif

记录output BufferInfo是在第几帧被使用，mDequeueCounter可以看作是当前解码的帧数。

    info->mDequeuedAt = ++mCodec->mDequeueCounter;
    info->mStatus = BufferInfo::OWNED_BY_US;

2.1 Executing

在 Executing 状态下，会检查output buffer flag 和 size：

• output buffer size为0，flag 不是 OMX_BUFFERFLAG_EOS，说明没有解出有效数据，重新回传给 OMX 使用；
• output buffer size为0，flag 是 OMX_BUFFERFLAG_EOS，ACodec 已经收到 EOS，重新把 buffer 交给 OMX；
• 其他情况说明数据有效，或者是flag是 OMX_BUFFERFLAG_EOS，需要把output buffer回传给上层。

        case RESUBMIT_BUFFERS:
        {
        	// 如果output buffer长度为0，flag 不是 OMX_BUFFERFLAG_EOS
        	// 如果output buffer长度为0，已将收到 EOS
        	// 重新把 output buffer 提交给 OMX
            if (rangeLength == 0 && (!(flags & OMX_BUFFERFLAG_EOS)
                    || mCodec->mPortEOS[kPortIndexOutput])) {
                ALOGV("[%s] calling fillBuffer %u",
                     mCodec->mComponentName.c_str(), info->mBufferID);

                err = mCodec->fillBuffer(info);
                if (err != OK) {
                    mCodec->signalError(OMX_ErrorUndefined, makeNoSideEffectStatus(err));
                    return true;
                }
                break;
            }

            sp<MediaCodecBuffer> buffer = info->mData;
            // ......
            // 设定 pts
            buffer->meta()->setInt64("timeUs", timeUs);
			// 解除 ACodec 引用
            info->mData.clear();
			// 调用 drainThisBuffer
            mCodec->mBufferChannel->drainThisBuffer(info->mBufferID, flags);

            info->mStatus = BufferInfo::OWNED_BY_DOWNSTREAM;
			// 如果 flag 为 OMX_BUFFERFLAG_EOS，将PortEOS置为true
            if (flags & OMX_BUFFERFLAG_EOS) {
                ALOGV("[%s] saw output EOS", mCodec->mComponentName.c_str());

                mCodec->mCallback->onEos(mCodec->mInputEOSResult);
                mCodec->mPortEOS[kPortIndexOutput] = true;
            }
            break;
        }

2.2 OutputPortSettingsChangedState

ACodec::BaseState::PortMode ACodec::OutputPortSettingsChangedState::getPortMode(
        OMX_U32 portIndex) {
    if (portIndex == kPortIndexOutput) {
        return FREE_BUFFERS;
    }

    CHECK_EQ(portIndex, (OMX_U32)kPortIndexInput);

    return RESUBMIT_BUFFERS;
}