OLD | NEW |
1 /* | 1 /* |
2 * Copyright (c) 2010 The WebM project authors. All Rights Reserved. | 2 * Copyright (c) 2010 The WebM project authors. All Rights Reserved. |
3 * | 3 * |
4 * Use of this source code is governed by a BSD-style license | 4 * Use of this source code is governed by a BSD-style license |
5 * that can be found in the LICENSE file in the root of the source | 5 * that can be found in the LICENSE file in the root of the source |
6 * tree. An additional intellectual property rights grant can be found | 6 * tree. An additional intellectual property rights grant can be found |
7 * in the file PATENTS. All contributing project authors may | 7 * in the file PATENTS. All contributing project authors may |
8 * be found in the AUTHORS file in the root of the source tree. | 8 * be found in the AUTHORS file in the root of the source tree. |
9 */ | 9 */ |
10 | 10 |
11 #include <assert.h> | 11 #include <assert.h> |
12 #include <math.h> | 12 #include <math.h> |
13 #include <stdio.h> | |
14 | 13 |
15 #include "./vp9_rtcd.h" | 14 #include "./vp9_rtcd.h" |
16 | 15 |
17 #include "vpx_mem/vpx_mem.h" | 16 #include "vpx_mem/vpx_mem.h" |
18 | 17 |
19 #include "vp9/common/vp9_common.h" | 18 #include "vp9/common/vp9_common.h" |
20 #include "vp9/common/vp9_entropy.h" | 19 #include "vp9/common/vp9_entropy.h" |
21 #include "vp9/common/vp9_entropymode.h" | 20 #include "vp9/common/vp9_entropymode.h" |
22 #include "vp9/common/vp9_idct.h" | 21 #include "vp9/common/vp9_idct.h" |
23 #include "vp9/common/vp9_mvref_common.h" | 22 #include "vp9/common/vp9_mvref_common.h" |
24 #include "vp9/common/vp9_pred_common.h" | 23 #include "vp9/common/vp9_pred_common.h" |
25 #include "vp9/common/vp9_quant_common.h" | 24 #include "vp9/common/vp9_quant_common.h" |
26 #include "vp9/common/vp9_reconinter.h" | 25 #include "vp9/common/vp9_reconinter.h" |
27 #include "vp9/common/vp9_reconintra.h" | 26 #include "vp9/common/vp9_reconintra.h" |
28 #include "vp9/common/vp9_seg_common.h" | 27 #include "vp9/common/vp9_seg_common.h" |
29 #include "vp9/common/vp9_systemdependent.h" | 28 #include "vp9/common/vp9_systemdependent.h" |
30 | 29 |
31 #include "vp9/encoder/vp9_cost.h" | 30 #include "vp9/encoder/vp9_cost.h" |
32 #include "vp9/encoder/vp9_encodemb.h" | 31 #include "vp9/encoder/vp9_encodemb.h" |
33 #include "vp9/encoder/vp9_encodemv.h" | 32 #include "vp9/encoder/vp9_encodemv.h" |
34 #include "vp9/encoder/vp9_encoder.h" | 33 #include "vp9/encoder/vp9_encoder.h" |
35 #include "vp9/encoder/vp9_mcomp.h" | 34 #include "vp9/encoder/vp9_mcomp.h" |
36 #include "vp9/encoder/vp9_quantize.h" | 35 #include "vp9/encoder/vp9_quantize.h" |
37 #include "vp9/encoder/vp9_ratectrl.h" | 36 #include "vp9/encoder/vp9_ratectrl.h" |
| 37 #include "vp9/encoder/vp9_rd.h" |
38 #include "vp9/encoder/vp9_rdopt.h" | 38 #include "vp9/encoder/vp9_rdopt.h" |
39 #include "vp9/encoder/vp9_tokenize.h" | |
40 #include "vp9/encoder/vp9_variance.h" | 39 #include "vp9/encoder/vp9_variance.h" |
41 | 40 |
42 #define RD_THRESH_MAX_FACT 64 | 41 #define RD_THRESH_MAX_FACT 64 |
43 #define RD_THRESH_INC 1 | 42 #define RD_THRESH_INC 1 |
44 #define RD_THRESH_POW 1.25 | |
45 #define RD_MULT_EPB_RATIO 64 | |
46 | |
47 /* Factor to weigh the rate for switchable interp filters */ | |
48 #define SWITCHABLE_INTERP_RATE_FACTOR 1 | |
49 | 43 |
50 #define LAST_FRAME_MODE_MASK 0xFFEDCD60 | 44 #define LAST_FRAME_MODE_MASK 0xFFEDCD60 |
51 #define GOLDEN_FRAME_MODE_MASK 0xFFDA3BB0 | 45 #define GOLDEN_FRAME_MODE_MASK 0xFFDA3BB0 |
52 #define ALT_REF_MODE_MASK 0xFFC648D0 | 46 #define ALT_REF_MODE_MASK 0xFFC648D0 |
53 | 47 |
54 #define MIN_EARLY_TERM_INDEX 3 | 48 #define MIN_EARLY_TERM_INDEX 3 |
55 | 49 |
56 typedef struct { | 50 typedef struct { |
57 PREDICTION_MODE mode; | 51 PREDICTION_MODE mode; |
58 MV_REFERENCE_FRAME ref_frame[2]; | 52 MV_REFERENCE_FRAME ref_frame[2]; |
(...skipping 62 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
121 | 115 |
122 static const REF_DEFINITION vp9_ref_order[MAX_REFS] = { | 116 static const REF_DEFINITION vp9_ref_order[MAX_REFS] = { |
123 {{LAST_FRAME, NONE}}, | 117 {{LAST_FRAME, NONE}}, |
124 {{GOLDEN_FRAME, NONE}}, | 118 {{GOLDEN_FRAME, NONE}}, |
125 {{ALTREF_FRAME, NONE}}, | 119 {{ALTREF_FRAME, NONE}}, |
126 {{LAST_FRAME, ALTREF_FRAME}}, | 120 {{LAST_FRAME, ALTREF_FRAME}}, |
127 {{GOLDEN_FRAME, ALTREF_FRAME}}, | 121 {{GOLDEN_FRAME, ALTREF_FRAME}}, |
128 {{INTRA_FRAME, NONE}}, | 122 {{INTRA_FRAME, NONE}}, |
129 }; | 123 }; |
130 | 124 |
131 // The baseline rd thresholds for breaking out of the rd loop for | |
132 // certain modes are assumed to be based on 8x8 blocks. | |
133 // This table is used to correct for blocks size. | |
134 // The factors here are << 2 (2 = x0.5, 32 = x8 etc). | |
135 static const uint8_t rd_thresh_block_size_factor[BLOCK_SIZES] = { | |
136 2, 3, 3, 4, 6, 6, 8, 12, 12, 16, 24, 24, 32 | |
137 }; | |
138 | |
139 static int raster_block_offset(BLOCK_SIZE plane_bsize, | 125 static int raster_block_offset(BLOCK_SIZE plane_bsize, |
140 int raster_block, int stride) { | 126 int raster_block, int stride) { |
141 const int bw = b_width_log2(plane_bsize); | 127 const int bw = b_width_log2(plane_bsize); |
142 const int y = 4 * (raster_block >> bw); | 128 const int y = 4 * (raster_block >> bw); |
143 const int x = 4 * (raster_block & ((1 << bw) - 1)); | 129 const int x = 4 * (raster_block & ((1 << bw) - 1)); |
144 return y * stride + x; | 130 return y * stride + x; |
145 } | 131 } |
146 static int16_t* raster_block_offset_int16(BLOCK_SIZE plane_bsize, | 132 static int16_t* raster_block_offset_int16(BLOCK_SIZE plane_bsize, |
147 int raster_block, int16_t *base) { | 133 int raster_block, int16_t *base) { |
148 const int stride = 4 * num_4x4_blocks_wide_lookup[plane_bsize]; | 134 const int stride = 4 * num_4x4_blocks_wide_lookup[plane_bsize]; |
149 return base + raster_block_offset(plane_bsize, raster_block, stride); | 135 return base + raster_block_offset(plane_bsize, raster_block, stride); |
150 } | 136 } |
151 | 137 |
152 static void fill_mode_costs(VP9_COMP *cpi) { | |
153 const FRAME_CONTEXT *const fc = &cpi->common.fc; | |
154 int i, j; | |
155 | |
156 for (i = 0; i < INTRA_MODES; i++) | |
157 for (j = 0; j < INTRA_MODES; j++) | |
158 vp9_cost_tokens(cpi->y_mode_costs[i][j], vp9_kf_y_mode_prob[i][j], | |
159 vp9_intra_mode_tree); | |
160 | |
161 // TODO(rbultje) separate tables for superblock costing? | |
162 vp9_cost_tokens(cpi->mbmode_cost, fc->y_mode_prob[1], vp9_intra_mode_tree); | |
163 vp9_cost_tokens(cpi->intra_uv_mode_cost[KEY_FRAME], | |
164 vp9_kf_uv_mode_prob[TM_PRED], vp9_intra_mode_tree); | |
165 vp9_cost_tokens(cpi->intra_uv_mode_cost[INTER_FRAME], | |
166 fc->uv_mode_prob[TM_PRED], vp9_intra_mode_tree); | |
167 | |
168 for (i = 0; i < SWITCHABLE_FILTER_CONTEXTS; ++i) | |
169 vp9_cost_tokens(cpi->switchable_interp_costs[i], | |
170 fc->switchable_interp_prob[i], vp9_switchable_interp_tree); | |
171 } | |
172 | |
173 static void fill_token_costs(vp9_coeff_cost *c, | |
174 vp9_coeff_probs_model (*p)[PLANE_TYPES]) { | |
175 int i, j, k, l; | |
176 TX_SIZE t; | |
177 for (t = TX_4X4; t <= TX_32X32; ++t) | |
178 for (i = 0; i < PLANE_TYPES; ++i) | |
179 for (j = 0; j < REF_TYPES; ++j) | |
180 for (k = 0; k < COEF_BANDS; ++k) | |
181 for (l = 0; l < BAND_COEFF_CONTEXTS(k); ++l) { | |
182 vp9_prob probs[ENTROPY_NODES]; | |
183 vp9_model_to_full_probs(p[t][i][j][k][l], probs); | |
184 vp9_cost_tokens((int *)c[t][i][j][k][0][l], probs, | |
185 vp9_coef_tree); | |
186 vp9_cost_tokens_skip((int *)c[t][i][j][k][1][l], probs, | |
187 vp9_coef_tree); | |
188 assert(c[t][i][j][k][0][l][EOB_TOKEN] == | |
189 c[t][i][j][k][1][l][EOB_TOKEN]); | |
190 } | |
191 } | |
192 | |
193 static const uint8_t rd_iifactor[32] = { | |
194 4, 4, 3, 2, 1, 0, 0, 0, | |
195 0, 0, 0, 0, 0, 0, 0, 0, | |
196 0, 0, 0, 0, 0, 0, 0, 0, | |
197 0, 0, 0, 0, 0, 0, 0, 0, | |
198 }; | |
199 | |
200 // 3* dc_qlookup[Q]*dc_qlookup[Q]; | |
201 | |
202 /* values are now correlated to quantizer */ | |
203 static int sad_per_bit16lut[QINDEX_RANGE]; | |
204 static int sad_per_bit4lut[QINDEX_RANGE]; | |
205 | |
206 void vp9_init_me_luts() { | |
207 int i; | |
208 | |
209 // Initialize the sad lut tables using a formulaic calculation for now | |
210 // This is to make it easier to resolve the impact of experimental changes | |
211 // to the quantizer tables. | |
212 for (i = 0; i < QINDEX_RANGE; i++) { | |
213 const double q = vp9_convert_qindex_to_q(i); | |
214 sad_per_bit16lut[i] = (int)(0.0418 * q + 2.4107); | |
215 sad_per_bit4lut[i] = (int)(0.063 * q + 2.742); | |
216 } | |
217 } | |
218 | |
219 int vp9_compute_rd_mult(const VP9_COMP *cpi, int qindex) { | |
220 const int q = vp9_dc_quant(qindex, 0); | |
221 // TODO(debargha): Adjust the function below | |
222 int rdmult = 88 * q * q / 25; | |
223 if (cpi->pass == 2 && (cpi->common.frame_type != KEY_FRAME)) { | |
224 if (cpi->twopass.next_iiratio > 31) | |
225 rdmult += (rdmult * rd_iifactor[31]) >> 4; | |
226 else | |
227 rdmult += (rdmult * rd_iifactor[cpi->twopass.next_iiratio]) >> 4; | |
228 } | |
229 return rdmult; | |
230 } | |
231 | |
232 static int compute_rd_thresh_factor(int qindex) { | |
233 // TODO(debargha): Adjust the function below | |
234 const int q = (int)(pow(vp9_dc_quant(qindex, 0) / 4.0, RD_THRESH_POW) * 5.12); | |
235 return MAX(q, 8); | |
236 } | |
237 | |
238 void vp9_initialize_me_consts(VP9_COMP *cpi, int qindex) { | |
239 cpi->mb.sadperbit16 = sad_per_bit16lut[qindex]; | |
240 cpi->mb.sadperbit4 = sad_per_bit4lut[qindex]; | |
241 } | |
242 | |
243 static void swap_block_ptr(MACROBLOCK *x, PICK_MODE_CONTEXT *ctx, | 138 static void swap_block_ptr(MACROBLOCK *x, PICK_MODE_CONTEXT *ctx, |
244 int m, int n, int min_plane, int max_plane) { | 139 int m, int n, int min_plane, int max_plane) { |
245 int i; | 140 int i; |
246 | 141 |
247 for (i = min_plane; i < max_plane; ++i) { | 142 for (i = min_plane; i < max_plane; ++i) { |
248 struct macroblock_plane *const p = &x->plane[i]; | 143 struct macroblock_plane *const p = &x->plane[i]; |
249 struct macroblockd_plane *const pd = &x->e_mbd.plane[i]; | 144 struct macroblockd_plane *const pd = &x->e_mbd.plane[i]; |
250 | 145 |
251 p->coeff = ctx->coeff_pbuf[i][m]; | 146 p->coeff = ctx->coeff_pbuf[i][m]; |
252 p->qcoeff = ctx->qcoeff_pbuf[i][m]; | 147 p->qcoeff = ctx->qcoeff_pbuf[i][m]; |
253 pd->dqcoeff = ctx->dqcoeff_pbuf[i][m]; | 148 pd->dqcoeff = ctx->dqcoeff_pbuf[i][m]; |
254 p->eobs = ctx->eobs_pbuf[i][m]; | 149 p->eobs = ctx->eobs_pbuf[i][m]; |
255 | 150 |
256 ctx->coeff_pbuf[i][m] = ctx->coeff_pbuf[i][n]; | 151 ctx->coeff_pbuf[i][m] = ctx->coeff_pbuf[i][n]; |
257 ctx->qcoeff_pbuf[i][m] = ctx->qcoeff_pbuf[i][n]; | 152 ctx->qcoeff_pbuf[i][m] = ctx->qcoeff_pbuf[i][n]; |
258 ctx->dqcoeff_pbuf[i][m] = ctx->dqcoeff_pbuf[i][n]; | 153 ctx->dqcoeff_pbuf[i][m] = ctx->dqcoeff_pbuf[i][n]; |
259 ctx->eobs_pbuf[i][m] = ctx->eobs_pbuf[i][n]; | 154 ctx->eobs_pbuf[i][m] = ctx->eobs_pbuf[i][n]; |
260 | 155 |
261 ctx->coeff_pbuf[i][n] = p->coeff; | 156 ctx->coeff_pbuf[i][n] = p->coeff; |
262 ctx->qcoeff_pbuf[i][n] = p->qcoeff; | 157 ctx->qcoeff_pbuf[i][n] = p->qcoeff; |
263 ctx->dqcoeff_pbuf[i][n] = pd->dqcoeff; | 158 ctx->dqcoeff_pbuf[i][n] = pd->dqcoeff; |
264 ctx->eobs_pbuf[i][n] = p->eobs; | 159 ctx->eobs_pbuf[i][n] = p->eobs; |
265 } | 160 } |
266 } | 161 } |
267 | 162 |
268 static void set_block_thresholds(const VP9_COMMON *cm, RD_OPT *rd) { | |
269 int i, bsize, segment_id; | |
270 | |
271 for (segment_id = 0; segment_id < MAX_SEGMENTS; ++segment_id) { | |
272 const int qindex = clamp(vp9_get_qindex(&cm->seg, segment_id, | |
273 cm->base_qindex) + cm->y_dc_delta_q, | |
274 0, MAXQ); | |
275 const int q = compute_rd_thresh_factor(qindex); | |
276 | |
277 for (bsize = 0; bsize < BLOCK_SIZES; ++bsize) { | |
278 // Threshold here seems unnecessarily harsh but fine given actual | |
279 // range of values used for cpi->sf.thresh_mult[]. | |
280 const int t = q * rd_thresh_block_size_factor[bsize]; | |
281 const int thresh_max = INT_MAX / t; | |
282 | |
283 if (bsize >= BLOCK_8X8) { | |
284 for (i = 0; i < MAX_MODES; ++i) | |
285 rd->threshes[segment_id][bsize][i] = | |
286 rd->thresh_mult[i] < thresh_max | |
287 ? rd->thresh_mult[i] * t / 4 | |
288 : INT_MAX; | |
289 } else { | |
290 for (i = 0; i < MAX_REFS; ++i) | |
291 rd->threshes[segment_id][bsize][i] = | |
292 rd->thresh_mult_sub8x8[i] < thresh_max | |
293 ? rd->thresh_mult_sub8x8[i] * t / 4 | |
294 : INT_MAX; | |
295 } | |
296 } | |
297 } | |
298 } | |
299 | |
300 void vp9_initialize_rd_consts(VP9_COMP *cpi) { | |
301 VP9_COMMON *const cm = &cpi->common; | |
302 MACROBLOCK *const x = &cpi->mb; | |
303 RD_OPT *const rd = &cpi->rd; | |
304 int i; | |
305 | |
306 vp9_clear_system_state(); | |
307 | |
308 rd->RDDIV = RDDIV_BITS; // in bits (to multiply D by 128) | |
309 rd->RDMULT = vp9_compute_rd_mult(cpi, cm->base_qindex + cm->y_dc_delta_q); | |
310 | |
311 x->errorperbit = rd->RDMULT / RD_MULT_EPB_RATIO; | |
312 x->errorperbit += (x->errorperbit == 0); | |
313 | |
314 x->select_tx_size = (cpi->sf.tx_size_search_method == USE_LARGESTALL && | |
315 cm->frame_type != KEY_FRAME) ? 0 : 1; | |
316 | |
317 set_block_thresholds(cm, rd); | |
318 | |
319 if (!cpi->sf.use_nonrd_pick_mode || cm->frame_type == KEY_FRAME) { | |
320 fill_token_costs(x->token_costs, cm->fc.coef_probs); | |
321 | |
322 for (i = 0; i < PARTITION_CONTEXTS; i++) | |
323 vp9_cost_tokens(cpi->partition_cost[i], get_partition_probs(cm, i), | |
324 vp9_partition_tree); | |
325 } | |
326 | |
327 if (!cpi->sf.use_nonrd_pick_mode || (cm->current_video_frame & 0x07) == 1 || | |
328 cm->frame_type == KEY_FRAME) { | |
329 fill_mode_costs(cpi); | |
330 | |
331 if (!frame_is_intra_only(cm)) { | |
332 vp9_build_nmv_cost_table(x->nmvjointcost, | |
333 cm->allow_high_precision_mv ? x->nmvcost_hp | |
334 : x->nmvcost, | |
335 &cm->fc.nmvc, cm->allow_high_precision_mv); | |
336 | |
337 for (i = 0; i < INTER_MODE_CONTEXTS; ++i) | |
338 vp9_cost_tokens((int *)cpi->inter_mode_cost[i], | |
339 cm->fc.inter_mode_probs[i], vp9_inter_mode_tree); | |
340 } | |
341 } | |
342 } | |
343 | |
344 static const int MAX_XSQ_Q10 = 245727; | |
345 | |
346 static void model_rd_norm(int xsq_q10, int *r_q10, int *d_q10) { | |
347 // NOTE: The tables below must be of the same size | |
348 | |
349 // The functions described below are sampled at the four most significant | |
350 // bits of x^2 + 8 / 256 | |
351 | |
352 // Normalized rate | |
353 // This table models the rate for a Laplacian source | |
354 // source with given variance when quantized with a uniform quantizer | |
355 // with given stepsize. The closed form expression is: | |
356 // Rn(x) = H(sqrt(r)) + sqrt(r)*[1 + H(r)/(1 - r)], | |
357 // where r = exp(-sqrt(2) * x) and x = qpstep / sqrt(variance), | |
358 // and H(x) is the binary entropy function. | |
359 static const int rate_tab_q10[] = { | |
360 65536, 6086, 5574, 5275, 5063, 4899, 4764, 4651, | |
361 4553, 4389, 4255, 4142, 4044, 3958, 3881, 3811, | |
362 3748, 3635, 3538, 3453, 3376, 3307, 3244, 3186, | |
363 3133, 3037, 2952, 2877, 2809, 2747, 2690, 2638, | |
364 2589, 2501, 2423, 2353, 2290, 2232, 2179, 2130, | |
365 2084, 2001, 1928, 1862, 1802, 1748, 1698, 1651, | |
366 1608, 1530, 1460, 1398, 1342, 1290, 1243, 1199, | |
367 1159, 1086, 1021, 963, 911, 864, 821, 781, | |
368 745, 680, 623, 574, 530, 490, 455, 424, | |
369 395, 345, 304, 269, 239, 213, 190, 171, | |
370 154, 126, 104, 87, 73, 61, 52, 44, | |
371 38, 28, 21, 16, 12, 10, 8, 6, | |
372 5, 3, 2, 1, 1, 1, 0, 0, | |
373 }; | |
374 // Normalized distortion | |
375 // This table models the normalized distortion for a Laplacian source | |
376 // source with given variance when quantized with a uniform quantizer | |
377 // with given stepsize. The closed form expression is: | |
378 // Dn(x) = 1 - 1/sqrt(2) * x / sinh(x/sqrt(2)) | |
379 // where x = qpstep / sqrt(variance) | |
380 // Note the actual distortion is Dn * variance. | |
381 static const int dist_tab_q10[] = { | |
382 0, 0, 1, 1, 1, 2, 2, 2, | |
383 3, 3, 4, 5, 5, 6, 7, 7, | |
384 8, 9, 11, 12, 13, 15, 16, 17, | |
385 18, 21, 24, 26, 29, 31, 34, 36, | |
386 39, 44, 49, 54, 59, 64, 69, 73, | |
387 78, 88, 97, 106, 115, 124, 133, 142, | |
388 151, 167, 184, 200, 215, 231, 245, 260, | |
389 274, 301, 327, 351, 375, 397, 418, 439, | |
390 458, 495, 528, 559, 587, 613, 637, 659, | |
391 680, 717, 749, 777, 801, 823, 842, 859, | |
392 874, 899, 919, 936, 949, 960, 969, 977, | |
393 983, 994, 1001, 1006, 1010, 1013, 1015, 1017, | |
394 1018, 1020, 1022, 1022, 1023, 1023, 1023, 1024, | |
395 }; | |
396 static const int xsq_iq_q10[] = { | |
397 0, 4, 8, 12, 16, 20, 24, 28, | |
398 32, 40, 48, 56, 64, 72, 80, 88, | |
399 96, 112, 128, 144, 160, 176, 192, 208, | |
400 224, 256, 288, 320, 352, 384, 416, 448, | |
401 480, 544, 608, 672, 736, 800, 864, 928, | |
402 992, 1120, 1248, 1376, 1504, 1632, 1760, 1888, | |
403 2016, 2272, 2528, 2784, 3040, 3296, 3552, 3808, | |
404 4064, 4576, 5088, 5600, 6112, 6624, 7136, 7648, | |
405 8160, 9184, 10208, 11232, 12256, 13280, 14304, 15328, | |
406 16352, 18400, 20448, 22496, 24544, 26592, 28640, 30688, | |
407 32736, 36832, 40928, 45024, 49120, 53216, 57312, 61408, | |
408 65504, 73696, 81888, 90080, 98272, 106464, 114656, 122848, | |
409 131040, 147424, 163808, 180192, 196576, 212960, 229344, 245728, | |
410 }; | |
411 /* | |
412 static const int tab_size = sizeof(rate_tab_q10) / sizeof(rate_tab_q10[0]); | |
413 assert(sizeof(dist_tab_q10) / sizeof(dist_tab_q10[0]) == tab_size); | |
414 assert(sizeof(xsq_iq_q10) / sizeof(xsq_iq_q10[0]) == tab_size); | |
415 assert(MAX_XSQ_Q10 + 1 == xsq_iq_q10[tab_size - 1]); | |
416 */ | |
417 int tmp = (xsq_q10 >> 2) + 8; | |
418 int k = get_msb(tmp) - 3; | |
419 int xq = (k << 3) + ((tmp >> k) & 0x7); | |
420 const int one_q10 = 1 << 10; | |
421 const int a_q10 = ((xsq_q10 - xsq_iq_q10[xq]) << 10) >> (2 + k); | |
422 const int b_q10 = one_q10 - a_q10; | |
423 *r_q10 = (rate_tab_q10[xq] * b_q10 + rate_tab_q10[xq + 1] * a_q10) >> 10; | |
424 *d_q10 = (dist_tab_q10[xq] * b_q10 + dist_tab_q10[xq + 1] * a_q10) >> 10; | |
425 } | |
426 | |
427 void vp9_model_rd_from_var_lapndz(unsigned int var, unsigned int n, | |
428 unsigned int qstep, int *rate, | |
429 int64_t *dist) { | |
430 // This function models the rate and distortion for a Laplacian | |
431 // source with given variance when quantized with a uniform quantizer | |
432 // with given stepsize. The closed form expressions are in: | |
433 // Hang and Chen, "Source Model for transform video coder and its | |
434 // application - Part I: Fundamental Theory", IEEE Trans. Circ. | |
435 // Sys. for Video Tech., April 1997. | |
436 if (var == 0) { | |
437 *rate = 0; | |
438 *dist = 0; | |
439 } else { | |
440 int d_q10, r_q10; | |
441 const uint64_t xsq_q10_64 = | |
442 ((((uint64_t)qstep * qstep * n) << 10) + (var >> 1)) / var; | |
443 const int xsq_q10 = xsq_q10_64 > MAX_XSQ_Q10 ? | |
444 MAX_XSQ_Q10 : (int)xsq_q10_64; | |
445 model_rd_norm(xsq_q10, &r_q10, &d_q10); | |
446 *rate = (n * r_q10 + 2) >> 2; | |
447 *dist = (var * (int64_t)d_q10 + 512) >> 10; | |
448 } | |
449 } | |
450 | |
451 static void model_rd_for_sb(VP9_COMP *cpi, BLOCK_SIZE bsize, | 163 static void model_rd_for_sb(VP9_COMP *cpi, BLOCK_SIZE bsize, |
452 MACROBLOCK *x, MACROBLOCKD *xd, | 164 MACROBLOCK *x, MACROBLOCKD *xd, |
453 int *out_rate_sum, int64_t *out_dist_sum) { | 165 int *out_rate_sum, int64_t *out_dist_sum) { |
454 // Note our transform coeffs are 8 times an orthogonal transform. | 166 // Note our transform coeffs are 8 times an orthogonal transform. |
455 // Hence quantizer step is also 8 times. To get effective quantizer | 167 // Hence quantizer step is also 8 times. To get effective quantizer |
456 // we need to divide by 8 before sending to modeling function. | 168 // we need to divide by 8 before sending to modeling function. |
457 int i; | 169 int i; |
458 int64_t rate_sum = 0; | 170 int64_t rate_sum = 0; |
459 int64_t dist_sum = 0; | 171 int64_t dist_sum = 0; |
460 const int ref = xd->mi[0]->mbmi.ref_frame[0]; | 172 const int ref = xd->mi[0]->mbmi.ref_frame[0]; |
(...skipping 31 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
492 pd->dequant[1] >> 3, &rate, &dist); | 204 pd->dequant[1] >> 3, &rate, &dist); |
493 rate_sum += rate; | 205 rate_sum += rate; |
494 dist_sum += dist; | 206 dist_sum += dist; |
495 } | 207 } |
496 } | 208 } |
497 | 209 |
498 *out_rate_sum = (int)rate_sum; | 210 *out_rate_sum = (int)rate_sum; |
499 *out_dist_sum = dist_sum << 4; | 211 *out_dist_sum = dist_sum << 4; |
500 } | 212 } |
501 | 213 |
502 static void model_rd_for_sb_y_tx(VP9_COMP *cpi, BLOCK_SIZE bsize, | |
503 TX_SIZE tx_size, | |
504 MACROBLOCK *x, MACROBLOCKD *xd, | |
505 int *out_rate_sum, int64_t *out_dist_sum, | |
506 int *out_skip) { | |
507 int j, k; | |
508 BLOCK_SIZE bs; | |
509 const struct macroblock_plane *const p = &x->plane[0]; | |
510 const struct macroblockd_plane *const pd = &xd->plane[0]; | |
511 const int width = 4 * num_4x4_blocks_wide_lookup[bsize]; | |
512 const int height = 4 * num_4x4_blocks_high_lookup[bsize]; | |
513 int rate_sum = 0; | |
514 int64_t dist_sum = 0; | |
515 const int t = 4 << tx_size; | |
516 | |
517 if (tx_size == TX_4X4) { | |
518 bs = BLOCK_4X4; | |
519 } else if (tx_size == TX_8X8) { | |
520 bs = BLOCK_8X8; | |
521 } else if (tx_size == TX_16X16) { | |
522 bs = BLOCK_16X16; | |
523 } else if (tx_size == TX_32X32) { | |
524 bs = BLOCK_32X32; | |
525 } else { | |
526 assert(0); | |
527 } | |
528 | |
529 *out_skip = 1; | |
530 for (j = 0; j < height; j += t) { | |
531 for (k = 0; k < width; k += t) { | |
532 int rate; | |
533 int64_t dist; | |
534 unsigned int sse; | |
535 cpi->fn_ptr[bs].vf(&p->src.buf[j * p->src.stride + k], p->src.stride, | |
536 &pd->dst.buf[j * pd->dst.stride + k], pd->dst.stride, | |
537 &sse); | |
538 // sse works better than var, since there is no dc prediction used | |
539 vp9_model_rd_from_var_lapndz(sse, t * t, pd->dequant[1] >> 3, | |
540 &rate, &dist); | |
541 rate_sum += rate; | |
542 dist_sum += dist; | |
543 *out_skip &= (rate < 1024); | |
544 } | |
545 } | |
546 | |
547 *out_rate_sum = rate_sum; | |
548 *out_dist_sum = dist_sum << 4; | |
549 } | |
550 | |
551 int64_t vp9_block_error_c(const int16_t *coeff, const int16_t *dqcoeff, | 214 int64_t vp9_block_error_c(const int16_t *coeff, const int16_t *dqcoeff, |
552 intptr_t block_size, int64_t *ssz) { | 215 intptr_t block_size, int64_t *ssz) { |
553 int i; | 216 int i; |
554 int64_t error = 0, sqcoeff = 0; | 217 int64_t error = 0, sqcoeff = 0; |
555 | 218 |
556 for (i = 0; i < block_size; i++) { | 219 for (i = 0; i < block_size; i++) { |
557 const int diff = coeff[i] - dqcoeff[i]; | 220 const int diff = coeff[i] - dqcoeff[i]; |
558 error += diff * diff; | 221 error += diff * diff; |
559 sqcoeff += coeff[i] * coeff[i]; | 222 sqcoeff += coeff[i] * coeff[i]; |
560 } | 223 } |
(...skipping 27 matching lines...) Expand all Loading... |
588 const int16_t *band_count = &band_counts[tx_size][1]; | 251 const int16_t *band_count = &band_counts[tx_size][1]; |
589 const int eob = p->eobs[block]; | 252 const int eob = p->eobs[block]; |
590 const int16_t *const qcoeff = BLOCK_OFFSET(p->qcoeff, block); | 253 const int16_t *const qcoeff = BLOCK_OFFSET(p->qcoeff, block); |
591 unsigned int (*token_costs)[2][COEFF_CONTEXTS][ENTROPY_TOKENS] = | 254 unsigned int (*token_costs)[2][COEFF_CONTEXTS][ENTROPY_TOKENS] = |
592 x->token_costs[tx_size][type][is_inter_block(mbmi)]; | 255 x->token_costs[tx_size][type][is_inter_block(mbmi)]; |
593 uint8_t token_cache[32 * 32]; | 256 uint8_t token_cache[32 * 32]; |
594 int pt = combine_entropy_contexts(*A, *L); | 257 int pt = combine_entropy_contexts(*A, *L); |
595 int c, cost; | 258 int c, cost; |
596 // Check for consistency of tx_size with mode info | 259 // Check for consistency of tx_size with mode info |
597 assert(type == PLANE_TYPE_Y ? mbmi->tx_size == tx_size | 260 assert(type == PLANE_TYPE_Y ? mbmi->tx_size == tx_size |
598 : get_uv_tx_size(mbmi) == tx_size); | 261 : get_uv_tx_size(mbmi, pd) == tx_size); |
599 | 262 |
600 if (eob == 0) { | 263 if (eob == 0) { |
601 // single eob token | 264 // single eob token |
602 cost = token_costs[0][0][pt][EOB_TOKEN]; | 265 cost = token_costs[0][0][pt][EOB_TOKEN]; |
603 c = 0; | 266 c = 0; |
604 } else { | 267 } else { |
605 int band_left = *band_count++; | 268 int band_left = *band_count++; |
606 | 269 |
607 // dc token | 270 // dc token |
608 int v = qcoeff[0]; | 271 int v = qcoeff[0]; |
(...skipping 105 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
714 args->this_dist += args->dist; | 377 args->this_dist += args->dist; |
715 args->this_sse += args->sse; | 378 args->this_sse += args->sse; |
716 args->this_rd += rd; | 379 args->this_rd += rd; |
717 | 380 |
718 if (args->this_rd > args->best_rd) { | 381 if (args->this_rd > args->best_rd) { |
719 args->skip = 1; | 382 args->skip = 1; |
720 return; | 383 return; |
721 } | 384 } |
722 } | 385 } |
723 | 386 |
724 void vp9_get_entropy_contexts(BLOCK_SIZE bsize, TX_SIZE tx_size, | |
725 const struct macroblockd_plane *pd, | |
726 ENTROPY_CONTEXT t_above[16], | |
727 ENTROPY_CONTEXT t_left[16]) { | |
728 const BLOCK_SIZE plane_bsize = get_plane_block_size(bsize, pd); | |
729 const int num_4x4_w = num_4x4_blocks_wide_lookup[plane_bsize]; | |
730 const int num_4x4_h = num_4x4_blocks_high_lookup[plane_bsize]; | |
731 const ENTROPY_CONTEXT *const above = pd->above_context; | |
732 const ENTROPY_CONTEXT *const left = pd->left_context; | |
733 | |
734 int i; | |
735 switch (tx_size) { | |
736 case TX_4X4: | |
737 vpx_memcpy(t_above, above, sizeof(ENTROPY_CONTEXT) * num_4x4_w); | |
738 vpx_memcpy(t_left, left, sizeof(ENTROPY_CONTEXT) * num_4x4_h); | |
739 break; | |
740 case TX_8X8: | |
741 for (i = 0; i < num_4x4_w; i += 2) | |
742 t_above[i] = !!*(const uint16_t *)&above[i]; | |
743 for (i = 0; i < num_4x4_h; i += 2) | |
744 t_left[i] = !!*(const uint16_t *)&left[i]; | |
745 break; | |
746 case TX_16X16: | |
747 for (i = 0; i < num_4x4_w; i += 4) | |
748 t_above[i] = !!*(const uint32_t *)&above[i]; | |
749 for (i = 0; i < num_4x4_h; i += 4) | |
750 t_left[i] = !!*(const uint32_t *)&left[i]; | |
751 break; | |
752 case TX_32X32: | |
753 for (i = 0; i < num_4x4_w; i += 8) | |
754 t_above[i] = !!*(const uint64_t *)&above[i]; | |
755 for (i = 0; i < num_4x4_h; i += 8) | |
756 t_left[i] = !!*(const uint64_t *)&left[i]; | |
757 break; | |
758 default: | |
759 assert(0 && "Invalid transform size."); | |
760 } | |
761 } | |
762 | |
763 static void txfm_rd_in_plane(MACROBLOCK *x, | 387 static void txfm_rd_in_plane(MACROBLOCK *x, |
764 int *rate, int64_t *distortion, | 388 int *rate, int64_t *distortion, |
765 int *skippable, int64_t *sse, | 389 int *skippable, int64_t *sse, |
766 int64_t ref_best_rd, int plane, | 390 int64_t ref_best_rd, int plane, |
767 BLOCK_SIZE bsize, TX_SIZE tx_size, | 391 BLOCK_SIZE bsize, TX_SIZE tx_size, |
768 int use_fast_coef_casting) { | 392 int use_fast_coef_casting) { |
769 MACROBLOCKD *const xd = &x->e_mbd; | 393 MACROBLOCKD *const xd = &x->e_mbd; |
770 const struct macroblockd_plane *const pd = &xd->plane[plane]; | 394 const struct macroblockd_plane *const pd = &xd->plane[plane]; |
771 struct rdcost_block_args args; | 395 struct rdcost_block_args args; |
772 vp9_zero(args); | 396 vp9_zero(args); |
(...skipping 113 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
886 cpi->tx_stepdown_count[max_tx_size - TX_16X16]++; | 510 cpi->tx_stepdown_count[max_tx_size - TX_16X16]++; |
887 } else if (rd[TX_8X8][1] < rd[TX_4X4][1]) { | 511 } else if (rd[TX_8X8][1] < rd[TX_4X4][1]) { |
888 tx_cache[TX_MODE_SELECT] = rd[TX_8X8][1]; | 512 tx_cache[TX_MODE_SELECT] = rd[TX_8X8][1]; |
889 cpi->tx_stepdown_count[max_tx_size - TX_8X8]++; | 513 cpi->tx_stepdown_count[max_tx_size - TX_8X8]++; |
890 } else { | 514 } else { |
891 tx_cache[TX_MODE_SELECT] = rd[TX_4X4][1]; | 515 tx_cache[TX_MODE_SELECT] = rd[TX_4X4][1]; |
892 cpi->tx_stepdown_count[max_tx_size - TX_4X4]++; | 516 cpi->tx_stepdown_count[max_tx_size - TX_4X4]++; |
893 } | 517 } |
894 } | 518 } |
895 | 519 |
896 static int64_t scaled_rd_cost(int rdmult, int rddiv, | |
897 int rate, int64_t dist, double scale) { | |
898 return (int64_t) (RDCOST(rdmult, rddiv, rate, dist) * scale); | |
899 } | |
900 | |
901 static void choose_tx_size_from_modelrd(VP9_COMP *cpi, MACROBLOCK *x, | |
902 int (*r)[2], int *rate, | |
903 int64_t *d, int64_t *distortion, | |
904 int *s, int *skip, int64_t *sse, | |
905 int64_t ref_best_rd, | |
906 BLOCK_SIZE bs) { | |
907 const TX_SIZE max_tx_size = max_txsize_lookup[bs]; | |
908 VP9_COMMON *const cm = &cpi->common; | |
909 MACROBLOCKD *const xd = &x->e_mbd; | |
910 MB_MODE_INFO *const mbmi = &xd->mi[0]->mbmi; | |
911 vp9_prob skip_prob = vp9_get_skip_prob(cm, xd); | |
912 int64_t rd[TX_SIZES][2] = {{INT64_MAX, INT64_MAX}, | |
913 {INT64_MAX, INT64_MAX}, | |
914 {INT64_MAX, INT64_MAX}, | |
915 {INT64_MAX, INT64_MAX}}; | |
916 TX_SIZE n, m; | |
917 int s0, s1; | |
918 double scale_rd[TX_SIZES] = {1.73, 1.44, 1.20, 1.00}; | |
919 const TX_SIZE max_mode_tx_size = tx_mode_to_biggest_tx_size[cm->tx_mode]; | |
920 int64_t best_rd = INT64_MAX; | |
921 TX_SIZE best_tx = TX_4X4; | |
922 | |
923 const vp9_prob *tx_probs = get_tx_probs2(max_tx_size, xd, &cm->fc.tx_probs); | |
924 assert(skip_prob > 0); | |
925 s0 = vp9_cost_bit(skip_prob, 0); | |
926 s1 = vp9_cost_bit(skip_prob, 1); | |
927 | |
928 for (n = TX_4X4; n <= max_tx_size; n++) { | |
929 double scale = scale_rd[n]; | |
930 r[n][1] = r[n][0]; | |
931 for (m = 0; m <= n - (n == max_tx_size); m++) { | |
932 if (m == n) | |
933 r[n][1] += vp9_cost_zero(tx_probs[m]); | |
934 else | |
935 r[n][1] += vp9_cost_one(tx_probs[m]); | |
936 } | |
937 if (s[n]) { | |
938 rd[n][0] = rd[n][1] = scaled_rd_cost(x->rdmult, x->rddiv, s1, d[n], | |
939 scale); | |
940 } else { | |
941 rd[n][0] = scaled_rd_cost(x->rdmult, x->rddiv, r[n][0] + s0, d[n], | |
942 scale); | |
943 rd[n][1] = scaled_rd_cost(x->rdmult, x->rddiv, r[n][1] + s0, d[n], | |
944 scale); | |
945 } | |
946 if (rd[n][1] < best_rd) { | |
947 best_rd = rd[n][1]; | |
948 best_tx = n; | |
949 } | |
950 } | |
951 | |
952 mbmi->tx_size = cm->tx_mode == TX_MODE_SELECT ? | |
953 best_tx : MIN(max_tx_size, max_mode_tx_size); | |
954 | |
955 // Actually encode using the chosen mode if a model was used, but do not | |
956 // update the r, d costs | |
957 txfm_rd_in_plane(x, rate, distortion, skip, | |
958 &sse[mbmi->tx_size], ref_best_rd, 0, bs, mbmi->tx_size, | |
959 cpi->sf.use_fast_coef_costing); | |
960 | |
961 if (max_tx_size == TX_32X32 && best_tx == TX_32X32) { | |
962 cpi->tx_stepdown_count[0]++; | |
963 } else if (max_tx_size >= TX_16X16 && best_tx == TX_16X16) { | |
964 cpi->tx_stepdown_count[max_tx_size - TX_16X16]++; | |
965 } else if (rd[TX_8X8][1] <= rd[TX_4X4][1]) { | |
966 cpi->tx_stepdown_count[max_tx_size - TX_8X8]++; | |
967 } else { | |
968 cpi->tx_stepdown_count[max_tx_size - TX_4X4]++; | |
969 } | |
970 } | |
971 | |
972 static void inter_super_block_yrd(VP9_COMP *cpi, MACROBLOCK *x, int *rate, | 520 static void inter_super_block_yrd(VP9_COMP *cpi, MACROBLOCK *x, int *rate, |
973 int64_t *distortion, int *skip, | 521 int64_t *distortion, int *skip, |
974 int64_t *psse, BLOCK_SIZE bs, | 522 int64_t *psse, BLOCK_SIZE bs, |
975 int64_t txfm_cache[TX_MODES], | 523 int64_t txfm_cache[TX_MODES], |
976 int64_t ref_best_rd) { | 524 int64_t ref_best_rd) { |
977 int r[TX_SIZES][2], s[TX_SIZES]; | 525 int r[TX_SIZES][2], s[TX_SIZES]; |
978 int64_t d[TX_SIZES], sse[TX_SIZES]; | 526 int64_t d[TX_SIZES], sse[TX_SIZES]; |
979 MACROBLOCKD *xd = &x->e_mbd; | 527 MACROBLOCKD *xd = &x->e_mbd; |
980 MB_MODE_INFO *const mbmi = &xd->mi[0]->mbmi; | 528 MB_MODE_INFO *const mbmi = &xd->mi[0]->mbmi; |
981 const TX_SIZE max_tx_size = max_txsize_lookup[bs]; | 529 const TX_SIZE max_tx_size = max_txsize_lookup[bs]; |
(...skipping 338 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
1320 | 868 |
1321 return best_rd; | 869 return best_rd; |
1322 } | 870 } |
1323 | 871 |
1324 static void super_block_uvrd(const VP9_COMP *cpi, MACROBLOCK *x, | 872 static void super_block_uvrd(const VP9_COMP *cpi, MACROBLOCK *x, |
1325 int *rate, int64_t *distortion, int *skippable, | 873 int *rate, int64_t *distortion, int *skippable, |
1326 int64_t *sse, BLOCK_SIZE bsize, | 874 int64_t *sse, BLOCK_SIZE bsize, |
1327 int64_t ref_best_rd) { | 875 int64_t ref_best_rd) { |
1328 MACROBLOCKD *const xd = &x->e_mbd; | 876 MACROBLOCKD *const xd = &x->e_mbd; |
1329 MB_MODE_INFO *const mbmi = &xd->mi[0]->mbmi; | 877 MB_MODE_INFO *const mbmi = &xd->mi[0]->mbmi; |
1330 const TX_SIZE uv_tx_size = get_uv_tx_size(mbmi); | 878 const TX_SIZE uv_tx_size = get_uv_tx_size(mbmi, &xd->plane[1]); |
1331 int plane; | 879 int plane; |
1332 int pnrate = 0, pnskip = 1; | 880 int pnrate = 0, pnskip = 1; |
1333 int64_t pndist = 0, pnsse = 0; | 881 int64_t pndist = 0, pnsse = 0; |
1334 | 882 |
1335 if (ref_best_rd < 0) | 883 if (ref_best_rd < 0) |
1336 goto term; | 884 goto term; |
1337 | 885 |
1338 if (is_inter_block(mbmi)) { | 886 if (is_inter_block(mbmi)) { |
1339 int plane; | 887 int plane; |
1340 for (plane = 1; plane < MAX_MB_PLANE; ++plane) | 888 for (plane = 1; plane < MAX_MB_PLANE; ++plane) |
(...skipping 99 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
1440 } else { | 988 } else { |
1441 rd_pick_intra_sbuv_mode(cpi, x, ctx, | 989 rd_pick_intra_sbuv_mode(cpi, x, ctx, |
1442 rate_uv, rate_uv_tokenonly, dist_uv, skip_uv, | 990 rate_uv, rate_uv_tokenonly, dist_uv, skip_uv, |
1443 bsize < BLOCK_8X8 ? BLOCK_8X8 : bsize, max_tx_size); | 991 bsize < BLOCK_8X8 ? BLOCK_8X8 : bsize, max_tx_size); |
1444 } | 992 } |
1445 *mode_uv = x->e_mbd.mi[0]->mbmi.uv_mode; | 993 *mode_uv = x->e_mbd.mi[0]->mbmi.uv_mode; |
1446 } | 994 } |
1447 | 995 |
1448 static int cost_mv_ref(const VP9_COMP *cpi, PREDICTION_MODE mode, | 996 static int cost_mv_ref(const VP9_COMP *cpi, PREDICTION_MODE mode, |
1449 int mode_context) { | 997 int mode_context) { |
1450 const MACROBLOCK *const x = &cpi->mb; | 998 assert(is_inter_mode(mode)); |
1451 const int segment_id = x->e_mbd.mi[0]->mbmi.segment_id; | 999 return cpi->inter_mode_cost[mode_context][INTER_OFFSET(mode)]; |
1452 | |
1453 // Don't account for mode here if segment skip is enabled. | |
1454 if (!vp9_segfeature_active(&cpi->common.seg, segment_id, SEG_LVL_SKIP)) { | |
1455 assert(is_inter_mode(mode)); | |
1456 return cpi->inter_mode_cost[mode_context][INTER_OFFSET(mode)]; | |
1457 } else { | |
1458 return 0; | |
1459 } | |
1460 } | 1000 } |
1461 | 1001 |
1462 static void joint_motion_search(VP9_COMP *cpi, MACROBLOCK *x, | 1002 static void joint_motion_search(VP9_COMP *cpi, MACROBLOCK *x, |
1463 BLOCK_SIZE bsize, | 1003 BLOCK_SIZE bsize, |
1464 int_mv *frame_mv, | 1004 int_mv *frame_mv, |
1465 int mi_row, int mi_col, | 1005 int mi_row, int mi_col, |
1466 int_mv single_newmv[MAX_REF_FRAMES], | 1006 int_mv single_newmv[MAX_REF_FRAMES], |
1467 int *rate_mv); | 1007 int *rate_mv); |
1468 | 1008 |
1469 static int set_and_cost_bmi_mvs(VP9_COMP *cpi, MACROBLOCKD *xd, int i, | 1009 static int set_and_cost_bmi_mvs(VP9_COMP *cpi, MACROBLOCKD *xd, int i, |
(...skipping 597 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
2067 *returntotrate = bsi->r; | 1607 *returntotrate = bsi->r; |
2068 *returndistortion = bsi->d; | 1608 *returndistortion = bsi->d; |
2069 *returnyrate = bsi->segment_yrate; | 1609 *returnyrate = bsi->segment_yrate; |
2070 *skippable = vp9_is_skippable_in_plane(x, BLOCK_8X8, 0); | 1610 *skippable = vp9_is_skippable_in_plane(x, BLOCK_8X8, 0); |
2071 *psse = bsi->sse; | 1611 *psse = bsi->sse; |
2072 mbmi->mode = bsi->modes[3]; | 1612 mbmi->mode = bsi->modes[3]; |
2073 | 1613 |
2074 return bsi->segment_rd; | 1614 return bsi->segment_rd; |
2075 } | 1615 } |
2076 | 1616 |
2077 static void mv_pred(VP9_COMP *cpi, MACROBLOCK *x, | |
2078 uint8_t *ref_y_buffer, int ref_y_stride, | |
2079 int ref_frame, BLOCK_SIZE block_size ) { | |
2080 MACROBLOCKD *xd = &x->e_mbd; | |
2081 MB_MODE_INFO *mbmi = &xd->mi[0]->mbmi; | |
2082 int_mv this_mv; | |
2083 int i; | |
2084 int zero_seen = 0; | |
2085 int best_index = 0; | |
2086 int best_sad = INT_MAX; | |
2087 int this_sad = INT_MAX; | |
2088 int max_mv = 0; | |
2089 | |
2090 uint8_t *src_y_ptr = x->plane[0].src.buf; | |
2091 uint8_t *ref_y_ptr; | |
2092 int row_offset, col_offset; | |
2093 int num_mv_refs = MAX_MV_REF_CANDIDATES + | |
2094 (cpi->sf.adaptive_motion_search && | |
2095 cpi->common.show_frame && | |
2096 block_size < cpi->sf.max_partition_size); | |
2097 | |
2098 MV pred_mv[3]; | |
2099 pred_mv[0] = mbmi->ref_mvs[ref_frame][0].as_mv; | |
2100 pred_mv[1] = mbmi->ref_mvs[ref_frame][1].as_mv; | |
2101 pred_mv[2] = x->pred_mv[ref_frame]; | |
2102 | |
2103 // Get the sad for each candidate reference mv | |
2104 for (i = 0; i < num_mv_refs; i++) { | |
2105 this_mv.as_mv = pred_mv[i]; | |
2106 | |
2107 max_mv = MAX(max_mv, | |
2108 MAX(abs(this_mv.as_mv.row), abs(this_mv.as_mv.col)) >> 3); | |
2109 // only need to check zero mv once | |
2110 if (!this_mv.as_int && zero_seen) | |
2111 continue; | |
2112 | |
2113 zero_seen = zero_seen || !this_mv.as_int; | |
2114 | |
2115 row_offset = this_mv.as_mv.row >> 3; | |
2116 col_offset = this_mv.as_mv.col >> 3; | |
2117 ref_y_ptr = ref_y_buffer + (ref_y_stride * row_offset) + col_offset; | |
2118 | |
2119 // Find sad for current vector. | |
2120 this_sad = cpi->fn_ptr[block_size].sdf(src_y_ptr, x->plane[0].src.stride, | |
2121 ref_y_ptr, ref_y_stride); | |
2122 | |
2123 // Note if it is the best so far. | |
2124 if (this_sad < best_sad) { | |
2125 best_sad = this_sad; | |
2126 best_index = i; | |
2127 } | |
2128 } | |
2129 | |
2130 // Note the index of the mv that worked best in the reference list. | |
2131 x->mv_best_ref_index[ref_frame] = best_index; | |
2132 x->max_mv_context[ref_frame] = max_mv; | |
2133 x->pred_mv_sad[ref_frame] = best_sad; | |
2134 } | |
2135 | |
2136 static void estimate_ref_frame_costs(const VP9_COMMON *cm, | 1617 static void estimate_ref_frame_costs(const VP9_COMMON *cm, |
2137 const MACROBLOCKD *xd, | 1618 const MACROBLOCKD *xd, |
2138 int segment_id, | 1619 int segment_id, |
2139 unsigned int *ref_costs_single, | 1620 unsigned int *ref_costs_single, |
2140 unsigned int *ref_costs_comp, | 1621 unsigned int *ref_costs_comp, |
2141 vp9_prob *comp_mode_p) { | 1622 vp9_prob *comp_mode_p) { |
2142 int seg_ref_active = vp9_segfeature_active(&cm->seg, segment_id, | 1623 int seg_ref_active = vp9_segfeature_active(&cm->seg, segment_id, |
2143 SEG_LVL_REF_FRAME); | 1624 SEG_LVL_REF_FRAME); |
2144 if (seg_ref_active) { | 1625 if (seg_ref_active) { |
2145 vpx_memset(ref_costs_single, 0, MAX_REF_FRAMES * sizeof(*ref_costs_single)); | 1626 vpx_memset(ref_costs_single, 0, MAX_REF_FRAMES * sizeof(*ref_costs_single)); |
(...skipping 62 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
2208 ctx->mic = *xd->mi[0]; | 1689 ctx->mic = *xd->mi[0]; |
2209 ctx->single_pred_diff = (int)comp_pred_diff[SINGLE_REFERENCE]; | 1690 ctx->single_pred_diff = (int)comp_pred_diff[SINGLE_REFERENCE]; |
2210 ctx->comp_pred_diff = (int)comp_pred_diff[COMPOUND_REFERENCE]; | 1691 ctx->comp_pred_diff = (int)comp_pred_diff[COMPOUND_REFERENCE]; |
2211 ctx->hybrid_pred_diff = (int)comp_pred_diff[REFERENCE_MODE_SELECT]; | 1692 ctx->hybrid_pred_diff = (int)comp_pred_diff[REFERENCE_MODE_SELECT]; |
2212 | 1693 |
2213 vpx_memcpy(ctx->tx_rd_diff, tx_size_diff, sizeof(ctx->tx_rd_diff)); | 1694 vpx_memcpy(ctx->tx_rd_diff, tx_size_diff, sizeof(ctx->tx_rd_diff)); |
2214 vpx_memcpy(ctx->best_filter_diff, best_filter_diff, | 1695 vpx_memcpy(ctx->best_filter_diff, best_filter_diff, |
2215 sizeof(*best_filter_diff) * SWITCHABLE_FILTER_CONTEXTS); | 1696 sizeof(*best_filter_diff) * SWITCHABLE_FILTER_CONTEXTS); |
2216 } | 1697 } |
2217 | 1698 |
2218 static void setup_pred_block(const MACROBLOCKD *xd, | 1699 static void setup_buffer_inter(VP9_COMP *cpi, MACROBLOCK *x, |
2219 struct buf_2d dst[MAX_MB_PLANE], | 1700 const TileInfo *const tile, |
2220 const YV12_BUFFER_CONFIG *src, | 1701 MV_REFERENCE_FRAME ref_frame, |
2221 int mi_row, int mi_col, | 1702 BLOCK_SIZE block_size, |
2222 const struct scale_factors *scale, | 1703 int mi_row, int mi_col, |
2223 const struct scale_factors *scale_uv) { | 1704 int_mv frame_nearest_mv[MAX_REF_FRAMES], |
2224 int i; | 1705 int_mv frame_near_mv[MAX_REF_FRAMES], |
2225 | 1706 struct buf_2d yv12_mb[4][MAX_MB_PLANE]) { |
2226 dst[0].buf = src->y_buffer; | |
2227 dst[0].stride = src->y_stride; | |
2228 dst[1].buf = src->u_buffer; | |
2229 dst[2].buf = src->v_buffer; | |
2230 dst[1].stride = dst[2].stride = src->uv_stride; | |
2231 #if CONFIG_ALPHA | |
2232 dst[3].buf = src->alpha_buffer; | |
2233 dst[3].stride = src->alpha_stride; | |
2234 #endif | |
2235 | |
2236 // TODO(jkoleszar): Make scale factors per-plane data | |
2237 for (i = 0; i < MAX_MB_PLANE; i++) { | |
2238 setup_pred_plane(dst + i, dst[i].buf, dst[i].stride, mi_row, mi_col, | |
2239 i ? scale_uv : scale, | |
2240 xd->plane[i].subsampling_x, xd->plane[i].subsampling_y); | |
2241 } | |
2242 } | |
2243 | |
2244 void vp9_setup_buffer_inter(VP9_COMP *cpi, MACROBLOCK *x, | |
2245 const TileInfo *const tile, | |
2246 MV_REFERENCE_FRAME ref_frame, | |
2247 BLOCK_SIZE block_size, | |
2248 int mi_row, int mi_col, | |
2249 int_mv frame_nearest_mv[MAX_REF_FRAMES], | |
2250 int_mv frame_near_mv[MAX_REF_FRAMES], | |
2251 struct buf_2d yv12_mb[4][MAX_MB_PLANE]) { | |
2252 const VP9_COMMON *cm = &cpi->common; | 1707 const VP9_COMMON *cm = &cpi->common; |
2253 const YV12_BUFFER_CONFIG *yv12 = get_ref_frame_buffer(cpi, ref_frame); | 1708 const YV12_BUFFER_CONFIG *yv12 = get_ref_frame_buffer(cpi, ref_frame); |
2254 MACROBLOCKD *const xd = &x->e_mbd; | 1709 MACROBLOCKD *const xd = &x->e_mbd; |
2255 MODE_INFO *const mi = xd->mi[0]; | 1710 MODE_INFO *const mi = xd->mi[0]; |
2256 int_mv *const candidates = mi->mbmi.ref_mvs[ref_frame]; | 1711 int_mv *const candidates = mi->mbmi.ref_mvs[ref_frame]; |
2257 const struct scale_factors *const sf = &cm->frame_refs[ref_frame - 1].sf; | 1712 const struct scale_factors *const sf = &cm->frame_refs[ref_frame - 1].sf; |
2258 | 1713 |
2259 // TODO(jkoleszar): Is the UV buffer ever used here? If so, need to make this | 1714 // TODO(jkoleszar): Is the UV buffer ever used here? If so, need to make this |
2260 // use the UV scaling factors. | 1715 // use the UV scaling factors. |
2261 setup_pred_block(xd, yv12_mb[ref_frame], yv12, mi_row, mi_col, sf, sf); | 1716 vp9_setup_pred_block(xd, yv12_mb[ref_frame], yv12, mi_row, mi_col, sf, sf); |
2262 | 1717 |
2263 // Gets an initial list of candidate vectors from neighbours and orders them | 1718 // Gets an initial list of candidate vectors from neighbours and orders them |
2264 vp9_find_mv_refs(cm, xd, tile, mi, ref_frame, candidates, mi_row, mi_col); | 1719 vp9_find_mv_refs(cm, xd, tile, mi, ref_frame, candidates, mi_row, mi_col); |
2265 | 1720 |
2266 // Candidate refinement carried out at encoder and decoder | 1721 // Candidate refinement carried out at encoder and decoder |
2267 vp9_find_best_ref_mvs(xd, cm->allow_high_precision_mv, candidates, | 1722 vp9_find_best_ref_mvs(xd, cm->allow_high_precision_mv, candidates, |
2268 &frame_nearest_mv[ref_frame], | 1723 &frame_nearest_mv[ref_frame], |
2269 &frame_near_mv[ref_frame]); | 1724 &frame_near_mv[ref_frame]); |
2270 | 1725 |
2271 // Further refinement that is encode side only to test the top few candidates | 1726 // Further refinement that is encode side only to test the top few candidates |
2272 // in full and choose the best as the centre point for subsequent searches. | 1727 // in full and choose the best as the centre point for subsequent searches. |
2273 // The current implementation doesn't support scaling. | 1728 // The current implementation doesn't support scaling. |
2274 if (!vp9_is_scaled(sf) && block_size >= BLOCK_8X8) | 1729 if (!vp9_is_scaled(sf) && block_size >= BLOCK_8X8) |
2275 mv_pred(cpi, x, yv12_mb[ref_frame][0].buf, yv12->y_stride, | 1730 vp9_mv_pred(cpi, x, yv12_mb[ref_frame][0].buf, yv12->y_stride, |
2276 ref_frame, block_size); | 1731 ref_frame, block_size); |
2277 } | |
2278 | |
2279 const YV12_BUFFER_CONFIG *vp9_get_scaled_ref_frame(const VP9_COMP *cpi, | |
2280 int ref_frame) { | |
2281 const VP9_COMMON *const cm = &cpi->common; | |
2282 const int ref_idx = cm->ref_frame_map[get_ref_frame_idx(cpi, ref_frame)]; | |
2283 const int scaled_idx = cpi->scaled_ref_idx[ref_frame - 1]; | |
2284 return (scaled_idx != ref_idx) ? &cm->frame_bufs[scaled_idx].buf : NULL; | |
2285 } | |
2286 | |
2287 int vp9_get_switchable_rate(const VP9_COMP *cpi) { | |
2288 const MACROBLOCKD *const xd = &cpi->mb.e_mbd; | |
2289 const MB_MODE_INFO *const mbmi = &xd->mi[0]->mbmi; | |
2290 const int ctx = vp9_get_pred_context_switchable_interp(xd); | |
2291 return SWITCHABLE_INTERP_RATE_FACTOR * | |
2292 cpi->switchable_interp_costs[ctx][mbmi->interp_filter]; | |
2293 } | 1732 } |
2294 | 1733 |
2295 static void single_motion_search(VP9_COMP *cpi, MACROBLOCK *x, | 1734 static void single_motion_search(VP9_COMP *cpi, MACROBLOCK *x, |
2296 BLOCK_SIZE bsize, | 1735 BLOCK_SIZE bsize, |
2297 int mi_row, int mi_col, | 1736 int mi_row, int mi_col, |
2298 int_mv *tmp_mv, int *rate_mv) { | 1737 int_mv *tmp_mv, int *rate_mv) { |
2299 MACROBLOCKD *xd = &x->e_mbd; | 1738 MACROBLOCKD *xd = &x->e_mbd; |
2300 const VP9_COMMON *cm = &cpi->common; | 1739 const VP9_COMMON *cm = &cpi->common; |
2301 MB_MODE_INFO *mbmi = &xd->mi[0]->mbmi; | 1740 MB_MODE_INFO *mbmi = &xd->mi[0]->mbmi; |
2302 struct buf_2d backup_yv12[MAX_MB_PLANE] = {{0, 0}}; | 1741 struct buf_2d backup_yv12[MAX_MB_PLANE] = {{0, 0}}; |
(...skipping 489 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
2792 if (rd / 2 > ref_best_rd) { | 2231 if (rd / 2 > ref_best_rd) { |
2793 restore_dst_buf(xd, orig_dst, orig_dst_stride); | 2232 restore_dst_buf(xd, orig_dst, orig_dst_stride); |
2794 return INT64_MAX; | 2233 return INT64_MAX; |
2795 } | 2234 } |
2796 } | 2235 } |
2797 | 2236 |
2798 if (cm->interp_filter == SWITCHABLE) | 2237 if (cm->interp_filter == SWITCHABLE) |
2799 *rate2 += vp9_get_switchable_rate(cpi); | 2238 *rate2 += vp9_get_switchable_rate(cpi); |
2800 | 2239 |
2801 if (!is_comp_pred) { | 2240 if (!is_comp_pred) { |
2802 if (!x->in_active_map || | 2241 if (cpi->allow_encode_breakout) { |
2803 vp9_segfeature_active(&cm->seg, mbmi->segment_id, SEG_LVL_SKIP)) { | |
2804 if (psse) | |
2805 *psse = 0; | |
2806 *distortion = 0; | |
2807 x->skip = 1; | |
2808 } else if (cpi->allow_encode_breakout && x->encode_breakout) { | |
2809 const BLOCK_SIZE y_size = get_plane_block_size(bsize, &xd->plane[0]); | 2242 const BLOCK_SIZE y_size = get_plane_block_size(bsize, &xd->plane[0]); |
2810 const BLOCK_SIZE uv_size = get_plane_block_size(bsize, &xd->plane[1]); | 2243 const BLOCK_SIZE uv_size = get_plane_block_size(bsize, &xd->plane[1]); |
2811 unsigned int var, sse; | 2244 unsigned int var, sse; |
2812 // Skipping threshold for ac. | 2245 // Skipping threshold for ac. |
2813 unsigned int thresh_ac; | 2246 unsigned int thresh_ac; |
2814 // Set a maximum for threshold to avoid big PSNR loss in low bitrate case. | 2247 // Skipping threshold for dc |
2815 // Use extreme low threshold for static frames to limit skipping. | 2248 unsigned int thresh_dc; |
2816 const unsigned int max_thresh = (cpi->allow_encode_breakout == | |
2817 ENCODE_BREAKOUT_LIMITED) ? 128 : 36000; | |
2818 // The encode_breakout input | |
2819 const unsigned int min_thresh = | |
2820 MIN(((unsigned int)x->encode_breakout << 4), max_thresh); | |
2821 | |
2822 // Calculate threshold according to dequant value. | |
2823 thresh_ac = (xd->plane[0].dequant[1] * xd->plane[0].dequant[1]) / 9; | |
2824 thresh_ac = clamp(thresh_ac, min_thresh, max_thresh); | |
2825 | 2249 |
2826 var = cpi->fn_ptr[y_size].vf(x->plane[0].src.buf, x->plane[0].src.stride, | 2250 var = cpi->fn_ptr[y_size].vf(x->plane[0].src.buf, x->plane[0].src.stride, |
2827 xd->plane[0].dst.buf, | 2251 xd->plane[0].dst.buf, |
2828 xd->plane[0].dst.stride, &sse); | 2252 xd->plane[0].dst.stride, &sse); |
2829 | 2253 |
2830 // Adjust threshold according to partition size. | 2254 if (x->encode_breakout > 0) { |
2831 thresh_ac >>= 8 - (b_width_log2_lookup[bsize] + | 2255 // Set a maximum for threshold to avoid big PSNR loss in low bitrate |
2832 b_height_log2_lookup[bsize]); | 2256 // case. Use extreme low threshold for static frames to limit skipping. |
| 2257 const unsigned int max_thresh = (cpi->allow_encode_breakout == |
| 2258 ENCODE_BREAKOUT_LIMITED) ? 128 : 36000; |
| 2259 // The encode_breakout input |
| 2260 const unsigned int min_thresh = |
| 2261 MIN(((unsigned int)x->encode_breakout << 4), max_thresh); |
| 2262 |
| 2263 // Calculate threshold according to dequant value. |
| 2264 thresh_ac = (xd->plane[0].dequant[1] * xd->plane[0].dequant[1]) / 9; |
| 2265 thresh_ac = clamp(thresh_ac, min_thresh, max_thresh); |
| 2266 |
| 2267 // Adjust threshold according to partition size. |
| 2268 thresh_ac >>= 8 - (b_width_log2_lookup[bsize] + |
| 2269 b_height_log2_lookup[bsize]); |
| 2270 thresh_dc = (xd->plane[0].dequant[0] * xd->plane[0].dequant[0] >> 6); |
| 2271 } else { |
| 2272 thresh_ac = 0; |
| 2273 thresh_dc = 0; |
| 2274 } |
2833 | 2275 |
2834 // Y skipping condition checking | 2276 // Y skipping condition checking |
2835 if (sse < thresh_ac || sse == 0) { | 2277 if (sse < thresh_ac || sse == 0) { |
2836 // Skipping threshold for dc | |
2837 unsigned int thresh_dc; | |
2838 | |
2839 thresh_dc = (xd->plane[0].dequant[0] * xd->plane[0].dequant[0] >> 6); | |
2840 | |
2841 // dc skipping checking | 2278 // dc skipping checking |
2842 if ((sse - var) < thresh_dc || sse == var) { | 2279 if ((sse - var) < thresh_dc || sse == var) { |
2843 unsigned int sse_u, sse_v; | 2280 unsigned int sse_u, sse_v; |
2844 unsigned int var_u, var_v; | 2281 unsigned int var_u, var_v; |
2845 | 2282 |
2846 var_u = cpi->fn_ptr[uv_size].vf(x->plane[1].src.buf, | 2283 var_u = cpi->fn_ptr[uv_size].vf(x->plane[1].src.buf, |
2847 x->plane[1].src.stride, | 2284 x->plane[1].src.stride, |
2848 xd->plane[1].dst.buf, | 2285 xd->plane[1].dst.buf, |
2849 xd->plane[1].dst.stride, &sse_u); | 2286 xd->plane[1].dst.stride, &sse_u); |
2850 | 2287 |
(...skipping 67 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
2918 restore_dst_buf(xd, orig_dst, orig_dst_stride); | 2355 restore_dst_buf(xd, orig_dst, orig_dst_stride); |
2919 return this_rd; // if 0, this will be re-calculated by caller | 2356 return this_rd; // if 0, this will be re-calculated by caller |
2920 } | 2357 } |
2921 | 2358 |
2922 void vp9_rd_pick_intra_mode_sb(VP9_COMP *cpi, MACROBLOCK *x, | 2359 void vp9_rd_pick_intra_mode_sb(VP9_COMP *cpi, MACROBLOCK *x, |
2923 int *returnrate, int64_t *returndist, | 2360 int *returnrate, int64_t *returndist, |
2924 BLOCK_SIZE bsize, | 2361 BLOCK_SIZE bsize, |
2925 PICK_MODE_CONTEXT *ctx, int64_t best_rd) { | 2362 PICK_MODE_CONTEXT *ctx, int64_t best_rd) { |
2926 VP9_COMMON *const cm = &cpi->common; | 2363 VP9_COMMON *const cm = &cpi->common; |
2927 MACROBLOCKD *const xd = &x->e_mbd; | 2364 MACROBLOCKD *const xd = &x->e_mbd; |
| 2365 struct macroblockd_plane *const pd = xd->plane; |
2928 int rate_y = 0, rate_uv = 0, rate_y_tokenonly = 0, rate_uv_tokenonly = 0; | 2366 int rate_y = 0, rate_uv = 0, rate_y_tokenonly = 0, rate_uv_tokenonly = 0; |
2929 int y_skip = 0, uv_skip = 0; | 2367 int y_skip = 0, uv_skip = 0; |
2930 int64_t dist_y = 0, dist_uv = 0, tx_cache[TX_MODES] = { 0 }; | 2368 int64_t dist_y = 0, dist_uv = 0, tx_cache[TX_MODES] = { 0 }; |
2931 TX_SIZE max_uv_tx_size; | 2369 TX_SIZE max_uv_tx_size; |
2932 x->skip_encode = 0; | 2370 x->skip_encode = 0; |
2933 ctx->skip = 0; | 2371 ctx->skip = 0; |
2934 xd->mi[0]->mbmi.ref_frame[0] = INTRA_FRAME; | 2372 xd->mi[0]->mbmi.ref_frame[0] = INTRA_FRAME; |
2935 | 2373 |
2936 if (bsize >= BLOCK_8X8) { | 2374 if (bsize >= BLOCK_8X8) { |
2937 if (rd_pick_intra_sby_mode(cpi, x, &rate_y, &rate_y_tokenonly, | 2375 if (rd_pick_intra_sby_mode(cpi, x, &rate_y, &rate_y_tokenonly, |
2938 &dist_y, &y_skip, bsize, tx_cache, | 2376 &dist_y, &y_skip, bsize, tx_cache, |
2939 best_rd) >= best_rd) { | 2377 best_rd) >= best_rd) { |
2940 *returnrate = INT_MAX; | 2378 *returnrate = INT_MAX; |
2941 return; | 2379 return; |
2942 } | 2380 } |
2943 max_uv_tx_size = get_uv_tx_size_impl(xd->mi[0]->mbmi.tx_size, bsize); | 2381 max_uv_tx_size = get_uv_tx_size_impl(xd->mi[0]->mbmi.tx_size, bsize, |
| 2382 pd[1].subsampling_x, |
| 2383 pd[1].subsampling_y); |
2944 rd_pick_intra_sbuv_mode(cpi, x, ctx, &rate_uv, &rate_uv_tokenonly, | 2384 rd_pick_intra_sbuv_mode(cpi, x, ctx, &rate_uv, &rate_uv_tokenonly, |
2945 &dist_uv, &uv_skip, bsize, max_uv_tx_size); | 2385 &dist_uv, &uv_skip, bsize, max_uv_tx_size); |
2946 } else { | 2386 } else { |
2947 y_skip = 0; | 2387 y_skip = 0; |
2948 if (rd_pick_intra_sub_8x8_y_mode(cpi, x, &rate_y, &rate_y_tokenonly, | 2388 if (rd_pick_intra_sub_8x8_y_mode(cpi, x, &rate_y, &rate_y_tokenonly, |
2949 &dist_y, best_rd) >= best_rd) { | 2389 &dist_y, best_rd) >= best_rd) { |
2950 *returnrate = INT_MAX; | 2390 *returnrate = INT_MAX; |
2951 return; | 2391 return; |
2952 } | 2392 } |
2953 max_uv_tx_size = get_uv_tx_size_impl(xd->mi[0]->mbmi.tx_size, bsize); | 2393 max_uv_tx_size = get_uv_tx_size_impl(xd->mi[0]->mbmi.tx_size, bsize, |
| 2394 pd[1].subsampling_x, |
| 2395 pd[1].subsampling_y); |
2954 rd_pick_intra_sbuv_mode(cpi, x, ctx, &rate_uv, &rate_uv_tokenonly, | 2396 rd_pick_intra_sbuv_mode(cpi, x, ctx, &rate_uv, &rate_uv_tokenonly, |
2955 &dist_uv, &uv_skip, BLOCK_8X8, max_uv_tx_size); | 2397 &dist_uv, &uv_skip, BLOCK_8X8, max_uv_tx_size); |
2956 } | 2398 } |
2957 | 2399 |
2958 if (y_skip && uv_skip) { | 2400 if (y_skip && uv_skip) { |
2959 *returnrate = rate_y + rate_uv - rate_y_tokenonly - rate_uv_tokenonly + | 2401 *returnrate = rate_y + rate_uv - rate_y_tokenonly - rate_uv_tokenonly + |
2960 vp9_cost_bit(vp9_get_skip_prob(cm, xd), 1); | 2402 vp9_cost_bit(vp9_get_skip_prob(cm, xd), 1); |
2961 *returndist = dist_y + dist_uv; | 2403 *returndist = dist_y + dist_uv; |
2962 vp9_zero(ctx->tx_rd_diff); | 2404 vp9_zero(ctx->tx_rd_diff); |
2963 } else { | 2405 } else { |
(...skipping 41 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
3005 int *returnrate, | 2447 int *returnrate, |
3006 int64_t *returndistortion, | 2448 int64_t *returndistortion, |
3007 BLOCK_SIZE bsize, | 2449 BLOCK_SIZE bsize, |
3008 PICK_MODE_CONTEXT *ctx, | 2450 PICK_MODE_CONTEXT *ctx, |
3009 int64_t best_rd_so_far) { | 2451 int64_t best_rd_so_far) { |
3010 VP9_COMMON *const cm = &cpi->common; | 2452 VP9_COMMON *const cm = &cpi->common; |
3011 RD_OPT *const rd_opt = &cpi->rd; | 2453 RD_OPT *const rd_opt = &cpi->rd; |
3012 MACROBLOCKD *const xd = &x->e_mbd; | 2454 MACROBLOCKD *const xd = &x->e_mbd; |
3013 MB_MODE_INFO *const mbmi = &xd->mi[0]->mbmi; | 2455 MB_MODE_INFO *const mbmi = &xd->mi[0]->mbmi; |
3014 const struct segmentation *const seg = &cm->seg; | 2456 const struct segmentation *const seg = &cm->seg; |
| 2457 struct macroblockd_plane *const pd = xd->plane; |
3015 PREDICTION_MODE this_mode; | 2458 PREDICTION_MODE this_mode; |
3016 MV_REFERENCE_FRAME ref_frame, second_ref_frame; | 2459 MV_REFERENCE_FRAME ref_frame, second_ref_frame; |
3017 unsigned char segment_id = mbmi->segment_id; | 2460 unsigned char segment_id = mbmi->segment_id; |
3018 int comp_pred, i; | 2461 int comp_pred, i; |
3019 int_mv frame_mv[MB_MODE_COUNT][MAX_REF_FRAMES]; | 2462 int_mv frame_mv[MB_MODE_COUNT][MAX_REF_FRAMES]; |
3020 struct buf_2d yv12_mb[4][MAX_MB_PLANE]; | 2463 struct buf_2d yv12_mb[4][MAX_MB_PLANE]; |
3021 int_mv single_newmv[MAX_REF_FRAMES] = { { 0 } }; | 2464 int_mv single_newmv[MAX_REF_FRAMES] = { { 0 } }; |
3022 static const int flag_list[4] = { 0, VP9_LAST_FLAG, VP9_GOLD_FLAG, | 2465 static const int flag_list[4] = { 0, VP9_LAST_FLAG, VP9_GOLD_FLAG, |
3023 VP9_ALT_FLAG }; | 2466 VP9_ALT_FLAG }; |
3024 int64_t best_rd = best_rd_so_far; | 2467 int64_t best_rd = best_rd_so_far; |
(...skipping 44 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
3069 for (i = 0; i < TX_SIZES; i++) | 2512 for (i = 0; i < TX_SIZES; i++) |
3070 rate_uv_intra[i] = INT_MAX; | 2513 rate_uv_intra[i] = INT_MAX; |
3071 for (i = 0; i < MAX_REF_FRAMES; ++i) | 2514 for (i = 0; i < MAX_REF_FRAMES; ++i) |
3072 x->pred_sse[i] = INT_MAX; | 2515 x->pred_sse[i] = INT_MAX; |
3073 | 2516 |
3074 *returnrate = INT_MAX; | 2517 *returnrate = INT_MAX; |
3075 | 2518 |
3076 for (ref_frame = LAST_FRAME; ref_frame <= ALTREF_FRAME; ++ref_frame) { | 2519 for (ref_frame = LAST_FRAME; ref_frame <= ALTREF_FRAME; ++ref_frame) { |
3077 x->pred_mv_sad[ref_frame] = INT_MAX; | 2520 x->pred_mv_sad[ref_frame] = INT_MAX; |
3078 if (cpi->ref_frame_flags & flag_list[ref_frame]) { | 2521 if (cpi->ref_frame_flags & flag_list[ref_frame]) { |
3079 vp9_setup_buffer_inter(cpi, x, tile, | 2522 setup_buffer_inter(cpi, x, tile, |
3080 ref_frame, bsize, mi_row, mi_col, | 2523 ref_frame, bsize, mi_row, mi_col, |
3081 frame_mv[NEARESTMV], frame_mv[NEARMV], yv12_mb); | 2524 frame_mv[NEARESTMV], frame_mv[NEARMV], yv12_mb); |
3082 } | 2525 } |
3083 frame_mv[NEWMV][ref_frame].as_int = INVALID_MV; | 2526 frame_mv[NEWMV][ref_frame].as_int = INVALID_MV; |
3084 frame_mv[ZEROMV][ref_frame].as_int = 0; | 2527 frame_mv[ZEROMV][ref_frame].as_int = 0; |
3085 } | 2528 } |
3086 | 2529 |
3087 for (ref_frame = LAST_FRAME; ref_frame <= ALTREF_FRAME; ++ref_frame) { | 2530 for (ref_frame = LAST_FRAME; ref_frame <= ALTREF_FRAME; ++ref_frame) { |
3088 // All modes from vp9_mode_order that use this frame as any ref | 2531 // All modes from vp9_mode_order that use this frame as any ref |
3089 static const int ref_frame_mask_all[] = { | 2532 static const int ref_frame_mask_all[] = { |
(...skipping 17 matching lines...) Expand all Loading... |
3107 } | 2550 } |
3108 } | 2551 } |
3109 // If the segment reference frame feature is enabled.... | 2552 // If the segment reference frame feature is enabled.... |
3110 // then do nothing if the current ref frame is not allowed.. | 2553 // then do nothing if the current ref frame is not allowed.. |
3111 if (vp9_segfeature_active(seg, segment_id, SEG_LVL_REF_FRAME) && | 2554 if (vp9_segfeature_active(seg, segment_id, SEG_LVL_REF_FRAME) && |
3112 vp9_get_segdata(seg, segment_id, SEG_LVL_REF_FRAME) != (int)ref_frame) { | 2555 vp9_get_segdata(seg, segment_id, SEG_LVL_REF_FRAME) != (int)ref_frame) { |
3113 mode_skip_mask |= ref_frame_mask_all[ref_frame]; | 2556 mode_skip_mask |= ref_frame_mask_all[ref_frame]; |
3114 } | 2557 } |
3115 } | 2558 } |
3116 | 2559 |
3117 // If the segment skip feature is enabled.... | |
3118 // then do nothing if the current mode is not allowed.. | |
3119 if (vp9_segfeature_active(seg, segment_id, SEG_LVL_SKIP)) { | |
3120 mode_skip_mask = ~(1 << THR_ZEROMV); | |
3121 inter_mode_mask = (1 << ZEROMV); | |
3122 } | |
3123 | |
3124 // Disable this drop out case if the ref frame | 2560 // Disable this drop out case if the ref frame |
3125 // segment level feature is enabled for this segment. This is to | 2561 // segment level feature is enabled for this segment. This is to |
3126 // prevent the possibility that we end up unable to pick any mode. | 2562 // prevent the possibility that we end up unable to pick any mode. |
3127 if (!vp9_segfeature_active(seg, segment_id, SEG_LVL_REF_FRAME)) { | 2563 if (!vp9_segfeature_active(seg, segment_id, SEG_LVL_REF_FRAME)) { |
3128 // Only consider ZEROMV/ALTREF_FRAME for alt ref frame, | 2564 // Only consider ZEROMV/ALTREF_FRAME for alt ref frame, |
3129 // unless ARNR filtering is enabled in which case we want | 2565 // unless ARNR filtering is enabled in which case we want |
3130 // an unfiltered alternative. We allow near/nearest as well | 2566 // an unfiltered alternative. We allow near/nearest as well |
3131 // because they may result in zero-zero MVs but be cheaper. | 2567 // because they may result in zero-zero MVs but be cheaper. |
3132 if (cpi->rc.is_src_frame_alt_ref && (cpi->oxcf.arnr_max_frames == 0)) { | 2568 if (cpi->rc.is_src_frame_alt_ref && (cpi->oxcf.arnr_max_frames == 0)) { |
3133 mode_skip_mask = | 2569 mode_skip_mask = |
(...skipping 18 matching lines...) Expand all Loading... |
3152 } | 2588 } |
3153 | 2589 |
3154 if (bsize > cpi->sf.max_intra_bsize) { | 2590 if (bsize > cpi->sf.max_intra_bsize) { |
3155 const int all_intra_modes = (1 << THR_DC) | (1 << THR_TM) | | 2591 const int all_intra_modes = (1 << THR_DC) | (1 << THR_TM) | |
3156 (1 << THR_H_PRED) | (1 << THR_V_PRED) | (1 << THR_D135_PRED) | | 2592 (1 << THR_H_PRED) | (1 << THR_V_PRED) | (1 << THR_D135_PRED) | |
3157 (1 << THR_D207_PRED) | (1 << THR_D153_PRED) | (1 << THR_D63_PRED) | | 2593 (1 << THR_D207_PRED) | (1 << THR_D153_PRED) | (1 << THR_D63_PRED) | |
3158 (1 << THR_D117_PRED) | (1 << THR_D45_PRED); | 2594 (1 << THR_D117_PRED) | (1 << THR_D45_PRED); |
3159 mode_skip_mask |= all_intra_modes; | 2595 mode_skip_mask |= all_intra_modes; |
3160 } | 2596 } |
3161 | 2597 |
3162 if (!x->in_active_map) { | |
3163 int mode_index; | |
3164 assert(cpi->ref_frame_flags & VP9_LAST_FLAG); | |
3165 if (frame_mv[NEARESTMV][LAST_FRAME].as_int == 0) | |
3166 mode_index = THR_NEARESTMV; | |
3167 else if (frame_mv[NEARMV][LAST_FRAME].as_int == 0) | |
3168 mode_index = THR_NEARMV; | |
3169 else | |
3170 mode_index = THR_ZEROMV; | |
3171 mode_skip_mask = ~(1 << mode_index); | |
3172 mode_skip_start = MAX_MODES; | |
3173 inter_mode_mask = (1 << NEARESTMV) | (1 << NEARMV) | (1 << ZEROMV) | | |
3174 (1 << NEWMV); | |
3175 } | |
3176 | |
3177 for (mode_index = 0; mode_index < MAX_MODES; ++mode_index) { | 2598 for (mode_index = 0; mode_index < MAX_MODES; ++mode_index) { |
3178 int mode_excluded = 0; | 2599 int mode_excluded = 0; |
3179 int64_t this_rd = INT64_MAX; | 2600 int64_t this_rd = INT64_MAX; |
3180 int disable_skip = 0; | 2601 int disable_skip = 0; |
3181 int compmode_cost = 0; | 2602 int compmode_cost = 0; |
3182 int rate2 = 0, rate_y = 0, rate_uv = 0; | 2603 int rate2 = 0, rate_y = 0, rate_uv = 0; |
3183 int64_t distortion2 = 0, distortion_y = 0, distortion_uv = 0; | 2604 int64_t distortion2 = 0, distortion_y = 0, distortion_uv = 0; |
3184 int skippable = 0; | 2605 int skippable = 0; |
3185 int64_t tx_cache[TX_MODES]; | 2606 int64_t tx_cache[TX_MODES]; |
3186 int i; | 2607 int i; |
(...skipping 69 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
3256 if (best_mode_index >= 0 && | 2677 if (best_mode_index >= 0 && |
3257 vp9_mode_order[best_mode_index].ref_frame[0] > INTRA_FRAME) | 2678 vp9_mode_order[best_mode_index].ref_frame[0] > INTRA_FRAME) |
3258 continue; | 2679 continue; |
3259 } | 2680 } |
3260 if (mode_search_skip_flags & FLAG_SKIP_INTRA_DIRMISMATCH) { | 2681 if (mode_search_skip_flags & FLAG_SKIP_INTRA_DIRMISMATCH) { |
3261 if (conditional_skipintra(this_mode, best_intra_mode)) | 2682 if (conditional_skipintra(this_mode, best_intra_mode)) |
3262 continue; | 2683 continue; |
3263 } | 2684 } |
3264 } | 2685 } |
3265 } else { | 2686 } else { |
3266 if (x->in_active_map && | 2687 const MV_REFERENCE_FRAME ref_frames[2] = {ref_frame, second_ref_frame}; |
3267 !vp9_segfeature_active(&cm->seg, mbmi->segment_id, SEG_LVL_SKIP)) { | 2688 if (!check_best_zero_mv(cpi, mbmi->mode_context, frame_mv, |
3268 const MV_REFERENCE_FRAME ref_frames[2] = {ref_frame, second_ref_frame}; | 2689 inter_mode_mask, this_mode, ref_frames)) |
3269 if (!check_best_zero_mv(cpi, mbmi->mode_context, frame_mv, | 2690 continue; |
3270 inter_mode_mask, this_mode, ref_frames)) | |
3271 continue; | |
3272 } | |
3273 } | 2691 } |
3274 | 2692 |
3275 mbmi->mode = this_mode; | 2693 mbmi->mode = this_mode; |
3276 mbmi->uv_mode = x->in_active_map ? DC_PRED : this_mode; | 2694 mbmi->uv_mode = DC_PRED; |
3277 mbmi->ref_frame[0] = ref_frame; | 2695 mbmi->ref_frame[0] = ref_frame; |
3278 mbmi->ref_frame[1] = second_ref_frame; | 2696 mbmi->ref_frame[1] = second_ref_frame; |
3279 // Evaluate all sub-pel filters irrespective of whether we can use | 2697 // Evaluate all sub-pel filters irrespective of whether we can use |
3280 // them for this frame. | 2698 // them for this frame. |
3281 mbmi->interp_filter = cm->interp_filter == SWITCHABLE ? EIGHTTAP | 2699 mbmi->interp_filter = cm->interp_filter == SWITCHABLE ? EIGHTTAP |
3282 : cm->interp_filter; | 2700 : cm->interp_filter; |
3283 x->skip = 0; | 2701 x->skip = 0; |
3284 set_ref_ptrs(cm, xd, ref_frame, second_ref_frame); | 2702 set_ref_ptrs(cm, xd, ref_frame, second_ref_frame); |
3285 | 2703 |
3286 // Select prediction reference frames. | 2704 // Select prediction reference frames. |
3287 for (i = 0; i < MAX_MB_PLANE; i++) { | 2705 for (i = 0; i < MAX_MB_PLANE; i++) { |
3288 xd->plane[i].pre[0] = yv12_mb[ref_frame][i]; | 2706 xd->plane[i].pre[0] = yv12_mb[ref_frame][i]; |
3289 if (comp_pred) | 2707 if (comp_pred) |
3290 xd->plane[i].pre[1] = yv12_mb[second_ref_frame][i]; | 2708 xd->plane[i].pre[1] = yv12_mb[second_ref_frame][i]; |
3291 } | 2709 } |
3292 | 2710 |
3293 for (i = 0; i < TX_MODES; ++i) | 2711 for (i = 0; i < TX_MODES; ++i) |
3294 tx_cache[i] = INT64_MAX; | 2712 tx_cache[i] = INT64_MAX; |
3295 | 2713 |
3296 if (ref_frame == INTRA_FRAME) { | 2714 if (ref_frame == INTRA_FRAME) { |
3297 TX_SIZE uv_tx; | 2715 TX_SIZE uv_tx; |
3298 intra_super_block_yrd(cpi, x, &rate_y, &distortion_y, &skippable, NULL, | 2716 intra_super_block_yrd(cpi, x, &rate_y, &distortion_y, &skippable, NULL, |
3299 bsize, tx_cache, best_rd); | 2717 bsize, tx_cache, best_rd); |
3300 | 2718 |
3301 if (rate_y == INT_MAX) | 2719 if (rate_y == INT_MAX) |
3302 continue; | 2720 continue; |
3303 | 2721 |
3304 uv_tx = get_uv_tx_size_impl(mbmi->tx_size, bsize); | 2722 uv_tx = get_uv_tx_size_impl(mbmi->tx_size, bsize, pd[1].subsampling_x, |
| 2723 pd[1].subsampling_y); |
3305 if (rate_uv_intra[uv_tx] == INT_MAX) { | 2724 if (rate_uv_intra[uv_tx] == INT_MAX) { |
3306 choose_intra_uv_mode(cpi, ctx, bsize, uv_tx, | 2725 choose_intra_uv_mode(cpi, ctx, bsize, uv_tx, |
3307 &rate_uv_intra[uv_tx], &rate_uv_tokenonly[uv_tx], | 2726 &rate_uv_intra[uv_tx], &rate_uv_tokenonly[uv_tx], |
3308 &dist_uv[uv_tx], &skip_uv[uv_tx], &mode_uv[uv_tx]); | 2727 &dist_uv[uv_tx], &skip_uv[uv_tx], &mode_uv[uv_tx]); |
3309 } | 2728 } |
3310 | 2729 |
3311 rate_uv = rate_uv_tokenonly[uv_tx]; | 2730 rate_uv = rate_uv_tokenonly[uv_tx]; |
3312 distortion_uv = dist_uv[uv_tx]; | 2731 distortion_uv = dist_uv[uv_tx]; |
3313 skippable = skippable && skip_uv[uv_tx]; | 2732 skippable = skippable && skip_uv[uv_tx]; |
3314 mbmi->uv_mode = mode_uv[uv_tx]; | 2733 mbmi->uv_mode = mode_uv[uv_tx]; |
(...skipping 23 matching lines...) Expand all Loading... |
3338 | 2757 |
3339 // Estimate the reference frame signaling cost and add it | 2758 // Estimate the reference frame signaling cost and add it |
3340 // to the rolling cost variable. | 2759 // to the rolling cost variable. |
3341 if (comp_pred) { | 2760 if (comp_pred) { |
3342 rate2 += ref_costs_comp[ref_frame]; | 2761 rate2 += ref_costs_comp[ref_frame]; |
3343 } else { | 2762 } else { |
3344 rate2 += ref_costs_single[ref_frame]; | 2763 rate2 += ref_costs_single[ref_frame]; |
3345 } | 2764 } |
3346 | 2765 |
3347 if (!disable_skip) { | 2766 if (!disable_skip) { |
3348 // Test for the condition where skip block will be activated | 2767 if (skippable) { |
3349 // because there are no non zero coefficients and make any | 2768 vp9_prob skip_prob = vp9_get_skip_prob(cm, xd); |
3350 // necessary adjustment for rate. Ignore if skip is coded at | |
3351 // segment level as the cost wont have been added in. | |
3352 // Is Mb level skip allowed (i.e. not coded at segment level). | |
3353 const int mb_skip_allowed = !vp9_segfeature_active(seg, segment_id, | |
3354 SEG_LVL_SKIP); | |
3355 | 2769 |
3356 if (skippable) { | |
3357 // Back out the coefficient coding costs | 2770 // Back out the coefficient coding costs |
3358 rate2 -= (rate_y + rate_uv); | 2771 rate2 -= (rate_y + rate_uv); |
3359 // for best yrd calculation | 2772 // for best yrd calculation |
3360 rate_uv = 0; | 2773 rate_uv = 0; |
3361 | 2774 |
3362 if (mb_skip_allowed) { | 2775 // Cost the skip mb case |
3363 int prob_skip_cost; | 2776 if (skip_prob) { |
3364 | 2777 int prob_skip_cost = vp9_cost_bit(skip_prob, 1); |
3365 // Cost the skip mb case | 2778 rate2 += prob_skip_cost; |
3366 vp9_prob skip_prob = vp9_get_skip_prob(cm, xd); | |
3367 if (skip_prob) { | |
3368 prob_skip_cost = vp9_cost_bit(skip_prob, 1); | |
3369 rate2 += prob_skip_cost; | |
3370 } | |
3371 } | 2779 } |
3372 } else if (mb_skip_allowed && ref_frame != INTRA_FRAME && !xd->lossless) { | 2780 } else if (ref_frame != INTRA_FRAME && !xd->lossless) { |
3373 if (RDCOST(x->rdmult, x->rddiv, rate_y + rate_uv, distortion2) < | 2781 if (RDCOST(x->rdmult, x->rddiv, rate_y + rate_uv, distortion2) < |
3374 RDCOST(x->rdmult, x->rddiv, 0, total_sse)) { | 2782 RDCOST(x->rdmult, x->rddiv, 0, total_sse)) { |
3375 // Add in the cost of the no skip flag. | 2783 // Add in the cost of the no skip flag. |
3376 rate2 += vp9_cost_bit(vp9_get_skip_prob(cm, xd), 0); | 2784 rate2 += vp9_cost_bit(vp9_get_skip_prob(cm, xd), 0); |
3377 } else { | 2785 } else { |
3378 // FIXME(rbultje) make this work for splitmv also | 2786 // FIXME(rbultje) make this work for splitmv also |
3379 rate2 += vp9_cost_bit(vp9_get_skip_prob(cm, xd), 1); | 2787 rate2 += vp9_cost_bit(vp9_get_skip_prob(cm, xd), 1); |
3380 distortion2 = total_sse; | 2788 distortion2 = total_sse; |
3381 assert(total_sse >= 0); | 2789 assert(total_sse >= 0); |
3382 rate2 -= (rate_y + rate_uv); | 2790 rate2 -= (rate_y + rate_uv); |
3383 rate_y = 0; | 2791 rate_y = 0; |
3384 rate_uv = 0; | 2792 rate_uv = 0; |
3385 this_skip2 = 1; | 2793 this_skip2 = 1; |
3386 } | 2794 } |
3387 } else if (mb_skip_allowed) { | 2795 } else { |
3388 // Add in the cost of the no skip flag. | 2796 // Add in the cost of the no skip flag. |
3389 rate2 += vp9_cost_bit(vp9_get_skip_prob(cm, xd), 0); | 2797 rate2 += vp9_cost_bit(vp9_get_skip_prob(cm, xd), 0); |
3390 } | 2798 } |
3391 | 2799 |
3392 // Calculate the final RD estimate for this mode. | 2800 // Calculate the final RD estimate for this mode. |
3393 this_rd = RDCOST(x->rdmult, x->rddiv, rate2, distortion2); | 2801 this_rd = RDCOST(x->rdmult, x->rddiv, rate2, distortion2); |
3394 } | 2802 } |
3395 | 2803 |
3396 if (ref_frame == INTRA_FRAME) { | 2804 if (ref_frame == INTRA_FRAME) { |
3397 // Keep record of best intra rd | 2805 // Keep record of best intra rd |
(...skipping 141 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
3539 | 2947 |
3540 if (best_mode_index < 0 || best_rd >= best_rd_so_far) | 2948 if (best_mode_index < 0 || best_rd >= best_rd_so_far) |
3541 return INT64_MAX; | 2949 return INT64_MAX; |
3542 | 2950 |
3543 // If we used an estimate for the uv intra rd in the loop above... | 2951 // If we used an estimate for the uv intra rd in the loop above... |
3544 if (cpi->sf.use_uv_intra_rd_estimate) { | 2952 if (cpi->sf.use_uv_intra_rd_estimate) { |
3545 // Do Intra UV best rd mode selection if best mode choice above was intra. | 2953 // Do Intra UV best rd mode selection if best mode choice above was intra. |
3546 if (vp9_mode_order[best_mode_index].ref_frame[0] == INTRA_FRAME) { | 2954 if (vp9_mode_order[best_mode_index].ref_frame[0] == INTRA_FRAME) { |
3547 TX_SIZE uv_tx_size; | 2955 TX_SIZE uv_tx_size; |
3548 *mbmi = best_mbmode; | 2956 *mbmi = best_mbmode; |
3549 uv_tx_size = get_uv_tx_size(mbmi); | 2957 uv_tx_size = get_uv_tx_size(mbmi, &xd->plane[1]); |
3550 rd_pick_intra_sbuv_mode(cpi, x, ctx, &rate_uv_intra[uv_tx_size], | 2958 rd_pick_intra_sbuv_mode(cpi, x, ctx, &rate_uv_intra[uv_tx_size], |
3551 &rate_uv_tokenonly[uv_tx_size], | 2959 &rate_uv_tokenonly[uv_tx_size], |
3552 &dist_uv[uv_tx_size], | 2960 &dist_uv[uv_tx_size], |
3553 &skip_uv[uv_tx_size], | 2961 &skip_uv[uv_tx_size], |
3554 bsize < BLOCK_8X8 ? BLOCK_8X8 : bsize, | 2962 bsize < BLOCK_8X8 ? BLOCK_8X8 : bsize, |
3555 uv_tx_size); | 2963 uv_tx_size); |
3556 } | 2964 } |
3557 } | 2965 } |
3558 | 2966 |
3559 assert((cm->interp_filter == SWITCHABLE) || | 2967 assert((cm->interp_filter == SWITCHABLE) || |
(...skipping 26 matching lines...) Expand all Loading... |
3586 if (best_tx_rd[i] == INT64_MAX) | 2994 if (best_tx_rd[i] == INT64_MAX) |
3587 best_tx_diff[i] = 0; | 2995 best_tx_diff[i] = 0; |
3588 else | 2996 else |
3589 best_tx_diff[i] = best_rd - best_tx_rd[i]; | 2997 best_tx_diff[i] = best_rd - best_tx_rd[i]; |
3590 } | 2998 } |
3591 } else { | 2999 } else { |
3592 vp9_zero(best_filter_diff); | 3000 vp9_zero(best_filter_diff); |
3593 vp9_zero(best_tx_diff); | 3001 vp9_zero(best_tx_diff); |
3594 } | 3002 } |
3595 | 3003 |
3596 if (!x->in_active_map) { | |
3597 assert(mbmi->ref_frame[0] == LAST_FRAME); | |
3598 assert(mbmi->ref_frame[1] == NONE); | |
3599 assert(mbmi->mode == NEARESTMV || | |
3600 mbmi->mode == NEARMV || | |
3601 mbmi->mode == ZEROMV); | |
3602 assert(frame_mv[mbmi->mode][LAST_FRAME].as_int == 0); | |
3603 assert(mbmi->mode == mbmi->uv_mode); | |
3604 } | |
3605 | |
3606 set_ref_ptrs(cm, xd, mbmi->ref_frame[0], mbmi->ref_frame[1]); | 3004 set_ref_ptrs(cm, xd, mbmi->ref_frame[0], mbmi->ref_frame[1]); |
3607 store_coding_context(x, ctx, best_mode_index, | 3005 store_coding_context(x, ctx, best_mode_index, |
3608 best_pred_diff, best_tx_diff, best_filter_diff); | 3006 best_pred_diff, best_tx_diff, best_filter_diff); |
3609 | 3007 |
3610 return best_rd; | 3008 return best_rd; |
3611 } | 3009 } |
3612 | 3010 |
| 3011 int64_t vp9_rd_pick_inter_mode_sb_seg_skip(VP9_COMP *cpi, MACROBLOCK *x, |
| 3012 int *returnrate, |
| 3013 int64_t *returndistortion, |
| 3014 BLOCK_SIZE bsize, |
| 3015 PICK_MODE_CONTEXT *ctx, |
| 3016 int64_t best_rd_so_far) { |
| 3017 VP9_COMMON *const cm = &cpi->common; |
| 3018 RD_OPT *const rd_opt = &cpi->rd; |
| 3019 MACROBLOCKD *const xd = &x->e_mbd; |
| 3020 MB_MODE_INFO *const mbmi = &xd->mi[0]->mbmi; |
| 3021 const struct segmentation *const seg = &cm->seg; |
| 3022 unsigned char segment_id = mbmi->segment_id; |
| 3023 const int comp_pred = 0; |
| 3024 int i; |
| 3025 int64_t best_tx_diff[TX_MODES]; |
| 3026 int64_t best_pred_diff[REFERENCE_MODES]; |
| 3027 int64_t best_filter_diff[SWITCHABLE_FILTER_CONTEXTS]; |
| 3028 unsigned int ref_costs_single[MAX_REF_FRAMES], ref_costs_comp[MAX_REF_FRAMES]; |
| 3029 vp9_prob comp_mode_p; |
| 3030 INTERP_FILTER best_filter = SWITCHABLE; |
| 3031 int64_t this_rd = INT64_MAX; |
| 3032 int rate2 = 0; |
| 3033 const int64_t distortion2 = 0; |
| 3034 |
| 3035 x->skip_encode = cpi->sf.skip_encode_frame && x->q_index < QIDX_SKIP_THRESH; |
| 3036 |
| 3037 estimate_ref_frame_costs(cm, xd, segment_id, ref_costs_single, ref_costs_comp, |
| 3038 &comp_mode_p); |
| 3039 |
| 3040 for (i = 0; i < MAX_REF_FRAMES; ++i) |
| 3041 x->pred_sse[i] = INT_MAX; |
| 3042 for (i = LAST_FRAME; i < MAX_REF_FRAMES; ++i) |
| 3043 x->pred_mv_sad[i] = INT_MAX; |
| 3044 |
| 3045 *returnrate = INT_MAX; |
| 3046 |
| 3047 assert(vp9_segfeature_active(seg, segment_id, SEG_LVL_SKIP)); |
| 3048 |
| 3049 mbmi->mode = ZEROMV; |
| 3050 mbmi->uv_mode = DC_PRED; |
| 3051 mbmi->ref_frame[0] = LAST_FRAME; |
| 3052 mbmi->ref_frame[1] = NONE; |
| 3053 mbmi->mv[0].as_int = 0; |
| 3054 x->skip = 1; |
| 3055 |
| 3056 // Search for best switchable filter by checking the variance of |
| 3057 // pred error irrespective of whether the filter will be used |
| 3058 rd_opt->mask_filter = 0; |
| 3059 for (i = 0; i < SWITCHABLE_FILTER_CONTEXTS; ++i) |
| 3060 rd_opt->filter_cache[i] = INT64_MAX; |
| 3061 |
| 3062 if (cm->interp_filter != BILINEAR) { |
| 3063 best_filter = EIGHTTAP; |
| 3064 if (cm->interp_filter == SWITCHABLE && |
| 3065 x->source_variance >= cpi->sf.disable_filter_search_var_thresh) { |
| 3066 int rs; |
| 3067 int best_rs = INT_MAX; |
| 3068 for (i = 0; i < SWITCHABLE_FILTERS; ++i) { |
| 3069 mbmi->interp_filter = i; |
| 3070 rs = vp9_get_switchable_rate(cpi); |
| 3071 if (rs < best_rs) { |
| 3072 best_rs = rs; |
| 3073 best_filter = mbmi->interp_filter; |
| 3074 } |
| 3075 } |
| 3076 } |
| 3077 } |
| 3078 // Set the appropriate filter |
| 3079 if (cm->interp_filter == SWITCHABLE) { |
| 3080 mbmi->interp_filter = best_filter; |
| 3081 rate2 += vp9_get_switchable_rate(cpi); |
| 3082 } else { |
| 3083 mbmi->interp_filter = cm->interp_filter; |
| 3084 } |
| 3085 |
| 3086 if (cm->reference_mode == REFERENCE_MODE_SELECT) |
| 3087 rate2 += vp9_cost_bit(comp_mode_p, comp_pred); |
| 3088 |
| 3089 // Estimate the reference frame signaling cost and add it |
| 3090 // to the rolling cost variable. |
| 3091 rate2 += ref_costs_single[LAST_FRAME]; |
| 3092 this_rd = RDCOST(x->rdmult, x->rddiv, rate2, distortion2); |
| 3093 |
| 3094 *returnrate = rate2; |
| 3095 *returndistortion = distortion2; |
| 3096 |
| 3097 if (this_rd >= best_rd_so_far) |
| 3098 return INT64_MAX; |
| 3099 |
| 3100 assert((cm->interp_filter == SWITCHABLE) || |
| 3101 (cm->interp_filter == mbmi->interp_filter)); |
| 3102 |
| 3103 update_rd_thresh_fact(cpi, bsize, THR_ZEROMV); |
| 3104 |
| 3105 vp9_zero(best_pred_diff); |
| 3106 vp9_zero(best_filter_diff); |
| 3107 vp9_zero(best_tx_diff); |
| 3108 |
| 3109 if (!x->select_tx_size) |
| 3110 swap_block_ptr(x, ctx, 1, 0, 0, MAX_MB_PLANE); |
| 3111 store_coding_context(x, ctx, THR_ZEROMV, |
| 3112 best_pred_diff, best_tx_diff, best_filter_diff); |
| 3113 |
| 3114 return this_rd; |
| 3115 } |
3613 | 3116 |
3614 int64_t vp9_rd_pick_inter_mode_sub8x8(VP9_COMP *cpi, MACROBLOCK *x, | 3117 int64_t vp9_rd_pick_inter_mode_sub8x8(VP9_COMP *cpi, MACROBLOCK *x, |
3615 const TileInfo *const tile, | 3118 const TileInfo *const tile, |
3616 int mi_row, int mi_col, | 3119 int mi_row, int mi_col, |
3617 int *returnrate, | 3120 int *returnrate, |
3618 int64_t *returndistortion, | 3121 int64_t *returndistortion, |
3619 BLOCK_SIZE bsize, | 3122 BLOCK_SIZE bsize, |
3620 PICK_MODE_CONTEXT *ctx, | 3123 PICK_MODE_CONTEXT *ctx, |
3621 int64_t best_rd_so_far) { | 3124 int64_t best_rd_so_far) { |
3622 VP9_COMMON *const cm = &cpi->common; | 3125 VP9_COMMON *const cm = &cpi->common; |
(...skipping 48 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
3671 for (i = 0; i < REFERENCE_MODES; ++i) | 3174 for (i = 0; i < REFERENCE_MODES; ++i) |
3672 best_pred_rd[i] = INT64_MAX; | 3175 best_pred_rd[i] = INT64_MAX; |
3673 for (i = 0; i < SWITCHABLE_FILTER_CONTEXTS; i++) | 3176 for (i = 0; i < SWITCHABLE_FILTER_CONTEXTS; i++) |
3674 best_filter_rd[i] = INT64_MAX; | 3177 best_filter_rd[i] = INT64_MAX; |
3675 rate_uv_intra = INT_MAX; | 3178 rate_uv_intra = INT_MAX; |
3676 | 3179 |
3677 *returnrate = INT_MAX; | 3180 *returnrate = INT_MAX; |
3678 | 3181 |
3679 for (ref_frame = LAST_FRAME; ref_frame <= ALTREF_FRAME; ref_frame++) { | 3182 for (ref_frame = LAST_FRAME; ref_frame <= ALTREF_FRAME; ref_frame++) { |
3680 if (cpi->ref_frame_flags & flag_list[ref_frame]) { | 3183 if (cpi->ref_frame_flags & flag_list[ref_frame]) { |
3681 vp9_setup_buffer_inter(cpi, x, tile, | 3184 setup_buffer_inter(cpi, x, tile, |
3682 ref_frame, bsize, mi_row, mi_col, | 3185 ref_frame, bsize, mi_row, mi_col, |
3683 frame_mv[NEARESTMV], frame_mv[NEARMV], | 3186 frame_mv[NEARESTMV], frame_mv[NEARMV], |
3684 yv12_mb); | 3187 yv12_mb); |
3685 } | 3188 } |
3686 frame_mv[NEWMV][ref_frame].as_int = INVALID_MV; | 3189 frame_mv[NEWMV][ref_frame].as_int = INVALID_MV; |
3687 frame_mv[ZEROMV][ref_frame].as_int = 0; | 3190 frame_mv[ZEROMV][ref_frame].as_int = 0; |
3688 } | 3191 } |
3689 | 3192 |
3690 for (ref_index = 0; ref_index < MAX_REFS; ++ref_index) { | 3193 for (ref_index = 0; ref_index < MAX_REFS; ++ref_index) { |
3691 int mode_excluded = 0; | 3194 int mode_excluded = 0; |
(...skipping 518 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
4210 } else { | 3713 } else { |
4211 vp9_zero(best_filter_diff); | 3714 vp9_zero(best_filter_diff); |
4212 } | 3715 } |
4213 | 3716 |
4214 set_ref_ptrs(cm, xd, mbmi->ref_frame[0], mbmi->ref_frame[1]); | 3717 set_ref_ptrs(cm, xd, mbmi->ref_frame[0], mbmi->ref_frame[1]); |
4215 store_coding_context(x, ctx, best_ref_index, | 3718 store_coding_context(x, ctx, best_ref_index, |
4216 best_pred_diff, best_tx_diff, best_filter_diff); | 3719 best_pred_diff, best_tx_diff, best_filter_diff); |
4217 | 3720 |
4218 return best_rd; | 3721 return best_rd; |
4219 } | 3722 } |
4220 | |
4221 void vp9_set_rd_speed_thresholds(VP9_COMP *cpi) { | |
4222 int i; | |
4223 RD_OPT *const rd = &cpi->rd; | |
4224 | |
4225 // Set baseline threshold values | |
4226 for (i = 0; i < MAX_MODES; ++i) | |
4227 rd->thresh_mult[i] = is_best_mode(cpi->oxcf.mode) ? -500 : 0; | |
4228 | |
4229 rd->thresh_mult[THR_NEARESTMV] = 0; | |
4230 rd->thresh_mult[THR_NEARESTG] = 0; | |
4231 rd->thresh_mult[THR_NEARESTA] = 0; | |
4232 | |
4233 rd->thresh_mult[THR_DC] += 1000; | |
4234 | |
4235 rd->thresh_mult[THR_NEWMV] += 1000; | |
4236 rd->thresh_mult[THR_NEWA] += 1000; | |
4237 rd->thresh_mult[THR_NEWG] += 1000; | |
4238 | |
4239 rd->thresh_mult[THR_NEARMV] += 1000; | |
4240 rd->thresh_mult[THR_NEARA] += 1000; | |
4241 rd->thresh_mult[THR_COMP_NEARESTLA] += 1000; | |
4242 rd->thresh_mult[THR_COMP_NEARESTGA] += 1000; | |
4243 | |
4244 rd->thresh_mult[THR_TM] += 1000; | |
4245 | |
4246 rd->thresh_mult[THR_COMP_NEARLA] += 1500; | |
4247 rd->thresh_mult[THR_COMP_NEWLA] += 2000; | |
4248 rd->thresh_mult[THR_NEARG] += 1000; | |
4249 rd->thresh_mult[THR_COMP_NEARGA] += 1500; | |
4250 rd->thresh_mult[THR_COMP_NEWGA] += 2000; | |
4251 | |
4252 rd->thresh_mult[THR_ZEROMV] += 2000; | |
4253 rd->thresh_mult[THR_ZEROG] += 2000; | |
4254 rd->thresh_mult[THR_ZEROA] += 2000; | |
4255 rd->thresh_mult[THR_COMP_ZEROLA] += 2500; | |
4256 rd->thresh_mult[THR_COMP_ZEROGA] += 2500; | |
4257 | |
4258 rd->thresh_mult[THR_H_PRED] += 2000; | |
4259 rd->thresh_mult[THR_V_PRED] += 2000; | |
4260 rd->thresh_mult[THR_D45_PRED ] += 2500; | |
4261 rd->thresh_mult[THR_D135_PRED] += 2500; | |
4262 rd->thresh_mult[THR_D117_PRED] += 2500; | |
4263 rd->thresh_mult[THR_D153_PRED] += 2500; | |
4264 rd->thresh_mult[THR_D207_PRED] += 2500; | |
4265 rd->thresh_mult[THR_D63_PRED] += 2500; | |
4266 | |
4267 /* disable frame modes if flags not set */ | |
4268 if (!(cpi->ref_frame_flags & VP9_LAST_FLAG)) { | |
4269 rd->thresh_mult[THR_NEWMV ] = INT_MAX; | |
4270 rd->thresh_mult[THR_NEARESTMV] = INT_MAX; | |
4271 rd->thresh_mult[THR_ZEROMV ] = INT_MAX; | |
4272 rd->thresh_mult[THR_NEARMV ] = INT_MAX; | |
4273 } | |
4274 if (!(cpi->ref_frame_flags & VP9_GOLD_FLAG)) { | |
4275 rd->thresh_mult[THR_NEARESTG ] = INT_MAX; | |
4276 rd->thresh_mult[THR_ZEROG ] = INT_MAX; | |
4277 rd->thresh_mult[THR_NEARG ] = INT_MAX; | |
4278 rd->thresh_mult[THR_NEWG ] = INT_MAX; | |
4279 } | |
4280 if (!(cpi->ref_frame_flags & VP9_ALT_FLAG)) { | |
4281 rd->thresh_mult[THR_NEARESTA ] = INT_MAX; | |
4282 rd->thresh_mult[THR_ZEROA ] = INT_MAX; | |
4283 rd->thresh_mult[THR_NEARA ] = INT_MAX; | |
4284 rd->thresh_mult[THR_NEWA ] = INT_MAX; | |
4285 } | |
4286 | |
4287 if ((cpi->ref_frame_flags & (VP9_LAST_FLAG | VP9_ALT_FLAG)) != | |
4288 (VP9_LAST_FLAG | VP9_ALT_FLAG)) { | |
4289 rd->thresh_mult[THR_COMP_ZEROLA ] = INT_MAX; | |
4290 rd->thresh_mult[THR_COMP_NEARESTLA] = INT_MAX; | |
4291 rd->thresh_mult[THR_COMP_NEARLA ] = INT_MAX; | |
4292 rd->thresh_mult[THR_COMP_NEWLA ] = INT_MAX; | |
4293 } | |
4294 if ((cpi->ref_frame_flags & (VP9_GOLD_FLAG | VP9_ALT_FLAG)) != | |
4295 (VP9_GOLD_FLAG | VP9_ALT_FLAG)) { | |
4296 rd->thresh_mult[THR_COMP_ZEROGA ] = INT_MAX; | |
4297 rd->thresh_mult[THR_COMP_NEARESTGA] = INT_MAX; | |
4298 rd->thresh_mult[THR_COMP_NEARGA ] = INT_MAX; | |
4299 rd->thresh_mult[THR_COMP_NEWGA ] = INT_MAX; | |
4300 } | |
4301 } | |
4302 | |
4303 void vp9_set_rd_speed_thresholds_sub8x8(VP9_COMP *cpi) { | |
4304 const SPEED_FEATURES *const sf = &cpi->sf; | |
4305 RD_OPT *const rd = &cpi->rd; | |
4306 int i; | |
4307 | |
4308 for (i = 0; i < MAX_REFS; ++i) | |
4309 rd->thresh_mult_sub8x8[i] = is_best_mode(cpi->oxcf.mode) ? -500 : 0; | |
4310 | |
4311 rd->thresh_mult_sub8x8[THR_LAST] += 2500; | |
4312 rd->thresh_mult_sub8x8[THR_GOLD] += 2500; | |
4313 rd->thresh_mult_sub8x8[THR_ALTR] += 2500; | |
4314 rd->thresh_mult_sub8x8[THR_INTRA] += 2500; | |
4315 rd->thresh_mult_sub8x8[THR_COMP_LA] += 4500; | |
4316 rd->thresh_mult_sub8x8[THR_COMP_GA] += 4500; | |
4317 | |
4318 // Check for masked out split cases. | |
4319 for (i = 0; i < MAX_REFS; i++) | |
4320 if (sf->disable_split_mask & (1 << i)) | |
4321 rd->thresh_mult_sub8x8[i] = INT_MAX; | |
4322 | |
4323 // disable mode test if frame flag is not set | |
4324 if (!(cpi->ref_frame_flags & VP9_LAST_FLAG)) | |
4325 rd->thresh_mult_sub8x8[THR_LAST] = INT_MAX; | |
4326 if (!(cpi->ref_frame_flags & VP9_GOLD_FLAG)) | |
4327 rd->thresh_mult_sub8x8[THR_GOLD] = INT_MAX; | |
4328 if (!(cpi->ref_frame_flags & VP9_ALT_FLAG)) | |
4329 rd->thresh_mult_sub8x8[THR_ALTR] = INT_MAX; | |
4330 if ((cpi->ref_frame_flags & (VP9_LAST_FLAG | VP9_ALT_FLAG)) != | |
4331 (VP9_LAST_FLAG | VP9_ALT_FLAG)) | |
4332 rd->thresh_mult_sub8x8[THR_COMP_LA] = INT_MAX; | |
4333 if ((cpi->ref_frame_flags & (VP9_GOLD_FLAG | VP9_ALT_FLAG)) != | |
4334 (VP9_GOLD_FLAG | VP9_ALT_FLAG)) | |
4335 rd->thresh_mult_sub8x8[THR_COMP_GA] = INT_MAX; | |
4336 } | |
OLD | NEW |