yading@10: /* yading@10: * Chinese AVS video (AVS1-P2, JiZhun profile) decoder. yading@10: * yading@10: * DSP functions yading@10: * yading@10: * Copyright (c) 2006 Stefan Gehrer yading@10: * yading@10: * This file is part of FFmpeg. yading@10: * yading@10: * FFmpeg is free software; you can redistribute it and/or yading@10: * modify it under the terms of the GNU Lesser General Public yading@10: * License as published by the Free Software Foundation; either yading@10: * version 2.1 of the License, or (at your option) any later version. yading@10: * yading@10: * FFmpeg is distributed in the hope that it will be useful, yading@10: * but WITHOUT ANY WARRANTY; without even the implied warranty of yading@10: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU yading@10: * Lesser General Public License for more details. yading@10: * yading@10: * You should have received a copy of the GNU Lesser General Public yading@10: * License along with FFmpeg; if not, write to the Free Software yading@10: * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA yading@10: */ yading@10: yading@10: #include yading@10: #include "dsputil.h" yading@10: #include "cavsdsp.h" yading@10: #include "libavutil/common.h" yading@10: yading@10: /***************************************************************************** yading@10: * yading@10: * in-loop deblocking filter yading@10: * yading@10: ****************************************************************************/ yading@10: yading@10: #define P2 p0_p[-3*stride] yading@10: #define P1 p0_p[-2*stride] yading@10: #define P0 p0_p[-1*stride] yading@10: #define Q0 p0_p[ 0*stride] yading@10: #define Q1 p0_p[ 1*stride] yading@10: #define Q2 p0_p[ 2*stride] yading@10: yading@10: static inline void loop_filter_l2(uint8_t *p0_p,int stride,int alpha, int beta) { yading@10: int p0 = P0; yading@10: int q0 = Q0; yading@10: yading@10: if(abs(p0-q0)>2) + 2; yading@10: if(abs(P2-p0) < beta && abs(p0-q0) < alpha) { yading@10: P0 = (P1 + p0 + s) >> 2; yading@10: P1 = (2*P1 + s) >> 2; yading@10: } else yading@10: P0 = (2*P1 + s) >> 2; yading@10: if(abs(Q2-q0) < beta && abs(q0-p0) < alpha) { yading@10: Q0 = (Q1 + q0 + s) >> 2; yading@10: Q1 = (2*Q1 + s) >> 2; yading@10: } else yading@10: Q0 = (2*Q1 + s) >> 2; yading@10: } yading@10: } yading@10: yading@10: static inline void loop_filter_l1(uint8_t *p0_p, int stride, int alpha, int beta, int tc) { yading@10: int p0 = P0; yading@10: int q0 = Q0; yading@10: yading@10: if(abs(p0-q0)>3,-tc, tc); yading@10: P0 = av_clip_uint8(p0+delta); yading@10: Q0 = av_clip_uint8(q0-delta); yading@10: if(abs(P2-p0)>3, -tc, tc); yading@10: P1 = av_clip_uint8(P1+delta); yading@10: } yading@10: if(abs(Q2-q0)>3, -tc, tc); yading@10: Q1 = av_clip_uint8(Q1-delta); yading@10: } yading@10: } yading@10: } yading@10: yading@10: static inline void loop_filter_c2(uint8_t *p0_p,int stride,int alpha, int beta) { yading@10: int p0 = P0; yading@10: int q0 = Q0; yading@10: yading@10: if(abs(p0-q0)>2) + 2; yading@10: if(abs(P2-p0) < beta && abs(p0-q0) < alpha) { yading@10: P0 = (P1 + p0 + s) >> 2; yading@10: } else yading@10: P0 = (2*P1 + s) >> 2; yading@10: if(abs(Q2-q0) < beta && abs(q0-p0) < alpha) { yading@10: Q0 = (Q1 + q0 + s) >> 2; yading@10: } else yading@10: Q0 = (2*Q1 + s) >> 2; yading@10: } yading@10: } yading@10: yading@10: static inline void loop_filter_c1(uint8_t *p0_p,int stride,int alpha, int beta, yading@10: int tc) { yading@10: if(abs(P0-Q0)>3, -tc, tc); yading@10: P0 = av_clip_uint8(P0+delta); yading@10: Q0 = av_clip_uint8(Q0-delta); yading@10: } yading@10: } yading@10: yading@10: #undef P0 yading@10: #undef P1 yading@10: #undef P2 yading@10: #undef Q0 yading@10: #undef Q1 yading@10: #undef Q2 yading@10: yading@10: static void cavs_filter_lv_c(uint8_t *d, int stride, int alpha, int beta, int tc, yading@10: int bs1, int bs2) { yading@10: int i; yading@10: if(bs1==2) yading@10: for(i=0;i<16;i++) yading@10: loop_filter_l2(d + i*stride,1,alpha,beta); yading@10: else { yading@10: if(bs1) yading@10: for(i=0;i<8;i++) yading@10: loop_filter_l1(d + i*stride,1,alpha,beta,tc); yading@10: if (bs2) yading@10: for(i=8;i<16;i++) yading@10: loop_filter_l1(d + i*stride,1,alpha,beta,tc); yading@10: } yading@10: } yading@10: yading@10: static void cavs_filter_lh_c(uint8_t *d, int stride, int alpha, int beta, int tc, yading@10: int bs1, int bs2) { yading@10: int i; yading@10: if(bs1==2) yading@10: for(i=0;i<16;i++) yading@10: loop_filter_l2(d + i,stride,alpha,beta); yading@10: else { yading@10: if(bs1) yading@10: for(i=0;i<8;i++) yading@10: loop_filter_l1(d + i,stride,alpha,beta,tc); yading@10: if (bs2) yading@10: for(i=8;i<16;i++) yading@10: loop_filter_l1(d + i,stride,alpha,beta,tc); yading@10: } yading@10: } yading@10: yading@10: static void cavs_filter_cv_c(uint8_t *d, int stride, int alpha, int beta, int tc, yading@10: int bs1, int bs2) { yading@10: int i; yading@10: if(bs1==2) yading@10: for(i=0;i<8;i++) yading@10: loop_filter_c2(d + i*stride,1,alpha,beta); yading@10: else { yading@10: if(bs1) yading@10: for(i=0;i<4;i++) yading@10: loop_filter_c1(d + i*stride,1,alpha,beta,tc); yading@10: if (bs2) yading@10: for(i=4;i<8;i++) yading@10: loop_filter_c1(d + i*stride,1,alpha,beta,tc); yading@10: } yading@10: } yading@10: yading@10: static void cavs_filter_ch_c(uint8_t *d, int stride, int alpha, int beta, int tc, yading@10: int bs1, int bs2) { yading@10: int i; yading@10: if(bs1==2) yading@10: for(i=0;i<8;i++) yading@10: loop_filter_c2(d + i,stride,alpha,beta); yading@10: else { yading@10: if(bs1) yading@10: for(i=0;i<4;i++) yading@10: loop_filter_c1(d + i,stride,alpha,beta,tc); yading@10: if (bs2) yading@10: for(i=4;i<8;i++) yading@10: loop_filter_c1(d + i,stride,alpha,beta,tc); yading@10: } yading@10: } yading@10: yading@10: /***************************************************************************** yading@10: * yading@10: * inverse transform yading@10: * yading@10: ****************************************************************************/ yading@10: yading@10: static void cavs_idct8_add_c(uint8_t *dst, int16_t *block, int stride) { yading@10: int i; yading@10: int16_t (*src)[8] = (int16_t(*)[8])block; yading@10: const uint8_t *cm = ff_cropTbl + MAX_NEG_CROP; yading@10: yading@10: src[0][0] += 8; yading@10: yading@10: for( i = 0; i < 8; i++ ) { yading@10: const int a0 = 3*src[i][1] - (src[i][7]<<1); yading@10: const int a1 = 3*src[i][3] + (src[i][5]<<1); yading@10: const int a2 = (src[i][3]<<1) - 3*src[i][5]; yading@10: const int a3 = (src[i][1]<<1) + 3*src[i][7]; yading@10: yading@10: const int b4 = ((a0 + a1 + a3)<<1) + a1; yading@10: const int b5 = ((a0 - a1 + a2)<<1) + a0; yading@10: const int b6 = ((a3 - a2 - a1)<<1) + a3; yading@10: const int b7 = ((a0 - a2 - a3)<<1) - a2; yading@10: yading@10: const int a7 = (src[i][2]<<2) - 10*src[i][6]; yading@10: const int a6 = (src[i][6]<<2) + 10*src[i][2]; yading@10: const int a5 = ((src[i][0] - src[i][4]) << 3) + 4; yading@10: const int a4 = ((src[i][0] + src[i][4]) << 3) + 4; yading@10: yading@10: const int b0 = a4 + a6; yading@10: const int b1 = a5 + a7; yading@10: const int b2 = a5 - a7; yading@10: const int b3 = a4 - a6; yading@10: yading@10: src[i][0] = (b0 + b4) >> 3; yading@10: src[i][1] = (b1 + b5) >> 3; yading@10: src[i][2] = (b2 + b6) >> 3; yading@10: src[i][3] = (b3 + b7) >> 3; yading@10: src[i][4] = (b3 - b7) >> 3; yading@10: src[i][5] = (b2 - b6) >> 3; yading@10: src[i][6] = (b1 - b5) >> 3; yading@10: src[i][7] = (b0 - b4) >> 3; yading@10: } yading@10: for( i = 0; i < 8; i++ ) { yading@10: const int a0 = 3*src[1][i] - (src[7][i]<<1); yading@10: const int a1 = 3*src[3][i] + (src[5][i]<<1); yading@10: const int a2 = (src[3][i]<<1) - 3*src[5][i]; yading@10: const int a3 = (src[1][i]<<1) + 3*src[7][i]; yading@10: yading@10: const int b4 = ((a0 + a1 + a3)<<1) + a1; yading@10: const int b5 = ((a0 - a1 + a2)<<1) + a0; yading@10: const int b6 = ((a3 - a2 - a1)<<1) + a3; yading@10: const int b7 = ((a0 - a2 - a3)<<1) - a2; yading@10: yading@10: const int a7 = (src[2][i]<<2) - 10*src[6][i]; yading@10: const int a6 = (src[6][i]<<2) + 10*src[2][i]; yading@10: const int a5 = (src[0][i] - src[4][i]) << 3; yading@10: const int a4 = (src[0][i] + src[4][i]) << 3; yading@10: yading@10: const int b0 = a4 + a6; yading@10: const int b1 = a5 + a7; yading@10: const int b2 = a5 - a7; yading@10: const int b3 = a4 - a6; yading@10: yading@10: dst[i + 0*stride] = cm[ dst[i + 0*stride] + ((b0 + b4) >> 7)]; yading@10: dst[i + 1*stride] = cm[ dst[i + 1*stride] + ((b1 + b5) >> 7)]; yading@10: dst[i + 2*stride] = cm[ dst[i + 2*stride] + ((b2 + b6) >> 7)]; yading@10: dst[i + 3*stride] = cm[ dst[i + 3*stride] + ((b3 + b7) >> 7)]; yading@10: dst[i + 4*stride] = cm[ dst[i + 4*stride] + ((b3 - b7) >> 7)]; yading@10: dst[i + 5*stride] = cm[ dst[i + 5*stride] + ((b2 - b6) >> 7)]; yading@10: dst[i + 6*stride] = cm[ dst[i + 6*stride] + ((b1 - b5) >> 7)]; yading@10: dst[i + 7*stride] = cm[ dst[i + 7*stride] + ((b0 - b4) >> 7)]; yading@10: } yading@10: } yading@10: yading@10: /***************************************************************************** yading@10: * yading@10: * motion compensation yading@10: * yading@10: ****************************************************************************/ yading@10: yading@10: #define CAVS_SUBPIX(OPNAME, OP, NAME, A, B, C, D, E, F) \ yading@10: static void OPNAME ## cavs_filt8_h_ ## NAME(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\ yading@10: const int h=8;\ yading@10: const uint8_t *cm = ff_cropTbl + MAX_NEG_CROP;\ yading@10: int i;\ yading@10: for(i=0; i>3] yading@10: #define op_put2(a, b) a = cm[((b)+64)>>7] yading@10: #define op_put3(a, b) a = cm[((b)+32)>>6] yading@10: #define op_put4(a, b) a = cm[((b)+512)>>10] yading@10: #define op_avg1(a, b) a = ((a)+cm[((b)+4)>>3] +1)>>1 yading@10: #define op_avg2(a, b) a = ((a)+cm[((b)+64)>>7] +1)>>1 yading@10: #define op_avg3(a, b) a = ((a)+cm[((b)+32)>>6] +1)>>1 yading@10: #define op_avg4(a, b) a = ((a)+cm[((b)+512)>>10]+1)>>1 yading@10: CAVS_SUBPIX(put_ , op_put1, hpel, 0, -1, 5, 5, -1, 0) yading@10: CAVS_SUBPIX(put_ , op_put2, qpel_l, -1, -2, 96, 42, -7, 0) yading@10: CAVS_SUBPIX(put_ , op_put2, qpel_r, 0, -7, 42, 96, -2, -1) yading@10: CAVS_SUBPIX_HV(put_, op_put3, jj, 0, -1, 5, 5, -1, 0, 0, -1, 5, 5, -1, 0, 0) yading@10: CAVS_SUBPIX_HV(put_, op_put4, ff, 0, -1, 5, 5, -1, 0, -1, -2, 96, 42, -7, 0, 0) yading@10: CAVS_SUBPIX_HV(put_, op_put4, ii, -1, -2, 96, 42, -7, 0, 0, -1, 5, 5, -1, 0, 0) yading@10: CAVS_SUBPIX_HV(put_, op_put4, kk, 0, -7, 42, 96, -2, -1, 0, -1, 5, 5, -1, 0, 0) yading@10: CAVS_SUBPIX_HV(put_, op_put4, qq, 0, -1, 5, 5, -1, 0, 0, -7, 42, 96, -2,-1, 0) yading@10: CAVS_SUBPIX_HV(put_, op_put2, egpr, 0, -1, 5, 5, -1, 0, 0, -1, 5, 5, -1, 0, 1) yading@10: CAVS_SUBPIX(avg_ , op_avg1, hpel, 0, -1, 5, 5, -1, 0) yading@10: CAVS_SUBPIX(avg_ , op_avg2, qpel_l, -1, -2, 96, 42, -7, 0) yading@10: CAVS_SUBPIX(avg_ , op_avg2, qpel_r, 0, -7, 42, 96, -2, -1) yading@10: CAVS_SUBPIX_HV(avg_, op_avg3, jj, 0, -1, 5, 5, -1, 0, 0, -1, 5, 5, -1, 0, 0) yading@10: CAVS_SUBPIX_HV(avg_, op_avg4, ff, 0, -1, 5, 5, -1, 0, -1, -2, 96, 42, -7, 0, 0) yading@10: CAVS_SUBPIX_HV(avg_, op_avg4, ii, -1, -2, 96, 42, -7, 0, 0, -1, 5, 5, -1, 0, 0) yading@10: CAVS_SUBPIX_HV(avg_, op_avg4, kk, 0, -7, 42, 96, -2, -1, 0, -1, 5, 5, -1, 0, 0) yading@10: CAVS_SUBPIX_HV(avg_, op_avg4, qq, 0, -1, 5, 5, -1, 0, 0, -7, 42, 96, -2,-1, 0) yading@10: CAVS_SUBPIX_HV(avg_, op_avg2, egpr, 0, -1, 5, 5, -1, 0, 0, -1, 5, 5, -1, 0, 1) yading@10: CAVS_MC(put_, 8) yading@10: CAVS_MC(put_, 16) yading@10: CAVS_MC(avg_, 8) yading@10: CAVS_MC(avg_, 16) yading@10: yading@10: #define ff_put_cavs_qpel8_mc00_c ff_put_pixels8x8_c yading@10: #define ff_avg_cavs_qpel8_mc00_c ff_avg_pixels8x8_c yading@10: #define ff_put_cavs_qpel16_mc00_c ff_put_pixels16x16_c yading@10: #define ff_avg_cavs_qpel16_mc00_c ff_avg_pixels16x16_c yading@10: yading@10: av_cold void ff_cavsdsp_init(CAVSDSPContext* c, AVCodecContext *avctx) { yading@10: #define dspfunc(PFX, IDX, NUM) \ yading@10: c->PFX ## _pixels_tab[IDX][ 0] = ff_ ## PFX ## NUM ## _mc00_c; \ yading@10: c->PFX ## _pixels_tab[IDX][ 1] = ff_ ## PFX ## NUM ## _mc10_c; \ yading@10: c->PFX ## _pixels_tab[IDX][ 2] = ff_ ## PFX ## NUM ## _mc20_c; \ yading@10: c->PFX ## _pixels_tab[IDX][ 3] = ff_ ## PFX ## NUM ## _mc30_c; \ yading@10: c->PFX ## _pixels_tab[IDX][ 4] = ff_ ## PFX ## NUM ## _mc01_c; \ yading@10: c->PFX ## _pixels_tab[IDX][ 5] = ff_ ## PFX ## NUM ## _mc11_c; \ yading@10: c->PFX ## _pixels_tab[IDX][ 6] = ff_ ## PFX ## NUM ## _mc21_c; \ yading@10: c->PFX ## _pixels_tab[IDX][ 7] = ff_ ## PFX ## NUM ## _mc31_c; \ yading@10: c->PFX ## _pixels_tab[IDX][ 8] = ff_ ## PFX ## NUM ## _mc02_c; \ yading@10: c->PFX ## _pixels_tab[IDX][ 9] = ff_ ## PFX ## NUM ## _mc12_c; \ yading@10: c->PFX ## _pixels_tab[IDX][10] = ff_ ## PFX ## NUM ## _mc22_c; \ yading@10: c->PFX ## _pixels_tab[IDX][11] = ff_ ## PFX ## NUM ## _mc32_c; \ yading@10: c->PFX ## _pixels_tab[IDX][12] = ff_ ## PFX ## NUM ## _mc03_c; \ yading@10: c->PFX ## _pixels_tab[IDX][13] = ff_ ## PFX ## NUM ## _mc13_c; \ yading@10: c->PFX ## _pixels_tab[IDX][14] = ff_ ## PFX ## NUM ## _mc23_c; \ yading@10: c->PFX ## _pixels_tab[IDX][15] = ff_ ## PFX ## NUM ## _mc33_c yading@10: dspfunc(put_cavs_qpel, 0, 16); yading@10: dspfunc(put_cavs_qpel, 1, 8); yading@10: dspfunc(avg_cavs_qpel, 0, 16); yading@10: dspfunc(avg_cavs_qpel, 1, 8); yading@10: c->cavs_filter_lv = cavs_filter_lv_c; yading@10: c->cavs_filter_lh = cavs_filter_lh_c; yading@10: c->cavs_filter_cv = cavs_filter_cv_c; yading@10: c->cavs_filter_ch = cavs_filter_ch_c; yading@10: c->cavs_idct8_add = cavs_idct8_add_c; yading@10: c->idct_perm = FF_NO_IDCT_PERM; yading@10: yading@10: if (ARCH_X86) yading@10: ff_cavsdsp_init_x86(c, avctx); yading@10: }