/*

 * Copyright (c) 2021, Alliance for Open Media. All rights reserved

 *

 * This source code is subject to the terms of the BSD 3-Clause Clear License

 * and the Alliance for Open Media Patent License 1.0. If the BSD 3-Clause Clear

 * License was not distributed with this source code in the LICENSE file, you

 * can obtain it at aomedia.org/license/software-license/bsd-3-c-c/.  If the

 * Alliance for Open Media Patent License 1.0 was not distributed with this

 * source code in the PATENTS file, you can obtain it at

 * aomedia.org/license/patent-license/.

 */

#include "config/avm_dsp_rtcd.h"

#include "config/av2_rtcd.h"

#include "av2/common/filter.h"

#include "av2/common/scale.h"

#include "avm_dsp/avm_filter.h"

// Note: Expect val to be in q4 precision

static INLINE int scaled_x(int val, const struct scale_factors *sf) {

  const int off =

      (sf->x_scale_fp - (1 << REF_SCALE_SHIFT)) * (1 << (SUBPEL_BITS - 1));

  const int64_t tval = (int64_t)val * sf->x_scale_fp + off;

  return (int)ROUND_POWER_OF_TWO_SIGNED_64(tval,

                                           REF_SCALE_SHIFT - SCALE_EXTRA_BITS);

}

// Note: Expect val to be in q4 precision

static INLINE int scaled_y(int val, const struct scale_factors *sf) {

  const int off =

      (sf->y_scale_fp - (1 << REF_SCALE_SHIFT)) * (1 << (SUBPEL_BITS - 1));

  const int64_t tval = (int64_t)val * sf->y_scale_fp + off;

  return (int)ROUND_POWER_OF_TWO_SIGNED_64(tval,

                                           REF_SCALE_SHIFT - SCALE_EXTRA_BITS);

}

// Note: Expect val to be in q16 precision

static INLINE int64_t scaled_warp_x(int64_t val,

                                    const struct scale_factors *sf) {

  const int64_t tval = (int64_t)val * sf->x_scale_fp;

  return (int64_t)ROUND_POWER_OF_TWO_SIGNED_64(tval, REF_SCALE_SHIFT);

}

// Note: Expect val to be in q16 precision

static INLINE int64_t scaled_warp_y(int64_t val,

                                    const struct scale_factors *sf) {

  const int64_t tval = (int64_t)val * sf->y_scale_fp;

  return (int64_t)ROUND_POWER_OF_TWO_SIGNED_64(tval, REF_SCALE_SHIFT);

}

// Note: Expect val to be in q16 precision

static INLINE int64_t un_scaled_warp(int64_t val,

                                     const struct scale_factors *sf) {

  (void)sf;

  return val;

}

// Note: Expect val to be in q4 precision

static int unscaled_value(int val, const struct scale_factors *sf) {

  (void)sf;

  return val * (1 << SCALE_EXTRA_BITS);

}

static int get_fixed_point_scale_factor(int other_size, int this_size) {

  // Calculate scaling factor once for each reference frame

  // and use fixed point scaling factors in decoding and encoding routines.

  // Hardware implementations can calculate scale factor in device driver

  // and use multiplication and shifting on hardware instead of division.

  return ((other_size << REF_SCALE_SHIFT) + this_size / 2) / this_size;

}

// Given the fixed point scale, calculate coarse point scale.

static int fixed_point_scale_to_coarse_point_scale(int scale_fp) {

  return ROUND_POWER_OF_TWO(scale_fp, REF_SCALE_SHIFT - SCALE_SUBPEL_BITS);

}

// Note: x and y are integer precision, mvq4 is q4 precision.

MV32 av2_scale_mv(const MV *mvq4, int x, int y,

                  const struct scale_factors *sf) {

  const int x_off_q4 = scaled_x(x << SUBPEL_BITS, sf);

  const int y_off_q4 = scaled_y(y << SUBPEL_BITS, sf);

  const MV32 res = { scaled_y((y << SUBPEL_BITS) + mvq4->row, sf) - y_off_q4,

                     scaled_x((x << SUBPEL_BITS) + mvq4->col, sf) - x_off_q4 };

  return res;

}

void av2_setup_scale_factors_for_frame(struct scale_factors *sf, int other_w,

                                       int other_h, int this_w, int this_h) {

  if (!valid_ref_frame_size(other_w, other_h, this_w, this_h)) {

    sf->x_scale_fp = REF_INVALID_SCALE;

    sf->y_scale_fp = REF_INVALID_SCALE;

    return;

  }

  sf->x_scale_fp = get_fixed_point_scale_factor(other_w, this_w);

  sf->y_scale_fp = get_fixed_point_scale_factor(other_h, this_h);

  sf->x_step_q4 = fixed_point_scale_to_coarse_point_scale(sf->x_scale_fp);

  sf->y_step_q4 = fixed_point_scale_to_coarse_point_scale(sf->y_scale_fp);

  if (av2_is_scaled(sf)) {

    sf->scale_value_x = scaled_x;

    sf->scale_value_y = scaled_y;

    sf->scale_value_warp_x = scaled_warp_x;

    sf->scale_value_warp_y = scaled_warp_y;

  } else {

    sf->scale_value_x = unscaled_value;

    sf->scale_value_y = unscaled_value;

    sf->scale_value_warp_x = un_scaled_warp;

    sf->scale_value_warp_y = un_scaled_warp;

  }

}