stagefright: add SoftAVCEnc using libavc

#ifeq ($(if $(wildcard external/libh264),1,0),1)

LOCAL_PATH := $(call my-dir)

include $(CLEAR_VARS)

LOCAL_MODULE            := libstagefright_soft_avcenc

LOCAL_MODULE_TAGS       := optional

LOCAL_STATIC_LIBRARIES  := libavcenc

LOCAL_SRC_FILES         := SoftAVCEnc.cpp

LOCAL_C_INCLUDES := $(TOP)/external/libavc/encoder

LOCAL_C_INCLUDES += $(TOP)/external/libavc/common

LOCAL_C_INCLUDES += $(TOP)/frameworks/av/media/libstagefright/include

LOCAL_C_INCLUDES += $(TOP)/frameworks/native/include/media/openmax

LOCAL_C_INCLUDES += $(TOP)/frameworks/av/media/libstagefright/include

LOCAL_C_INCLUDES += $(TOP)/frameworks/native/include/media/hardware

LOCAL_C_INCLUDES += $(TOP)/frameworks/native/include/media/openmax

LOCAL_SHARED_LIBRARIES  := libstagefright

LOCAL_SHARED_LIBRARIES  += libstagefright_omx

LOCAL_SHARED_LIBRARIES  += libstagefright_foundation

LOCAL_SHARED_LIBRARIES  += libutils

LOCAL_SHARED_LIBRARIES  += liblog

LOCAL_LDFLAGS := -Wl,-Bsymbolic

include $(BUILD_SHARED_LIBRARY)

#endif

/*

 * Copyright 2012 The Android Open Source Project

 *

 * Licensed under the Apache License, Version 2.0 (the "License");

 * you may not use this file except in compliance with the License.

 * You may obtain a copy of the License at

 *

 *      http://www.apache.org/licenses/LICENSE-2.0

 *

 * Unless required by applicable law or agreed to in writing, software

 * distributed under the License is distributed on an "AS IS" BASIS,

 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

 * See the License for the specific language governing permissions and

 * limitations under the License.

 */

#ifndef __SOFT_AVC_ENC_H__

#define __SOFT_AVC_ENC_H__

#include <media/stagefright/MediaBuffer.h>

#include <media/stagefright/foundation/ABase.h>

#include <utils/Vector.h>

#include "SoftVideoEncoderOMXComponent.h"

