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

use crate::prelude::*;
use beet_core::prelude::*;
use bevy::asset::AssetLoader;
use bevy::asset::LoadContext;
use bevy::asset::io::Reader;
use bevy::tasks::ConditionalSendFuture;
use serde::de::DeserializeOwned;
use std::marker::PhantomData;

#[derive(Default)]
pub struct QTableLoader<State: StateSpace, Action: ActionSpace> {
	phantom: PhantomData<(State, Action)>,
}

impl<
	State: StateSpace + DeserializeOwned,
	Action: ActionSpace + DeserializeOwned,
> AssetLoader for QTableLoader<State, Action>
{
	type Asset = QTable<State, Action>;
	type Settings = ();
	type Error = BevyError;

	fn load(
		&self,
		reader: &mut dyn Reader,
		_settings: &Self::Settings,
		_load_context: &mut LoadContext,
	) -> impl ConditionalSendFuture<Output = Result<Self::Asset, Self::Error>>
	{
		Box::pin(async move {
			let mut bytes = Vec::new();
			reader.read_to_end(&mut bytes).await?;
			let table = bevy::scene::ron::de::from_bytes::<
				QTable<State, Action>,
			>(&bytes)?;
			Ok(table)
		})
	}
}