Statistics
| Branch: | Revision:

ffmpeg / libpostproc / postprocess_internal.h @ 16e0bf73

History | View | Annotate | Download (5.56 KB)

1 c41d972d Michael Niedermayer
/*
2 b78e7197 Diego Biurrun
 * 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 c41d972d Michael Niedermayer
21 b304569a Michael Niedermayer
/**
22
 * @file postprocess_internal.h
23
 * internal api header.
24
 */
25
26 5b21bdab Diego Biurrun
#ifndef FFMPEG_POSTPROCESS_INTERNAL_H
27
#define FFMPEG_POSTPROCESS_INTERNAL_H
28 699b3f99 Måns Rullgård
29 6ab6c7c3 Luca Barbato
#include "avutil.h"
30 53eecfde Diego Biurrun
#include "postprocess.h"
31 6ab6c7c3 Luca Barbato
32 bb270c08 Diego Biurrun
#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 c41d972d Michael Niedermayer
46
// Experimental vertical filters
47 bb270c08 Diego Biurrun
#define V_X1_FILTER     0x0200                  // 512
48
#define V_A_DEBLOCK     0x0400
49 c41d972d Michael Niedermayer
50
// Experimental horizontal filters
51 bb270c08 Diego Biurrun
#define H_X1_FILTER     0x2000                  // 8192
52
#define H_A_DEBLOCK     0x4000
53 c41d972d Michael Niedermayer
54 b304569a Michael Niedermayer
/// select between full y range (255-0) or standart one (234-16)
55 bb270c08 Diego Biurrun
#define FULL_Y_RANGE    0x8000                  // 32768
56 c41d972d Michael Niedermayer
57
//Deinterlacing Filters
58 16e0bf73 Diego Biurrun
#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 c41d972d Michael Niedermayer
66 bb270c08 Diego Biurrun
#define TEMP_NOISE_FILTER               0x100000
67
#define FORCE_QUANT                     0x200000
68 c41d972d Michael Niedermayer
69 755bfeab Diego Biurrun
//use if you want a faster postprocessing code
70
//cannot differentiate between chroma & luma filters (both on or both off)
71
//obviously the -pp option on the command line has no effect except turning the here selected
72 c41d972d Michael Niedermayer
//filters on
73
//#define COMPILE_TIME_MODE 0x77
74
75 134eb1e5 Michael Niedermayer
static inline int CLIP(int a){
76 16e0bf73 Diego Biurrun
    if(a&256) return ((a)>>31)^(-1);
77
    else      return a;
78 134eb1e5 Michael Niedermayer
}
79 b304569a Michael Niedermayer
/**
80
 * Postprocessng filter.
81
 */
82 c41d972d Michael Niedermayer
struct PPFilter{
83 16e0bf73 Diego Biurrun
    const char *shortName;
84
    const char *longName;
85
    int chromDefault;       ///< is chrominance filtering on by default if this filter is manually activated
86
    int minLumQuality;      ///< minimum quality to turn luminance filtering on
87
    int minChromQuality;    ///< minimum quality to turn chrominance filtering on
88
    int mask;               ///< Bitmask to turn this filter on
89 c41d972d Michael Niedermayer
};
90
91 b304569a Michael Niedermayer
/**
92
 * Postprocessng mode.
93
 */
94 c41d972d Michael Niedermayer
typedef struct PPMode{
95 16e0bf73 Diego Biurrun
    int lumMode;                    ///< acivates filters for luminance
96
    int chromMode;                  ///< acivates filters for chrominance
97
    int error;                      ///< non zero on error
98 c41d972d Michael Niedermayer
99 16e0bf73 Diego Biurrun
    int minAllowedY;                ///< for brigtness correction
100
    int maxAllowedY;                ///< for brihtness correction
101
    float maxClippedThreshold;      ///< amount of "black" u r willing to loose to get a brightness corrected picture
102 c41d972d Michael Niedermayer
103 16e0bf73 Diego Biurrun
    int maxTmpNoise[3];             ///< for Temporal Noise Reducing filter (Maximal sum of abs differences)
104 c41d972d Michael Niedermayer
105 16e0bf73 Diego Biurrun
    int baseDcDiff;
106
    int flatnessThreshold;
107 c41d972d Michael Niedermayer
108 16e0bf73 Diego Biurrun
    int forcedQuant;                ///< quantizer if FORCE_QUANT is used
109 c41d972d Michael Niedermayer
} PPMode;
110
111 b304569a Michael Niedermayer
/**
112
 * postprocess context.
113
 */