namespace android {

struct MediaBuffer;

#define CODEC_MAX_CORES          4

#define LEN_STATUS_BUFFER        (10  * 1024)

#define MAX_VBV_BUFF_SIZE        (120 * 16384)

#define MAX_NUM_IO_BUFS           3

#define DEFAULT_MAX_REF_FRM         1

#define DEFAULT_MAX_REORDER_FRM     0

#define DEFAULT_QP_MIN              10

#define DEFAULT_QP_MAX              40

#define DEFAULT_MAX_BITRATE         20000000

#define DEFAULT_MAX_SRCH_RANGE_X    256

#define DEFAULT_MAX_SRCH_RANGE_Y    256

#define DEFAULT_MAX_FRAMERATE       120000

#define DEFAULT_NUM_CORES           1

#define DEFAULT_NUM_CORES_PRE_ENC   0

#define DEFAULT_FPS                 30

#define DEFAULT_ENC_SPEED           IVE_NORMAL

#define DEFAULT_MEM_REC_CNT         0

#define DEFAULT_RECON_ENABLE        0

#define DEFAULT_CHKSUM_ENABLE       0

#define DEFAULT_START_FRM           0

#define DEFAULT_NUM_FRMS            0xFFFFFFFF

#define DEFAULT_INP_COLOR_FORMAT       IV_YUV_420SP_VU

#define DEFAULT_RECON_COLOR_FORMAT     IV_YUV_420P

#define DEFAULT_LOOPBACK            0

#define DEFAULT_SRC_FRAME_RATE      30

#define DEFAULT_TGT_FRAME_RATE      30

#define DEFAULT_MAX_WD              1920

#define DEFAULT_MAX_HT              1920

#define DEFAULT_MAX_LEVEL           40

#define DEFAULT_STRIDE              0

#define DEFAULT_WD                  1280

#define DEFAULT_HT                  720

#define DEFAULT_PSNR_ENABLE         0

#define DEFAULT_ME_SPEED            100

#define DEFAULT_ENABLE_FAST_SAD     0

#define DEFAULT_ENABLE_ALT_REF      0

#define DEFAULT_RC_MODE             IVE_RC_STORAGE

#define DEFAULT_BITRATE             6000000

#define DEFAULT_I_QP                22

#define DEFAULT_I_QP_MAX            DEFAULT_QP_MAX

#define DEFAULT_I_QP_MIN            DEFAULT_QP_MIN

#define DEFAULT_P_QP                28

#define DEFAULT_P_QP_MAX            DEFAULT_QP_MAX

#define DEFAULT_P_QP_MIN            DEFAULT_QP_MIN

#define DEFAULT_B_QP                22

#define DEFAULT_B_QP_MAX            DEFAULT_QP_MAX

#define DEFAULT_B_QP_MIN            DEFAULT_QP_MIN

#define DEFAULT_AIR                 IVE_AIR_MODE_NONE

#define DEFAULT_AIR_REFRESH_PERIOD  30

#define DEFAULT_SRCH_RNG_X          64

#define DEFAULT_SRCH_RNG_Y          48

#define DEFAULT_I_INTERVAL          30

#define DEFAULT_IDR_INTERVAL        1000

#define DEFAULT_B_FRAMES            0

#define DEFAULT_DISABLE_DEBLK_LEVEL 0

#define DEFAULT_HPEL                1

#define DEFAULT_QPEL                1

#define DEFAULT_I4                  1

#define DEFAULT_EPROFILE            IV_PROFILE_BASE

#define DEFAULT_SLICE_MODE          IVE_SLICE_MODE_NONE

#define DEFAULT_SLICE_PARAM         256

#define DEFAULT_ARCH                ARCH_ARM_A9Q

#define DEFAULT_SOC                 SOC_GENERIC

#define DEFAULT_INTRA4x4            0

#define STRLENGTH                   500

#define MIN(a, b) ((a) < (b))? (a) : (b)

#define MAX(a, b) ((a) > (b))? (a) : (b)

#define ALIGN16(x) ((((x) + 15) >> 4) << 4)

#define ALIGN128(x) ((((x) + 127) >> 7) << 7)

#define ALIGN4096(x) ((((x) + 4095) >> 12) << 12)

/** Used to remove warnings about unused parameters */

#define UNUSED(x) ((void)(x))

/** Get time */

#define GETTIME(a, b) gettimeofday(a, b);

/** Compute difference between start and end */

#define TIME_DIFF(start, end, diff) \

    diff = ((end.tv_sec - start.tv_sec) * 1000000) + \

            (end.tv_usec - start.tv_usec);

#define ive_aligned_malloc(alignment, size) memalign(alignment, size)

#define ive_aligned_free(buf) free(buf)

struct SoftAVC : public SoftVideoEncoderOMXComponent {

    SoftAVC(

            const char *name,

            const OMX_CALLBACKTYPE *callbacks,

            OMX_PTR appData,

            OMX_COMPONENTTYPE **component);

    // Override SimpleSoftOMXComponent methods

    virtual OMX_ERRORTYPE internalGetParameter(

            OMX_INDEXTYPE index, OMX_PTR params);

    virtual OMX_ERRORTYPE internalSetParameter(

            OMX_INDEXTYPE index, const OMX_PTR params);

    virtual void onQueueFilled(OMX_U32 portIndex);

protected:

    virtual ~SoftAVC();

private:

    enum {

        kNumBuffers = 2,

    };

    // OMX input buffer's timestamp and flags

    typedef struct {

        int64_t mTimeUs;

        int32_t mFlags;

    } InputBufferInfo;

    int32_t  mStride;

    uint32_t mFrameRate;

    struct timeval mTimeStart;   // Time at the start of decode()

    struct timeval mTimeEnd;     // Time at the end of decode()

    // If a request for a change it bitrate has been received.

    bool mBitrateUpdated;

