1use crate::layout::subpixel_snap_rect;
8use crate::motion::CASpring;
9use crate::theme::MacTheme;
10use crate::tokens::ProgressSizeMetrics;
11use egui::{
12 Align2, CornerRadius, FontId, Pos2, Rect, Response, Sense, Stroke, StrokeKind, Ui, Vec2, Widget,
13};
14
15#[derive(Debug, Clone, Copy, PartialEq, Eq, Default)]
17pub enum MacProgressStyle {
18 #[default]
20 Bar,
21 Spinning,
23}
24
25#[derive(Debug, Clone, Copy, PartialEq, Eq, Default)]
27pub enum MacProgressControlSize {
28 #[default]
30 Regular,
31 Small,
33 Mini,
35}
36
37pub struct MacProgressIndicator<'a> {
56 style: MacProgressStyle,
57 control_size: MacProgressControlSize,
58 progress: Option<f32>,
59 time: f64,
60 label: Option<&'a str>,
61 show_percentage: bool,
62 width: Option<f32>,
63 enabled: bool,
64 theme: Option<&'a MacTheme>,
65}
66
67impl<'a> Default for MacProgressIndicator<'a> {
68 fn default() -> Self {
69 Self::new()
70 }
71}
72
73impl<'a> MacProgressIndicator<'a> {
74 pub fn new() -> Self {
76 Self {
77 style: MacProgressStyle::Bar,
78 control_size: MacProgressControlSize::Regular,
79 progress: Some(0.0),
80 time: 0.0,
81 label: None,
82 show_percentage: false,
83 width: None,
84 enabled: true,
85 theme: None,
86 }
87 }
88
89 pub fn bar(progress: f32) -> Self {
91 Self::new().style(MacProgressStyle::Bar).progress(progress)
92 }
93
94 pub fn spinning(time: f64) -> Self {
96 Self::new()
97 .style(MacProgressStyle::Spinning)
98 .indeterminate(time)
99 }
100
101 pub fn style(mut self, style: MacProgressStyle) -> Self {
103 self.style = style;
104 self
105 }
106
107 pub fn control_size(mut self, size: MacProgressControlSize) -> Self {
109 self.control_size = size;
110 self
111 }
112
113 pub fn progress(mut self, progress: f32) -> Self {
115 self.progress = Some(progress.clamp(0.0, 1.0));
116 self
117 }
118
119 pub fn indeterminate(mut self, time: f64) -> Self {
121 self.progress = None;
122 self.time = time;
123 self
124 }
125
126 pub fn time(mut self, time: f64) -> Self {
128 self.time = time;
129 self
130 }
131
132 pub fn label(mut self, label: &'a str) -> Self {
134 self.label = Some(label);
135 self
136 }
137
138 pub fn show_percentage(mut self, show: bool) -> Self {
140 self.show_percentage = show;
141 self
142 }
143
144 pub fn width(mut self, width: f32) -> Self {
146 self.width = Some(width);
147 self
148 }
149
150 pub fn enabled(mut self, enabled: bool) -> Self {
152 self.enabled = enabled;
153 self
154 }
155
156 pub fn theme(mut self, theme: &'a MacTheme) -> Self {
158 self.theme = Some(theme);
159 self
160 }
161
162 pub fn clamped_progress(&self) -> Option<f32> {
164 self.progress.map(|p| p.clamp(0.0, 1.0))
165 }
166
167 fn size_metrics<'t>(&self, theme: &'t MacTheme) -> &'t ProgressSizeMetrics {
168 let pm = theme.progress_indicator();
169 match self.control_size {
170 MacProgressControlSize::Regular => &pm.regular,
171 MacProgressControlSize::Small => &pm.small,
172 MacProgressControlSize::Mini => &pm.mini,
173 }
174 }
175
176 pub fn show(self, ui: &mut Ui) -> Response {
178 ui.add(self)
179 }
180}
181
182impl<'a> Widget for MacProgressIndicator<'a> {
183 fn ui(self, ui: &mut Ui) -> Response {
184 let theme_cow = MacTheme::resolve(self.theme, ui.ctx());
185 let theme: &MacTheme = &theme_cow;
186
187 let pm = theme.progress_indicator();
188 let sm = self.size_metrics(theme);
189 let font_sz = sm.font_size.points();
190 let label_spacing = pm.label_spacing.points();
191 let pct_scale = pm.percentage_scale.points();
192
193 let status_text: Option<String> = match (self.label, self.show_percentage, self.progress) {
194 (Some(lbl), true, Some(p)) => {
195 let pct = (p.clamp(0.0, 1.0) * pct_scale).round() as u32;
196 Some(format!("{lbl} ({pct}%)"))
197 }
198 (Some(lbl), _, _) => Some(lbl.to_string()),
199 (None, true, Some(p)) => {
200 let pct = (p.clamp(0.0, 1.0) * pct_scale).round() as u32;
201 Some(format!("{pct}%"))
202 }
203 (None, _, _) => None,
204 };
205
206 match self.style {
207 MacProgressStyle::Bar => {
208 let bar_w = self.width.unwrap_or_else(|| pm.default_bar_width.points());
209 let bar_h = sm.bar_height.points();
210 let total_h = if status_text.is_some() {
211 bar_h + label_spacing + font_sz
212 } else {
213 bar_h
214 };
215
216 let (raw_rect, response) =
217 ui.allocate_exact_size(Vec2::new(bar_w, total_h), Sense::hover());
218 let rect = subpixel_snap_rect(raw_rect, ui.ctx().pixels_per_point());
219 let track_rect = Rect::from_min_size(rect.min, Vec2::new(rect.width(), bar_h));
220 let radius = CornerRadius::same(sm.bar_corner_radius.points() as u8);
221 let painter = ui.painter_at(rect);
222
223 painter.rect(
224 track_rect,
225 radius,
226 theme.control(),
227 Stroke::new(pm.border_width.points(), theme.separator()),
228 StrokeKind::Inside,
229 );
230
231 let fill_color = if self.enabled {
232 theme.control_accent()
233 } else {
234 theme.selected_control_inactive()
235 };
236
237 if let Some(target_p) = self.clamped_progress() {
238 let spring_preset = &theme.motion().swiftui_smooth;
239 let spring = CASpring::new(spring_preset.response, spring_preset.damping_ratio);
240 let anim_id = response.id.with("spring_progress");
241 let now = if self.time > 0.0 {
242 self.time
243 } else {
244 ui.input(|i| i.time)
245 };
246 let smooth_p = spring
247 .animate_f32_at(ui.ctx(), anim_id, target_p, now, true)
248 .clamp(0.0, 1.0);
249
250 if smooth_p > 0.0 {
251 let fill_w = track_rect.width() * smooth_p;
252 let fill_rect =
253 Rect::from_min_size(track_rect.min, Vec2::new(fill_w, bar_h));
254 painter.rect_filled(fill_rect, radius, fill_color);
255 }
256 } else {
257 let pill_ratio = pm.indeterminate_pill_ratio.points().clamp(0.0, 1.0);
258 let pill_w = track_rect.width() * pill_ratio;
259 let travel = (track_rect.width() - pill_w).max(0.0);
260 let freq = pm.bar_indeterminate_speed.points() as f64;
261 let tau = pm.tau_radians.points() as f64;
262 let phase = (((self.time * freq * tau).sin() + 1.0) * 0.5) as f32;
263 let pill_x = track_rect.left() + travel * phase;
264 let pill_rect = Rect::from_min_size(
265 Pos2::new(pill_x, track_rect.top()),
266 Vec2::new(pill_w, bar_h),
267 );
268 painter.rect_filled(pill_rect, radius, fill_color);
269 if self.time > 0.0 && crate::layout::is_window_active(ui.ctx()) {
270 ui.ctx()
271 .request_repaint_after(std::time::Duration::from_millis(33));
272 }
273 }
274
275 if let Some(text) = status_text {
276 let label_y = track_rect.bottom() + label_spacing;
277 painter.text(
278 Pos2::new(track_rect.left(), label_y),
279 Align2::LEFT_TOP,
280 text,
281 FontId::proportional(font_sz),
282 theme.secondary_label(),
283 );
284 }
285
286 response
287 }
288 MacProgressStyle::Spinning => {
289 let spin_sz = sm.spinner_size.points();
290 let (raw_rect, response) =
291 ui.allocate_exact_size(Vec2::splat(spin_sz), Sense::hover());
292 let rect = subpixel_snap_rect(raw_rect, ui.ctx().pixels_per_point());
293 let painter = ui.painter_at(rect);
294
295 let center = rect.center();
296 let spoke_w = sm.spoke_width.points();
297 let tip_r = spoke_w * 0.5;
298 let spin_r = spin_sz * 0.5;
299 let outer_r = (spin_r - tip_r).max(0.0);
300 let inner_ratio = pm.spoke_inner_ratio.points();
301 let inner_r = outer_r * inner_ratio;
302 let spoke_count = (pm.spoke_count.points().round() as usize).max(1);
303 let tau = pm.tau_radians.points();
304 let min_alpha = pm.min_spoke_alpha.points();
305 let speed_hz = pm.spin_speed_hz.points() as f64;
306 let active_step =
307 ((self.time * speed_hz * spoke_count as f64).floor() as usize) % spoke_count;
308
309 let base_color = theme.secondary_label();
310
311 for idx in 0..spoke_count {
312 let angle = (idx as f32 / spoke_count as f32) * tau;
313 let dir = Vec2::new(angle.sin(), -angle.cos());
314 let p0 = center + dir * inner_r;
315 let p1 = center + dir * outer_r;
316
317 let dist = (active_step + spoke_count - idx) % spoke_count;
318 let decay = 1.0 - (dist as f32 / spoke_count as f32);
319 let factor = min_alpha + (1.0 - min_alpha) * decay;
320 let spoke_color = base_color.gamma_multiply(factor.clamp(0.0, 1.0));
321
322 painter.line_segment([p0, p1], Stroke::new(spoke_w, spoke_color));
323 painter.circle_filled(p0, tip_r, spoke_color);
324 painter.circle_filled(p1, tip_r, spoke_color);
325 }
326
327 if self.time > 0.0 && crate::layout::is_window_active(ui.ctx()) {
328 let step_secs = (1.0 / (speed_hz * spoke_count as f64)).min(1.0) as f32;
329 ui.ctx().request_repaint_after_secs(step_secs);
330 }
331 response
332 }
333 }
334 }
335}