use aaronia_rtsa::version;
use aaronia_rtsa::ApiHandle;
use aaronia_rtsa::Device;
use num_complex::Complex32;

fn main() -> Result<(), Box<dyn std::error::Error>> {
    println!("RTSA library version: {}", version());

    let mut api = ApiHandle::new()?;
    api.rescan_devices()?;
    let d = api.devices()?;
    println!("devices {d:?}");

    let mut dev = api.get_device()?;
    dev.open()?;
    dev.set("device/receiverchannel", "Rx1")?;
    dev.set("device/outputformat", "iq")?;
    dev.set("device/receiverclock", "92MHz")?;
    dev.set("main/decimation", "1 / 64")?;
    dev.set("main/centerfreq", "810e6")?;
    dev.set("main/reflevel", "-20")?;
    dev.connect()?;
    dev.start()?;

    let mut s = rx(&mut dev)?;

    dev.stop()?;
    dev.disconnect()?;
    dev.close()?;

    plot(&mut s);

    Ok(())
}

const N: usize = 8192 * 2;
fn rx(dev: &mut Device) -> Result<[Complex32; N], aaronia_rtsa::Error> {
    let mut samples = [Complex32::new(0.0, 0.0); N];
    let mut i = 0;

    while i < N {
        let p = dev.packet(0)?;
        let cur = p.samples();
        let n = std::cmp::min(N - i, cur.len());
        samples[i..i + n].copy_from_slice(&cur[0..n]);
        i += n;
        dev.consume(0)?;
    }

    Ok(samples)
}

fn plot(s: &mut [num_complex::Complex32]) {
    use gnuplot::*;

    let mut planner = rustfft::FftPlanner::new();
    planner.plan_fft_forward(s.len()).process(s);

    let abs = s.iter().map(|s| s.norm_sqr().log10());

    let mut fg = Figure::new();
    fg.axes2d().set_title("Spectrum", &[]).lines(
        0..s.len(),
        abs,
        &[LineWidth(3.0), Color("blue"), LineStyle(DotDash)],
    );
    fg.show().unwrap();
}