summaryrefslogtreecommitdiff
path: root/src/gui.rs
blob: 42e062d75acfe862fc44ec8a968d858c1bc9ef10 (plain)
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
use gtk;
use gtk::prelude::*;
use std::cell::RefCell;
use portaudio as pa;
use std::rc::Rc;
use std::sync::Arc;
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="GtkComboBoxText" id="dropdown">
      </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));

    {
        let state_for_dropdown = state.clone();
        
        let dropdown: gtk::ComboBoxText = try!(
            gtk_builder.get_object("dropdown").ok_or("GUI does not contain an object with id 'dropdown'")
        );
        dropdown.connect_changed(move |dropdown: &gtk::ComboBoxText| {
            match state_for_dropdown.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_for_dropdown.borrow().pa, selected_mic, mic_sender.clone()).ok();
            if stream.is_none() {
                writeln!(io::stderr(), "Failed to open audio channel").ok();
            }
            state_for_dropdown.borrow_mut().pa_stream = stream;
        });
    }

    let async_thread = thread::spawn(move || {
        for samples in mic_receiver {
            let frequency_domain = ::transforms::fft(samples, 44100.0);
            
            let max_frequency = frequency_domain.iter()
                .fold(None as Option<::transforms::FrequencyBucket>, |max, next|
                      if max.is_none() || max.clone().unwrap().intensity < next.intensity { Some(next.clone()) } else { max }
                ).unwrap().max_freq;
            println!("{}Hz", max_frequency.floor());
        }
    });

    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: &gtk::ComboBoxText, microphones: Vec<(u32, String)>) {
    for (index, name) in microphones {
        dropdown.append(Some(format!("{}", index).as_ref()), name.as_ref());
    }
}