1 /******************************************************************************
2 *
3 * Copyright (C) 2015 The Android Open Source Project
4 *
5 * Licensed under the Apache License, Version 2.0 (the "License");
6 * you may not use this file except in compliance with the License.
7 * You may obtain a copy of the License at:
8 *
9 * http://www.apache.org/licenses/LICENSE-2.0
10 *
11 * Unless required by applicable law or agreed to in writing, software
12 * distributed under the License is distributed on an "AS IS" BASIS,
13 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14 * See the License for the specific language governing permissions and
15 * limitations under the License.
16 *
17 *****************************************************************************
18 * Originally developed and contributed by Ittiam Systems Pvt. Ltd, Bangalore
19 */
20 #include <stdio.h>
21 #include "iv_datatypedef.h"
22 #include "iv.h"
23
24 #include "icv.h"
25 #include "ideint.h"
26
27 #include "impeg2_buf_mgr.h"
28 #include "impeg2_disp_mgr.h"
29 #include "impeg2_defs.h"
30 #include "impeg2_platform_macros.h"
31 #include "impeg2_inter_pred.h"
32 #include "impeg2_idct.h"
33 #include "impeg2_globals.h"
34 #include "impeg2_mem_func.h"
35 #include "impeg2_format_conv.h"
36 #include "impeg2_macros.h"
37
38 #include "impeg2d.h"
39 #include "impeg2d_bitstream.h"
40 #include "impeg2d_structs.h"
41 #include "impeg2d_globals.h"
42 #include "impeg2d_mc.h"
43 #include "impeg2d_pic_proc.h"
44 #include "impeg2d_deinterlace.h"
45
46 typedef struct
47 {
48 IVD_ARCH_T ivd_arch;
49 ICV_ARCH_T icv_arch;
50 }arch_map_t;
51
52 static const arch_map_t gas_impeg2d_arch_mapping[] =
53 {
54 {ARCH_ARM_NONEON, ICV_ARM_NONEON},
55 {ARCH_ARM_A9Q, ICV_ARM_A9Q},
56 {ARCH_ARM_A9A, ICV_ARM_A9A},
57 {ARCH_ARM_A9, ICV_ARM_A9},
58 {ARCH_ARM_A7, ICV_ARM_A7},
59 {ARCH_ARM_A5, ICV_ARM_A5},
60 {ARCH_ARM_A15, ICV_ARM_A15},
61 {ARCH_ARM_NEONINTR, ICV_ARM_NEONINTR},
62 {ARCH_ARMV8_GENERIC, ICV_ARMV8_GENERIC},
63 {ARCH_X86_GENERIC, ICV_X86_GENERIC},
64 {ARCH_X86_SSSE3, ICV_X86_SSSE3},
65 {ARCH_X86_SSE42, ICV_X86_SSE42},
66 {ARCH_X86_AVX2, ICV_X86_AVX2},
67 {ARCH_MIPS_GENERIC, ICV_MIPS_GENERIC},
68 {ARCH_MIPS_32, ICV_MIPS_32},
69 {ARCH_RISCV_GENERIC, ICV_RISCV_GENERIC},
70 };
71
72
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)73 static void impeg2d_get_pic(icv_pic_t *ps_dst,
74 UWORD8 *pu1_buf_y,
75 UWORD8 *pu1_buf_u,
76 UWORD8 *pu1_buf_v,
77 WORD32 wd,
78 WORD32 ht,
79 WORD32 strd)
80 {
81 ps_dst->ai4_wd[0] = wd;
82 ps_dst->ai4_wd[1] = wd / 2;
83 ps_dst->ai4_wd[2] = wd / 2;
84
85 ps_dst->ai4_ht[0] = ht;
86 ps_dst->ai4_ht[1] = ht / 2;
87 ps_dst->ai4_ht[2] = ht / 2;
88
89 ps_dst->ai4_strd[0] = strd;
90 ps_dst->ai4_strd[1] = strd / 2;
91 ps_dst->ai4_strd[2] = strd / 2;
92
93 ps_dst->apu1_buf[0] = pu1_buf_y;
94 ps_dst->apu1_buf[1] = pu1_buf_u;
95 ps_dst->apu1_buf[2] = pu1_buf_v;
96
97 ps_dst->e_color_fmt = ICV_YUV420P;
98 }
impeg2d_get_flds(icv_pic_t * ps_frm,icv_pic_t * ps_top_fld,icv_pic_t * ps_bot_fld)99 static void impeg2d_get_flds(icv_pic_t *ps_frm,
100 icv_pic_t *ps_top_fld,
101 icv_pic_t *ps_bot_fld)
102 {
103 ps_top_fld->ai4_wd[0] = ps_frm->ai4_wd[0];
104 ps_top_fld->ai4_wd[1] = ps_frm->ai4_wd[1];
105 ps_top_fld->ai4_wd[2] = ps_frm->ai4_wd[2];
106
107 ps_top_fld->ai4_ht[0] = ps_frm->ai4_ht[0] / 2;
108 ps_top_fld->ai4_ht[1] = ps_frm->ai4_ht[1] / 2;
109 ps_top_fld->ai4_ht[2] = ps_frm->ai4_ht[2] / 2;
110
111 ps_top_fld->ai4_strd[0] = ps_frm->ai4_strd[0] * 2;
112 ps_top_fld->ai4_strd[1] = ps_frm->ai4_strd[1] * 2;
113 ps_top_fld->ai4_strd[2] = ps_frm->ai4_strd[2] * 2;
114
115 ps_top_fld->e_color_fmt = ps_frm->e_color_fmt;
116
117 /* Copy top field structure to bottom field, since properties of both fields are same */
118 *ps_bot_fld = *ps_top_fld;
119
120 /* Initialize the addresses for top field */
121 ps_top_fld->apu1_buf[0] = ps_frm->apu1_buf[0];
122 ps_top_fld->apu1_buf[1] = ps_frm->apu1_buf[1];
123 ps_top_fld->apu1_buf[2] = ps_frm->apu1_buf[2];
124
125 /* Initialize the addresses for bottom field */
126 ps_bot_fld->apu1_buf[0] = ps_frm->apu1_buf[0] + ps_frm->ai4_strd[0];
127 ps_bot_fld->apu1_buf[1] = ps_frm->apu1_buf[1] + ps_frm->ai4_strd[1];
128 ps_bot_fld->apu1_buf[2] = ps_frm->apu1_buf[2] + ps_frm->ai4_strd[2];
129
130 return;
131 }
132
133
impeg2d_get_arch(IVD_ARCH_T e_arch)134 static ICV_ARCH_T impeg2d_get_arch(IVD_ARCH_T e_arch)
135 {
136 ICV_ARCH_T ret_arch;
137 WORD32 num_entries, i;
138
139 ret_arch = ICV_ARM_A9;
140 num_entries = sizeof(gas_impeg2d_arch_mapping) / sizeof(gas_impeg2d_arch_mapping[0]);
141 for(i = 0; i < num_entries; i++)
142 {
143 if(e_arch == gas_impeg2d_arch_mapping[i].ivd_arch)
144 {
145 ret_arch = gas_impeg2d_arch_mapping[i].icv_arch;
146 break;
147 }
148 }
149 return ret_arch;
150 }
151
152 /******************************************************************************
153 * Function Name : impeg2d_deinterlace
154 *
155 * Description : Deinterlace current picture
156 *
157 * Arguments :
158 * dec : Decoder Context
159 *
160 * Values Returned : 0 on success, -1 on error
161 ******************************************************************************/
impeg2d_deint_ctxt_size(void)162 WORD32 impeg2d_deint_ctxt_size(void)
163 {
164 return ideint_ctxt_size();
165 }
166
167 /******************************************************************************
168 * Function Name : impeg2d_deinterlace
169 *
170 * Description : Deinterlace current picture
171 *
172 * Arguments :
173 * dec : Decoder Context
174 *
175 * Values Returned : 0 on success, -1 on error
176 ******************************************************************************/
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)177 WORD32 impeg2d_deinterlace(dec_state_t *ps_dec,
178 pic_buf_t *ps_src_pic,
179 iv_yuv_buf_t *ps_disp_frm_buf,
180 WORD32 start_row,
181 WORD32 num_rows)
182 {
183 icv_pic_t as_inp_flds[3];
184 IDEINT_ERROR_T ret;
185 icv_pic_t s_src_frm;
186 icv_pic_t s_dst_frm;
187 UWORD8 *pu1_dst_y, *pu1_dst_u, *pu1_dst_v;
188 ideint_params_t s_params;
189
190 if((NULL == ps_src_pic) || (NULL == ps_src_pic->pu1_y) || (0 == num_rows))
191 return -1;
192
193 s_params.e_arch = impeg2d_get_arch(ps_dec->e_processor_arch);
194 s_params.e_soc = ICV_SOC_GENERIC;
195 s_params.e_mode = IDEINT_MODE_SPATIAL;
196 s_params.i4_cur_fld_top = ps_dec->u2_top_field_first;
197 s_params.i4_disable_weave = 0;
198 s_params.pf_aligned_alloc = NULL;
199 s_params.pf_aligned_free = NULL;
200
201 impeg2d_get_pic(&s_src_frm, ps_src_pic->pu1_y, ps_src_pic->pu1_u,
202 ps_src_pic->pu1_v, ps_dec->u2_horizontal_size,
203 ps_dec->u2_vertical_size, ps_dec->u2_frame_width);
204 impeg2d_get_flds(&s_src_frm, &as_inp_flds[1], &as_inp_flds[2]);
205
206 if(ps_dec->ps_deint_pic)
207 {
208 icv_pic_t s_prv_frm;
209 icv_pic_t s_fld;
210 impeg2d_get_pic(&s_prv_frm, ps_dec->ps_deint_pic->pu1_y,
211 ps_dec->ps_deint_pic->pu1_u,
212 ps_dec->ps_deint_pic->pu1_v, ps_dec->u2_horizontal_size,
213 ps_dec->u2_vertical_size, ps_dec->u2_frame_width);
214 impeg2d_get_flds(&s_prv_frm, &s_fld, &as_inp_flds[0]);
215 }
216 else
217 {
218 as_inp_flds[0].apu1_buf[0] = NULL;
219 as_inp_flds[0].apu1_buf[1] = NULL;
220 as_inp_flds[0].apu1_buf[2] = NULL;
221 }
222
223 pu1_dst_y = ps_disp_frm_buf->pv_y_buf;
224 pu1_dst_u = ps_disp_frm_buf->pv_u_buf;
225 pu1_dst_v = ps_disp_frm_buf->pv_v_buf;
226
227 /* Use intermediate buffer as output to deinterlacer,
228 * if color format is not 420P
229 */
230 if(IV_YUV_420P != ps_dec->i4_chromaFormat)
231 {
232 UWORD8 *pu1_buf_y;
233 UWORD8 *pu1_buf_u;
234 UWORD8 *pu1_buf_v;
235 WORD32 wd = ALIGN16(ps_dec->u2_horizontal_size);
236 WORD32 ht = ALIGN16(ps_dec->u2_vertical_size);
237
238 pu1_buf_y = ps_dec->pu1_deint_fmt_buf;
239 pu1_buf_u = pu1_buf_y + wd * ht;
240 pu1_buf_v = pu1_buf_u + wd * ht / 4;
241
242 pu1_dst_u = pu1_buf_u;
243 pu1_dst_v = pu1_buf_v;
244
245 if((ps_dec->i4_chromaFormat != IV_YUV_420SP_UV) &&
246 (ps_dec->i4_chromaFormat != IV_YUV_420SP_VU))
247 {
248 pu1_dst_y = pu1_buf_y;
249 }
250
251 }
252 impeg2d_get_pic(&s_dst_frm, pu1_dst_y, pu1_dst_u, pu1_dst_v,
253 ps_dec->u2_horizontal_size, ps_dec->u2_vertical_size,
254 ps_dec->u4_frm_buf_stride);
255
256
257 ret = ideint_process(ps_dec->pv_deinterlacer_ctxt, &as_inp_flds[0],
258 &as_inp_flds[1], &as_inp_flds[2], &s_dst_frm,
259 &s_params, start_row, num_rows);
260
261 if(IDEINT_ERROR_NONE != ret)
262 {
263 return -1;
264 }
265
266 /* Format convert deinterlacer output if required*/
267 if(IV_YUV_420P != ps_dec->i4_chromaFormat)
268 {
269 pic_buf_t s_src_pic;
270
271 s_src_pic = *ps_src_pic;
272 s_src_pic.pu1_y = pu1_dst_y;
273 s_src_pic.pu1_u = pu1_dst_u;
274 s_src_pic.pu1_v = pu1_dst_v;
275
276 impeg2d_format_convert(ps_dec,
277 &s_src_pic,
278 ps_disp_frm_buf,
279 start_row,
280 num_rows);
281
282 }
283 return 0;
284
285 }
286