2017-08-01 18:44:01 +00:00
|
|
|
extern crate gstreamer as gst;
|
2017-08-17 14:58:15 +00:00
|
|
|
use gst::prelude::*;
|
2017-08-01 18:44:01 +00:00
|
|
|
extern crate gstreamer_app as gst_app;
|
2017-08-11 14:09:32 +00:00
|
|
|
extern crate gstreamer_audio as gst_audio;
|
2017-08-01 18:44:01 +00:00
|
|
|
|
2017-08-03 15:26:52 +00:00
|
|
|
extern crate glib;
|
|
|
|
|
2017-08-14 17:45:35 +00:00
|
|
|
extern crate byte_slice_cast;
|
|
|
|
use byte_slice_cast::*;
|
|
|
|
|
2017-08-01 18:44:01 +00:00
|
|
|
use std::u64;
|
|
|
|
use std::i16;
|
|
|
|
use std::i32;
|
|
|
|
|
2017-08-04 09:54:11 +00:00
|
|
|
pub mod utils;
|
2017-08-03 15:26:52 +00:00
|
|
|
|
2017-08-17 14:58:15 +00:00
|
|
|
fn create_pipeline() -> Result<gst::Pipeline, utils::ExampleError> {
|
2017-08-04 12:36:12 +00:00
|
|
|
gst::init().map_err(utils::ExampleError::InitFailed)?;
|
2017-08-01 18:44:01 +00:00
|
|
|
let pipeline = gst::Pipeline::new(None);
|
2017-08-04 09:54:11 +00:00
|
|
|
let src = utils::create_element("audiotestsrc")?;
|
|
|
|
let sink = utils::create_element("appsink")?;
|
2017-08-03 15:26:52 +00:00
|
|
|
|
|
|
|
pipeline
|
|
|
|
.add_many(&[&src, &sink])
|
|
|
|
.expect("Unable to add elements in the pipeline");
|
2017-08-01 18:44:01 +00:00
|
|
|
|
2017-08-04 09:54:11 +00:00
|
|
|
utils::link_elements(&src, &sink)?;
|
2017-08-03 15:26:52 +00:00
|
|
|
|
|
|
|
let appsink = sink.clone()
|
2017-08-17 14:58:15 +00:00
|
|
|
.dynamic_cast::<gst_app::AppSink>()
|
2017-08-03 15:26:52 +00:00
|
|
|
.expect("Sink element is expected to be an appsink!");
|
2017-08-01 18:44:01 +00:00
|
|
|
|
2017-08-17 14:58:15 +00:00
|
|
|
appsink.set_caps(&gst::Caps::new_simple(
|
2017-08-01 18:44:01 +00:00
|
|
|
"audio/x-raw",
|
|
|
|
&[
|
2017-08-11 14:09:32 +00:00
|
|
|
("format", &gst_audio::AUDIO_FORMAT_S16.to_string()),
|
2017-08-02 17:15:16 +00:00
|
|
|
("layout", &"interleaved"),
|
|
|
|
("channels", &(1i32)),
|
2017-08-17 14:58:15 +00:00
|
|
|
("rate", &gst::IntRange::<i32>::new(1, i32::MAX)),
|
2017-08-01 18:44:01 +00:00
|
|
|
],
|
|
|
|
));
|
|
|
|
|
2017-08-17 14:58:15 +00:00
|
|
|
appsink.set_callbacks(gst_app::AppSinkCallbacks::new(
|
2017-08-01 18:44:01 +00:00
|
|
|
/* eos */
|
|
|
|
|_| {},
|
|
|
|
/* new_preroll */
|
2017-08-17 14:58:15 +00:00
|
|
|
|_| gst::FlowReturn::Ok,
|
2017-08-01 18:44:01 +00:00
|
|
|
/* new_samples */
|
|
|
|
|appsink| {
|
|
|
|
let sample = match appsink.pull_sample() {
|
2017-08-17 14:58:15 +00:00
|
|
|
None => return gst::FlowReturn::Eos,
|
2017-08-01 18:44:01 +00:00
|
|
|
Some(sample) => sample,
|
|
|
|
};
|
|
|
|
|
2017-08-03 15:26:52 +00:00
|
|
|
let buffer = sample
|
|
|
|
.get_buffer()
|
|
|
|
.expect("Unable to extract buffer from the sample");
|
2017-08-11 14:09:32 +00:00
|
|
|
|
2017-08-11 12:20:43 +00:00
|
|
|
let map = buffer
|
|
|
|
.map_readable()
|
|
|
|
.expect("Unable to map buffer for reading");
|
2017-08-11 14:09:32 +00:00
|
|
|
|
2017-08-14 17:45:35 +00:00
|
|
|
let samples = if let Ok(samples) = map.as_slice().as_slice_of::<i16>() {
|
2017-08-11 14:09:32 +00:00
|
|
|
samples
|
|
|
|
} else {
|
2017-08-17 14:58:15 +00:00
|
|
|
return gst::FlowReturn::Error;
|
2017-08-11 14:09:32 +00:00
|
|
|
};
|
|
|
|
|
|
|
|
let sum: f64 = samples
|
|
|
|
.iter()
|
2017-08-01 18:44:01 +00:00
|
|
|
.map(|sample| {
|
2017-08-11 14:09:32 +00:00
|
|
|
let f = (*sample as f64) / (i16::MAX as f64);
|
2017-08-01 18:44:01 +00:00
|
|
|
f * f
|
|
|
|
})
|
|
|
|
.sum();
|
2017-08-11 14:09:32 +00:00
|
|
|
let rms = (sum / (samples.len() as f64)).sqrt();
|
2017-08-01 18:44:01 +00:00
|
|
|
println!("rms: {}", rms);
|
|
|
|
|
2017-08-17 14:58:15 +00:00
|
|
|
gst::FlowReturn::Ok
|
2017-08-01 18:44:01 +00:00
|
|
|
},
|
|
|
|
));
|
|
|
|
|
2017-08-03 15:26:52 +00:00
|
|
|
Ok(pipeline)
|
|
|
|
}
|
|
|
|
|
2017-08-04 09:54:11 +00:00
|
|
|
fn main_loop() -> Result<(), utils::ExampleError> {
|
2017-08-03 15:26:52 +00:00
|
|
|
let pipeline = create_pipeline()?;
|
2017-08-01 18:44:01 +00:00
|
|
|
|
2017-08-04 09:54:11 +00:00
|
|
|
utils::set_state(&pipeline, gst::State::Playing)?;
|
2017-08-03 15:26:52 +00:00
|
|
|
|
|
|
|
let bus = pipeline
|
|
|
|
.get_bus()
|
|
|
|
.expect("Pipeline without bus. Shouldn't happen!");
|
2017-08-01 18:44:01 +00:00
|
|
|
|
|
|
|
loop {
|
2017-08-17 14:58:15 +00:00
|
|
|
use gst::MessageView;
|
|
|
|
|
2017-08-01 18:44:01 +00:00
|
|
|
let msg = match bus.timed_pop(u64::MAX) {
|
|
|
|
None => break,
|
|
|
|
Some(msg) => msg,
|
|
|
|
};
|
|
|
|
|
|
|
|
match msg.view() {
|
|
|
|
MessageView::Eos(..) => break,
|
|
|
|
MessageView::Error(err) => {
|
2017-08-04 09:54:11 +00:00
|
|
|
utils::set_state(&pipeline, gst::State::Null)?;
|
|
|
|
return Err(utils::ExampleError::ElementError(
|
2017-08-01 18:44:01 +00:00
|
|
|
msg.get_src().get_path_string(),
|
|
|
|
err.get_error(),
|
2017-08-03 15:26:52 +00:00
|
|
|
err.get_debug().unwrap(),
|
|
|
|
));
|
2017-08-01 18:44:01 +00:00
|
|
|
}
|
|
|
|
_ => (),
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2017-08-04 09:54:11 +00:00
|
|
|
utils::set_state(&pipeline, gst::State::Null)?;
|
2017-08-03 15:26:52 +00:00
|
|
|
|
|
|
|
Ok(())
|
|
|
|
}
|
|
|
|
|
|
|
|
fn main() {
|
|
|
|
match main_loop() {
|
|
|
|
Ok(r) => r,
|
|
|
|
Err(e) => eprintln!("Error! {}", e),
|
|
|
|
}
|
2017-08-01 18:44:01 +00:00
|
|
|
}
|