
# Objective - Some imports are only used with certain features. ## Solution - Gate them! ## Testing - CI
76 lines
2.1 KiB
Rust
76 lines
2.1 KiB
Rust
use crate::ron;
|
|
use bevy_ecs::{
|
|
reflect::AppTypeRegistry,
|
|
world::{FromWorld, World},
|
|
};
|
|
use bevy_reflect::TypeRegistryArc;
|
|
use thiserror::Error;
|
|
|
|
#[cfg(feature = "serialize")]
|
|
use {
|
|
crate::{serde::SceneDeserializer, DynamicScene},
|
|
bevy_asset::{io::Reader, AssetLoader, LoadContext},
|
|
serde::de::DeserializeSeed,
|
|
};
|
|
|
|
/// Asset loader for a Bevy dynamic scene (`.scn` / `.scn.ron`).
|
|
///
|
|
/// The loader handles assets serialized with [`DynamicScene::serialize`].
|
|
#[derive(Debug)]
|
|
pub struct SceneLoader {
|
|
#[cfg_attr(
|
|
not(feature = "serialize"),
|
|
expect(dead_code, reason = "only used with `serialize` feature")
|
|
)]
|
|
type_registry: TypeRegistryArc,
|
|
}
|
|
|
|
impl FromWorld for SceneLoader {
|
|
fn from_world(world: &mut World) -> Self {
|
|
let type_registry = world.resource::<AppTypeRegistry>();
|
|
SceneLoader {
|
|
type_registry: type_registry.0.clone(),
|
|
}
|
|
}
|
|
}
|
|
|
|
/// Possible errors that can be produced by [`SceneLoader`]
|
|
#[non_exhaustive]
|
|
#[derive(Debug, Error)]
|
|
pub enum SceneLoaderError {
|
|
/// An [IO Error](std::io::Error)
|
|
#[error("Error while trying to read the scene file: {0}")]
|
|
Io(#[from] std::io::Error),
|
|
/// A [RON Error](ron::error::SpannedError)
|
|
#[error("Could not parse RON: {0}")]
|
|
RonSpannedError(#[from] ron::error::SpannedError),
|
|
}
|
|
|
|
#[cfg(feature = "serialize")]
|
|
impl AssetLoader for SceneLoader {
|
|
type Asset = DynamicScene;
|
|
type Settings = ();
|
|
type Error = SceneLoaderError;
|
|
|
|
async fn load(
|
|
&self,
|
|
reader: &mut dyn Reader,
|
|
_settings: &(),
|
|
_load_context: &mut LoadContext<'_>,
|
|
) -> Result<Self::Asset, Self::Error> {
|
|
let mut bytes = Vec::new();
|
|
reader.read_to_end(&mut bytes).await?;
|
|
let mut deserializer = ron::de::Deserializer::from_bytes(&bytes)?;
|
|
let scene_deserializer = SceneDeserializer {
|
|
type_registry: &self.type_registry.read(),
|
|
};
|
|
Ok(scene_deserializer
|
|
.deserialize(&mut deserializer)
|
|
.map_err(|e| deserializer.span_error(e))?)
|
|
}
|
|
|
|
fn extensions(&self) -> &[&str] {
|
|
&["scn", "scn.ron"]
|
|
}
|
|
}
|