#![feature(test)]

extern crate deflate;
extern crate test;
extern crate flate2;

use std::io::Write;
use std::io;

use test::Bencher;
use flate2::Compression;
use flate2::write;
use deflate::{CompressionOptions, deflate_bytes_zlib_conf, deflate_bytes_zlib};

fn load_from_file(name: &str) -> Vec<u8> {
    use std::fs::File;
    use std::io::Read;
    let mut input = Vec::new();
    let mut f = File::open(name).unwrap();

    f.read_to_end(&mut input).unwrap();
    input
}

fn get_test_data() -> Vec<u8> {
    use std::env;
    let path = env::var("TEST_FILE").unwrap_or_else(|_| "tests/pg11.txt".to_string());
    load_from_file(&path)
}

#[bench]
fn test_file_zlib_def(b: &mut Bencher) {
    let test_data = get_test_data();

    b.iter(|| deflate_bytes_zlib(&test_data));
}

#[bench]
fn test_file_zlib_best(b: &mut Bencher) {
    let test_data = get_test_data();

    b.iter(|| {
        deflate_bytes_zlib_conf(&test_data, CompressionOptions::high())
    });
}

#[bench]
fn test_file_zlib_fast(b: &mut Bencher) {
    let test_data = get_test_data();

    b.iter(|| {
        deflate_bytes_zlib_conf(&test_data, CompressionOptions::fast())
    });
}

#[bench]
fn test_file_zlib_rle(b: &mut Bencher) {
    let test_data = get_test_data();

    b.iter(|| {
        deflate_bytes_zlib_conf(&test_data, CompressionOptions::rle())
    });
}


fn deflate_bytes_flate2_zlib(level: Compression, input: &[u8]) -> Vec<u8> {
    use flate2::write::ZlibEncoder;
    use std::io::Write;

    let mut e = ZlibEncoder::new(Vec::with_capacity(input.len() / 3), level);
    e.write_all(input).unwrap();
    e.finish().unwrap()
}

#[bench]
fn test_file_zlib_flate2_def(b: &mut Bencher) {
    let test_data = get_test_data();
    b.iter(|| {
        deflate_bytes_flate2_zlib(Compression::Default, &test_data)
    });
}

#[bench]
fn test_file_zlib_flate2_best(b: &mut Bencher) {
    let test_data = get_test_data();
    b.iter(|| deflate_bytes_flate2_zlib(Compression::Best, &test_data));
}

#[bench]
fn test_file_zlib_flate2_fast(b: &mut Bencher) {
    let test_data = get_test_data();
    b.iter(|| deflate_bytes_flate2_zlib(Compression::Fast, &test_data));
}

#[derive(Copy, Clone)]
struct Dummy {}

impl Write for Dummy {
    fn write(&mut self, buf: &[u8]) -> io::Result<usize> {
        Ok(buf.len())
    }

    fn flush(&mut self) -> io::Result<()> {
        Ok(())
    }
}

#[bench]
fn writer_create(b: &mut Bencher) {
    use deflate::write::DeflateEncoder;
    b.iter(|| DeflateEncoder::new(Dummy {}, CompressionOptions::fast()));
}

#[bench]
fn writer_create_flate2(b: &mut Bencher) {
    b.iter(|| write::DeflateEncoder::new(Dummy {}, Compression::Fast));
}