use criterion::Bencher;
use time::ext::{NumericalDuration, NumericalStdDuration};
use time::macros::{datetime, offset};

setup_benchmark! {
    "PrimitiveDateTime",

    // All getters are trivially dispatched to the relevant field, and do not need to be benchmarked
    // a second time.

    // region: attach offset
    fn assume_offset(ben: &mut Bencher<'_>) {
        ben.iter(|| datetime!("2019-01-01 0:00").assume_offset(offset!("UTC")));
        ben.iter(|| datetime!("2019-01-01 0:00").assume_offset(offset!("-1")));
    }

    fn assume_utc(ben: &mut Bencher<'_>) {
        ben.iter(|| datetime!("2019-01-01 0:00").assume_utc());
    }
    // endregion attach offset

    // region: trait impls
    fn add_duration(ben: &mut Bencher<'_>) {
        let a = 5.days();
        let b = 1.days();
        let c = 2.seconds();
        let d = (-2).seconds();
        let e = 1.hours();

        ben.iter(|| datetime!("2019-01-01 0:00") + a);
        ben.iter(|| datetime!("2019-12-31 0:00") + b);
        ben.iter(|| datetime!("2019-12-31 23:59:59") + c);
        ben.iter(|| datetime!("2020-01-01 0:00:01") + d);
        ben.iter(|| datetime!("1999-12-31 23:00") + e);
    }

    fn add_std_duration(ben: &mut Bencher<'_>) {
        let a = 5.std_days();
        let b = 1.std_days();
        let c = 2.std_seconds();

        ben.iter(|| datetime!("2019-01-01 0:00") + a);
        ben.iter(|| datetime!("2019-12-31 0:00") + b);
        ben.iter(|| datetime!("2019-12-31 23:59:59") + c);
    }

    fn add_assign_duration(ben: &mut Bencher<'_>) {
        let a = 1.days();
        let b = 1.seconds();
        iter_batched_ref!(
            ben,
            || datetime!("2019-01-01 0:00"),
            [
                |datetime| *datetime += a,
                |datetime| *datetime += b,
            ]
        );
    }

    fn add_assign_std_duration(ben: &mut Bencher<'_>) {
        let a = 1.std_days();
        let b = 1.std_seconds();
        iter_batched_ref!(
            ben,
            || datetime!("2019-01-01 0:00"),
            [
                |datetime| *datetime += a,
                |datetime| *datetime += b,
            ]
        );
    }

    fn sub_duration(ben: &mut Bencher<'_>) {
        let a = 5.days();
        let b = 1.days();
        let c = 2.seconds();
        let d = (-2).seconds();
        let e = (-1).hours();

        ben.iter(|| datetime!("2019-01-06 0:00") - a);
        ben.iter(|| datetime!("2020-01-01 0:00") - b);
        ben.iter(|| datetime!("2020-01-01 0:00:01") - c);
        ben.iter(|| datetime!("2019-12-31 23:59:59") - d);
        ben.iter(|| datetime!("1999-12-31 23:00") - e);
    }

    fn sub_std_duration(ben: &mut Bencher<'_>) {
        let a = 5.std_days();
        let b = 1.std_days();
        let c = 2.std_seconds();

        ben.iter(|| datetime!("2019-01-06 0:00") - a);
        ben.iter(|| datetime!("2020-01-01 0:00") - b);
        ben.iter(|| datetime!("2020-01-01 0:00:01") - c);
    }

    fn sub_assign_duration(ben: &mut Bencher<'_>) {
        let a = 1.days();
        let b = 1.seconds();
        iter_batched_ref!(
            ben,
            || datetime!("2019-01-01 0:00"),
            [
                |datetime| *datetime -= a,
                |datetime| *datetime -= b,
            ]
        );
    }

    fn sub_assign_std_duration(ben: &mut Bencher<'_>) {
        let a = 1.std_days();
        let b = 1.std_seconds();
        iter_batched_ref!(
            ben,
            || datetime!("2019-01-01 0:00"),
            [
                |datetime| *datetime -= a,
                |datetime| *datetime -= b,
            ]
        );
    }

    fn sub_datetime(ben: &mut Bencher<'_>) {
        ben.iter(|| datetime!("2019-01-02 0:00") - datetime!("2019-01-01 0:00"));
        ben.iter(|| datetime!("2019-01-01 0:00") - datetime!("2019-01-02 0:00"));
        ben.iter(|| datetime!("2020-01-01 0:00") - datetime!("2019-12-31 0:00"));
        ben.iter(|| datetime!("2019-12-31 0:00") - datetime!("2020-01-01 0:00"));
    }

    fn ord(ben: &mut Bencher<'_>) {
        ben.iter(|| datetime!("2019-01-01 0:00").partial_cmp(&datetime!("2019-01-01 0:00")));
        ben.iter(|| datetime!("2019-01-01 0:00").partial_cmp(&datetime!("2020-01-01 0:00")));
        ben.iter(|| datetime!("2019-01-01 0:00").partial_cmp(&datetime!("2019-02-01 0:00")));
        ben.iter(|| datetime!("2019-01-01 0:00").partial_cmp(&datetime!("2019-01-02 0:00")));
        ben.iter(|| datetime!("2019-01-01 0:00").partial_cmp(&datetime!("2019-01-01 1:00")));
        ben.iter(|| datetime!("2019-01-01 0:00").partial_cmp(&datetime!("2019-01-01 0:01")));
        ben.iter(|| datetime!("2019-01-01 0:00").partial_cmp(&datetime!("2019-01-01 0:00:01")));
        ben.iter(|| datetime!("2019-01-01 0:00").partial_cmp(&datetime!("2019-01-01 0:00:00.000_000_001")));
        ben.iter(|| datetime!("2020-01-01 0:00").partial_cmp(&datetime!("2019-01-01 0:00")));
        ben.iter(|| datetime!("2019-02-01 0:00").partial_cmp(&datetime!("2019-01-01 0:00")));
        ben.iter(|| datetime!("2019-01-02 0:00").partial_cmp(&datetime!("2019-01-01 0:00")));
        ben.iter(|| datetime!("2019-01-01 1:00").partial_cmp(&datetime!("2019-01-01 0:00")));
        ben.iter(|| datetime!("2019-01-01 0:01").partial_cmp(&datetime!("2019-01-01 0:00")));
        ben.iter(|| datetime!("2019-01-01 0:00:01").partial_cmp(&datetime!("2019-01-01 0:00")));
        ben.iter(|| datetime!("2019-01-01 0:00:00.000_000_001").partial_cmp(&datetime!("2019-01-01 0:00")));
    }
    // endregion trait impls
}
