[Merged by Bors] - Introduce SystemLabel's for RenderAssetPlugin, and change Image preparation system to run before others

crates/bevy_render/src/render_asset.rs

@@ -39,23 +39,63 @@ pub trait RenderAsset: Asset {
     ) -> Result<Self::PreparedAsset, PrepareAssetError<Self::ExtractedAsset>>;
 }
 
+#[derive(Clone, Hash, Debug, PartialEq, Eq, SystemLabel)]
+pub enum PrepareAssetLabel {
+    PreAssetPrepare,
+    AssetPrepare,
+    PostAssetPrepare,
+}
+
+impl Default for PrepareAssetLabel {
+    fn default() -> Self {
+        Self::AssetPrepare
+    }
+}
+
 /// This plugin extracts the changed assets from the "app world" into the "render world"
 /// and prepares them for the GPU. They can then be accessed from the [`RenderAssets`] resource.
 ///
 /// Therefore it sets up the [`RenderStage::Extract`](crate::RenderStage::Extract) and
 /// [`RenderStage::Prepare`](crate::RenderStage::Prepare) steps for the specified [`RenderAsset`].
-pub struct RenderAssetPlugin<A: RenderAsset>(PhantomData<fn() -> A>);
+pub struct RenderAssetPlugin<A: RenderAsset> {
+    prepare_asset_label: PrepareAssetLabel,
+    phantom: PhantomData<fn() -> A>,
+}
+
+impl<A: RenderAsset> RenderAssetPlugin<A> {
+    pub fn with_prepare_asset_label(prepare_asset_label: PrepareAssetLabel) -> Self {
+        Self {
+            prepare_asset_label,
+            phantom: PhantomData,
+        }
+    }
+}
 
 impl<A: RenderAsset> Default for RenderAssetPlugin<A> {
     fn default() -> Self {
-        Self(PhantomData)
+        Self {
+            prepare_asset_label: Default::default(),
+            phantom: PhantomData,
+        }
     }
 }
 
 impl<A: RenderAsset> Plugin for RenderAssetPlugin<A> {
     fn build(&self, app: &mut App) {
         if let Ok(render_app) = app.get_sub_app_mut(RenderApp) {
-            let prepare_asset_system = PrepareAssetSystem::<A>::system(&mut render_app.world);
+            let prepare_asset_system = PrepareAssetSystem::<A>::system(&mut render_app.world)
+                .label(self.prepare_asset_label.clone());
+
+            let prepare_asset_system = match self.prepare_asset_label {
+                PrepareAssetLabel::PreAssetPrepare => prepare_asset_system,
+                PrepareAssetLabel::AssetPrepare => {
+                    prepare_asset_system.after(PrepareAssetLabel::PreAssetPrepare)
+                }
+                PrepareAssetLabel::PostAssetPrepare => {
+                    prepare_asset_system.after(PrepareAssetLabel::AssetPrepare)
+                }
+            };
+
             render_app
                 .init_resource::<ExtractedAssets<A>>()
                 .init_resource::<RenderAssets<A>>()

crates/bevy_render/src/texture/mod.rs

@@ -14,7 +14,10 @@ pub use hdr_texture_loader::*;
 pub use image_texture_loader::*;
 pub use texture_cache::*;
 
-use crate::{render_asset::RenderAssetPlugin, RenderApp, RenderStage};
+use crate::{
+    render_asset::{PrepareAssetLabel, RenderAssetPlugin},
+    RenderApp, RenderStage,
+};
 use bevy_app::{App, Plugin};
 use bevy_asset::{AddAsset, Assets};
 
@@ -40,8 +43,10 @@ impl Plugin for ImagePlugin {
             app.init_asset_loader::<HdrTextureLoader>();
         }
 
-        app.add_plugin(RenderAssetPlugin::<Image>::default())
-            .add_asset::<Image>();
+        app.add_plugin(RenderAssetPlugin::<Image>::with_prepare_asset_label(
+            PrepareAssetLabel::PreAssetPrepare,
+        ))
+        .add_asset::<Image>();
         app.world
             .get_resource_mut::<Assets<Image>>()
             .unwrap()