1use egui::{Align2, Color32, FontId, Pos2, Rect, Sense, Stroke, Ui, vec2};
8
9pub mod a11y;
10pub mod caps;
11pub mod clip;
12pub mod clipboard;
13pub mod deckfx;
14pub mod edges;
15pub mod effects;
16pub mod focus;
17pub mod labels3d;
18pub mod harness;
19pub mod look;
20pub mod nav;
21pub mod overlay;
22pub mod rabbit;
23pub mod scroll_engine;
24pub mod testmatrix; pub mod theme;
27pub mod trace; pub use a11y::{Semantics, node as a11y_node, stable_id};
30pub use caps::FacetCaps;
31pub use clip::{ArrowColumnRef, ClipKind, ClipPayload, CopySource, PasteTarget};
32pub use clipboard::ClipAction;
33pub use deckfx::{DeckFx, DeckRaven};
34pub use look::{Action, KeyMap, Palette};
35pub use nav::{Dir4, Navigable, nearest_in_direction};
36pub use rabbit::{Rabbit, RabbitMesh, rabbit_mesh, rabbit_outline};
37pub use scroll_engine::SmoothScroll;
38pub use theme::{Theme, set_theme, theme};
39
40pub use look::Theme as LookTheme;
43
44#[derive(Clone)]
47pub struct Node {
48 pub label: String,
49 pub color: Color32,
50}
51
52#[derive(Clone, Copy)]
54pub struct Edge {
55 pub src: usize,
56 pub dst: usize,
57}
58
59#[derive(Default, Clone)]
61pub struct Scene {
62 pub nodes: Vec<Node>,
63 pub edges: Vec<Edge>,
64}
65
66impl Scene {
67 pub fn new() -> Self {
68 Self::default()
69 }
70 pub fn node(&mut self, label: impl Into<String>, color: Color32) -> usize {
72 self.nodes.push(Node { label: label.into(), color });
73 self.nodes.len() - 1
74 }
75 pub fn edge(&mut self, src: usize, dst: usize) {
76 self.edges.push(Edge { src, dst });
77 }
78 pub fn is_empty(&self) -> bool {
79 self.nodes.is_empty()
80 }
81}
82
83#[derive(Clone, Copy, PartialEq, Eq, Default)]
85pub enum Layout {
86 #[default]
87 Circular,
88 Force,
91}
92
93pub fn draw(ui: &mut Ui, scene: &Scene, layout: Layout, empty_hint: &str) {
96 let (rect, _) = ui.allocate_exact_size(ui.available_size(), Sense::hover());
97 let th = theme(ui);
98 let painter = ui.painter_at(rect);
99 let n = scene.nodes.len();
100 if n == 0 {
101 painter.text(rect.center(), Align2::CENTER_CENTER, empty_hint, FontId::proportional(13.0), th.text_dim);
102 return;
103 }
104 let pos = positions(layout, scene, rect);
105 for e in &scene.edges {
106 if e.src < n && e.dst < n {
107 painter.line_segment([pos[e.src], pos[e.dst]], Stroke::new(0.6, th.edge));
108 }
109 }
110 for (i, node) in scene.nodes.iter().enumerate() {
111 painter.circle_filled(pos[i], 5.0, node.color);
112 }
113 if n <= 60 {
114 for (i, node) in scene.nodes.iter().enumerate() {
115 painter.text(pos[i] + vec2(7.0, 0.0), Align2::LEFT_CENTER, &node.label, FontId::proportional(10.0), th.text);
116 }
117 }
118}
119
120fn positions(layout: Layout, scene: &Scene, rect: Rect) -> Vec<Pos2> {
121 let n = scene.nodes.len();
122 let center = rect.center();
123 let radius = rect.size().min_elem() * 0.42;
124 let circular = |i: usize| {
125 let a = std::f32::consts::TAU * (i as f32) / (n as f32);
126 vec2(a.cos(), a.sin())
127 };
128 match layout {
129 Layout::Circular => (0..n).map(|i| center + radius * circular(i)).collect(),
130 Layout::Force => {
131 let mut p: Vec<egui::Vec2> = (0..n).map(circular).collect();
133 let k = (1.0 / (n.max(1) as f32).sqrt()).clamp(0.05, 1.0);
134 for _ in 0..120 {
135 let mut disp = vec![egui::Vec2::ZERO; n];
136 for i in 0..n {
137 for j in (i + 1)..n {
138 let d = p[i] - p[j];
139 let dist = d.length().max(1e-3);
140 let f = k * k / dist;
141 let dir = d / dist;
142 disp[i] += dir * f;
143 disp[j] -= dir * f;
144 }
145 }
146 for e in &scene.edges {
147 if e.src < n && e.dst < n {
148 let d = p[e.src] - p[e.dst];
149 let dist = d.length().max(1e-3);
150 let f = dist * dist / k;
151 let dir = d / dist;
152 disp[e.src] -= dir * f;
153 disp[e.dst] += dir * f;
154 }
155 }
156 for i in 0..n {
157 let dl = disp[i].length().max(1e-3);
158 p[i] += disp[i] / dl * dl.min(0.04); }
160 }
161 let (mut mn, mut mx) = (egui::vec2(f32::MAX, f32::MAX), egui::vec2(f32::MIN, f32::MIN));
163 for v in &p {
164 mn.x = mn.x.min(v.x);
165 mn.y = mn.y.min(v.y);
166 mx.x = mx.x.max(v.x);
167 mx.y = mx.y.max(v.y);
168 }
169 let span = (mx - mn).max(egui::vec2(1e-3, 1e-3));
170 p.iter()
171 .map(|v| center + egui::vec2(((v.x - mn.x) / span.x - 0.5) * 2.0 * radius, ((v.y - mn.y) / span.y - 0.5) * 2.0 * radius))
172 .collect()
173 }
174 }
175}
176
177pub trait Facet {
187 fn title(&self) -> &str;
188 fn ui(&mut self, ui: &mut Ui);
189 fn state_json(&self) -> serde_json::Value;
190
191 fn caps(&self) -> FacetCaps {
195 FacetCaps::NONE
196 }
197
198 fn scale(&self) -> f32 {
200 1.0
201 }
202 fn set_scale(&mut self, _scale: f32) {}
204
205 fn selection_json(&self) -> serde_json::Value {
208 serde_json::Value::Null
209 }
210
211 fn copy(&mut self) -> Option<String> {
214 None
215 }
216 fn cut(&mut self) -> Option<String> {
218 self.copy()
219 }
220 fn paste(&mut self, _text: &str) -> bool {
222 false
223 }
224
225 fn as_any_mut(&mut self) -> Option<&mut dyn std::any::Any> {
231 None
232 }
233}
234
235pub struct FacetDeck {
240 facets: Vec<Box<dyn Facet>>,
241 active: usize,
242 fx: DeckFx,
245 raven: Option<DeckRaven>,
247}
248
249impl FacetDeck {
250 pub fn new(facets: Vec<Box<dyn Facet>>) -> Self {
251 Self { facets, active: 0, fx: DeckFx::OFF, raven: None }
252 }
253 pub fn active(&self) -> usize {
254 self.active
255 }
256
257 pub fn active_title(&self) -> Option<&str> {
260 self.facets.get(self.active).map(|f| f.title())
261 }
262
263 pub fn titles(&self) -> Vec<&str> {
266 self.facets.iter().map(|f| f.title()).collect()
267 }
268
269 pub fn set_active_by_title(&mut self, title: &str) -> bool {
275 match self.facets.iter().position(|f| f.title() == title) {
276 Some(i) => {
277 self.active = i;
278 true
279 }
280 None => false,
281 }
282 }
283
284 pub fn facet_mut<T: std::any::Any>(&mut self, title: &str) -> Option<&mut T> {
289 self.facets
290 .iter_mut()
291 .find(|f| f.title() == title)
292 .and_then(|f| f.as_any_mut())
293 .and_then(|a| a.downcast_mut::<T>())
294 }
295
296 pub fn replace_facet(&mut self, title: &str, facet: Box<dyn Facet>) -> bool {
302 if let Some(slot) = self.facets.iter_mut().find(|f| f.title() == title) {
303 *slot = facet;
304 true
305 } else {
306 false
307 }
308 }
309
310 pub fn with_fx(mut self, fx: DeckFx) -> Self {
314 self.fx = fx;
315 self
316 }
317 pub fn fx(&self) -> &DeckFx {
319 &self.fx
320 }
321 pub fn fx_mut(&mut self) -> &mut DeckFx {
323 &mut self.fx
324 }
325 pub fn set_palette(&mut self, i: usize) {
328 self.fx.set_palette(i);
329 }
330 pub fn cycle_palette(&mut self) -> usize {
333 self.fx.cycle_palette()
334 }
335
336 pub fn send_raven(&mut self, target: Rect) {
342 let theme = self.effective_theme();
343 self.raven = Some(DeckRaven::new(target, &theme));
344 harness::trail(
345 harness::Kind::Render,
346 format!("raven launched → perch ({:.0},{:.0})", target.center().x, target.top()),
347 );
348 }
349 pub fn has_raven(&self) -> bool {
351 self.raven.is_some()
352 }
353 pub fn raven_perched(&self) -> bool {
355 self.raven.as_ref().map(|r| r.is_perched()).unwrap_or(false)
356 }
357 pub fn clear_raven(&mut self) {
359 self.raven = None;
360 }
361
362 fn effective_theme(&self) -> Theme {
366 self.fx.theme().unwrap_or_default()
367 }
368
369 pub fn palette_picker(&mut self, ui: &mut Ui) -> Option<usize> {
381 let mut sel = self.fx.palette().unwrap_or(0);
382 let before = sel;
383 ui.horizontal_wrapped(|ui| {
384 ui.label("Palette:");
385 for (i, ctor) in Theme::ALL.iter().enumerate() {
386 ui.selectable_value(&mut sel, i, ctor().name);
387 }
388 });
389 if sel != before {
390 self.fx.set_palette(sel);
391 }
392 (sel != before).then_some(sel)
393 }
394
395 pub fn active_caps(&self) -> FacetCaps {
397 self.facets.get(self.active).map(|f| f.caps()).unwrap_or(FacetCaps::NONE)
398 }
399
400 fn active_scale(&self) -> f32 {
402 self.facets.get(self.active).map(|f| f.scale()).unwrap_or(1.0)
403 }
404
405 fn scale_active(&mut self, k: f32) {
407 if let Some(f) = self.facets.get_mut(self.active) {
408 let s = (f.scale() * k).clamp(0.25, 4.0);
409 f.set_scale(s);
410 }
411 }
412
413 fn reset_scale(&mut self) {
415 if let Some(f) = self.facets.get_mut(self.active) {
416 f.set_scale(1.0);
417 }
418 }
419
420 pub fn ui(&mut self, ui: &mut Ui) {
423 if let Some(theme) = self.fx.theme() {
427 set_theme(ui.ctx(), theme);
428 }
429
430 let titles: Vec<String> = self.facets.iter().map(|f| f.title().to_string()).collect();
431 ui.horizontal_wrapped(|ui| {
436 for (i, t) in titles.iter().enumerate() {
437 ui.selectable_value(&mut self.active, i, t);
438 }
439 });
440
441 let caps = self.active_caps();
442
443 if caps.scalable {
445 ui.horizontal(|ui| {
446 if ui.button("−").on_hover_text("Zoom out (Ctrl-−)").clicked() {
447 self.scale_active(1.0 / 1.1);
448 }
449 ui.label(format!("{:.0}%", self.active_scale() * 100.0));
450 if ui.button("+").on_hover_text("Zoom in (Ctrl-+)").clicked() {
451 self.scale_active(1.1);
452 }
453 if ui.button("Reset").on_hover_text("Reset zoom (Ctrl-0)").clicked() {
454 self.reset_scale();
455 }
456 });
457 }
458
459 if caps.scalable {
462 let (cmd, plus, minus, zero) = ui.input(|i| {
463 (
464 i.modifiers.command,
465 i.key_pressed(egui::Key::Plus) || i.key_pressed(egui::Key::Equals),
466 i.key_pressed(egui::Key::Minus),
467 i.key_pressed(egui::Key::Num0),
468 )
469 });
470 if cmd {
471 if plus {
472 self.scale_active(1.1);
473 }
474 if minus {
475 self.scale_active(1.0 / 1.1);
476 }
477 if zero {
478 self.reset_scale();
479 }
480 }
481 }
482
483 self.route_clipboard(ui.ctx());
486
487 ui.separator();
488 let content = ui.scope(|ui| {
491 if let Some(f) = self.facets.get_mut(self.active) {
492 f.ui(ui);
493 }
494 });
495 let content_rect = content.response.rect;
496
497 if self.fx.glow && content_rect.is_positive() {
499 let theme = self.effective_theme();
500 let time = ui.input(|i| i.time);
501 let painter = ui.painter_at(content_rect);
502 deckfx::paint_active_glow(&painter, content_rect.shrink(2.0), &theme, &self.fx, time);
503 ui.ctx().request_repaint(); }
505
506 self.drive_raven(ui.ctx());
508 }
509
510 fn drive_raven(&mut self, ctx: &egui::Context) {
513 let Some(raven) = self.raven.as_mut() else { return };
514 raven.sprite.update(ctx);
515 let painter =
516 ctx.layer_painter(egui::LayerId::new(egui::Order::Foreground, egui::Id::new("facett_deck_raven")));
517 raven.sprite.paint(&painter);
518 }
519
520 fn route_clipboard(&mut self, ctx: &egui::Context) {
523 let caps = self.active_caps();
524 if !(caps.copyable || caps.cuttable || caps.pasteable) {
525 return;
526 }
527 for action in clipboard::poll(ctx) {
528 let Some(f) = self.facets.get_mut(self.active) else { continue };
529 match action {
530 ClipAction::Copy if caps.copyable => {
531 if let Some(t) = f.copy() {
532 clipboard::put(ctx, t);
533 }
534 }
535 ClipAction::Cut if caps.cuttable => {
536 if let Some(t) = f.cut() {
537 clipboard::put(ctx, t);
538 }
539 }
540 ClipAction::Paste(s) if caps.pasteable => {
541 f.paste(&s);
542 }
543 _ => {}
546 }
547 }
548 }
549
550 pub fn state_json(&self) -> serde_json::Value {
554 let mut facets = serde_json::Map::new();
555 let mut caps = serde_json::Map::new();
556 for f in &self.facets {
557 facets.insert(f.title().to_string(), f.state_json());
558 caps.insert(f.title().to_string(), f.caps().to_json());
559 }
560 serde_json::json!({
561 "active": self.facets.get(self.active).map(|f| f.title()),
562 "facets": facets,
563 "caps": caps,
564 })
565 }
566}
567
568pub fn hash_color(s: &str) -> Color32 {
570 let mut h: u32 = 2166136261;
571 for b in s.bytes() {
572 h = (h ^ b as u32).wrapping_mul(16777619);
573 }
574 Color32::from_rgb((h & 0xFF) as u8 | 0x60, ((h >> 8) & 0xFF) as u8 | 0x60, ((h >> 16) & 0xFF) as u8 | 0x60)
575}
576
577#[cfg(test)]
578mod tests {
579 use super::*;
580
581 #[test]
582 fn scene_builds() {
583 let mut s = Scene::new();
584 let a = s.node("Person", hash_color("Person"));
585 let b = s.node("Company", hash_color("Company"));
586 s.edge(a, b);
587 assert_eq!(s.nodes.len(), 2);
588 assert_eq!(s.edges.len(), 1);
589 assert!(!s.is_empty());
590 }
591
592 #[test]
593 fn force_layout_produces_finite_bounded_positions() {
594 let mut scene = Scene::new();
595 for i in 0..12 { scene.node(format!("n{i}"), hash_color("n")); }
596 for i in 0..12 { scene.edge(i, (i + 1) % 12); }
597 let rect = egui::Rect::from_min_size(egui::pos2(0.0, 0.0), egui::vec2(400.0, 400.0));
598 let pos = positions(Layout::Force, &scene, rect);
599 assert_eq!(pos.len(), 12);
600 for p in &pos {
601 assert!(p.x.is_finite() && p.y.is_finite(), "finite");
602 assert!(rect.expand(50.0).contains(*p), "roughly within the rect");
603 }
604 }
605
606 #[test]
607 fn hash_color_is_stable() {
608 assert_eq!(hash_color("Person"), hash_color("Person"));
609 assert_ne!(hash_color("Person"), hash_color("Company"));
610 }
611
612 struct Stub(&'static str);
614 impl Facet for Stub {
615 fn title(&self) -> &str {
616 self.0
617 }
618 fn ui(&mut self, ui: &mut Ui) {
619 ui.label(self.0);
620 }
621 fn state_json(&self) -> serde_json::Value {
622 serde_json::json!({ "t": self.0 })
623 }
624 }
625
626 #[test]
627 fn deck_fx_is_off_by_default() {
628 let deck = FacetDeck::new(vec![Box::new(Stub("a"))]);
629 assert_eq!(*deck.fx(), DeckFx::OFF, "no effects until the host opts in");
630 assert!(!deck.has_raven());
631 assert!(!deck.fx().glow);
632 assert!(deck.fx().palette().is_none());
633 }
634
635 #[test]
636 fn deck_cycle_palette_walks_theme_all() {
637 let mut deck = FacetDeck::new(vec![Box::new(Stub("a"))]);
638 let first = deck.cycle_palette();
639 assert_eq!(first, 0);
640 assert_eq!(deck.fx().theme().map(|t| t.name), Some(Theme::ALL[0]().name));
641 for _ in 1..Theme::ALL.len() {
643 deck.cycle_palette();
644 }
645 assert_eq!(deck.cycle_palette(), 0, "wraps back to the first palette");
646 }
647
648 #[test]
649 fn deck_send_raven_launches_and_perches_after_a_full_flight() {
650 use crate::effects::RAVEN_FLIGHT_SECS;
651 let mut deck = FacetDeck::new(vec![Box::new(Stub("rows"))]);
652 assert!(!deck.has_raven());
653 let target = egui::Rect::from_min_size(egui::pos2(120.0, 80.0), egui::vec2(200.0, 28.0));
654 deck.send_raven(target);
655 assert!(deck.has_raven(), "raven summoned");
656 assert!(!deck.raven_perched(), "not perched at launch");
657
658 if let Some(r) = deck.raven.as_mut() {
660 r.sprite.advance(RAVEN_FLIGHT_SECS + 0.1);
661 }
662 assert!(deck.raven_perched(), "perched after the flight duration");
663
664 deck.clear_raven();
665 assert!(!deck.has_raven());
666 }
667
668 #[test]
675 fn palette_picker_does_not_pin_without_a_user_click() {
676 let mut deck = FacetDeck::new(vec![Box::new(Stub("a"))]);
677 assert!(deck.fx().palette().is_none(), "starts with no override");
678 let ctx = egui::Context::default();
679 let mut chosen = Some(7usize);
680 let _ = ctx.run(egui::RawInput::default(), |ctx| {
681 egui::CentralPanel::default().show(ctx, |ui| {
682 chosen = deck.palette_picker(ui);
684 });
685 });
686 assert_eq!(chosen, None, "drawing the picker reports no selection without a click");
687 assert!(
688 deck.fx().palette().is_none(),
689 "drawing the picker must not pin index 0 — that would clobber the host's own theme each frame"
690 );
691 }
692
693 #[test]
694 fn deck_palette_override_applies_theme_in_a_ui_pass() {
695 let mut deck = FacetDeck::new(vec![Box::new(Stub("a"))]);
696 deck.set_palette(1); let ctx = egui::Context::default();
698 let mut seen = "";
699 let _ = ctx.run(egui::RawInput::default(), |ctx| {
700 egui::CentralPanel::default().show(ctx, |ui| {
701 deck.ui(ui);
702 seen = theme(ui).name;
703 });
704 });
705 assert_eq!(seen, Theme::ALL[1]().name, "deck applied its palette override");
706 }
707}