use env_logger::Env;
use plonky2::circuit_builder::CircuitBuilder;
use plonky2::circuit_data::CircuitConfig;
use plonky2::field::crandall_field::CrandallField;
use plonky2::field::extension_field::Extendable;
use plonky2::field::field::Field;
use plonky2::fri::FriConfig;
use plonky2::witness::PartialWitness;

fn main() {
    // Set the default log filter. This can be overridden using the `RUST_LOG` environment variable,
    // e.g. `RUST_LOG=debug`.
    // We default to debug for now, since there aren't many logs anyway, but we should probably
    // change this to info or warn later.
    env_logger::Builder::from_env(Env::default().default_filter_or("debug")).init();

    bench_prove::<CrandallField, 4>();
}

fn bench_prove<F: Field + Extendable<D>, const D: usize>() {
    let config = CircuitConfig {
        num_wires: 134,
        num_routed_wires: 27,
        security_bits: 128,
        rate_bits: 3,
        num_challenges: 3,
        fri_config: FriConfig {
            proof_of_work_bits: 1,
            rate_bits: 3,
            reduction_arity_bits: vec![2, 2, 2, 2, 2],
            num_query_rounds: 1,
        },
    };

    let mut builder = CircuitBuilder::<F, D>::new(config);

    let zero = builder.zero();
    let zero_ext = builder.zero_extension();

    let mut state = [zero; 12];
    for _ in 0..10000 {
        state = builder.permute(state);
    }

    // Random other gates.
    builder.add(zero, zero);
    builder.add_extension(zero_ext, zero_ext);

    let prover = builder.build_prover();
    let inputs = PartialWitness::new();
    prover.prove(inputs);
}