postprocess_internal.h 5.7 KB

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