use criterion::black_box;
use lambdaworks_math::{
    cyclic_group::IsGroup,
    elliptic_curve::{
        short_weierstrass::{
            curves::bls12_381::{
                curve::BLS12381Curve, pairing::BLS12381AtePairing, twist::BLS12381TwistCurve,
            },
            point::ShortWeierstrassProjectivePoint,
            traits::Compress,
        },
        traits::{IsEllipticCurve, IsPairing},
    },
};

use rand::{rngs::StdRng, Rng, SeedableRng};
#[allow(dead_code)]
type G1 = ShortWeierstrassProjectivePoint<BLS12381Curve>;
#[allow(dead_code)]
type G2 = ShortWeierstrassProjectivePoint<BLS12381TwistCurve>;

#[allow(dead_code)]
pub fn rand_points_g1() -> (G1, G1, u128, u128) {
    let mut rng = StdRng::seed_from_u64(42);
    let a_val = rng.gen();
    let b_val = rng.gen();
    let a = BLS12381Curve::generator().operate_with_self(a_val);
    let b = BLS12381Curve::generator().operate_with_self(b_val);
    (a, b, a_val, b_val)
}

#[allow(dead_code)]
pub fn rand_points_g2() -> (G2, G2, u128, u128) {
    let mut rng = StdRng::seed_from_u64(42);
    let a_val = rng.gen();
    let b_val = rng.gen();
    let a = BLS12381TwistCurve::generator().operate_with_self(a_val);
    let b = BLS12381TwistCurve::generator().operate_with_self(b_val);
    (a, b, a_val, b_val)
}

#[allow(dead_code)]
pub fn bls12_381_operate_with_g1() {
    let (a, b, _, _) = rand_points_g1();
    let _ = black_box(black_box(&a).operate_with(black_box(&b)));
}

#[allow(dead_code)]
pub fn bls12_381_operate_with_g2() {
    let (a, b, _, _) = rand_points_g1();
    let _ = black_box(black_box(&a).operate_with(black_box(&b)));
}

#[allow(dead_code)]
pub fn bls12_381_operate_with_self_g1() {
    let (a, _, _, b_val) = rand_points_g1();
    let _ = black_box(black_box(&a).operate_with_self(black_box(b_val)));
}

#[allow(dead_code)]
pub fn bls12_381_operate_with_self_g2() {
    let (a, _, _, b_val) = rand_points_g2();
    let _ = black_box(black_box(&a).operate_with_self(black_box(b_val)));
}

#[allow(dead_code)]
pub fn bls12_381_double_g1() {
    let (a, _, _, _) = rand_points_g1();
    let _ = black_box(black_box(&a).operate_with_self(black_box(2u64)));
}

#[allow(dead_code)]
pub fn bls12_381_double_g2() {
    let (a, _, _, _) = rand_points_g2();
    let _ = black_box(black_box(&a).operate_with_self(black_box(2u64)));
}

#[allow(dead_code)]
pub fn bls12_381_neg_g1() {
    let (a, _, _, _) = rand_points_g1();
    let _ = black_box(black_box(&a).neg());
}

#[allow(dead_code)]
pub fn bls12_381_neg_g2() {
    let (a, _, _, _) = rand_points_g2();
    let _ = black_box(black_box(&a).neg());
}

#[allow(dead_code)]
pub fn bls12_381_compress_g1() {
    let (a, _, _, _) = rand_points_g1();
    let _ = black_box(BLS12381Curve::compress_g1_point(black_box(&a)));
}

#[allow(dead_code)]
pub fn bls12_381_decompress_g1() {
    let (a, _, _, _) = rand_points_g1();
    let a: [u8; 48] = BLS12381Curve::compress_g1_point(&a);
    let _ = black_box(BLS12381Curve::decompress_g1_point(&mut black_box(a))).unwrap();
}

#[allow(dead_code)]
pub fn bls12_381_subgroup_check_g1() {
    let (a, _, _, _) = rand_points_g1();
    let _ = black_box(black_box(&a.is_in_subgroup()));
}

#[allow(dead_code)]
pub fn bls12_381_ate_pairing() {
    let (a, _, _, _) = rand_points_g1();
    let (_, b, _, _) = rand_points_g2();
    let _ = black_box(BLS12381AtePairing::compute(black_box(&a), black_box(&b)));
}