LCOV - code coverage report
Current view: top level - src/libpostproc - postprocess.c (source / functions) Hit Total Coverage
Test: coverage.info Lines: 391 467 83.7 %
Date: 2017-04-24 03:48:25 Functions: 20 22 90.9 %

          Line data    Source code
       1             : /*
       2             :  * Copyright (C) 2001-2003 Michael Niedermayer (michaelni@gmx.at)
       3             :  *
       4             :  * AltiVec optimizations (C) 2004 Romain Dolbeau <romain@dolbeau.org>
       5             :  *
       6             :  * This file is part of FFmpeg.
       7             :  *
       8             :  * FFmpeg is free software; you can redistribute it and/or modify
       9             :  * it under the terms of the GNU General Public License as published by
      10             :  * the Free Software Foundation; either version 2 of the License, or
      11             :  * (at your option) any later version.
      12             :  *
      13             :  * FFmpeg is distributed in the hope that it will be useful,
      14             :  * but WITHOUT ANY WARRANTY; without even the implied warranty of
      15             :  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
      16             :  * GNU General Public License for more details.
      17             :  *
      18             :  * You should have received a copy of the GNU General Public License
      19             :  * along with FFmpeg; if not, write to the Free Software
      20             :  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
      21             :  */
      22             : 
      23             : /**
      24             :  * @file
      25             :  * postprocessing.
      26             :  */
      27             : 
      28             : /*
      29             :                         C       MMX     MMX2    3DNow   AltiVec
      30             : isVertDC                Ec      Ec                      Ec
      31             : isVertMinMaxOk          Ec      Ec                      Ec
      32             : doVertLowPass           E               e       e       Ec
      33             : doVertDefFilter         Ec      Ec      e       e       Ec
      34             : isHorizDC               Ec      Ec                      Ec
      35             : isHorizMinMaxOk         a       E                       Ec
      36             : doHorizLowPass          E               e       e       Ec
      37             : doHorizDefFilter        Ec      Ec      e       e       Ec
      38             : do_a_deblock            Ec      E       Ec      E
      39             : deRing                  E               e       e*      Ecp
      40             : Vertical RKAlgo1        E               a       a
      41             : Horizontal RKAlgo1                      a       a
      42             : Vertical X1#            a               E       E
      43             : Horizontal X1#          a               E       E
      44             : LinIpolDeinterlace      e               E       E*
      45             : CubicIpolDeinterlace    a               e       e*
      46             : LinBlendDeinterlace     e               E       E*
      47             : MedianDeinterlace#      E       Ec      Ec
      48             : TempDeNoiser#           E               e       e       Ec
      49             : 
      50             : * I do not have a 3DNow! CPU -> it is untested, but no one said it does not work so it seems to work
      51             : # more or less selfinvented filters so the exactness is not too meaningful
      52             : E = Exact implementation
      53             : e = almost exact implementation (slightly different rounding,...)
      54             : a = alternative / approximate impl
      55             : c = checked against the other implementations (-vo md5)
      56             : p = partially optimized, still some work to do
      57             : */
      58             : 
      59             : /*
      60             : TODO:
      61             : reduce the time wasted on the mem transfer
      62             : unroll stuff if instructions depend too much on the prior one
      63             : move YScale thing to the end instead of fixing QP
      64             : write a faster and higher quality deblocking filter :)
      65             : make the mainloop more flexible (variable number of blocks at once
      66             :         (the if/else stuff per block is slowing things down)
      67             : compare the quality & speed of all filters
      68             : split this huge file
      69             : optimize c versions
      70             : try to unroll inner for(x=0 ... loop to avoid these damn if(x ... checks
      71             : ...
      72             : */
      73             : 
      74             : //Changelog: use git log
      75             : 
      76             : #include "config.h"
      77             : #include "libavutil/avutil.h"
      78             : #include "libavutil/avassert.h"
      79             : #include "libavutil/intreadwrite.h"
      80             : #include <inttypes.h>
      81             : #include <stdio.h>
      82             : #include <stdlib.h>
      83             : #include <string.h>
      84             : //#undef HAVE_MMXEXT_INLINE
      85             : //#define HAVE_AMD3DNOW_INLINE
      86             : //#undef HAVE_MMX_INLINE
      87             : //#undef ARCH_X86
      88             : //#define DEBUG_BRIGHTNESS
      89             : #include "postprocess.h"
      90             : #include "postprocess_internal.h"
      91             : #include "libavutil/avstring.h"
      92             : 
      93             : #include "libavutil/ffversion.h"
      94             : const char postproc_ffversion[] = "FFmpeg version " FFMPEG_VERSION;
      95             : 
      96        5207 : unsigned postproc_version(void)
      97             : {
      98             :     av_assert0(LIBPOSTPROC_VERSION_MICRO >= 100);
      99        5207 :     return LIBPOSTPROC_VERSION_INT;
     100             : }
     101             : 
     102        5207 : const char *postproc_configuration(void)
     103             : {
     104        5207 :     return FFMPEG_CONFIGURATION;
     105             : }
     106             : 
     107           0 : const char *postproc_license(void)
     108             : {
     109             : #define LICENSE_PREFIX "libpostproc license: "
     110           0 :     return LICENSE_PREFIX FFMPEG_LICENSE + sizeof(LICENSE_PREFIX) - 1;
     111             : }
     112             : 
     113             : #if HAVE_ALTIVEC_H
     114             : #include <altivec.h>
     115             : #endif
     116             : 
     117             : #define GET_MODE_BUFFER_SIZE 500
     118             : #define OPTIONS_ARRAY_SIZE 10
     119             : #define BLOCK_SIZE 8
     120             : #define TEMP_STRIDE 8
     121             : //#define NUM_BLOCKS_AT_ONCE 16 //not used yet
     122             : 
     123             : #if ARCH_X86 && HAVE_INLINE_ASM
     124             : DECLARE_ASM_CONST(8, uint64_t, w05)= 0x0005000500050005LL;
     125             : DECLARE_ASM_CONST(8, uint64_t, w04)= 0x0004000400040004LL;
     126             : DECLARE_ASM_CONST(8, uint64_t, w20)= 0x0020002000200020LL;
     127             : DECLARE_ASM_CONST(8, uint64_t, b00)= 0x0000000000000000LL;
     128             : DECLARE_ASM_CONST(8, uint64_t, b01)= 0x0101010101010101LL;
     129             : DECLARE_ASM_CONST(8, uint64_t, b02)= 0x0202020202020202LL;
     130             : DECLARE_ASM_CONST(8, uint64_t, b08)= 0x0808080808080808LL;
     131             : DECLARE_ASM_CONST(8, uint64_t, b80)= 0x8080808080808080LL;
     132             : #endif
     133             : 
     134             : DECLARE_ASM_CONST(8, int, deringThreshold)= 20;
     135             : 
     136             : 
     137             : static const struct PPFilter filters[]=
     138             : {
     139             :     {"hb", "hdeblock",              1, 1, 3, H_DEBLOCK},
     140             :     {"vb", "vdeblock",              1, 2, 4, V_DEBLOCK},
     141             : /*  {"hr", "rkhdeblock",            1, 1, 3, H_RK1_FILTER},
     142             :     {"vr", "rkvdeblock",            1, 2, 4, V_RK1_FILTER},*/
     143             :     {"h1", "x1hdeblock",            1, 1, 3, H_X1_FILTER},
     144             :     {"v1", "x1vdeblock",            1, 2, 4, V_X1_FILTER},
     145             :     {"ha", "ahdeblock",             1, 1, 3, H_A_DEBLOCK},
     146             :     {"va", "avdeblock",             1, 2, 4, V_A_DEBLOCK},
     147             :     {"dr", "dering",                1, 5, 6, DERING},
     148             :     {"al", "autolevels",            0, 1, 2, LEVEL_FIX},
     149             :     {"lb", "linblenddeint",         1, 1, 4, LINEAR_BLEND_DEINT_FILTER},
     150             :     {"li", "linipoldeint",          1, 1, 4, LINEAR_IPOL_DEINT_FILTER},
     151             :     {"ci", "cubicipoldeint",        1, 1, 4, CUBIC_IPOL_DEINT_FILTER},
     152             :     {"md", "mediandeint",           1, 1, 4, MEDIAN_DEINT_FILTER},
     153             :     {"fd", "ffmpegdeint",           1, 1, 4, FFMPEG_DEINT_FILTER},
     154             :     {"l5", "lowpass5",              1, 1, 4, LOWPASS5_DEINT_FILTER},
     155             :     {"tn", "tmpnoise",              1, 7, 8, TEMP_NOISE_FILTER},
     156             :     {"fq", "forcequant",            1, 0, 0, FORCE_QUANT},
     157             :     {"be", "bitexact",              1, 0, 0, BITEXACT},
     158             :     {"vi", "visualize",             1, 0, 0, VISUALIZE},
     159             :     {NULL, NULL,0,0,0,0} //End Marker
     160             : };
     161             : 
     162             : static const char * const replaceTable[]=
     163             : {
     164             :     "default",      "hb:a,vb:a,dr:a",
     165             :     "de",           "hb:a,vb:a,dr:a",
     166             :     "fast",         "h1:a,v1:a,dr:a",
     167             :     "fa",           "h1:a,v1:a,dr:a",
     168             :     "ac",           "ha:a:128:7,va:a,dr:a",
     169             :     NULL //End Marker
     170             : };
     171             : 
     172             : /* The horizontal functions exist only in C because the MMX
     173             :  * code is faster with vertical filters and transposing. */
     174             : 
     175             : /**
     176             :  * Check if the given 8x8 Block is mostly "flat"
     177             :  */
     178       36864 : static inline int isHorizDC_C(const uint8_t src[], int stride, const PPContext *c)
     179             : {
     180       36864 :     int numEq= 0;
     181             :     int y;
     182       36864 :     const int dcOffset= ((c->nonBQP*c->ppMode.baseDcDiff)>>8) + 1;
     183       36864 :     const int dcThreshold= dcOffset*2 + 1;
     184             : 
     185      331776 :     for(y=0; y<BLOCK_SIZE; y++){
     186      294912 :         numEq += ((unsigned)(src[0] - src[1] + dcOffset)) < dcThreshold;
     187      294912 :         numEq += ((unsigned)(src[1] - src[2] + dcOffset)) < dcThreshold;
     188      294912 :         numEq += ((unsigned)(src[2] - src[3] + dcOffset)) < dcThreshold;
     189      294912 :         numEq += ((unsigned)(src[3] - src[4] + dcOffset)) < dcThreshold;
     190      294912 :         numEq += ((unsigned)(src[4] - src[5] + dcOffset)) < dcThreshold;
     191      294912 :         numEq += ((unsigned)(src[5] - src[6] + dcOffset)) < dcThreshold;
     192      294912 :         numEq += ((unsigned)(src[6] - src[7] + dcOffset)) < dcThreshold;
     193      294912 :         src+= stride;
     194             :     }
     195       36864 :     return numEq > c->ppMode.flatnessThreshold;
     196             : }
     197             : 
     198             : /**
     199             :  * Check if the middle 8x8 Block in the given 8x16 block is flat
     200             :  */
     201       36608 : static inline int isVertDC_C(const uint8_t src[], int stride, const PPContext *c)
     202             : {
     203       36608 :     int numEq= 0;
     204             :     int y;
     205       36608 :     const int dcOffset= ((c->nonBQP*c->ppMode.baseDcDiff)>>8) + 1;
     206       36608 :     const int dcThreshold= dcOffset*2 + 1;
     207             : 
     208       36608 :     src+= stride*4; // src points to begin of the 8x8 Block
     209      292864 :     for(y=0; y<BLOCK_SIZE-1; y++){
     210      256256 :         numEq += ((unsigned)(src[0] - src[0+stride] + dcOffset)) < dcThreshold;
     211      256256 :         numEq += ((unsigned)(src[1] - src[1+stride] + dcOffset)) < dcThreshold;
     212      256256 :         numEq += ((unsigned)(src[2] - src[2+stride] + dcOffset)) < dcThreshold;
     213      256256 :         numEq += ((unsigned)(src[3] - src[3+stride] + dcOffset)) < dcThreshold;
     214      256256 :         numEq += ((unsigned)(src[4] - src[4+stride] + dcOffset)) < dcThreshold;
     215      256256 :         numEq += ((unsigned)(src[5] - src[5+stride] + dcOffset)) < dcThreshold;
     216      256256 :         numEq += ((unsigned)(src[6] - src[6+stride] + dcOffset)) < dcThreshold;
     217      256256 :         numEq += ((unsigned)(src[7] - src[7+stride] + dcOffset)) < dcThreshold;
     218      256256 :         src+= stride;
     219             :     }
     220       36608 :     return numEq > c->ppMode.flatnessThreshold;
     221             : }
     222             : 
     223         701 : static inline int isHorizMinMaxOk_C(const uint8_t src[], int stride, int QP)
     224             : {
     225             :     int i;
     226        1149 :     for(i=0; i<2; i++){
     227        1020 :         if((unsigned)(src[0] - src[5] + 2*QP) > 4*QP) return 0;
     228         652 :         src += stride;
     229         652 :         if((unsigned)(src[2] - src[7] + 2*QP) > 4*QP) return 0;
     230         510 :         src += stride;
     231         510 :         if((unsigned)(src[4] - src[1] + 2*QP) > 4*QP) return 0;
     232         508 :         src += stride;
     233         508 :         if((unsigned)(src[6] - src[3] + 2*QP) > 4*QP) return 0;
     234         448 :         src += stride;
     235             :     }
     236         129 :     return 1;
     237             : }
     238             : 
     239         352 : static inline int isVertMinMaxOk_C(const uint8_t src[], int stride, int QP)
     240             : {
     241             :     int x;
     242         352 :     src+= stride*4;
     243         611 :     for(x=0; x<BLOCK_SIZE; x+=4){
     244         526 :         if((unsigned)(src[  x + 0*stride] - src[  x + 5*stride] + 2*QP) > 4*QP) return 0;
     245         349 :         if((unsigned)(src[1+x + 2*stride] - src[1+x + 7*stride] + 2*QP) > 4*QP) return 0;
     246         291 :         if((unsigned)(src[2+x + 4*stride] - src[2+x + 1*stride] + 2*QP) > 4*QP) return 0;
     247         291 :         if((unsigned)(src[3+x + 6*stride] - src[3+x + 3*stride] + 2*QP) > 4*QP) return 0;
     248             :     }
     249          85 :     return 1;
     250             : }
     251             : 
     252       36864 : static inline int horizClassify_C(const uint8_t src[], int stride, const PPContext *c)
     253             : {
     254       36864 :     if( isHorizDC_C(src, stride, c) ){
     255         701 :         return isHorizMinMaxOk_C(src, stride, c->QP);
     256             :     }else{
     257       36163 :         return 2;
     258             :     }
     259             : }
     260             : 
     261       36608 : static inline int vertClassify_C(const uint8_t src[], int stride, const PPContext *c)
     262             : {
     263       36608 :     if( isVertDC_C(src, stride, c) ){
     264         352 :         return isVertMinMaxOk_C(src, stride, c->QP);
     265             :     }else{
     266       36256 :         return 2;
     267             :     }
     268             : }
     269             : 
     270       36163 : static inline void doHorizDefFilter_C(uint8_t dst[], int stride, const PPContext *c)
     271             : {
     272             :     int y;
     273      325467 :     for(y=0; y<BLOCK_SIZE; y++){
     274      289304 :         const int middleEnergy= 5*(dst[4] - dst[3]) + 2*(dst[2] - dst[5]);
     275             : 
     276      289304 :         if(FFABS(middleEnergy) < 8*c->QP){
     277      188700 :             const int q=(dst[3] - dst[4])/2;
     278      188700 :             const int leftEnergy=  5*(dst[2] - dst[1]) + 2*(dst[0] - dst[3]);
     279      188700 :             const int rightEnergy= 5*(dst[6] - dst[5]) + 2*(dst[4] - dst[7]);
     280             : 
     281      188700 :             int d= FFABS(middleEnergy) - FFMIN( FFABS(leftEnergy), FFABS(rightEnergy) );
     282      188700 :             d= FFMAX(d, 0);
     283             : 
     284      188700 :             d= (5*d + 32) >> 6;
     285      188700 :             d*= FFSIGN(-middleEnergy);
     286             : 
     287      188700 :             if(q>0)
     288             :             {
     289       43347 :                 d = FFMAX(d, 0);
     290       43347 :                 d = FFMIN(d, q);
     291             :             }
     292             :             else
     293             :             {
     294      145353 :                 d = FFMIN(d, 0);
     295      145353 :                 d = FFMAX(d, q);
     296             :             }
     297             : 
     298      188700 :             dst[3]-= d;
     299      188700 :             dst[4]+= d;
     300             :         }
     301      289304 :         dst+= stride;
     302             :     }
     303       36163 : }
     304             : 
     305             : /**
     306             :  * Do a horizontal low pass filter on the 10x8 block (dst points to middle 8x8 Block)
     307             :  * using the 9-Tap Filter (1,1,2,2,4,2,2,1,1)/16 (C version)
     308             :  */
     309         129 : static inline void doHorizLowPass_C(uint8_t dst[], int stride, const PPContext *c)
     310             : {
     311             :     int y;
     312        1161 :     for(y=0; y<BLOCK_SIZE; y++){
     313        1032 :         const int first= FFABS(dst[-1] - dst[0]) < c->QP ? dst[-1] : dst[0];
     314        1032 :         const int last= FFABS(dst[8] - dst[7]) < c->QP ? dst[8] : dst[7];
     315             : 
     316             :         int sums[10];
     317        1032 :         sums[0] = 4*first + dst[0] + dst[1] + dst[2] + 4;
     318        1032 :         sums[1] = sums[0] - first  + dst[3];
     319        1032 :         sums[2] = sums[1] - first  + dst[4];
     320        1032 :         sums[3] = sums[2] - first  + dst[5];
     321        1032 :         sums[4] = sums[3] - first  + dst[6];
     322        1032 :         sums[5] = sums[4] - dst[0] + dst[7];
     323        1032 :         sums[6] = sums[5] - dst[1] + last;
     324        1032 :         sums[7] = sums[6] - dst[2] + last;
     325        1032 :         sums[8] = sums[7] - dst[3] + last;
     326        1032 :         sums[9] = sums[8] - dst[4] + last;
     327             : 
     328        1032 :         dst[0]= (sums[0] + sums[2] + 2*dst[0])>>4;
     329        1032 :         dst[1]= (sums[1] + sums[3] + 2*dst[1])>>4;
     330        1032 :         dst[2]= (sums[2] + sums[4] + 2*dst[2])>>4;
     331        1032 :         dst[3]= (sums[3] + sums[5] + 2*dst[3])>>4;
     332        1032 :         dst[4]= (sums[4] + sums[6] + 2*dst[4])>>4;
     333        1032 :         dst[5]= (sums[5] + sums[7] + 2*dst[5])>>4;
     334        1032 :         dst[6]= (sums[6] + sums[8] + 2*dst[6])>>4;
     335        1032 :         dst[7]= (sums[7] + sums[9] + 2*dst[7])>>4;
     336             : 
     337        1032 :         dst+= stride;
     338             :     }
     339         129 : }
     340             : 
     341             : /**
     342             :  * Experimental Filter 1 (Horizontal)
     343             :  * will not damage linear gradients
     344             :  * Flat blocks should look like they were passed through the (1,1,2,2,4,2,2,1,1) 9-Tap filter
     345             :  * can only smooth blocks at the expected locations (it cannot smooth them if they did move)
     346             :  * MMX2 version does correct clipping C version does not
     347             :  * not identical with the vertical one
     348             :  */
     349       11520 : static inline void horizX1Filter(uint8_t *src, int stride, int QP)
     350             : {
     351             :     int y;
     352             :     static uint64_t lut[256];
     353       11520 :     if(!lut[255])
     354             :     {
     355             :         int i;
     356     2960640 :         for(i=0; i<256; i++)
     357             :         {
     358     2949120 :             int v= i < 128 ? 2*i : 2*(i-256);
     359             : /*
     360             : //Simulate 112242211 9-Tap filter
     361             :             uint64_t a= (v/16)  & 0xFF;
     362             :             uint64_t b= (v/8)   & 0xFF;
     363             :             uint64_t c= (v/4)   & 0xFF;
     364             :             uint64_t d= (3*v/8) & 0xFF;
     365             : */
     366             : //Simulate piecewise linear interpolation
     367     2949120 :             uint64_t a= (v/16)   & 0xFF;
     368     2949120 :             uint64_t b= (v*3/16) & 0xFF;
     369     2949120 :             uint64_t c= (v*5/16) & 0xFF;
     370     2949120 :             uint64_t d= (7*v/16) & 0xFF;
     371     2949120 :             uint64_t A= (0x100 - a)&0xFF;
     372     2949120 :             uint64_t B= (0x100 - b)&0xFF;
     373     2949120 :             uint64_t C= (0x100 - c)&0xFF;
     374     2949120 :             uint64_t D= (0x100 - c)&0xFF;
     375             : 
     376     8847360 :             lut[i]   = (a<<56) | (b<<48) | (c<<40) | (d<<32) |
     377     5898240 :                        (D<<24) | (C<<16) | (B<<8)  | (A);
     378             :             //lut[i] = (v<<32) | (v<<24);
     379             :         }
     380             :     }
     381             : 
     382      103680 :     for(y=0; y<BLOCK_SIZE; y++){
     383       92160 :         int a= src[1] - src[2];
     384       92160 :         int b= src[3] - src[4];
     385       92160 :         int c= src[5] - src[6];
     386             : 
     387       92160 :         int d= FFMAX(FFABS(b) - (FFABS(a) + FFABS(c))/2, 0);
     388             : 
     389       92160 :         if(d < QP){
     390       84620 :             int v = d * FFSIGN(-b);
     391             : 
     392       84620 :             src[1] +=v/8;
     393       84620 :             src[2] +=v/4;
     394       84620 :             src[3] +=3*v/8;
     395       84620 :             src[4] -=3*v/8;
     396       84620 :             src[5] -=v/4;
     397       84620 :             src[6] -=v/8;
     398             :         }
     399       92160 :         src+=stride;
     400             :     }
     401       11520 : }
     402             : 
     403             : /**
     404             :  * accurate deblock filter
     405             :  */
     406       22960 : static av_always_inline void do_a_deblock_C(uint8_t *src, int step,
     407             :                                             int stride, const PPContext *c, int mode)
     408             : {
     409             :     int y;
     410       22960 :     const int QP= c->QP;
     411       22960 :     const int dcOffset= ((c->nonBQP*c->ppMode.baseDcDiff)>>8) + 1;
     412       22960 :     const int dcThreshold= dcOffset*2 + 1;
     413             : //START_TIMER
     414       22960 :     src+= step*4; // src points to begin of the 8x8 Block
     415      206640 :     for(y=0; y<8; y++){
     416      183680 :         int numEq= 0;
     417             : 
     418      183680 :         numEq += ((unsigned)(src[-1*step] - src[0*step] + dcOffset)) < dcThreshold;
     419      183680 :         numEq += ((unsigned)(src[ 0*step] - src[1*step] + dcOffset)) < dcThreshold;
     420      183680 :         numEq += ((unsigned)(src[ 1*step] - src[2*step] + dcOffset)) < dcThreshold;
     421      183680 :         numEq += ((unsigned)(src[ 2*step] - src[3*step] + dcOffset)) < dcThreshold;
     422      183680 :         numEq += ((unsigned)(src[ 3*step] - src[4*step] + dcOffset)) < dcThreshold;
     423      183680 :         numEq += ((unsigned)(src[ 4*step] - src[5*step] + dcOffset)) < dcThreshold;
     424      183680 :         numEq += ((unsigned)(src[ 5*step] - src[6*step] + dcOffset)) < dcThreshold;
     425      183680 :         numEq += ((unsigned)(src[ 6*step] - src[7*step] + dcOffset)) < dcThreshold;
     426      183680 :         numEq += ((unsigned)(src[ 7*step] - src[8*step] + dcOffset)) < dcThreshold;
     427      183680 :         if(numEq > c->ppMode.flatnessThreshold){
     428             :             int min, max, x;
     429             : 
     430       82036 :             if(src[0] > src[step]){
     431       17783 :                 max= src[0];
     432       17783 :                 min= src[step];
     433             :             }else{
     434       64253 :                 max= src[step];
     435       64253 :                 min= src[0];
     436             :             }
     437      328144 :             for(x=2; x<8; x+=2){
     438      246108 :                 if(src[x*step] > src[(x+1)*step]){
     439       53874 :                         if(src[x    *step] > max) max= src[ x   *step];
     440       53874 :                         if(src[(x+1)*step] < min) min= src[(x+1)*step];
     441             :                 }else{
     442      192234 :                         if(src[(x+1)*step] > max) max= src[(x+1)*step];
     443      192234 :                         if(src[ x   *step] < min) min= src[ x   *step];
     444             :                 }
     445             :             }
     446       82036 :             if(max-min < 2*QP){
     447       42585 :                 const int first= FFABS(src[-1*step] - src[0]) < QP ? src[-1*step] : src[0];
     448       42585 :                 const int last= FFABS(src[8*step] - src[7*step]) < QP ? src[8*step] : src[7*step];
     449             : 
     450             :                 int sums[10];
     451       42585 :                 sums[0] = 4*first + src[0*step] + src[1*step] + src[2*step] + 4;
     452       42585 :                 sums[1] = sums[0] - first       + src[3*step];
     453       42585 :                 sums[2] = sums[1] - first       + src[4*step];
     454       42585 :                 sums[3] = sums[2] - first       + src[5*step];
     455       42585 :                 sums[4] = sums[3] - first       + src[6*step];
     456       42585 :                 sums[5] = sums[4] - src[0*step] + src[7*step];
     457       42585 :                 sums[6] = sums[5] - src[1*step] + last;
     458       42585 :                 sums[7] = sums[6] - src[2*step] + last;
     459       42585 :                 sums[8] = sums[7] - src[3*step] + last;
     460       42585 :                 sums[9] = sums[8] - src[4*step] + last;
     461             : 
     462       42585 :                 if (mode & VISUALIZE) {
     463           0 :                     src[0*step] =
     464           0 :                     src[1*step] =
     465           0 :                     src[2*step] =
     466           0 :                     src[3*step] =
     467           0 :                     src[4*step] =
     468           0 :                     src[5*step] =
     469           0 :                     src[6*step] =
     470           0 :                     src[7*step] = 128;
     471             :                 }
     472       42585 :                 src[0*step]= (sums[0] + sums[2] + 2*src[0*step])>>4;
     473       42585 :                 src[1*step]= (sums[1] + sums[3] + 2*src[1*step])>>4;
     474       42585 :                 src[2*step]= (sums[2] + sums[4] + 2*src[2*step])>>4;
     475       42585 :                 src[3*step]= (sums[3] + sums[5] + 2*src[3*step])>>4;
     476       42585 :                 src[4*step]= (sums[4] + sums[6] + 2*src[4*step])>>4;
     477       42585 :                 src[5*step]= (sums[5] + sums[7] + 2*src[5*step])>>4;
     478       42585 :                 src[6*step]= (sums[6] + sums[8] + 2*src[6*step])>>4;
     479       42585 :                 src[7*step]= (sums[7] + sums[9] + 2*src[7*step])>>4;
     480             :             }
     481             :         }else{
     482      101644 :             const int middleEnergy= 5*(src[4*step] - src[3*step]) + 2*(src[2*step] - src[5*step]);
     483             : 
     484      101644 :             if(FFABS(middleEnergy) < 8*QP){
     485       86530 :                 const int q=(src[3*step] - src[4*step])/2;
     486       86530 :                 const int leftEnergy=  5*(src[2*step] - src[1*step]) + 2*(src[0*step] - src[3*step]);
     487       86530 :                 const int rightEnergy= 5*(src[6*step] - src[5*step]) + 2*(src[4*step] - src[7*step]);
     488             : 
     489       86530 :                 int d= FFABS(middleEnergy) - FFMIN( FFABS(leftEnergy), FFABS(rightEnergy) );
     490       86530 :                 d= FFMAX(d, 0);
     491             : 
     492       86530 :                 d= (5*d + 32) >> 6;
     493       86530 :                 d*= FFSIGN(-middleEnergy);
     494             : 
     495       86530 :                 if(q>0){
     496       27119 :                     d = FFMAX(d, 0);
     497       27119 :                     d = FFMIN(d, q);
     498             :                 }else{
     499       59411 :                     d = FFMIN(d, 0);
     500       59411 :                     d = FFMAX(d, q);
     501             :                 }
     502             : 
     503       86530 :                 if ((mode & VISUALIZE) && d) {
     504           0 :                     d= (d < 0) ? 32 : -32;
     505           0 :                     src[3*step]= av_clip_uint8(src[3*step] - d);
     506           0 :                     src[4*step]= av_clip_uint8(src[4*step] + d);
     507           0 :                     d = 0;
     508             :                 }
     509             : 
     510       86530 :                 src[3*step]-= d;
     511       86530 :                 src[4*step]+= d;
     512             :             }
     513             :         }
     514             : 
     515      183680 :         src += stride;
     516             :     }
     517             : /*if(step==16){
     518             :     STOP_TIMER("step16")
     519             : }else{
     520             :     STOP_TIMER("stepX")
     521             : }*/
     522       22960 : }
     523             : 
     524             : //Note: we have C, MMX, MMX2, 3DNOW version there is no 3DNOW+MMX2 one
     525             : //Plain C versions
     526             : //we always compile C for testing which needs bitexactness
     527             : #define TEMPLATE_PP_C 1
     528             : #include "postprocess_template.c"
     529             : 
     530             : #if HAVE_ALTIVEC
     531             : #   define TEMPLATE_PP_ALTIVEC 1
     532             : #   include "postprocess_altivec_template.c"
     533             : #   include "postprocess_template.c"
     534             : #endif
     535             : 
     536             : #if ARCH_X86 && HAVE_INLINE_ASM
     537             : #    if CONFIG_RUNTIME_CPUDETECT
     538             : #        define TEMPLATE_PP_MMX 1
     539             : #        include "postprocess_template.c"
     540             : #        define TEMPLATE_PP_MMXEXT 1
     541             : #        include "postprocess_template.c"
     542             : #        define TEMPLATE_PP_3DNOW 1
     543             : #        include "postprocess_template.c"
     544             : #        define TEMPLATE_PP_SSE2 1
     545             : #        include "postprocess_template.c"
     546             : #    else
     547             : #        if HAVE_SSE2_INLINE
     548             : #            define TEMPLATE_PP_SSE2 1
     549             : #            include "postprocess_template.c"
     550             : #        elif HAVE_MMXEXT_INLINE
     551             : #            define TEMPLATE_PP_MMXEXT 1
     552             : #            include "postprocess_template.c"
     553             : #        elif HAVE_AMD3DNOW_INLINE
     554             : #            define TEMPLATE_PP_3DNOW 1
     555             : #            include "postprocess_template.c"
     556             : #        elif HAVE_MMX_INLINE
     557             : #            define TEMPLATE_PP_MMX 1
     558             : #            include "postprocess_template.c"
     559             : #        endif
     560             : #    endif
     561             : #endif
     562             : 
     563             : typedef void (*pp_fn)(const uint8_t src[], int srcStride, uint8_t dst[], int dstStride, int width, int height,
     564             :                       const QP_STORE_T QPs[], int QPStride, int isColor, PPContext *c2);
     565             : 
     566         123 : static inline void postProcess(const uint8_t src[], int srcStride, uint8_t dst[], int dstStride, int width, int height,
     567             :         const QP_STORE_T QPs[], int QPStride, int isColor, pp_mode *vm, pp_context *vc)
     568             : {
     569         123 :     pp_fn pp = postProcess_C;
     570         123 :     PPContext *c= (PPContext *)vc;
     571         123 :     PPMode *ppMode= (PPMode *)vm;
     572         123 :     c->ppMode= *ppMode; //FIXME
     573             : 
     574         123 :     if (!(ppMode->lumMode & BITEXACT)) {
     575             : #if CONFIG_RUNTIME_CPUDETECT
     576             : #if ARCH_X86 && HAVE_INLINE_ASM
     577             :         // ordered per speed fastest first
     578          15 :         if      (c->cpuCaps & AV_CPU_FLAG_SSE2)     pp = postProcess_SSE2;
     579          15 :         else if (c->cpuCaps & AV_CPU_FLAG_MMXEXT)   pp = postProcess_MMX2;
     580          15 :         else if (c->cpuCaps & AV_CPU_FLAG_3DNOW)    pp = postProcess_3DNow;
     581          15 :         else if (c->cpuCaps & AV_CPU_FLAG_MMX)      pp = postProcess_MMX;
     582             : #elif HAVE_ALTIVEC
     583             :         if      (c->cpuCaps & AV_CPU_FLAG_ALTIVEC)  pp = postProcess_altivec;
     584             : #endif
     585             : #else /* CONFIG_RUNTIME_CPUDETECT */
     586             : #if     HAVE_SSE2_INLINE
     587             :         pp = postProcess_SSE2;
     588             : #elif   HAVE_MMXEXT_INLINE
     589             :         pp = postProcess_MMX2;
     590             : #elif HAVE_AMD3DNOW_INLINE
     591             :         pp = postProcess_3DNow;
     592             : #elif HAVE_MMX_INLINE
     593             :         pp = postProcess_MMX;
     594             : #elif HAVE_ALTIVEC
     595             :         pp = postProcess_altivec;
     596             : #endif
     597             : #endif /* !CONFIG_RUNTIME_CPUDETECT */
     598             :     }
     599             : 
     600         123 :     pp(src, srcStride, dst, dstStride, width, height, QPs, QPStride, isColor, c);
     601         123 : }
     602             : 
     603             : /* -pp Command line Help
     604             : */
     605             : const char pp_help[] =
     606             : "Available postprocessing filters:\n"
     607             : "Filters                        Options\n"
     608             : "short  long name       short   long option     Description\n"
     609             : "*      *               a       autoq           CPU power dependent enabler\n"
     610             : "                       c       chrom           chrominance filtering enabled\n"
     611             : "                       y       nochrom         chrominance filtering disabled\n"
     612             : "                       n       noluma          luma filtering disabled\n"
     613             : "hb     hdeblock        (2 threshold)           horizontal deblocking filter\n"
     614             : "       1. difference factor: default=32, higher -> more deblocking\n"
     615             : "       2. flatness threshold: default=39, lower -> more deblocking\n"
     616             : "                       the h & v deblocking filters share these\n"
     617             : "                       so you can't set different thresholds for h / v\n"
     618             : "vb     vdeblock        (2 threshold)           vertical deblocking filter\n"
     619             : "ha     hadeblock       (2 threshold)           horizontal deblocking filter\n"
     620             : "va     vadeblock       (2 threshold)           vertical deblocking filter\n"
     621             : "h1     x1hdeblock                              experimental h deblock filter 1\n"
     622             : "v1     x1vdeblock                              experimental v deblock filter 1\n"
     623             : "dr     dering                                  deringing filter\n"
     624             : "al     autolevels                              automatic brightness / contrast\n"
     625             : "                       f        fullyrange     stretch luminance to (0..255)\n"
     626             : "lb     linblenddeint                           linear blend deinterlacer\n"
     627             : "li     linipoldeint                            linear interpolating deinterlace\n"
     628             : "ci     cubicipoldeint                          cubic interpolating deinterlacer\n"
     629             : "md     mediandeint                             median deinterlacer\n"
     630             : "fd     ffmpegdeint                             ffmpeg deinterlacer\n"
     631             : "l5     lowpass5                                FIR lowpass deinterlacer\n"
     632             : "de     default                                 hb:a,vb:a,dr:a\n"
     633             : "fa     fast                                    h1:a,v1:a,dr:a\n"
     634             : "ac                                             ha:a:128:7,va:a,dr:a\n"
     635             : "tn     tmpnoise        (3 threshold)           temporal noise reducer\n"
     636             : "                     1. <= 2. <= 3.            larger -> stronger filtering\n"
     637             : "fq     forceQuant      <quantizer>             force quantizer\n"
     638             : "Usage:\n"
     639             : "<filterName>[:<option>[:<option>...]][[,|/][-]<filterName>[:<option>...]]...\n"
     640             : "long form example:\n"
     641             : "vdeblock:autoq/hdeblock:autoq/linblenddeint    default,-vdeblock\n"
     642             : "short form example:\n"
     643             : "vb:a/hb:a/lb                                   de,-vb\n"
     644             : "more examples:\n"
     645             : "tn:64:128:256\n"
     646             : "\n"
     647             : ;
     648             : 
     649          56 : pp_mode *pp_get_mode_by_name_and_quality(const char *name, int quality)
     650             : {
     651             :     char temp[GET_MODE_BUFFER_SIZE];
     652          56 :     char *p= temp;
     653             :     static const char filterDelimiters[] = ",/";
     654             :     static const char optionDelimiters[] = ":|";
     655             :     struct PPMode *ppMode;
     656             :     char *filterToken;
     657             : 
     658          56 :     if (!name)  {
     659           0 :         av_log(NULL, AV_LOG_ERROR, "pp: Missing argument\n");
     660           0 :         return NULL;
     661             :     }
     662             : 
     663          56 :     if (!strcmp(name, "help")) {
     664             :         const char *p;
     665           0 :         for (p = pp_help; strchr(p, '\n'); p = strchr(p, '\n') + 1) {
     666           0 :             av_strlcpy(temp, p, FFMIN(sizeof(temp), strchr(p, '\n') - p + 2));
     667           0 :             av_log(NULL, AV_LOG_INFO, "%s", temp);
     668             :         }
     669           0 :         return NULL;
     670             :     }
     671             : 
     672          56 :     ppMode= av_malloc(sizeof(PPMode));
     673          56 :     if (!ppMode)
     674           0 :         return NULL;
     675             : 
     676          56 :     ppMode->lumMode= 0;
     677          56 :     ppMode->chromMode= 0;
     678          56 :     ppMode->maxTmpNoise[0]= 700;
     679          56 :     ppMode->maxTmpNoise[1]= 1500;
     680          56 :     ppMode->maxTmpNoise[2]= 3000;
     681          56 :     ppMode->maxAllowedY= 234;
     682          56 :     ppMode->minAllowedY= 16;
     683          56 :     ppMode->baseDcDiff= 256/8;
     684          56 :     ppMode->flatnessThreshold= 56-16-1;
     685          56 :     ppMode->maxClippedThreshold= (AVRational){1,100};
     686          56 :     ppMode->error=0;
     687             : 
     688          56 :     memset(temp, 0, GET_MODE_BUFFER_SIZE);
     689          56 :     av_strlcpy(temp, name, GET_MODE_BUFFER_SIZE - 1);
     690             : 
     691          56 :     av_log(NULL, AV_LOG_DEBUG, "pp: %s\n", name);
     692             : 
     693         224 :     for(;;){
     694             :         const char *filterName;
     695         280 :         int q= 1000000; //PP_QUALITY_MAX;
     696         280 :         int chrom=-1;
     697         280 :         int luma=-1;
     698             :         const char *option;
     699             :         const char *options[OPTIONS_ARRAY_SIZE];
     700             :         int i;
     701         280 :         int filterNameOk=0;
     702         280 :         int numOfUnknownOptions=0;
     703         280 :         int enable=1; //does the user want us to enabled or disabled the filter
     704             :         char *tokstate;
     705             : 
     706         280 :         filterToken= av_strtok(p, filterDelimiters, &tokstate);
     707         336 :         if(!filterToken) break;
     708         224 :         p+= strlen(filterToken) + 1; // p points to next filterToken
     709         224 :         filterName= av_strtok(filterToken, optionDelimiters, &tokstate);
     710         224 :         if (!filterName) {
     711           0 :             ppMode->error++;
     712           0 :             break;
     713             :         }
     714         224 :         av_log(NULL, AV_LOG_DEBUG, "pp: %s::%s\n", filterToken, filterName);
     715             : 
     716         224 :         if(*filterName == '-'){
     717           0 :             enable=0;
     718           0 :             filterName++;
     719             :         }
     720             : 
     721             :         for(;;){ //for all options
     722         266 :             option= av_strtok(NULL, optionDelimiters, &tokstate);
     723         245 :             if(!option) break;
     724             : 
     725          21 :             av_log(NULL, AV_LOG_DEBUG, "pp: option: %s\n", option);
     726          21 :             if(!strcmp("autoq", option) || !strcmp("a", option)) q= quality;
     727          21 :             else if(!strcmp("nochrom", option) || !strcmp("y", option)) chrom=0;
     728          21 :             else if(!strcmp("chrom", option) || !strcmp("c", option)) chrom=1;
     729          21 :             else if(!strcmp("noluma", option) || !strcmp("n", option)) luma=0;
     730             :             else{
     731          21 :                 options[numOfUnknownOptions] = option;
     732          21 :                 numOfUnknownOptions++;
     733             :             }
     734          21 :             if(numOfUnknownOptions >= OPTIONS_ARRAY_SIZE-1) break;
     735             :         }
     736         224 :         options[numOfUnknownOptions] = NULL;
     737             : 
     738             :         /* replace stuff from the replace Table */
     739        1344 :         for(i=0; replaceTable[2*i]; i++){
     740        1120 :             if(!strcmp(replaceTable[2*i], filterName)){
     741           0 :                 size_t newlen = strlen(replaceTable[2*i + 1]);
     742             :                 int plen;
     743             :                 int spaceLeft;
     744             : 
     745           0 :                 p--, *p=',';
     746             : 
     747           0 :                 plen= strlen(p);
     748           0 :                 spaceLeft= p - temp + plen;
     749           0 :                 if(spaceLeft + newlen  >= GET_MODE_BUFFER_SIZE - 1){
     750           0 :                     ppMode->error++;
     751           0 :                     break;
     752             :                 }
     753           0 :                 memmove(p + newlen, p, plen+1);
     754           0 :                 memcpy(p, replaceTable[2*i + 1], newlen);
     755           0 :                 filterNameOk=1;
     756             :             }
     757             :         }
     758             : 
     759        4256 :         for(i=0; filters[i].shortName; i++){
     760        4032 :             if(   !strcmp(filters[i].longName, filterName)
     761        4032 :                || !strcmp(filters[i].shortName, filterName)){
     762         224 :                 ppMode->lumMode &= ~filters[i].mask;
     763         224 :                 ppMode->chromMode &= ~filters[i].mask;
     764             : 
     765         224 :                 filterNameOk=1;
     766         224 :                 if(!enable) break; // user wants to disable it
     767             : 
     768         224 :                 if(q >= filters[i].minLumQuality && luma)
     769         224 :                     ppMode->lumMode|= filters[i].mask;
     770         224 :                 if(chrom==1 || (chrom==-1 && filters[i].chromDefault))
     771         203 :                     if(q >= filters[i].minChromQuality)
     772         203 :                             ppMode->chromMode|= filters[i].mask;
     773             : 
     774         224 :                 if(filters[i].mask == LEVEL_FIX){
     775             :                     int o;
     776          21 :                     ppMode->minAllowedY= 16;
     777          21 :                     ppMode->maxAllowedY= 234;
     778          21 :                     for(o=0; options[o]; o++){
     779           0 :                         if(  !strcmp(options[o],"fullyrange")
     780           0 :                            ||!strcmp(options[o],"f")){
     781           0 :                             ppMode->minAllowedY= 0;
     782           0 :                             ppMode->maxAllowedY= 255;
     783           0 :                             numOfUnknownOptions--;
     784             :                         }
     785             :                     }
     786             :                 }
     787         203 :                 else if(filters[i].mask == TEMP_NOISE_FILTER)
     788             :                 {
     789             :                     int o;
     790          21 :                     int numOfNoises=0;
     791             : 
     792          21 :                     for(o=0; options[o]; o++){
     793             :                         char *tail;
     794           0 :                         ppMode->maxTmpNoise[numOfNoises]=
     795           0 :                             strtol(options[o], &tail, 0);
     796           0 :                         if(tail!=options[o]){
     797           0 :                             numOfNoises++;
     798           0 :                             numOfUnknownOptions--;
     799           0 :                             if(numOfNoises >= 3) break;
     800             :                         }
     801             :                     }
     802             :                 }
     803         182 :                 else if(filters[i].mask == V_DEBLOCK   || filters[i].mask == H_DEBLOCK
     804         196 :                      || filters[i].mask == V_A_DEBLOCK || filters[i].mask == H_A_DEBLOCK){
     805             :                     int o;
     806             : 
     807          70 :                     for(o=0; options[o] && o<2; o++){
     808             :                         char *tail;
     809          14 :                         int val= strtol(options[o], &tail, 0);
     810          14 :                         if(tail==options[o]) break;
     811             : 
     812          14 :                         numOfUnknownOptions--;
     813          14 :                         if(o==0) ppMode->baseDcDiff= val;
     814           7 :                         else ppMode->flatnessThreshold= val;
     815             :                     }
     816             :                 }
     817         126 :                 else if(filters[i].mask == FORCE_QUANT){
     818             :                     int o;
     819           7 :                     ppMode->forcedQuant= 15;
     820             : 
     821          14 :                     for(o=0; options[o] && o<1; o++){
     822             :                         char *tail;
     823           7 :                         int val= strtol(options[o], &tail, 0);
     824           7 :                         if(tail==options[o]) break;
     825             : 
     826           7 :                         numOfUnknownOptions--;
     827           7 :                         ppMode->forcedQuant= val;
     828             :                     }
     829             :                 }
     830             :             }
     831             :         }
     832         224 :         if(!filterNameOk) ppMode->error++;
     833         224 :         ppMode->error += numOfUnknownOptions;
     834             :     }
     835             : 
     836          56 :     av_log(NULL, AV_LOG_DEBUG, "pp: lumMode=%X, chromMode=%X\n", ppMode->lumMode, ppMode->chromMode);
     837          56 :     if(ppMode->error){
     838           0 :         av_log(NULL, AV_LOG_ERROR, "%d errors in postprocess string \"%s\"\n", ppMode->error, name);
     839           0 :         av_free(ppMode);
     840           0 :         return NULL;
     841             :     }
     842          56 :     return ppMode;
     843             : }
     844             : 
     845          56 : void pp_free_mode(pp_mode *mode){
     846          56 :     av_free(mode);
     847          56 : }
     848             : 
     849         224 : static void reallocAlign(void **p, int size){
     850         224 :     av_free(*p);
     851         224 :     *p= av_mallocz(size);
     852         224 : }
     853             : 
     854          16 : static void reallocBuffers(PPContext *c, int width, int height, int stride, int qpStride){
     855          16 :     int mbWidth = (width+15)>>4;
     856          16 :     int mbHeight= (height+15)>>4;
     857             :     int i;
     858             : 
     859          16 :     c->stride= stride;
     860          16 :     c->qpStride= qpStride;
     861             : 
     862          16 :     reallocAlign((void **)&c->tempDst, stride*24+32);
     863          16 :     reallocAlign((void **)&c->tempSrc, stride*24);
     864          16 :     reallocAlign((void **)&c->tempBlocks, 2*16*8);
     865          16 :     reallocAlign((void **)&c->yHistogram, 256*sizeof(uint64_t));
     866        4112 :     for(i=0; i<256; i++)
     867        4096 :             c->yHistogram[i]= width*height/64*15/256;
     868             : 
     869          64 :     for(i=0; i<3; i++){
     870             :         //Note: The +17*1024 is just there so I do not have to worry about r/w over the end.
     871          48 :         reallocAlign((void **)&c->tempBlurred[i], stride*mbHeight*16 + 17*1024);
     872          48 :         reallocAlign((void **)&c->tempBlurredPast[i], 256*((height+7)&(~7))/2 + 17*1024);//FIXME size
     873             :     }
     874             : 
     875          16 :     reallocAlign((void **)&c->deintTemp, 2*width+32);
     876          16 :     reallocAlign((void **)&c->nonBQPTable, qpStride*mbHeight*sizeof(QP_STORE_T));
     877          16 :     reallocAlign((void **)&c->stdQPTable, qpStride*mbHeight*sizeof(QP_STORE_T));
     878          16 :     reallocAlign((void **)&c->forcedQPTable, mbWidth*sizeof(QP_STORE_T));
     879          16 : }
     880             : 
     881           0 : static const char * context_to_name(void * ptr) {
     882           0 :     return "postproc";
     883             : }
     884             : 
     885             : static const AVClass av_codec_context_class = { "Postproc", context_to_name, NULL };
     886             : 
     887           8 : av_cold pp_context *pp_get_context(int width, int height, int cpuCaps){
     888           8 :     PPContext *c= av_mallocz(sizeof(PPContext));
     889           8 :     int stride= FFALIGN(width, 16);  //assumed / will realloc if needed
     890           8 :     int qpStride= (width+15)/16 + 2; //assumed / will realloc if needed
     891             : 
     892           8 :     if (!c)
     893           0 :         return NULL;
     894             : 
     895           8 :     c->av_class = &av_codec_context_class;
     896           8 :     if(cpuCaps&PP_FORMAT){
     897           8 :         c->hChromaSubSample= cpuCaps&0x3;
     898           8 :         c->vChromaSubSample= (cpuCaps>>4)&0x3;
     899             :     }else{
     900           0 :         c->hChromaSubSample= 1;
     901           0 :         c->vChromaSubSample= 1;
     902             :     }
     903           8 :     if (cpuCaps & PP_CPU_CAPS_AUTO) {
     904           8 :         c->cpuCaps = av_get_cpu_flags();
     905             :     } else {
     906           0 :         c->cpuCaps = 0;
     907           0 :         if (cpuCaps & PP_CPU_CAPS_MMX)      c->cpuCaps |= AV_CPU_FLAG_MMX;
     908           0 :         if (cpuCaps & PP_CPU_CAPS_MMX2)     c->cpuCaps |= AV_CPU_FLAG_MMXEXT;
     909           0 :         if (cpuCaps & PP_CPU_CAPS_3DNOW)    c->cpuCaps |= AV_CPU_FLAG_3DNOW;
     910           0 :         if (cpuCaps & PP_CPU_CAPS_ALTIVEC)  c->cpuCaps |= AV_CPU_FLAG_ALTIVEC;
     911             :     }
     912             : 
     913           8 :     reallocBuffers(c, width, height, stride, qpStride);
     914             : 
     915           8 :     c->frameNum=-1;
     916             : 
     917           8 :     return c;
     918             : }
     919             : 
     920           8 : av_cold void pp_free_context(void *vc){
     921           8 :     PPContext *c = (PPContext*)vc;
     922             :     int i;
     923             : 
     924          32 :     for(i=0; i<FF_ARRAY_ELEMS(c->tempBlurred); i++)
     925          24 :         av_free(c->tempBlurred[i]);
     926          32 :     for(i=0; i<FF_ARRAY_ELEMS(c->tempBlurredPast); i++)
     927          24 :         av_free(c->tempBlurredPast[i]);
     928             : 
     929           8 :     av_free(c->tempBlocks);
     930           8 :     av_free(c->yHistogram);
     931           8 :     av_free(c->tempDst);
     932           8 :     av_free(c->tempSrc);
     933           8 :     av_free(c->deintTemp);
     934           8 :     av_free(c->stdQPTable);
     935           8 :     av_free(c->nonBQPTable);
     936           8 :     av_free(c->forcedQPTable);
     937             : 
     938           8 :     memset(c, 0, sizeof(PPContext));
     939             : 
     940           8 :     av_free(c);
     941           8 : }
     942             : 
     943          41 : void  pp_postprocess(const uint8_t * src[3], const int srcStride[3],
     944             :                      uint8_t * dst[3], const int dstStride[3],
     945             :                      int width, int height,
     946             :                      const QP_STORE_T *QP_store,  int QPStride,
     947             :                      pp_mode *vm,  void *vc, int pict_type)
     948             : {
     949          41 :     int mbWidth = (width+15)>>4;
     950          41 :     int mbHeight= (height+15)>>4;
     951          41 :     PPMode *mode = vm;
     952          41 :     PPContext *c = vc;
     953          41 :     int minStride= FFMAX(FFABS(srcStride[0]), FFABS(dstStride[0]));
     954          41 :     int absQPStride = FFABS(QPStride);
     955             : 
     956             :     // c->stride and c->QPStride are always positive
     957          41 :     if(c->stride < minStride || c->qpStride < absQPStride)
     958          16 :         reallocBuffers(c, width, height,
     959           8 :                        FFMAX(minStride, c->stride),
     960           8 :                        FFMAX(c->qpStride, absQPStride));
     961             : 
     962          41 :     if(!QP_store || (mode->lumMode & FORCE_QUANT)){
     963             :         int i;
     964          21 :         QP_store= c->forcedQPTable;
     965          21 :         absQPStride = QPStride = 0;
     966          21 :         if(mode->lumMode & FORCE_QUANT)
     967           5 :             for(i=0; i<mbWidth; i++) c->forcedQPTable[i]= mode->forcedQuant;
     968             :         else
     969          16 :             for(i=0; i<mbWidth; i++) c->forcedQPTable[i]= 1;
     970             :     }
     971             : 
     972          41 :     if(pict_type & PP_PICT_TYPE_QP2){
     973             :         int i;
     974           0 :         const int count= FFMAX(mbHeight * absQPStride, mbWidth);
     975           0 :         for(i=0; i<(count>>2); i++){
     976           0 :             AV_WN32(c->stdQPTable + (i<<2), AV_RN32(QP_store + (i<<2)) >> 1 & 0x7F7F7F7F);
     977             :         }
     978           0 :         for(i<<=2; i<count; i++){
     979           0 :             c->stdQPTable[i] = QP_store[i]>>1;
     980             :         }
     981           0 :         QP_store= c->stdQPTable;
     982           0 :         QPStride= absQPStride;
     983             :     }
     984             : 
     985             :     if(0){
     986             :         int x,y;
     987             :         for(y=0; y<mbHeight; y++){
     988             :             for(x=0; x<mbWidth; x++){
     989             :                 av_log(c, AV_LOG_INFO, "%2d ", QP_store[x + y*QPStride]);
     990             :             }
     991             :             av_log(c, AV_LOG_INFO, "\n");
     992             :         }
     993             :         av_log(c, AV_LOG_INFO, "\n");
     994             :     }
     995             : 
     996          41 :     if((pict_type&7)!=3){
     997          38 :         if (QPStride >= 0){
     998             :             int i;
     999          38 :             const int count= FFMAX(mbHeight * QPStride, mbWidth);
    1000        1834 :             for(i=0; i<(count>>2); i++){
    1001        1796 :                 AV_WN32(c->nonBQPTable + (i<<2), AV_RN32(QP_store + (i<<2)) & 0x3F3F3F3F);
    1002             :             }
    1003          84 :             for(i<<=2; i<count; i++){
    1004          46 :                 c->nonBQPTable[i] = QP_store[i] & 0x3F;
    1005             :             }
    1006             :         } else {
    1007             :             int i,j;
    1008           0 :             for(i=0; i<mbHeight; i++) {
    1009           0 :                 for(j=0; j<absQPStride; j++) {
    1010           0 :                     c->nonBQPTable[i*absQPStride+j] = QP_store[i*QPStride+j] & 0x3F;
    1011             :                 }
    1012             :             }
    1013             :         }
    1014             :     }
    1015             : 
    1016          41 :     av_log(c, AV_LOG_DEBUG, "using npp filters 0x%X/0x%X\n",
    1017             :            mode->lumMode, mode->chromMode);
    1018             : 
    1019          41 :     postProcess(src[0], srcStride[0], dst[0], dstStride[0],
    1020             :                 width, height, QP_store, QPStride, 0, mode, c);
    1021             : 
    1022          41 :     if (!(src[1] && src[2] && dst[1] && dst[2]))
    1023           0 :         return;
    1024             : 
    1025          41 :     width  = (width )>>c->hChromaSubSample;
    1026          41 :     height = (height)>>c->vChromaSubSample;
    1027             : 
    1028          41 :     if(mode->chromMode){
    1029          41 :         postProcess(src[1], srcStride[1], dst[1], dstStride[1],
    1030             :                     width, height, QP_store, QPStride, 1, mode, c);
    1031          41 :         postProcess(src[2], srcStride[2], dst[2], dstStride[2],
    1032             :                     width, height, QP_store, QPStride, 2, mode, c);
    1033             :     }
    1034           0 :     else if(srcStride[1] == dstStride[1] && srcStride[2] == dstStride[2]){
    1035           0 :         linecpy(dst[1], src[1], height, srcStride[1]);
    1036           0 :         linecpy(dst[2], src[2], height, srcStride[2]);
    1037             :     }else{
    1038             :         int y;
    1039           0 :         for(y=0; y<height; y++){
    1040           0 :             memcpy(&(dst[1][y*dstStride[1]]), &(src[1][y*srcStride[1]]), width);
    1041           0 :             memcpy(&(dst[2][y*dstStride[2]]), &(src[2][y*srcStride[2]]), width);
    1042             :         }
    1043             :     }
    1044             : }

Generated by: LCOV version 1.13