use pylon;
use image;
use std::error::Error;

fn main() -> Result<(), Box<dyn Error>> {
  pylon::initialize();
  let devices = pylon::Device::enumerate_devices()?;
  println!("Number of devices: {}", devices);

  let mut dev = pylon::Device::create_device_by_index(0)?;
  dev.open()?;

  let s = dev.get_string_feature("DeviceModelName")?;
  println!("Device: {}", s);

  dev.set_string_feature("AcquisitionMode", "Continuous")?;
  dev.set_float_feature("AcquisitionFrameRateAbs", 20.0)?;
  let mut stream = dev.get_stream_grabber(0)?;
  dev.execute_command("AcquisitionStart")?;

  for n in 0..100 {
    let img = stream.grab()?;
    let path = format!("/tmp/image_{:02}.png", n);
    println!("Save to: {}", path);
    if let pylon::Image::Grey8(img) = img {
      let img = image::GrayImage::from_vec(img.width() as u32, img.height() as u32, img.buf).unwrap();
      img.save(path)?;
    }
  }

  dev.execute_command("AcquisitionStop")?;

  Ok(())
}