#![cfg_attr(not(feature = "no-f16-f128"), feature(f16))]
#![cfg_attr(not(feature = "no-f16-f128"), feature(f128))]
// makes configuration easier
#![allow(unused_macros)]
#![allow(unused_imports)]

use compiler_builtins::float::Float;
use rustc_apfloat::{Float as _, FloatConvert as _};
use testcrate::*;

mod int_to_float {
    use super::*;

    macro_rules! i_to_f {
        ($($from:ty, $into:ty, $fn:ident);*;) => {
            $(
                #[test]
                fn $fn() {
                    use compiler_builtins::float::conv::$fn;
                    use compiler_builtins::int::Int;

                    fuzz(N, |x: $from| {
                        let f0 = x as $into;
                        let f1: $into = $fn(x);
                        // This makes sure that the conversion produced the best rounding possible, and does
                        // this independent of `x as $into` rounding correctly.
                        // This assumes that float to integer conversion is correct.
                        let y_minus_ulp = <$into>::from_bits(f1.to_bits().wrapping_sub(1)) as $from;
                        let y = f1 as $from;
                        let y_plus_ulp = <$into>::from_bits(f1.to_bits().wrapping_add(1)) as $from;
                        let error_minus = <$from as Int>::abs_diff(y_minus_ulp, x);
                        let error = <$from as Int>::abs_diff(y, x);
                        let error_plus = <$from as Int>::abs_diff(y_plus_ulp, x);
                        // The first two conditions check that none of the two closest float values are
                        // strictly closer in representation to `x`. The second makes sure that rounding is
                        // towards even significand if two float values are equally close to the integer.
                        if error_minus < error
                            || error_plus < error
                            || ((error_minus == error || error_plus == error)
                                && ((f0.to_bits() & 1) != 0))
                        {
                            if !cfg!(any(
                                target_arch = "powerpc",
                                target_arch = "powerpc64"
                            )) {
                                panic!(
                                    "incorrect rounding by {}({}): {}, ({}, {}, {}), errors ({}, {}, {})",
                                    stringify!($fn),
                                    x,
                                    f1.to_bits(),
                                    y_minus_ulp,
                                    y,
                                    y_plus_ulp,
                                    error_minus,
                                    error,
                                    error_plus,
                                );
                            }
                        }
                        // Test against native conversion. We disable testing on all `x86` because of
                        // rounding bugs with `i686`. `powerpc` also has the same rounding bug.
                        if f0 != f1 && !cfg!(any(
                            target_arch = "x86",
                            target_arch = "powerpc",
                            target_arch = "powerpc64"
                        )) {
                            panic!(
                                "{}({}): std: {}, builtins: {}",
                                stringify!($fn),
                                x,
                                f0,
                                f1,
                            );
                        }
                    });
                }
            )*
        };
    }

    i_to_f! {
        u32, f32, __floatunsisf;
        u32, f64, __floatunsidf;
        i32, f32, __floatsisf;
        i32, f64, __floatsidf;
        u64, f32, __floatundisf;
        u64, f64, __floatundidf;
        i64, f32, __floatdisf;
        i64, f64, __floatdidf;
        u128, f32, __floatuntisf;
        u128, f64, __floatuntidf;
        i128, f32, __floattisf;
        i128, f64, __floattidf;
    }
}

// PowerPC tests are failing on LLVM 13: https://github.com/rust-lang/rust/issues/88520
#[cfg(not(target_arch = "powerpc64"))]
mod f_to_i {
    use super::*;

    macro_rules! f_to_i {
        ($x:ident, $f_ty:ty, $apfloat_ty:ident, $sys_available:meta, $($i_ty:ty, $fn:ident);*;) => {
            $(
                // it is undefined behavior in the first place to do conversions with NaNs
                if !apfloat_fallback!(
                    $f_ty, $apfloat_ty, $sys_available, |x: FloatTy| x.is_nan() => no_convert, $x
                ) {
                    let conv0 = apfloat_fallback!(
                        $f_ty, $apfloat_ty, $sys_available,
                        // Use an `as` cast when the builtin is available on the system.
                        |x| x as $i_ty;
                        // When the builtin is not available, we need to use a different conversion
                        // method (since apfloat doesn't support `as` casting).
                        |x: $f_ty| {
                            use compiler_builtins::int::MinInt;

                            let apf = FloatTy::from_bits(x.to_bits().into());
                            let bits: usize = <$i_ty>::BITS.try_into().unwrap();

                            let err_fn = || panic!(
                                "Unable to convert value {x:?} to type {}:", stringify!($i_ty)
                            );

                            if <$i_ty>::SIGNED {
                               <$i_ty>::try_from(apf.to_i128(bits).value).ok().unwrap_or_else(err_fn)
                            } else {
                               <$i_ty>::try_from(apf.to_u128(bits).value).ok().unwrap_or_else(err_fn)
                            }
                        },
                        $x
                    );
                    let conv1: $i_ty = $fn($x);
                    if conv0 != conv1 {
                        panic!("{}({:?}): std: {:?}, builtins: {:?}", stringify!($fn), $x, conv0, conv1);
                    }
                }
            )*
        };
    }

