postprocess_internal.h 5.6 KB

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