dsp 0.11.0

Digital Signal Processing
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63

//! Nodes for saving and reading data to/from file
//! 
use std::fs::File;
use byteorder::{ReadBytesExt, WriteBytesExt}; 
use byteorder::LittleEndian;


/// Save binary data into a file
/// 
/// Example
/// 
/// ```
/// use dsp::core::file::FileSink;
/// 
/// let mut node = FileSink::new("target/file.dat");
/// let input_buffer = vec![1.;100];
/// node.read_buffer(&input_buffer);
/// ```
pub struct FileSink {
    file: Option<File>,
}

impl FileSink {
    pub fn new(file_name: &str) -> FileSink {
        if let Ok(file) = File::create(file_name) {
            FileSink {file: Some(file)}
        } else {
            FileSink{file: None}
        }
    }

    pub fn read_buffer(&mut self, input_buffer: &[f32]) {
        let mut file = self.file.as_ref().unwrap();
        for v in input_buffer {
            file.write_f32::<LittleEndian>(*v).expect("Can't reead from file");
        }
        let _ = file.sync_all();
    }
}


/// Load binary data from a file
/// 
pub struct FileSource {
    file: File,
}

impl FileSource {
    pub fn new(file_name: &str) -> FileSource {
        let file = File::open(file_name).expect("Can't open file");
        FileSource {file}
    }

    pub fn write_buffer(&mut self, output_buffer: &mut [f32]) {
        for i in 0..output_buffer.len() {
            if let Ok(v) = self.file.read_f32::<LittleEndian>() {
                output_buffer[i] = v;
            } else {
                break
            }
        }
    }
}