/*
 *  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_mem/vpx_mem.h"
#include "vp9/common/vp9_reconintra.h"
#include "vp9_rtcd.h"

#if CONFIG_NEWBINTRAMODES
static int find_grad_measure(uint8_t *x, int stride, int n, int tx, int ty,
                             int dx, int dy) {
  int i, j;
  int count = 0, gsum = 0, gdiv;
  /* TODO: Make this code more efficient by breaking up into two loops */
  for (i = -ty; i < n; ++i)
    for (j = -tx; j < n; ++j) {
      int g;
      if (i >= 0 && j >= 0) continue;
      if (i + dy >= 0 && j + dx >= 0) continue;
      if (i + dy < -ty || i + dy >= n || j + dx < -tx || j + dx >= n) continue;
      g = abs(x[(i + dy) * stride + j + dx] - x[i * stride + j]);
      gsum += g * g;
      count++;
    }
  gdiv = (dx * dx + dy * dy) * count;
  return ((gsum << 8) + (gdiv >> 1)) / gdiv;
}

#if CONTEXT_PRED_REPLACEMENTS == 6
B_PREDICTION_MODE vp9_find_dominant_direction(uint8_t *ptr,
                                              int stride, int n,
                                              int tx, int ty) {
  int g[8], i, imin, imax;
  g[1] = find_grad_measure(ptr, stride, n, tx, ty,  2, 1);
  g[2] = find_grad_measure(ptr, stride, n, tx, ty,  1, 1);
  g[3] = find_grad_measure(ptr, stride, n, tx, ty,  1, 2);
  g[5] = find_grad_measure(ptr, stride, n, tx, ty, -1, 2);
  g[6] = find_grad_measure(ptr, stride, n, tx, ty, -1, 1);
  g[7] = find_grad_measure(ptr, stride, n, tx, ty, -2, 1);
  imin = 1;
  for (i = 2; i < 8; i += 1 + (i == 3))
    imin = (g[i] < g[imin] ? i : imin);
  imax = 1;
  for (i = 2; i < 8; i += 1 + (i == 3))
    imax = (g[i] > g[imax] ? i : imax);
  /*
  printf("%d %d %d %d %d %d = %d %d\n",
         g[1], g[2], g[3], g[5], g[6], g[7], imin, imax);
         */
  switch (imin) {
    case 1:
      return B_D153_PRED;
    case 2:
      return B_D135_PRED;
    case 3:
      return B_D117_PRED;
    case 5:
      return B_D63_PRED;
    case 6:
      return B_D45_PRED;
    case 7:
      return B_D27_PRED;
    default:
      assert(0);
  }
}
#elif CONTEXT_PRED_REPLACEMENTS == 4
B_PREDICTION_MODE vp9_find_dominant_direction(uint8_t *ptr,
                                              int stride, int n,
                                              int tx, int ty) {
  int g[8], i, imin, imax;
  g[1] = find_grad_measure(ptr, stride, n, tx, ty,  2, 1);
  g[3] = find_grad_measure(ptr, stride, n, tx, ty,  1, 2);
  g[5] = find_grad_measure(ptr, stride, n, tx, ty, -1, 2);
  g[7] = find_grad_measure(ptr, stride, n, tx, ty, -2, 1);
  imin = 1;
  for (i = 3; i < 8; i+=2)
    imin = (g[i] < g[imin] ? i : imin);
  imax = 1;
  for (i = 3; i < 8; i+=2)
    imax = (g[i] > g[imax] ? i : imax);
  /*
  printf("%d %d %d %d = %d %d\n",
         g[1], g[3], g[5], g[7], imin, imax);
         */
  switch (imin) {
    case 1:
      return B_D153_PRED;
    case 3:
      return B_D117_PRED;
    case 5:
      return B_D63_PRED;
    case 7:
      return B_D27_PRED;
    default:
      assert(0);
  }
}
#elif CONTEXT_PRED_REPLACEMENTS == 0
B_PREDICTION_MODE vp9_find_dominant_direction(uint8_t *ptr,
                                              int stride, int n,
                                              int tx, int ty) {
  int g[8], i, imin, imax;
  g[0] = find_grad_measure(ptr, stride, n, tx, ty,  1, 0);
  g[1] = find_grad_measure(ptr, stride, n, tx, ty,  2, 1);
  g[2] = find_grad_measure(ptr, stride, n, tx, ty,  1, 1);
  g[3] = find_grad_measure(ptr, stride, n, tx, ty,  1, 2);
  g[4] = find_grad_measure(ptr, stride, n, tx, ty,  0, 1);
  g[5] = find_grad_measure(ptr, stride, n, tx, ty, -1, 2);
  g[6] = find_grad_measure(ptr, stride, n, tx, ty, -1, 1);
  g[7] = find_grad_measure(ptr, stride, n, tx, ty, -2, 1);
  imax = 0;
  for (i = 1; i < 8; i++)
    imax = (g[i] > g[imax] ? i : imax);
  imin = 0;
  for (i = 1; i < 8; i++)
    imin = (g[i] < g[imin] ? i : imin);

  switch (imin) {
    case 0:
      return B_H_PRED;
    case 1:
      return B_D153_PRED;
    case 2:
      return B_D135_PRED;
    case 3:
      return B_D117_PRED;
    case 4:
      return B_V_PRED;
    case 5:
      return B_D63_PRED;
    case 6:
      return B_D45_PRED;
    case 7:
      return B_D27_PRED;
    default:
      assert(0);
  }
}
#endif