    #[test]
    fn f32_to_int() {
        use compiler_builtins::float::conv::{
            __fixsfdi, __fixsfsi, __fixsfti, __fixunssfdi, __fixunssfsi, __fixunssfti,
        };

        fuzz_float(N, |x: f32| {
            f_to_i!(x, f32, Single, all(),
                u32, __fixunssfsi;
                u64, __fixunssfdi;
                u128, __fixunssfti;
                i32, __fixsfsi;
                i64, __fixsfdi;
                i128, __fixsfti;
            );
        });
    }

    #[test]
    fn f64_to_int() {
        use compiler_builtins::float::conv::{
            __fixdfdi, __fixdfsi, __fixdfti, __fixunsdfdi, __fixunsdfsi, __fixunsdfti,
        };

        fuzz_float(N, |x: f64| {
            f_to_i!(x, f64, Double, all(),
                u32, __fixunsdfsi;
                u64, __fixunsdfdi;
                u128, __fixunsdfti;
                i32, __fixdfsi;
                i64, __fixdfdi;
                i128, __fixdfti;
            );
        });
    }

    #[test]
    #[cfg(not(feature = "no-f16-f128"))]
    fn f128_to_int() {
        #[cfg(any(target_arch = "powerpc", target_arch = "powerpc64"))]
        use compiler_builtins::float::conv::{
            __fixkfdi as __fixtfdi, __fixkfsi as __fixtfsi, __fixkfti as __fixtfti,
            __fixunskfdi as __fixunstfdi, __fixunskfsi as __fixunstfsi,
            __fixunskfti as __fixunstfti,
        };
        #[cfg(not(any(target_arch = "powerpc", target_arch = "powerpc64")))]
        use compiler_builtins::float::conv::{
            __fixtfdi, __fixtfsi, __fixtfti, __fixunstfdi, __fixunstfsi, __fixunstfti,
        };

        fuzz_float(N, |x: f128| {
            f_to_i!(
                x,
                f128,
                Quad,
                not(feature = "no-sys-f128-int-convert"),
                u32, __fixunstfsi;
                u64, __fixunstfdi;
                u128, __fixunstfti;
                i32, __fixtfsi;
                i64, __fixtfdi;
                i128, __fixtfti;
            );
        });
    }
}

macro_rules! conv {
    ($fX:ident, $fD:ident, $fn:ident, $apfloatX:ident, $apfloatD:ident) => {
        fuzz_float(N, |x: $fX| {
            let tmp0: $apfloatD = $apfloatX::from_bits(x.to_bits().into())
                .convert(&mut false)
                .value;
            let tmp0 = $fD::from_bits(tmp0.to_bits().try_into().unwrap());
            let tmp1: $fD = $fn(x);
            if !Float::eq_repr(tmp0, tmp1) {
                panic!(
                    "{}({x:?}): apfloat: {tmp0:?}, builtins: {tmp1:?}",
                    stringify!($fn)
                );
            }
        })
    };
}

macro_rules! extend {
    ($fX:ident, $fD:ident, $fn:ident) => {
        #[test]
        fn $fn() {
            use compiler_builtins::float::extend::$fn;

            fuzz_float(N, |x: $fX| {
                let tmp0 = x as $fD;
                let tmp1: $fD = $fn(x);
                if !Float::eq_repr(tmp0, tmp1) {
                    panic!(
                        "{}({}): std: {}, builtins: {}",
                        stringify!($fn),
                        x,
                        tmp0,
                        tmp1
                    );
                }
            });
        }
    };
}

// PowerPC tests are failing on LLVM 13: https://github.com/rust-lang/rust/issues/88520
#[cfg(not(target_arch = "powerpc64"))]
mod float_extend {
    use super::*;

    extend!(f32, f64, __extendsfdf2);

    #[test]
    fn conv() {
        use compiler_builtins::float::extend::__extendsfdf2;
        use rustc_apfloat::ieee::{Double, Single};

        conv!(f32, f64, __extendsfdf2, Single, Double);
    }
}

#[cfg(not(feature = "no-f16-f128"))]
#[cfg(not(any(target_arch = "powerpc", target_arch = "powerpc64")))]
mod float_extend_f128 {
    use super::*;

    #[test]
    fn conv() {
        use compiler_builtins::float::extend::{
            __extenddftf2, __extendhfsf2, __extendhftf2, __extendsftf2, __gnu_h2f_ieee,
        };
        use rustc_apfloat::ieee::{Double, Half, Quad, Single};

        // FIXME(f16_f128): Also do extend!() for `f16` and `f128` when builtins are in nightly
        conv!(f16, f32, __extendhfsf2, Half, Single);
        conv!(f16, f32, __gnu_h2f_ieee, Half, Single);
        conv!(f16, f128, __extendhftf2, Half, Quad);
        conv!(f32, f128, __extendsftf2, Single, Quad);
        conv!(f64, f128, __extenddftf2, Double, Quad);
    }
}

