postprocess_internal.h 5.6 KB

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