reconintra.c 3.3 KB

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