beet_ml 0.0.8

Machine Learning actions built upon beet_flow
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

use bevy::asset::LoadState;
use beet_core::prelude::*;

pub fn workspace_asset_plugin() -> AssetPlugin {
	AssetPlugin {
		file_path: "../../assets".into(),
		..default()
	}
}

pub fn block_on_asset_load<'a, A: Asset>(
	app: &'a mut App,
	path: &'static str,
) -> Result<Handle<A>> {
	let now = std::time::Instant::now();
	let handle = app
		.world_mut()
		.resource_mut::<AssetServer>()
		.load::<A>(path);
	loop {
		match app
			.world_mut()
			.resource_mut::<AssetServer>()
			.load_state(handle.id())
		{
			LoadState::Loaded => return Ok(handle),
			LoadState::Failed(err) => {
				bevybail!("Asset load failed {:?}\n{}", path, err);
			}
			LoadState::NotLoaded => {
				bevybail!("Asset not loaded {:?}", path);
			}
			LoadState::Loading => {
				// wait patiently 😴
			}
		}
		app.update();
		if now.elapsed().as_secs() > 10 {
			bevybail!("Timeout: block_on_asset_load");
		}
	}
}