114 c41d972d Michael Niedermayer
typedef struct PPContext{
115 16e0bf73 Diego Biurrun
    /**
116
     * info on struct for av_log
117
     */
118
    const AVClass *av_class;
119 e7becfb2 Diego Biurrun
120 16e0bf73 Diego Biurrun
    uint8_t *tempBlocks; ///<used for the horizontal code
121 c41d972d Michael Niedermayer
122 16e0bf73 Diego Biurrun
    /**
123
     * luma histogram.
124
     * we need 64bit here otherwise we'll going to have a problem
125
     * after watching a black picture for 5 hours
126
     */
127
    uint64_t *yHistogram;
128 c41d972d Michael Niedermayer
129 16e0bf73 Diego Biurrun
    DECLARE_ALIGNED(8, uint64_t, packedYOffset);
130
    DECLARE_ALIGNED(8, uint64_t, packedYScale);
131 c41d972d Michael Niedermayer
132 16e0bf73 Diego Biurrun
    /** Temporal noise reducing buffers */
133
    uint8_t *tempBlured[3];
134
    int32_t *tempBluredPast[3];
135 c41d972d Michael Niedermayer
136 16e0bf73 Diego Biurrun
    /** Temporary buffers for handling the last row(s) */
137
    uint8_t *tempDst;
138
    uint8_t *tempSrc;
139 c41d972d Michael Niedermayer
140 16e0bf73 Diego Biurrun
    uint8_t *deintTemp;
141 c41d972d Michael Niedermayer
142 16e0bf73 Diego Biurrun
    DECLARE_ALIGNED(8, uint64_t, pQPb);
143
    DECLARE_ALIGNED(8, uint64_t, pQPb2);
144 c41d972d Michael Niedermayer
145 16e0bf73 Diego Biurrun
    DECLARE_ALIGNED(8, uint64_t, mmxDcOffset[64]);
146
    DECLARE_ALIGNED(8, uint64_t, mmxDcThreshold[64]);
147 88c0bc7e Michael Niedermayer
148 16e0bf73 Diego Biurrun
    QP_STORE_T *stdQPTable;       ///< used to fix MPEG2 style qscale
149
    QP_STORE_T *nonBQPTable;
150
    QP_STORE_T *forcedQPTable;
151 88c0bc7e Michael Niedermayer
152 16e0bf73 Diego Biurrun
    int QP;
153
    int nonBQP;
154 c41d972d Michael Niedermayer
155 16e0bf73 Diego Biurrun
    int frameNum;
156 115329f1 Diego Biurrun
157 16e0bf73 Diego Biurrun
    int cpuCaps;
158 115329f1 Diego Biurrun
159 16e0bf73 Diego Biurrun
    int qpStride; ///<size of qp buffers (needed to realloc them if needed)
160
    int stride;   ///<size of some buffers (needed to realloc them if needed)
161 115329f1 Diego Biurrun
162 16e0bf73 Diego Biurrun
    int hChromaSubSample;
163
    int vChromaSubSample;
164 c41d972d Michael Niedermayer
165 16e0bf73 Diego Biurrun
    PPMode ppMode;
166 c41d972d Michael Niedermayer
} PPContext;
167
168 0426af31 Michael Niedermayer
169 16e0bf73 Diego Biurrun
static inline void linecpy(void *dest, const void *src, int lines, int stride) {
170
    if (stride > 0) {
171
        memcpy(dest, src, lines*stride);
172
    } else {
173
        memcpy((uint8_t*)dest+(lines-1)*stride, (const uint8_t*)src+(lines-1)*stride, -lines*stride);
174
    }
175 655f688d Jindřich Makovička
}
176 699b3f99 Måns Rullgård
177 5b21bdab Diego Biurrun
#endif /* FFMPEG_POSTPROCESS_INTERNAL_H */