start .iq reading
This commit is contained in:
@ -0,0 +1,56 @@
|
||||
use num_complex::Complex;
|
||||
use std::error::Error;
|
||||
use std::fs::File;
|
||||
use std::io::{BufRead, BufReader, Read};
|
||||
use std::os::unix::process::CommandExt;
|
||||
|
||||
pub type IqSample = Complex<f32>;
|
||||
|
||||
// Data chunk
|
||||
pub struct IqChunk {
|
||||
pub samples: Vec<IqSample>,
|
||||
}
|
||||
|
||||
pub struct FileSource {
|
||||
// Buffer
|
||||
pub reader: BufReader<File>,
|
||||
// Hackrf I & Q 8 bits
|
||||
pub raw_buffer: Vec<u8>,
|
||||
// Size of a sample chunk
|
||||
pub chunk_samples_size: usize,
|
||||
}
|
||||
|
||||
impl FileSource {
|
||||
pub fn new(file_path: &str, chunk_samples_size: usize) -> Self {
|
||||
let file = File::open(file_path).unwrap();
|
||||
|
||||
// Init buffer with size 16 Mo
|
||||
let reader = BufReader::with_capacity(16 * 1024 * 1024, file);
|
||||
|
||||
// 1 sample = 2 bytes (1 byte : I, 1 byte : Q)
|
||||
let raw_reader = vec![0; chunk_samples_size * 2];
|
||||
|
||||
Self {
|
||||
reader,
|
||||
raw_buffer: raw_reader,
|
||||
chunk_samples_size,
|
||||
}
|
||||
}
|
||||
|
||||
pub fn read_chunk(&mut self) -> Option<IqChunk> {
|
||||
// TODO : match for EOF
|
||||
self.reader.read_exact(&mut self.raw_buffer);
|
||||
|
||||
// Output samples
|
||||
let mut samples = Vec::with_capacity(self.chunk_samples_size);
|
||||
|
||||
// Buffer read
|
||||
for iq in self.raw_buffer.chunks(2) {
|
||||
let i = (iq[0] as f32) / 128.0;
|
||||
let q = (iq[1] as f32) / 128.0;
|
||||
samples.push(Complex::new(i, q));
|
||||
}
|
||||
|
||||
Some(IqChunk { samples })
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user