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