[*] refactor: image_composition support background-remove-mask

Author: heng30

lib/background-remover/src/remover.rs

@@ -244,7 +244,7 @@ impl BackgroundRemover {
         let mut mask = vec![0u8; (height * width) as usize];
         for i in 0..mask.len() {
             let clamped = data[i].max(0.0).min(1.0); // Clamp to [0, 1] and convert to 0-255
-            mask[i] = (clamped * 255.0) as u8; //foreground (high) -> 255, background (low) -> 0
+            mask[i] = (clamped * 255.0) as u8; // foreground (high) -> 255, background (low) -> 0
         }
 
         let mask_image: GrayImage = ImageBuffer::from_raw(width, height, mask)

lib/camera/examples/background_removal_demo.rs

@@ -0,0 +1,75 @@
+use anyhow::Result;
+use camera::{
+    ShapeCircle,
+    image_composition::{MixPositionWithPadding, Shape, ShapeBase, mix_images_rgb},
+};
+use image::RgbImage;
+
+fn main() -> Result<()> {
+    env_logger::Builder::from_env(env_logger::Env::default().default_filter_or("info")).init();
+
+    log::info!("=== Background Removal Demo (RGB) ===");
+
+    log::info!("Loading camera image and mask...");
+    let camera_image = image::open("./examples/test.png")
+        .map_err(|e| anyhow::anyhow!("Failed to load camera image: {}", e))?
+        .to_rgb8();
+
+    let mask = image::open("./examples/mask.png")
+        .map_err(|e| anyhow::anyhow!("Failed to load mask: {}", e))?
+        .to_luma8();
+
+    log::info!(
+        "  ✓ Loaded: {}x{}",
+        camera_image.width(),
+        camera_image.height()
+    );
+
+    let background = RgbImage::from_fn(1920, 1080, |x, y| {
+        let r = (x as f32 / 1920.0 * 180.0) as u8 + 40;
+        let g = (y as f32 / 1080.0 * 180.0) as u8 + 40;
+        let b = 150;
+        image::Rgb([r, g, b])
+    });
+
+    log::info!("1. Rectangle with background removal (no zoom)");
+    let circle_no_zoom = ShapeCircle::default().with_radius(250).with_base(
+        ShapeBase::default()
+            .with_pos(MixPositionWithPadding::BottomRight((80, 80)))
+            .with_zoom(1.0)
+            .with_clip_pos((0.3, 0.2)),
+    );
+
+    let result_no_zoom = mix_images_rgb(
+        background.clone(),
+        camera_image.clone(),
+        Some(mask.clone()),
+        Shape::Circle(circle_no_zoom),
+    )?;
+    result_no_zoom.save("tmp/bg_removal_no_zoom.png")?;
+    log::info!("  ✓ Saved: tmp/bg_removal_no_zoom.png");
+
+    log::info!("2. Rectangle with background removal (1.5x zoom)");
+    let circle_zoom = ShapeCircle::default().with_radius(250).with_base(
+        ShapeBase::default()
+            .with_pos(MixPositionWithPadding::BottomRight((80, 80)))
+            .with_zoom(1.5)
+            .with_clip_pos((0.3, 0.2)),
+    );
+
+    let result_zoom = mix_images_rgb(
+        background.clone(),
+        camera_image.clone(),
+        Some(mask.clone()),
+        Shape::Circle(circle_zoom),
+    )?;
+    result_zoom.save("tmp/bg_removal_zoom.png")?;
+    log::info!("  ✓ Saved: tmp/bg_removal_zoom.png");
+
+    log::info!("=== All tests completed! ===");
+    log::info!("Generated files:");
+    log::info!("  - tmp/bg_removal_no_zoom.png  (No zoom, with mask)");
+    log::info!("  - tmp/bg_removal_zoom.png     (1.5x zoom, with mask)");
+
+    Ok(())
+}

lib/camera/examples/circle_demo.rs

@@ -33,7 +33,7 @@ fn main() -> Result<()> {
             .with_clip_pos((0.5, 0.5)),
     );
 
-    let result1 = mix_images(bg1, test_image.clone(), Shape::Circle(circle1))?;
+    let result1 = mix_images(bg1, test_image.clone(), None, Shape::Circle(circle1))?;
     result1.save("tmp/circle_zoom_1.0x.png")?;
     log::info!("   ✓ Saved circle_zoom_1.0x.png");
 
@@ -47,7 +47,7 @@ fn main() -> Result<()> {
             .with_clip_pos((0.2, 0.3)),
     );
 
-    let result2 = mix_images(bg2, test_image.clone(), Shape::Circle(circle2))?;
+    let result2 = mix_images(bg2, test_image.clone(), None, Shape::Circle(circle2))?;
     result2.save("tmp/circle_zoom_2.0x.png")?;
     log::info!("   ✓ Saved circle_zoom_2.0x.png");
 
@@ -56,7 +56,7 @@ fn main() -> Result<()> {
     let circle3 = ShapeCircle::default()
         .with_radius(100)
         .with_base(ShapeBase::default().with_border_width(3).with_zoom(0.5));
-    let result3 = mix_images(bg3, test_image.clone(), Shape::Circle(circle3))?;
+    let result3 = mix_images(bg3, test_image.clone(), None, Shape::Circle(circle3))?;
     result3.save("tmp/circle_zoom_0.5x.png")?;
     log::info!("   ✓ Saved circle_zoom_0.5x.png");
 

lib/camera/examples/image_composition_demo.rs

@@ -37,6 +37,7 @@ fn main() -> Result<(), Box<dyn std::error::Error>> {
     let result_rect = mix_images(
         background_rgba.clone(),
         camera_image_rgba.clone(),
+        None,
         Shape::Rectangle(rect),
     )?;
     result_rect.save("tmp/composition_rectangle_rgba.png")?;
@@ -52,6 +53,7 @@ fn main() -> Result<(), Box<dyn std::error::Error>> {
     let result_circle = mix_images(
         background_rgba.clone(),
         camera_image_rgba.clone(),
+        None,
         Shape::Circle(circle),
     )?;
     result_circle.save("tmp/composition_circle_rgba.png")?;
@@ -85,6 +87,7 @@ fn main() -> Result<(), Box<dyn std::error::Error>> {
     let result_rect_rgb = mix_images_rgb(
         background_rgb.clone(),
         camera_image_rgb.clone(),
+        None,
         Shape::Rectangle(rect_rgb),
     )?;
     result_rect_rgb.save("tmp/composition_rectangle_rgb.png")?;
@@ -100,6 +103,7 @@ fn main() -> Result<(), Box<dyn std::error::Error>> {
     let result_circle_rgb = mix_images_rgb(
         background_rgb.clone(),
         camera_image_rgb.clone(),
+        None,
         Shape::Circle(circle_rgb),
     )?;
     result_circle_rgb.save("tmp/composition_circle_rgb.png")?;
@@ -115,6 +119,7 @@ fn main() -> Result<(), Box<dyn std::error::Error>> {
     let result_plain_rgb = mix_images_rgb(
         background_rgb.clone(),
         camera_image_rgb.clone(),
+        None,
         Shape::Rectangle(plain_rect_rgb),
     )?;
     result_plain_rgb.save("tmp/composition_rect_plain_rgb.png")?;
@@ -130,6 +135,7 @@ fn main() -> Result<(), Box<dyn std::error::Error>> {
     let result_plain_rgb = mix_images_rgb(
         background_rgb.clone(),
         camera_image_rgb.clone(),
+        None,
         Shape::Circle(plain_circle_rgb),
     )?;
     result_plain_rgb.save("tmp/composition_circle_plain_rgb.png")?;

lib/camera/examples/rectangle-demo.rs lib/camera/examples/rectangle_demo.rslib/camera/examples/rectangle-demo.rs renamed to lib/camera/examples/rectangle_demo.rs

@@ -27,7 +27,7 @@ fn main() -> Result<()> {
             .with_border_color(Rgba([255, 0, 0, 255]))
             .with_clip_pos((0.1, 0.1)),
     );
-    let result1 = mix_images(bg1, test_image.clone(), Shape::Rectangle(rect1))?;
+    let result1 = mix_images(bg1, test_image.clone(), None, Shape::Rectangle(rect1))?;
     result1.save("tmp/rect_zoom_1.0x.png")?;
     log::info!("   ✓ Saved zoom_1.0x.png (original size)");
 
@@ -41,7 +41,7 @@ fn main() -> Result<()> {
             .with_border_color(Rgba([255, 0, 0, 255]))
             .with_clip_pos((0.25, 0.25)),
     );
-    let result2 = mix_images(bg2, test_image.clone(), Shape::Rectangle(rect2))?;
+    let result2 = mix_images(bg2, test_image.clone(), None, Shape::Rectangle(rect2))?;
     result2.save("tmp/rect_zoom_2.0x.png")?;
     log::info!("   ✓ Saved zoom_2.0x.png (center cropped, 2x magnified)");
 
@@ -54,7 +54,7 @@ fn main() -> Result<()> {
             .with_border_color(Rgba([255, 0, 0, 255]))
             .with_clip_pos((0.30, 0.30)),
     );
-    let result3 = mix_images(bg3, test_image.clone(), Shape::Rectangle(rect3))?;
+    let result3 = mix_images(bg3, test_image.clone(), None, Shape::Rectangle(rect3))?;
     result3.save("tmp/rect_zoom_0.5x.png")?;
     log::info!("   ✓ Saved zoom_0.5x.png (shrunk with black padding)");