1*437bfbebSnyanmisaka /*
2*437bfbebSnyanmisaka * Copyright 2021 Rockchip Electronics Co. LTD
3*437bfbebSnyanmisaka *
4*437bfbebSnyanmisaka * Licensed under the Apache License, Version 2.0 (the "License");
5*437bfbebSnyanmisaka * you may not use this file except in compliance with the License.
6*437bfbebSnyanmisaka * You may obtain a copy of the License at
7*437bfbebSnyanmisaka *
8*437bfbebSnyanmisaka * http://www.apache.org/licenses/LICENSE-2.0
9*437bfbebSnyanmisaka *
10*437bfbebSnyanmisaka * Unless required by applicable law or agreed to in writing, software
11*437bfbebSnyanmisaka * distributed under the License is distributed on an "AS IS" BASIS,
12*437bfbebSnyanmisaka * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13*437bfbebSnyanmisaka * See the License for the specific language governing permissions and
14*437bfbebSnyanmisaka * limitations under the License.
15*437bfbebSnyanmisaka */
16*437bfbebSnyanmisaka
17*437bfbebSnyanmisaka #define MODULE_TAG "hal_avs2d_rkv"
18*437bfbebSnyanmisaka
19*437bfbebSnyanmisaka #include <string.h>
20*437bfbebSnyanmisaka #include <stdio.h>
21*437bfbebSnyanmisaka
22*437bfbebSnyanmisaka #include "mpp_log.h"
23*437bfbebSnyanmisaka #include "mpp_mem.h"
24*437bfbebSnyanmisaka #include "mpp_common.h"
25*437bfbebSnyanmisaka #include "mpp_debug.h"
26*437bfbebSnyanmisaka #include "mpp_bitput.h"
27*437bfbebSnyanmisaka
28*437bfbebSnyanmisaka #include "avs2d_syntax.h"
29*437bfbebSnyanmisaka #include "hal_avs2d_api.h"
30*437bfbebSnyanmisaka #include "hal_avs2d_rkv.h"
31*437bfbebSnyanmisaka #include "mpp_dec_cb_param.h"
32*437bfbebSnyanmisaka #include "vdpu34x_avs2d.h"
33*437bfbebSnyanmisaka
34*437bfbebSnyanmisaka #define VDPU34X_FAST_REG_SET_CNT (3)
35*437bfbebSnyanmisaka #define MAX_REF_NUM (8)
36*437bfbebSnyanmisaka #define AVS2_RKV_SHPH_SIZE (1408 / 8) /* bytes */
37*437bfbebSnyanmisaka #define AVS2_RKV_SCALIST_SIZE (80 + 128) /* bytes */
38*437bfbebSnyanmisaka #define VDPU34x_TOTAL_REG_CNT (278)
39*437bfbebSnyanmisaka
40*437bfbebSnyanmisaka #define AVS2_RKV_SHPH_ALIGNED_SIZE (MPP_ALIGN(AVS2_RKV_SHPH_SIZE, SZ_4K))
41*437bfbebSnyanmisaka #define AVS2_RKV_SCALIST_ALIGNED_SIZE (MPP_ALIGN(AVS2_RKV_SCALIST_SIZE, SZ_4K))
42*437bfbebSnyanmisaka #define AVS2_RKV_STREAM_INFO_SET_SIZE (AVS2_RKV_SHPH_ALIGNED_SIZE + \
43*437bfbebSnyanmisaka AVS2_RKV_SCALIST_ALIGNED_SIZE)
44*437bfbebSnyanmisaka #define AVS2_ALL_TBL_BUF_SIZE(cnt) (AVS2_RKV_STREAM_INFO_SET_SIZE * (cnt))
45*437bfbebSnyanmisaka #define AVS2_SHPH_OFFSET(pos) (AVS2_RKV_STREAM_INFO_SET_SIZE * (pos))
46*437bfbebSnyanmisaka #define AVS2_SCALIST_OFFSET(pos) (AVS2_SHPH_OFFSET(pos) + AVS2_RKV_SHPH_ALIGNED_SIZE)
47*437bfbebSnyanmisaka
48*437bfbebSnyanmisaka #define COLMV_COMPRESS_EN (1)
49*437bfbebSnyanmisaka #define COLMV_BLOCK_SIZE (16)
50*437bfbebSnyanmisaka #define COLMV_BYTES (16)
51*437bfbebSnyanmisaka
52*437bfbebSnyanmisaka typedef struct avs2d_buf_t {
53*437bfbebSnyanmisaka RK_U32 valid;
54*437bfbebSnyanmisaka RK_U32 offset_shph;
55*437bfbebSnyanmisaka RK_U32 offset_sclst;
56*437bfbebSnyanmisaka Vdpu34xAvs2dRegSet *regs;
57*437bfbebSnyanmisaka } Avs2dRkvBuf_t;
58*437bfbebSnyanmisaka
59*437bfbebSnyanmisaka typedef struct avs2d_reg_ctx_t {
60*437bfbebSnyanmisaka Avs2dRkvBuf_t reg_buf[VDPU34X_FAST_REG_SET_CNT];
61*437bfbebSnyanmisaka
62*437bfbebSnyanmisaka RK_U32 shph_offset;
63*437bfbebSnyanmisaka RK_U32 sclst_offset;
64*437bfbebSnyanmisaka
65*437bfbebSnyanmisaka Vdpu34xAvs2dRegSet *regs;
66*437bfbebSnyanmisaka
67*437bfbebSnyanmisaka RK_U8 shph_dat[AVS2_RKV_SHPH_SIZE];
68*437bfbebSnyanmisaka RK_U8 scalist_dat[AVS2_RKV_SCALIST_SIZE];
69*437bfbebSnyanmisaka
70*437bfbebSnyanmisaka MppBuffer bufs;
71*437bfbebSnyanmisaka RK_S32 bufs_fd;
72*437bfbebSnyanmisaka void *bufs_ptr;
73*437bfbebSnyanmisaka
74*437bfbebSnyanmisaka MppBuffer rcb_buf[VDPU34X_FAST_REG_SET_CNT];
75*437bfbebSnyanmisaka RK_S32 rcb_buf_size;
76*437bfbebSnyanmisaka Vdpu34xRcbInfo rcb_info[RCB_BUF_COUNT];
77*437bfbebSnyanmisaka RK_U32 reg_out[VDPU34x_TOTAL_REG_CNT];
78*437bfbebSnyanmisaka
79*437bfbebSnyanmisaka } Avs2dRkvRegCtx_t;
80*437bfbebSnyanmisaka
81*437bfbebSnyanmisaka MPP_RET hal_avs2d_rkv_deinit(void *hal);
avs2d_ver_align(RK_U32 val)82*437bfbebSnyanmisaka static RK_U32 avs2d_ver_align(RK_U32 val)
83*437bfbebSnyanmisaka {
84*437bfbebSnyanmisaka return MPP_ALIGN(val, 16);
85*437bfbebSnyanmisaka }
86*437bfbebSnyanmisaka
avs2d_hor_align(RK_U32 val)87*437bfbebSnyanmisaka static RK_U32 avs2d_hor_align(RK_U32 val)
88*437bfbebSnyanmisaka {
89*437bfbebSnyanmisaka
90*437bfbebSnyanmisaka return MPP_ALIGN(val, 16);
91*437bfbebSnyanmisaka }
92*437bfbebSnyanmisaka
avs2d_len_align(RK_U32 val)93*437bfbebSnyanmisaka static RK_U32 avs2d_len_align(RK_U32 val)
94*437bfbebSnyanmisaka {
95*437bfbebSnyanmisaka return (2 * MPP_ALIGN(val, 16));
96*437bfbebSnyanmisaka }
97*437bfbebSnyanmisaka
avs2d_hor_align_64(RK_U32 val)98*437bfbebSnyanmisaka static RK_U32 avs2d_hor_align_64(RK_U32 val)
99*437bfbebSnyanmisaka {
100*437bfbebSnyanmisaka return MPP_ALIGN(val, 64);
101*437bfbebSnyanmisaka }
102*437bfbebSnyanmisaka
prepare_header(Avs2dHalCtx_t * p_hal,RK_U8 * data,RK_U32 len)103*437bfbebSnyanmisaka static MPP_RET prepare_header(Avs2dHalCtx_t *p_hal, RK_U8 *data, RK_U32 len)
104*437bfbebSnyanmisaka {
105*437bfbebSnyanmisaka RK_U32 i, j;
106*437bfbebSnyanmisaka BitputCtx_t bp;
107*437bfbebSnyanmisaka RK_U64 *bit_buf = (RK_U64 *)data;
108*437bfbebSnyanmisaka Avs2dSyntax_t *syntax = &p_hal->syntax;
109*437bfbebSnyanmisaka PicParams_Avs2d *pp = &syntax->pp;
110*437bfbebSnyanmisaka AlfParams_Avs2d *alfp = &syntax->alfp;
111*437bfbebSnyanmisaka RefParams_Avs2d *refp = &syntax->refp;
112*437bfbebSnyanmisaka WqmParams_Avs2d *wqmp = &syntax->wqmp;
113*437bfbebSnyanmisaka
114*437bfbebSnyanmisaka memset(data, 0, len);
115*437bfbebSnyanmisaka
116*437bfbebSnyanmisaka mpp_set_bitput_ctx(&bp, bit_buf, len);
117*437bfbebSnyanmisaka //!< sequence header syntax
118*437bfbebSnyanmisaka mpp_put_bits(&bp, pp->chroma_format_idc, 2);
119*437bfbebSnyanmisaka mpp_put_bits(&bp, pp->pic_width_in_luma_samples, 16);
120*437bfbebSnyanmisaka mpp_put_bits(&bp, pp->pic_height_in_luma_samples, 16);
121*437bfbebSnyanmisaka mpp_put_bits(&bp, pp->bit_depth_luma_minus8, 3);
122*437bfbebSnyanmisaka mpp_put_bits(&bp, pp->bit_depth_chroma_minus8, 3);
123*437bfbebSnyanmisaka mpp_put_bits(&bp, pp->lcu_size, 3);
124*437bfbebSnyanmisaka mpp_put_bits(&bp, pp->progressive_sequence, 1);
125*437bfbebSnyanmisaka mpp_put_bits(&bp, pp->field_coded_sequence, 1);
126*437bfbebSnyanmisaka mpp_put_bits(&bp, pp->multi_hypothesis_skip_enable_flag, 1);
127*437bfbebSnyanmisaka mpp_put_bits(&bp, pp->dual_hypothesis_prediction_enable_flag, 1);
128*437bfbebSnyanmisaka mpp_put_bits(&bp, pp->weighted_skip_enable_flag, 1);
129*437bfbebSnyanmisaka mpp_put_bits(&bp, pp->asymmetrc_motion_partitions_enable_flag, 1);
130*437bfbebSnyanmisaka mpp_put_bits(&bp, pp->nonsquare_quadtree_transform_enable_flag, 1);
131*437bfbebSnyanmisaka mpp_put_bits(&bp, pp->nonsquare_intra_prediction_enable_flag, 1);
132*437bfbebSnyanmisaka mpp_put_bits(&bp, pp->secondary_transform_enable_flag, 1);
133*437bfbebSnyanmisaka mpp_put_bits(&bp, pp->sample_adaptive_offset_enable_flag, 1);
134*437bfbebSnyanmisaka mpp_put_bits(&bp, pp->adaptive_loop_filter_enable_flag, 1);
135*437bfbebSnyanmisaka mpp_put_bits(&bp, pp->pmvr_enable_flag, 1);
136*437bfbebSnyanmisaka mpp_put_bits(&bp, pp->cross_slice_loopfilter_enable_flag, 1);
137*437bfbebSnyanmisaka //!< picture header syntax
138*437bfbebSnyanmisaka mpp_put_bits(&bp, pp->picture_type, 3);
139*437bfbebSnyanmisaka mpp_put_bits(&bp, refp->ref_pic_num, 3);
140*437bfbebSnyanmisaka mpp_put_bits(&bp, pp->scene_reference_enable_flag, 1);
141*437bfbebSnyanmisaka mpp_put_bits(&bp, pp->bottom_field_picture_flag, 1);
142*437bfbebSnyanmisaka mpp_put_bits(&bp, pp->fixed_picture_qp, 1);
143*437bfbebSnyanmisaka mpp_put_bits(&bp, pp->picture_qp, 7);
144*437bfbebSnyanmisaka mpp_put_bits(&bp, pp->loop_filter_disable_flag, 1);
145*437bfbebSnyanmisaka mpp_put_bits(&bp, pp->alpha_c_offset, 5);
146*437bfbebSnyanmisaka mpp_put_bits(&bp, pp->beta_offset, 5);
147*437bfbebSnyanmisaka //!< weight quant param
148*437bfbebSnyanmisaka mpp_put_bits(&bp, wqmp->chroma_quant_param_delta_cb, 6);
149*437bfbebSnyanmisaka mpp_put_bits(&bp, wqmp->chroma_quant_param_delta_cr, 6);
150*437bfbebSnyanmisaka mpp_put_bits(&bp, wqmp->pic_weight_quant_enable_flag, 1);
151*437bfbebSnyanmisaka //!< alf param
152*437bfbebSnyanmisaka mpp_put_bits(&bp, alfp->enable_pic_alf_y, 1);
153*437bfbebSnyanmisaka mpp_put_bits(&bp, alfp->enable_pic_alf_cb, 1);
154*437bfbebSnyanmisaka mpp_put_bits(&bp, alfp->enable_pic_alf_cr, 1);
155*437bfbebSnyanmisaka
156*437bfbebSnyanmisaka if (alfp->enable_pic_alf_y) {
157*437bfbebSnyanmisaka RK_U32 alf_filter_num = alfp->alf_filter_num_minus1 + 1;
158*437bfbebSnyanmisaka mpp_put_bits(&bp, alfp->alf_filter_num_minus1, 4);
159*437bfbebSnyanmisaka
160*437bfbebSnyanmisaka for (i = 0; i < 16; i++)
161*437bfbebSnyanmisaka mpp_put_bits(&bp, alfp->alf_coeff_idx_tab[i], 4);
162*437bfbebSnyanmisaka
163*437bfbebSnyanmisaka for (i = 0; i < alf_filter_num; i++) {
164*437bfbebSnyanmisaka for (j = 0; j < 9; j++) {
165*437bfbebSnyanmisaka mpp_put_bits(&bp, alfp->alf_coeff_y[i][j], 7);
166*437bfbebSnyanmisaka }
167*437bfbebSnyanmisaka }
168*437bfbebSnyanmisaka }
169*437bfbebSnyanmisaka
170*437bfbebSnyanmisaka if (alfp->enable_pic_alf_cb) {
171*437bfbebSnyanmisaka for (j = 0; j < 9; j++)
172*437bfbebSnyanmisaka mpp_put_bits(&bp, alfp->alf_coeff_cb[j], 7);
173*437bfbebSnyanmisaka }
174*437bfbebSnyanmisaka
175*437bfbebSnyanmisaka if (alfp->enable_pic_alf_cr) {
176*437bfbebSnyanmisaka for (j = 0; j < 9; j++)
177*437bfbebSnyanmisaka mpp_put_bits(&bp, alfp->alf_coeff_cr[j], 7);
178*437bfbebSnyanmisaka }
179*437bfbebSnyanmisaka
180*437bfbebSnyanmisaka mpp_put_align(&bp, 128, 0);
181*437bfbebSnyanmisaka
182*437bfbebSnyanmisaka return MPP_OK;
183*437bfbebSnyanmisaka }
184*437bfbebSnyanmisaka
prepare_scalist(Avs2dHalCtx_t * p_hal,RK_U8 * data,RK_U32 len)185*437bfbebSnyanmisaka static MPP_RET prepare_scalist(Avs2dHalCtx_t *p_hal, RK_U8 *data, RK_U32 len)
186*437bfbebSnyanmisaka {
187*437bfbebSnyanmisaka RK_U32 i, j;
188*437bfbebSnyanmisaka RK_U32 size_id, block_size;
189*437bfbebSnyanmisaka BitputCtx_t bp;
190*437bfbebSnyanmisaka RK_U64 *bit_buf = (RK_U64 *)data;
191*437bfbebSnyanmisaka Avs2dSyntax_t *syntax = &p_hal->syntax;
192*437bfbebSnyanmisaka WqmParams_Avs2d *wqmp = &syntax->wqmp;
193*437bfbebSnyanmisaka
194*437bfbebSnyanmisaka if (!wqmp->pic_weight_quant_enable_flag)
195*437bfbebSnyanmisaka return MPP_OK;
196*437bfbebSnyanmisaka
197*437bfbebSnyanmisaka memset(data, 0, len);
198*437bfbebSnyanmisaka
199*437bfbebSnyanmisaka mpp_set_bitput_ctx(&bp, bit_buf, len);
200*437bfbebSnyanmisaka
201*437bfbebSnyanmisaka for (size_id = 0; size_id < 2; size_id++) {
202*437bfbebSnyanmisaka block_size = MPP_MIN(1 << (size_id + 2), 8);
203*437bfbebSnyanmisaka for (i = 0; i < block_size; i++) {
204*437bfbebSnyanmisaka for (j = 0 ; j < block_size; j++)
205*437bfbebSnyanmisaka //!< row col reversed
206*437bfbebSnyanmisaka mpp_put_bits(&bp, wqmp->wq_matrix[size_id][size_id * j + i], 8);
207*437bfbebSnyanmisaka }
208*437bfbebSnyanmisaka }
209*437bfbebSnyanmisaka
210*437bfbebSnyanmisaka return MPP_OK;
211*437bfbebSnyanmisaka }
212*437bfbebSnyanmisaka
get_frame_fd(Avs2dHalCtx_t * p_hal,RK_S32 idx)213*437bfbebSnyanmisaka static RK_S32 get_frame_fd(Avs2dHalCtx_t *p_hal, RK_S32 idx)
214*437bfbebSnyanmisaka {
215*437bfbebSnyanmisaka RK_S32 ret_fd = 0;
216*437bfbebSnyanmisaka MppBuffer mbuffer = NULL;
217*437bfbebSnyanmisaka
218*437bfbebSnyanmisaka mpp_buf_slot_get_prop(p_hal->frame_slots, idx, SLOT_BUFFER, &mbuffer);
219*437bfbebSnyanmisaka ret_fd = mpp_buffer_get_fd(mbuffer);
220*437bfbebSnyanmisaka
221*437bfbebSnyanmisaka return ret_fd;
222*437bfbebSnyanmisaka }
223*437bfbebSnyanmisaka
get_packet_fd(Avs2dHalCtx_t * p_hal,RK_S32 idx)224*437bfbebSnyanmisaka static RK_S32 get_packet_fd(Avs2dHalCtx_t *p_hal, RK_S32 idx)
225*437bfbebSnyanmisaka {
226*437bfbebSnyanmisaka RK_S32 ret_fd = 0;
227*437bfbebSnyanmisaka MppBuffer mbuffer = NULL;
228*437bfbebSnyanmisaka
229*437bfbebSnyanmisaka mpp_buf_slot_get_prop(p_hal->packet_slots, idx, SLOT_BUFFER, &mbuffer);
230*437bfbebSnyanmisaka ret_fd = mpp_buffer_get_fd(mbuffer);
231*437bfbebSnyanmisaka
232*437bfbebSnyanmisaka return ret_fd;
233*437bfbebSnyanmisaka }
234*437bfbebSnyanmisaka
init_common_regs(Vdpu34xAvs2dRegSet * regs)235*437bfbebSnyanmisaka static MPP_RET init_common_regs(Vdpu34xAvs2dRegSet *regs)
236*437bfbebSnyanmisaka {
237*437bfbebSnyanmisaka Vdpu34xRegCommon *common = ®s->common;
238*437bfbebSnyanmisaka
239*437bfbebSnyanmisaka common->reg009.dec_mode = 3; // AVS2
240*437bfbebSnyanmisaka common->reg015.rlc_mode = 0;
241*437bfbebSnyanmisaka
242*437bfbebSnyanmisaka common->reg011.buf_empty_en = 1;
243*437bfbebSnyanmisaka common->reg011.dec_timeout_e = 1;
244*437bfbebSnyanmisaka
245*437bfbebSnyanmisaka common->reg010.dec_e = 1;
246*437bfbebSnyanmisaka
247*437bfbebSnyanmisaka common->reg013.h26x_error_mode = 0;
248*437bfbebSnyanmisaka common->reg013.colmv_error_mode = 0;
249*437bfbebSnyanmisaka common->reg013.h26x_streamd_error_mode = 0;
250*437bfbebSnyanmisaka common->reg021.inter_error_prc_mode = 0;
251*437bfbebSnyanmisaka common->reg021.error_deb_en = 0;
252*437bfbebSnyanmisaka common->reg021.error_intra_mode = 0;
253*437bfbebSnyanmisaka
254*437bfbebSnyanmisaka if (mpp_get_soc_type() == ROCKCHIP_SOC_RK3588) {
255*437bfbebSnyanmisaka common->reg024.cabac_err_en_lowbits = 0;
256*437bfbebSnyanmisaka common->reg025.cabac_err_en_highbits = 0;
257*437bfbebSnyanmisaka common->reg026.swreg_block_gating_e = 0xfffef;
258*437bfbebSnyanmisaka } else {
259*437bfbebSnyanmisaka common->reg024.cabac_err_en_lowbits = 0xffffffdf;
260*437bfbebSnyanmisaka common->reg025.cabac_err_en_highbits = 0x3dffffff;
261*437bfbebSnyanmisaka common->reg026.swreg_block_gating_e = 0xfffff;
262*437bfbebSnyanmisaka }
263*437bfbebSnyanmisaka
264*437bfbebSnyanmisaka common->reg026.reg_cfg_gating_en = 1;
265*437bfbebSnyanmisaka common->reg032_timeout_threshold = 0x3fffff;
266*437bfbebSnyanmisaka
267*437bfbebSnyanmisaka common->reg011.dec_clkgate_e = 1;
268*437bfbebSnyanmisaka common->reg011.dec_e_strmd_clkgate_dis = 0;
269*437bfbebSnyanmisaka common->reg011.dec_timeout_e = 1;
270*437bfbebSnyanmisaka
271*437bfbebSnyanmisaka common->reg013.timeout_mode = 1;
272*437bfbebSnyanmisaka common->reg013.stmerror_waitdecfifo_empty = 1;
273*437bfbebSnyanmisaka common->reg012.colmv_compress_en = COLMV_COMPRESS_EN;
274*437bfbebSnyanmisaka common->reg012.wr_ddr_align_en = 1;
275*437bfbebSnyanmisaka common->reg012.info_collect_en = 1;
276*437bfbebSnyanmisaka common->reg012.error_info_en = 0;
277*437bfbebSnyanmisaka
278*437bfbebSnyanmisaka return MPP_OK;
279*437bfbebSnyanmisaka }
280*437bfbebSnyanmisaka
281*437bfbebSnyanmisaka //TODO calc rcb buffer size;
282*437bfbebSnyanmisaka /*
283*437bfbebSnyanmisaka static void avs2d_refine_rcb_size(Vdpu34xRcbInfo *rcb_info,
284*437bfbebSnyanmisaka Vdpu34xAvs2dRegSet *hw_regs,
285*437bfbebSnyanmisaka RK_S32 width, RK_S32 height, void *dxva)
286*437bfbebSnyanmisaka {
287*437bfbebSnyanmisaka (void) rcb_info;
288*437bfbebSnyanmisaka (void) hw_regs;
289*437bfbebSnyanmisaka (void) width;
290*437bfbebSnyanmisaka (void) height;
291*437bfbebSnyanmisaka (void) dxva;
292*437bfbebSnyanmisaka return;
293*437bfbebSnyanmisaka }
294*437bfbebSnyanmisaka */
295*437bfbebSnyanmisaka
hal_avs2d_rcb_info_update(void * hal,Vdpu34xAvs2dRegSet * hw_regs)296*437bfbebSnyanmisaka static void hal_avs2d_rcb_info_update(void *hal, Vdpu34xAvs2dRegSet *hw_regs)
297*437bfbebSnyanmisaka {
298*437bfbebSnyanmisaka MPP_RET ret = MPP_OK;
299*437bfbebSnyanmisaka Avs2dHalCtx_t *p_hal = (Avs2dHalCtx_t *)hal;
300*437bfbebSnyanmisaka Avs2dRkvRegCtx_t *reg_ctx = (Avs2dRkvRegCtx_t *)p_hal->reg_ctx;
301*437bfbebSnyanmisaka RK_S32 width = p_hal->syntax.pp.pic_width_in_luma_samples;
302*437bfbebSnyanmisaka RK_S32 height = p_hal->syntax.pp.pic_height_in_luma_samples;
303*437bfbebSnyanmisaka RK_S32 i = 0;
304*437bfbebSnyanmisaka RK_S32 loop = p_hal->fast_mode ? MPP_ARRAY_ELEMS(reg_ctx->reg_buf) : 1;
305*437bfbebSnyanmisaka
306*437bfbebSnyanmisaka (void) hw_regs;
307*437bfbebSnyanmisaka
308*437bfbebSnyanmisaka reg_ctx->rcb_buf_size = vdpu34x_get_rcb_buf_size(reg_ctx->rcb_info, width, height);
309*437bfbebSnyanmisaka //avs2d_refine_rcb_size(reg_ctx->rcb_info, hw_regs, width, height, (void *)&p_hal->syntax);
310*437bfbebSnyanmisaka
311*437bfbebSnyanmisaka for (i = 0; i < loop; i++) {
312*437bfbebSnyanmisaka MppBuffer rcb_buf = NULL;
313*437bfbebSnyanmisaka
314*437bfbebSnyanmisaka if (reg_ctx->rcb_buf[i]) {
315*437bfbebSnyanmisaka mpp_buffer_put(reg_ctx->rcb_buf[i]);
316*437bfbebSnyanmisaka reg_ctx->rcb_buf[i] = NULL;
317*437bfbebSnyanmisaka }
318*437bfbebSnyanmisaka
319*437bfbebSnyanmisaka ret = mpp_buffer_get(p_hal->buf_group, &rcb_buf, reg_ctx->rcb_buf_size);
320*437bfbebSnyanmisaka
321*437bfbebSnyanmisaka if (ret)
322*437bfbebSnyanmisaka mpp_err_f("AVS2D mpp_buffer_group_get failed\n");
323*437bfbebSnyanmisaka
324*437bfbebSnyanmisaka reg_ctx->rcb_buf[i] = rcb_buf;
325*437bfbebSnyanmisaka }
326*437bfbebSnyanmisaka }
327*437bfbebSnyanmisaka
fill_registers(Avs2dHalCtx_t * p_hal,Vdpu34xAvs2dRegSet * p_regs,HalTaskInfo * task)328*437bfbebSnyanmisaka static MPP_RET fill_registers(Avs2dHalCtx_t *p_hal, Vdpu34xAvs2dRegSet *p_regs, HalTaskInfo *task)
329*437bfbebSnyanmisaka {
330*437bfbebSnyanmisaka MPP_RET ret = MPP_OK;
331*437bfbebSnyanmisaka RK_U32 i;
332*437bfbebSnyanmisaka MppFrame mframe = NULL;
333*437bfbebSnyanmisaka Avs2dSyntax_t *syntax = &p_hal->syntax;
334*437bfbebSnyanmisaka PicParams_Avs2d *pp = &syntax->pp;
335*437bfbebSnyanmisaka RefParams_Avs2d *refp = &syntax->refp;
336*437bfbebSnyanmisaka HalDecTask *task_dec = &task->dec;
337*437bfbebSnyanmisaka Vdpu34xRegCommon *common = &p_regs->common;
338*437bfbebSnyanmisaka RK_U32 is_fbc = 0;
339*437bfbebSnyanmisaka HalBuf *mv_buf = NULL;
340*437bfbebSnyanmisaka
341*437bfbebSnyanmisaka mpp_buf_slot_get_prop(p_hal->frame_slots, task_dec->output, SLOT_FRAME_PTR, &mframe);
342*437bfbebSnyanmisaka is_fbc = MPP_FRAME_FMT_IS_FBC(mpp_frame_get_fmt(mframe));
343*437bfbebSnyanmisaka
344*437bfbebSnyanmisaka //!< caculate the yuv_frame_size
345*437bfbebSnyanmisaka {
346*437bfbebSnyanmisaka RK_U32 hor_virstride = 0;
347*437bfbebSnyanmisaka RK_U32 ver_virstride = 0;
348*437bfbebSnyanmisaka RK_U32 y_virstride = 0;
349*437bfbebSnyanmisaka
350*437bfbebSnyanmisaka hor_virstride = mpp_frame_get_hor_stride(mframe);
351*437bfbebSnyanmisaka ver_virstride = mpp_frame_get_ver_stride(mframe);
352*437bfbebSnyanmisaka y_virstride = hor_virstride * ver_virstride;
353*437bfbebSnyanmisaka AVS2D_HAL_TRACE("is_fbc %d y_virstride %d, hor_virstride %d, ver_virstride %d\n", is_fbc, y_virstride, hor_virstride, ver_virstride);
354*437bfbebSnyanmisaka
355*437bfbebSnyanmisaka if (is_fbc) {
356*437bfbebSnyanmisaka RK_U32 fbc_hdr_stride = mpp_frame_get_fbc_hdr_stride(mframe);
357*437bfbebSnyanmisaka RK_U32 fbd_offset = MPP_ALIGN(fbc_hdr_stride * (ver_virstride + 16) / 16, SZ_4K);
358*437bfbebSnyanmisaka
359*437bfbebSnyanmisaka common->reg012.fbc_e = 1;
360*437bfbebSnyanmisaka common->reg018.y_hor_virstride = fbc_hdr_stride / 16;
361*437bfbebSnyanmisaka common->reg019.uv_hor_virstride = fbc_hdr_stride / 16;
362*437bfbebSnyanmisaka common->reg020_fbc_payload_off.payload_st_offset = fbd_offset >> 4;
363*437bfbebSnyanmisaka } else {
364*437bfbebSnyanmisaka common->reg012.fbc_e = 0;
365*437bfbebSnyanmisaka common->reg018.y_hor_virstride = hor_virstride / 16;
366*437bfbebSnyanmisaka common->reg019.uv_hor_virstride = hor_virstride / 16;
367*437bfbebSnyanmisaka common->reg020_y_virstride.y_virstride = y_virstride / 16;
368*437bfbebSnyanmisaka }
369*437bfbebSnyanmisaka common->reg013.cur_pic_is_idr = (pp->picture_type == 0 || pp->picture_type == 4 || pp->picture_type == 5);
370*437bfbebSnyanmisaka }
371*437bfbebSnyanmisaka
372*437bfbebSnyanmisaka // set current
373*437bfbebSnyanmisaka {
374*437bfbebSnyanmisaka RK_S32 fd = -1;
375*437bfbebSnyanmisaka p_regs->avs2d_param.reg65_cur_top_poc = mpp_frame_get_poc(mframe);
376*437bfbebSnyanmisaka p_regs->avs2d_param.reg66_cur_bot_poc = 0;
377*437bfbebSnyanmisaka fd = get_frame_fd(p_hal, task_dec->output);
378*437bfbebSnyanmisaka mpp_assert(fd >= 0);
379*437bfbebSnyanmisaka p_regs->common_addr.reg130_decout_base = fd;
380*437bfbebSnyanmisaka mv_buf = hal_bufs_get_buf(p_hal->cmv_bufs, task_dec->output);
381*437bfbebSnyanmisaka p_regs->common_addr.reg131_colmv_cur_base = mpp_buffer_get_fd(mv_buf->buf[0]);
382*437bfbebSnyanmisaka AVS2D_HAL_TRACE("cur frame index %d, fd %d, colmv fd %d", task_dec->output, fd, p_regs->common_addr.reg131_colmv_cur_base);
383*437bfbebSnyanmisaka }
384*437bfbebSnyanmisaka
385*437bfbebSnyanmisaka // set reference
386*437bfbebSnyanmisaka {
387*437bfbebSnyanmisaka RK_U64 ref_flag = 0;
388*437bfbebSnyanmisaka RK_S32 valid_slot = -1;
389*437bfbebSnyanmisaka RK_U32 *ref_low = (RK_U32 *)&p_regs->avs2d_param.reg99;
390*437bfbebSnyanmisaka RK_U32 *ref_hight = (RK_U32 *)&p_regs->avs2d_param.reg100;
391*437bfbebSnyanmisaka RK_U32 err_ref_base = 0;
392*437bfbebSnyanmisaka
393*437bfbebSnyanmisaka AVS2D_HAL_TRACE("num of ref %d", refp->ref_pic_num);
394*437bfbebSnyanmisaka
395*437bfbebSnyanmisaka for (i = 0; i < refp->ref_pic_num; i++) {
396*437bfbebSnyanmisaka if (task_dec->refer[i] < 0)
397*437bfbebSnyanmisaka continue;
398*437bfbebSnyanmisaka
399*437bfbebSnyanmisaka valid_slot = i;
400*437bfbebSnyanmisaka break;
401*437bfbebSnyanmisaka }
402*437bfbebSnyanmisaka
403*437bfbebSnyanmisaka for (i = 0; i < refp->ref_pic_num; i++) {
404*437bfbebSnyanmisaka MppFrame frame_ref = NULL;
405*437bfbebSnyanmisaka
406*437bfbebSnyanmisaka RK_S32 slot_idx = task_dec->refer[i] < 0 ? task_dec->refer[valid_slot] : task_dec->refer[i];
407*437bfbebSnyanmisaka
408*437bfbebSnyanmisaka if (slot_idx < 0) {
409*437bfbebSnyanmisaka AVS2D_HAL_DBG(AVS2D_HAL_DBG_ERROR, "missing ref, could not found valid ref");
410*437bfbebSnyanmisaka task->dec.flags.ref_err = 1;
411*437bfbebSnyanmisaka return ret = MPP_ERR_UNKNOW;
412*437bfbebSnyanmisaka }
413*437bfbebSnyanmisaka
414*437bfbebSnyanmisaka mpp_buf_slot_get_prop(p_hal->frame_slots, slot_idx, SLOT_FRAME_PTR, &frame_ref);
415*437bfbebSnyanmisaka
416*437bfbebSnyanmisaka if (frame_ref) {
417*437bfbebSnyanmisaka RK_U32 frm_flag = 1 << 3;
418*437bfbebSnyanmisaka
419*437bfbebSnyanmisaka if (pp->bottom_field_picture_flag)
420*437bfbebSnyanmisaka frm_flag |= 1 << 2;
421*437bfbebSnyanmisaka
422*437bfbebSnyanmisaka if (pp->field_coded_sequence)
423*437bfbebSnyanmisaka frm_flag |= 1;
424*437bfbebSnyanmisaka
425*437bfbebSnyanmisaka ref_flag |= frm_flag << (i * 8);
426*437bfbebSnyanmisaka
427*437bfbebSnyanmisaka p_regs->avs2d_addr.ref_base[i] = get_frame_fd(p_hal, slot_idx);
428*437bfbebSnyanmisaka mv_buf = hal_bufs_get_buf(p_hal->cmv_bufs, slot_idx);
429*437bfbebSnyanmisaka p_regs->avs2d_addr.colmv_base[i] = mpp_buffer_get_fd(mv_buf->buf[0]);
430*437bfbebSnyanmisaka
431*437bfbebSnyanmisaka p_regs->avs2d_param.reg67_098_ref_poc[i] = mpp_frame_get_poc(frame_ref);
432*437bfbebSnyanmisaka
433*437bfbebSnyanmisaka if (!err_ref_base && !mpp_frame_get_errinfo(frame_ref))
434*437bfbebSnyanmisaka err_ref_base = p_regs->avs2d_addr.ref_base[i];
435*437bfbebSnyanmisaka
436*437bfbebSnyanmisaka AVS2D_HAL_TRACE("ref_base[%d] index=%d, fd = %d, colmv %d, poc %d",
437*437bfbebSnyanmisaka i, slot_idx, p_regs->avs2d_addr.ref_base[i],
438*437bfbebSnyanmisaka p_regs->avs2d_addr.colmv_base[i], p_regs->avs2d_param.reg67_098_ref_poc[i]);
439*437bfbebSnyanmisaka }
440*437bfbebSnyanmisaka }
441*437bfbebSnyanmisaka
442*437bfbebSnyanmisaka if (p_hal->syntax.refp.scene_ref_enable && p_hal->syntax.refp.scene_ref_slot_idx >= 0) {
443*437bfbebSnyanmisaka MppFrame scene_ref = NULL;
444*437bfbebSnyanmisaka RK_S32 slot_idx = p_hal->syntax.refp.scene_ref_slot_idx;
445*437bfbebSnyanmisaka RK_S32 replace_idx = p_hal->syntax.refp.scene_ref_replace_pos;
446*437bfbebSnyanmisaka
447*437bfbebSnyanmisaka mpp_buf_slot_get_prop(p_hal->frame_slots, slot_idx, SLOT_FRAME_PTR, &scene_ref);
448*437bfbebSnyanmisaka
449*437bfbebSnyanmisaka if (scene_ref) {
450*437bfbebSnyanmisaka p_regs->avs2d_addr.ref_base[replace_idx] = get_frame_fd(p_hal, slot_idx);
451*437bfbebSnyanmisaka mv_buf = hal_bufs_get_buf(p_hal->cmv_bufs, slot_idx);
452*437bfbebSnyanmisaka p_regs->avs2d_addr.colmv_base[replace_idx] = mpp_buffer_get_fd(mv_buf->buf[0]);
453*437bfbebSnyanmisaka p_regs->avs2d_param.reg67_098_ref_poc[replace_idx] = mpp_frame_get_poc(scene_ref);
454*437bfbebSnyanmisaka }
455*437bfbebSnyanmisaka }
456*437bfbebSnyanmisaka
457*437bfbebSnyanmisaka *ref_low = (RK_U32) (ref_flag & 0xffffffff);
458*437bfbebSnyanmisaka *ref_hight = (RK_U32) ((ref_flag >> 32) & 0xffffffff);
459*437bfbebSnyanmisaka
460*437bfbebSnyanmisaka p_regs->common_addr.reg132_error_ref_base = err_ref_base;
461*437bfbebSnyanmisaka }
462*437bfbebSnyanmisaka
463*437bfbebSnyanmisaka // set rlc
464*437bfbebSnyanmisaka {
465*437bfbebSnyanmisaka p_regs->common_addr.reg128_rlc_base = get_packet_fd(p_hal, task_dec->input);
466*437bfbebSnyanmisaka AVS2D_HAL_TRACE("packet fd %d from slot %d", p_regs->common_addr.reg128_rlc_base, task_dec->input);
467*437bfbebSnyanmisaka p_regs->common_addr.reg129_rlcwrite_base = p_regs->common_addr.reg128_rlc_base;
468*437bfbebSnyanmisaka common->reg016_str_len = MPP_ALIGN(mpp_packet_get_length(task_dec->input_packet), 16) + 64;
469*437bfbebSnyanmisaka }
470*437bfbebSnyanmisaka
471*437bfbebSnyanmisaka return ret;
472*437bfbebSnyanmisaka }
473*437bfbebSnyanmisaka
hal_avs2d_rkv_deinit(void * hal)474*437bfbebSnyanmisaka MPP_RET hal_avs2d_rkv_deinit(void *hal)
475*437bfbebSnyanmisaka {
476*437bfbebSnyanmisaka MPP_RET ret = MPP_OK;
477*437bfbebSnyanmisaka RK_U32 i, loop;
478*437bfbebSnyanmisaka Avs2dHalCtx_t *p_hal = (Avs2dHalCtx_t *)hal;
479*437bfbebSnyanmisaka Avs2dRkvRegCtx_t *reg_ctx = (Avs2dRkvRegCtx_t *)p_hal->reg_ctx;
480*437bfbebSnyanmisaka
481*437bfbebSnyanmisaka AVS2D_HAL_TRACE("In.");
482*437bfbebSnyanmisaka
483*437bfbebSnyanmisaka INP_CHECK(ret, NULL == reg_ctx);
484*437bfbebSnyanmisaka
485*437bfbebSnyanmisaka //!< malloc buffers
486*437bfbebSnyanmisaka loop = p_hal->fast_mode ? MPP_ARRAY_ELEMS(reg_ctx->reg_buf) : 1;
487*437bfbebSnyanmisaka for (i = 0; i < loop; i++) {
488*437bfbebSnyanmisaka if (reg_ctx->rcb_buf[i]) {
489*437bfbebSnyanmisaka mpp_buffer_put(reg_ctx->rcb_buf[i]);
490*437bfbebSnyanmisaka reg_ctx->rcb_buf[i] = NULL;
491*437bfbebSnyanmisaka }
492*437bfbebSnyanmisaka
493*437bfbebSnyanmisaka MPP_FREE(reg_ctx->reg_buf[i].regs);
494*437bfbebSnyanmisaka }
495*437bfbebSnyanmisaka
496*437bfbebSnyanmisaka if (reg_ctx->bufs) {
497*437bfbebSnyanmisaka mpp_buffer_put(reg_ctx->bufs);
498*437bfbebSnyanmisaka reg_ctx->bufs = NULL;
499*437bfbebSnyanmisaka }
500*437bfbebSnyanmisaka
501*437bfbebSnyanmisaka if (p_hal->cmv_bufs) {
502*437bfbebSnyanmisaka hal_bufs_deinit(p_hal->cmv_bufs);
503*437bfbebSnyanmisaka p_hal->cmv_bufs = NULL;
504*437bfbebSnyanmisaka }
505*437bfbebSnyanmisaka
506*437bfbebSnyanmisaka MPP_FREE(p_hal->reg_ctx);
507*437bfbebSnyanmisaka
508*437bfbebSnyanmisaka __RETURN:
509*437bfbebSnyanmisaka AVS2D_HAL_TRACE("Out. ret %d", ret);
510*437bfbebSnyanmisaka return ret;
511*437bfbebSnyanmisaka }
512*437bfbebSnyanmisaka
hal_avs2d_rkv_init(void * hal,MppHalCfg * cfg)513*437bfbebSnyanmisaka MPP_RET hal_avs2d_rkv_init(void *hal, MppHalCfg *cfg)
514*437bfbebSnyanmisaka {
515*437bfbebSnyanmisaka MPP_RET ret = MPP_OK;
516*437bfbebSnyanmisaka RK_U32 i, loop;
517*437bfbebSnyanmisaka Avs2dRkvRegCtx_t *reg_ctx;
518*437bfbebSnyanmisaka Avs2dHalCtx_t *p_hal = (Avs2dHalCtx_t *)hal;
519*437bfbebSnyanmisaka
520*437bfbebSnyanmisaka AVS2D_HAL_TRACE("In.");
521*437bfbebSnyanmisaka
522*437bfbebSnyanmisaka INP_CHECK(ret, NULL == p_hal);
523*437bfbebSnyanmisaka
524*437bfbebSnyanmisaka MEM_CHECK(ret, p_hal->reg_ctx = mpp_calloc_size(void, sizeof(Avs2dRkvRegCtx_t)));
525*437bfbebSnyanmisaka reg_ctx = (Avs2dRkvRegCtx_t *)p_hal->reg_ctx;
526*437bfbebSnyanmisaka
527*437bfbebSnyanmisaka //!< malloc buffers
528*437bfbebSnyanmisaka loop = p_hal->fast_mode ? MPP_ARRAY_ELEMS(reg_ctx->reg_buf) : 1;
529*437bfbebSnyanmisaka FUN_CHECK(ret = mpp_buffer_get(p_hal->buf_group, ®_ctx->bufs, AVS2_ALL_TBL_BUF_SIZE(loop)));
530*437bfbebSnyanmisaka reg_ctx->bufs_fd = mpp_buffer_get_fd(reg_ctx->bufs);
531*437bfbebSnyanmisaka reg_ctx->bufs_ptr = mpp_buffer_get_ptr(reg_ctx->bufs);
532*437bfbebSnyanmisaka
533*437bfbebSnyanmisaka for (i = 0; i < loop; i++) {
534*437bfbebSnyanmisaka reg_ctx->reg_buf[i].regs = mpp_calloc(Vdpu34xAvs2dRegSet, 1);
535*437bfbebSnyanmisaka init_common_regs(reg_ctx->reg_buf[i].regs);
536*437bfbebSnyanmisaka reg_ctx->reg_buf[i].offset_shph = AVS2_SHPH_OFFSET(i);
537*437bfbebSnyanmisaka reg_ctx->reg_buf[i].offset_sclst = AVS2_SCALIST_OFFSET(i);
538*437bfbebSnyanmisaka }
539*437bfbebSnyanmisaka
540*437bfbebSnyanmisaka if (!p_hal->fast_mode) {
541*437bfbebSnyanmisaka reg_ctx->regs = reg_ctx->reg_buf[0].regs;
542*437bfbebSnyanmisaka reg_ctx->shph_offset = reg_ctx->reg_buf[0].offset_shph;
543*437bfbebSnyanmisaka reg_ctx->sclst_offset = reg_ctx->reg_buf[0].offset_sclst;
544*437bfbebSnyanmisaka }
545*437bfbebSnyanmisaka
546*437bfbebSnyanmisaka if (MPP_FRAME_FMT_IS_FBC(cfg->cfg->base.out_fmt))
547*437bfbebSnyanmisaka mpp_slots_set_prop(p_hal->frame_slots, SLOTS_HOR_ALIGN, avs2d_hor_align_64);
548*437bfbebSnyanmisaka else
549*437bfbebSnyanmisaka mpp_slots_set_prop(p_hal->frame_slots, SLOTS_HOR_ALIGN, avs2d_hor_align);
550*437bfbebSnyanmisaka
551*437bfbebSnyanmisaka mpp_slots_set_prop(p_hal->frame_slots, SLOTS_VER_ALIGN, avs2d_ver_align);
552*437bfbebSnyanmisaka mpp_slots_set_prop(p_hal->frame_slots, SLOTS_LEN_ALIGN, avs2d_len_align);
553*437bfbebSnyanmisaka
554*437bfbebSnyanmisaka __RETURN:
555*437bfbebSnyanmisaka AVS2D_HAL_TRACE("Out. ret %d", ret);
556*437bfbebSnyanmisaka (void)cfg;
557*437bfbebSnyanmisaka return ret;
558*437bfbebSnyanmisaka __FAILED:
559*437bfbebSnyanmisaka hal_avs2d_rkv_deinit(p_hal);
560*437bfbebSnyanmisaka AVS2D_HAL_TRACE("Out. ret %d", ret);
561*437bfbebSnyanmisaka return ret;
562*437bfbebSnyanmisaka }
563*437bfbebSnyanmisaka
set_up_colmv_buf(void * hal)564*437bfbebSnyanmisaka static MPP_RET set_up_colmv_buf(void *hal)
565*437bfbebSnyanmisaka {
566*437bfbebSnyanmisaka MPP_RET ret = MPP_OK;
567*437bfbebSnyanmisaka Avs2dHalCtx_t *p_hal = (Avs2dHalCtx_t *)hal;
568*437bfbebSnyanmisaka Avs2dSyntax_t *syntax = &p_hal->syntax;
569*437bfbebSnyanmisaka PicParams_Avs2d *pp = &syntax->pp;
570*437bfbebSnyanmisaka RK_U32 mv_size = 0;
571*437bfbebSnyanmisaka RK_U32 ctu_size = 1 << (p_hal->syntax.pp.lcu_size);
572*437bfbebSnyanmisaka RK_U32 width = p_hal->syntax.pp.pic_width_in_luma_samples;
573*437bfbebSnyanmisaka RK_U32 height = p_hal->syntax.pp.pic_height_in_luma_samples;
574*437bfbebSnyanmisaka
575*437bfbebSnyanmisaka mv_size = vdpu34x_get_colmv_size(width, height, ctu_size, COLMV_BYTES,
576*437bfbebSnyanmisaka COLMV_BLOCK_SIZE, COLMV_COMPRESS_EN);
577*437bfbebSnyanmisaka if (pp->field_coded_sequence)
578*437bfbebSnyanmisaka mv_size *= 2;
579*437bfbebSnyanmisaka AVS2D_HAL_TRACE("mv_size %d", mv_size);
580*437bfbebSnyanmisaka
581*437bfbebSnyanmisaka if (p_hal->cmv_bufs == NULL || p_hal->mv_size < mv_size) {
582*437bfbebSnyanmisaka size_t size = mv_size;
583*437bfbebSnyanmisaka
584*437bfbebSnyanmisaka if (p_hal->cmv_bufs) {
585*437bfbebSnyanmisaka hal_bufs_deinit(p_hal->cmv_bufs);
586*437bfbebSnyanmisaka p_hal->cmv_bufs = NULL;
587*437bfbebSnyanmisaka }
588*437bfbebSnyanmisaka
589*437bfbebSnyanmisaka hal_bufs_init(&p_hal->cmv_bufs);
590*437bfbebSnyanmisaka if (p_hal->cmv_bufs == NULL) {
591*437bfbebSnyanmisaka mpp_err_f("colmv bufs init fail");
592*437bfbebSnyanmisaka ret = MPP_ERR_INIT;
593*437bfbebSnyanmisaka goto __RETURN;
594*437bfbebSnyanmisaka }
595*437bfbebSnyanmisaka
596*437bfbebSnyanmisaka p_hal->mv_size = mv_size;
597*437bfbebSnyanmisaka p_hal->mv_count = mpp_buf_slot_get_count(p_hal->frame_slots);
598*437bfbebSnyanmisaka hal_bufs_setup(p_hal->cmv_bufs, p_hal->mv_count, 1, &size);
599*437bfbebSnyanmisaka }
600*437bfbebSnyanmisaka
601*437bfbebSnyanmisaka __RETURN:
602*437bfbebSnyanmisaka return ret;
603*437bfbebSnyanmisaka }
604*437bfbebSnyanmisaka
hal_avs2d_rkv_gen_regs(void * hal,HalTaskInfo * task)605*437bfbebSnyanmisaka MPP_RET hal_avs2d_rkv_gen_regs(void *hal, HalTaskInfo *task)
606*437bfbebSnyanmisaka {
607*437bfbebSnyanmisaka MPP_RET ret = MPP_OK;
608*437bfbebSnyanmisaka Avs2dRkvRegCtx_t *reg_ctx;
609*437bfbebSnyanmisaka Avs2dHalCtx_t *p_hal = (Avs2dHalCtx_t *)hal;
610*437bfbebSnyanmisaka Vdpu34xAvs2dRegSet *regs = NULL;
611*437bfbebSnyanmisaka
612*437bfbebSnyanmisaka AVS2D_HAL_TRACE("In.");
613*437bfbebSnyanmisaka
614*437bfbebSnyanmisaka INP_CHECK(ret, NULL == p_hal);
615*437bfbebSnyanmisaka
616*437bfbebSnyanmisaka if ((task->dec.flags.parse_err || task->dec.flags.ref_err) &&
617*437bfbebSnyanmisaka !p_hal->cfg->base.disable_error) {
618*437bfbebSnyanmisaka ret = MPP_NOK;
619*437bfbebSnyanmisaka goto __RETURN;
620*437bfbebSnyanmisaka }
621*437bfbebSnyanmisaka
622*437bfbebSnyanmisaka ret = set_up_colmv_buf(p_hal);
623*437bfbebSnyanmisaka if (ret)
624*437bfbebSnyanmisaka goto __RETURN;
625*437bfbebSnyanmisaka
626*437bfbebSnyanmisaka reg_ctx = (Avs2dRkvRegCtx_t *)p_hal->reg_ctx;
627*437bfbebSnyanmisaka
628*437bfbebSnyanmisaka if (p_hal->fast_mode) {
629*437bfbebSnyanmisaka RK_U32 i = 0;
630*437bfbebSnyanmisaka
631*437bfbebSnyanmisaka for (i = 0; i < MPP_ARRAY_ELEMS(reg_ctx->reg_buf); i++) {
632*437bfbebSnyanmisaka if (!reg_ctx->reg_buf[i].valid) {
633*437bfbebSnyanmisaka task->dec.reg_index = i;
634*437bfbebSnyanmisaka regs = reg_ctx->reg_buf[i].regs;
635*437bfbebSnyanmisaka reg_ctx->shph_offset = reg_ctx->reg_buf[i].offset_shph;
636*437bfbebSnyanmisaka reg_ctx->sclst_offset = reg_ctx->reg_buf[i].offset_sclst;
637*437bfbebSnyanmisaka reg_ctx->regs = reg_ctx->reg_buf[i].regs;
638*437bfbebSnyanmisaka reg_ctx->reg_buf[i].valid = 1;
639*437bfbebSnyanmisaka break;
640*437bfbebSnyanmisaka }
641*437bfbebSnyanmisaka }
642*437bfbebSnyanmisaka
643*437bfbebSnyanmisaka mpp_assert(regs);
644*437bfbebSnyanmisaka }
645*437bfbebSnyanmisaka
646*437bfbebSnyanmisaka regs = reg_ctx->regs;
647*437bfbebSnyanmisaka
648*437bfbebSnyanmisaka prepare_header(p_hal, reg_ctx->shph_dat, sizeof(reg_ctx->shph_dat));
649*437bfbebSnyanmisaka prepare_scalist(p_hal, reg_ctx->scalist_dat, sizeof(reg_ctx->scalist_dat));
650*437bfbebSnyanmisaka
651*437bfbebSnyanmisaka ret = fill_registers(p_hal, regs, task);
652*437bfbebSnyanmisaka
653*437bfbebSnyanmisaka if (ret)
654*437bfbebSnyanmisaka goto __RETURN;
655*437bfbebSnyanmisaka
656*437bfbebSnyanmisaka {
657*437bfbebSnyanmisaka memcpy(reg_ctx->bufs_ptr + reg_ctx->shph_offset, reg_ctx->shph_dat, sizeof(reg_ctx->shph_dat));
658*437bfbebSnyanmisaka memcpy(reg_ctx->bufs_ptr + reg_ctx->sclst_offset, reg_ctx->scalist_dat, sizeof(reg_ctx->scalist_dat));
659*437bfbebSnyanmisaka regs->common.reg012.scanlist_addr_valid_en = 1;
660*437bfbebSnyanmisaka
661*437bfbebSnyanmisaka regs->avs2d_addr.head_base = reg_ctx->bufs_fd;
662*437bfbebSnyanmisaka mpp_dev_set_reg_offset(p_hal->dev, 161, reg_ctx->shph_offset);
663*437bfbebSnyanmisaka
664*437bfbebSnyanmisaka regs->avs2d_param.reg105.head_len = AVS2_RKV_SHPH_SIZE / 16;
665*437bfbebSnyanmisaka regs->avs2d_param.reg105.head_len -= (regs->avs2d_param.reg105.head_len > 0) ? 1 : 0;
666*437bfbebSnyanmisaka
667*437bfbebSnyanmisaka regs->avs2d_addr.scanlist_addr = reg_ctx->bufs_fd;
668*437bfbebSnyanmisaka mpp_dev_set_reg_offset(p_hal->dev, 180, reg_ctx->sclst_offset);
669*437bfbebSnyanmisaka }
670*437bfbebSnyanmisaka
671*437bfbebSnyanmisaka if (avs2d_hal_debug & AVS2D_HAL_DBG_IN) {
672*437bfbebSnyanmisaka FILE *fp_shph = NULL;
673*437bfbebSnyanmisaka char name[50];
674*437bfbebSnyanmisaka snprintf(name, sizeof(name), "/data/tmp/rkv_shph_%03d.bin", p_hal->frame_no);
675*437bfbebSnyanmisaka fp_shph = fopen(name, "wb");
676*437bfbebSnyanmisaka fwrite(reg_ctx->bufs_ptr + reg_ctx->shph_offset, 1, sizeof(reg_ctx->shph_dat), fp_shph);
677*437bfbebSnyanmisaka fclose(fp_shph);
678*437bfbebSnyanmisaka }
679*437bfbebSnyanmisaka
680*437bfbebSnyanmisaka if (avs2d_hal_debug & AVS2D_HAL_DBG_IN) {
681*437bfbebSnyanmisaka FILE *fp_scalist = NULL;
682*437bfbebSnyanmisaka char name[50];
683*437bfbebSnyanmisaka snprintf(name, sizeof(name), "/data/tmp/rkv_scalist_%03d.bin", p_hal->frame_no);
684*437bfbebSnyanmisaka fp_scalist = fopen(name, "wb");
685*437bfbebSnyanmisaka fwrite(reg_ctx->bufs_ptr + reg_ctx->sclst_offset, 1, sizeof(reg_ctx->scalist_dat), fp_scalist);
686*437bfbebSnyanmisaka fclose(fp_scalist);
687*437bfbebSnyanmisaka }
688*437bfbebSnyanmisaka
689*437bfbebSnyanmisaka // set rcb
690*437bfbebSnyanmisaka {
691*437bfbebSnyanmisaka hal_avs2d_rcb_info_update(p_hal, regs);
692*437bfbebSnyanmisaka vdpu34x_setup_rcb(®s->common_addr, p_hal->dev, p_hal->fast_mode ?
693*437bfbebSnyanmisaka reg_ctx->rcb_buf[task->dec.reg_index] : reg_ctx->rcb_buf[0],
694*437bfbebSnyanmisaka reg_ctx->rcb_info);
695*437bfbebSnyanmisaka
696*437bfbebSnyanmisaka }
697*437bfbebSnyanmisaka
698*437bfbebSnyanmisaka if (avs2d_hal_debug & AVS2D_HAL_DBG_IN) {
699*437bfbebSnyanmisaka FILE *fp_rcb = NULL;
700*437bfbebSnyanmisaka char name[50];
701*437bfbebSnyanmisaka void *base = NULL;
702*437bfbebSnyanmisaka snprintf(name, sizeof(name), "/data/tmp/rkv_rcb_%03d.bin", p_hal->frame_no);
703*437bfbebSnyanmisaka fp_rcb = fopen(name, "wb");
704*437bfbebSnyanmisaka base = mpp_buffer_get_ptr(reg_ctx->rcb_buf[0]);
705*437bfbebSnyanmisaka fwrite(base, 1, reg_ctx->rcb_buf_size, fp_rcb);
706*437bfbebSnyanmisaka fclose(fp_rcb);
707*437bfbebSnyanmisaka
708*437bfbebSnyanmisaka }
709*437bfbebSnyanmisaka
710*437bfbebSnyanmisaka vdpu34x_setup_statistic(®s->common, ®s->statistic);
711*437bfbebSnyanmisaka mpp_buffer_sync_end(reg_ctx->bufs);
712*437bfbebSnyanmisaka
713*437bfbebSnyanmisaka /* enable reference frame usage feedback */
714*437bfbebSnyanmisaka regs->statistic.reg265.perf_cnt0_sel = 42;
715*437bfbebSnyanmisaka regs->statistic.reg266_perf_cnt0 = 0;
716*437bfbebSnyanmisaka
717*437bfbebSnyanmisaka __RETURN:
718*437bfbebSnyanmisaka AVS2D_HAL_TRACE("Out. ret %d", ret);
719*437bfbebSnyanmisaka return ret;
720*437bfbebSnyanmisaka }
721*437bfbebSnyanmisaka
hal_avs2d_rkv_dump_reg_write(void * hal,Vdpu34xAvs2dRegSet * regs)722*437bfbebSnyanmisaka static MPP_RET hal_avs2d_rkv_dump_reg_write(void *hal, Vdpu34xAvs2dRegSet *regs)
723*437bfbebSnyanmisaka {
724*437bfbebSnyanmisaka MPP_RET ret = MPP_OK;
725*437bfbebSnyanmisaka Avs2dHalCtx_t *p_hal = (Avs2dHalCtx_t *)hal;
726*437bfbebSnyanmisaka FILE *fp_reg = NULL;
727*437bfbebSnyanmisaka RK_U32 i = 0;
728*437bfbebSnyanmisaka char name[50];
729*437bfbebSnyanmisaka snprintf(name, sizeof(name), "/data/tmp/rkv_reg_write_%03d.txt", p_hal->frame_no);
730*437bfbebSnyanmisaka fp_reg = fopen(name , "w+");
731*437bfbebSnyanmisaka
732*437bfbebSnyanmisaka fprintf(fp_reg, "********Frame num %d\n", p_hal->frame_no);
733*437bfbebSnyanmisaka for (i = 0; i < 8; i++)
734*437bfbebSnyanmisaka fprintf(fp_reg, "Write reg[%03d] : 0x%08x\n", i, 0);
735*437bfbebSnyanmisaka
736*437bfbebSnyanmisaka for (i = 0; i < sizeof(Vdpu34xRegCommon) / sizeof(RK_U32); i++)
737*437bfbebSnyanmisaka fprintf(fp_reg, "Write reg[%03d] : 0x%08x\n", (RK_U32)(i + OFFSET_COMMON_REGS / sizeof(RK_U32)),
738*437bfbebSnyanmisaka ((RK_U32 *)®s->common)[i]);
739*437bfbebSnyanmisaka
740*437bfbebSnyanmisaka for (i = 0; i < 63 - 32; i++)
741*437bfbebSnyanmisaka fprintf(fp_reg, "Write reg[%03d] : 0x%08x\n", i + 33, 0);
742*437bfbebSnyanmisaka
743*437bfbebSnyanmisaka for (i = 0; i < sizeof(Vdpu34xRegAvs2dParam) / sizeof(RK_U32); i++)
744*437bfbebSnyanmisaka fprintf(fp_reg, "Write reg[%03d] : 0x%08x\n", (RK_U32)(i + OFFSET_CODEC_PARAMS_REGS / sizeof(RK_U32)),
745*437bfbebSnyanmisaka ((RK_U32 *)®s->avs2d_param)[i]);
746*437bfbebSnyanmisaka
747*437bfbebSnyanmisaka for (i = 0; i < 127 - 112; i++)
748*437bfbebSnyanmisaka fprintf(fp_reg, "Write reg[%03d] : 0x%08x\n", i + 113, 0);
749*437bfbebSnyanmisaka
750*437bfbebSnyanmisaka for (i = 0; i < sizeof(Vdpu34xRegCommonAddr) / sizeof(RK_U32); i++)
751*437bfbebSnyanmisaka fprintf(fp_reg, "Write reg[%03d] : 0x%08x\n", (RK_U32)(i + OFFSET_COMMON_ADDR_REGS / sizeof(RK_U32)),
752*437bfbebSnyanmisaka ((RK_U32 *)®s->common_addr)[i]);
753*437bfbebSnyanmisaka
754*437bfbebSnyanmisaka for (i = 0; i < 159 - 142; i++)
755*437bfbebSnyanmisaka fprintf(fp_reg, "Write reg[%03d] : 0x%08x\n", i + 143, 0);
756*437bfbebSnyanmisaka
757*437bfbebSnyanmisaka
758*437bfbebSnyanmisaka for (i = 0; i < sizeof(Vdpu34xRegAvs2dAddr) / sizeof(RK_U32); i++ )
759*437bfbebSnyanmisaka fprintf(fp_reg, "Write reg[%03d] : 0x%08x\n", (RK_U32)(i + OFFSET_CODEC_ADDR_REGS / sizeof(RK_U32)),
760*437bfbebSnyanmisaka ((RK_U32 *)®s->avs2d_addr)[i]);
761*437bfbebSnyanmisaka
762*437bfbebSnyanmisaka for (i = 0; i < 223 - 197; i++)
763*437bfbebSnyanmisaka fprintf(fp_reg, "Write reg[%03d] : 0x%08x\n", i + 198, 0);
764*437bfbebSnyanmisaka
765*437bfbebSnyanmisaka for (i = 0; i < sizeof(Vdpu34xRegIrqStatus) / sizeof(RK_U32); i++ )
766*437bfbebSnyanmisaka fprintf(fp_reg, "Write reg[%03d] : 0x%08x\n", (RK_U32)(i + OFFSET_INTERRUPT_REGS / sizeof(RK_U32)),
767*437bfbebSnyanmisaka ((RK_U32 *)®s->irq_status)[i]);
768*437bfbebSnyanmisaka
769*437bfbebSnyanmisaka for (i = 0; i < 255 - 237; i++)
770*437bfbebSnyanmisaka fprintf(fp_reg, "Write reg[%03d] : 0x%08x\n", i + 238, 0);
771*437bfbebSnyanmisaka
772*437bfbebSnyanmisaka for (i = 0; i < sizeof(Vdpu34xRegStatistic) / sizeof(RK_U32); i++ )
773*437bfbebSnyanmisaka fprintf(fp_reg, "Write reg[%03d] : 0x%08x\n", (RK_U32)(i + OFFSET_STATISTIC_REGS / sizeof(RK_U32)),
774*437bfbebSnyanmisaka ((RK_U32 *)®s->statistic)[i]);
775*437bfbebSnyanmisaka
776*437bfbebSnyanmisaka fclose(fp_reg);
777*437bfbebSnyanmisaka return ret;
778*437bfbebSnyanmisaka }
779*437bfbebSnyanmisaka
hal_avs2d_rkv_dump_stream(void * hal,HalTaskInfo * task)780*437bfbebSnyanmisaka static MPP_RET hal_avs2d_rkv_dump_stream(void *hal, HalTaskInfo *task)
781*437bfbebSnyanmisaka {
782*437bfbebSnyanmisaka MPP_RET ret = MPP_OK;
783*437bfbebSnyanmisaka Avs2dHalCtx_t *p_hal = (Avs2dHalCtx_t *)hal;
784*437bfbebSnyanmisaka
785*437bfbebSnyanmisaka FILE *fp_stream = NULL;
786*437bfbebSnyanmisaka char name[50];
787*437bfbebSnyanmisaka MppBuffer buffer = NULL;
788*437bfbebSnyanmisaka void *base = NULL;
789*437bfbebSnyanmisaka mpp_buf_slot_get_prop(p_hal->packet_slots, task->dec.input, SLOT_BUFFER, &buffer);
790*437bfbebSnyanmisaka base = mpp_buffer_get_ptr(buffer);
791*437bfbebSnyanmisaka snprintf(name, sizeof(name), "/data/tmp/rkv_stream_in_%03d.bin", p_hal->frame_no);
792*437bfbebSnyanmisaka fp_stream = fopen(name, "wb");
793*437bfbebSnyanmisaka fwrite(base, 1, mpp_packet_get_length(task->dec.input_packet), fp_stream);
794*437bfbebSnyanmisaka fclose(fp_stream);
795*437bfbebSnyanmisaka
796*437bfbebSnyanmisaka return ret;
797*437bfbebSnyanmisaka }
798*437bfbebSnyanmisaka
hal_avs2d_rkv_start(void * hal,HalTaskInfo * task)799*437bfbebSnyanmisaka MPP_RET hal_avs2d_rkv_start(void *hal, HalTaskInfo *task)
800*437bfbebSnyanmisaka {
801*437bfbebSnyanmisaka MPP_RET ret = MPP_OK;
802*437bfbebSnyanmisaka Vdpu34xAvs2dRegSet *regs = NULL;
803*437bfbebSnyanmisaka Avs2dRkvRegCtx_t *reg_ctx;
804*437bfbebSnyanmisaka MppDev dev = NULL;
805*437bfbebSnyanmisaka Avs2dHalCtx_t *p_hal = (Avs2dHalCtx_t *)hal;
806*437bfbebSnyanmisaka
807*437bfbebSnyanmisaka AVS2D_HAL_TRACE("In.");
808*437bfbebSnyanmisaka INP_CHECK(ret, NULL == p_hal);
809*437bfbebSnyanmisaka
810*437bfbebSnyanmisaka if ((task->dec.flags.parse_err || task->dec.flags.ref_err) &&
811*437bfbebSnyanmisaka !p_hal->cfg->base.disable_error) {
812*437bfbebSnyanmisaka ret = MPP_NOK;
813*437bfbebSnyanmisaka goto __RETURN;
814*437bfbebSnyanmisaka }
815*437bfbebSnyanmisaka
816*437bfbebSnyanmisaka reg_ctx = (Avs2dRkvRegCtx_t *)p_hal->reg_ctx;
817*437bfbebSnyanmisaka regs = p_hal->fast_mode ? reg_ctx->reg_buf[task->dec.reg_index].regs : reg_ctx->regs;
818*437bfbebSnyanmisaka dev = p_hal->dev;
819*437bfbebSnyanmisaka
820*437bfbebSnyanmisaka p_hal->frame_no++;
821*437bfbebSnyanmisaka
822*437bfbebSnyanmisaka do {
823*437bfbebSnyanmisaka MppDevRegWrCfg wr_cfg;
824*437bfbebSnyanmisaka MppDevRegRdCfg rd_cfg;
825*437bfbebSnyanmisaka
826*437bfbebSnyanmisaka wr_cfg.reg = ®s->common;
827*437bfbebSnyanmisaka wr_cfg.size = sizeof(regs->common);
828*437bfbebSnyanmisaka wr_cfg.offset = OFFSET_COMMON_REGS;
829*437bfbebSnyanmisaka
830*437bfbebSnyanmisaka ret = mpp_dev_ioctl(dev, MPP_DEV_REG_WR, &wr_cfg);
831*437bfbebSnyanmisaka
832*437bfbebSnyanmisaka if (ret) {
833*437bfbebSnyanmisaka mpp_err_f("set register write failed %d\n", ret);
834*437bfbebSnyanmisaka break;
835*437bfbebSnyanmisaka }
836*437bfbebSnyanmisaka
837*437bfbebSnyanmisaka wr_cfg.reg = ®s->avs2d_param;
838*437bfbebSnyanmisaka wr_cfg.size = sizeof(regs->avs2d_param);
839*437bfbebSnyanmisaka wr_cfg.offset = OFFSET_CODEC_PARAMS_REGS;
840*437bfbebSnyanmisaka
841*437bfbebSnyanmisaka ret = mpp_dev_ioctl(dev, MPP_DEV_REG_WR, &wr_cfg);
842*437bfbebSnyanmisaka
843*437bfbebSnyanmisaka if (ret) {
844*437bfbebSnyanmisaka mpp_err_f("set register write failed %d\n", ret);
845*437bfbebSnyanmisaka break;
846*437bfbebSnyanmisaka }
847*437bfbebSnyanmisaka
848*437bfbebSnyanmisaka wr_cfg.reg = ®s->common_addr;
849*437bfbebSnyanmisaka wr_cfg.size = sizeof(regs->common_addr);
850*437bfbebSnyanmisaka wr_cfg.offset = OFFSET_COMMON_ADDR_REGS;
851*437bfbebSnyanmisaka
852*437bfbebSnyanmisaka ret = mpp_dev_ioctl(dev, MPP_DEV_REG_WR, &wr_cfg);
853*437bfbebSnyanmisaka
854*437bfbebSnyanmisaka if (ret) {
855*437bfbebSnyanmisaka mpp_err_f("set register write failed %d\n", ret);
856*437bfbebSnyanmisaka break;
857*437bfbebSnyanmisaka }
858*437bfbebSnyanmisaka
859*437bfbebSnyanmisaka wr_cfg.reg = ®s->avs2d_addr;
860*437bfbebSnyanmisaka wr_cfg.size = sizeof(regs->avs2d_addr);
861*437bfbebSnyanmisaka wr_cfg.offset = OFFSET_CODEC_ADDR_REGS;
862*437bfbebSnyanmisaka
863*437bfbebSnyanmisaka ret = mpp_dev_ioctl(dev, MPP_DEV_REG_WR, &wr_cfg);
864*437bfbebSnyanmisaka
865*437bfbebSnyanmisaka if (ret) {
866*437bfbebSnyanmisaka mpp_err_f("set register write failed %d\n", ret);
867*437bfbebSnyanmisaka break;
868*437bfbebSnyanmisaka }
869*437bfbebSnyanmisaka
870*437bfbebSnyanmisaka wr_cfg.reg = ®s->statistic;
871*437bfbebSnyanmisaka wr_cfg.size = sizeof(regs->statistic);
872*437bfbebSnyanmisaka wr_cfg.offset = OFFSET_STATISTIC_REGS;
873*437bfbebSnyanmisaka ret = mpp_dev_ioctl(dev, MPP_DEV_REG_WR, &wr_cfg);
874*437bfbebSnyanmisaka
875*437bfbebSnyanmisaka if (ret) {
876*437bfbebSnyanmisaka mpp_err_f("set register write failed %d\n", ret);
877*437bfbebSnyanmisaka break;
878*437bfbebSnyanmisaka }
879*437bfbebSnyanmisaka
880*437bfbebSnyanmisaka rd_cfg.reg = ®s->irq_status;
881*437bfbebSnyanmisaka rd_cfg.size = sizeof(regs->irq_status);
882*437bfbebSnyanmisaka rd_cfg.offset = OFFSET_INTERRUPT_REGS;
883*437bfbebSnyanmisaka ret = mpp_dev_ioctl(dev, MPP_DEV_REG_RD, &rd_cfg);
884*437bfbebSnyanmisaka
885*437bfbebSnyanmisaka if (ret) {
886*437bfbebSnyanmisaka mpp_err_f("set register read failed %d\n", ret);
887*437bfbebSnyanmisaka break;
888*437bfbebSnyanmisaka }
889*437bfbebSnyanmisaka
890*437bfbebSnyanmisaka rd_cfg.reg = ®s->avs2d_param;
891*437bfbebSnyanmisaka rd_cfg.size = sizeof(regs->avs2d_param);
892*437bfbebSnyanmisaka rd_cfg.offset = OFFSET_CODEC_PARAMS_REGS;
893*437bfbebSnyanmisaka ret = mpp_dev_ioctl(dev, MPP_DEV_REG_RD, &rd_cfg);
894*437bfbebSnyanmisaka
895*437bfbebSnyanmisaka if (ret) {
896*437bfbebSnyanmisaka mpp_err_f("set register read failed %d\n", ret);
897*437bfbebSnyanmisaka break;
898*437bfbebSnyanmisaka }
899*437bfbebSnyanmisaka
900*437bfbebSnyanmisaka rd_cfg.reg = ®s->statistic;
901*437bfbebSnyanmisaka rd_cfg.size = sizeof(regs->statistic);
902*437bfbebSnyanmisaka rd_cfg.offset = OFFSET_STATISTIC_REGS;
903*437bfbebSnyanmisaka ret = mpp_dev_ioctl(dev, MPP_DEV_REG_RD, &rd_cfg);
904*437bfbebSnyanmisaka
905*437bfbebSnyanmisaka if (ret) {
906*437bfbebSnyanmisaka mpp_err_f("set register write failed %d\n", ret);
907*437bfbebSnyanmisaka break;
908*437bfbebSnyanmisaka }
909*437bfbebSnyanmisaka
910*437bfbebSnyanmisaka if (avs2d_hal_debug & AVS2D_HAL_DBG_REG) {
911*437bfbebSnyanmisaka memset(reg_ctx->reg_out, 0, sizeof(reg_ctx->reg_out));
912*437bfbebSnyanmisaka rd_cfg.reg = reg_ctx->reg_out;
913*437bfbebSnyanmisaka rd_cfg.size = sizeof(reg_ctx->reg_out);
914*437bfbebSnyanmisaka rd_cfg.offset = 0;
915*437bfbebSnyanmisaka ret = mpp_dev_ioctl(dev, MPP_DEV_REG_RD, &rd_cfg);
916*437bfbebSnyanmisaka }
917*437bfbebSnyanmisaka
918*437bfbebSnyanmisaka // rcb info for sram
919*437bfbebSnyanmisaka vdpu34x_set_rcbinfo(dev, reg_ctx->rcb_info);
920*437bfbebSnyanmisaka
921*437bfbebSnyanmisaka if (avs2d_hal_debug & AVS2D_HAL_DBG_IN)
922*437bfbebSnyanmisaka hal_avs2d_rkv_dump_stream(hal, task);
923*437bfbebSnyanmisaka
924*437bfbebSnyanmisaka if (avs2d_hal_debug & AVS2D_HAL_DBG_REG)
925*437bfbebSnyanmisaka hal_avs2d_rkv_dump_reg_write(hal, regs);
926*437bfbebSnyanmisaka
927*437bfbebSnyanmisaka // send request to hardware
928*437bfbebSnyanmisaka ret = mpp_dev_ioctl(dev, MPP_DEV_CMD_SEND, NULL);
929*437bfbebSnyanmisaka if (ret) {
930*437bfbebSnyanmisaka mpp_err_f("send cmd failed %d\n", ret);
931*437bfbebSnyanmisaka break;
932*437bfbebSnyanmisaka }
933*437bfbebSnyanmisaka
934*437bfbebSnyanmisaka } while (0);
935*437bfbebSnyanmisaka
936*437bfbebSnyanmisaka __RETURN:
937*437bfbebSnyanmisaka AVS2D_HAL_TRACE("Out.");
938*437bfbebSnyanmisaka return ret;
939*437bfbebSnyanmisaka }
940*437bfbebSnyanmisaka
941*437bfbebSnyanmisaka
fetch_data(RK_U32 fmt,RK_U8 * line,RK_U32 num)942*437bfbebSnyanmisaka static RK_U8 fetch_data(RK_U32 fmt, RK_U8 *line, RK_U32 num)
943*437bfbebSnyanmisaka {
944*437bfbebSnyanmisaka RK_U32 offset = 0;
945*437bfbebSnyanmisaka RK_U32 value = 0;
946*437bfbebSnyanmisaka
947*437bfbebSnyanmisaka if (fmt == MPP_FMT_YUV420SP_10BIT) {
948*437bfbebSnyanmisaka offset = (num * 2) & 7;
949*437bfbebSnyanmisaka value = (line[num * 10 / 8] >> offset) |
950*437bfbebSnyanmisaka (line[num * 10 / 8 + 1] << (8 - offset));
951*437bfbebSnyanmisaka
952*437bfbebSnyanmisaka value = (value & 0x3ff) >> 2;
953*437bfbebSnyanmisaka } else if (fmt == MPP_FMT_YUV420SP) {
954*437bfbebSnyanmisaka value = line[num];
955*437bfbebSnyanmisaka }
956*437bfbebSnyanmisaka
957*437bfbebSnyanmisaka return value;
958*437bfbebSnyanmisaka }
959*437bfbebSnyanmisaka
hal_avs2d_rkv_dump_yuv(void * hal,HalTaskInfo * task)960*437bfbebSnyanmisaka static MPP_RET hal_avs2d_rkv_dump_yuv(void *hal, HalTaskInfo *task)
961*437bfbebSnyanmisaka {
962*437bfbebSnyanmisaka MPP_RET ret = MPP_OK;
963*437bfbebSnyanmisaka Avs2dHalCtx_t *p_hal = (Avs2dHalCtx_t *)hal;
964*437bfbebSnyanmisaka
965*437bfbebSnyanmisaka MppFrameFormat fmt = MPP_FMT_YUV420SP;
966*437bfbebSnyanmisaka RK_U32 vir_w = 0;
967*437bfbebSnyanmisaka RK_U32 vir_h = 0;
968*437bfbebSnyanmisaka RK_U32 i = 0;
969*437bfbebSnyanmisaka RK_U32 j = 0;
970*437bfbebSnyanmisaka FILE *fp_stream = NULL;
971*437bfbebSnyanmisaka char name[50];
972*437bfbebSnyanmisaka MppBuffer buffer = NULL;
973*437bfbebSnyanmisaka MppFrame frame;
974*437bfbebSnyanmisaka void *base = NULL;
975*437bfbebSnyanmisaka
976*437bfbebSnyanmisaka ret = mpp_buf_slot_get_prop(p_hal->frame_slots, task->dec.output, SLOT_FRAME_PTR, &frame);
977*437bfbebSnyanmisaka
978*437bfbebSnyanmisaka if (ret != MPP_OK || frame == NULL)
979*437bfbebSnyanmisaka mpp_log_f("failed to get frame slot %d", task->dec.output);
980*437bfbebSnyanmisaka
981*437bfbebSnyanmisaka ret = mpp_buf_slot_get_prop(p_hal->frame_slots, task->dec.output, SLOT_BUFFER, &buffer);
982*437bfbebSnyanmisaka
983*437bfbebSnyanmisaka if (ret != MPP_OK || buffer == NULL)
984*437bfbebSnyanmisaka mpp_log_f("failed to get frame buffer slot %d", task->dec.output);
985*437bfbebSnyanmisaka
986*437bfbebSnyanmisaka AVS2D_HAL_TRACE("frame slot %d, fd %d\n", task->dec.output, mpp_buffer_get_fd(buffer));
987*437bfbebSnyanmisaka base = mpp_buffer_get_ptr(buffer);
988*437bfbebSnyanmisaka vir_w = mpp_frame_get_hor_stride(frame);
989*437bfbebSnyanmisaka vir_h = mpp_frame_get_ver_stride(frame);
990*437bfbebSnyanmisaka fmt = mpp_frame_get_fmt(frame);
991*437bfbebSnyanmisaka snprintf(name, sizeof(name), "/data/tmp/rkv_out_%dx%d_nv12_%03d.yuv", vir_w, vir_h,
992*437bfbebSnyanmisaka p_hal->frame_no);
993*437bfbebSnyanmisaka fp_stream = fopen(name, "wb");
994*437bfbebSnyanmisaka /* if format is fbc, write fbc header first */
995*437bfbebSnyanmisaka if (MPP_FRAME_FMT_IS_FBC(fmt)) {
996*437bfbebSnyanmisaka RK_U32 header_size = 0;
997*437bfbebSnyanmisaka
998*437bfbebSnyanmisaka header_size = vir_w * vir_h / 16;
999*437bfbebSnyanmisaka fwrite(base, 1, header_size, fp_stream);
1000*437bfbebSnyanmisaka base += header_size;
1001*437bfbebSnyanmisaka }
1002*437bfbebSnyanmisaka
1003*437bfbebSnyanmisaka if (fmt != MPP_FMT_YUV420SP_10BIT) {
1004*437bfbebSnyanmisaka fwrite(base, 1, vir_w * vir_h * 3 / 2, fp_stream);
1005*437bfbebSnyanmisaka } else {
1006*437bfbebSnyanmisaka RK_U8 tmp = 0;
1007*437bfbebSnyanmisaka for (i = 0; i < vir_h; i++) {
1008*437bfbebSnyanmisaka for (j = 0; j < vir_w; j++) {
1009*437bfbebSnyanmisaka tmp = fetch_data(fmt, base, j);
1010*437bfbebSnyanmisaka fwrite(&tmp, 1, 1, fp_stream);
1011*437bfbebSnyanmisaka }
1012*437bfbebSnyanmisaka base += vir_w;
1013*437bfbebSnyanmisaka }
1014*437bfbebSnyanmisaka
1015*437bfbebSnyanmisaka for (i = 0; i < vir_h / 2; i++) {
1016*437bfbebSnyanmisaka for (j = 0; j < vir_w; j++) {
1017*437bfbebSnyanmisaka tmp = fetch_data(fmt, base, j);
1018*437bfbebSnyanmisaka fwrite(&tmp, 1, 1, fp_stream);
1019*437bfbebSnyanmisaka }
1020*437bfbebSnyanmisaka base += vir_w;
1021*437bfbebSnyanmisaka }
1022*437bfbebSnyanmisaka }
1023*437bfbebSnyanmisaka fclose(fp_stream);
1024*437bfbebSnyanmisaka
1025*437bfbebSnyanmisaka return ret;
1026*437bfbebSnyanmisaka }
1027*437bfbebSnyanmisaka
hal_avs2d_rkv_wait(void * hal,HalTaskInfo * task)1028*437bfbebSnyanmisaka MPP_RET hal_avs2d_rkv_wait(void *hal, HalTaskInfo *task)
1029*437bfbebSnyanmisaka {
1030*437bfbebSnyanmisaka MPP_RET ret = MPP_OK;
1031*437bfbebSnyanmisaka Avs2dHalCtx_t *p_hal = (Avs2dHalCtx_t *)hal;
1032*437bfbebSnyanmisaka Avs2dRkvRegCtx_t *reg_ctx;
1033*437bfbebSnyanmisaka Vdpu34xAvs2dRegSet *p_regs;
1034*437bfbebSnyanmisaka
1035*437bfbebSnyanmisaka INP_CHECK(ret, NULL == p_hal);
1036*437bfbebSnyanmisaka reg_ctx = (Avs2dRkvRegCtx_t *)p_hal->reg_ctx;
1037*437bfbebSnyanmisaka p_regs = p_hal->fast_mode ? reg_ctx->reg_buf[task->dec.reg_index].regs : reg_ctx->regs;
1038*437bfbebSnyanmisaka
1039*437bfbebSnyanmisaka if ((task->dec.flags.parse_err || task->dec.flags.ref_err) &&
1040*437bfbebSnyanmisaka !p_hal->cfg->base.disable_error) {
1041*437bfbebSnyanmisaka AVS2D_HAL_DBG(AVS2D_HAL_DBG_ERROR, "found task error.\n");
1042*437bfbebSnyanmisaka ret = MPP_NOK;
1043*437bfbebSnyanmisaka goto __RETURN;
1044*437bfbebSnyanmisaka } else {
1045*437bfbebSnyanmisaka ret = mpp_dev_ioctl(p_hal->dev, MPP_DEV_CMD_POLL, NULL);
1046*437bfbebSnyanmisaka if (ret)
1047*437bfbebSnyanmisaka mpp_err_f("poll cmd failed %d\n", ret);
1048*437bfbebSnyanmisaka }
1049*437bfbebSnyanmisaka
1050*437bfbebSnyanmisaka if (avs2d_hal_debug & AVS2D_HAL_DBG_OUT)
1051*437bfbebSnyanmisaka hal_avs2d_rkv_dump_yuv(hal, task);
1052*437bfbebSnyanmisaka
1053*437bfbebSnyanmisaka if (avs2d_hal_debug & AVS2D_HAL_DBG_REG) {
1054*437bfbebSnyanmisaka FILE *fp_reg = NULL;
1055*437bfbebSnyanmisaka RK_U32 i = 0;
1056*437bfbebSnyanmisaka char name[50];
1057*437bfbebSnyanmisaka snprintf(name, sizeof(name), "/data/tmp/rkv_reg_read_%03d.txt", p_hal->frame_no);
1058*437bfbebSnyanmisaka fp_reg = fopen(name , "w+");
1059*437bfbebSnyanmisaka
1060*437bfbebSnyanmisaka for (i = 0; i < 278; i++)
1061*437bfbebSnyanmisaka fprintf(fp_reg, "%08x\n", reg_ctx->reg_out[i]);
1062*437bfbebSnyanmisaka
1063*437bfbebSnyanmisaka fclose(fp_reg);
1064*437bfbebSnyanmisaka }
1065*437bfbebSnyanmisaka
1066*437bfbebSnyanmisaka AVS2D_HAL_TRACE("read reg[224] 0x%08x\n", p_regs->irq_status.reg224);
1067*437bfbebSnyanmisaka
1068*437bfbebSnyanmisaka if (p_hal->dec_cb) {
1069*437bfbebSnyanmisaka DecCbHalDone param;
1070*437bfbebSnyanmisaka
1071*437bfbebSnyanmisaka param.task = (void *)&task->dec;
1072*437bfbebSnyanmisaka param.regs = (RK_U32 *)p_regs;
1073*437bfbebSnyanmisaka
1074*437bfbebSnyanmisaka if (p_regs->irq_status.reg224.dec_error_sta ||
1075*437bfbebSnyanmisaka (!p_regs->irq_status.reg224.dec_rdy_sta) ||
1076*437bfbebSnyanmisaka p_regs->irq_status.reg224.buf_empty_sta ||
1077*437bfbebSnyanmisaka p_regs->irq_status.reg226.strmd_error_status ||
1078*437bfbebSnyanmisaka p_regs->irq_status.reg227.colmv_error_ref_picidx ||
1079*437bfbebSnyanmisaka p_regs->irq_status.reg225.strmd_detect_error_flag)
1080*437bfbebSnyanmisaka param.hard_err = 1;
1081*437bfbebSnyanmisaka else
1082*437bfbebSnyanmisaka param.hard_err = 0;
1083*437bfbebSnyanmisaka
1084*437bfbebSnyanmisaka task->dec.flags.ref_used = p_regs->statistic.reg266_perf_cnt0;
1085*437bfbebSnyanmisaka task->dec.flags.ref_info_valid = 1;
1086*437bfbebSnyanmisaka
1087*437bfbebSnyanmisaka if (task->dec.flags.ref_miss) {
1088*437bfbebSnyanmisaka RK_U32 ref_hw_usage = p_regs->statistic.reg266_perf_cnt0;
1089*437bfbebSnyanmisaka
1090*437bfbebSnyanmisaka AVS2D_HAL_TRACE("hal frame %d ref miss %x hard_err %d hw_usage %x", p_hal->frame_no,
1091*437bfbebSnyanmisaka task->dec.flags.ref_miss, param.hard_err, ref_hw_usage);
1092*437bfbebSnyanmisaka }
1093*437bfbebSnyanmisaka
1094*437bfbebSnyanmisaka AVS2D_HAL_TRACE("hal frame %d hard_err= %d", p_hal->frame_no, param.hard_err);
1095*437bfbebSnyanmisaka
1096*437bfbebSnyanmisaka mpp_callback(p_hal->dec_cb, ¶m);
1097*437bfbebSnyanmisaka }
1098*437bfbebSnyanmisaka
1099*437bfbebSnyanmisaka memset(&p_regs->irq_status.reg224, 0, sizeof(RK_U32));
1100*437bfbebSnyanmisaka
1101*437bfbebSnyanmisaka if (p_hal->fast_mode)
1102*437bfbebSnyanmisaka reg_ctx->reg_buf[task->dec.reg_index].valid = 0;
1103*437bfbebSnyanmisaka
1104*437bfbebSnyanmisaka __RETURN:
1105*437bfbebSnyanmisaka AVS2D_HAL_TRACE("Out. ret %d", ret);
1106*437bfbebSnyanmisaka return ret;
1107*437bfbebSnyanmisaka }
1108*437bfbebSnyanmisaka
1109*437bfbebSnyanmisaka const MppHalApi hal_avs2d_rkvdpu = {
1110*437bfbebSnyanmisaka .name = "avs2d_rkvdpu",
1111*437bfbebSnyanmisaka .type = MPP_CTX_DEC,
1112*437bfbebSnyanmisaka .coding = MPP_VIDEO_CodingAVS2,
1113*437bfbebSnyanmisaka .ctx_size = sizeof(Avs2dRkvRegCtx_t),
1114*437bfbebSnyanmisaka .flag = 0,
1115*437bfbebSnyanmisaka .init = hal_avs2d_rkv_init,
1116*437bfbebSnyanmisaka .deinit = hal_avs2d_rkv_deinit,
1117*437bfbebSnyanmisaka .reg_gen = hal_avs2d_rkv_gen_regs,
1118*437bfbebSnyanmisaka .start = hal_avs2d_rkv_start,
1119*437bfbebSnyanmisaka .wait = hal_avs2d_rkv_wait,
1120*437bfbebSnyanmisaka .reset = NULL,
1121*437bfbebSnyanmisaka .flush = NULL,
1122*437bfbebSnyanmisaka .control = NULL,
1123*437bfbebSnyanmisaka };
1124