/*
 *  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_ports/config.h"
#include "vpx_ports/arm.h"
#include "vp9/encoder/vp9_variance.h"
#include "vp9/encoder/vp9_onyx_int.h"

extern void (*vp9_yv12_copy_partial_frame_ptr)(YV12_BUFFER_CONFIG *src_ybc, YV12_BUFFER_CONFIG *dst_ybc, int Fraction);
extern void vp9_yv12_copy_partial_frame(YV12_BUFFER_CONFIG *src_ybc, YV12_BUFFER_CONFIG *dst_ybc, int Fraction);
extern void vpxyv12_copy_partial_frame_neon(YV12_BUFFER_CONFIG *src_ybc, YV12_BUFFER_CONFIG *dst_ybc, int Fraction);

void vp9_arch_arm_encoder_init(VP9_COMP *cpi) {
#if CONFIG_RUNTIME_CPU_DETECT
  int flags = cpi->common.rtcd.flags;

#if HAVE_ARMV5TE
  if (flags & HAS_EDSP) {
  }
#endif

#if HAVE_ARMV6
  if (flags & HAS_MEDIA) {
    cpi->rtcd.variance.sad16x16              = vp9_sad16x16_armv6;
    /*cpi->rtcd.variance.sad16x8               = vp9_sad16x8_c;
    cpi->rtcd.variance.sad8x16               = vp9_sad8x16_c;
    cpi->rtcd.variance.sad8x8                = vp9_sad8x8_c;
    cpi->rtcd.variance.sad4x4                = vp9_sad4x4_c;*/

    /*cpi->rtcd.variance.var4x4                = vp9_variance4x4_c;*/
    cpi->rtcd.variance.var8x8                = vp9_variance8x8_armv6;
    /*cpi->rtcd.variance.var8x16               = vp9_variance8x16_c;
    cpi->rtcd.variance.var16x8               = vp9_variance16x8_c;*/
    cpi->rtcd.variance.var16x16              = vp9_variance16x16_armv6;

    /*cpi->rtcd.variance.subpixvar4x4          = vp9_sub_pixel_variance4x4_c;*/
    cpi->rtcd.variance.subpixvar8x8          = vp9_sub_pixel_variance8x8_armv6;
    /*cpi->rtcd.variance.subpixvar8x16         = vp9_sub_pixel_variance8x16_c;
    cpi->rtcd.variance.subpixvar16x8         = vp9_sub_pixel_variance16x8_c;*/
    cpi->rtcd.variance.subpixvar16x16        = vp9_sub_pixel_variance16x16_armv6;
    cpi->rtcd.variance.halfpixvar16x16_h     = vp9_variance_halfpixvar16x16_h_armv6;
    cpi->rtcd.variance.halfpixvar16x16_v     = vp9_variance_halfpixvar16x16_v_armv6;
    cpi->rtcd.variance.halfpixvar16x16_hv    = vp9_variance_halfpixvar16x16_hv_armv6;

    cpi->rtcd.variance.mse16x16              = vp9_mse16x16_armv6;
    /*cpi->rtcd.variance.getmbss               = vp9_get_mb_ss_c;*/

    cpi->rtcd.fdct.short4x4                  = vp9_short_fdct4x4_armv6;
    cpi->rtcd.fdct.short8x4                  = vp9_short_fdct8x4_armv6;
    cpi->rtcd.fdct.fast4x4                   = vp9_short_fdct4x4_armv6;
    cpi->rtcd.fdct.fast8x4                   = vp9_short_fdct8x4_armv6;
    cpi->rtcd.fdct.walsh_short4x4            = vp9_short_walsh4x4_armv6;

    /*cpi->rtcd.encodemb.berr                  = vp9_block_error_c;
    cpi->rtcd.encodemb.mberr                 = vp9_mbblock_error_c;
    cpi->rtcd.encodemb.mbuverr               = vp9_mbuverror_c;*/
    cpi->rtcd.encodemb.subb                  = vp9_subtract_b_armv6;
    cpi->rtcd.encodemb.submby                = vp9_subtract_mby_armv6;
    cpi->rtcd.encodemb.submbuv               = vp9_subtract_mbuv_armv6;

    /*cpi->rtcd.quantize.quantb                = vp8_regular_quantize_b;*/
    cpi->rtcd.quantize.fastquantb            = vp8_fast_quantize_b_armv6;
  }
