
# Objective Fixes #15417. ## Solution - Remove the `labeled_assets` fields from `LoadedAsset` and `ErasedLoadedAsset`. - Created new structs `CompleteLoadedAsset` and `CompleteErasedLoadedAsset` to hold the `labeled_subassets`. - When a subasset is `LoadContext::finish`ed, it produces a `CompleteLoadedAsset`. - When a `CompleteLoadedAsset` is added to a `LoadContext` (as a subasset), their `labeled_assets` are merged, reporting any overlaps. One important detail to note: nested subassets with overlapping names could in theory have been used in the past for the purposes of asset preprocessing. Even though there was no way to access these "shadowed" nested subassets, asset preprocessing does get access to these nested subassets. This does not seem like a case we should support though. It is confusing at best. ## Testing - This is just a refactor. --- ## Migration Guide - Most uses of `LoadedAsset` and `ErasedLoadedAsset` should be replaced with `CompleteLoadedAsset` and `CompleteErasedLoadedAsset` respectively.
664 lines
25 KiB
Rust
664 lines
25 KiB
Rust
use crate::{
|
|
io::{AssetReaderError, MissingAssetSourceError, MissingProcessedAssetReaderError, Reader},
|
|
loader_builders::{Deferred, NestedLoader, StaticTyped},
|
|
meta::{AssetHash, AssetMeta, AssetMetaDyn, ProcessedInfoMinimal, Settings},
|
|
path::AssetPath,
|
|
Asset, AssetLoadError, AssetServer, AssetServerMode, Assets, Handle, UntypedAssetId,
|
|
UntypedHandle,
|
|
};
|
|
use alloc::{
|
|
boxed::Box,
|
|
string::{String, ToString},
|
|
vec::Vec,
|
|
};
|
|
use atomicow::CowArc;
|
|
use bevy_ecs::world::World;
|
|
use bevy_log::warn;
|
|
use bevy_platform_support::collections::{HashMap, HashSet};
|
|
use bevy_tasks::{BoxedFuture, ConditionalSendFuture};
|
|
use core::any::{Any, TypeId};
|
|
use downcast_rs::{impl_downcast, Downcast};
|
|
use ron::error::SpannedError;
|
|
use serde::{Deserialize, Serialize};
|
|
use std::path::{Path, PathBuf};
|
|
use thiserror::Error;
|
|
|
|
/// Loads an [`Asset`] from a given byte [`Reader`]. This can accept [`AssetLoader::Settings`], which configure how the [`Asset`]
|
|
/// should be loaded.
|
|
///
|
|
/// This trait is generally used in concert with [`AssetReader`](crate::io::AssetReader) to load assets from a byte source.
|
|
///
|
|
/// For a complementary version of this trait that can save assets, see [`AssetSaver`](crate::saver::AssetSaver).
|
|
pub trait AssetLoader: Send + Sync + 'static {
|
|
/// The top level [`Asset`] loaded by this [`AssetLoader`].
|
|
type Asset: Asset;
|
|
/// The settings type used by this [`AssetLoader`].
|
|
type Settings: Settings + Default + Serialize + for<'a> Deserialize<'a>;
|
|
/// The type of [error](`std::error::Error`) which could be encountered by this loader.
|
|
type Error: Into<Box<dyn core::error::Error + Send + Sync + 'static>>;
|
|
/// Asynchronously loads [`AssetLoader::Asset`] (and any other labeled assets) from the bytes provided by [`Reader`].
|
|
fn load(
|
|
&self,
|
|
reader: &mut dyn Reader,
|
|
settings: &Self::Settings,
|
|
load_context: &mut LoadContext,
|
|
) -> impl ConditionalSendFuture<Output = Result<Self::Asset, Self::Error>>;
|
|
|
|
/// Returns a list of extensions supported by this [`AssetLoader`], without the preceding dot.
|
|
/// Note that users of this [`AssetLoader`] may choose to load files with a non-matching extension.
|
|
fn extensions(&self) -> &[&str] {
|
|
&[]
|
|
}
|
|
}
|
|
|
|
/// Provides type-erased access to an [`AssetLoader`].
|
|
pub trait ErasedAssetLoader: Send + Sync + 'static {
|
|
/// Asynchronously loads the asset(s) from the bytes provided by [`Reader`].
|
|
fn load<'a>(
|
|
&'a self,
|
|
reader: &'a mut dyn Reader,
|
|
meta: &'a dyn AssetMetaDyn,
|
|
load_context: LoadContext<'a>,
|
|
) -> BoxedFuture<
|
|
'a,
|
|
Result<CompleteErasedLoadedAsset, Box<dyn core::error::Error + Send + Sync + 'static>>,
|
|
>;
|
|
|
|
/// Returns a list of extensions supported by this asset loader, without the preceding dot.
|
|
fn extensions(&self) -> &[&str];
|
|
/// Deserializes metadata from the input `meta` bytes into the appropriate type (erased as [`Box<dyn AssetMetaDyn>`]).
|
|
fn deserialize_meta(&self, meta: &[u8]) -> Result<Box<dyn AssetMetaDyn>, DeserializeMetaError>;
|
|
/// Returns the default meta value for the [`AssetLoader`] (erased as [`Box<dyn AssetMetaDyn>`]).
|
|
fn default_meta(&self) -> Box<dyn AssetMetaDyn>;
|
|
/// Returns the type name of the [`AssetLoader`].
|
|
fn type_name(&self) -> &'static str;
|
|
/// Returns the [`TypeId`] of the [`AssetLoader`].
|
|
fn type_id(&self) -> TypeId;
|
|
/// Returns the type name of the top-level [`Asset`] loaded by the [`AssetLoader`].
|
|
fn asset_type_name(&self) -> &'static str;
|
|
/// Returns the [`TypeId`] of the top-level [`Asset`] loaded by the [`AssetLoader`].
|
|
fn asset_type_id(&self) -> TypeId;
|
|
}
|
|
|
|
impl<L> ErasedAssetLoader for L
|
|
where
|
|
L: AssetLoader + Send + Sync,
|
|
{
|
|
/// Processes the asset in an asynchronous closure.
|
|
fn load<'a>(
|
|
&'a self,
|
|
reader: &'a mut dyn Reader,
|
|
meta: &'a dyn AssetMetaDyn,
|
|
mut load_context: LoadContext<'a>,
|
|
) -> BoxedFuture<
|
|
'a,
|
|
Result<CompleteErasedLoadedAsset, Box<dyn core::error::Error + Send + Sync + 'static>>,
|
|
> {
|
|
Box::pin(async move {
|
|
let settings = meta
|
|
.loader_settings()
|
|
.expect("Loader settings should exist")
|
|
.downcast_ref::<L::Settings>()
|
|
.expect("AssetLoader settings should match the loader type");
|
|
let asset = <L as AssetLoader>::load(self, reader, settings, &mut load_context)
|
|
.await
|
|
.map_err(Into::into)?;
|
|
Ok(load_context.finish(asset).into())
|
|
})
|
|
}
|
|
|
|
fn extensions(&self) -> &[&str] {
|
|
<L as AssetLoader>::extensions(self)
|
|
}
|
|
|
|
fn deserialize_meta(&self, meta: &[u8]) -> Result<Box<dyn AssetMetaDyn>, DeserializeMetaError> {
|
|
let meta = AssetMeta::<L, ()>::deserialize(meta)?;
|
|
Ok(Box::new(meta))
|
|
}
|
|
|
|
fn default_meta(&self) -> Box<dyn AssetMetaDyn> {
|
|
Box::new(AssetMeta::<L, ()>::new(crate::meta::AssetAction::Load {
|
|
loader: self.type_name().to_string(),
|
|
settings: L::Settings::default(),
|
|
}))
|
|
}
|
|
|
|
fn type_name(&self) -> &'static str {
|
|
core::any::type_name::<L>()
|
|
}
|
|
|
|
fn type_id(&self) -> TypeId {
|
|
TypeId::of::<L>()
|
|
}
|
|
|
|
fn asset_type_name(&self) -> &'static str {
|
|
core::any::type_name::<L::Asset>()
|
|
}
|
|
|
|
fn asset_type_id(&self) -> TypeId {
|
|
TypeId::of::<L::Asset>()
|
|
}
|
|
}
|
|
|
|
pub(crate) struct LabeledAsset {
|
|
pub(crate) asset: ErasedLoadedAsset,
|
|
pub(crate) handle: UntypedHandle,
|
|
}
|
|
|
|
/// The successful result of an [`AssetLoader::load`] call. This contains the loaded "root" asset and any other "labeled" assets produced
|
|
/// by the loader. It also holds the input [`AssetMeta`] (if it exists) and tracks dependencies:
|
|
/// * normal dependencies: dependencies that must be loaded as part of this asset load (ex: assets a given asset has handles to).
|
|
/// * Loader dependencies: dependencies whose actual asset values are used during the load process
|
|
pub struct LoadedAsset<A: Asset> {
|
|
pub(crate) value: A,
|
|
pub(crate) dependencies: HashSet<UntypedAssetId>,
|
|
pub(crate) loader_dependencies: HashMap<AssetPath<'static>, AssetHash>,
|
|
}
|
|
|
|
impl<A: Asset> LoadedAsset<A> {
|
|
/// Create a new loaded asset. This will use [`VisitAssetDependencies`](crate::VisitAssetDependencies) to populate `dependencies`.
|
|
pub fn new_with_dependencies(value: A) -> Self {
|
|
let mut dependencies = <HashSet<_>>::default();
|
|
value.visit_dependencies(&mut |id| {
|
|
dependencies.insert(id);
|
|
});
|
|
LoadedAsset {
|
|
value,
|
|
dependencies,
|
|
loader_dependencies: HashMap::default(),
|
|
}
|
|
}
|
|
|
|
/// Cast (and take ownership) of the [`Asset`] value of the given type.
|
|
pub fn take(self) -> A {
|
|
self.value
|
|
}
|
|
|
|
/// Retrieves a reference to the internal [`Asset`] type.
|
|
pub fn get(&self) -> &A {
|
|
&self.value
|
|
}
|
|
}
|
|
|
|
impl<A: Asset> From<A> for LoadedAsset<A> {
|
|
fn from(asset: A) -> Self {
|
|
LoadedAsset::new_with_dependencies(asset)
|
|
}
|
|
}
|
|
|
|
/// A "type erased / boxed" counterpart to [`LoadedAsset`]. This is used in places where the loaded type is not statically known.
|
|
pub struct ErasedLoadedAsset {
|
|
pub(crate) value: Box<dyn AssetContainer>,
|
|
pub(crate) dependencies: HashSet<UntypedAssetId>,
|
|
pub(crate) loader_dependencies: HashMap<AssetPath<'static>, AssetHash>,
|
|
}
|
|
|
|
impl<A: Asset> From<LoadedAsset<A>> for ErasedLoadedAsset {
|
|
fn from(asset: LoadedAsset<A>) -> Self {
|
|
ErasedLoadedAsset {
|
|
value: Box::new(asset.value),
|
|
dependencies: asset.dependencies,
|
|
loader_dependencies: asset.loader_dependencies,
|
|
}
|
|
}
|
|
}
|
|
|
|
impl ErasedLoadedAsset {
|
|
/// Cast (and take ownership) of the [`Asset`] value of the given type. This will return [`Some`] if
|
|
/// the stored type matches `A` and [`None`] if it does not.
|
|
pub fn take<A: Asset>(self) -> Option<A> {
|
|
self.value.downcast::<A>().map(|a| *a).ok()
|
|
}
|
|
|
|
/// Retrieves a reference to the internal [`Asset`] type, if it matches the type `A`. Otherwise returns [`None`].
|
|
pub fn get<A: Asset>(&self) -> Option<&A> {
|
|
self.value.downcast_ref::<A>()
|
|
}
|
|
|
|
/// Retrieves the [`TypeId`] of the stored [`Asset`] type.
|
|
pub fn asset_type_id(&self) -> TypeId {
|
|
(*self.value).type_id()
|
|
}
|
|
|
|
/// Retrieves the `type_name` of the stored [`Asset`] type.
|
|
pub fn asset_type_name(&self) -> &'static str {
|
|
self.value.asset_type_name()
|
|
}
|
|
|
|
/// Cast this loaded asset as the given type. If the type does not match,
|
|
/// the original type-erased asset is returned.
|
|
pub fn downcast<A: Asset>(mut self) -> Result<LoadedAsset<A>, ErasedLoadedAsset> {
|
|
match self.value.downcast::<A>() {
|
|
Ok(value) => Ok(LoadedAsset {
|
|
value: *value,
|
|
dependencies: self.dependencies,
|
|
loader_dependencies: self.loader_dependencies,
|
|
}),
|
|
Err(value) => {
|
|
self.value = value;
|
|
Err(self)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
/// A type erased container for an [`Asset`] value that is capable of inserting the [`Asset`] into a [`World`]'s [`Assets`] collection.
|
|
pub trait AssetContainer: Downcast + Any + Send + Sync + 'static {
|
|
fn insert(self: Box<Self>, id: UntypedAssetId, world: &mut World);
|
|
fn asset_type_name(&self) -> &'static str;
|
|
}
|
|
|
|
impl_downcast!(AssetContainer);
|
|
|
|
impl<A: Asset> AssetContainer for A {
|
|
fn insert(self: Box<Self>, id: UntypedAssetId, world: &mut World) {
|
|
world.resource_mut::<Assets<A>>().insert(id.typed(), *self);
|
|
}
|
|
|
|
fn asset_type_name(&self) -> &'static str {
|
|
core::any::type_name::<A>()
|
|
}
|
|
}
|
|
|
|
/// A loaded asset and all its loaded subassets.
|
|
pub struct CompleteLoadedAsset<A: Asset> {
|
|
/// The loaded asset.
|
|
pub(crate) asset: LoadedAsset<A>,
|
|
/// The subassets by their label.
|
|
pub(crate) labeled_assets: HashMap<CowArc<'static, str>, LabeledAsset>,
|
|
}
|
|
|
|
impl<A: Asset> CompleteLoadedAsset<A> {
|
|
/// Take ownership of the stored [`Asset`] value.
|
|
pub fn take(self) -> A {
|
|
self.asset.value
|
|
}
|
|
|
|
/// Returns the stored asset.
|
|
pub fn get_asset(&self) -> &LoadedAsset<A> {
|
|
&self.asset
|
|
}
|
|
|
|
/// Returns the [`ErasedLoadedAsset`] for the given label, if it exists.
|
|
pub fn get_labeled(
|
|
&self,
|
|
label: impl Into<CowArc<'static, str>>,
|
|
) -> Option<&ErasedLoadedAsset> {
|
|
self.labeled_assets.get(&label.into()).map(|a| &a.asset)
|
|
}
|
|
|
|
/// Iterate over all labels for "labeled assets" in the loaded asset
|
|
pub fn iter_labels(&self) -> impl Iterator<Item = &str> {
|
|
self.labeled_assets.keys().map(|s| &**s)
|
|
}
|
|
}
|
|
|
|
/// A "type erased / boxed" counterpart to [`CompleteLoadedAsset`]. This is used in places where the
|
|
/// loaded type is not statically known.
|
|
pub struct CompleteErasedLoadedAsset {
|
|
/// The loaded asset.
|
|
pub(crate) asset: ErasedLoadedAsset,
|
|
/// The subassets by their label.
|
|
pub(crate) labeled_assets: HashMap<CowArc<'static, str>, LabeledAsset>,
|
|
}
|
|
|
|
impl CompleteErasedLoadedAsset {
|
|
/// Cast (and take ownership) of the [`Asset`] value of the given type. This will return
|
|
/// [`Some`] if the stored type matches `A` and [`None`] if it does not.
|
|
pub fn take<A: Asset>(self) -> Option<A> {
|
|
self.asset.take()
|
|
}
|
|
|
|
/// Returns the stored asset.
|
|
pub fn get_asset(&self) -> &ErasedLoadedAsset {
|
|
&self.asset
|
|
}
|
|
|
|
/// Returns the [`ErasedLoadedAsset`] for the given label, if it exists.
|
|
pub fn get_labeled(
|
|
&self,
|
|
label: impl Into<CowArc<'static, str>>,
|
|
) -> Option<&ErasedLoadedAsset> {
|
|
self.labeled_assets.get(&label.into()).map(|a| &a.asset)
|
|
}
|
|
|
|
/// Iterate over all labels for "labeled assets" in the loaded asset
|
|
pub fn iter_labels(&self) -> impl Iterator<Item = &str> {
|
|
self.labeled_assets.keys().map(|s| &**s)
|
|
}
|
|
|
|
/// Cast this loaded asset as the given type. If the type does not match,
|
|
/// the original type-erased asset is returned.
|
|
pub fn downcast<A: Asset>(
|
|
mut self,
|
|
) -> Result<CompleteLoadedAsset<A>, CompleteErasedLoadedAsset> {
|
|
match self.asset.downcast::<A>() {
|
|
Ok(asset) => Ok(CompleteLoadedAsset {
|
|
asset,
|
|
labeled_assets: self.labeled_assets,
|
|
}),
|
|
Err(asset) => {
|
|
self.asset = asset;
|
|
Err(self)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
impl<A: Asset> From<CompleteLoadedAsset<A>> for CompleteErasedLoadedAsset {
|
|
fn from(value: CompleteLoadedAsset<A>) -> Self {
|
|
Self {
|
|
asset: value.asset.into(),
|
|
labeled_assets: value.labeled_assets,
|
|
}
|
|
}
|
|
}
|
|
|
|
/// An error that occurs when attempting to call [`NestedLoader::load`] which
|
|
/// is configured to work [immediately].
|
|
///
|
|
/// [`NestedLoader::load`]: crate::NestedLoader::load
|
|
/// [immediately]: crate::Immediate
|
|
#[derive(Error, Debug)]
|
|
#[error("Failed to load dependency {dependency:?} {error}")]
|
|
pub struct LoadDirectError {
|
|
pub dependency: AssetPath<'static>,
|
|
pub error: AssetLoadError,
|
|
}
|
|
|
|
/// An error that occurs while deserializing [`AssetMeta`].
|
|
#[derive(Error, Debug, Clone, PartialEq, Eq)]
|
|
pub enum DeserializeMetaError {
|
|
#[error("Failed to deserialize asset meta: {0:?}")]
|
|
DeserializeSettings(#[from] SpannedError),
|
|
#[error("Failed to deserialize minimal asset meta: {0:?}")]
|
|
DeserializeMinimal(SpannedError),
|
|
}
|
|
|
|
/// A context that provides access to assets in [`AssetLoader`]s, tracks dependencies, and collects asset load state.
|
|
///
|
|
/// Any asset state accessed by [`LoadContext`] will be tracked and stored for use in dependency events and asset preprocessing.
|
|
pub struct LoadContext<'a> {
|
|
pub(crate) asset_server: &'a AssetServer,
|
|
pub(crate) should_load_dependencies: bool,
|
|
populate_hashes: bool,
|
|
asset_path: AssetPath<'static>,
|
|
pub(crate) dependencies: HashSet<UntypedAssetId>,
|
|
/// Direct dependencies used by this loader.
|
|
pub(crate) loader_dependencies: HashMap<AssetPath<'static>, AssetHash>,
|
|
pub(crate) labeled_assets: HashMap<CowArc<'static, str>, LabeledAsset>,
|
|
}
|
|
|
|
impl<'a> LoadContext<'a> {
|
|
/// Creates a new [`LoadContext`] instance.
|
|
pub(crate) fn new(
|
|
asset_server: &'a AssetServer,
|
|
asset_path: AssetPath<'static>,
|
|
should_load_dependencies: bool,
|
|
populate_hashes: bool,
|
|
) -> Self {
|
|
Self {
|
|
asset_server,
|
|
asset_path,
|
|
populate_hashes,
|
|
should_load_dependencies,
|
|
dependencies: HashSet::default(),
|
|
loader_dependencies: HashMap::default(),
|
|
labeled_assets: HashMap::default(),
|
|
}
|
|
}
|
|
|
|
/// Begins a new labeled asset load. Use the returned [`LoadContext`] to load
|
|
/// dependencies for the new asset and call [`LoadContext::finish`] to finalize the asset load.
|
|
/// When finished, make sure you call [`LoadContext::add_labeled_asset`] to add the results back to the parent
|
|
/// context.
|
|
/// Prefer [`LoadContext::labeled_asset_scope`] when possible, which will automatically add
|
|
/// the labeled [`LoadContext`] back to the parent context.
|
|
/// [`LoadContext::begin_labeled_asset`] exists largely to enable parallel asset loading.
|
|
///
|
|
/// See [`AssetPath`] for more on labeled assets.
|
|
///
|
|
/// ```no_run
|
|
/// # use bevy_asset::{Asset, LoadContext};
|
|
/// # use bevy_reflect::TypePath;
|
|
/// # #[derive(Asset, TypePath, Default)]
|
|
/// # struct Image;
|
|
/// # let load_context: LoadContext = panic!();
|
|
/// let mut handles = Vec::new();
|
|
/// for i in 0..2 {
|
|
/// let mut labeled = load_context.begin_labeled_asset();
|
|
/// handles.push(std::thread::spawn(move || {
|
|
/// (i.to_string(), labeled.finish(Image::default()))
|
|
/// }));
|
|
/// }
|
|
///
|
|
/// for handle in handles {
|
|
/// let (label, loaded_asset) = handle.join().unwrap();
|
|
/// load_context.add_loaded_labeled_asset(label, loaded_asset);
|
|
/// }
|
|
/// ```
|
|
pub fn begin_labeled_asset(&self) -> LoadContext {
|
|
LoadContext::new(
|
|
self.asset_server,
|
|
self.asset_path.clone(),
|
|
self.should_load_dependencies,
|
|
self.populate_hashes,
|
|
)
|
|
}
|
|
|
|
/// Creates a new [`LoadContext`] for the given `label`. The `load` function is responsible for loading an [`Asset`] of
|
|
/// type `A`. `load` will be called immediately and the result will be used to finalize the [`LoadContext`], resulting in a new
|
|
/// [`LoadedAsset`], which is registered under the `label` label.
|
|
///
|
|
/// This exists to remove the need to manually call [`LoadContext::begin_labeled_asset`] and then manually register the
|
|
/// result with [`LoadContext::add_labeled_asset`].
|
|
///
|
|
/// See [`AssetPath`] for more on labeled assets.
|
|
pub fn labeled_asset_scope<A: Asset>(
|
|
&mut self,
|
|
label: String,
|
|
load: impl FnOnce(&mut LoadContext) -> A,
|
|
) -> Handle<A> {
|
|
let mut context = self.begin_labeled_asset();
|
|
let asset = load(&mut context);
|
|
let complete_asset = context.finish(asset);
|
|
self.add_loaded_labeled_asset(label, complete_asset)
|
|
}
|
|
|
|
/// This will add the given `asset` as a "labeled [`Asset`]" with the `label` label.
|
|
///
|
|
/// # Warning
|
|
///
|
|
/// This will not assign dependencies to the given `asset`. If adding an asset
|
|
/// with dependencies generated from calls such as [`LoadContext::load`], use
|
|
/// [`LoadContext::labeled_asset_scope`] or [`LoadContext::begin_labeled_asset`] to generate a
|
|
/// new [`LoadContext`] to track the dependencies for the labeled asset.
|
|
///
|
|
/// See [`AssetPath`] for more on labeled assets.
|
|
pub fn add_labeled_asset<A: Asset>(&mut self, label: String, asset: A) -> Handle<A> {
|
|
self.labeled_asset_scope(label, |_| asset)
|
|
}
|
|
|
|
/// Add a [`LoadedAsset`] that is a "labeled sub asset" of the root path of this load context.
|
|
/// This can be used in combination with [`LoadContext::begin_labeled_asset`] to parallelize
|
|
/// sub asset loading.
|
|
///
|
|
/// See [`AssetPath`] for more on labeled assets.
|
|
pub fn add_loaded_labeled_asset<A: Asset>(
|
|
&mut self,
|
|
label: impl Into<CowArc<'static, str>>,
|
|
loaded_asset: CompleteLoadedAsset<A>,
|
|
) -> Handle<A> {
|
|
let label = label.into();
|
|
let CompleteLoadedAsset {
|
|
asset,
|
|
labeled_assets,
|
|
} = loaded_asset;
|
|
let loaded_asset: ErasedLoadedAsset = asset.into();
|
|
let labeled_path = self.asset_path.clone().with_label(label.clone());
|
|
let handle = self
|
|
.asset_server
|
|
.get_or_create_path_handle(labeled_path, None);
|
|
self.labeled_assets.insert(
|
|
label,
|
|
LabeledAsset {
|
|
asset: loaded_asset,
|
|
handle: handle.clone().untyped(),
|
|
},
|
|
);
|
|
for (label, asset) in labeled_assets {
|
|
if self.labeled_assets.insert(label.clone(), asset).is_some() {
|
|
warn!("A labeled asset with the label \"{label}\" already exists. Replacing with the new asset.");
|
|
}
|
|
}
|
|
handle
|
|
}
|
|
|
|
/// Returns `true` if an asset with the label `label` exists in this context.
|
|
///
|
|
/// See [`AssetPath`] for more on labeled assets.
|
|
pub fn has_labeled_asset<'b>(&self, label: impl Into<CowArc<'b, str>>) -> bool {
|
|
let path = self.asset_path.clone().with_label(label.into());
|
|
!self.asset_server.get_handles_untyped(&path).is_empty()
|
|
}
|
|
|
|
/// "Finishes" this context by populating the final [`Asset`] value.
|
|
pub fn finish<A: Asset>(self, value: A) -> CompleteLoadedAsset<A> {
|
|
CompleteLoadedAsset {
|
|
asset: LoadedAsset {
|
|
value,
|
|
dependencies: self.dependencies,
|
|
loader_dependencies: self.loader_dependencies,
|
|
},
|
|
labeled_assets: self.labeled_assets,
|
|
}
|
|
}
|
|
|
|
/// Gets the source path for this load context.
|
|
pub fn path(&self) -> &Path {
|
|
self.asset_path.path()
|
|
}
|
|
|
|
/// Gets the source asset path for this load context.
|
|
pub fn asset_path(&self) -> &AssetPath<'static> {
|
|
&self.asset_path
|
|
}
|
|
|
|
/// Reads the asset at the given path and returns its bytes
|
|
pub async fn read_asset_bytes<'b, 'c>(
|
|
&'b mut self,
|
|
path: impl Into<AssetPath<'c>>,
|
|
) -> Result<Vec<u8>, ReadAssetBytesError> {
|
|
let path = path.into();
|
|
let source = self.asset_server.get_source(path.source())?;
|
|
let asset_reader = match self.asset_server.mode() {
|
|
AssetServerMode::Unprocessed { .. } => source.reader(),
|
|
AssetServerMode::Processed { .. } => source.processed_reader()?,
|
|
};
|
|
let mut reader = asset_reader.read(path.path()).await?;
|
|
let hash = if self.populate_hashes {
|
|
// NOTE: ensure meta is read while the asset bytes reader is still active to ensure transactionality
|
|
// See `ProcessorGatedReader` for more info
|
|
let meta_bytes = asset_reader.read_meta_bytes(path.path()).await?;
|
|
let minimal: ProcessedInfoMinimal = ron::de::from_bytes(&meta_bytes)
|
|
.map_err(DeserializeMetaError::DeserializeMinimal)?;
|
|
let processed_info = minimal
|
|
.processed_info
|
|
.ok_or(ReadAssetBytesError::MissingAssetHash)?;
|
|
processed_info.full_hash
|
|
} else {
|
|
Default::default()
|
|
};
|
|
let mut bytes = Vec::new();
|
|
reader
|
|
.read_to_end(&mut bytes)
|
|
.await
|
|
.map_err(|source| ReadAssetBytesError::Io {
|
|
path: path.path().to_path_buf(),
|
|
source,
|
|
})?;
|
|
self.loader_dependencies.insert(path.clone_owned(), hash);
|
|
Ok(bytes)
|
|
}
|
|
|
|
/// Returns a handle to an asset of type `A` with the label `label`. This [`LoadContext`] must produce an asset of the
|
|
/// given type and the given label or the dependencies of this asset will never be considered "fully loaded". However you
|
|
/// can call this method before _or_ after adding the labeled asset.
|
|
pub fn get_label_handle<'b, A: Asset>(
|
|
&mut self,
|
|
label: impl Into<CowArc<'b, str>>,
|
|
) -> Handle<A> {
|
|
let path = self.asset_path.clone().with_label(label);
|
|
let handle = self.asset_server.get_or_create_path_handle::<A>(path, None);
|
|
self.dependencies.insert(handle.id().untyped());
|
|
handle
|
|
}
|
|
|
|
pub(crate) async fn load_direct_internal(
|
|
&mut self,
|
|
path: AssetPath<'static>,
|
|
meta: &dyn AssetMetaDyn,
|
|
loader: &dyn ErasedAssetLoader,
|
|
reader: &mut dyn Reader,
|
|
) -> Result<CompleteErasedLoadedAsset, LoadDirectError> {
|
|
let complete_asset = self
|
|
.asset_server
|
|
.load_with_meta_loader_and_reader(
|
|
&path,
|
|
meta,
|
|
loader,
|
|
reader,
|
|
false,
|
|
self.populate_hashes,
|
|
)
|
|
.await
|
|
.map_err(|error| LoadDirectError {
|
|
dependency: path.clone(),
|
|
error,
|
|
})?;
|
|
let info = meta.processed_info().as_ref();
|
|
let hash = info.map(|i| i.full_hash).unwrap_or_default();
|
|
self.loader_dependencies.insert(path, hash);
|
|
Ok(complete_asset)
|
|
}
|
|
|
|
/// Create a builder for loading nested assets in this context.
|
|
#[must_use]
|
|
pub fn loader(&mut self) -> NestedLoader<'a, '_, StaticTyped, Deferred> {
|
|
NestedLoader::new(self)
|
|
}
|
|
|
|
/// Retrieves a handle for the asset at the given path and adds that path as a dependency of the asset.
|
|
/// If the current context is a normal [`AssetServer::load`], an actual asset load will be kicked off immediately, which ensures the load happens
|
|
/// as soon as possible.
|
|
/// "Normal loads" kicked from within a normal Bevy App will generally configure the context to kick off loads immediately.
|
|
/// If the current context is configured to not load dependencies automatically (ex: [`AssetProcessor`](crate::processor::AssetProcessor)),
|
|
/// a load will not be kicked off automatically. It is then the calling context's responsibility to begin a load if necessary.
|
|
///
|
|
/// If you need to override asset settings, asset type, or load directly, please see [`LoadContext::loader`].
|
|
pub fn load<'b, A: Asset>(&mut self, path: impl Into<AssetPath<'b>>) -> Handle<A> {
|
|
self.loader().load(path)
|
|
}
|
|
}
|
|
|
|
/// An error produced when calling [`LoadContext::read_asset_bytes`]
|
|
#[derive(Error, Debug)]
|
|
pub enum ReadAssetBytesError {
|
|
#[error(transparent)]
|
|
DeserializeMetaError(#[from] DeserializeMetaError),
|
|
#[error(transparent)]
|
|
AssetReaderError(#[from] AssetReaderError),
|
|
#[error(transparent)]
|
|
MissingAssetSourceError(#[from] MissingAssetSourceError),
|
|
#[error(transparent)]
|
|
MissingProcessedAssetReaderError(#[from] MissingProcessedAssetReaderError),
|
|
/// Encountered an I/O error while loading an asset.
|
|
#[error("Encountered an io error while loading asset at `{}`: {source}", path.display())]
|
|
Io {
|
|
path: PathBuf,
|
|
source: std::io::Error,
|
|
},
|
|
#[error("The LoadContext for this read_asset_bytes call requires hash metadata, but it was not provided. This is likely an internal implementation error.")]
|
|
MissingAssetHash,
|
|
}
|