mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer-rs.git
synced 2024-06-02 13:29:29 +00:00
422 lines
13 KiB
Rust
422 lines
13 KiB
Rust
// Take a look at the license at the top of the repository in the LICENSE file.
|
|
|
|
use crate::utils::HasStreamLock;
|
|
use crate::video_codec_state::{InNegotiation, Readable, VideoCodecState, VideoCodecStateContext};
|
|
use crate::VideoCodecFrame;
|
|
use crate::VideoDecoder;
|
|
use crate::VideoFormat;
|
|
#[cfg(any(feature = "v1_16", feature = "dox"))]
|
|
#[cfg_attr(feature = "dox", doc(cfg(feature = "v1_16")))]
|
|
use crate::VideoInterlaceMode;
|
|
use glib::prelude::*;
|
|
use glib::translate::*;
|
|
use std::mem;
|
|
use std::ptr;
|
|
|
|
extern "C" {
|
|
fn _gst_video_decoder_error(
|
|
dec: *mut ffi::GstVideoDecoder,
|
|
weight: i32,
|
|
domain: glib::ffi::GQuark,
|
|
code: i32,
|
|
txt: *mut libc::c_char,
|
|
debug: *mut libc::c_char,
|
|
file: *const libc::c_char,
|
|
function: *const libc::c_char,
|
|
line: i32,
|
|
) -> gst::ffi::GstFlowReturn;
|
|
}
|
|
|
|
pub trait VideoDecoderExtManual: 'static {
|
|
#[cfg(any(feature = "v1_12", feature = "dox"))]
|
|
#[cfg_attr(feature = "dox", doc(cfg(feature = "v1_12")))]
|
|
fn allocate_output_frame(
|
|
&self,
|
|
frame: &mut VideoCodecFrame,
|
|
params: Option<&gst::BufferPoolAcquireParams>,
|
|
) -> Result<gst::FlowSuccess, gst::FlowError>;
|
|
|
|
#[doc(alias = "get_frame")]
|
|
fn frame(&self, frame_number: i32) -> Option<VideoCodecFrame>;
|
|
#[doc(alias = "get_frames")]
|
|
fn frames(&self) -> Vec<VideoCodecFrame>;
|
|
#[doc(alias = "get_oldest_frame")]
|
|
fn oldest_frame(&self) -> Option<VideoCodecFrame>;
|
|
|
|
#[doc(alias = "get_allocator")]
|
|
fn allocator(&self) -> (Option<gst::Allocator>, gst::AllocationParams);
|
|
|
|
fn finish_frame(&self, frame: VideoCodecFrame) -> Result<gst::FlowSuccess, gst::FlowError>;
|
|
fn release_frame(&self, frame: VideoCodecFrame);
|
|
fn drop_frame(&self, frame: VideoCodecFrame) -> Result<gst::FlowSuccess, gst::FlowError>;
|
|
|
|
#[doc(alias = "get_latency")]
|
|
fn latency(&self) -> (gst::ClockTime, gst::ClockTime);
|
|
fn set_latency(&self, min_latency: gst::ClockTime, max_latency: gst::ClockTime);
|
|
|
|
#[doc(alias = "get_output_state")]
|
|
fn output_state(&self) -> Option<VideoCodecState<'static, Readable>>;
|
|
fn set_output_state(
|
|
&self,
|
|
fmt: VideoFormat,
|
|
width: u32,
|
|
height: u32,
|
|
reference: Option<&VideoCodecState<Readable>>,
|
|
) -> Result<VideoCodecState<InNegotiation>, gst::FlowError>;
|
|
#[cfg(any(feature = "v1_16", feature = "dox"))]
|
|
#[cfg_attr(feature = "dox", doc(cfg(feature = "v1_16")))]
|
|
fn set_interlaced_output_state(
|
|
&self,
|
|
fmt: VideoFormat,
|
|
mode: VideoInterlaceMode,
|
|
width: u32,
|
|
height: u32,
|
|
reference: Option<&VideoCodecState<Readable>>,
|
|
) -> Result<VideoCodecState<InNegotiation>, gst::FlowError>;
|
|
|
|
fn negotiate<'a>(
|
|
&'a self,
|
|
output_state: VideoCodecState<'a, InNegotiation<'a>>,
|
|
) -> Result<(), gst::FlowError>;
|
|
|
|
#[allow(clippy::too_many_arguments)]
|
|
fn error<T: gst::MessageErrorDomain>(
|
|
&self,
|
|
weight: i32,
|
|
code: T,
|
|
message: Option<&str>,
|
|
debug: Option<&str>,
|
|
file: &str,
|
|
function: &str,
|
|
line: u32,
|
|
) -> Result<gst::FlowSuccess, gst::FlowError>;
|
|
}
|
|
|
|
impl<O: IsA<VideoDecoder>> VideoDecoderExtManual for O {
|
|
#[cfg(any(feature = "v1_12", feature = "dox"))]
|
|
#[cfg_attr(feature = "dox", doc(cfg(feature = "v1_12")))]
|
|
fn allocate_output_frame(
|
|
&self,
|
|
frame: &mut VideoCodecFrame,
|
|
params: Option<&gst::BufferPoolAcquireParams>,
|
|
) -> Result<gst::FlowSuccess, gst::FlowError> {
|
|
unsafe {
|
|
let params_ptr = params.to_glib_none().0 as *mut _;
|
|
gst::FlowSuccess::try_from_glib(
|
|
ffi::gst_video_decoder_allocate_output_frame_with_params(
|
|
self.as_ref().to_glib_none().0,
|
|
frame.to_glib_none().0,
|
|
params_ptr,
|
|
),
|
|
)
|
|
}
|
|
}
|
|
|
|
fn allocator(&self) -> (Option<gst::Allocator>, gst::AllocationParams) {
|
|
unsafe {
|
|
let mut allocator = ptr::null_mut();
|
|
let mut params = mem::zeroed();
|
|
ffi::gst_video_decoder_get_allocator(
|
|
self.as_ref().to_glib_none().0,
|
|
&mut allocator,
|
|
&mut params,
|
|
);
|
|
(from_glib_full(allocator), params.into())
|
|
}
|
|
}
|
|
|
|
fn finish_frame(&self, frame: VideoCodecFrame) -> Result<gst::FlowSuccess, gst::FlowError> {
|
|
unsafe {
|
|
gst::FlowSuccess::try_from_glib(ffi::gst_video_decoder_finish_frame(
|
|
self.as_ref().to_glib_none().0,
|
|
frame.into_ptr(),
|
|
))
|
|
}
|
|
}
|
|
|
|
fn release_frame(&self, frame: VideoCodecFrame) {
|
|
unsafe {
|
|
ffi::gst_video_decoder_release_frame(self.as_ref().to_glib_none().0, frame.into_ptr())
|
|
}
|
|
}
|
|
|
|
fn drop_frame(&self, frame: VideoCodecFrame) -> Result<gst::FlowSuccess, gst::FlowError> {
|
|
unsafe {
|
|
gst::FlowSuccess::try_from_glib(ffi::gst_video_decoder_drop_frame(
|
|
self.as_ref().to_glib_none().0,
|
|
frame.into_ptr(),
|
|
))
|
|
}
|
|
}
|
|
|
|
fn latency(&self) -> (gst::ClockTime, gst::ClockTime) {
|
|
let mut min_latency = gst::ffi::GST_CLOCK_TIME_NONE;
|
|
let mut max_latency = gst::ffi::GST_CLOCK_TIME_NONE;
|
|
|
|
unsafe {
|
|
ffi::gst_video_decoder_get_latency(
|
|
self.as_ref().to_glib_none().0,
|
|
&mut min_latency,
|
|
&mut max_latency,
|
|
);
|
|
|
|
(from_glib(min_latency), from_glib(max_latency))
|
|
}
|
|
}
|
|
|
|
fn set_latency(&self, min_latency: gst::ClockTime, max_latency: gst::ClockTime) {
|
|
unsafe {
|
|
ffi::gst_video_decoder_set_latency(
|
|
self.as_ref().to_glib_none().0,
|
|
min_latency.into_glib(),
|
|
max_latency.into_glib(),
|
|
);
|
|
}
|
|
}
|
|
|
|
fn frame(&self, frame_number: i32) -> Option<VideoCodecFrame> {
|
|
let frame = unsafe {
|
|
ffi::gst_video_decoder_get_frame(self.as_ref().to_glib_none().0, frame_number)
|
|
};
|
|
|
|
if frame.is_null() {
|
|
None
|
|
} else {
|
|
unsafe { Some(VideoCodecFrame::new(frame, self.as_ref())) }
|
|
}
|
|
}
|
|
|
|
fn frames(&self) -> Vec<VideoCodecFrame> {
|
|
unsafe {
|
|
let frames = ffi::gst_video_decoder_get_frames(self.as_ref().to_glib_none().0);
|
|
let mut iter: *const glib::ffi::GList = frames;
|
|
let mut vec = Vec::new();
|
|
|
|
while !iter.is_null() {
|
|
let frame_ptr = Ptr::from((*iter).data);
|
|
/* transfer ownership of the frame */
|
|
let frame = VideoCodecFrame::new(frame_ptr, self.as_ref());
|
|
vec.push(frame);
|
|
iter = (*iter).next;
|
|
}
|
|
|
|
glib::ffi::g_list_free(frames);
|
|
vec
|
|
}
|
|
}
|
|
|
|
fn oldest_frame(&self) -> Option<VideoCodecFrame> {
|
|
let frame =
|
|
unsafe { ffi::gst_video_decoder_get_oldest_frame(self.as_ref().to_glib_none().0) };
|
|
|
|
if frame.is_null() {
|
|
None
|
|
} else {
|
|
unsafe { Some(VideoCodecFrame::new(frame, self.as_ref())) }
|
|
}
|
|
}
|
|
|
|
fn output_state(&self) -> Option<VideoCodecState<'static, Readable>> {
|
|
let state =
|
|
unsafe { ffi::gst_video_decoder_get_output_state(self.as_ref().to_glib_none().0) };
|
|
|
|
if state.is_null() {
|
|
None
|
|
} else {
|
|
unsafe { Some(VideoCodecState::<Readable>::new(state)) }
|
|
}
|
|
}
|
|
|
|
fn set_output_state(
|
|
&self,
|
|
fmt: VideoFormat,
|
|
width: u32,
|
|
height: u32,
|
|
reference: Option<&VideoCodecState<Readable>>,
|
|
) -> Result<VideoCodecState<InNegotiation>, gst::FlowError> {
|
|
let state = unsafe {
|
|
let reference = match reference {
|
|
Some(reference) => reference.as_mut_ptr(),
|
|
None => ptr::null_mut(),
|
|
};
|
|
ffi::gst_video_decoder_set_output_state(
|
|
self.as_ref().to_glib_none().0,
|
|
fmt.into_glib(),
|
|
width,
|
|
height,
|
|
reference,
|
|
)
|
|
};
|
|
|
|
if state.is_null() {
|
|
Err(gst::FlowError::NotNegotiated)
|
|
} else {
|
|
unsafe { Ok(VideoCodecState::<InNegotiation>::new(state, self.as_ref())) }
|
|
}
|
|
}
|
|
|
|
#[cfg(any(feature = "v1_16", feature = "dox"))]
|
|
#[cfg_attr(feature = "dox", doc(cfg(feature = "v1_16")))]
|
|
fn set_interlaced_output_state(
|
|
&self,
|
|
fmt: VideoFormat,
|
|
mode: VideoInterlaceMode,
|
|
width: u32,
|
|
height: u32,
|
|
reference: Option<&VideoCodecState<Readable>>,
|
|
) -> Result<VideoCodecState<InNegotiation>, gst::FlowError> {
|
|
let state = unsafe {
|
|
let reference = match reference {
|
|
Some(reference) => reference.as_mut_ptr(),
|
|
None => ptr::null_mut(),
|
|
};
|
|
ffi::gst_video_decoder_set_interlaced_output_state(
|
|
self.as_ref().to_glib_none().0,
|
|
fmt.into_glib(),
|
|
mode.into_glib(),
|
|
width,
|
|
height,
|
|
reference,
|
|
)
|
|
};
|
|
|
|
if state.is_null() {
|
|
Err(gst::FlowError::NotNegotiated)
|
|
} else {
|
|
unsafe { Ok(VideoCodecState::<InNegotiation>::new(state, self.as_ref())) }
|
|
}
|
|
}
|
|
|
|
fn negotiate<'a>(
|
|
&'a self,
|
|
output_state: VideoCodecState<'a, InNegotiation<'a>>,
|
|
) -> Result<(), gst::FlowError> {
|
|
// Consume output_state so user won't be able to modify it anymore
|
|
let self_ptr = self.to_glib_none().0 as *const gst::ffi::GstElement;
|
|
assert_eq!(output_state.context.element_as_ptr(), self_ptr);
|
|
|
|
let ret = unsafe {
|
|
from_glib(ffi::gst_video_decoder_negotiate(
|
|
self.as_ref().to_glib_none().0,
|
|
))
|
|
};
|
|
if ret {
|
|
Ok(())
|
|
} else {
|
|
Err(gst::FlowError::NotNegotiated)
|
|
}
|
|
}
|
|
fn error<T: gst::MessageErrorDomain>(
|
|
&self,
|
|
weight: i32,
|
|
code: T,
|
|
message: Option<&str>,
|
|
debug: Option<&str>,
|
|
file: &str,
|
|
function: &str,
|
|
line: u32,
|
|
) -> Result<gst::FlowSuccess, gst::FlowError> {
|
|
unsafe {
|
|
gst::FlowSuccess::try_from_glib(_gst_video_decoder_error(
|
|
self.as_ref().to_glib_none().0,
|
|
weight,
|
|
T::domain().into_glib(),
|
|
code.code(),
|
|
message.to_glib_full(),
|
|
debug.to_glib_full(),
|
|
file.to_glib_none().0,
|
|
function.to_glib_none().0,
|
|
line as i32,
|
|
))
|
|
}
|
|
}
|
|
}
|
|
|
|
impl HasStreamLock for VideoDecoder {
|
|
fn stream_lock(&self) -> *mut glib::ffi::GRecMutex {
|
|
let decoder_sys: *const ffi::GstVideoDecoder = self.to_glib_none().0;
|
|
unsafe { &(*decoder_sys).stream_lock as *const _ as usize as *mut _ }
|
|
}
|
|
|
|
fn element_as_ptr(&self) -> *const gst::ffi::GstElement {
|
|
let decoder_sys: *const ffi::GstVideoDecoder = self.to_glib_none().0;
|
|
decoder_sys as *const gst::ffi::GstElement
|
|
}
|
|
}
|
|
|
|
#[macro_export]
|
|
macro_rules! video_decoder_error(
|
|
($obj:expr, $weight:expr, $err:expr, ($msg:expr), [$debug:expr]) => { {
|
|
use $crate::prelude::VideoDecoderExtManual;
|
|
$obj.error(
|
|
$weight,
|
|
$err,
|
|
Some($msg),
|
|
Some($debug),
|
|
file!(),
|
|
module_path!(),
|
|
line!(),
|
|
)
|
|
}};
|
|
($obj:expr, $weight:expr, $err:expr, ($msg:expr)) => { {
|
|
use $crate::prelude::VideoDecoderExtManual;
|
|
$obj.error(
|
|
$weight,
|
|
$err,
|
|
Some($msg),
|
|
None,
|
|
file!(),
|
|
module_path!(),
|
|
line!(),
|
|
)
|
|
}};
|
|
($obj:expr, $weight:expr, $err:expr, [$debug:expr]) => { {
|
|
use $crate::prelude::VideoDecoderExtManual;
|
|
$obj.error(
|
|
$weight,
|
|
$err,
|
|
None,
|
|
Some($debug),
|
|
file!(),
|
|
module_path!(),
|
|
line!(),
|
|
)
|
|
}};
|
|
($obj:expr, $weight:expr, $err:expr, ($($msg:tt)*), [$($debug:tt)*]) => { {
|
|
use $crate::prelude::VideoDecoderExtManual;
|
|
$obj.error(
|
|
$weight,
|
|
$err,
|
|
Some(&format!($($msg)*)),
|
|
Some(&format!($($debug)*)),
|
|
file!(),
|
|
module_path!(),
|
|
line!(),
|
|
)
|
|
}};
|
|
($obj:expr, $weight:expr, $err:expr, ($($msg:tt)*)) => { {
|
|
use $crate::prelude::VideoDecoderExtManual;
|
|
$obj.error(
|
|
$weight,
|
|
$err,
|
|
Some(&format!($($msg)*)),
|
|
None,
|
|
file!(),
|
|
module_path!(),
|
|
line!(),
|
|
)
|
|
}};
|
|
($obj:expr, $weight:expr, $err:expr, [$($debug:tt)*]) => { {
|
|
use $crate::prelude::VideoDecoderExtManual;
|
|
$obj.error(
|
|
$weight,
|
|
$err,
|
|
None,
|
|
Some(&format!($($debug)*)),
|
|
file!(),
|
|
module_path!(),
|
|
line!(),
|
|
)
|
|
}};
|
|
);
|