use std::convert::TryInto;
use std::thread;
use std::time::Instant;

use plonky2::field::crandall_field::CrandallField;
use plonky2::field::field::Field;
use plonky2::rescue::rescue;

type F = CrandallField;

// 113 wire polys, 3 Z polys, 4 parts of quotient poly.
const PROVER_POLYS: usize = 113 + 3 + 4;

fn main() {
    const THREADS: usize = 12;
    const LDE_BITS: i32 = 3;
    const W: usize = 12;
    const HASHES_PER_POLY: usize = (1 << (13 + LDE_BITS)) / 6;

    let threads = (0..THREADS)
        .map(|_i| {
            thread::spawn(move || {
                let mut x: [F; W] = F::rand_vec(W).try_into().unwrap();

                let hashes_per_thread = HASHES_PER_POLY * PROVER_POLYS / THREADS;
                let start = Instant::now();
                for _ in 0..hashes_per_thread {
                    x = rescue(x);
                }
                let duration = start.elapsed();
                println!("took {:?}", duration);
                println!(
                    "avg {:?}us",
                    duration.as_secs_f64() * 1e6 / (hashes_per_thread as f64)
                );
                println!("result {:?}", x);
            })
        })
        .collect::<Vec<_>>();

    for t in threads {
        t.join().expect("oops");
    }
}