// Copyright © 2024 Mikhail Hogrefe
//
// This file is part of Malachite.
//
// Malachite is free software: you can redistribute it and/or modify it under the terms of the GNU
// Lesser General Public License (LGPL) as published by the Free Software Foundation; either version
// 3 of the License, or (at your option) any later version. See <https://www.gnu.org/licenses/>.

use malachite_base::num::float::NiceFloat;
use malachite_base::random::EXAMPLE_SEED;
use malachite_base::test_util::stats::moments::MomentStats;
use malachite_float::random::random_non_negative_finite_floats;
use malachite_float::test_util::random::random_floats_helper_helper;

fn random_non_negative_finite_floats_helper(
    mean_sci_exponent_abs_numerator: u64,
    mean_sci_exponent_abs_denominator: u64,
    mean_precision_numerator: u64,
    mean_precision_denominator: u64,
    mean_zero_p_numerator: u64,
    mean_zero_p_denominator: u64,
    expected_values: &[&str],
    expected_values_hex: &[&str],
    expected_common_values: &[(&str, usize)],
    expected_common_values_hex: &[(&str, usize)],
    expected_median: (&str, Option<&str>),
    expected_median_hex: (&str, Option<&str>),
    expected_moment_stats: MomentStats,
) {
    random_floats_helper_helper(
        random_non_negative_finite_floats(
            EXAMPLE_SEED,
            mean_sci_exponent_abs_numerator,
            mean_sci_exponent_abs_denominator,
            mean_precision_numerator,
            mean_precision_denominator,
            mean_zero_p_numerator,
            mean_zero_p_denominator,
        ),
        expected_values,
        expected_values_hex,
        expected_common_values,
        expected_common_values_hex,
        expected_median,
        expected_median_hex,
        expected_moment_stats,
    );
}

