postprocess_internal.h 5.63 KB
Newer Older
Michael Niedermayer's avatar
Michael Niedermayer committed
1
/*
2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19
 * Copyright (C) 2001-2002 Michael Niedermayer (michaelni@gmx.at)
 *
 * This file is part of FFmpeg.
 *
 * FFmpeg 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.
 *
 * FFmpeg 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 FFmpeg; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 */
Michael Niedermayer's avatar
Michael Niedermayer committed
20

Michael Niedermayer's avatar
Michael Niedermayer committed
21
/**
22
 * @file
23
 * internal API header.
Michael Niedermayer's avatar
Michael Niedermayer committed
24 25
 */

26 27
#ifndef POSTPROC_POSTPROCESS_INTERNAL_H
#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"
Diego Biurrun's avatar
Diego Biurrun committed
33
#include "postprocess.h"
Luca Barbato's avatar
Luca Barbato committed
34

35 36 37 38 39 40 41 42 43 44 45 46 47
#define V_DEBLOCK       0x01
#define H_DEBLOCK       0x02
#define DERING          0x04
#define LEVEL_FIX       0x08 ///< Brightness & Contrast

#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)
Michael Niedermayer's avatar
Michael Niedermayer committed
48 49

// Experimental vertical filters
50 51
#define V_X1_FILTER     0x0200                  // 512
#define V_A_DEBLOCK     0x0400
Michael Niedermayer's avatar
Michael Niedermayer committed
52 53

// Experimental horizontal filters
54 55
#define H_X1_FILTER     0x2000                  // 8192
#define H_A_DEBLOCK     0x4000
Michael Niedermayer's avatar
Michael Niedermayer committed
56

57
/// select between full y range (255-0) or standard one (234-16)
58
#define FULL_Y_RANGE    0x8000                  // 32768
Michael Niedermayer's avatar
Michael Niedermayer committed
59 60

//Deinterlacing Filters
61 62 63 64 65 66 67
#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
#define LOWPASS5_DEINT_FILTER           0x800000
Michael Niedermayer's avatar
Michael Niedermayer committed
68

69 70
#define TEMP_NOISE_FILTER               0x100000
#define FORCE_QUANT                     0x200000
71
#define BITEXACT                        0x1000000
Michael Niedermayer's avatar
Michael Niedermayer committed
72

Diego Biurrun's avatar
Diego Biurrun committed
73 74 75
//use if you want a faster postprocessing code
//cannot differentiate between chroma & luma filters (both on or both off)
//obviously the -pp option on the command line has no effect except turning the here selected
Michael Niedermayer's avatar
Michael Niedermayer committed
76 77 78
//filters on
//#define COMPILE_TIME_MODE 0x77

79 80
#define CLIP av_clip_uint8

Michael Niedermayer's avatar
Michael Niedermayer committed
81
/**
82
 * Postprocessing filter.
Michael Niedermayer's avatar
Michael Niedermayer committed
83
 */
Michael Niedermayer's avatar
Michael Niedermayer committed
84
struct PPFilter{
85 86 87 88 89 90
    const char *shortName;
    const char *longName;
    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
Michael Niedermayer's avatar
Michael Niedermayer committed
91 92
};

Michael Niedermayer's avatar
Michael Niedermayer committed
93
/**
94
 * Postprocessing mode.
Michael Niedermayer's avatar
Michael Niedermayer committed
95
 */
