chiron 0.1.0

Graphical and command-line interface tools for Biomechanics Foundation
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89

use super::notifications::Toast;
use super::plot::PlotData;
use super::settings::StripedTable;
use crate::ui::windows::AddWindowEvent;
use super::EguiTab;
use bevy::prelude::*;
use bevy_c3d::prelude::*;
use egui_extras::{Column, TableBuilder};

pub fn draw_marker_data_view(ui: &mut egui::Ui, world: &mut World) {
    let c3d_loaded = world.resource_scope::<C3dState, bool>(|world, c3d_state| {
        let c3d_asset = world.get_resource::<Assets<C3dAsset>>();
        if let Some(c3d_asset) = c3d_asset {
            let c3d_asset = c3d_asset.get(&c3d_state.handle);
            if let Some(_) = c3d_asset {
                return true;
            }
        }
        false
    });
    if !c3d_loaded {
        return;
    }
    world.resource_scope::<C3dState, _>(|world, c3d_state| {
        world.resource_scope::<Assets<C3dAsset>, _>(|world, c3d_asset| {
            let c3d_asset = c3d_asset.get(&c3d_state.handle);
            if let Some(c3d_asset) = c3d_asset {
                draw_marker_data(ui, world, &c3d_asset.c3d);
            }
        });
    });
}

fn draw_marker_data(ui: &mut egui::Ui, world: &mut World, c3d: &C3d) {
    world.resource_scope::<StripedTable, _>(|world, striped_table| {
        TableBuilder::new(ui)
            .striped(striped_table.0)
            .resizable(true)
            .column(Column::initial(30.0).at_least(30.0).clip(true))
            .column(Column::remainder())
            .header(20.0, |mut header| {
                header.col(|ui| {
                    ui.label("#");
                });
                header.col(|ui| {
                    ui.label("Marker Name");
                });
            })
            .body(|body| {
                body.rows(18.0, c3d.points.points.cols(), |mut row| {
                    let i = row.index();
                    row.col(|ui| {
                        ui.label(i.to_string());
                    });
                    row.col(|ui| {
                        let marker = if let Some(label) = c3d.points.labels.get(i) {
                            label
                        } else {
                            "Unknown"
                        };
                        if ui.button(marker).clicked() {
                            for (j, dimension) in ["X", "Y", "Z"].iter().enumerate() {
                                let data: Vec<[f64; 2]> = c3d
                                    .points
                                    .iter_col(i)
                                    .enumerate()
                                    .map(|(i, v)| match j {
                                        0 => [i as f64, v[0] as f64],
                                        1 => [i as f64, v[1] as f64],
                                        2 => [i as f64, v[2] as f64],
                                        _ => unreachable!(),
                                    })
                                    .collect();
                                world.send_event(AddWindowEvent {
                                    window: EguiTab::PlotView(PlotData {
                                        title: format!("{} {}", marker, dimension),
                                        data,
                                    }),
                                });
                                world.send_event(Toast::info(
                                    format!("{} {}", marker, dimension).as_str(),
                                ));
                            }
                        }
                    });
                });
            });
    });
}