2017-11-11 14:53:03 +00:00
|
|
|
#[macro_use]
|
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-11-11 14:53:03 +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::i16;
|
|
|
|
use std::i32;
|
2017-11-11 14:53:03 +00:00
|
|
|
use std::error::Error as StdError;
|
2017-08-01 18:44:01 +00:00
|
|
|
|
2017-11-11 14:53:03 +00:00
|
|
|
extern crate failure;
|
2017-08-03 15:26:52 +00:00
|
|
|
|
2017-11-11 14:53:03 +00:00
|
|
|
#[macro_use]
|
|
|
|
extern crate failure_derive;
|
|
|
|
|
|
|
|
#[allow(unused_imports)]
|
|
|
|
use failure::{Error, Fail};
|
2017-08-03 15:26:52 +00:00
|
|
|
|
2017-11-11 14:53:03 +00:00
|
|
|
#[derive(Debug, Fail)]
|
|
|
|
#[fail(display = "Missing element {}", _0)]
|
|
|
|
struct MissingElement(&'static str);
|
|
|
|
|
|
|
|
#[derive(Debug, Fail)]
|
|
|
|
#[fail(display = "Received error from {}: {} (debug: {:?})", src, error, debug)]
|
|
|
|
struct ErrorMessage {
|
|
|
|
src: String,
|
|
|
|
error: String,
|
|
|
|
debug: Option<String>,
|
|
|
|
#[cause] cause: glib::Error,
|
|
|
|
}
|
2017-08-01 18:44:01 +00:00
|
|
|
|
2017-11-11 14:53:03 +00:00
|
|
|
fn create_pipeline() -> Result<gst::Pipeline, Error> {
|
|
|
|
gst::init()?;
|
|
|
|
|
|
|
|
let pipeline = gst::Pipeline::new(None);
|
|
|
|
let src =
|
|
|
|
gst::ElementFactory::make("audiotestsrc", None).ok_or(MissingElement("audiotestsrc"))?;
|
|
|
|
let sink = gst::ElementFactory::make("appsink", None).ok_or(MissingElement("appsink"))?;
|
|
|
|
|
|
|
|
pipeline.add_many(&[&src, &sink])?;
|
|
|
|
src.link(&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-11-11 14:53:03 +00:00
|
|
|
let buffer = if let Some(buffer) = sample.get_buffer() {
|
|
|
|
buffer
|
|
|
|
} else {
|
|
|
|
gst_element_error!(
|
|
|
|
appsink,
|
|
|
|
gst::ResourceError::Failed,
|
|
|
|
("Failed to get buffer from appsink")
|
|
|
|
);
|
2017-08-11 14:09:32 +00:00
|
|
|
|
2017-11-11 14:53:03 +00:00
|
|
|
return gst::FlowReturn::Error;
|
|
|
|
};
|
|
|
|
|
|
|
|
let map = if let Some(map) = buffer.map_readable() {
|
|
|
|
map
|
|
|
|
} else {
|
|
|
|
gst_element_error!(
|
|
|
|
appsink,
|
|
|
|
gst::ResourceError::Failed,
|
|
|
|
("Failed to map buffer readable")
|
|
|
|
);
|
|
|
|
|
|
|
|
return gst::FlowReturn::Error;
|
|
|
|
};
|
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-11-11 14:53:03 +00:00
|
|
|
gst_element_error!(
|
|
|
|
appsink,
|
|
|
|
gst::ResourceError::Failed,
|
|
|
|
("Failed to interprete buffer as S16 PCM")
|
|
|
|
);
|
|
|
|
|
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-09-10 12:10:04 +00:00
|
|
|
let f = f64::from(*sample) / f64::from(i16::MAX);
|
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-11-11 15:56:37 +00:00
|
|
|
fn main_loop(pipeline: gst::Pipeline) -> Result<(), Error> {
|
2017-11-11 14:53:03 +00:00
|
|
|
pipeline.set_state(gst::State::Playing).into_result()?;
|
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
|
|
|
|
2017-09-13 16:35:35 +00:00
|
|
|
while let Some(msg) = bus.timed_pop(gst::CLOCK_TIME_NONE) {
|
2017-08-17 14:58:15 +00:00
|
|
|
use gst::MessageView;
|
|
|
|
|
2017-08-01 18:44:01 +00:00
|
|
|
match msg.view() {
|
|
|
|
MessageView::Eos(..) => break,
|
|
|
|
MessageView::Error(err) => {
|
2017-11-11 14:53:03 +00:00
|
|
|
pipeline.set_state(gst::State::Null).into_result()?;
|
|
|
|
Err(ErrorMessage {
|
|
|
|
src: msg.get_src().get_path_string(),
|
|
|
|
error: err.get_error().description().into(),
|
|
|
|
debug: err.get_debug(),
|
|
|
|
cause: err.get_error(),
|
|
|
|
})?;
|
2017-08-01 18:44:01 +00:00
|
|
|
}
|
|
|
|
_ => (),
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2017-11-11 14:53:03 +00:00
|
|
|
pipeline.set_state(gst::State::Null).into_result()?;
|
2017-08-03 15:26:52 +00:00
|
|
|
|
|
|
|
Ok(())
|
|
|
|
}
|
|
|
|
|
|
|
|
fn main() {
|
2017-11-11 15:56:37 +00:00
|
|
|
match create_pipeline().and_then(main_loop) {
|
2017-08-03 15:26:52 +00:00
|
|
|
Ok(r) => r,
|
|
|
|
Err(e) => eprintln!("Error! {}", e),
|
|
|
|
}
|
2017-08-01 18:44:01 +00:00
|
|
|
}
|