2002-11-02 13:58:14 +00:00
|
|
|
/*
|
|
|
|
Copyright (C) 2001-2002 Michael Niedermayer (michaelni@gmx.at)
|
|
|
|
|
|
|
|
This program is free software; you can redistribute it and/or modify
|
|
|
|
it under the terms of the GNU General Public License as published by
|
|
|
|
the Free Software Foundation; either version 2 of the License, or
|
|
|
|
(at your option) any later version.
|
|
|
|
|
|
|
|
This program 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 General Public License for more details.
|
|
|
|
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
|
|
along with this program; if not, write to the Free Software
|
|
|
|
Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
|
|
*/
|
|
|
|
|
2003-03-06 13:51:18 +00:00
|
|
|
/**
|
|
|
|
* @file postprocess_internal.h
|
|
|
|
* internal api header.
|
|
|
|
*/
|
|
|
|
|
2002-11-02 13:58:14 +00:00
|
|
|
#define V_DEBLOCK 0x01
|
|
|
|
#define H_DEBLOCK 0x02
|
|
|
|
#define DERING 0x04
|
2003-03-06 13:51:18 +00:00
|
|
|
#define LEVEL_FIX 0x08 ///< Brightness & Contrast
|
2002-11-02 13:58:14 +00:00
|
|
|
|
|
|
|
#define LUM_V_DEBLOCK V_DEBLOCK // 1
|
|
|
|
#define LUM_H_DEBLOCK H_DEBLOCK // 2
|
|
|
|
#define CHROM_V_DEBLOCK (V_DEBLOCK<<4) // 16
|
|
|
|
#define CHROM_H_DEBLOCK (H_DEBLOCK<<4) // 32
|
|
|
|
#define LUM_DERING DERING // 4
|
|
|
|
#define CHROM_DERING (DERING<<4) // 64
|
|
|
|
#define LUM_LEVEL_FIX LEVEL_FIX // 8
|
|
|
|
#define CHROM_LEVEL_FIX (LEVEL_FIX<<4) // 128 (not implemented yet)
|
|
|
|
|
|
|
|
// Experimental vertical filters
|
|
|
|
#define V_X1_FILTER 0x0200 // 512
|
2004-05-27 15:57:20 +00:00
|
|
|
#define V_A_DEBLOCK 0x0400
|
2002-11-02 13:58:14 +00:00
|
|
|
|
|
|
|
// Experimental horizontal filters
|
|
|
|
#define H_X1_FILTER 0x2000 // 8192
|
2004-05-27 15:57:20 +00:00
|
|
|
#define H_A_DEBLOCK 0x4000
|
2002-11-02 13:58:14 +00:00
|
|
|
|
2003-03-06 13:51:18 +00:00
|
|
|
/// select between full y range (255-0) or standart one (234-16)
|
2002-11-02 13:58:14 +00:00
|
|
|
#define FULL_Y_RANGE 0x8000 // 32768
|
|
|
|
|
|
|
|
//Deinterlacing Filters
|
|
|
|
#define LINEAR_IPOL_DEINT_FILTER 0x10000 // 65536
|
|
|
|
#define LINEAR_BLEND_DEINT_FILTER 0x20000 // 131072
|
|
|
|
#define CUBIC_BLEND_DEINT_FILTER 0x8000 // (not implemented yet)
|
|
|
|
#define CUBIC_IPOL_DEINT_FILTER 0x40000 // 262144
|
|
|
|
#define MEDIAN_DEINT_FILTER 0x80000 // 524288
|
|
|
|
#define FFMPEG_DEINT_FILTER 0x400000
|
2003-03-29 13:31:12 +00:00
|
|
|
#define LOWPASS5_DEINT_FILTER 0x800000
|
2002-11-02 13:58:14 +00:00
|
|
|
|
|
|
|
#define TEMP_NOISE_FILTER 0x100000
|
|
|
|
#define FORCE_QUANT 0x200000
|
|
|
|
|
|
|
|
//use if u want a faster postprocessing code
|
|
|
|
//cant differentiate between chroma & luma filters (both on or both off)
|
|
|
|
//obviosly the -pp option at the commandline has no effect except turning the here selected
|
|
|
|
//filters on
|
|
|
|
//#define COMPILE_TIME_MODE 0x77
|
|
|
|
|
2003-03-29 13:31:12 +00:00
|
|
|
#if 1
|
|
|
|
static inline int CLIP(int a){
|
|
|
|
if(a&256) return ((a)>>31)^(-1);
|
|
|
|
else return a;
|
|
|
|
}
|
|
|
|
//#define CLIP(a) (((a)&256) ? ((a)>>31)^(-1) : (a))
|
|
|
|
#elif 0
|
|
|
|
#define CLIP(a) clip_tab[a]
|
|
|
|
#else
|
|
|
|
#define CLIP(a) (a)
|
|
|
|
#endif
|
2003-03-06 13:51:18 +00:00
|
|
|
/**
|
|
|
|
* Postprocessng filter.
|
|
|
|
*/
|
2002-11-02 13:58:14 +00:00
|
|
|
struct PPFilter{
|
|
|
|
char *shortName;
|
|
|
|
char *longName;
|
2003-03-06 13:51:18 +00:00
|
|
|
int chromDefault; ///< is chrominance filtering on by default if this filter is manually activated
|
|
|
|
int minLumQuality; ///< minimum quality to turn luminance filtering on
|
|
|
|
int minChromQuality; ///< minimum quality to turn chrominance filtering on
|
|
|
|
int mask; ///< Bitmask to turn this filter on
|
2002-11-02 13:58:14 +00:00
|
|
|
};
|
|
|
|
|
2003-03-06 13:51:18 +00:00
|
|
|
/**
|
|
|
|
* Postprocessng mode.
|
|
|
|
*/
|
2002-11-02 13:58:14 +00:00
|
|
|
typedef struct PPMode{
|
2003-03-06 13:51:18 +00:00
|
|
|
int lumMode; ///< acivates filters for luminance
|
|
|
|
int chromMode; ///< acivates filters for chrominance
|
|
|
|
int error; ///< non zero on error
|
2002-11-02 13:58:14 +00:00
|
|
|
|
2003-03-06 13:51:18 +00:00
|
|
|
int minAllowedY; ///< for brigtness correction
|
|
|
|
int maxAllowedY; ///< for brihtness correction
|
|
|
|
float maxClippedThreshold; ///< amount of "black" u r willing to loose to get a brightness corrected picture
|
2002-11-02 13:58:14 +00:00
|
|
|
|
2003-03-06 13:51:18 +00:00
|
|
|
int maxTmpNoise[3]; ///< for Temporal Noise Reducing filter (Maximal sum of abs differences)
|
2002-11-02 13:58:14 +00:00
|
|
|
|
|
|
|
int baseDcDiff;
|
|
|
|
int flatnessThreshold;
|
|
|
|
|
2003-03-06 13:51:18 +00:00
|
|
|
int forcedQuant; ///< quantizer if FORCE_QUANT is used
|
2002-11-02 13:58:14 +00:00
|
|
|
} PPMode;
|
|
|
|
|
2003-03-06 13:51:18 +00:00
|
|
|
/**
|
|
|
|
* postprocess context.
|
|
|
|
*/
|
2002-11-02 13:58:14 +00:00
|
|
|
typedef struct PPContext{
|
2003-03-06 13:51:18 +00:00
|
|
|
uint8_t *tempBlocks; ///<used for the horizontal code
|
2002-11-02 13:58:14 +00:00
|
|
|
|
2003-03-06 13:51:18 +00:00
|
|
|
/**
|
|
|
|
* luma histogram.
|
|
|
|
* we need 64bit here otherwise we'll going to have a problem
|
|
|
|
* after watching a black picture for 5 hours
|
|
|
|
*/
|
2002-11-02 13:58:14 +00:00
|
|
|
uint64_t *yHistogram;
|
|
|
|
|
|
|
|
uint64_t __attribute__((aligned(8))) packedYOffset;
|
|
|
|
uint64_t __attribute__((aligned(8))) packedYScale;
|
|
|
|
|
2003-03-06 13:51:18 +00:00
|
|
|
/** Temporal noise reducing buffers */
|
2002-11-02 13:58:14 +00:00
|
|
|
uint8_t *tempBlured[3];
|
|
|
|
int32_t *tempBluredPast[3];
|
|
|
|
|
2003-03-06 13:51:18 +00:00
|
|
|
/** Temporary buffers for handling the last row(s) */
|
2002-11-02 13:58:14 +00:00
|
|
|
uint8_t *tempDst;
|
|
|
|
uint8_t *tempSrc;
|
|
|
|
|
|
|
|
uint8_t *deintTemp;
|
|
|
|
|
|
|
|
uint64_t __attribute__((aligned(8))) pQPb;
|
|
|
|
uint64_t __attribute__((aligned(8))) pQPb2;
|
|
|
|
|
2004-01-04 17:29:51 +00:00
|
|
|
uint64_t __attribute__((aligned(8))) mmxDcOffset[64];
|
|
|
|
uint64_t __attribute__((aligned(8))) mmxDcThreshold[64];
|
2002-12-29 00:57:23 +00:00
|
|
|
|
2003-04-18 12:45:34 +00:00
|
|
|
QP_STORE_T *stdQPTable; ///< used to fix MPEG2 style qscale
|
2002-11-02 13:58:14 +00:00
|
|
|
QP_STORE_T *nonBQPTable;
|
2002-12-29 00:57:23 +00:00
|
|
|
QP_STORE_T *forcedQPTable;
|
|
|
|
|
2002-11-02 13:58:14 +00:00
|
|
|
int QP;
|
|
|
|
int nonBQP;
|
|
|
|
|
|
|
|
int frameNum;
|
|
|
|
|
|
|
|
int cpuCaps;
|
2002-12-29 00:57:23 +00:00
|
|
|
|
2003-04-18 12:45:34 +00:00
|
|
|
int qpStride; ///<size of qp buffers (needed to realloc them if needed)
|
|
|
|
int stride; ///<size of some buffers (needed to realloc them if needed)
|
2003-01-05 19:10:42 +00:00
|
|
|
|
|
|
|
int hChromaSubSample;
|
|
|
|
int vChromaSubSample;
|
2002-11-02 13:58:14 +00:00
|
|
|
|
|
|
|
PPMode ppMode;
|
|
|
|
} PPContext;
|
|
|
|
|
2003-04-18 12:45:34 +00:00
|
|
|
|