#![feature(test)]
#![feature(try_from)]
use std::convert::TryFrom;

#[macro_use]
extern crate enum_tryfrom_derive;
extern crate enum_tryfrom;
extern crate rand;
extern crate test;

use test::Bencher;
use std::mem::transmute;
use rand::{Rng, SeedableRng, StdRng};

#[derive(PartialEq,Eq,Debug,TryFromPrimitive)]
#[TryFromPrimitiveType="u8"]
enum Foo {
    ZerothFoo = 0,
    FirstFoo,
    SecondFoo,
    ThirdFoo,
}

#[bench]
fn bench_raw(b: &mut Bencher) {
    let mut values = (0..255).map(|x| x % 4).collect::<Vec<u8>>();
    let mut rng = StdRng::from_seed(&[4]);
    rng.shuffle(&mut values);

    b.iter(|| {
        values.iter().map(|x| {
            if *x > 3 as u8 {panic!("x out of range!")};
            unsafe{transmute::<u8, Foo>(*x)}
        }).collect::<Vec<Foo>>()
    })
}

#[bench]
fn bench_from(b: &mut Bencher) {
    let mut values = (0..255).map(|x| x % 4).collect::<Vec<u8>>();
    let mut rng = StdRng::from_seed(&[4]);
    rng.shuffle(&mut values);

    b.iter(|| {
        values.iter().map(|x| Foo::try_from(*x).unwrap()).collect::<Vec<Foo>>()
    })
}