yading@11: /*
yading@11:  * Copyright (C) 2001-2002 Michael Niedermayer (michaelni@gmx.at)
yading@11:  *
yading@11:  * This file is part of FFmpeg.
yading@11:  *
yading@11:  * FFmpeg is free software; you can redistribute it and/or modify
yading@11:  * it under the terms of the GNU General Public License as published by
yading@11:  * the Free Software Foundation; either version 2 of the License, or
yading@11:  * (at your option) any later version.
yading@11:  *
yading@11:  * FFmpeg is distributed in the hope that it will be useful,
yading@11:  * but WITHOUT ANY WARRANTY; without even the implied warranty of
yading@11:  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
yading@11:  * GNU General Public License for more details.
yading@11:  *
yading@11:  * You should have received a copy of the GNU General Public License
yading@11:  * along with FFmpeg; if not, write to the Free Software
yading@11:  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
yading@11:  */
yading@11: 
yading@11: /**
yading@11:  * @file
yading@11:  * internal API header.
yading@11:  */
yading@11: 
yading@11: #ifndef POSTPROC_POSTPROCESS_INTERNAL_H
yading@11: #define POSTPROC_POSTPROCESS_INTERNAL_H
yading@11: 
yading@11: #include <string.h>
yading@11: #include "libavutil/avutil.h"
yading@11: #include "libavutil/intmath.h"
yading@11: #include "libavutil/log.h"
yading@11: #include "postprocess.h"
yading@11: 
yading@11: #define V_DEBLOCK       0x01
yading@11: #define H_DEBLOCK       0x02
yading@11: #define DERING          0x04
yading@11: #define LEVEL_FIX       0x08 ///< Brightness & Contrast
yading@11: 
yading@11: #define LUM_V_DEBLOCK   V_DEBLOCK               //   1
yading@11: #define LUM_H_DEBLOCK   H_DEBLOCK               //   2
yading@11: #define CHROM_V_DEBLOCK (V_DEBLOCK<<4)          //  16
yading@11: #define CHROM_H_DEBLOCK (H_DEBLOCK<<4)          //  32
yading@11: #define LUM_DERING      DERING                  //   4
yading@11: #define CHROM_DERING    (DERING<<4)             //  64
yading@11: #define LUM_LEVEL_FIX   LEVEL_FIX               //   8
yading@11: #define CHROM_LEVEL_FIX (LEVEL_FIX<<4)          // 128 (not implemented yet)
yading@11: 
yading@11: // Experimental vertical filters
yading@11: #define V_X1_FILTER     0x0200                  // 512
yading@11: #define V_A_DEBLOCK     0x0400
yading@11: 
yading@11: // Experimental horizontal filters
yading@11: #define H_X1_FILTER     0x2000                  // 8192
yading@11: #define H_A_DEBLOCK     0x4000
yading@11: 
yading@11: /// select between full y range (255-0) or standard one (234-16)
yading@11: #define FULL_Y_RANGE    0x8000                  // 32768
yading@11: 
yading@11: //Deinterlacing Filters
yading@11: #define LINEAR_IPOL_DEINT_FILTER        0x10000 // 65536
yading@11: #define LINEAR_BLEND_DEINT_FILTER       0x20000 // 131072
yading@11: #define CUBIC_BLEND_DEINT_FILTER        0x8000  // (not implemented yet)
yading@11: #define CUBIC_IPOL_DEINT_FILTER         0x40000 // 262144
yading@11: #define MEDIAN_DEINT_FILTER             0x80000 // 524288
yading@11: #define FFMPEG_DEINT_FILTER             0x400000
yading@11: #define LOWPASS5_DEINT_FILTER           0x800000
yading@11: 
yading@11: #define TEMP_NOISE_FILTER               0x100000
yading@11: #define FORCE_QUANT                     0x200000
yading@11: #define BITEXACT                        0x1000000
yading@11: 
yading@11: //use if you want a faster postprocessing code
yading@11: //cannot differentiate between chroma & luma filters (both on or both off)
yading@11: //obviously the -pp option on the command line has no effect except turning the here selected
yading@11: //filters on
yading@11: //#define COMPILE_TIME_MODE 0x77
yading@11: 
yading@11: #define CLIP av_clip_uint8
yading@11: 
yading@11: /**
yading@11:  * Postprocessing filter.
yading@11:  */
yading@11: struct PPFilter{
yading@11:     const char *shortName;
yading@11:     const char *longName;
yading@11:     int chromDefault;       ///< is chrominance filtering on by default if this filter is manually activated
yading@11:     int minLumQuality;      ///< minimum quality to turn luminance filtering on
yading@11:     int minChromQuality;    ///< minimum quality to turn chrominance filtering on
yading@11:     int mask;               ///< Bitmask to turn this filter on
yading@11: };
yading@11: 
yading@11: /**
yading@11:  * Postprocessing mode.
yading@11:  */
yading@11: typedef struct PPMode{
yading@11:     int lumMode;                    ///< activates filters for luminance
yading@11:     int chromMode;                  ///< activates filters for chrominance
yading@11:     int error;                      ///< non zero on error
yading@11: 
yading@11:     int minAllowedY;                ///< for brightness correction
yading@11:     int maxAllowedY;                ///< for brightness correction
yading@11:     float maxClippedThreshold;      ///< amount of "black" you are willing to lose to get a brightness-corrected picture
yading@11: 
yading@11:     int maxTmpNoise[3];             ///< for Temporal Noise Reducing filter (Maximal sum of abs differences)
yading@11: 
yading@11:     int baseDcDiff;
yading@11:     int flatnessThreshold;
yading@11: 
yading@11:     int forcedQuant;                ///< quantizer if FORCE_QUANT is used
yading@11: } PPMode;
yading@11: 
yading@11: /**
yading@11:  * postprocess context.
yading@11:  */
yading@11: typedef struct PPContext{
yading@11:     /**
yading@11:      * info on struct for av_log
yading@11:      */
yading@11:     const AVClass *av_class;
yading@11: 
yading@11:     uint8_t *tempBlocks; ///<used for the horizontal code
yading@11: 
yading@11:     /**
yading@11:      * luma histogram.
yading@11:      * we need 64bit here otherwise we'll going to have a problem
yading@11:      * after watching a black picture for 5 hours
yading@11:      */
yading@11:     uint64_t *yHistogram;
yading@11: 
yading@11:     DECLARE_ALIGNED(8, uint64_t, packedYOffset);
yading@11:     DECLARE_ALIGNED(8, uint64_t, packedYScale);
yading@11: 
yading@11:     /** Temporal noise reducing buffers */
yading@11:     uint8_t *tempBlurred[3];
yading@11:     int32_t *tempBlurredPast[3];
yading@11: 
yading@11:     /** Temporary buffers for handling the last row(s) */
yading@11:     uint8_t *tempDst;
yading@11:     uint8_t *tempSrc;
yading@11: 
yading@11:     uint8_t *deintTemp;
yading@11: 
yading@11:     DECLARE_ALIGNED(8, uint64_t, pQPb);
yading@11:     DECLARE_ALIGNED(8, uint64_t, pQPb2);
yading@11: 
yading@11:     DECLARE_ALIGNED(8, uint64_t, mmxDcOffset)[64];
yading@11:     DECLARE_ALIGNED(8, uint64_t, mmxDcThreshold)[64];
yading@11: 
yading@11:     QP_STORE_T *stdQPTable;       ///< used to fix MPEG2 style qscale
yading@11:     QP_STORE_T *nonBQPTable;
yading@11:     QP_STORE_T *forcedQPTable;
yading@11: 
yading@11:     int QP;
yading@11:     int nonBQP;
yading@11: 
yading@11:     int frameNum;
yading@11: 
yading@11:     int cpuCaps;
yading@11: 
yading@11:     int qpStride; ///<size of qp buffers (needed to realloc them if needed)
yading@11:     int stride;   ///<size of some buffers (needed to realloc them if needed)
yading@11: 
yading@11:     int hChromaSubSample;
yading@11:     int vChromaSubSample;
yading@11: 
yading@11:     PPMode ppMode;
yading@11: } PPContext;
yading@11: 
yading@11: 
yading@11: static inline void linecpy(void *dest, const void *src, int lines, int stride) {
yading@11:     if (stride > 0) {
yading@11:         memcpy(dest, src, lines*stride);
yading@11:     } else {
yading@11:         memcpy((uint8_t*)dest+(lines-1)*stride, (const uint8_t*)src+(lines-1)*stride, -lines*stride);
yading@11:     }
yading@11: }
yading@11: 
yading@11: #endif /* POSTPROC_POSTPROCESS_INTERNAL_H */