// use crate::ASICamera2::BayerPattern;
//
#![allow(non_camel_case_types)]
#![allow(non_snake_case)]
#![allow(dead_code)]
#![feature(alloc_layout_extra)]
mod asicam;
mod qhyccd;

use crate::asicam::ASICamera2::{ControlType, ImageType};
use crate::asicam::Camera;

fn main() {
    operate_qhy();
}

fn operate_qhy() {
    use crate::qhyccd::Control;
    println!("Operating on qhy camera ... or i'll die trying");
    let mut camera = qhyccd::acquire(0).unwrap();
    camera.set_exposure_ms(4500).unwrap();
    camera.set_param(Control::Gain, 100.0).unwrap();
    camera.set_param(Control::Offset, 000.0).unwrap();
    camera.set_param(Control::USBTraffic, 50.0).unwrap();
//    camera.set_target_temp(0.0).unwrap();
    camera.set_param(Control::Cooler, -15.0).unwrap();
    println!("Gain: {:?}", camera.get_param_limits(Control::Gain));
    println!("Offset: {:?}", camera.get_param_limits(Control::Offset));
    println!("USBTraffic: {:?}", camera.get_param_limits(Control::USBTraffic));
    println!("Gama: {:?}", camera.get_param_limits(Control::Gamma));
    println!("WB Red: {:?}", camera.get_param_limits(Control::CONTROL_WBR));
    println!("WB Green: {:?}", camera.get_param_limits(Control::CONTROL_WBG));
    println!("WB Blue: {:?}", camera.get_param_limits(Control::CONTROL_WBB));
    println!("Binning modes:");
    println!("1x1: {}", camera.has_param(Control::Bin1x1Mode));
    println!("2x2: {}", camera.has_param(Control::Bin2x2Mode));
    println!("3x3: {}", camera.has_param(Control::Bin3x3Mode));
    println!("4x4: {}", camera.has_param(Control::Bin4x4Mode));
//    camera.set_param(Control::Speed, 1.0).unwrap();
    println!("current temp: {}", camera.get_param(Control::CurTemp));
    camera.set_defaults().unwrap();
    camera.set_param(Control::CONTROL_WBR, 4027.0).unwrap();
    camera.set_param(Control::CONTROL_WBG, 4027.0).unwrap();
    camera.set_param(Control::CONTROL_WBB, 4027.0).unwrap();
//    camera.set_bin_mode(2).unwrap();
    let object = "jupiter";
    for i in 0..10 {
        camera.take_image(&format!("{}_{}.png", object, i));//, i));
    }
    camera.release().unwrap();
}

fn operate_asi() {
    println!("Operating on asi camera ... or i'll die trying");
    let mut camera = asicam::acquire(0).unwrap();

    println!("{:?}", camera);
    camera.set_control_value(ControlType::TargetTemp, -100).unwrap();
    camera.set_control_value(ControlType::CoolerOn, 1).unwrap();
    std::thread::sleep(std::time::Duration::from_millis(500));
    println!("Camera temperature is currently {:?}", camera.get_control_value(ControlType::Temperature).unwrap());

    /*
    for exposure in [2000, 5000, 10000, 30000].iter() {
        camera.set_control_value(ControlType::Exposure, *exposure).unwrap();
        for gain in [450, 375, 325, 250, 200].iter() {
            camera.set_control_value(ControlType::Gain, *gain).unwrap();
            for offset in [100, 80, 60, 40, 20, 0].iter() {
                camera.set_control_value(ControlType::Offset, *offset).unwrap();
                take_calibration_images(&camera, 1, &format!("roof_gain_{:03}_offset_{:03}_exposure_{:06}", gain, offset, exposure));
            }
        }
    }
    */
    camera.set_exposure_ms(45000).unwrap();
//    camera.set_control_value(ControlType::Exposure, 70000000).unwrap();
    camera.set_control_value(ControlType::Gain, 350).unwrap();
    camera.set_control_value(ControlType::Offset, 0).unwrap();
    camera.set_control_value(ControlType::HardwareBin, 0).unwrap();
    camera.set_roi_format(camera.width, camera.height, 1, ImageType::RGB24).unwrap();
    take_calibration_images(&camera, 40, "dark_gain_350_exposure_45000");
    /*
    for exposure in [1000 * 1000 * 10].iter() {
        camera.set_control_value(ControlType::Exposure, *exposure).unwrap();
        for gain in [450, 375, 325, 250, 200].iter() {
            camera.set_control_value(ControlType::Gain, *gain).unwrap();
            for offset in [100, 80, 70, 60, 40, 0].iter() {
                camera.set_control_value(ControlType::Offset, *offset).unwrap();
                take_calibration_images(
                    &camera,
                    30,
                    &format!("images/gain_{:03}_offset_{:03}_exposure_{:06}", gain, offset, exposure));
            }
        }
    }
    */

    println!("Done!");
}

fn take_calibration_images(camera: &Camera, count: u32, path_fragment: &str) {
    for i in 0..count {
        println!("{} image {:06}", path_fragment,  i);
        let temp = camera.get_control_value(ControlType::Temperature).unwrap();
        println!("Camera temperature is currently {:?}", temp);
        camera.take_image(&format!("{}_{:06}_temp_{:03}.png", path_fragment, i, temp)).unwrap();
    }
}