/*

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

 *

 * This source code is subject to the terms of the BSD 2 Clause License and

 * the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License

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

 * obtain it at www.aomedia.org/license/software. 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 www.aomedia.org/license/patent.

 */

#include <emmintrin.h>

#include "config/av1_rtcd.h"

void av1_interp_cubic_rate_dist_sse2(const double *p1, const double *p2,

                                     double x, double rate_dist_f[2]) {

  const __m128d half = _mm_set1_pd(0.5);

  const __m128d two = _mm_set1_pd(2.0);

  const __m128d three = _mm_set1_pd(3.0);

  const __m128d four = _mm_set1_pd(4.0);

  const __m128d five = _mm_set1_pd(5.0);

  const __m128d reg_x = _mm_set1_pd(x);

  const __m128d reg_p0 = _mm_set_pd(p2[0], p1[0]);

  const __m128d reg_p1 = _mm_set_pd(p2[1], p1[1]);

  const __m128d reg_p2 = _mm_set_pd(p2[2], p1[2]);

  const __m128d reg_p3 = _mm_set_pd(p2[3], p1[3]);

  // To ensure that results are bit-identical to the C code, we need to perform

  // exactly the same sequence of operations here as in the C code.

  // reg_res_0 = x * (3.0 * (p[1] - p[2]) + p[3] - p[0])

  __m128d reg_res_0 = _mm_sub_pd(reg_p1, reg_p2);

  reg_res_0 = _mm_mul_pd(three, reg_res_0);

  reg_res_0 = _mm_add_pd(reg_res_0, reg_p3);

  reg_res_0 = _mm_sub_pd(reg_res_0, reg_p0);

  reg_res_0 = _mm_mul_pd(reg_x, reg_res_0);

  // reg_res_1 = 2.0 * p[0] - 5.0 * p[1] + 4.0 * p[2]- p[3]

  const __m128d regp0_x_2 = _mm_mul_pd(two, reg_p0);

  const __m128d regp1_x_5 = _mm_mul_pd(five, reg_p1);

  const __m128d regp2_x_4 = _mm_mul_pd(four, reg_p2);

  __m128d reg_res_1 = _mm_sub_pd(regp0_x_2, regp1_x_5);

  reg_res_1 = _mm_add_pd(reg_res_1, regp2_x_4);

  reg_res_1 = _mm_sub_pd(reg_res_1, reg_p3);

  // reg_res_2 = x * (reg_res_1 + reg_res_0)

  __m128d reg_res_2 = _mm_add_pd(reg_res_1, reg_res_0);

  reg_res_2 = _mm_mul_pd(reg_x, reg_res_2);

  // reg_res_3 = p[2] - p[0] + reg_res_2

  __m128d reg_res_3 = _mm_sub_pd(reg_p2, reg_p0);

  reg_res_3 = _mm_add_pd(reg_res_3, reg_res_2);

  // reg_res_4 = p[1] + 0.5 * x * reg_res_3

  __m128d reg_res_4 = _mm_mul_pd(_mm_mul_pd(half, reg_x), reg_res_3);

  reg_res_4 = _mm_add_pd(reg_p1, reg_res_4);

  _mm_storeu_pd(rate_dist_f, reg_res_4);

}