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