Michael Niedermayer's avatar
Michael Niedermayer committed
96
typedef struct PPMode{
97 98
    int lumMode;                    ///< activates filters for luminance
    int chromMode;                  ///< activates filters for chrominance
99
    int error;                      ///< non zero on error
Michael Niedermayer's avatar
Michael Niedermayer committed
100

101 102
    int minAllowedY;                ///< for brightness correction
    int maxAllowedY;                ///< for brightness correction
103
    float maxClippedThreshold;      ///< amount of "black" you are willing to lose to get a brightness-corrected picture
Michael Niedermayer's avatar
Michael Niedermayer committed
104

105
    int maxTmpNoise[3];             ///< for Temporal Noise Reducing filter (Maximal sum of abs differences)
Michael Niedermayer's avatar
Michael Niedermayer committed
106

107 108
    int baseDcDiff;
    int flatnessThreshold;
Michael Niedermayer's avatar
Michael Niedermayer committed
109

110
    int forcedQuant;                ///< quantizer if FORCE_QUANT is used
Michael Niedermayer's avatar
Michael Niedermayer committed
111 112
} PPMode;

Michael Niedermayer's avatar
Michael Niedermayer committed
113 114 115
/**
 * postprocess context.
 */
Michael Niedermayer's avatar
Michael Niedermayer committed
116
typedef struct PPContext{
117 118 119 120
    /**
     * info on struct for av_log
     */
    const AVClass *av_class;
121

122
    uint8_t *tempBlocks; ///<used for the horizontal code
Michael Niedermayer's avatar
Michael Niedermayer committed
123

124 125 126 127 128 129
    /**
     * luma histogram.
     * we need 64bit here otherwise we'll going to have a problem
     * after watching a black picture for 5 hours
     */
    uint64_t *yHistogram;
Michael Niedermayer's avatar
Michael Niedermayer committed
130

131 132
    DECLARE_ALIGNED(8, uint64_t, packedYOffset);
    DECLARE_ALIGNED(8, uint64_t, packedYScale);
Michael Niedermayer's avatar
Michael Niedermayer committed
133

134
    /** Temporal noise reducing buffers */
135 136
    uint8_t *tempBlurred[3];
    int32_t *tempBlurredPast[3];
Michael Niedermayer's avatar
Michael Niedermayer committed
137

138 139 140
    /** Temporary buffers for handling the last row(s) */
    uint8_t *tempDst;
    uint8_t *tempSrc;
Michael Niedermayer's avatar
Michael Niedermayer committed
141

142
    uint8_t *deintTemp;
Michael Niedermayer's avatar
Michael Niedermayer committed
143

144 145
    DECLARE_ALIGNED(8, uint64_t, pQPb);
    DECLARE_ALIGNED(8, uint64_t, pQPb2);
Michael Niedermayer's avatar
Michael Niedermayer committed
146

147 148
    DECLARE_ALIGNED(8, uint64_t, mmxDcOffset)[64];
    DECLARE_ALIGNED(8, uint64_t, mmxDcThreshold)[64];
149

150 151 152
    QP_STORE_T *stdQPTable;       ///< used to fix MPEG2 style qscale
    QP_STORE_T *nonBQPTable;
    QP_STORE_T *forcedQPTable;
153

154 155
    int QP;
    int nonBQP;
Michael Niedermayer's avatar
Michael Niedermayer committed
156

157
    int frameNum;
158

159
    int cpuCaps;
160

161 162
    int qpStride; ///<size of qp buffers (needed to realloc them if needed)
    int stride;   ///<size of some buffers (needed to realloc them if needed)
163

164 165
    int hChromaSubSample;
    int vChromaSubSample;
Michael Niedermayer's avatar
Michael Niedermayer committed
166

167
    PPMode ppMode;
Michael Niedermayer's avatar
Michael Niedermayer committed
168 169
} PPContext;

Michael Niedermayer's avatar
Michael Niedermayer committed
170

171 172 173 174 175 176
static inline void linecpy(void *dest, const void *src, int lines, int stride) {
    if (stride > 0) {
        memcpy(dest, src, lines*stride);
    } else {
        memcpy((uint8_t*)dest+(lines-1)*stride, (const uint8_t*)src+(lines-1)*stride, -lines*stride);
    }
177
}
178

179
#endif /* POSTPROC_POSTPROCESS_INTERNAL_H */