Mercurial > hg > forks > libbpg
comparison x265/source/common/lowres.h @ 0:772086c29cc7
Initial import.
author | Matti Hamalainen <ccr@tnsp.org> |
---|---|
date | Wed, 16 Nov 2016 11:16:33 +0200 |
parents | |
children |
comparison
equal
deleted
inserted
replaced
-1:000000000000 | 0:772086c29cc7 |
---|---|
1 /***************************************************************************** | |
2 * Copyright (C) 2013 x265 project | |
3 * | |
4 * Authors: Gopu Govindaswamy <gopu@multicorewareinc.com> | |
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_LOWRES_H | |
25 #define X265_LOWRES_H | |
26 | |
27 #include "primitives.h" | |
28 #include "common.h" | |
29 #include "picyuv.h" | |
30 #include "mv.h" | |
31 | |
32 namespace X265_NS { | |
33 // private namespace | |
34 | |
35 struct ReferencePlanes | |
36 { | |
37 ReferencePlanes() { memset(this, 0, sizeof(ReferencePlanes)); } | |
38 | |
39 pixel* fpelPlane[3]; | |
40 pixel* lowresPlane[4]; | |
41 PicYuv* reconPic; | |
42 | |
43 bool isWeighted; | |
44 bool isLowres; | |
45 | |
46 intptr_t lumaStride; | |
47 intptr_t chromaStride; | |
48 | |
49 struct { | |
50 int weight; | |
51 int offset; | |
52 int shift; | |
53 int round; | |
54 } w[3]; | |
55 | |
56 pixel* getLumaAddr(uint32_t ctuAddr, uint32_t absPartIdx) { return fpelPlane[0] + reconPic->m_cuOffsetY[ctuAddr] + reconPic->m_buOffsetY[absPartIdx]; } | |
57 pixel* getCbAddr(uint32_t ctuAddr, uint32_t absPartIdx) { return fpelPlane[1] + reconPic->m_cuOffsetC[ctuAddr] + reconPic->m_buOffsetC[absPartIdx]; } | |
58 pixel* getCrAddr(uint32_t ctuAddr, uint32_t absPartIdx) { return fpelPlane[2] + reconPic->m_cuOffsetC[ctuAddr] + reconPic->m_buOffsetC[absPartIdx]; } | |
59 | |
60 /* lowres motion compensation, you must provide a buffer and stride for QPEL averaged pixels | |
61 * in case QPEL is required. Else it returns a pointer to the HPEL pixels */ | |
62 inline pixel *lowresMC(intptr_t blockOffset, const MV& qmv, pixel *buf, intptr_t& outstride) | |
63 { | |
64 if ((qmv.x | qmv.y) & 1) | |
65 { | |
66 int hpelA = (qmv.y & 2) | ((qmv.x & 2) >> 1); | |
67 pixel *frefA = lowresPlane[hpelA] + blockOffset + (qmv.x >> 2) + (qmv.y >> 2) * lumaStride; | |
68 int qmvx = qmv.x + (qmv.x & 1); | |
69 int qmvy = qmv.y + (qmv.y & 1); | |
70 int hpelB = (qmvy & 2) | ((qmvx & 2) >> 1); | |
71 pixel *frefB = lowresPlane[hpelB] + blockOffset + (qmvx >> 2) + (qmvy >> 2) * lumaStride; | |
72 primitives.pu[LUMA_8x8].pixelavg_pp(buf, outstride, frefA, lumaStride, frefB, lumaStride, 32); | |
73 return buf; | |
74 } | |
75 else | |
76 { | |
77 outstride = lumaStride; | |
78 int hpel = (qmv.y & 2) | ((qmv.x & 2) >> 1); | |
79 return lowresPlane[hpel] + blockOffset + (qmv.x >> 2) + (qmv.y >> 2) * lumaStride; | |
80 } | |
81 } | |
82 | |
83 inline int lowresQPelCost(pixel *fenc, intptr_t blockOffset, const MV& qmv, pixelcmp_t comp) | |
84 { | |
85 if ((qmv.x | qmv.y) & 1) | |
86 { | |
87 ALIGN_VAR_16(pixel, subpelbuf[8 * 8]); | |
88 int hpelA = (qmv.y & 2) | ((qmv.x & 2) >> 1); | |
89 pixel *frefA = lowresPlane[hpelA] + blockOffset + (qmv.x >> 2) + (qmv.y >> 2) * lumaStride; | |
90 int qmvx = qmv.x + (qmv.x & 1); | |
91 int qmvy = qmv.y + (qmv.y & 1); | |
92 int hpelB = (qmvy & 2) | ((qmvx & 2) >> 1); | |
93 pixel *frefB = lowresPlane[hpelB] + blockOffset + (qmvx >> 2) + (qmvy >> 2) * lumaStride; | |
94 primitives.pu[LUMA_8x8].pixelavg_pp(subpelbuf, 8, frefA, lumaStride, frefB, lumaStride, 32); | |
95 return comp(fenc, FENC_STRIDE, subpelbuf, 8); | |
96 } | |
97 else | |
98 { | |
99 int hpel = (qmv.y & 2) | ((qmv.x & 2) >> 1); | |
100 pixel *fref = lowresPlane[hpel] + blockOffset + (qmv.x >> 2) + (qmv.y >> 2) * lumaStride; | |
101 return comp(fenc, FENC_STRIDE, fref, lumaStride); | |
102 } | |
103 } | |
104 }; | |
105 | |
106 /* lowres buffers, sizes and strides */ | |
107 struct Lowres : public ReferencePlanes | |
108 { | |
109 pixel *buffer[4]; | |
110 | |
111 int frameNum; // Presentation frame number | |
112 int sliceType; // Slice type decided by lookahead | |
113 int width; // width of lowres frame in pixels | |
114 int lines; // height of lowres frame in pixel lines | |
115 int leadingBframes; // number of leading B frames for P or I | |
116 | |
117 bool bScenecut; // Set to false if the frame cannot possibly be part of a real scenecut. | |
118 bool bKeyframe; | |
119 bool bLastMiniGopBFrame; | |
120 | |
121 /* lookahead output data */ | |
122 int64_t costEst[X265_BFRAME_MAX + 2][X265_BFRAME_MAX + 2]; | |
123 int64_t costEstAq[X265_BFRAME_MAX + 2][X265_BFRAME_MAX + 2]; | |
124 int32_t* rowSatds[X265_BFRAME_MAX + 2][X265_BFRAME_MAX + 2]; | |
125 int intraMbs[X265_BFRAME_MAX + 2]; | |
126 int32_t* intraCost; | |
127 uint8_t* intraMode; | |
128 int64_t satdCost; | |
129 uint16_t* lowresCostForRc; | |
130 uint16_t(*lowresCosts[X265_BFRAME_MAX + 2][X265_BFRAME_MAX + 2]); | |
131 int32_t* lowresMvCosts[2][X265_BFRAME_MAX + 1]; | |
132 MV* lowresMvs[2][X265_BFRAME_MAX + 1]; | |
133 uint32_t maxBlocksInRow; | |
134 uint32_t maxBlocksInCol; | |
135 | |
136 /* used for vbvLookahead */ | |
137 int plannedType[X265_LOOKAHEAD_MAX + 1]; | |
138 int64_t plannedSatd[X265_LOOKAHEAD_MAX + 1]; | |
139 int indB; | |
140 int bframes; | |
141 | |
142 /* rate control / adaptive quant data */ | |
143 double* qpAqOffset; // AQ QP offset values for each 16x16 CU | |
144 double* qpCuTreeOffset; // cuTree QP offset values for each 16x16 CU | |
145 int* invQscaleFactor; // qScale values for qp Aq Offsets | |
146 uint64_t wp_ssd[3]; // This is different than SSDY, this is sum(pixel^2) - sum(pixel)^2 for entire frame | |
147 uint64_t wp_sum[3]; | |
148 | |
149 /* cutree intermediate data */ | |
150 uint16_t* propagateCost; | |
151 double weightedCostDelta[X265_BFRAME_MAX + 2]; | |
152 | |
153 bool create(PicYuv *origPic, int _bframes, bool bAqEnabled); | |
154 void destroy(); | |
155 void init(PicYuv *origPic, int poc); | |
156 }; | |
157 } | |
158 | |
159 #endif // ifndef X265_LOWRES_H |