[*] refactor

Author: heng30

lib/video-editor/src/filters/traits.rs

@@ -12,6 +12,19 @@ use std::{
     time::Duration,
 };
 
+// 滤镜效果作用的位置
+#[derive(Debug, Clone, Copy, PartialEq, Eq, serde::Serialize, serde::Deserialize)]
+pub enum EffectPosition {
+    Start, // 作用于 segment 开头
+    End,   // 作用于 segment 结尾
+}
+
+impl Default for EffectPosition {
+    fn default() -> Self {
+        Self::Start
+    }
+}
+
 #[derive(Debug, Clone)]
 pub struct VideoFilterConfig {
     pub output_width: u32,

lib/video-editor/src/filters/video.rs

@@ -1,20 +1,24 @@
+pub mod border;
 pub mod chroma;
 pub mod crop;
 pub mod fade_in;
 pub mod fade_out;
 pub mod flip;
 pub mod moving;
+pub mod opacity;
 pub mod slide;
 pub mod transform;
 pub mod wipe;
 pub mod zoom;
 
+pub use border::BorderFilter;
 pub use chroma::ChromaKeyFilter;
 pub use crop::CropFilter;
 pub use fade_in::FadeInFilter;
 pub use fade_out::FadeOutFilter;
 pub use flip::{FlipDirection, FlipFilter};
 pub use moving::{EasingFunction, MovingFilter};
+pub use opacity::OpacityFilter;
 pub use slide::{SlideDirection, SlideFilter};
 pub use transform::TransformFilter;
 pub use wipe::{WipeDirection, WipeFilter};
@@ -31,6 +35,8 @@ pub fn all_filter_names() -> &'static [&'static str] {
         FlipFilter::NAME,
         FadeInFilter::NAME,
         FadeOutFilter::NAME,
+        BorderFilter::NAME,
+        OpacityFilter::NAME,
         ChromaKeyFilter::NAME,
     ]
 }

lib/video-editor/src/filters/video/border.rs

@@ -0,0 +1,83 @@
+use crate::{
+    Result,
+    filters::traits::{VideoData, VideoFilter},
+    tracks::video_frame_cache::VideoImage,
+};
+use image::Rgba;
+
+#[derive(Debug, Clone, serde::Serialize, serde::Deserialize)]
+pub struct BorderFilter {
+    pub size: u32,           // border width in pixels
+    pub color: [u8; 4],      // RGBA color
+    pub corner_radius: u32,  // rounded corner radius
+}
+
+impl Default for BorderFilter {
+    fn default() -> Self {
+        Self {
+            size: 0,
+            color: [0, 0, 0, 255],
+            corner_radius: 0,
+        }
+    }
+}
+
+impl BorderFilter {
+    pub const NAME: &'static str = "border";
+
+    pub fn new(size: u32, color: [u8; 4], corner_radius: u32) -> Self {
+        Self { size, color, corner_radius }
+    }
+
+    fn draw_border(&self, image: &mut image::RgbaImage) {
+        let width = image.width();
+        let height = image.height();
+        let border_color = Rgba(self.color);
+        let size = self.size.min(width / 2).min(height / 2);
+
+        if size == 0 {
+            return;
+        }
+
+        // Draw top border
+        for y in 0..size {
+            for x in 0..width {
+                image.put_pixel(x, y, border_color);
+            }
+        }
+
+        // Draw bottom border
+        for y in (height - size)..height {
+            for x in 0..width {
+                image.put_pixel(x, y, border_color);
+            }
+        }
+
+        // Draw left border
+        for y in size..(height - size) {
+            for x in 0..size {
+                image.put_pixel(x, y, border_color);
+            }
+        }
+
+        // Draw right border
+        for y in size..(height - size) {
+            for x in (width - size)..width {
+                image.put_pixel(x, y, border_color);
+            }
+        }
+    }
+}
+
+impl VideoFilter for BorderFilter {
+    crate::impl_default_video_filter!(BorderFilter);
+
+    fn apply(&self, data: &mut VideoData) -> Result<()> {
+        for frame in &mut data.frames {
+            if let VideoImage::Image { buffer, .. } = frame {
+                self.draw_border(buffer);
+            }
+        }
+        Ok(())
+    }
+}

lib/video-editor/src/filters/video/flip.rs

@@ -3,7 +3,7 @@ use crate::{
     filters::traits::{VideoData, VideoFilter},
     tracks::video_frame_cache::VideoImage,
 };
-use image::{imageops, RgbaImage};
+use image::{RgbaImage, imageops};
 
 #[derive(Debug, Clone, Copy, PartialEq, Eq, serde::Serialize, serde::Deserialize)]
 pub enum FlipDirection {

lib/video-editor/src/filters/video/opacity.rs

@@ -0,0 +1,40 @@
+use crate::{
+    Result,
+    filters::traits::{VideoData, VideoFilter},
+    tracks::video_frame_cache::VideoImage,
+};
+use rayon::prelude::*;
+
+#[derive(Debug, Clone, serde::Serialize, serde::Deserialize)]
+pub struct OpacityFilter {
+    pub opacity: f32, // 0.0 = fully transparent, 1.0 = fully opaque
+}
+
+impl Default for OpacityFilter {
+    fn default() -> Self {
+        Self { opacity: 1.0 }
+    }
+}
+
+impl OpacityFilter {
+    pub const NAME: &'static str = "opacity";
+
+    pub fn new(opacity: f32) -> Self {
+        Self { opacity: opacity.clamp(0.0, 1.0) }
+    }
+}
+
+impl VideoFilter for OpacityFilter {
+    crate::impl_default_video_filter!(OpacityFilter);
+
+    fn apply(&self, data: &mut VideoData) -> Result<()> {
+        for frame in data.frames.iter_mut() {
+            if let VideoImage::Image { buffer, .. } = frame {
+                buffer.par_pixels_mut().for_each(|pixel| {
+                    pixel.0[3] = ((pixel.0[3] as f32) * self.opacity).clamp(0.0, 255.0) as u8;
+                });
+            }
+        }
+        Ok(())
+    }
+}