1 /*
2 * Copyright (c) 2010 The WebM project authors. All Rights Reserved.
3 *
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
6 * tree. An additional intellectual property rights grant can be found
7 * in the file PATENTS. All contributing project authors may
8 * be found in the AUTHORS file in the root of the source tree.
9 */
10
11 #include "./vpx_config.h"
12 #include "./vpx_dsp_rtcd.h"
13 #include "./vp8_rtcd.h"
14 #include "vpx_mem/vpx_mem.h"
15 #include "vpx_ports/vpx_once.h"
16 #include "blockd.h"
17 #include "vp8/common/reconintra.h"
18 #include "vp8/common/reconintra4x4.h"
19
20 enum {
21 SIZE_16,
22 SIZE_8,
23 NUM_SIZES,
24 };
25
26 typedef void (*intra_pred_fn)(uint8_t *dst, ptrdiff_t stride,
27 const uint8_t *above, const uint8_t *left);
28
29 static intra_pred_fn pred[4][NUM_SIZES];
30 static intra_pred_fn dc_pred[2][2][NUM_SIZES];
31
vp8_init_intra_predictors_internal(void)32 static void vp8_init_intra_predictors_internal(void) {
33 #define INIT_SIZE(sz) \
34 pred[V_PRED][SIZE_##sz] = vpx_v_predictor_##sz##x##sz; \
35 pred[H_PRED][SIZE_##sz] = vpx_h_predictor_##sz##x##sz; \
36 pred[TM_PRED][SIZE_##sz] = vpx_tm_predictor_##sz##x##sz; \
37 \
38 dc_pred[0][0][SIZE_##sz] = vpx_dc_128_predictor_##sz##x##sz; \
39 dc_pred[0][1][SIZE_##sz] = vpx_dc_top_predictor_##sz##x##sz; \
40 dc_pred[1][0][SIZE_##sz] = vpx_dc_left_predictor_##sz##x##sz; \
41 dc_pred[1][1][SIZE_##sz] = vpx_dc_predictor_##sz##x##sz
42
43 INIT_SIZE(16);
44 INIT_SIZE(8);
45 vp8_init_intra4x4_predictors_internal();
46 }
47
vp8_build_intra_predictors_mby_s(MACROBLOCKD * x,unsigned char * yabove_row,unsigned char * yleft,int left_stride,unsigned char * ypred_ptr,int y_stride)48 void vp8_build_intra_predictors_mby_s(MACROBLOCKD *x, unsigned char *yabove_row,
49 unsigned char *yleft, int left_stride,
50 unsigned char *ypred_ptr, int y_stride) {
51 MB_PREDICTION_MODE mode = x->mode_info_context->mbmi.mode;
52 DECLARE_ALIGNED(16, uint8_t, yleft_col[16]);
53 int i;
54 intra_pred_fn fn;
55
56 for (i = 0; i < 16; ++i) {
57 yleft_col[i] = yleft[i * left_stride];
58 }
59
60 if (mode == DC_PRED) {
61 fn = dc_pred[x->left_available][x->up_available][SIZE_16];
62 } else {
63 fn = pred[mode][SIZE_16];
64 }
65
66 fn(ypred_ptr, y_stride, yabove_row, yleft_col);
67 }
68
vp8_build_intra_predictors_mbuv_s(MACROBLOCKD * x,unsigned char * uabove_row,unsigned char * vabove_row,unsigned char * uleft,unsigned char * vleft,int left_stride,unsigned char * upred_ptr,unsigned char * vpred_ptr,int pred_stride)69 void vp8_build_intra_predictors_mbuv_s(
70 MACROBLOCKD *x, unsigned char *uabove_row, unsigned char *vabove_row,
71 unsigned char *uleft, unsigned char *vleft, int left_stride,
72 unsigned char *upred_ptr, unsigned char *vpred_ptr, int pred_stride) {
73 MB_PREDICTION_MODE uvmode = x->mode_info_context->mbmi.uv_mode;
74 #if HAVE_VSX
75 /* Power PC implementation uses "vec_vsx_ld" to read 16 bytes from
76 uleft_col and vleft_col. Play it safe by reserving enough stack
77 space here. */
78 unsigned char uleft_col[16];
79 unsigned char vleft_col[16];
80 #else
81 unsigned char uleft_col[8];
82 unsigned char vleft_col[8];
83 #endif
84 int i;
85 intra_pred_fn fn;
86
87 for (i = 0; i < 8; ++i) {
88 uleft_col[i] = uleft[i * left_stride];
89 vleft_col[i] = vleft[i * left_stride];
90 }
91
92 if (uvmode == DC_PRED) {
93 fn = dc_pred[x->left_available][x->up_available][SIZE_8];
94 } else {
95 fn = pred[uvmode][SIZE_8];
96 }
97
98 fn(upred_ptr, pred_stride, uabove_row, uleft_col);
99 fn(vpred_ptr, pred_stride, vabove_row, vleft_col);
100 }
101
vp8_init_intra_predictors(void)102 void vp8_init_intra_predictors(void) {
103 once(vp8_init_intra_predictors_internal);
104 }
105