use xilinx_dma::AxiDma;
use xilinx_dma::DmaBuffer;
use xilinx_dma::Error;

fn main() -> Result<(), Error> {
    let dma_buffer_h2d = DmaBuffer::new("udmabuf0")?;
    let dma_buffer_d2h = DmaBuffer::new("udmabuf1")?;
    println!("{:?}", dma_buffer_h2d);
    println!("{:?}", dma_buffer_d2h);

    // do not use the whole buffer
    let max_items = 128;
    let items = std::cmp::min(max_items, dma_buffer_h2d.size() / 4);
    let items = std::cmp::min(items, dma_buffer_d2h.size() / 4);

    let slice_h2d = &mut dma_buffer_h2d.slice::<u32>()[0..items];
    let slice_d2h = &mut dma_buffer_d2h.slice::<u32>()[0..items];

    for i in slice_d2h.iter_mut() {
        *i = 0;
    }

    for i in slice_h2d.iter_mut() {
        *i = fastrand::u32(0..1024);
    }

    let mut dma_h2d = AxiDma::new("uio4")?;
    let mut dma_d2h = AxiDma::new("uio5")?;
    println!("{:?}", dma_h2d);
    println!("{:?}", dma_d2h);

    dma_h2d.start_h2d(&dma_buffer_h2d, items * 4)?;
    dma_d2h.start_d2h(&dma_buffer_d2h, items * 4)?;
    println!("transfers started");

    dma_h2d.wait_h2d()?;
    println!("h2d done");
    dma_d2h.wait_d2h()?;
    println!("d2h done");

    dma_h2d.status_h2d();
    dma_d2h.status_d2h();

    for i in 0..items {
        assert_eq!(slice_d2h[i], slice_h2d[i] + 123);
    }

    Ok(())
}