hackrf-nusb 0.3.0

//! Asynchronous HackRF receive example.
//!
//! The default invocation prints usage and exits without touching USB. Pass
//! `--run` only when a HackRF is connected and USB permissions are configured.

use std::env;

#[cfg(any(feature = "smol", feature = "tokio"))]
use futures_lite::future::block_on;
#[cfg(any(feature = "smol", feature = "tokio"))]
use hackrf_nusb::{Complex32, Device};

#[cfg(any(feature = "smol", feature = "tokio"))]
const EXAMPLE_FREQ_HZ: u64 = 100_000_000;
#[cfg(any(feature = "smol", feature = "tokio"))]
const EXAMPLE_SAMPLE_RATE_HZ: u32 = 10_000_000;

fn main() -> hackrf_nusb::Result<()> {
    let args = env::args().skip(1).collect::<Vec<_>>();
    if !args.iter().any(|arg| arg == "--run") {
        print_usage();
        return Ok(());
    }

    #[cfg(not(any(feature = "smol", feature = "tokio")))]
    {
        eprintln!(
            "The async example needs nusb runtime integration.\n\
             Run with one feature enabled, for example:\n\
               cargo run --features smol --example rx_async -- --run --rx"
        );
        Ok(())
    }

    #[cfg(any(feature = "smol", feature = "tokio"))]
    block_on(run(args))
}

#[cfg(any(feature = "smol", feature = "tokio"))]
async fn run(args: Vec<String>) -> hackrf_nusb::Result<()> {
    let run_rx = args.iter().any(|arg| arg == "--rx");
    let mut device = Device::builder()
        .frequency_hz(EXAMPLE_FREQ_HZ)
        .sample_rate_hz(EXAMPLE_SAMPLE_RATE_HZ)
        .lna_gain_db(16)
        .vga_gain_db(20)
        .bias_tee(false)
        .open()
        .await?;

    println!(
        "opened {} firmware={} serial={}",
        device.info().board_name(),
        device.info().firmware_version,
        device
            .info()
            .serial
            .map_or_else(|| "unknown".to_owned(), |serial| format!("{serial:032x}")),
    );

    if run_rx {
        let mut rx = device.rx_stream()?;
        let mut samples = [Complex32::default(); 4096];
        rx.start().await?;
        let count = rx.read(&mut samples, None).await?;
        let stats = rx.stop().await?;
        drop(rx);
        println!(
            "read {count} samples; first={:?}; {stats:?}",
            samples.first()
        );
    } else {
        println!("RX not started; add --rx for a short receive smoke test.");
    }

    device.shutdown().await?;
    Ok(())
}

fn print_usage() {
    eprintln!(
        "This example opens and configures real HackRF hardware asynchronously.\n\
         It is gated to avoid accidental USB access during checks/tests.\n\n\
         Run:\n\
           cargo run --features smol --example rx_async -- --run\n\
           cargo run --features smol --example rx_async -- --run --rx"
    );
}