From: Kostya Shishkov <kostya.shishkov@gmail.com>
Date: Mon, 24 Mar 2025 16:12:59 +0000 (+0100)
Subject: core/scale: fill planes not present in input format (e.g. grayscale -> YUV)
X-Git-Url: https://git.nihav.org/?a=commitdiff_plain;h=d03595c58cda3ce27ac9e1952e6bde2dfcc90624;p=nihav.git

core/scale: fill planes not present in input format (e.g. grayscale -> YUV)
---

diff --git a/nihav-core/src/scale/fill.rs b/nihav-core/src/scale/fill.rs
new file mode 100644
index 0000000..ffd2695
--- /dev/null
+++ b/nihav-core/src/scale/fill.rs
@@ -0,0 +1,89 @@
+use super::*;
+use super::kernel::Kernel;
+
+struct FillKernel {}
+
+impl FillKernel {
+    fn new() -> Self { Self{} }
+}
+
+impl Kernel for FillKernel {
+    fn init(&mut self, in_fmt: &ScaleInfo, dest_fmt: &ScaleInfo, _options: &[(String, String)]) -> ScaleResult<NABufferType> {
+        let mut dfmt = dest_fmt.fmt;
+        for (i, chr) in dfmt.comp_info.iter_mut().flatten().enumerate() {
+            chr.packed = false;
+            chr.comp_offs = i as u8;
+            chr.next_elem = 0;
+        }
+        let res = alloc_video_buffer(NAVideoInfo::new(in_fmt.width, in_fmt.height, false, dfmt), 3);
+        if res.is_err() { return Err(ScaleError::AllocError); }
+        Ok(res.unwrap())
+    }
+    fn process(&mut self, pic_in: &NABufferType, pic_out: &mut NABufferType) {
+        if let Some(ref vbuf) = pic_in.get_vbuf() {
+            if let Some(ref mut dbuf) = pic_out.get_vbuf() {
+                let ifmt = vbuf.get_info().format;
+                let ofmt = dbuf.get_info().format;
+
+                let num_icore = usize::from(ifmt.components) - if ifmt.alpha { 1 } else { 0 };
+                let num_ocore = usize::from(ofmt.components) - if ofmt.alpha { 1 } else { 0 };
+
+                for plane_no in 0..num_icore.min(num_ocore) {
+                    let splane_off = vbuf.get_offset(plane_no);
+                    let splane_stride = vbuf.get_stride(plane_no);
+                    let src = vbuf.get_data();
+
+                    let dplane_off = dbuf.get_offset(plane_no);
+                    let dplane_stride = dbuf.get_stride(plane_no);
+                    let (width, height) = dbuf.get_dimensions(plane_no);
+                    let dst = dbuf.get_data_mut().unwrap();
+
+                    for (dline, sline) in dst[dplane_off..].chunks_mut(dplane_stride)
+                            .zip(src[splane_off..].chunks(splane_stride)).take(height) {
+                        dline[..width].copy_from_slice(&sline[..width]);
+                    }
+                }
+                if num_icore < num_ocore {
+                    let fill_val = match ofmt.model {
+                            ColorModel::RGB(_) => 0x00,
+                            ColorModel::YUV(_) => 0x80,
+                            _ => unimplemented!(),
+                        };
+                    for plane_no in num_icore..num_ocore {
+                        let plane_off = dbuf.get_offset(plane_no);
+                        let plane_stride = dbuf.get_stride(plane_no);
+                        let (width, height) = dbuf.get_dimensions(plane_no);
+                        let dst = dbuf.get_data_mut().unwrap();
+
+                        for line in dst[plane_off..].chunks_mut(plane_stride).take(height) {
+                            for el in line.iter_mut().take(width) {
+                                *el = fill_val;
+                            }
+                        }
+                    }
+                }
+                if !ifmt.alpha && ofmt.alpha {
+                    let alpha_plane = usize::from(ofmt.components) - 1;
+                    let alpha_off = dbuf.get_offset(alpha_plane);
+                    let alpha_stride = dbuf.get_stride(alpha_plane);
+                    let (width, height) = dbuf.get_dimensions(alpha_plane);
+                    let dst = dbuf.get_data_mut().unwrap();
+
+                    for line in dst[alpha_off..].chunks_mut(alpha_stride).take(height) {
+                        for el in line.iter_mut().take(width) {
+                            *el = 0xFF;
+                        }
+                    }
+                }
+            } else {
+                unreachable!();
+            }
+        } else {
+            unimplemented!();
+        }
+    }
+}
+
+pub fn create_fill() -> Box<dyn Kernel> {
+    Box::new(FillKernel::new())
+}
diff --git a/nihav-core/src/scale/mod.rs b/nihav-core/src/scale/mod.rs
index 8054523..93d42a4 100644
--- a/nihav-core/src/scale/mod.rs
+++ b/nihav-core/src/scale/mod.rs
@@ -20,6 +20,7 @@ use crate::frame::*;
 mod kernel;
 
 mod colorcvt;
+mod fill;
 mod repack;
 #[allow(clippy::module_inception)]
 mod scale;
@@ -91,6 +92,7 @@ const KERNELS: &[KernelDesc] = &[
     KernelDesc { name: "depal",         create: repack::create_depal },
     KernelDesc { name: "palette",       create: palette::create_palettise },
     KernelDesc { name: "scale",         create: scale::create_scale },
+    KernelDesc { name: "fill",          create: fill::create_fill },
     KernelDesc { name: "rgb_to_yuv",    create: colorcvt::create_rgb2yuv },
     KernelDesc { name: "yuv_to_rgb",    create: colorcvt::create_yuv2rgb },
 ];
@@ -359,6 +361,16 @@ fn build_pipeline(ifmt: &ScaleInfo, ofmt: &ScaleInfo, just_convert: bool, option
         cur_fmt = new_stage.fmt_out;
         add_stage!(stages, new_stage);
     }
+    let icomponents = cur_fmt.fmt.components - if cur_fmt.fmt.alpha { 1 } else { 0 };
+    let ocomponents = ofmt.fmt.components - if ofmt.fmt.alpha { 1 } else { 0 };
+    if !needs_palettise && ((!cur_fmt.fmt.alpha && ofmt.fmt.alpha) || (icomponents < ocomponents)) {
+        if debug {
+            println!("[adding fill]");
+        }
+        let new_stage = Stage::new("fill", &cur_fmt, ofmt, options)?;
+        cur_fmt = new_stage.fmt_out;
+        add_stage!(stages, new_stage);
+    }
     if needs_pack && !needs_palettise {
         if debug {
             println!("[adding pack]");