Commit | Line | Data |
---|---|---|
2ba45a60 DM |
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 | |
23 | * internal API header. | |
24 | */ | |
25 | ||
26 | #ifndef POSTPROC_POSTPROCESS_INTERNAL_H | |
27 | #define POSTPROC_POSTPROCESS_INTERNAL_H | |
28 | ||
29 | #include <string.h> | |
30 | #include "libavutil/avutil.h" | |
31 | #include "libavutil/intmath.h" | |
32 | #include "libavutil/log.h" | |
33 | #include "postprocess.h" | |
34 | ||
35 | #define V_DEBLOCK 0x01 | |
36 | #define H_DEBLOCK 0x02 | |
37 | #define DERING 0x04 | |
38 | #define LEVEL_FIX 0x08 ///< Brightness & Contrast | |
39 | ||
40 | #define LUM_V_DEBLOCK V_DEBLOCK // 1 | |
41 | #define LUM_H_DEBLOCK H_DEBLOCK // 2 | |
42 | #define CHROM_V_DEBLOCK (V_DEBLOCK<<4) // 16 | |
43 | #define CHROM_H_DEBLOCK (H_DEBLOCK<<4) // 32 | |
44 | #define LUM_DERING DERING // 4 | |
45 | #define CHROM_DERING (DERING<<4) // 64 | |
46 | #define LUM_LEVEL_FIX LEVEL_FIX // 8 | |
47 | #define CHROM_LEVEL_FIX (LEVEL_FIX<<4) // 128 (not implemented yet) | |
48 | ||
49 | // Experimental vertical filters | |
50 | #define V_X1_FILTER 0x0200 // 512 | |
51 | #define V_A_DEBLOCK 0x0400 | |
52 | ||
53 | // Experimental horizontal filters | |
54 | #define H_X1_FILTER 0x2000 // 8192 | |
55 | #define H_A_DEBLOCK 0x4000 | |
56 | ||
57 | /// select between full y range (255-0) or standard one (234-16) | |
58 | #define FULL_Y_RANGE 0x8000 // 32768 | |
59 | ||
60 | //Deinterlacing Filters | |
61 | #define LINEAR_IPOL_DEINT_FILTER 0x10000 // 65536 | |
62 | #define LINEAR_BLEND_DEINT_FILTER 0x20000 // 131072 | |
63 | #define CUBIC_BLEND_DEINT_FILTER 0x8000 // (not implemented yet) | |
64 | #define CUBIC_IPOL_DEINT_FILTER 0x40000 // 262144 | |
65 | #define MEDIAN_DEINT_FILTER 0x80000 // 524288 | |
66 | #define FFMPEG_DEINT_FILTER 0x400000 | |
67 | #define LOWPASS5_DEINT_FILTER 0x800000 | |
68 | ||
69 | #define TEMP_NOISE_FILTER 0x100000 | |
70 | #define FORCE_QUANT 0x200000 | |
71 | #define BITEXACT 0x1000000 | |
f6fa7814 | 72 | #define VISUALIZE 0x2000000 |
2ba45a60 DM |
73 | |
74 | //use if you want a faster postprocessing code | |
75 | //cannot differentiate between chroma & luma filters (both on or both off) | |
76 | //obviously the -pp option on the command line has no effect except turning the here selected | |
77 | //filters on | |
78 | //#define COMPILE_TIME_MODE 0x77 | |
79 | ||
2ba45a60 DM |
80 | /** |
81 | * Postprocessing filter. | |
82 | */ | |
83 | struct PPFilter{ | |
84 | const char *shortName; | |
85 | const char *longName; | |
86 | int chromDefault; ///< is chrominance filtering on by default if this filter is manually activated | |
87 | int minLumQuality; ///< minimum quality to turn luminance filtering on | |
88 | int minChromQuality; ///< minimum quality to turn chrominance filtering on | |
89 | int mask; ///< Bitmask to turn this filter on | |
90 | }; | |
91 | ||
92 | /** | |
93 | * Postprocessing mode. | |
94 | */ | |
95 | typedef struct PPMode{ | |
96 | int lumMode; ///< activates filters for luminance | |
97 | int chromMode; ///< activates filters for chrominance | |
98 | int error; ///< non zero on error | |
99 | ||
100 | int minAllowedY; ///< for brightness correction | |
101 | int maxAllowedY; ///< for brightness correction | |
102 | float maxClippedThreshold; ///< amount of "black" you are willing to lose to get a brightness-corrected picture | |
103 | ||
104 | int maxTmpNoise[3]; ///< for Temporal Noise Reducing filter (Maximal sum of abs differences) | |
105 | ||
106 | int baseDcDiff; | |
107 | int flatnessThreshold; | |
108 | ||
109 | int forcedQuant; ///< quantizer if FORCE_QUANT is used | |
110 | } PPMode; | |
111 | ||
112 | /** | |
113 | * postprocess context. | |
114 | */ | |
115 | typedef struct PPContext{ | |
116 | /** | |
117 | * info on struct for av_log | |
118 | */ | |
119 | const AVClass *av_class; | |
120 | ||
121 | uint8_t *tempBlocks; ///<used for the horizontal code | |
122 | ||
123 | /** | |
124 | * luma histogram. | |
125 | * we need 64bit here otherwise we'll going to have a problem | |
126 | * after watching a black picture for 5 hours | |
127 | */ | |
128 | uint64_t *yHistogram; | |
129 | ||
130 | DECLARE_ALIGNED(8, uint64_t, packedYOffset); | |
131 | DECLARE_ALIGNED(8, uint64_t, packedYScale); | |
132 | ||
133 | /** Temporal noise reducing buffers */ | |
134 | uint8_t *tempBlurred[3]; | |
135 | int32_t *tempBlurredPast[3]; | |
136 | ||
137 | /** Temporary buffers for handling the last row(s) */ | |
138 | uint8_t *tempDst; | |
139 | uint8_t *tempSrc; | |
140 | ||
141 | uint8_t *deintTemp; | |
142 | ||
143 | DECLARE_ALIGNED(8, uint64_t, pQPb); | |
144 | DECLARE_ALIGNED(8, uint64_t, pQPb2); | |
145 | ||
146 | DECLARE_ALIGNED(8, uint64_t, mmxDcOffset)[64]; | |
147 | DECLARE_ALIGNED(8, uint64_t, mmxDcThreshold)[64]; | |
148 | ||
149 | QP_STORE_T *stdQPTable; ///< used to fix MPEG2 style qscale | |
150 | QP_STORE_T *nonBQPTable; | |
151 | QP_STORE_T *forcedQPTable; | |
152 | ||
153 | int QP; | |
154 | int nonBQP; | |
155 | ||
156 | int frameNum; | |
157 | ||
158 | int cpuCaps; | |
159 | ||
160 | int qpStride; ///<size of qp buffers (needed to realloc them if needed) | |
161 | int stride; ///<size of some buffers (needed to realloc them if needed) | |
162 | ||
163 | int hChromaSubSample; | |
164 | int vChromaSubSample; | |
165 | ||
166 | PPMode ppMode; | |
167 | } PPContext; | |
168 | ||
169 | ||
170 | static inline void linecpy(void *dest, const void *src, int lines, int stride) { | |
171 | if (stride > 0) { | |
172 | memcpy(dest, src, lines*stride); | |
173 | } else { | |
174 | memcpy((uint8_t*)dest+(lines-1)*stride, (const uint8_t*)src+(lines-1)*stride, -lines*stride); | |
175 | } | |
176 | } | |
177 | ||
178 | #endif /* POSTPROC_POSTPROCESS_INTERNAL_H */ |