use std;

extern crate hound;

type FileReader = std::io::BufReader<std::fs::File>;

pub fn float_sample_iterator<'a>(reader: &'a mut hound::WavReader<FileReader>)
    -> Box<Iterator<Item=f32> + 'a> {
    match reader.spec().sample_format {
        hound::SampleFormat::Float => Box::new(reader.samples::<f32>().map(|x| x.unwrap())),
        hound::SampleFormat::Int => match reader.spec().bits_per_sample {
            8 => Box::new(reader.samples::<i8>().map(|x| (x.unwrap() as f32) / (i16::max_value() as f32))),
            16 => Box::new(reader.samples::<i16>().map(|x| (x.unwrap() as f32) / (i16::max_value() as f32))),
            32 => Box::new(reader.samples::<i32>().map(|x| (x.unwrap() as f32) / (i32::max_value() as f32))),
            _ => panic!("Unsupported sample rate")
        }
    }
}
