xref: /frameworks/av/media/libstagefright/codecs/on2/h264dec/omxdl/reference/vc/m4p2/src/omxVCM4P2_FindMVpred.c

/*
 * Copyright (C) 2007-2008 ARM Limited
 *
 * 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.
 *
 */
/**
 *
 * File Name:  omxVCM4P2_FindMVpred.c
 * OpenMAX DL: v1.0.2
 * Revision:   9641
 * Date:       Thursday, February 7, 2008
 *
 *
 *
 *
 * Description:
 * Contains module for predicting MV of MB
 *
 */

#include "omxtypes.h"
#include "armOMX.h"
#include "omxVC.h"

#include "armCOMM.h"

/**
 * Function:  omxVCM4P2_FindMVpred   (6.2.3.1.1)
 *
 * Description:
 * Predicts a motion vector for the current block using the procedure
 * specified in [ISO14496-2], subclause 7.6.5.  The resulting predicted MV is
 * returned in pDstMVPred. If the parameter pDstMVPredME if is not NULL then
 * the set of three MV candidates used for prediction is also returned,
 * otherwise pDstMVPredMEis NULL upon return.
 *
 * Input Arguments:
 *
 *   pSrcMVCurMB - pointer to the MV buffer associated with the current Y
 *            macroblock; a value of NULL indicates unavailability.
 *   pSrcCandMV1 - pointer to the MV buffer containing the 4 MVs associated
 *            with the MB located to the left of the current MB; set to NULL
 *            if there is no MB to the left.
 *   pSrcCandMV2 - pointer to the MV buffer containing the 4 MVs associated
 *            with the MB located above the current MB; set to NULL if there
 *            is no MB located above the current MB.
 *   pSrcCandMV3 - pointer to the MV buffer containing the 4 MVs associated
 *            with the MB located to the right and above the current MB; set
 *            to NULL if there is no MB located to the above-right.
 *   iBlk - the index of block in the current macroblock
 *   pDstMVPredME - MV candidate return buffer;  if set to NULL then
 *            prediction candidate MVs are not returned and pDstMVPredME will
 *            be NULL upon function return; if pDstMVPredME is non-NULL then it
 *            must point to a buffer containing sufficient space for three
 *            return MVs.
 *
 * Output Arguments:
 *
 *   pDstMVPred - pointer to the predicted motion vector
 *   pDstMVPredME - if non-NULL upon input then pDstMVPredME  points upon
 *            return to a buffer containing the three motion vector candidates
 *            used for prediction as specified in [ISO14496-2], subclause
 *            7.6.5, otherwise if NULL upon input then pDstMVPredME is NULL
 *            upon output.
 *
 * Return Value:
 *
 *    OMX_Sts_NoErr - no error
 *    OMX_Sts_BadArgErr - bad arguments; returned under any of the following
 *              conditions:
 *    -    the pointer pDstMVPred is NULL
 *    -    the parameter iBlk does not fall into the range 0 <= iBlk<=3
 *
 */

OMXResult omxVCM4P2_FindMVpred(
     const OMXVCMotionVector* pSrcMVCurMB,
     const OMXVCMotionVector* pSrcCandMV1,
     const OMXVCMotionVector* pSrcCandMV2,
     const OMXVCMotionVector* pSrcCandMV3,
     OMXVCMotionVector* pDstMVPred,
     OMXVCMotionVector* pDstMVPredME,
     OMX_INT iBlk
 )
{
    OMXVCMotionVector CandMV;
	const OMXVCMotionVector *pCandMV1;
    const OMXVCMotionVector *pCandMV2;
    const OMXVCMotionVector *pCandMV3;

    /* Argument error checks */
	armRetArgErrIf(iBlk!=0 && pSrcMVCurMB == NULL, OMX_Sts_BadArgErr);
    armRetArgErrIf(pDstMVPred == NULL, OMX_Sts_BadArgErr);
    armRetArgErrIf((iBlk < 0) || (iBlk > 3), OMX_Sts_BadArgErr);

    CandMV.dx = CandMV.dy = 0;
	/* Based on the position of the block extract the motion vectors and
       the tranperancy status */


    /* Set the default value for these to be used if pSrcCandMV[1|2|3] == NULL */
    pCandMV1 = pCandMV2 = pCandMV3 = &CandMV;


    switch (iBlk)
    {
        case 0:
        {
            if(pSrcCandMV1 != NULL)
            {
			    pCandMV1 = &pSrcCandMV1[1];
			}
			if(pSrcCandMV2 != NULL)
            {
				pCandMV2 = &pSrcCandMV2[2];
			}
			if(pSrcCandMV3 != NULL)
            {
				pCandMV3 = &pSrcCandMV3[2];
			}
			if ((pSrcCandMV1 == NULL) && (pSrcCandMV2 == NULL))
            {
                pCandMV1 = pCandMV2 = pCandMV3;
            }
            else if((pSrcCandMV1 == NULL) && (pSrcCandMV3 == NULL))
            {
                pCandMV1 = pCandMV3 = pCandMV2;
            }
            else if((pSrcCandMV2 == NULL) && (pSrcCandMV3 == NULL))
            {
                pCandMV2 = pCandMV3 = pCandMV1;
            }
            break;
        }
        case 1:
        {
            pCandMV1 = &pSrcMVCurMB[0];
			if(pSrcCandMV2 != NULL)
            {
				pCandMV2 = &pSrcCandMV2[3];
			}
			if(pSrcCandMV3 != NULL)
            {
				pCandMV3 = &pSrcCandMV3[2];
			}
			if((pSrcCandMV2 == NULL) && (pSrcCandMV3 == NULL))
            {
                pCandMV2 = pCandMV3 = pCandMV1;
            }
            break;
        }
        case 2:
        {
            if(pSrcCandMV1 != NULL)
            {
				pCandMV1 = &pSrcCandMV1[3];
			}
			pCandMV2 = &pSrcMVCurMB[0];
			pCandMV3 = &pSrcMVCurMB[1];
			break;
        }
        case 3:
        {
            pCandMV1 = &pSrcMVCurMB[2];
			pCandMV2 = &pSrcMVCurMB[0];
			pCandMV3 = &pSrcMVCurMB[1];
			break;
        }
    }

    /* Find the median of the 3 candidate MV's */
    pDstMVPred->dx = armMedianOf3 (pCandMV1->dx, pCandMV2->dx, pCandMV3->dx);
    pDstMVPred->dy = armMedianOf3 (pCandMV1->dy, pCandMV2->dy, pCandMV3->dy);

    if (pDstMVPredME != NULL)
    {
        /* Store the candidate MV's into the pDstMVPredME, these can be used
           in the fast algorithm if implemented */
        pDstMVPredME[0].dx = pCandMV1->dx;
        pDstMVPredME[0].dy = pCandMV1->dy;
        pDstMVPredME[1].dx = pCandMV2->dx;
        pDstMVPredME[1].dy = pCandMV2->dy;
        pDstMVPredME[2].dx = pCandMV3->dx;
        pDstMVPredME[2].dy = pCandMV3->dy;
    }

    return OMX_Sts_NoErr;
}


/* End of file */