/******************************************************************************
 *
 * Copyright (C) 2022 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.
 *
 *****************************************************************************
 * Originally developed and contributed by Ittiam Systems Pvt. Ltd, Bangalore
 */

/**
*******************************************************************************
* @file
*  isvce_rc_utils.c
*
* @brief
*  Contains get gpp function required by the SVC encoder
*
* @author
*  ittiam
*
* @par List of Functions:
*  - isvce_get_gpp()
*  - isvce_rc_utils_init()
*  - isvce_get_rc_utils_data_size()
*  - isvce_compute_gpp()
*  - isvce_get_gpp_function_selector()
*
* @remarks
*  None
*
*******************************************************************************
*/

#include "ih264_typedefs.h"
#include "ih264_macros.h"
#include "isvc_structs.h"
#include "isvce_rc_utils.h"
#include "isvce_rc_utils_private_defs.h"

/**
*******************************************************************************
*
* @brief
*   get gpp function
*
* @par Description:
*   computes gradient per pixel value for a given frame
*
* @param[in] ps_input_buf
*  pointer to yuv buffer properties
*
* @returns
*  calculated gpp value
*
* @remarks
*  none
*
*******************************************************************************
*/

static DOUBLE isvce_get_gpp(yuv_buf_props_t *ps_input_buf)
{
    UWORD32 i, j;

    DOUBLE d_gpp_y = 0;
    DOUBLE d_gpp_u = 0;
    DOUBLE d_gpp_v = 0;

    DOUBLE d_gpp = 0;

    UWORD32 u4_width = ps_input_buf->u4_width;
    UWORD32 u4_height = ps_input_buf->u4_height;

    UWORD8 *pu1_input_buf = (UWORD8 *) ps_input_buf->as_component_bufs[0].pv_data;
    WORD32 i4_input_stride = ps_input_buf->as_component_bufs[0].i4_data_stride;

    for(i = 0; i < u4_height - 1; i++)
    {
        for(j = 0; j < u4_width - 1; j++)
        {
            UWORD8 u1_cur_pix = pu1_input_buf[j];
            UWORD8 u1_bot_pix = pu1_input_buf[i4_input_stride + j];
            UWORD8 u1_right_pix = pu1_input_buf[j + 1];

            d_gpp_y += (ABS(u1_cur_pix - u1_bot_pix) + ABS(u1_cur_pix - u1_right_pix));
        }
        pu1_input_buf += i4_input_stride;
    }

    pu1_input_buf = (UWORD8 *) ps_input_buf->as_component_bufs[1].pv_data;
    i4_input_stride = ps_input_buf->as_component_bufs[1].i4_data_stride;

    for(i = 0; i < (u4_height >> 1) - 1; i++)
    {
        for(j = 0; j < u4_width - 2; j += 2)
        {
            UWORD8 u1_cur_pix = pu1_input_buf[j];
            UWORD8 u1_bot_pix = pu1_input_buf[i4_input_stride + j];
            UWORD8 u1_right_pix = pu1_input_buf[j + 2];

            d_gpp_u += (ABS(u1_cur_pix - u1_bot_pix) + ABS(u1_cur_pix - u1_right_pix));

            u1_cur_pix = pu1_input_buf[j + 1];
            u1_bot_pix = pu1_input_buf[i4_input_stride + j + 1];
            u1_right_pix = pu1_input_buf[j + 2 + 1];

            d_gpp_v += (ABS(u1_cur_pix - u1_bot_pix) + ABS(u1_cur_pix - u1_right_pix));
        }
        pu1_input_buf += i4_input_stride;
    }

    d_gpp_y /= (u4_width * u4_height);
    d_gpp_u /= ((u4_width >> 1) * (u4_height >> 1));
    d_gpp_v /= ((u4_width >> 1) * (u4_height >> 1));

    d_gpp = (DOUBLE) ((4 * d_gpp_y) + d_gpp_u + d_gpp_v) / 6;

    return d_gpp;
}

