tutorial: Add a Bin subclass example

This is a simple bin containing a progressreport element. The messages
of the progressreport are caught and printed to stdout instead of
forwarding to the application.
This commit is contained in:
Sebastian Dröge 2019-04-24 12:00:14 +03:00
parent 8486fb0b41
commit b6cd1cf6a9
3 changed files with 184 additions and 0 deletions

View file

@ -4,6 +4,7 @@ version = "0.5.0"
authors = ["Sebastian Dröge <sebastian@centricular.com>"] authors = ["Sebastian Dröge <sebastian@centricular.com>"]
repository = "https://gitlab.freedesktop.org/gstreamer/gst-plugins-rs" repository = "https://gitlab.freedesktop.org/gstreamer/gst-plugins-rs"
license = "MIT/Apache-2.0" license = "MIT/Apache-2.0"
edition = "2018"
[dependencies] [dependencies]
glib = { git = "https://github.com/gtk-rs/glib", features = ["subclassing"] } glib = { git = "https://github.com/gtk-rs/glib", features = ["subclassing"] }

View file

@ -18,6 +18,7 @@ extern crate byte_slice_cast;
extern crate num_traits; extern crate num_traits;
mod identity; mod identity;
mod progressbin;
mod rgb2gray; mod rgb2gray;
mod sinesrc; mod sinesrc;
@ -27,6 +28,7 @@ fn plugin_init(plugin: &gst::Plugin) -> Result<(), glib::BoolError> {
rgb2gray::register(plugin)?; rgb2gray::register(plugin)?;
sinesrc::register(plugin)?; sinesrc::register(plugin)?;
identity::register(plugin)?; identity::register(plugin)?;
progressbin::register(plugin)?;
Ok(()) Ok(())
} }

View file

@ -0,0 +1,181 @@
// Copyright (C) 2019 Sebastian Dröge <sebastian@centricular.com>
//
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
// option. This file may not be copied, modified, or distributed
// except according to those terms.
use glib;
use glib::prelude::*;
use glib::subclass;
use glib::subclass::prelude::*;
use gst;
use gst::prelude::*;
use gst::subclass::prelude::*;
// Struct containing all the element data
struct ProgressBin {
cat: gst::DebugCategory,
progress: gst::Element,
srcpad: gst::GhostPad,
sinkpad: gst::GhostPad,
}
// This trait registers our type with the GObject object system and
// provides the entry points for creating a new instance and setting
// up the class data
impl ObjectSubclass for ProgressBin {
const NAME: &'static str = "RsProgressBin";
type ParentType = gst::Bin;
type Instance = gst::subclass::ElementInstanceStruct<Self>;
type Class = subclass::simple::ClassStruct<Self>;
// This macro provides some boilerplate.
glib_object_subclass!();
// Called when a new instance is to be created. We need to return an instance
// of our struct here and also get the class struct passed in case it's needed
fn new_with_class(klass: &subclass::simple::ClassStruct<Self>) -> Self {
// Create our two ghostpads from the templates that were registered with
// the class. We don't provide a target for them yet because we can only
// do so after the progressreport element was added to the bin.
//
// We do that and adding the pads inside glib::Object::constructed() later.
let templ = klass.get_pad_template("sink").unwrap();
let sinkpad = gst::GhostPad::new_no_target_from_template(Some("sink"), &templ).unwrap();
let templ = klass.get_pad_template("src").unwrap();
let srcpad = gst::GhostPad::new_no_target_from_template(Some("src"), &templ).unwrap();
// Create the progressreport element.
let progress = gst::ElementFactory::make("progressreport", Some("progress")).unwrap();
// Don't let progressreport print to stdout itself
progress.set_property("silent", &true).unwrap();
// Return an instance of our struct and also include our debug category here.
// The debug category will be used later whenever we need to put something
// into the debug logs
Self {
cat: gst::DebugCategory::new(
"rsprogressbin",
gst::DebugColorFlags::empty(),
"Progress printing Bin",
),
progress,
srcpad,
sinkpad,
}
}
// Called exactly once when registering the type. Used for
// setting up metadata for all instances, e.g. the name and
// classification and the pad templates with their caps.
//
// Actual instances can create pads based on those pad templates
// with a subset of the caps given here.
fn class_init(klass: &mut subclass::simple::ClassStruct<Self>) {
// Set the element specific metadata. This information is what
// is visible from gst-inspect-1.0 and can also be programatically
// retrieved from the gst::Registry after initial registration
// without having to load the plugin in memory.
klass.set_metadata(
"ProgressBin",
"Generic",
"Prints progress information to stdout",
"Sebastian Dröge <sebastian@centricular.com>",
);
// Create and add pad templates for our sink and source pad. These
// are later used for actually creating the pads and beforehand
// already provide information to GStreamer about all possible
// pads that could exist for this type.
// Our element can accept any possible caps on both pads
let caps = gst::Caps::new_any();
let src_pad_template = gst::PadTemplate::new(
"src",
gst::PadDirection::Src,
gst::PadPresence::Always,
&caps,
)
.unwrap();
klass.add_pad_template(src_pad_template);
let sink_pad_template = gst::PadTemplate::new(
"sink",
gst::PadDirection::Sink,
gst::PadPresence::Always,
&caps,
)
.unwrap();
klass.add_pad_template(sink_pad_template);
}
}
// Implementation of glib::Object virtual methods
impl ObjectImpl for ProgressBin {
// This macro provides some boilerplate
glib_object_impl!();
// Called right after construction of a new instance
fn constructed(&self, obj: &glib::Object) {
// Call the parent class' ::constructed() implementation first
self.parent_constructed(obj);
// Here we actually add the pads we created in ProgressBin::new() to the
// element so that GStreamer is aware of their existence.
let bin = obj.downcast_ref::<gst::Bin>().unwrap();
// Add the progressreport element to the bin.
bin.add(&self.progress).unwrap();
// Then set the ghost pad targets to the corresponding pads of the progressreport element.
self.sinkpad
.set_target(Some(&self.progress.get_static_pad("sink").unwrap()))
.unwrap();
self.srcpad
.set_target(Some(&self.progress.get_static_pad("src").unwrap()))
.unwrap();
// And finally add the two ghostpads to the bin.
bin.add_pad(&self.sinkpad).unwrap();
bin.add_pad(&self.srcpad).unwrap();
}
}
// Implementation of gst::Element virtual methods
impl ElementImpl for ProgressBin {}
// Implementation of gst::Bin virtual methods
impl BinImpl for ProgressBin {
fn handle_message(&self, bin: &gst::Bin, msg: gst::Message) {
use gst::MessageView;
match msg.view() {
// If this is the progressreport message, we print the status
// to stdout. Otherwise we pass through to the default message
// handling of the parent class, i.e. forwarding to the parent
// bins and the application.
MessageView::Element(ref msg)
if msg.get_src().as_ref() == Some(self.progress.upcast_ref())
&& msg
.get_structure()
.map(|s| s.get_name() == "progress")
.unwrap_or(false) =>
{
let s = msg.get_structure().unwrap();
let percent = s.get::<f64>("percent-double").unwrap();
println!("progress: {:5.1}%", percent);
}
_ => self.parent_handle_message(bin, msg),
}
}
}
// Registers the type for our element, and then registers in GStreamer under
// the name "rsprogressbin" for being able to instantiate it via e.g.
// gst::ElementFactory::make().
pub fn register(plugin: &gst::Plugin) -> Result<(), glib::BoolError> {
gst::Element::register(Some(plugin), "rsprogressbin", 0, ProgressBin::get_type())
}