Commit | Line | Data |
---|---|---|
72b9787e JB |
1 | /***************************************************************************** |
2 | * Copyright (C) 2014 x265 project | |
3 | * | |
4 | * Authors: Steve Borho <steve@borho.org> | |
5 | * | |
6 | * This program 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 | * This program 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 this program; if not, write to the Free Software | |
18 | * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02111, USA. | |
19 | * | |
20 | * This program is also available under a commercial proprietary license. | |
21 | * For more information, contact us at license @ x265.com. | |
22 | *****************************************************************************/ | |
23 | ||
24 | #ifndef X265_QUANT_H | |
25 | #define X265_QUANT_H | |
26 | ||
27 | #include "common.h" | |
28 | #include "scalinglist.h" | |
29 | #include "contexts.h" | |
30 | ||
31 | namespace x265 { | |
32 | // private namespace | |
33 | ||
34 | class CUData; | |
35 | class Entropy; | |
36 | struct TUEntropyCodingParameters; | |
37 | ||
38 | struct QpParam | |
39 | { | |
40 | int rem; | |
41 | int per; | |
42 | int qp; | |
43 | int64_t lambda2; /* FIX8 */ | |
44 | int64_t lambda; /* FIX8 */ | |
45 | ||
46 | QpParam() : qp(MAX_INT) {} | |
47 | ||
48 | void setQpParam(int qpScaled) | |
49 | { | |
50 | if (qp != qpScaled) | |
51 | { | |
52 | rem = qpScaled % 6; | |
53 | per = qpScaled / 6; | |
54 | qp = qpScaled; | |
55 | lambda2 = (int64_t)(x265_lambda2_tab[qp - QP_BD_OFFSET] * 256. + 0.5); | |
56 | lambda = (int64_t)(x265_lambda_tab[qp - QP_BD_OFFSET] * 256. + 0.5); | |
57 | } | |
58 | } | |
59 | }; | |
60 | ||
b53f7c52 JB |
61 | #define MAX_NUM_TR_COEFFS MAX_TR_SIZE * MAX_TR_SIZE /* Maximum number of transform coefficients, for a 32x32 transform */ |
62 | #define MAX_NUM_TR_CATEGORIES 16 /* 32, 16, 8, 4 transform categories each for luma and chroma */ | |
63 | ||
64 | // NOTE: MUST be 16-byte aligned for asm code | |
65 | struct NoiseReduction | |
66 | { | |
67 | /* 0 = luma 4x4, 1 = luma 8x8, 2 = luma 16x16, 3 = luma 32x32 | |
68 | * 4 = chroma 4x4, 5 = chroma 8x8, 6 = chroma 16x16, 7 = chroma 32x32 | |
69 | * Intra 0..7 - Inter 8..15 */ | |
70 | uint16_t offsetDenoise[MAX_NUM_TR_CATEGORIES][MAX_NUM_TR_COEFFS]; | |
71 | uint32_t residualSum[MAX_NUM_TR_CATEGORIES][MAX_NUM_TR_COEFFS]; | |
72 | uint32_t count[MAX_NUM_TR_CATEGORIES]; | |
73 | }; | |
74 | ||
72b9787e JB |
75 | class Quant |
76 | { | |
77 | protected: | |
78 | ||
79 | const ScalingList* m_scalingList; | |
80 | Entropy* m_entropyCoder; | |
81 | ||
82 | QpParam m_qpParam[3]; | |
83 | ||
84 | bool m_useRDOQ; | |
85 | int64_t m_psyRdoqScale; | |
b53f7c52 JB |
86 | int16_t* m_resiDctCoeff; |
87 | int16_t* m_fencDctCoeff; | |
72b9787e JB |
88 | int16_t* m_fencShortBuf; |
89 | ||
90 | enum { IEP_RATE = 32768 }; /* FIX15 cost of an equal probable bit */ | |
91 | ||
92 | public: | |
93 | ||
94 | NoiseReduction* m_nr; | |
95 | NoiseReduction* m_frameNr; // Array of NR structures, one for each frameEncoder | |
96 | ||
97 | Quant(); | |
98 | ~Quant(); | |
99 | ||
100 | /* one-time setup */ | |
101 | bool init(bool useRDOQ, double psyScale, const ScalingList& scalingList, Entropy& entropy); | |
102 | bool allocNoiseReduction(const x265_param& param); | |
103 | ||
104 | /* CU setup */ | |
105 | void setQPforQuant(const CUData& ctu); | |
106 | ||
b53f7c52 | 107 | uint32_t transformNxN(const CUData& cu, const pixel* fenc, uint32_t fencStride, const int16_t* residual, uint32_t resiStride, coeff_t* coeff, |
72b9787e JB |
108 | uint32_t log2TrSize, TextType ttype, uint32_t absPartIdx, bool useTransformSkip); |
109 | ||
b53f7c52 | 110 | void invtransformNxN(bool transQuantBypass, int16_t* residual, uint32_t resiStride, const coeff_t* coeff, |
72b9787e JB |
111 | uint32_t log2TrSize, TextType ttype, bool bIntra, bool useTransformSkip, uint32_t numSig); |
112 | ||
113 | /* static methods shared with entropy.cpp */ | |
114 | static uint32_t calcPatternSigCtx(uint64_t sigCoeffGroupFlag64, uint32_t cgPosX, uint32_t cgPosY, uint32_t log2TrSizeCG); | |
115 | static uint32_t getSigCtxInc(uint32_t patternSigCtx, uint32_t log2TrSize, uint32_t trSize, uint32_t blkPos, bool bIsLuma, uint32_t firstSignificanceMapContext); | |
116 | static uint32_t getSigCoeffGroupCtxInc(uint64_t sigCoeffGroupFlag64, uint32_t cgPosX, uint32_t cgPosY, uint32_t log2TrSizeCG); | |
117 | ||
118 | protected: | |
119 | ||
120 | void setChromaQP(int qpin, TextType ttype, int chFmt); | |
121 | ||
122 | uint32_t signBitHidingHDQ(int16_t* qcoeff, int32_t* deltaU, uint32_t numSig, const TUEntropyCodingParameters &codingParameters); | |
123 | ||
b53f7c52 | 124 | uint32_t rdoQuant(const CUData& cu, int16_t* dstCoeff, uint32_t log2TrSize, TextType ttype, uint32_t absPartIdx, bool usePsy); |
72b9787e JB |
125 | inline uint32_t getRateLast(uint32_t posx, uint32_t posy) const; |
126 | }; | |
127 | ||
128 | static inline uint32_t getGroupIdx(const uint32_t idx) | |
129 | { | |
130 | // TODO: Why is this not a table lookup? | |
131 | ||
132 | uint32_t group = (idx >> 3); | |
133 | ||
134 | if (idx >= 24) | |
135 | group = 2; | |
136 | uint32_t groupIdx = ((idx >> (group + 1)) - 2) + 4 + (group << 1); | |
137 | if (idx <= 3) | |
138 | groupIdx = idx; | |
139 | ||
140 | #ifdef _DEBUG | |
141 | static const uint8_t g_groupIdx[32] = { 0, 1, 2, 3, 4, 4, 5, 5, 6, 6, 6, 6, 7, 7, 7, 7, 8, 8, 8, 8, 8, 8, 8, 8, 9, 9, 9, 9, 9, 9, 9, 9 }; | |
142 | assert(groupIdx == g_groupIdx[idx]); | |
143 | #endif | |
144 | ||
145 | return groupIdx; | |
146 | } | |
147 | ||
148 | } | |
149 | ||
150 | #endif // ifndef X265_QUANT_H |