// SPDX-License-Identifier: MIT
/*
 * Copyright 2023 Advanced Micro Devices, Inc.
 *
 * Permission is hereby granted, free of charge, to any person obtaining a
 * copy of this software and associated documentation files (the "Software"),
 * to deal in the Software without restriction, including without limitation
 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
 * and/or sell copies of the Software, and to permit persons to whom the
 * Software is furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
 * THE COPYRIGHT HOLDER(S) OR AUTHOR(S) BE LIABLE FOR ANY CLAIM, DAMAGES OR
 * OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
 * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
 * OTHER DEALINGS IN THE SOFTWARE.
 *
 * Authors: AMD
 *
 */
#include "dm_services.h"
#include "bw_fixed.h"

#define MAX_I64 \
 ((int64_t)((1ULL << 63) - 1))

#define MIN_I64 \
 (-MAX_I64 - 1)

#define FRACTIONAL_PART_MASK \
 ((1ULL << BW_FIXED_BITS_PER_FRACTIONAL_PART) - 1)

#define GET_FRACTIONAL_PART(x) \
 (FRACTIONAL_PART_MASK & (x))

static uint64_t abs_i64(int64_t arg)
{
 if (arg >= 0)
  return (uint64_t)(arg);
 else
  return (uint64_t)(-arg);
}

struct bw_fixed bw_int_to_fixed_nonconst(int64_t value)
{
 struct bw_fixed res;

 ASSERT(value < BW_FIXED_MAX_I32 && value > BW_FIXED_MIN_I32);
 res.value = value << BW_FIXED_BITS_PER_FRACTIONAL_PART;
 return res;
}

struct bw_fixed bw_frc_to_fixed(int64_t numerator, int64_t denominator)
{
 struct bw_fixed res;
 bool arg1_negative = numerator < 0;
 bool arg2_negative = denominator < 0;
 uint64_t arg1_value;
 uint64_t arg2_value;
 uint64_t remainder;

 /* determine integer part */
 uint64_t res_value;

 ASSERT(denominator != 0);

 arg1_value = abs_i64(numerator);
 arg2_value = abs_i64(denominator);
 res_value = div64_u64_rem(arg1_value, arg2_value, &remainder);

 ASSERT(res_value <= BW_FIXED_MAX_I32);

 /* determine fractional part */
 {
  uint32_t i = BW_FIXED_BITS_PER_FRACTIONAL_PART;

  do {
   remainder <<= 1;

   res_value <<= 1;

   if (remainder >= arg2_value) {
    res_value |= 1;
    remainder -= arg2_value;
   }
  } while (--i != 0);
 }

 /* round up LSB */
 {
  uint64_t summand = (remainder << 1) >= arg2_value;

  ASSERT(res_value <= MAX_I64 - summand);

  res_value += summand;
 }

 res.value = (int64_t)(res_value);

 if (arg1_negative ^ arg2_negative)
  res.value = -res.value;
 return res;
}

struct bw_fixed bw_floor2(const struct bw_fixed arg,
     const struct bw_fixed significance)
{
 struct bw_fixed result;
 int64_t multiplicand;

 multiplicand = div64_s64(arg.value, abs_i64(significance.value));
 result.value = abs_i64(significance.value) * multiplicand;
 ASSERT(abs_i64(result.value) <= abs_i64(arg.value));
 return result;
}

struct bw_fixed bw_ceil2(const struct bw_fixed arg,
    const struct bw_fixed significance)
{
 struct bw_fixed result;
 int64_t multiplicand;

 multiplicand = div64_s64(arg.value, abs_i64(significance.value));
 result.value = abs_i64(significance.value) * multiplicand;
 if (abs_i64(result.value) < abs_i64(arg.value)) {
  if (arg.value < 0)
   result.value -= abs_i64(significance.value);
  else
   result.value += abs_i64(significance.value);
 }
 return result;
}

struct bw_fixed bw_mul(const struct bw_fixed arg1, const struct bw_fixed arg2)
{
 struct bw_fixed res;

 bool arg1_negative = arg1.value < 0;
 bool arg2_negative = arg2.value < 0;

 uint64_t arg1_value = abs_i64(arg1.value);
 uint64_t arg2_value = abs_i64(arg2.value);

 uint64_t arg1_int = BW_FIXED_GET_INTEGER_PART(arg1_value);
 uint64_t arg2_int = BW_FIXED_GET_INTEGER_PART(arg2_value);

 uint64_t arg1_fra = GET_FRACTIONAL_PART(arg1_value);
 uint64_t arg2_fra = GET_FRACTIONAL_PART(arg2_value);

 uint64_t tmp;

 res.value = arg1_int * arg2_int;

 ASSERT(res.value <= BW_FIXED_MAX_I32);

 res.value <<= BW_FIXED_BITS_PER_FRACTIONAL_PART;

 tmp = arg1_int * arg2_fra;

 ASSERT(tmp <= (uint64_t)(MAX_I64 - res.value));

 res.value += tmp;

 tmp = arg2_int * arg1_fra;

 ASSERT(tmp <= (uint64_t)(MAX_I64 - res.value));

 res.value += tmp;

 tmp = arg1_fra * arg2_fra;

 tmp = (tmp >> BW_FIXED_BITS_PER_FRACTIONAL_PART) +
  (tmp >= (uint64_t)(bw_frc_to_fixed(1, 2).value));

 ASSERT(tmp <= (uint64_t)(MAX_I64 - res.value));

 res.value += tmp;

 if (arg1_negative ^ arg2_negative)
  res.value = -res.value;
 return res;
}