mirrors/gst-plugins-rs
mirrors/gst-plugins-rs
1
1
Fork 0
mirror of https://gitlab.freedesktop.org/gstreamer/gst-plugins-rs.git synced 2025-01-23 17:38:20 +00:00
Code Issues Projects Releases Wiki Activity

dav1ddec: Add 8/16 bit grayscale support

Browse source
This commit is contained in:
Sebastian Dröge 2022-02-03 13:23:51 +02:00
parent 9570fcaddd
commit 7bb42f3f5e
1 changed files with 39 additions and 19 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

58
video/dav1d/src/dav1ddec/imp.rs
View file

@ -50,15 +50,24 @@ impl Dav1dDec {
) -> gst_video::VideoFormat { ) -> gst_video::VideoFormat {
let bpc = pic.bits_per_component(); let bpc = pic.bits_per_component();
let format_desc = match (pic.pixel_layout(), bpc) { let format_desc = match (pic.pixel_layout(), bpc) {
// (dav1d::PixelLayout::I400, Some(dav1d::BitsPerComponent(8))) => "GRAY8", (dav1d::PixelLayout::I400, Some(dav1d::BitsPerComponent(8))) => "GRAY8",
#[cfg(target_endian = "little")]
(dav1d::PixelLayout::I400, Some(dav1d::BitsPerComponent(16))) => "GRAY16_LE",
#[cfg(target_endian = "big")]
(dav1d::PixelLayout::I400, Some(dav1d::BitsPerComponent(16))) => "GRAY16_BE",
// (dav1d::PixelLayout::I400, Some(dav1d::BitsPerComponent(10))) => "GRAY10_LE32", // (dav1d::PixelLayout::I400, Some(dav1d::BitsPerComponent(10))) => "GRAY10_LE32",
(dav1d::PixelLayout::I400, _) => return gst_video::VideoFormat::Unknown,
(dav1d::PixelLayout::I420, _) => "I420", (dav1d::PixelLayout::I420, _) => "I420",
(dav1d::PixelLayout::I422, Some(dav1d::BitsPerComponent(8))) => "Y42B", (dav1d::PixelLayout::I422, Some(dav1d::BitsPerComponent(8))) => "Y42B",
(dav1d::PixelLayout::I422, _) => "I422", (dav1d::PixelLayout::I422, _) => "I422",
(dav1d::PixelLayout::I444, _) => "Y444", (dav1d::PixelLayout::I444, _) => "Y444",
(dav1d::PixelLayout::Unknown, _) => { (layout, bpc) => {
gst_warning!(CAT, obj: element, "Unsupported dav1d format"); gst_warning!(
CAT,
obj: element,
"Unsupported dav1d format {:?}/{:?}",
layout,
bpc
);
return gst_video::VideoFormat::Unknown; return gst_video::VideoFormat::Unknown;
} }
}; };
@ -202,18 +211,25 @@ impl Dav1dDec {
let mut out_buffer = gst::Buffer::new(); let mut out_buffer = gst::Buffer::new();
let mut_buffer = out_buffer.get_mut().unwrap(); let mut_buffer = out_buffer.get_mut().unwrap();
// FIXME: For gray support we would need to deal only with the Y component. let components = if info.is_yuv() {
assert!(info.is_yuv()); const YUV_COMPONENTS: [dav1d::PlanarImageComponent; 3] = [
for component in [ dav1d::PlanarImageComponent::Y,
dav1d::PlanarImageComponent::Y, dav1d::PlanarImageComponent::U,
dav1d::PlanarImageComponent::U, dav1d::PlanarImageComponent::V,
dav1d::PlanarImageComponent::V, ];
] &YUV_COMPONENTS[..]
.iter() } else if info.is_gray() {
{ const GRAY_COMPONENTS: [dav1d::PlanarImageComponent; 1] =
let dest_stride: u32 = info.stride()[*component as usize].try_into().unwrap(); [dav1d::PlanarImageComponent::Y];
let plane = pic.plane(*component);
let (src_stride, height) = pic.plane_data_geometry(*component); &GRAY_COMPONENTS[..]
} else {
unreachable!();
};
for &component in components {
let dest_stride: u32 = info.stride()[component as usize].try_into().unwrap();
let plane = pic.plane(component);
let (src_stride, height) = pic.plane_data_geometry(component);
let mem = if video_meta_supported || src_stride == dest_stride { let mem = if video_meta_supported || src_stride == dest_stride {
gst::Memory::from_slice(plane) gst::Memory::from_slice(plane)
} else { } else {
@ -335,12 +351,16 @@ impl Dav1dDec {
fn video_output_formats() -> Vec<glib::SendValue> { fn video_output_formats() -> Vec<glib::SendValue> {
let values = [ let values = [
// gst_video::VideoFormat::Gray8, gst_video::VideoFormat::Gray8,
#[cfg(target_endian = "little")]
gst_video::VideoFormat::Gray16Le,
#[cfg(target_endian = "big")]
gst_video::VideoFormat::Gray16Be,
// #[cfg(target_endian = "little")]
// gst_video::VideoFormat::Gray10Le32,
gst_video::VideoFormat::I420, gst_video::VideoFormat::I420,
gst_video::VideoFormat::Y42b, gst_video::VideoFormat::Y42b,
gst_video::VideoFormat::Y444, gst_video::VideoFormat::Y444,
// #[cfg(target_endian = "little")]
// gst_video::VideoFormat::Gray10Le32,
#[cfg(target_endian = "little")] #[cfg(target_endian = "little")]
gst_video::VideoFormat::I42010le, gst_video::VideoFormat::I42010le,
#[cfg(target_endian = "little")] #[cfg(target_endian = "little")]