    bool mKeyFrameRequested;

#ifdef FILE_DUMP_ENABLE

    char mInFile[200];

    char mOutFile[200];

#endif /* FILE_DUMP_ENABLE */

    IV_COLOR_FORMAT_T mIvVideoColorFormat;

    int32_t  mIDRFrameRefreshIntervalInSec;

    IV_PROFILE_T mAVCEncProfile;

    WORD32   mAVCEncLevel;

    int64_t  mNumInputFrames;

    int64_t  mPrevTimestampUs;

    bool     mStarted;

    bool     mSpsPpsHeaderReceived;

    bool     mSawInputEOS;

    bool     mSignalledError;

    bool     mIntra4x4;

    bool     mEnableFastSad;

    bool     mEnableAltRef;

    bool    mReconEnable;

    bool    mPSNREnable;

    IVE_SPEED_CONFIG     mEncSpeed;

    uint8_t *mConversionBuffer;

    iv_obj_t *mCodecCtx;         // Codec context

    iv_mem_rec_t *mMemRecords;   // Memory records requested by the codec

    size_t mNumMemRecords;       // Number of memory records requested by codec

    size_t mNumCores;            // Number of cores used by the codec

    UWORD32 mHeaderGenerated;

    IV_ARCH_T mArch;

    IVE_SLICE_MODE_T mSliceMode;

    UWORD32 mSliceParam;

    bool mHalfPelEnable;

    UWORD32 mIInterval;

    UWORD32 mIDRInterval;

    UWORD32 mDisableDeblkLevel;

    IVE_AIR_MODE_T mAIRMode;

    UWORD32 mAIRRefreshPeriod;

    OMX_ERRORTYPE initEncParams();

    OMX_ERRORTYPE initEncoder();

    OMX_ERRORTYPE releaseEncoder();

    // Verifies the component role tried to be set to this OMX component is

    // strictly video_encoder.avc

    OMX_ERRORTYPE internalSetRoleParams(

        const OMX_PARAM_COMPONENTROLETYPE *role);

    // Updates bitrate to reflect port settings.

    OMX_ERRORTYPE internalSetBitrateParams(

        const OMX_VIDEO_PARAM_BITRATETYPE *bitrate);

    OMX_ERRORTYPE setConfig(

        OMX_INDEXTYPE index, const OMX_PTR _params);

    // Handles port definition changes.

    OMX_ERRORTYPE internalSetPortParams(

        const OMX_PARAM_PORTDEFINITIONTYPE *port);

    OMX_ERRORTYPE internalSetFormatParams(

        const OMX_VIDEO_PARAM_PORTFORMATTYPE *format);

    OMX_ERRORTYPE setFrameType(IV_PICTURE_CODING_TYPE_T  e_frame_type);

    OMX_ERRORTYPE setQp();

    OMX_ERRORTYPE setEncMode(IVE_ENC_MODE_T e_enc_mode);

    OMX_ERRORTYPE setDimensions();

    OMX_ERRORTYPE setNumCores();

    OMX_ERRORTYPE setFrameRate();

    OMX_ERRORTYPE setIpeParams();

    OMX_ERRORTYPE setBitRate();

    OMX_ERRORTYPE setAirParams();

    OMX_ERRORTYPE setMeParams();

    OMX_ERRORTYPE setGopParams();

    OMX_ERRORTYPE setProfileParams();

    OMX_ERRORTYPE setDeblockParams();

    OMX_ERRORTYPE setVbvParams();

    void logVersion();

    OMX_ERRORTYPE setEncodeArgs(

        ive_video_encode_ip_t *ps_encode_ip,

        ive_video_encode_op_t *ps_encode_op,

        OMX_BUFFERHEADERTYPE *inputBufferHeader,

        OMX_BUFFERHEADERTYPE *outputBufferHeader);

    DISALLOW_EVIL_CONSTRUCTORS(SoftAVC);

};

#ifdef FILE_DUMP_ENABLE

#define INPUT_DUMP_PATH     "/sdcard/media/avce_input"

#define INPUT_DUMP_EXT      "yuv"

#define OUTPUT_DUMP_PATH    "/sdcard/media/avce_output"

#define OUTPUT_DUMP_EXT     "h264"

#define GENERATE_FILE_NAMES() {                         \