#[test]
fn test_random_non_negative_finite_floats() {
    // mean |sci_exponent| 1, mean precision 2, zero probability 1/10
    let values = &[
        "8.0", "1.0", "0.8", "0.0", "4.0", "2.0", "1.0", "2.0", "0.8", "0.9536", "2.0", "3.0e1",
        "0.5", "0.0", "1.0", "2.0", "0.5", "0.0", "1.5", "0.1",
    ];
    let values_hex = &[
        "0x8.0#1",
        "0x1.0#1",
        "0x0.c#2",
        "0x0.0",
        "0x4.0#1",
        "0x2.0#3",
        "0x1.0#1",
        "0x2.0#1",
        "0x0.c#2",
        "0x0.f42#12",
        "0x2.0#1",
        "0x2.0E+1#1",
        "0x0.8#1",
        "0x0.0",
        "0x1.0#1",
        "0x2.0#2",
        "0x0.8#1",
        "0x0.0",
        "0x1.8#2",
        "0x0.2#1",
    ];
    let common_values = &[
        ("1.0", 149586),
        ("0.0", 100224),
        ("2.0", 74929),
        ("0.5", 74827),
        ("1.0", 37902),
        ("4.0", 37612),
        ("0.2", 37602),
        ("1.5", 37600),
        ("0.5", 18934),
        ("0.8", 18834),
    ];
    let common_values_hex = &[
        ("0x1.0#1", 149586),
        ("0x0.0", 100224),
        ("0x2.0#1", 74929),
        ("0x0.8#1", 74827),
        ("0x1.0#2", 37902),
        ("0x4.0#1", 37612),
        ("0x0.4#1", 37602),
        ("0x1.8#2", 37600),
        ("0x0.8#2", 18934),
        ("0x0.c#2", 18834),
    ];
    let sample_median = ("1.0", None);
    let sample_median_hex = ("0x1.0#1", None);
    let sample_moment_stats = MomentStats {
        mean: NiceFloat(8.693371936934735),
        standard_deviation: NiceFloat(1671.7840625385452),
        skewness: NiceFloat(718.2460436055444),
        excess_kurtosis: NiceFloat(578520.0792964109),
    };
    random_non_negative_finite_floats_helper(
        1,
        1,
        2,
        1,
        1,
        10,
        values,
        values_hex,
        common_values,
        common_values_hex,
        sample_median,
        sample_median_hex,
        sample_moment_stats,
    );

    // mean |sci_exponent| 1, mean precision 64, zero probability 1/10
    let values = &[
        "8.288616155488807101232282300740634937242230275473095947256061292584667312127693953525219\
        033439102",
        "1.0",
        "0.6651592248",
        "0.0",
        "4.2345242873846765390825841",
        "2.38320785690539951782446111823006788535",
        "1.036853813",
        "2.4027026671793265640566992824",
        "0.52666405862440172",
        "0.5366",
        "3.42175",
        "61.701080114490962438",
        "0.8336708437",
        "0.0",
        "1.0",
        "2.5",
        "0.658027364385",
        "0.0",
        "1.246",
        "0.14",
    ];
    let values_hex = &[
        "0x8.49e2bf94ebf30cb80717aab089ffe81a4240bd550c84a499a4753e306f26fd3913222c258d89033b#324",
        "0x1.0#1",
        "0x0.aa47dffd#32",
        "0x0.0",
        "0x4.3c09c8a06f55060972e84#86",
        "0x2.6219e8fcfa9bb9ef41add9535f62860#125",
        "0x1.096f406#28",
        "0x2.671785a18c13ffbbe8b27f4#93",
        "0x0.86d374abc53b0a#55",
        "0x0.896#12",
        "0x3.6bf8#16",
        "0x3d.b379fc839d57b1c#65",
        "0x0.d56b73d18#33",
        "0x0.0",
        "0x1.0#2",
        "0x2.8#4",
        "0x0.a8747b39e8#37",
        "0x0.0",
        "0x1.3f#9",
        "0x0.24#4",
    ];
    let common_values = &[
        ("0.0", 100224),
        ("1.0", 4643),
        ("2.0", 2373),
        ("0.5", 2353),
        ("1.5", 2327),
        ("1.0", 2294),
        ("0.5", 1219),
        ("4.0", 1204),
        ("2.0", 1195),
        ("0.8", 1191),
    ];
    let common_values_hex = &[
        ("0x0.0", 100224),
        ("0x1.0#1", 4643),
        ("0x2.0#1", 2373),
        ("0x0.8#1", 2353),
        ("0x1.8#2", 2327),
        ("0x1.0#2", 2294),
        ("0x0.8#2", 1219),
        ("0x4.0#1", 1204),
        ("0x2.0#2", 1195),
        ("0x0.c#2", 1191),
    ];
    let sample_median = (
        "1.31267178118571998379",
        Some("1.31268190290298901432879852014280461789620659172002407"),
    );
    let sample_median_hex = (
        "0x1.500b42029321dc7ec#67",
        Some("0x1.500bebd304f66efb8d7a481eef659ad171581e14dea8#175"),
    );
    let sample_moment_stats = MomentStats {
        mean: NiceFloat(10.510498034617717),
        standard_deviation: NiceFloat(1688.742389206287),
        skewness: NiceFloat(594.3533044311691),
        excess_kurtosis: NiceFloat(387838.0226770939),
    };
    random_non_negative_finite_floats_helper(
        1,
        1,
        64,
        1,
        1,
        10,
        values,
        values_hex,
        common_values,
        common_values_hex,
        sample_median,
        sample_median_hex,
        sample_moment_stats,
    );

    // mean |sci_exponent| 64, mean precision 2, zero probability 1/10
    let values = &[
        "1.0e9", "2.0e-6", "24.0", "0.0", "5.0e-7", "0.06", "8.0e22", "7.0e19", "0.05", "6.4e7",
        "2.0e38", "7.0e-9", "2.0e15", "0.0", "3.0e11", "1.0e-8", "6.0e14", "0.0", "0.05",
        "2.0e-34",
    ];
    let values_hex = &[
        "0x4.0E+7#1",
        "0x0.00002#1",
        "0x18.0#2",
        "0x0.0",
        "0x8.0E-6#1",
        "0x0.10#3",
        "0x1.0E+19#1",
        "0x4.0E+16#1",
        "0x0.0c#2",
        "0x3.d08E+6#12",
        "0x8.0E+31#1",
        "0x2.0E-7#1",
        "0x8.0E+12#1",
        "0x0.0",
        "0x4.0E+9#1",
        "0x4.0E-7#2",
        "0x2.0E+12#1",
        "0x0.0",
        "0x0.0c#2",
        "0x1.0E-28#1",
    ];
    let common_values = &[
        ("0.0", 100224),
        ("1.0", 3531),
        ("2.0", 3503),
        ("4.0", 3399),
        ("0.5", 3381),
        ("0.1", 3375),
        ("0.2", 3369),
        ("0.06", 3330),
        ("8.0", 3283),
        ("2.0e1", 3250),
    ];
    let common_values_hex = &[
        ("0x0.0", 100224),
        ("0x1.0#1", 3531),
        ("0x2.0#1", 3503),
        ("0x4.0#1", 3399),
        ("0x0.8#1", 3381),
        ("0x0.2#1", 3375),
        ("0x0.4#1", 3369),
        ("0x0.1#1", 3330),
        ("0x8.0#1", 3283),
        ("0x1.0E+1#1", 3250),
    ];
    let sample_median = ("0.006", Some("0.0059"));
    let sample_median_hex = ("0x0.018#3", Some("0x0.018#4"));
    let sample_moment_stats = MomentStats {
        mean: NiceFloat(2.237414368630999e242),
        standard_deviation: NiceFloat(f64::INFINITY),
        skewness: NiceFloat(f64::NAN),
        excess_kurtosis: NiceFloat(f64::NAN),
    };
    random_non_negative_finite_floats_helper(
        64,
        1,
        2,
        1,
        1,
        10,
        values,
        values_hex,
        common_values,
        common_values_hex,
        sample_median,
        sample_median_hex,
        sample_moment_stats,
    );

    // mean |sci_exponent| 64, mean precision 64, zero probability 1/10
    let values = &[
        "1112479228.653802418557662930660020738554074733226813063166716380919457294269645928321492\
        4893705523",
        "2.0e-6",
        "21.285095192",
        "0.0",
        "5.0479463188465553988010694e-7",
        "0.074475245528293734932014409944689621417",
        "7.83424591e22",
        "88644082373352750967.8179473",
        "0.032916503664025107",
        "3.601e7",
        "2.91091e38",
        "1.43659021971959067597e-8",
        "3.7545197012e15",
        "0.0",
        "3.0e11",
        "1.9e-8",
        "7.4087294826e14",
        "0.0",
        "0.0389",
        "2.2e-34",
    ];
    let values_hex = &[
        "0x424f15fc.a75f9865c038bd55844fff40d21205eaa8642524cd23a9f1837937e9c89911612c6c4819d8#324",
        "0x0.00002#1",
        "0x15.48fbffa#32",
        "0x0.0",
        "0x8.78139140deaa0c12e5d08E-6#86",
        "0x0.1310cf47e7d4ddcf7a0d6eca9afb1430#125",
        "0x1.096f406E+19#28",
        "0x4ce2f0b431827ff77.d164fe8#93",
        "0x0.086d374abc53b0a#55",
        "0x2.258E+6#12",
        "0xd.afeE+31#16",
        "0x3.db379fc839d57b1cE-7#65",
        "0xd.56b73d18E+12#33",
        "0x0.0",
        "0x4.0E+9#2",
        "0x5.0E-7#4",
        "0x2.a1d1ece7aE+12#37",
        "0x0.0",
        "0x0.09f8#9",
        "0x1.2E-28#4",
    ];
    let common_values = &[
        ("0.0", 100224),
        ("8.0", 129),
        ("4.0", 120),
        ("2.0", 107),
        ("0.06", 107),
        ("1.0e3", 106),
        ("3.0e1", 105),
        ("4.0e3", 104),
        ("0.5", 102),
        ("1.0e2", 102),
    ];
    let common_values_hex = &[
        ("0x0.0", 100224),
        ("0x8.0#1", 129),
        ("0x4.0#1", 120),
        ("0x2.0#1", 107),
        ("0x0.1#1", 107),
        ("0x4.0E+2#1", 106),
        ("0x2.0E+1#1", 105),
        ("0x1.0E+3#1", 104),
        ("0x0.8#1", 102),
        ("0x8.0E+1#1", 102),
    ];
    let sample_median = ("0.007447481", Some("0.0074479832945"));
    let sample_median_hex = ("0x0.01e8140#21", Some("0x0.01e81c6cabc#35"));
    let sample_moment_stats = MomentStats {
        mean: NiceFloat(3.6946902524964283e242),
        standard_deviation: NiceFloat(f64::INFINITY),
        skewness: NiceFloat(f64::NAN),
        excess_kurtosis: NiceFloat(f64::NAN),
    };
    random_non_negative_finite_floats_helper(
        64,
        1,
        64,
        1,
        1,
        10,
        values,
        values_hex,
        common_values,
        common_values_hex,
        sample_median,
        sample_median_hex,
        sample_moment_stats,
    );
}

#[test]
#[should_panic]
fn random_non_negative_finite_floats_fail_1() {
    random_non_negative_finite_floats(EXAMPLE_SEED, 1, 0, 2, 1, 1, 10);
}

#[test]
#[should_panic]
fn random_non_negative_finite_floats_fail_2() {
    random_non_negative_finite_floats(EXAMPLE_SEED, 0, 1, 2, 1, 1, 10);
}

#[test]
#[should_panic]
fn random_non_negative_finite_floats_fail_3() {
    random_non_negative_finite_floats(EXAMPLE_SEED, 1, 1, 1, 0, 1, 10);
}

#[test]
#[should_panic]
fn random_non_negative_finite_floats_fail_4() {
    random_non_negative_finite_floats(EXAMPLE_SEED, 1, 1, 1, 1, 1, 10);
}

#[test]
#[should_panic]
fn random_non_negative_finite_floats_fail_5() {
    random_non_negative_finite_floats(EXAMPLE_SEED, 1, 1, 2, 1, 1, 0);
}

#[test]
#[should_panic]
fn random_non_negative_finite_floats_fail_6() {
    random_non_negative_finite_floats(EXAMPLE_SEED, 1, 1, 2, 1, 2, 1);
}