/******************************************************************************
*
* Copyright (C) 2015 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
*/
#include <stdio.h>
#include "iv_datatypedef.h"
#include "iv.h"
#include "icv.h"
#include "ideint.h"
#include "impeg2_buf_mgr.h"
#include "impeg2_disp_mgr.h"
#include "impeg2_defs.h"
#include "impeg2_platform_macros.h"
#include "impeg2_inter_pred.h"
#include "impeg2_idct.h"
#include "impeg2_globals.h"
#include "impeg2_mem_func.h"
#include "impeg2_format_conv.h"
#include "impeg2_macros.h"
#include "impeg2d.h"
#include "impeg2d_bitstream.h"
#include "impeg2d_structs.h"
#include "impeg2d_globals.h"
#include "impeg2d_mc.h"
#include "impeg2d_pic_proc.h"
#include "impeg2d_deinterlace.h"
typedef struct
{
IVD_ARCH_T ivd_arch;
ICV_ARCH_T icv_arch;
}arch_map_t;
static const arch_map_t gas_impeg2d_arch_mapping[] =
{
{ARCH_ARM_NONEON, ICV_ARM_NONEON},
{ARCH_ARM_A9Q, ICV_ARM_A9Q},
{ARCH_ARM_A9A, ICV_ARM_A9A},
{ARCH_ARM_A9, ICV_ARM_A9},
{ARCH_ARM_A7, ICV_ARM_A7},
{ARCH_ARM_A5, ICV_ARM_A5},
{ARCH_ARM_A15, ICV_ARM_A15},
{ARCH_ARM_NEONINTR, ICV_ARM_NEONINTR},
{ARCH_ARMV8_GENERIC, ICV_ARMV8_GENERIC},
{ARCH_X86_GENERIC, ICV_X86_GENERIC},
{ARCH_X86_SSSE3, ICV_X86_SSSE3},
{ARCH_X86_SSE42, ICV_X86_SSE42},
{ARCH_X86_AVX2, ICV_X86_AVX2},
{ARCH_MIPS_GENERIC, ICV_MIPS_GENERIC},
{ARCH_MIPS_32, ICV_MIPS_32}
};
static void impeg2d_get_pic(icv_pic_t *ps_dst,
UWORD8 *pu1_buf_y,
UWORD8 *pu1_buf_u,
UWORD8 *pu1_buf_v,
WORD32 wd,
WORD32 ht,
WORD32 strd)
{
ps_dst->ai4_wd[0] = wd;
ps_dst->ai4_wd[1] = wd / 2;
ps_dst->ai4_wd[2] = wd / 2;
ps_dst->ai4_ht[0] = ht;
ps_dst->ai4_ht[1] = ht / 2;
ps_dst->ai4_ht[2] = ht / 2;
ps_dst->ai4_strd[0] = strd;
ps_dst->ai4_strd[1] = strd / 2;
ps_dst->ai4_strd[2] = strd / 2;
ps_dst->apu1_buf[0] = pu1_buf_y;
ps_dst->apu1_buf[1] = pu1_buf_u;
ps_dst->apu1_buf[2] = pu1_buf_v;
ps_dst->e_color_fmt = ICV_YUV420P;
}
static void impeg2d_get_flds(icv_pic_t *ps_frm,
icv_pic_t *ps_top_fld,
icv_pic_t *ps_bot_fld)
{
ps_top_fld->ai4_wd[0] = ps_frm->ai4_wd[0];
ps_top_fld->ai4_wd[1] = ps_frm->ai4_wd[1];
ps_top_fld->ai4_wd[2] = ps_frm->ai4_wd[2];
ps_top_fld->ai4_ht[0] = ps_frm->ai4_ht[0] / 2;
ps_top_fld->ai4_ht[1] = ps_frm->ai4_ht[1] / 2;
ps_top_fld->ai4_ht[2] = ps_frm->ai4_ht[2] / 2;
ps_top_fld->ai4_strd[0] = ps_frm->ai4_strd[0] * 2;
ps_top_fld->ai4_strd[1] = ps_frm->ai4_strd[1] * 2;
ps_top_fld->ai4_strd[2] = ps_frm->ai4_strd[2] * 2;
ps_top_fld->e_color_fmt = ps_frm->e_color_fmt;
/* Copy top field structure to bottom field, since properties of both fields are same */
*ps_bot_fld = *ps_top_fld;
/* Initialize the addresses for top field */
ps_top_fld->apu1_buf[0] = ps_frm->apu1_buf[0];
ps_top_fld->apu1_buf[1] = ps_frm->apu1_buf[1];
ps_top_fld->apu1_buf[2] = ps_frm->apu1_buf[2];
/* Initialize the addresses for bottom field */
ps_bot_fld->apu1_buf[0] = ps_frm->apu1_buf[0] + ps_frm->ai4_strd[0];
ps_bot_fld->apu1_buf[1] = ps_frm->apu1_buf[1] + ps_frm->ai4_strd[1];
ps_bot_fld->apu1_buf[2] = ps_frm->apu1_buf[2] + ps_frm->ai4_strd[2];
return;
}
static ICV_ARCH_T impeg2d_get_arch(IVD_ARCH_T e_arch)
{
ICV_ARCH_T ret_arch;
WORD32 num_entries, i;
ret_arch = ICV_ARM_A9;
num_entries = sizeof(gas_impeg2d_arch_mapping) / sizeof(gas_impeg2d_arch_mapping[0]);
for(i = 0; i < num_entries; i++)
{
if(e_arch == gas_impeg2d_arch_mapping[i].ivd_arch)
{
ret_arch = gas_impeg2d_arch_mapping[i].icv_arch;
break;
}
}
return ret_arch;
}
/******************************************************************************
* Function Name : impeg2d_deinterlace
*
* Description : Deinterlace current picture
*
* Arguments :
* dec : Decoder Context
*
* Values Returned : 0 on success, -1 on error
******************************************************************************/
WORD32 impeg2d_deint_ctxt_size(void)
{
return ideint_ctxt_size();
}
/******************************************************************************
* Function Name : impeg2d_deinterlace
*
* Description : Deinterlace current picture
*
* Arguments :
* dec : Decoder Context
*
* Values Returned : 0 on success, -1 on error
******************************************************************************/
WORD32 impeg2d_deinterlace(dec_state_t *ps_dec,
pic_buf_t *ps_src_pic,
iv_yuv_buf_t *ps_disp_frm_buf,
WORD32 start_row,
WORD32 num_rows)
{
icv_pic_t as_inp_flds[3];
IDEINT_ERROR_T ret;
icv_pic_t s_src_frm;
icv_pic_t s_dst_frm;
UWORD8 *pu1_dst_y, *pu1_dst_u, *pu1_dst_v;
ideint_params_t s_params;
if((NULL == ps_src_pic) || (NULL == ps_src_pic->pu1_y) || (0 == num_rows))
return -1;
s_params.e_arch = impeg2d_get_arch(ps_dec->e_processor_arch);
s_params.e_soc = ICV_SOC_GENERIC;
s_params.e_mode = IDEINT_MODE_SPATIAL;
s_params.i4_cur_fld_top = ps_dec->u2_top_field_first;
s_params.i4_disable_weave = 0;
s_params.pf_aligned_alloc = NULL;
s_params.pf_aligned_free = NULL;
impeg2d_get_pic(&s_src_frm, ps_src_pic->pu1_y, ps_src_pic->pu1_u,
ps_src_pic->pu1_v, ps_dec->u2_horizontal_size,
ps_dec->u2_vertical_size, ps_dec->u2_frame_width);
impeg2d_get_flds(&s_src_frm, &as_inp_flds[1], &as_inp_flds[2]);
if(ps_dec->ps_deint_pic)
{
icv_pic_t s_prv_frm;
icv_pic_t s_fld;
impeg2d_get_pic(&s_prv_frm, ps_dec->ps_deint_pic->pu1_y,
ps_dec->ps_deint_pic->pu1_u,
ps_dec->ps_deint_pic->pu1_v, ps_dec->u2_horizontal_size,
ps_dec->u2_vertical_size, ps_dec->u2_frame_width);
impeg2d_get_flds(&s_prv_frm, &s_fld, &as_inp_flds[0]);
}
else
{
as_inp_flds[0].apu1_buf[0] = NULL;
as_inp_flds[0].apu1_buf[1] = NULL;
as_inp_flds[0].apu1_buf[2] = NULL;
}
pu1_dst_y = ps_disp_frm_buf->pv_y_buf;
pu1_dst_u = ps_disp_frm_buf->pv_u_buf;
pu1_dst_v = ps_disp_frm_buf->pv_v_buf;
/* Use intermediate buffer as output to deinterlacer,
* if color format is not 420P
*/
if(IV_YUV_420P != ps_dec->i4_chromaFormat)
{
UWORD8 *pu1_buf_y;
UWORD8 *pu1_buf_u;
UWORD8 *pu1_buf_v;
WORD32 wd = ALIGN16(ps_dec->u2_horizontal_size);
WORD32 ht = ALIGN16(ps_dec->u2_vertical_size);
pu1_buf_y = ps_dec->pu1_deint_fmt_buf;
pu1_buf_u = pu1_buf_y + wd * ht;
pu1_buf_v = pu1_buf_u + wd * ht / 4;
pu1_dst_u = pu1_buf_u;
pu1_dst_v = pu1_buf_v;
if((ps_dec->i4_chromaFormat != IV_YUV_420SP_UV) &&
(ps_dec->i4_chromaFormat != IV_YUV_420SP_VU))
{
pu1_dst_y = pu1_buf_y;
}
}
impeg2d_get_pic(&s_dst_frm, pu1_dst_y, pu1_dst_u, pu1_dst_v,
ps_dec->u2_horizontal_size, ps_dec->u2_vertical_size,
ps_dec->u4_frm_buf_stride);
ret = ideint_process(ps_dec->pv_deinterlacer_ctxt, &as_inp_flds[0],
&as_inp_flds[1], &as_inp_flds[2], &s_dst_frm,
&s_params, start_row, num_rows);
if(IDEINT_ERROR_NONE != ret)
{
return -1;
}
/* Format convert deinterlacer output if required*/
if(IV_YUV_420P != ps_dec->i4_chromaFormat)
{
pic_buf_t s_src_pic;
s_src_pic = *ps_src_pic;
s_src_pic.pu1_y = pu1_dst_y;
s_src_pic.pu1_u = pu1_dst_u;
s_src_pic.pu1_v = pu1_dst_v;
impeg2d_format_convert(ps_dec,
&s_src_pic,
ps_disp_frm_buf,
start_row,
num_rows);
}
return 0;
}