1e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu/*
2e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu* Copyright (c) 2009-2011 Intel Corporation.  All rights reserved.
3e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu*
4e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu* Licensed under the Apache License, Version 2.0 (the "License");
5e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu* you may not use this file except in compliance with the License.
6e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu* You may obtain a copy of the License at
7e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu*
8e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu* http://www.apache.org/licenses/LICENSE-2.0
9e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu*
10e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu* Unless required by applicable law or agreed to in writing, software
11e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu* distributed under the License is distributed on an "AS IS" BASIS,
12e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu* See the License for the specific language governing permissions and
14e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu* limitations under the License.
15e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu*/
16e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu
17e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu
18e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu#define LOG_TAG "OMXVideoEncoderMPEG4"
19e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu#include "OMXVideoEncoderMPEG4.h"
20e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu
2119b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sangstatic const char *MPEG4_MIME_TYPE = "video/mpeg4";
22e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu
23e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy QiuOMXVideoEncoderMPEG4::OMXVideoEncoderMPEG4() {
24e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu    LOGV("OMXVideoEncoderMPEG4 is constructed.");
25e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu    BuildHandlerList();
2619b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang    mVideoEncoder = createVideoEncoder(MPEG4_MIME_TYPE);
2719b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang    if (!mVideoEncoder) LOGE("OMX_ErrorInsufficientResources");
28e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu}
29e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu
30e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy QiuOMXVideoEncoderMPEG4::~OMXVideoEncoderMPEG4() {
31e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu    LOGV("OMXVideoEncoderMPEG4 is destructed.");
32e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu}
33e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu
34e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy QiuOMX_ERRORTYPE OMXVideoEncoderMPEG4::InitOutputPortFormatSpecific(OMX_PARAM_PORTDEFINITIONTYPE *paramPortDefinitionOutput) {
35e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu    // OMX_VIDEO_PARAM_MPEG4TYPE
36e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu    memset(&mParamMpeg4, 0, sizeof(mParamMpeg4));
37e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu    SetTypeHeader(&mParamMpeg4, sizeof(mParamMpeg4));
38e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu    mParamMpeg4.nPortIndex = OUTPORT_INDEX;
39e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu    mParamMpeg4.eProfile = OMX_VIDEO_MPEG4ProfileSimple;
4019b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang    // TODO: Check eLevel (Level3)
41e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu    mParamMpeg4.eLevel = OMX_VIDEO_MPEG4Level5; //OMX_VIDEO_MPEG4Level3;
42e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu
43e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu    // override OMX_PARAM_PORTDEFINITIONTYPE
44e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu    paramPortDefinitionOutput->nBufferCountActual = OUTPORT_ACTUAL_BUFFER_COUNT;
45e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu    paramPortDefinitionOutput->nBufferCountMin = OUTPORT_MIN_BUFFER_COUNT;
46e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu    paramPortDefinitionOutput->nBufferSize = OUTPORT_BUFFER_SIZE;
47e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu    paramPortDefinitionOutput->format.video.cMIMEType = (OMX_STRING)MPEG4_MIME_TYPE;
48e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu    paramPortDefinitionOutput->format.video.eCompressionFormat = OMX_VIDEO_CodingMPEG4;
49e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu
50e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu    // override OMX_VIDEO_PARAM_PROFILELEVELTYPE
51e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu    // TODO: check if profile/level supported is correct
52e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu    mParamProfileLevel.eProfile = mParamMpeg4.eProfile;
53e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu    mParamProfileLevel.eLevel = mParamMpeg4.eLevel; //OMX_VIDEO_MPEG4Level5;
54e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu
5519b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang    // override OMX_VIDEO_CONFIG_INTEL_BITRATETYPE
5619b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang    mConfigIntelBitrate.nInitialQP = 15;  // Initial QP for I frames
57e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu    return OMX_ErrorNone;
58e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu}
59e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu
6019b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo SangOMX_ERRORTYPE OMXVideoEncoderMPEG4::SetVideoEncoderParam(void) {
6119b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang
6219b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang    if (!mEncoderParams) {
6319b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang        LOGE("NULL pointer: mEncoderParams");
6419b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang        return OMX_ErrorBadParameter;
6519b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang    }
6619b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang
6719b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang    mVideoEncoder->getParameters(mEncoderParams);
6819b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang    mEncoderParams->profile = (VAProfile)PROFILE_MPEG4SIMPLE;
6919b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang    return OMXVideoEncoderBase::SetVideoEncoderParam();
7019b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang}
7119b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang
72e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy QiuOMX_ERRORTYPE OMXVideoEncoderMPEG4::ProcessorInit(void) {
73e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu    return OMXVideoEncoderBase::ProcessorInit();
74e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu}
75e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu
76e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy QiuOMX_ERRORTYPE OMXVideoEncoderMPEG4::ProcessorDeinit(void) {
77e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu    return OMXVideoEncoderBase::ProcessorDeinit();
78e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu}
79e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu
80e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy QiuOMX_ERRORTYPE OMXVideoEncoderMPEG4::ProcessorProcess(
8119b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang    OMX_BUFFERHEADERTYPE **buffers,
8219b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang    buffer_retain_t *retains,
830541f11be2f78b558476772f1f877e140f9d844bywan    OMX_U32) {
8419b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang
8519b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang    VideoEncOutputBuffer outBuf;
8619b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang    VideoEncRawBuffer inBuf;
8719b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang    Encode_Status ret = ENCODE_SUCCESS;
8819b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang
8919b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang    OMX_U32 outfilledlen = 0;
9019b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang    OMX_S64 outtimestamp = 0;
9119b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang    OMX_U32 outflags = 0;
9219b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang    OMX_ERRORTYPE oret = OMX_ErrorNone;
9319b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang
9419b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang
9519b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang    LOGV_IF(buffers[INPORT_INDEX]->nFlags & OMX_BUFFERFLAG_EOS,
9619b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang            "%s(),%d: got OMX_BUFFERFLAG_EOS\n", __func__, __LINE__);
9719b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang
9819b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang    if (!buffers[INPORT_INDEX]->nFilledLen) {
9919b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang        LOGV("%s(),%d: input buffer's nFilledLen is zero\n",  __func__, __LINE__);
10019b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang        goto out;
10119b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang    }
10219b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang
1034ce57b82fa719f2aa9247c7f4a44853715a29987Guoliang Ji    inBuf.data = buffers[INPORT_INDEX]->pBuffer + buffers[INPORT_INDEX]->nOffset;
1044ce57b82fa719f2aa9247c7f4a44853715a29987Guoliang Ji    inBuf.size = buffers[INPORT_INDEX]->nFilledLen;
105dbb4b89361b47ddf1988c654c5cc8e07818dd5ecZhao Liang    inBuf.type = FTYPE_UNKNOWN;
106dbb4b89361b47ddf1988c654c5cc8e07818dd5ecZhao Liang    inBuf.flag = 0;
107dbb4b89361b47ddf1988c654c5cc8e07818dd5ecZhao Liang    inBuf.timeStamp = buffers[INPORT_INDEX]->nTimeStamp;
1084ce57b82fa719f2aa9247c7f4a44853715a29987Guoliang Ji
10919b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang    LOGV("inBuf.data=%x, size=%d", (unsigned)inBuf.data, inBuf.size);
11019b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang
11119b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang    outBuf.data =
11219b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang        buffers[OUTPORT_INDEX]->pBuffer + buffers[OUTPORT_INDEX]->nOffset;
11319b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang    outBuf.dataSize = 0;
11419b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang    outBuf.bufferSize = buffers[OUTPORT_INDEX]->nAllocLen - buffers[OUTPORT_INDEX]->nOffset;
11519b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang
11619b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang
11719b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang    if (mFrameRetrieved) {
11819b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang        // encode and setConfig need to be thread safe
11919b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang        pthread_mutex_unlock(&mSerializationLock);
12019b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang        ret = mVideoEncoder->encode(&inBuf);
12119b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang        pthread_mutex_unlock(&mSerializationLock);
12219b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang
12319b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang        CHECK_ENCODE_STATUS("encode");
12419b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang        mFrameRetrieved = OMX_FALSE;
12519b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang
12619b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang        // This is for buffer contention, we won't release current buffer
12719b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang        // but the last input buffer
12819b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang        ports[INPORT_INDEX]->ReturnAllRetainedBuffers();
12919b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang    }
13019b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang
13119b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang    if (mFirstFrame) {
13219b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang        LOGV("mFirstFrame\n");
13319b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang        outBuf.format = OUTPUT_CODEC_DATA;
13419b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang        ret = mVideoEncoder->getOutput(&outBuf);
13519b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang        CHECK_ENCODE_STATUS("getOutput");
13619b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang        // Return code could not be ENCODE_BUFFER_TOO_SMALL
13719b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang        // If we don't return error, we will have dead lock issue
13819b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang        if (ret == ENCODE_BUFFER_TOO_SMALL) {
13919b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang            return OMX_ErrorUndefined;
14019b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang        }
14119b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang
14219b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang        LOGV("output codec data size = %d", outBuf.dataSize);
14319b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang
14419b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang        outflags |= OMX_BUFFERFLAG_CODECCONFIG;
14519b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang        outflags |= OMX_BUFFERFLAG_ENDOFFRAME;
14619b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang        outflags |= OMX_BUFFERFLAG_SYNCFRAME;
14719b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang
148f536dbda76fc8fddd355af82875a4521463cc4beChang Ying        retains[INPORT_INDEX] = BUFFER_RETAIN_GETAGAIN;
14919b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang        outfilledlen = outBuf.dataSize;
15019b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang        mFirstFrame = OMX_FALSE;
15119b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang    } else {
15288c8d2ab4d97fe48ed25c3adff105f2cef5a2d36Zhao Liang        if (mSyncEncoding == OMX_FALSE && mFrameInputCount == 1) {
15330736b960fca5337c0e89021f2ff8779475b4ef8Zhao Liang            retains[INPORT_INDEX] = BUFFER_RETAIN_ACCUMULATE;
15430736b960fca5337c0e89021f2ff8779475b4ef8Zhao Liang            retains[OUTPORT_INDEX] = BUFFER_RETAIN_GETAGAIN;
15530736b960fca5337c0e89021f2ff8779475b4ef8Zhao Liang            mFrameRetrieved = OMX_TRUE;
15630736b960fca5337c0e89021f2ff8779475b4ef8Zhao Liang            goto out;
15730736b960fca5337c0e89021f2ff8779475b4ef8Zhao Liang        }
15830736b960fca5337c0e89021f2ff8779475b4ef8Zhao Liang
15919b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang        outBuf.format = OUTPUT_EVERYTHING;
16019b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang        mVideoEncoder->getOutput(&outBuf);
16119b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang        CHECK_ENCODE_STATUS("getOutput");
16219b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang
16319b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang        LOGV("output data size = %d", outBuf.dataSize);
16419b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang
16519b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang
16619b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang        outfilledlen = outBuf.dataSize;
167dbb4b89361b47ddf1988c654c5cc8e07818dd5ecZhao Liang        outtimestamp = outBuf.timeStamp;
16819b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang
16919b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang        if (outBuf.flag & ENCODE_BUFFERFLAG_SYNCFRAME) {
17019b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang            outflags |= OMX_BUFFERFLAG_SYNCFRAME;
17119b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang        }
17219b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang
17319b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang        if (outBuf.flag & ENCODE_BUFFERFLAG_ENDOFFRAME) {
17419b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang            LOGV("Get buffer done\n");
17519b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang            outflags |= OMX_BUFFERFLAG_ENDOFFRAME;
17619b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang            mFrameRetrieved = OMX_TRUE;
17788c8d2ab4d97fe48ed25c3adff105f2cef5a2d36Zhao Liang            if (mSyncEncoding)
17888c8d2ab4d97fe48ed25c3adff105f2cef5a2d36Zhao Liang                retains[INPORT_INDEX] = BUFFER_RETAIN_NOT_RETAIN;
17988c8d2ab4d97fe48ed25c3adff105f2cef5a2d36Zhao Liang            else
18088c8d2ab4d97fe48ed25c3adff105f2cef5a2d36Zhao Liang                retains[INPORT_INDEX] = BUFFER_RETAIN_ACCUMULATE;
18119b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang
18219b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang        } else {
18319b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang            retains[INPORT_INDEX] = BUFFER_RETAIN_GETAGAIN;  //get again
18419b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang
18519b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang        }
18619b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang
18719b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang    }
18819b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang
18919b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang    if (outfilledlen > 0) {
19019b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang        retains[OUTPORT_INDEX] = BUFFER_RETAIN_NOT_RETAIN;
19119b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang    } else {
19219b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang        retains[OUTPORT_INDEX] = BUFFER_RETAIN_GETAGAIN;
19319b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang    }
19419b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang
19519b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang
19619b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang
19719b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang#if SHOW_FPS
19819b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang    {
19919b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang        struct timeval t;
20019b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang        OMX_TICKS current_ts, interval_ts;
20119b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang        float current_fps, average_fps;
20219b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang
20319b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang        t.tv_sec = t.tv_usec = 0;
20419b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang        gettimeofday(&t, NULL);
20519b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang
20619b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang        current_ts =
20719b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang            (nsecs_t)t.tv_sec * 1000000000 + (nsecs_t)t.tv_usec * 1000;
20819b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang        interval_ts = current_ts - lastTs;
20919b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang        lastTs = current_ts;
21019b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang
21119b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang        current_fps = (float)1000000000 / (float)interval_ts;
21219b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang        average_fps = (current_fps + lastFps) / 2;
21319b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang        lastFps = current_fps;
21419b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang
21519b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang        LOGV("FPS = %2.1f\n", average_fps);
21619b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang    }
21719b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang#endif
21819b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang
21919b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sangout:
22019b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang
22119b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang    if (retains[OUTPORT_INDEX] != BUFFER_RETAIN_GETAGAIN) {
22219b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang        buffers[OUTPORT_INDEX]->nFilledLen = outfilledlen;
22319b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang        buffers[OUTPORT_INDEX]->nTimeStamp = outtimestamp;
22419b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang        buffers[OUTPORT_INDEX]->nFlags = outflags;
22519b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang    }
22619b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang
22719b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang    if (retains[INPORT_INDEX] == BUFFER_RETAIN_NOT_RETAIN ||
22819b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang            retains[INPORT_INDEX] == BUFFER_RETAIN_ACCUMULATE ) {
22919b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang        mFrameInputCount ++;
23019b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang    }
23119b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang
23219b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang    if (retains[OUTPORT_INDEX] == BUFFER_RETAIN_NOT_RETAIN)
23319b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang        mFrameOutputCount ++;
23419b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang
23519b2ab9f325bdbf3afe530e943fa5a0c0020b308Shuduo Sang    return oret;
236e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu
237e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu}
238e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu
239e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy QiuOMX_ERRORTYPE OMXVideoEncoderMPEG4::BuildHandlerList(void) {
240e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu    OMXVideoEncoderBase::BuildHandlerList();
241e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu    AddHandler(OMX_IndexParamVideoMpeg4, GetParamVideoMpeg4, SetParamVideoMpeg4);
24230736b960fca5337c0e89021f2ff8779475b4ef8Zhao Liang    AddHandler(OMX_IndexParamVideoProfileLevelQuerySupported, GetParamVideoProfileLevelQuerySupported, SetParamVideoProfileLevelQuerySupported);
243e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu    return OMX_ErrorNone;
244e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu}
245e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu
246cae415b84125ffc14bfdb121394a60b74d506f48Yanli
247cae415b84125ffc14bfdb121394a60b74d506f48YanliOMX_ERRORTYPE OMXVideoEncoderMPEG4::GetParamVideoProfileLevelQuerySupported(OMX_PTR pStructure) {
248cae415b84125ffc14bfdb121394a60b74d506f48Yanli    OMX_ERRORTYPE ret;
249cae415b84125ffc14bfdb121394a60b74d506f48Yanli    OMX_VIDEO_PARAM_PROFILELEVELTYPE *p = (OMX_VIDEO_PARAM_PROFILELEVELTYPE *)pStructure;
250cae415b84125ffc14bfdb121394a60b74d506f48Yanli    CHECK_TYPE_HEADER(p);
251cae415b84125ffc14bfdb121394a60b74d506f48Yanli    CHECK_PORT_INDEX(p, OUTPORT_INDEX);
252cae415b84125ffc14bfdb121394a60b74d506f48Yanli
253cae415b84125ffc14bfdb121394a60b74d506f48Yanli    struct ProfileLevelTable {
254cae415b84125ffc14bfdb121394a60b74d506f48Yanli        OMX_U32 profile;
255cae415b84125ffc14bfdb121394a60b74d506f48Yanli        OMX_U32 level;
256cae415b84125ffc14bfdb121394a60b74d506f48Yanli    } plTable[] = {
257cae415b84125ffc14bfdb121394a60b74d506f48Yanli        {OMX_VIDEO_MPEG4ProfileSimple, OMX_VIDEO_MPEG4Level5},
258cae415b84125ffc14bfdb121394a60b74d506f48Yanli    };
259cae415b84125ffc14bfdb121394a60b74d506f48Yanli
260cae415b84125ffc14bfdb121394a60b74d506f48Yanli    OMX_U32 count = sizeof(plTable)/sizeof(ProfileLevelTable);
261cae415b84125ffc14bfdb121394a60b74d506f48Yanli    CHECK_ENUMERATION_RANGE(p->nProfileIndex,count);
262cae415b84125ffc14bfdb121394a60b74d506f48Yanli
263cae415b84125ffc14bfdb121394a60b74d506f48Yanli    p->eProfile = plTable[p->nProfileIndex].profile;
264cae415b84125ffc14bfdb121394a60b74d506f48Yanli    p->eLevel = plTable[p->nProfileIndex].level;
265cae415b84125ffc14bfdb121394a60b74d506f48Yanli
266cae415b84125ffc14bfdb121394a60b74d506f48Yanli    return OMX_ErrorNone;
267cae415b84125ffc14bfdb121394a60b74d506f48Yanli}
268cae415b84125ffc14bfdb121394a60b74d506f48Yanli
2690541f11be2f78b558476772f1f877e140f9d844bywanOMX_ERRORTYPE OMXVideoEncoderMPEG4::SetParamVideoProfileLevelQuerySupported(OMX_PTR) {
270cae415b84125ffc14bfdb121394a60b74d506f48Yanli    LOGW("SetParamVideoMpeg4ProfileLevel is not supported.");
271cae415b84125ffc14bfdb121394a60b74d506f48Yanli    return OMX_ErrorUnsupportedSetting;
272cae415b84125ffc14bfdb121394a60b74d506f48Yanli}
273cae415b84125ffc14bfdb121394a60b74d506f48Yanli
274e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy QiuOMX_ERRORTYPE OMXVideoEncoderMPEG4::GetParamVideoMpeg4(OMX_PTR pStructure) {
275e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu    OMX_ERRORTYPE ret;
276e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu    OMX_VIDEO_PARAM_MPEG4TYPE *p = (OMX_VIDEO_PARAM_MPEG4TYPE *)pStructure;
277e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu    CHECK_TYPE_HEADER(p);
278e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu    CHECK_PORT_INDEX(p, OUTPORT_INDEX);
279e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu
280e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu    memcpy(p, &mParamMpeg4, sizeof(*p));
281e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu    return OMX_ErrorNone;
282e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu}
283e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu
284e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy QiuOMX_ERRORTYPE OMXVideoEncoderMPEG4::SetParamVideoMpeg4(OMX_PTR pStructure) {
285e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu    OMX_ERRORTYPE ret;
286e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu    OMX_VIDEO_PARAM_MPEG4TYPE *p = (OMX_VIDEO_PARAM_MPEG4TYPE *)pStructure;
287e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu    CHECK_TYPE_HEADER(p);
288e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu    CHECK_PORT_INDEX(p, OUTPORT_INDEX);
289e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu    CHECK_SET_PARAM_STATE();
290e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu
291e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu    // TODO: do we need to check if port is enabled?
292e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu    // TODO: see SetPortMpeg4Param implementation - Can we make simple copy????
293e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu    memcpy(&mParamMpeg4, p, sizeof(mParamMpeg4));
294e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu    return OMX_ErrorNone;
295e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu}
296e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu
297e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu
298e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy QiuDECLARE_OMX_COMPONENT("OMX.Intel.VideoEncoder.MPEG4", "video_encoder.mpeg4", OMXVideoEncoderMPEG4);
299e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu
300e7ace334fb7c64f7b32aa3746e5a11bcefa60bfbAndy Qiu
301