33768ba79befae56c1d9676b28e5e70c5dab0dc6
[libav.git] / 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 FFMPEG_POSTPROCESS_INTERNAL_H
27 #define FFMPEG_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 static inline int CLIP(int a){
80 if(a&256) return ((a)>>31)^(-1);
81 else return a;
82 }
83 /**
84 * Postprocessng filter.
85 */
86 struct PPFilter{
87 const char *shortName;
88 const char *longName;
89 int chromDefault; ///< is chrominance filtering on by default if this filter is manually activated
90 int minLumQuality; ///< minimum quality to turn luminance filtering on
91 int minChromQuality; ///< minimum quality to turn chrominance filtering on
92 int mask; ///< Bitmask to turn this filter on
93 };
94
95 /**
96 * Postprocessng mode.
97 */
98 typedef struct PPMode{
99 int lumMode; ///< acivates filters for luminance
100 int chromMode; ///< acivates filters for chrominance
101 int error; ///< non zero on error
102
103 int minAllowedY; ///< for brigtness correction
104 int maxAllowedY; ///< for brihtness correction
105 float maxClippedThreshold; ///< amount of "black" u r willing to loose to get a brightness corrected picture
106
107 int maxTmpNoise[3]; ///< for Temporal Noise Reducing filter (Maximal sum of abs differences)
108
109 int baseDcDiff;
110 int flatnessThreshold;
111
112 int forcedQuant; ///< quantizer if FORCE_QUANT is used
113 } PPMode;
114
115 /**
116 * postprocess context.
117 */
118 typedef struct PPContext{
119 /**
120 * info on struct for av_log
121 */
122 const AVClass *av_class;
123
124 uint8_t *tempBlocks; ///<used for the horizontal code
125
126 /**
127 * luma histogram.
128 * we need 64bit here otherwise we'll going to have a problem
129 * after watching a black picture for 5 hours
130 */
131 uint64_t *yHistogram;
132
133 DECLARE_ALIGNED(8, uint64_t, packedYOffset);
134 DECLARE_ALIGNED(8, uint64_t, packedYScale);
135
136 /** Temporal noise reducing buffers */
137 uint8_t *tempBlured[3];
138 int32_t *tempBluredPast[3];
139
140 /** Temporary buffers for handling the last row(s) */
141 uint8_t *tempDst;
142 uint8_t *tempSrc;
143
144 uint8_t *deintTemp;
145
146 DECLARE_ALIGNED(8, uint64_t, pQPb);
147 DECLARE_ALIGNED(8, uint64_t, pQPb2);
148
149 DECLARE_ALIGNED(8, uint64_t, mmxDcOffset[64]);
150 DECLARE_ALIGNED(8, uint64_t, mmxDcThreshold[64]);
151
152 QP_STORE_T *stdQPTable; ///< used to fix MPEG2 style qscale
153 QP_STORE_T *nonBQPTable;
154 QP_STORE_T *forcedQPTable;
155
156 int QP;
157 int nonBQP;
158
159 int frameNum;
160
161 int cpuCaps;
162
163 int qpStride; ///<size of qp buffers (needed to realloc them if needed)
164 int stride; ///<size of some buffers (needed to realloc them if needed)
165
166 int hChromaSubSample;
167 int vChromaSubSample;
168
169 PPMode ppMode;
170 } PPContext;
171
172
173 static inline void linecpy(void *dest, void *src, int lines, int stride)
174 {
175 if (stride > 0) {
176 memcpy(dest, src, lines*stride);
177 } else {
178 memcpy(dest+(lines-1)*stride, src+(lines-1)*stride, -lines*stride);
179 }
180 }
181
182 #endif /* FFMPEG_POSTPROCESS_INTERNAL_H */