    GETTIME(&mTimeStart, NULL);                         \

    strcpy(mInFile, "");                                \

    sprintf(mInFile, "%s_%ld.%ld.%s", INPUT_DUMP_PATH,  \

            mTimeStart.tv_sec, mTimeStart.tv_usec,      \

            INPUT_DUMP_EXT);                            \

    strcpy(mOutFile, "");                               \

    sprintf(mOutFile, "%s_%ld.%ld.%s", OUTPUT_DUMP_PATH,\

            mTimeStart.tv_sec, mTimeStart.tv_usec,      \

            OUTPUT_DUMP_EXT);                           \

}

#define CREATE_DUMP_FILE(m_filename) {                  \

    FILE *fp = fopen(m_filename, "wb");                 \

    if (fp != NULL) {                                   \

        ALOGD("Opened file %s", m_filename);            \

        fclose(fp);                                     \

    } else {                                            \

        ALOGD("Could not open file %s", m_filename);    \

    }                                                   \

}

#define DUMP_TO_FILE(m_filename, m_buf, m_size)         \

{                                                       \

    FILE *fp = fopen(m_filename, "ab");                 \

    if (fp != NULL && m_buf != NULL) {                  \

        int i;                                          \

        i = fwrite(m_buf, 1, m_size, fp);               \

        ALOGD("fwrite ret %d to write %d", i, m_size);  \

        if (i != (int)m_size) {                         \

            ALOGD("Error in fwrite, returned %d", i);   \

            perror("Error in write to file");           \

        }                                               \

        fclose(fp);                                     \

    } else {                                            \

        ALOGD("Could not write to file %s", m_filename);\

    }                                                   \

}

#else /* FILE_DUMP_ENABLE */

#define INPUT_DUMP_PATH

#define INPUT_DUMP_EXT

#define OUTPUT_DUMP_PATH

#define OUTPUT_DUMP_EXT

#define GENERATE_FILE_NAMES()

#define CREATE_DUMP_FILE(m_filename)

#define DUMP_TO_FILE(m_filename, m_buf, m_size)

#endif /* FILE_DUMP_ENABLE */

}  // namespace android

#endif  // __SOFT_AVC_ENC_H__

            <Limit name="bitrate" range="1-128000" />

        </MediaCodec>

        <MediaCodec name="OMX.google.h264.encoder" type="video/avc">

            <!-- profiles and levels:  ProfileBaseline : Level2 -->

            <Limit name="size" min="16x16" max="896x896" />

            <Limit name="alignment" value="16x16" />

            <!-- profiles and levels:  ProfileBaseline : Level41 -->

            <Limit name="size" min="16x16" max="1920x1088" />

            <Limit name="alignment" value="2x2" />

            <Limit name="block-size" value="16x16" />

            <Limit name="blocks-per-second" range="1-11880" />

            <Limit name="bitrate" range="1-2000000" />

            <Limit name="blocks-per-second" range="1-244800" />

            <Limit name="bitrate" range="1-12000000" />

        </MediaCodec>

        <MediaCodec name="OMX.google.mpeg4.encoder" type="video/mp4v-es">

            <!-- profiles and levels:  ProfileCore : Level2 -->

    { "OMX.google.amrwb.decoder", "amrdec", "audio_decoder.amrwb" },

    { "OMX.google.amrwb.encoder", "amrwbenc", "audio_encoder.amrwb" },

    { "OMX.google.h264.decoder", "avcdec", "video_decoder.avc" },

    { "OMX.google.h264.encoder", "h264enc", "video_encoder.avc" },

    { "OMX.google.h264.encoder", "avcenc", "video_encoder.avc" },

    { "OMX.google.hevc.decoder", "hevcdec", "video_decoder.hevc" },

    { "OMX.google.g711.alaw.decoder", "g711dec", "audio_decoder.g711alaw" },

    { "OMX.google.g711.mlaw.decoder", "g711dec", "audio_decoder.g711mlaw" },