/*

 *  Copyright (c) 2010 The WebM project authors. All Rights Reserved.

 *

 *  Use of this source code is governed by a BSD-style license

 *  that can be found in the LICENSE file in the root of the source

 *  tree. An additional intellectual property rights grant can be found

 *  in the file PATENTS.  All contributing project authors may

 *  be found in the AUTHORS file in the root of the source tree.

 */

#include "./vpx_config.h"

#include "./vpx_dsp_rtcd.h"

#include "./vp8_rtcd.h"

#include "vpx_mem/vpx_mem.h"

#include "vpx_ports/vpx_once.h"

#include "blockd.h"

#include "vp8/common/reconintra.h"

#include "vp8/common/reconintra4x4.h"

enum {

  SIZE_16,

  SIZE_8,

  NUM_SIZES,

};

typedef void (*intra_pred_fn)(uint8_t *dst, ptrdiff_t stride,

                              const uint8_t *above, const uint8_t *left);

static intra_pred_fn pred[4][NUM_SIZES];

static intra_pred_fn dc_pred[2][2][NUM_SIZES];

static void vp8_init_intra_predictors_internal(void) {

#define INIT_SIZE(sz)                                           \

  pred[V_PRED][SIZE_##sz] = vpx_v_predictor_##sz##x##sz;        \

  pred[H_PRED][SIZE_##sz] = vpx_h_predictor_##sz##x##sz;        \

  pred[TM_PRED][SIZE_##sz] = vpx_tm_predictor_##sz##x##sz;      \

                                                                \

  dc_pred[0][0][SIZE_##sz] = vpx_dc_128_predictor_##sz##x##sz;  \

  dc_pred[0][1][SIZE_##sz] = vpx_dc_top_predictor_##sz##x##sz;  \

  dc_pred[1][0][SIZE_##sz] = vpx_dc_left_predictor_##sz##x##sz; \

  dc_pred[1][1][SIZE_##sz] = vpx_dc_predictor_##sz##x##sz

  INIT_SIZE(16);

  INIT_SIZE(8);

  vp8_init_intra4x4_predictors_internal();

}

void 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) {

  MB_PREDICTION_MODE mode = x->mode_info_context->mbmi.mode;

  DECLARE_ALIGNED(16, uint8_t, yleft_col[16]);

  int i;

  intra_pred_fn fn;

  for (i = 0; i < 16; ++i) {

    yleft_col[i] = yleft[i * left_stride];

  }

  if (mode == DC_PRED) {

    fn = dc_pred[x->left_available][x->up_available][SIZE_16];

  } else {

    fn = pred[mode][SIZE_16];

  }

  fn(ypred_ptr, y_stride, yabove_row, yleft_col);

}

void 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) {

  MB_PREDICTION_MODE uvmode = x->mode_info_context->mbmi.uv_mode;

#if HAVE_VSX

  /* Power PC implementation uses "vec_vsx_ld" to read 16 bytes from

     uleft_col and vleft_col. Play it safe by reserving enough stack

     space here. */

  unsigned char uleft_col[16];

  unsigned char vleft_col[16];

#else

  unsigned char uleft_col[8];

  unsigned char vleft_col[8];

#endif

  int i;

  intra_pred_fn fn;

  for (i = 0; i < 8; ++i) {

    uleft_col[i] = uleft[i * left_stride];

    vleft_col[i] = vleft[i * left_stride];

  }

  if (uvmode == DC_PRED) {

    fn = dc_pred[x->left_available][x->up_available][SIZE_8];

  } else {

    fn = pred[uvmode][SIZE_8];

  }

  fn(upred_ptr, pred_stride, uabove_row, uleft_col);

  fn(vpred_ptr, pred_stride, vabove_row, vleft_col);

}

void vp8_init_intra_predictors(void) {

  once(vp8_init_intra_predictors_internal);

}