#[cfg(not(feature = "no-f16-f128"))]
#[cfg(any(target_arch = "powerpc", target_arch = "powerpc64"))]
mod float_extend_f128_ppc {
    use super::*;

    #[test]
    fn conv() {
        use compiler_builtins::float::extend::{
            __extenddfkf2, __extendhfkf2, __extendhfsf2, __extendsfkf2, __gnu_h2f_ieee,
        };
        use rustc_apfloat::ieee::{Double, Half, Quad, Single};

        // FIXME(f16_f128): Also do extend!() for `f16` and `f128` when builtins are in nightly
        conv!(f16, f32, __extendhfsf2, Half, Single);
        conv!(f16, f32, __gnu_h2f_ieee, Half, Single);
        conv!(f16, f128, __extendhfkf2, Half, Quad);
        conv!(f32, f128, __extendsfkf2, Single, Quad);
        conv!(f64, f128, __extenddfkf2, Double, Quad);
    }
}

#[cfg(target_arch = "arm")]
mod float_extend_arm {
    use super::*;

    extend!(f32, f64, __extendsfdf2vfp);

    #[test]
    fn conv() {
        use compiler_builtins::float::extend::__extendsfdf2vfp;
        use rustc_apfloat::ieee::{Double, Single};

        conv!(f32, f64, __extendsfdf2vfp, Single, Double);
    }
}

macro_rules! trunc {
    ($fX:ident, $fD:ident, $fn:ident) => {
        #[test]
        fn $fn() {
            use compiler_builtins::float::trunc::$fn;

            fuzz_float(N, |x: $fX| {
                let tmp0 = x as $fD;
                let tmp1: $fD = $fn(x);
                if !Float::eq_repr(tmp0, tmp1) {
                    panic!(
                        "{}({}): std: {}, builtins: {}",
                        stringify!($fn),
                        x,
                        tmp0,
                        tmp1
                    );
                }
            });
        }
    };
}

// PowerPC tests are failing on LLVM 13: https://github.com/rust-lang/rust/issues/88520
#[cfg(not(target_arch = "powerpc64"))]
mod float_trunc {
    use super::*;

    trunc!(f64, f32, __truncdfsf2);

    #[test]
    fn conv() {
        use compiler_builtins::float::trunc::__truncdfsf2;
        use rustc_apfloat::ieee::{Double, Single};

        conv!(f64, f32, __truncdfsf2, Double, Single);
    }
}

#[cfg(not(feature = "no-f16-f128"))]
#[cfg(not(any(target_arch = "powerpc", target_arch = "powerpc64")))]
mod float_trunc_f128 {
    use super::*;

    #[test]
    fn conv() {
        use compiler_builtins::float::trunc::{__gnu_f2h_ieee, __truncdfhf2, __truncsfhf2};
        use compiler_builtins::float::trunc::{__trunctfdf2, __trunctfhf2, __trunctfsf2};
        use rustc_apfloat::ieee::{Double, Half, Quad, Single};

        // FIXME(f16_f128): Also do trunc!() for `f16` and `f128` when builtins are in nightly
        conv!(f32, f16, __truncsfhf2, Single, Half);
        conv!(f32, f16, __gnu_f2h_ieee, Single, Half);
        conv!(f64, f16, __truncdfhf2, Double, Half);
        conv!(f128, f16, __trunctfhf2, Quad, Half);
        conv!(f128, f32, __trunctfsf2, Quad, Single);
        conv!(f128, f64, __trunctfdf2, Quad, Double);
    }
}

#[cfg(not(feature = "no-f16-f128"))]
#[cfg(any(target_arch = "powerpc", target_arch = "powerpc64"))]
mod float_trunc_f128_ppc {
    use super::*;

    #[test]
    fn conv() {
        use compiler_builtins::float::trunc::{__gnu_f2h_ieee, __truncdfhf2, __truncsfhf2};
        use compiler_builtins::float::trunc::{__trunckfdf2, __trunckfhf2, __trunckfsf2};
        use rustc_apfloat::ieee::{Double, Half, Quad, Single};

        // FIXME(f16_f128): Also do trunc!() for `f16` and `f128` when builtins are in nightly
        conv!(f32, f16, __truncsfhf2, Single, Half);
        conv!(f32, f16, __gnu_f2h_ieee, Single, Half);
        conv!(f64, f16, __truncdfhf2, Double, Half);
        conv!(f128, f16, __trunckfhf2, Quad, Half);
        conv!(f128, f32, __trunckfsf2, Quad, Single);
        conv!(f128, f64, __trunckfdf2, Quad, Double);
    }
}

#[cfg(target_arch = "arm")]
mod float_trunc_arm {
    use super::*;

    trunc!(f64, f32, __truncdfsf2vfp);

    #[test]
    fn conv() {
        use compiler_builtins::float::trunc::__truncdfsf2vfp;
        use rustc_apfloat::ieee::{Double, Single};

        conv!(f64, f32, __truncdfsf2vfp, Double, Single)
    }
}