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