add --debug to write a file that can be played with cat /tmp/page.rawf32audio | sox -r 16000 -b 32 -t f32 -e floating-point - -d
parent
6b54e500cd
commit
226bedb80e
27
src/main.rs
27
src/main.rs
|
|
@ -5,6 +5,8 @@ use std::time::{Duration, Instant};
|
||||||
use chrono;
|
use chrono;
|
||||||
use clap::Parser;
|
use clap::Parser;
|
||||||
use std::thread;
|
use std::thread;
|
||||||
|
use std::fs::File;
|
||||||
|
use std::io::Write;
|
||||||
|
|
||||||
#[derive(Parser, Debug)]
|
#[derive(Parser, Debug)]
|
||||||
struct Flags {
|
struct Flags {
|
||||||
|
|
@ -24,6 +26,9 @@ struct Flags {
|
||||||
stream_tail: f32,
|
stream_tail: f32,
|
||||||
|
|
||||||
wav: Option<String>,
|
wav: Option<String>,
|
||||||
|
|
||||||
|
#[arg(long, default_value = "false")]
|
||||||
|
debug: bool,
|
||||||
}
|
}
|
||||||
|
|
||||||
fn main() {
|
fn main() {
|
||||||
|
|
@ -49,6 +54,10 @@ fn main() {
|
||||||
w.transcribe(&audio_data);
|
w.transcribe(&audio_data);
|
||||||
},
|
},
|
||||||
None => {
|
None => {
|
||||||
|
match &flags.debug {
|
||||||
|
true => { File::create("/tmp/page.rawf32audio").unwrap(); },
|
||||||
|
false => {},
|
||||||
|
};
|
||||||
let mut buffer = vec![];
|
let mut buffer = vec![];
|
||||||
let mut last = Instant::now();
|
let mut last = Instant::now();
|
||||||
new_listener().listen(move |data: Vec<f32>| {
|
new_listener().listen(move |data: Vec<f32>| {
|
||||||
|
|
@ -59,8 +68,22 @@ fn main() {
|
||||||
Err(msg) => eprintln!("{}", msg),
|
Err(msg) => eprintln!("{}", msg),
|
||||||
};
|
};
|
||||||
|
|
||||||
for i in stream_retain..buffer.len() {
|
match &flags.debug {
|
||||||
buffer[i - stream_retain] = buffer[i]
|
true => {
|
||||||
|
let mut f = File::options().append(true).open("/tmp/page.rawf32audio").unwrap();
|
||||||
|
let mut wav_data = vec![];
|
||||||
|
for i in buffer.iter() {
|
||||||
|
for j in i.to_le_bytes() {
|
||||||
|
wav_data.push(j);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
f.write_all(wav_data.as_slice()).unwrap();
|
||||||
|
},
|
||||||
|
false => {},
|
||||||
|
};
|
||||||
|
|
||||||
|
for i in 0..stream_retain {
|
||||||
|
buffer[i] = buffer[buffer.len() - stream_retain + i];
|
||||||
}
|
}
|
||||||
buffer.truncate(stream_retain);
|
buffer.truncate(stream_retain);
|
||||||
last = Instant::now();
|
last = Instant::now();
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue