bevy_animation_graph_editor 0.2.0

Animation graph editor for the Bevy game engine
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

mod egui_inspector_impls;
mod egui_nodes;
mod graph_saving;
mod graph_show;
mod graph_update;
mod ui;

use bevy::{asset::LoadedFolder, prelude::*, utils::HashSet};
use bevy_animation_graph::core::{animation_graph::AnimationGraph, plugin::AnimationGraphPlugin};
use bevy_egui::EguiPlugin;
use bevy_inspector_egui::{bevy_egui, DefaultInspectorConfigPlugin};
use clap::Parser;
use egui_inspector_impls::BetterInspectorPlugin;
use graph_saving::{save_graph_system, SaveGraph};
use std::path::PathBuf;
use ui::UiState;

#[derive(Parser, Resource)]
struct Cli {
    #[arg(short, long)]
    asset_source: PathBuf,
}

/// Keeps a handle to the folder so that it does not get unloaded
#[derive(Resource)]
struct GraphHandles {
    #[allow(dead_code)]
    folder: Handle<LoadedFolder>,
    unsaved: HashSet<Handle<AnimationGraph>>,
}

fn main() {
    let cli = Cli::parse();

    let mut app = App::new();

    app //
        .add_event::<SaveGraph>()
        .add_plugins(
            DefaultPlugins.set(AssetPlugin {
                file_path: std::fs::canonicalize(&cli.asset_source)
                    .unwrap()
                    .to_string_lossy()
                    .into(),
                ..Default::default()
            }),
        )
        .add_plugins(EguiPlugin)
        .add_plugins(AnimationGraphPlugin)
        .add_plugins(DefaultInspectorConfigPlugin)
        .add_plugins(BetterInspectorPlugin)
        .insert_resource(UiState::new())
        .insert_resource(cli)
        .add_systems(Startup, (load_target_graph, ui::setup_system))
        .add_systems(
            Update,
            (
                ui::show_ui_system,
                ui::graph_save_event_system,
                ui::scene_spawner_system,
                save_graph_system,
            )
                .chain(),
        );

    app.run();
}

fn load_target_graph(mut commands: Commands, asset_server: Res<AssetServer>) {
    commands.insert_resource(GraphHandles {
        folder: asset_server.load_folder(""),
        unsaved: HashSet::default(),
    });
}