/******************************************************************************
 *
 * 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_fmt_conv.c
*
* @brief
*  Contains functions for format conversion or frame copy of output buffer
*
* @author
*  ittiam
*
* @par List of Functions:
*  - isvce_fmt_conv()
*
* @remarks
*  None
*
*******************************************************************************
*/
#include "ih264_typedefs.h"
#include "ih264_macros.h"
/* Dependencies of ih264_buf_mgr.h */
/* Dependencies of ih264_list.h */
#include "ih264_error.h"
/* Dependencies of ih264_common_tables.h */
#include "ih264_defs.h"
#include "ih264_structs.h"
#include "ih264_buf_mgr.h"
#include "ih264_common_tables.h"
#include "ih264_list.h"
#include "ih264_platform_macros.h"
#include "ih264_trans_data.h"
#include "ih264_size_defs.h"
/* Dependencies of ih264e_cabac_structs.h */
#include "ih264_cabac_tables.h"
/* Dependencies of ime_structs.h */
#include "ime_defs.h"
#include "ime_distortion_metrics.h"
/* Dependencies of ih264e_structs.h */
#include "iv2.h"
#include "ive2.h"
#include "ih264_defs.h"
#include "ih264_deblk_edge_filters.h"
#include "ih264_inter_pred_filters.h"
#include "ih264_structs.h"
#include "ih264_trans_quant_itrans_iquant.h"
/* Dependencies of ih264e_bitstream.h */
#include "ih264e_error.h"
#include "ih264e_bitstream.h"
#include "ih264e_cabac_structs.h"
#include "irc_cntrl_param.h"
#include "irc_frame_info_collector.h"
#include "ime_statistics.h"
#include "ime_structs.h"
/* Dependencies of 'ih264e_utils.h' */
#include "ih264e_defs.h"
#include "ih264e_structs.h"
#include "ih264e_fmt_conv.h"
#include "isvce_structs.h"

IH264E_ERROR_T isvce_fmt_conv(isvce_codec_t *ps_codec, svc_au_buf_t *ps_pic, UWORD8 *pu1_y_dst,
                              UWORD8 *pu1_u_dst, UWORD8 *pu1_v_dst, UWORD32 u4_dst_y_strd,
                              UWORD32 u4_dst_uv_strd, WORD32 cur_row, WORD32 num_rows)
{
    IH264E_ERROR_T ret = IH264E_SUCCESS;
    UWORD8 *pu1_y_src, *pu1_uv_src;
    UWORD8 *pu1_y_dst_tmp, *pu1_uv_dst_tmp;
    UWORD8 *pu1_u_dst_tmp, *pu1_v_dst_tmp;
    WORD32 is_u_first;
    UWORD8 *pu1_luma;
    UWORD8 *pu1_chroma;
    WORD32 wd;

    WORD32 src_y_strd;
    WORD32 src_uv_strd;

    WORD32 layer_id = ps_pic->u1_num_spatial_layers - 1;

    if(0 == num_rows)
    {
        return ret;
    }

    pu1_luma = ps_pic->ps_layer_yuv_buf_props[layer_id].as_component_bufs[0].pv_data;
    pu1_chroma = ps_pic->ps_layer_yuv_buf_props[layer_id].as_component_bufs[1].pv_data;

    src_y_strd = ps_pic->ps_layer_yuv_buf_props[layer_id].as_component_bufs[0].i4_data_stride;
    src_uv_strd = ps_pic->ps_layer_yuv_buf_props[layer_id].as_component_bufs[1].i4_data_stride;

    wd = ps_codec->s_cfg.u4_disp_wd;
    is_u_first = (IV_YUV_420SP_UV == ps_codec->e_codec_color_format) ? 1 : 0;

    /* In case of 420P output luma copy is disabled for shared mode */
    {
        pu1_y_src = pu1_luma + cur_row * src_y_strd;
        pu1_uv_src = pu1_chroma + (cur_row / 2) * src_uv_strd;

        pu1_y_dst_tmp = pu1_y_dst + cur_row * u4_dst_y_strd;
        pu1_uv_dst_tmp = pu1_u_dst + (cur_row / 2) * u4_dst_uv_strd;
        pu1_u_dst_tmp = pu1_u_dst + (cur_row / 2) * u4_dst_uv_strd;
        pu1_v_dst_tmp = pu1_v_dst + (cur_row / 2) * u4_dst_uv_strd;

        /* If the call is non-blocking and there are no rows to be copied then
         * return */
        /* In non-shared mode, reference buffers are in 420SP UV format,
         * if output also is in 420SP_UV, then just copy
         * if output is in 420SP_VU then swap UV values
         */
        if((IV_YUV_420SP_UV == ps_codec->s_cfg.e_recon_color_fmt) ||
           (IV_YUV_420SP_VU == ps_codec->s_cfg.e_recon_color_fmt))
        {
            ih264e_fmt_conv_420sp_to_420sp(pu1_y_src, pu1_uv_src, pu1_y_dst_tmp, pu1_uv_dst_tmp, wd,
                                           num_rows, ps_codec->i4_rec_strd, ps_codec->i4_rec_strd,
                                           u4_dst_y_strd, u4_dst_uv_strd);
        }
        else if(IV_YUV_420P == ps_codec->s_cfg.e_recon_color_fmt)
        {
            ih264e_fmt_conv_420sp_to_420p(pu1_y_src, pu1_uv_src, pu1_y_dst_tmp, pu1_u_dst_tmp,
                                          pu1_v_dst_tmp, wd, num_rows, ps_codec->i4_rec_strd,
                                          ps_codec->i4_rec_strd, u4_dst_y_strd, u4_dst_uv_strd,
                                          is_u_first, 0);
        }
    }
    return (ret);
}