B_PREDICTION_MODE vp9_find_bpred_context(MACROBLOCKD *xd, int block_idx,
                                         uint8_t *ptr, int stride) {
  const int have_top = (block_idx >> 2) || xd->up_available;
  const int have_left = (block_idx & 3)  || xd->left_available;
  int tx = have_left ? 4 : 0;
  int ty = have_top ? 4 : 0;
  if (!have_left && !have_top)
    return B_DC_PRED;
  return vp9_find_dominant_direction(ptr, stride, 4, tx, ty);
}

void vp9_intra4x4_predict(MACROBLOCKD *xd,
                          int block_idx,
                          int b_mode,
                          uint8_t *predictor,
                          int ps) {
  int i, r, c;
  const int have_top = (block_idx >> 2) || xd->up_available;
  const int have_left = (block_idx & 3)  || xd->left_available;
  const int have_right = (block_idx & 3) != 3 || xd->right_available;
  uint8_t left[4], above[8], top_left;
  /*
   * 127 127 127 .. 127 127 127 127 127 127
   * 129  A   B  ..  Y   Z
   * 129  C   D  ..  W   X
   * 129  E   F  ..  U   V
   * 129  G   H  ..  S   T   T   T   T   T
   *  ..
   */

  if (have_left) {
    uint8_t *left_ptr = predictor - 1;
    const int stride = ps;

    left[0] = left_ptr[0 * stride];
    left[1] = left_ptr[1 * stride];
    left[2] = left_ptr[2 * stride];
    left[3] = left_ptr[3 * stride];
  } else {
    left[0] = left[1] = left[2] = left[3] = 129;
  }

  if (have_top) {
    uint8_t *above_ptr = predictor - ps;
    top_left = have_left ? above_ptr[-1] : 127;

    above[0] = above_ptr[0];
    above[1] = above_ptr[1];
    above[2] = above_ptr[2];
    above[3] = above_ptr[3];
    if (((block_idx & 3) != 3) ||
        (have_right && block_idx == 3 &&
         ((xd->mb_index != 3 && xd->sb_index != 3) ||
          ((xd->mb_index & 1) == 0 && xd->sb_index == 3)))) {
      above[4] = above_ptr[4];
      above[5] = above_ptr[5];
      above[6] = above_ptr[6];
      above[7] = above_ptr[7];
    } else if (have_right) {
      uint8_t *above_right = above_ptr + 4;

      if (xd->sb_index == 3 && (xd->mb_index & 1))
        above_right -= 32 * ps;
      if (xd->mb_index == 3)
        above_right -= 16 * ps;
      above_right -= (block_idx & ~3) * ps;

      /* use a more distant above-right (from closest available top-right
       * corner), but with a "localized DC" (similar'ish to TM-pred):
       *
       *  A   B   C   D   E   F   G   H
       *  I   J   K   L
       *  M   N   O   P
       *  Q   R   S   T
       *  U   V   W   X   x1  x2  x3  x4
       *
       * Where:
       * x1 = clip_pixel(E + X - D)
       * x2 = clip_pixel(F + X - D)
       * x3 = clip_pixel(G + X - D)
       * x4 = clip_pixel(H + X - D)
       *
       * This is applied anytime when we use a "distant" above-right edge
       * that is not immediately top-right to the block that we're going
       * to do intra prediction for.
       */
      above[4] = clip_pixel(above_right[0] + above_ptr[3] - above_right[-1]);
      above[5] = clip_pixel(above_right[1] + above_ptr[3] - above_right[-1]);
      above[6] = clip_pixel(above_right[2] + above_ptr[3] - above_right[-1]);
      above[7] = clip_pixel(above_right[3] + above_ptr[3] - above_right[-1]);
    } else {
      // extend edge
      above[4] = above[5] = above[6] = above[7] = above[3];
    }
  } else {
    above[0] = above[1] = above[2] = above[3] = 127;
    above[4] = above[5] = above[6] = above[7] = 127;
    top_left = 127;
  }

#if CONFIG_NEWBINTRAMODES
  if (b_mode == B_CONTEXT_PRED)
    b_mode = xd->mode_info_context->bmi[block_idx].as_mode.context;
#endif

  switch (b_mode) {
    case B_DC_PRED: {
      int expected_dc = 128;
      if (have_top || have_left) {
        int average = 0;
        int count = 0;
        if (have_top) {
          for (i = 0; i < 4; i++)
            average += above[i];
          count += 4;
        }
        if (have_left) {
          for (i = 0; i < 4; i++)
            average += left[i];
          count += 4;
        }
        expected_dc = (average + (count >> 1)) / count;
      }
      for (r = 0; r < 4; r++) {
        for (c = 0; c < 4; c++)
          predictor[c] = expected_dc;
        predictor += ps;
      }
    }
    break;
    case B_TM_PRED: {
      /* prediction similar to true_motion prediction */
      for (r = 0; r < 4; r++) {
        for (c = 0; c < 4; c++)
          predictor[c] = clip_pixel(above[c] - top_left + left[r]);
        predictor += ps;
      }
    }
    break;
    case B_V_PRED:
      for (r = 0; r < 4; r++) {
        for (c = 0; c < 4; c++)
          predictor[c] = above[c];
        predictor += ps;
      }
      break;
    case B_H_PRED:
      for (r = 0; r < 4; r++) {
        for (c = 0; c < 4; c++)
          predictor[c] = left[r];
        predictor += ps;
      }
      break;
    case B_D45_PRED: {
      uint8_t *p = above;

      predictor[0 * ps + 0] = ROUND_POWER_OF_TWO(p[0] + p[1] * 2 + p[2], 2);
      predictor[0 * ps + 1] =
        predictor[1 * ps + 0] = ROUND_POWER_OF_TWO(p[1] + p[2] * 2 + p[3], 2);
      predictor[0 * ps + 2] =
        predictor[1 * ps + 1] =
          predictor[2 * ps + 0] = ROUND_POWER_OF_TWO(p[2] + p[3] * 2 + p[4], 2);
      predictor[0 * ps + 3] =
        predictor[1 * ps + 2] =
          predictor[2 * ps + 1] =
            predictor[3 * ps + 0] =
              ROUND_POWER_OF_TWO(p[3] + p[4] * 2 + p[5], 2);
      predictor[1 * ps + 3] =
        predictor[2 * ps + 2] =
          predictor[3 * ps + 1] = ROUND_POWER_OF_TWO(p[4] + p[5] * 2 + p[6], 2);
      predictor[2 * ps + 3] =
        predictor[3 * ps + 2] = ROUND_POWER_OF_TWO(p[5] + p[6] * 2 + p[7], 2);
      predictor[3 * ps + 3] = ROUND_POWER_OF_TWO(p[6] + p[7] * 2 + p[7], 2);

    }
    break;
    case B_D135_PRED: {
      uint8_t p[9] = { left[3], left[2], left[1], left[0],
                       top_left,
                       above[0], above[1], above[2], above[3] };

      predictor[3 * ps + 0] = ROUND_POWER_OF_TWO(p[0] + p[1] * 2 + p[2], 2);
      predictor[3 * ps + 1] =
        predictor[2 * ps + 0] = ROUND_POWER_OF_TWO(p[1] + p[2] * 2 + p[3], 2);
      predictor[3 * ps + 2] =
        predictor[2 * ps + 1] =
          predictor[1 * ps + 0] = ROUND_POWER_OF_TWO(p[2] + p[3] * 2 + p[4], 2);
      predictor[3 * ps + 3] =
        predictor[2 * ps + 2] =
          predictor[1 * ps + 1] =
            predictor[0 * ps + 0] =
              ROUND_POWER_OF_TWO(p[3] + p[4] * 2 + p[5], 2);
      predictor[2 * ps + 3] =
        predictor[1 * ps + 2] =
          predictor[0 * ps + 1] = ROUND_POWER_OF_TWO(p[4] + p[5] * 2 + p[6], 2);
      predictor[1 * ps + 3] =
        predictor[0 * ps + 2] = ROUND_POWER_OF_TWO(p[5] + p[6] * 2 + p[7], 2);
      predictor[0 * ps + 3] = ROUND_POWER_OF_TWO(p[6] + p[7] * 2 + p[8], 2);

    }
    break;
    case B_D117_PRED: {
      uint8_t p[9] = { left[3], left[2], left[1], left[0],
                       top_left,
                       above[0], above[1], above[2], above[3] };

      predictor[3 * ps + 0] = ROUND_POWER_OF_TWO(p[1] + p[2] * 2 + p[3], 2);
      predictor[2 * ps + 0] = ROUND_POWER_OF_TWO(p[2] + p[3] * 2 + p[4], 2);
      predictor[3 * ps + 1] =
        predictor[1 * ps + 0] = ROUND_POWER_OF_TWO(p[3] + p[4] * 2 + p[5], 2);
      predictor[2 * ps + 1] =
        predictor[0 * ps + 0] = ROUND_POWER_OF_TWO(p[4] + p[5], 1);
      predictor[3 * ps + 2] =
        predictor[1 * ps + 1] = ROUND_POWER_OF_TWO(p[4] + p[5] * 2 + p[6], 2);
      predictor[2 * ps + 2] =
        predictor[0 * ps + 1] = ROUND_POWER_OF_TWO(p[5] + p[6], 1);
      predictor[3 * ps + 3] =
        predictor[1 * ps + 2] = ROUND_POWER_OF_TWO(p[5] + p[6] * 2 + p[7], 2);
      predictor[2 * ps + 3] =
        predictor[0 * ps + 2] = ROUND_POWER_OF_TWO(p[6] + p[7], 1);
      predictor[1 * ps + 3] = ROUND_POWER_OF_TWO(p[6] + p[7] * 2 + p[8], 2);
      predictor[0 * ps + 3] = ROUND_POWER_OF_TWO(p[7] + p[8], 1);

    }
    break;
    case B_D63_PRED: {
      uint8_t *p = above;

      predictor[0 * ps + 0] = ROUND_POWER_OF_TWO(p[0] + p[1], 1);
      predictor[1 * ps + 0] = ROUND_POWER_OF_TWO(p[0] + p[1] * 2 + p[2], 2);
      predictor[2 * ps + 0] =
        predictor[0 * ps + 1] = ROUND_POWER_OF_TWO(p[1] + p[2], 1);
      predictor[1 * ps + 1] =
        predictor[3 * ps + 0] = ROUND_POWER_OF_TWO(p[1] + p[2] * 2 + p[3], 2);
      predictor[2 * ps + 1] =
        predictor[0 * ps + 2] = ROUND_POWER_OF_TWO(p[2] + p[3], 1);
      predictor[3 * ps + 1] =
        predictor[1 * ps + 2] = ROUND_POWER_OF_TWO(p[2] + p[3] * 2 + p[4], 2);
      predictor[0 * ps + 3] =
        predictor[2 * ps + 2] = ROUND_POWER_OF_TWO(p[3] + p[4], 1);
      predictor[1 * ps + 3] =
        predictor[3 * ps + 2] = ROUND_POWER_OF_TWO(p[3] + p[4] * 2 + p[5], 2);
      predictor[2 * ps + 3] = ROUND_POWER_OF_TWO(p[4] + p[5] * 2 + p[6], 2);
      predictor[3 * ps + 3] = ROUND_POWER_OF_TWO(p[5] + p[6] * 2 + p[7], 2);
    }
    break;
    case B_D153_PRED: {
      uint8_t p[9] = { left[3], left[2], left[1], left[0],
                       top_left,
                       above[0], above[1], above[2], above[3] };

      predictor[3 * ps + 0] = ROUND_POWER_OF_TWO(p[0] + p[1], 1);
      predictor[3 * ps + 1] = ROUND_POWER_OF_TWO(p[0] + p[1] * 2 + p[2], 2);
      predictor[2 * ps + 0] =
        predictor[3 * ps + 2] = ROUND_POWER_OF_TWO(p[1] + p[2], 1);
      predictor[2 * ps + 1] =
        predictor[3 * ps + 3] = ROUND_POWER_OF_TWO(p[1] + p[2] * 2 + p[3], 2);
      predictor[2 * ps + 2] =
        predictor[1 * ps + 0] = ROUND_POWER_OF_TWO(p[2] + p[3], 1);
      predictor[2 * ps + 3] =
        predictor[1 * ps + 1] = ROUND_POWER_OF_TWO(p[2] + p[3] * 2 + p[4], 2);
      predictor[1 * ps + 2] =
        predictor[0 * ps + 0] = ROUND_POWER_OF_TWO(p[3] + p[4], 1);
      predictor[1 * ps + 3] =
        predictor[0 * ps + 1] = ROUND_POWER_OF_TWO(p[3] + p[4] * 2 + p[5], 2);
      predictor[0 * ps + 2] = ROUND_POWER_OF_TWO(p[4] + p[5] * 2 + p[6], 2);
      predictor[0 * ps + 3] = ROUND_POWER_OF_TWO(p[5] + p[6] * 2 + p[7], 2);
    }
    break;
    case B_D27_PRED: {
      uint8_t *p = left;
      predictor[0 * ps + 0] = ROUND_POWER_OF_TWO(p[0] + p[1], 1);
      predictor[0 * ps + 1] = ROUND_POWER_OF_TWO(p[0] + p[1] * 2 + p[2], 2);
      predictor[0 * ps + 2] =
        predictor[1 * ps + 0] = ROUND_POWER_OF_TWO(p[1] + p[2], 1);
      predictor[0 * ps + 3] =
        predictor[1 * ps + 1] = ROUND_POWER_OF_TWO(p[1] + p[2] * 2 + p[3], 2);
      predictor[1 * ps + 2] =
        predictor[2 * ps + 0] = ROUND_POWER_OF_TWO(p[2] + p[3], 1);
      predictor[1 * ps + 3] =
        predictor[2 * ps + 1] = ROUND_POWER_OF_TWO(p[2] + p[3] * 2 + p[3], 2);
      predictor[2 * ps + 2] =
        predictor[2 * ps + 3] =
          predictor[3 * ps + 0] =
            predictor[3 * ps + 1] =
              predictor[3 * ps + 2] =
                predictor[3 * ps + 3] = p[3];
    }
    break;

#if CONFIG_NEWBINTRAMODES
    case B_CONTEXT_PRED:
    break;
    /*
    case B_CORNER_PRED:
    corner_predictor(predictor, 16, 4, above, left);
    break;
    */
#endif
  }
}
#endif