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
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
|
use gtk;
use gtk::prelude::*;
use std::cell::RefCell;
use portaudio as pa;
use std::rc::Rc;
use std::io;
use std::io::Write;
use std::thread;
use std::sync::mpsc::*;
const GUI_XML: &'static str = r#"
<interface>
<object class="GtkWindow" id="window">
<property name="title">Rusty Microphone</property>
<child>
<object class="GtkVBox">
<child>
<object class="GtkComboBoxText" id="dropdown">
</object>
</child>
<child>
<object class="GtkLabel" id="pitch-label">
<property name="label">Hello world</property>
</object>
</child>
</object>
</child>
</object>
</interface>
"#;
struct ApplicationState {
pa: pa::PortAudio,
pa_stream: Option<pa::Stream<pa::NonBlocking, pa::Input<f32>>>
}
pub fn start_gui() -> Result<(), String> {
let pa = try!(::audio::init().map_err(|e| e.to_string()));
let microphones = try!(::audio::get_device_list(&pa).map_err(|e| e.to_string()));
let (mic_sender, mic_receiver) = channel();
let state = Rc::new(RefCell::new(ApplicationState {
pa: pa,
pa_stream: None
}));
try!(gtk::init().map_err(|_| "Failed to initialize GTK."));
let gtk_builder = try!(create_window(microphones));
connect_dropdown_choose_microphone(>k_builder, mic_sender, state.clone());
let (pitch_sender, pitch_receiver) = channel();
start_processing_audio(mic_receiver, pitch_sender);
let pitch_label: gtk::Label = gtk_builder.get_object("pitch-label").expect("GUI does not contain an object with id 'pitch-label'");
gtk::timeout_add(100, move || {
let mut pitch = None;
loop {
let next = pitch_receiver.try_recv().ok();
if next.is_none() {
break;
}
pitch = next;
}
match pitch {
Some(pitch) => {pitch_label.set_label(pitch.as_ref());},
None => {}
};
gtk::Continue(true)
});
gtk::main();
Ok(())
}
fn create_window(microphones: Vec<(u32, String)>) -> Result<gtk::Builder, String> {
let gtk_builder = gtk::Builder::new_from_string(GUI_XML);
let window: gtk::Window = try!(
gtk_builder.get_object("window")
.ok_or("GUI does not contain an object with id 'window'")
);
window.set_default_size(300, 300);
window.connect_delete_event(|_, _| {
gtk::main_quit();
Inhibit(false)
});
window.show_all();
let dropdown: gtk::ComboBoxText = try!(
gtk_builder.get_object("dropdown")
.ok_or("GUI does not contain an object with id 'dropdown'")
);
set_dropdown_items(&dropdown, microphones);
Ok(gtk_builder)
}
fn set_dropdown_items(dropdown: >k::ComboBoxText, microphones: Vec<(u32, String)>) {
for (index, name) in microphones {
dropdown.append(Some(format!("{}", index).as_ref()), name.as_ref());
}
}
fn connect_dropdown_choose_microphone(gtk_builder: >k::Builder, mic_sender: Sender<Vec<f64>>, state: Rc<RefCell<ApplicationState>>) {
let dropdown: gtk::ComboBoxText = gtk_builder.get_object("dropdown").expect("GUI does not contain an object with id 'dropdown'");
dropdown.connect_changed(move |dropdown: >k::ComboBoxText| {
match state.borrow_mut().pa_stream {
Some(ref mut stream) => {stream.stop().ok();},
_ => {}
}
let selected_mic = dropdown.get_active_id().and_then(|id| id.parse().ok()).expect("Dropdown did not change to a valid value");
let stream = ::audio::start_listening(&state.borrow().pa, selected_mic, mic_sender.clone()).ok();
if stream.is_none() {
writeln!(io::stderr(), "Failed to open audio channel").ok();
}
state.borrow_mut().pa_stream = stream;
});
}
fn start_processing_audio(mic_receiver: Receiver<Vec<f64>>, pitch_sender: Sender<String>) {
thread::spawn(move || {
for samples in mic_receiver {
let frequency_domain = ::transforms::fft(samples, 44100.0);
let fundamental = ::transforms::find_fundamental_frequency(&frequency_domain);
let pitch = match fundamental {
Some(fundamental) => ::transforms::hz_to_pitch(fundamental),
None => "".to_string()
};
pitch_sender.send(pitch).ok();
}
});
}
|