#endif

#if HAVE_ARMV7
  if (flags & HAS_NEON) {
    cpi->rtcd.variance.sad16x16              = vp9_sad16x16_neon;
    cpi->rtcd.variance.sad16x8               = vp9_sad16x8_neon;
    cpi->rtcd.variance.sad8x16               = vp9_sad8x16_neon;
    cpi->rtcd.variance.sad8x8                = vp9_sad8x8_neon;
    cpi->rtcd.variance.sad4x4                = vp9_sad4x4_neon;

    /*cpi->rtcd.variance.var4x4                = vp9_variance4x4_c;*/
    cpi->rtcd.variance.var8x8                = vp9_variance8x8_neon;
    cpi->rtcd.variance.var8x16               = vp9_variance8x16_neon;
    cpi->rtcd.variance.var16x8               = vp9_variance16x8_neon;
    cpi->rtcd.variance.var16x16              = vp9_variance16x16_neon;

    /*cpi->rtcd.variance.subpixvar4x4          = vp9_sub_pixel_variance4x4_c;*/
    cpi->rtcd.variance.subpixvar8x8          = vp9_sub_pixel_variance8x8_neon;
    /*cpi->rtcd.variance.subpixvar8x16         = vp9_sub_pixel_variance8x16_c;
    cpi->rtcd.variance.subpixvar16x8         = vp9_sub_pixel_variance16x8_c;*/
    cpi->rtcd.variance.subpixvar16x16        = vp9_sub_pixel_variance16x16_neon;
    cpi->rtcd.variance.halfpixvar16x16_h     = vp9_variance_halfpixvar16x16_h_neon;
    cpi->rtcd.variance.halfpixvar16x16_v     = vp9_variance_halfpixvar16x16_v_neon;
    cpi->rtcd.variance.halfpixvar16x16_hv    = vp9_variance_halfpixvar16x16_hv_neon;

    cpi->rtcd.variance.mse16x16              = vp9_mse16x16_neon;
    /*cpi->rtcd.variance.getmbss               = vp9_get_mb_ss_c;*/

    cpi->rtcd.fdct.short4x4                  = vp9_short_fdct4x4_neon;
    cpi->rtcd.fdct.short8x4                  = vp9_short_fdct8x4_neon;
    cpi->rtcd.fdct.fast4x4                   = vp9_short_fdct4x4_neon;
    cpi->rtcd.fdct.fast8x4                   = vp9_short_fdct8x4_neon;
    cpi->rtcd.fdct.walsh_short4x4            = vp9_short_walsh4x4_neon;

    /*cpi->rtcd.encodemb.berr                  = vp9_block_error_c;
    cpi->rtcd.encodemb.mberr                 = vp9_mbblock_error_c;
    cpi->rtcd.encodemb.mbuverr               = vp9_mbuverror_c;*/
    cpi->rtcd.encodemb.subb                  = vp9_subtract_b_neon;
    cpi->rtcd.encodemb.submby                = vp9_subtract_mby_neon;
    cpi->rtcd.encodemb.submbuv               = vp9_subtract_mbuv_neon;

    /*cpi->rtcd.quantize.quantb                = vp8_regular_quantize_b;
    cpi->rtcd.quantize.quantb_pair           = vp8_regular_quantize_b_pair;*/
    cpi->rtcd.quantize.fastquantb            = vp8_fast_quantize_b_neon;
    cpi->rtcd.quantize.fastquantb_pair       = vp8_fast_quantize_b_pair_neon;
  }
#endif

#if HAVE_ARMV7
#if CONFIG_RUNTIME_CPU_DETECT
  if (flags & HAS_NEON)
#endif
  {
    vp9_yv12_copy_partial_frame_ptr = vpxyv12_copy_partial_frame_neon;
  }
#endif
#endif
}