/**
*******************************************************************************
*
* @brief
*   gets the memory size required for compute gpp
*
* @par Description:
*   returns the memory required by the rc utils context and state structs
*   for allocation.
*
* @returns
*
* @remarks
*
*
*******************************************************************************
*/

UWORD32 isvce_get_rc_utils_data_size() { return sizeof(svc_rc_utils_state_t); }

/**
*******************************************************************************
*
* @brief
*   compute gpp process
*
* @par Description:
*   calls the function to compute gpp
*
* @param[in] ps_svc_rc_utils_ctxt
*  pointer to svc rc utils context
*
* @param[in] ps_input_buf
*  pointer to yuv buffer properties
*
* @returns
*  calculated gpp value
*
* @remarks
*  none
*
*******************************************************************************
*/

DOUBLE isvce_compute_gpp(svc_rc_utils_ctxt_t *ps_svc_rc_utils_ctxt, yuv_buf_props_t *ps_input_buf)
{
    svc_rc_utils_state_t *ps_rc_utils_state =
        (svc_rc_utils_state_t *) ps_svc_rc_utils_ctxt->pv_rc_utils_state;

    return ps_rc_utils_state->pf_get_gpp(ps_input_buf);
}

/**
*******************************************************************************
*
* @brief
*   selects which function to call for get gpp based on e_arch
*
* @par Description:
*
* @param[in] ps_rc_utils_state
*  pointer to svc rc utils state
*
* @param[in] e_arch
*  architecure type
*
* @returns
*
* @remarks
*
*******************************************************************************
*/

static void isvce_get_gpp_function_selector(svc_rc_utils_state_t *ps_rc_utils_state,
                                            IV_ARCH_T e_arch)
{
    switch(e_arch)
    {
#if defined(X86)
        case ARCH_X86_SSE42:
        {
            ps_rc_utils_state->pf_get_gpp = isvce_get_gpp_sse42;

            break;
        }
#elif defined(ARMV8)
        case ARCH_ARM_A53:
        case ARCH_ARM_A57:
        case ARCH_ARM_V8_NEON:
        {
            ps_rc_utils_state->pf_get_gpp = isvce_get_gpp_neon;

            break;
        }
#elif defined(ARM) && !defined(DISABLE_NEON)
        case ARCH_ARM_A9Q:
        case ARCH_ARM_A9A:
        case ARCH_ARM_A9:
        case ARCH_ARM_A7:
        case ARCH_ARM_A5:
        case ARCH_ARM_A15:
        {
            ps_rc_utils_state->pf_get_gpp = isvce_get_gpp_neon;

            break;
        }
#endif
        default:
        {
            ps_rc_utils_state->pf_get_gpp = isvce_get_gpp;

            break;
        }
    }
}

/**
*******************************************************************************
*
* @brief
*   initializes the rc utils context
*
* @par Description:
*   initializes the rc utils context
*
* @param[in] ps_svc_rc_utils_ctxt
*   pointer to svc rc utils context
*
* @param[in] ps_mem_rec
*   pointer to memory allocated to compute gpp process
*
* @param[in] e_arch
*   architecure type
*
* @returns
*
* @remarks
*  none
*
*******************************************************************************
*/

void isvce_rc_utils_init(svc_rc_utils_ctxt_t *ps_svc_rc_utils_ctxt, iv_mem_rec_t *ps_mem_rec,
                         IV_ARCH_T e_arch)
{
    svc_rc_utils_state_t *ps_rc_utils_state;

    UWORD8 *pu1_buf = (UWORD8 *) ps_mem_rec->pv_base;

    ps_rc_utils_state = (svc_rc_utils_state_t *) pu1_buf;

    ps_svc_rc_utils_ctxt->pv_rc_utils_state = ps_rc_utils_state;

    isvce_get_gpp_function_selector(ps_rc_utils_state, e_arch);
}