libpostproc/postprocess_internal.h

   1 /*
   2  * Copyright (C) 2001-2002 Michael Niedermayer (michaelni@gmx.at)
   3  *
   4  * This file is part of FFmpeg.
   5  *
   6  * FFmpeg is free software; you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License as published by
   8  * the Free Software Foundation; either version 2 of the License, or
   9  * (at your option) any later version.
  10  *
  11  * FFmpeg is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  * GNU General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * along with FFmpeg; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  19  */
  20
  21 /**
  22  * @file postprocess_internal.h
  23  * internal api header.
  24  */
  25
  26 #ifndef POSTPROCESS_INTERNAL_H
  27 #define POSTPROCESS_INTERNAL_H
  28
  29 #include "avutil.h"
  30 #include "postprocess.h"
  31
  32 #define V_DEBLOCK       0x01
  33 #define H_DEBLOCK       0x02
  34 #define DERING          0x04
  35 #define LEVEL_FIX       0x08 ///< Brightness & Contrast
  36
  37 #define LUM_V_DEBLOCK   V_DEBLOCK               //   1
  38 #define LUM_H_DEBLOCK   H_DEBLOCK               //   2
  39 #define CHROM_V_DEBLOCK (V_DEBLOCK<<4)          //  16
  40 #define CHROM_H_DEBLOCK (H_DEBLOCK<<4)          //  32
  41 #define LUM_DERING      DERING                  //   4
  42 #define CHROM_DERING    (DERING<<4)             //  64
  43 #define LUM_LEVEL_FIX   LEVEL_FIX               //   8
  44 #define CHROM_LEVEL_FIX (LEVEL_FIX<<4)          // 128 (not implemented yet)
  45
  46 // Experimental vertical filters
  47 #define V_X1_FILTER     0x0200                  // 512
  48 #define V_A_DEBLOCK     0x0400
  49
  50 // Experimental horizontal filters
  51 #define H_X1_FILTER     0x2000                  // 8192
  52 #define H_A_DEBLOCK     0x4000
  53
  54 /// select between full y range (255-0) or standart one (234-16)
  55 #define FULL_Y_RANGE    0x8000                  // 32768
  56
  57 //Deinterlacing Filters
  58 #define        LINEAR_IPOL_DEINT_FILTER         0x10000 // 65536
  59 #define        LINEAR_BLEND_DEINT_FILTER        0x20000 // 131072
  60 #define        CUBIC_BLEND_DEINT_FILTER         0x8000  // (not implemented yet)
  61 #define        CUBIC_IPOL_DEINT_FILTER          0x40000 // 262144
  62 #define        MEDIAN_DEINT_FILTER              0x80000 // 524288
  63 #define        FFMPEG_DEINT_FILTER              0x400000
  64 #define        LOWPASS5_DEINT_FILTER            0x800000
  65
  66 #define TEMP_NOISE_FILTER               0x100000
  67 #define FORCE_QUANT                     0x200000
  68
  69 #if ( defined(__PIC__) || defined(__pic__) ) && ! defined(PIC)
  70 #    define PIC
  71 #endif
  72
  73 //use if you want a faster postprocessing code
  74 //cannot differentiate between chroma & luma filters (both on or both off)
  75 //obviously the -pp option on the command line has no effect except turning the here selected
  76 //filters on
  77 //#define COMPILE_TIME_MODE 0x77
  78
  79 #if 1
  80 static inline int CLIP(int a){
  81         if(a&256) return ((a)>>31)^(-1);
  82         else      return a;
  83 }
  84 //#define CLIP(a) (((a)&256) ? ((a)>>31)^(-1) : (a))
  85 #elif 0
  86 #define CLIP(a) clip_tab[a]
  87 #else
  88 #define CLIP(a) (a)
  89 #endif
  90 /**
  91  * Postprocessng filter.
  92  */
  93 struct PPFilter{
  94         const char *shortName;
  95         const char *longName;
  96         int chromDefault;       ///< is chrominance filtering on by default if this filter is manually activated
  97         int minLumQuality;      ///< minimum quality to turn luminance filtering on
  98         int minChromQuality;    ///< minimum quality to turn chrominance filtering on
  99         int mask;               ///< Bitmask to turn this filter on
 100 };
 101
 102 /**
 103  * Postprocessng mode.
 104  */
 105 typedef struct PPMode{
 106         int lumMode;                    ///< acivates filters for luminance
 107         int chromMode;                  ///< acivates filters for chrominance
 108         int error;                      ///< non zero on error
 109
 110         int minAllowedY;                ///< for brigtness correction
 111         int maxAllowedY;                ///< for brihtness correction
 112         float maxClippedThreshold;      ///< amount of "black" u r willing to loose to get a brightness corrected picture
 113
 114         int maxTmpNoise[3];             ///< for Temporal Noise Reducing filter (Maximal sum of abs differences)
 115
 116         int baseDcDiff;
 117         int flatnessThreshold;
 118
 119         int forcedQuant;                ///< quantizer if FORCE_QUANT is used
 120 } PPMode;
 121
 122 /**
 123  * postprocess context.
 124  */
 125 typedef struct PPContext{
 126         /**
 127          * info on struct for av_log
 128          */
 129         AVClass *av_class;
 130
 131         uint8_t *tempBlocks; ///<used for the horizontal code
 132
 133         /**
 134          * luma histogram.
 135          * we need 64bit here otherwise we'll going to have a problem
 136          * after watching a black picture for 5 hours
 137          */
 138         uint64_t *yHistogram;
 139
 140         DECLARE_ALIGNED(8, uint64_t, packedYOffset);
 141         DECLARE_ALIGNED(8, uint64_t, packedYScale);
 142
 143         /** Temporal noise reducing buffers */
 144         uint8_t *tempBlured[3];
 145         int32_t *tempBluredPast[3];
 146
 147         /** Temporary buffers for handling the last row(s) */
 148         uint8_t *tempDst;
 149         uint8_t *tempSrc;
 150
 151         uint8_t *deintTemp;
 152
 153         DECLARE_ALIGNED(8, uint64_t, pQPb);
 154         DECLARE_ALIGNED(8, uint64_t, pQPb2);
 155
 156         DECLARE_ALIGNED(8, uint64_t, mmxDcOffset[64]);
 157         DECLARE_ALIGNED(8, uint64_t, mmxDcThreshold[64]);
 158
 159         QP_STORE_T *stdQPTable;       ///< used to fix MPEG2 style qscale
 160         QP_STORE_T *nonBQPTable;
 161         QP_STORE_T *forcedQPTable;
 162
 163         int QP;
 164         int nonBQP;
 165
 166         int frameNum;
 167
 168         int cpuCaps;
 169
 170         int qpStride; ///<size of qp buffers (needed to realloc them if needed)
 171         int stride;   ///<size of some buffers (needed to realloc them if needed)
 172
 173         int hChromaSubSample;
 174         int vChromaSubSample;
 175
 176         PPMode ppMode;
 177 } PPContext;
 178
 179
 180 static inline void linecpy(void *dest, void *src, int lines, int stride)
 181 {
 182         if (stride > 0) {
 183                 memcpy(dest, src, lines*stride);
 184         } else {
 185                 memcpy(dest+(lines-1)*stride, src+(lines-1)*stride, -lines*stride);
 186         }
 187 }
 188
 189 #endif // POSTPROCESS_INTERNAL_H