rustmotion_components/
audio_spectrum.rs

1use schemars::JsonSchema;
2use serde::{Deserialize, Serialize};
3use skia_safe::{Canvas, Color, Paint, PaintStyle, Point, Rect};
4
5use rustmotion_core::css::CssStyle;
6use rustmotion_core::engine::animator::AnimatedProperties;
7use rustmotion_core::engine::layout_pass::BoxLayout;
8use rustmotion_core::engine::renderer::audio_analysis::audio_analysis_cache;
9use rustmotion_core::engine::renderer::parse_hex_color;
10use rustmotion_core::schema::TimelineStep;
11use rustmotion_core::traits::{PaintCtx, Painter, TimingConfig};
12
13fn default_bars() -> u32 {
14    16
15}
16fn default_color() -> String {
17    "#38bdf8".to_string()
18}
19fn default_bar_gap() -> f32 {
20    2.0
21}
22fn default_min_height() -> f32 {
23    2.0
24}
25
26#[derive(Debug, Clone, Default, PartialEq, Serialize, Deserialize, JsonSchema)]
27#[serde(rename_all = "snake_case")]
28pub enum SpectrumMode {
29    #[default]
30    Bars,
31    Radial,
32}
33
34#[derive(Debug, Serialize, Deserialize, JsonSchema)]
35pub struct AudioSpectrum {
36    /// Source audio track (src path). If None, uses the first track in the cache.
37    #[serde(default)]
38    pub track: Option<String>,
39    /// Number of frequency bars to display (resampled from 16 internal bands).
40    #[serde(default = "default_bars")]
41    pub bars: u32,
42    /// Display mode: bars (vertical) or radial.
43    #[serde(default)]
44    pub mode: SpectrumMode,
45    /// Bar color as hex string.
46    #[serde(default = "default_color")]
47    pub color: String,
48    /// Gap between bars in pixels.
49    #[serde(default = "default_bar_gap")]
50    pub bar_gap: f32,
51    /// Minimum bar height in pixels.
52    #[serde(default = "default_min_height")]
53    pub min_height: f32,
54    #[serde(flatten)]
55    pub timing: TimingConfig,
56    #[serde(default)]
57    pub style: CssStyle,
58    #[serde(default)]
59    pub timeline: Vec<TimelineStep>,
60    #[serde(default)]
61    pub stagger: Option<f32>,
62}
63
64rustmotion_core::impl_traits!(AudioSpectrum {
65    Animatable => animation,
66    Timed => timing,
67    Styled => style,
68});
69
70impl AudioSpectrum {
71    fn get_band_values(&self, time: f64) -> Vec<f32> {
72        let cache = audio_analysis_cache();
73        let analysis = if let Some(ref src) = self.track {
74            cache.get(src).map(|r| r.clone())
75        } else {
76            cache.iter().next().map(|r| r.value().clone())
77        };
78
79        let n = self.bars.max(1) as usize;
80
81        let Some(analysis) = analysis else {
82            return vec![0.0; n];
83        };
84
85        let num_bands = 16usize;
86        (0..n)
87            .map(|i| {
88                let band_f = i as f32 * (num_bands as f32 - 1.0) / (n as f32 - 1.0).max(1.0);
89                let band_lo = band_f as usize;
90                let band_hi = (band_lo + 1).min(num_bands - 1);
91                let frac = band_f - band_lo as f32;
92                let v_lo = analysis.band_at(time, band_lo as u8);
93                let v_hi = analysis.band_at(time, band_hi as u8);
94                v_lo * (1.0 - frac) + v_hi * frac
95            })
96            .collect()
97    }
98}
99
100impl Painter for AudioSpectrum {
101    fn paint_content(
102        &self,
103        canvas: &Canvas,
104        layout: &BoxLayout,
105        _props: &AnimatedProperties,
106        ctx: &PaintCtx,
107    ) {
108        let w = layout.width;
109        let h = layout.height;
110        let n = self.bars.max(1) as usize;
111        let values = self.get_band_values(ctx.scenario_time);
112        let (r, g, b, a) = parse_hex_color(&self.color);
113        let color = Color::from_argb(a, r, g, b);
114
115        match self.mode {
116            SpectrumMode::Bars => {
117                let total_gap = self.bar_gap * (n as f32 - 1.0);
118                let bar_w = ((w - total_gap) / n as f32).max(1.0);
119                let mut paint = Paint::default();
120                paint.set_color(color);
121                paint.set_style(PaintStyle::Fill);
122                paint.set_anti_alias(true);
123
124                for (i, &v) in values.iter().enumerate() {
125                    let bar_h = (v * h).max(self.min_height);
126                    let x = i as f32 * (bar_w + self.bar_gap);
127                    let y = h - bar_h;
128                    canvas.draw_rect(Rect::from_xywh(x, y, bar_w, bar_h), &paint);
129                }
130            }
131            SpectrumMode::Radial => {
132                let cx = w / 2.0;
133                let cy = h / 2.0;
134                let max_r = cx.min(cy);
135                let inner_r = max_r * 0.3;
136
137                let mut paint = Paint::default();
138                paint.set_color(color);
139                paint.set_style(PaintStyle::Stroke);
140                paint.set_stroke_width(2.0);
141                paint.set_anti_alias(true);
142
143                for (i, &v) in values.iter().enumerate() {
144                    let angle = (i as f32 / n as f32) * 2.0 * std::f32::consts::PI;
145                    let bar_len = (v * (max_r - inner_r)).max(self.min_height);
146                    let x0 = cx + inner_r * angle.cos();
147                    let y0 = cy + inner_r * angle.sin();
148                    let x1 = cx + (inner_r + bar_len) * angle.cos();
149                    let y1 = cy + (inner_r + bar_len) * angle.sin();
150                    canvas.draw_line(Point::new(x0, y0), Point::new(x1, y1), &paint);
151                }
152            }
153        }
154    }
155}
rustmotion_components/audio_spectrum.rs

rustmotion_components/
audio_spectrum.rs
