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
90
91

use crate::visualizer::C3dFrame;
use bevy::prelude::*;
use bevy_c3d::*;

pub struct MarkerPlugin;

impl Plugin for MarkerPlugin {
    fn build(&self, app: &mut App) {
        app.add_systems(Update, markers)
            .add_systems(PostUpdate, add_markers);
    }
}

#[derive(Component)]
pub struct Marker;

pub fn markers(
    c3d_frame: Res<C3dFrame>,
    mut query: Query<(&mut Transform, &Marker)>,
    c3d_state: ResMut<C3dState>,
    c3d_assets: Res<Assets<C3dAsset>>,
) {
    if !c3d_frame.updated() {
        return;
    }
    if !c3d_state.loaded {
        return;
    }
    let asset = c3d_assets.get(&c3d_state.handle);
    match asset {
        Some(asset) => {
            let point_data = &asset.c3d.points.points;
            let frame = c3d_frame.frame() as usize;
            if frame >= point_data.rows() {
                return;
            }
            for (i, (mut transform, _)) in query.iter_mut().enumerate() {
                transform.translation = Vec3::new(
                    point_data[frame][i][0] as f32 / 1000.0,
                    point_data[frame][i][1] as f32 / 1000.0,
                    point_data[frame][i][2] as f32 / 1000.0,
                );
            }
        }
        None => {}
    }
}

pub fn add_markers(
    mut events: EventReader<C3dLoadedEvent>,
    c3d_state: ResMut<C3dState>,
    c3d_assets: Res<Assets<C3dAsset>>,
    mut commands: Commands,
    mut meshes: ResMut<Assets<Mesh>>,
    mut materials: ResMut<Assets<StandardMaterial>>,
    markers: Query<Entity, With<Marker>>,
) {
    if let Some(_) = events.read().last() {
        let asset = c3d_assets.get(&c3d_state.handle);
        if asset.is_none() {
            return;
        }
        let asset = asset.unwrap();
        for marker in markers.iter() {
            commands.entity(marker).despawn();
        }
        for i in 0..asset.c3d.points.labels.len() {
            let matrix = Mat4::from_scale_rotation_translation(
                Vec3::new(1.0, 1.0, 1.0),
                Quat::from_rotation_y(0.0),
                Vec3::new(
                    asset.c3d.points.points[0][i][0] as f32 / 1000.0,
                    asset.c3d.points.points[0][i][1] as f32 / 1000.0,
                    asset.c3d.points.points[0][i][2] as f32 / 1000.0,
                ),
            );
            commands.spawn((
                PbrBundle {
                    mesh: meshes.add(Sphere::new(0.01).mesh()),
                    material: materials.add(StandardMaterial {
                        base_color: Color::rgb_u8(255, 0, 255),
                        ..default()
                    }),
                    transform: Transform::from_matrix(matrix),
                    ..default()
                },
                Marker,
            ));
        }
    }
}