diff --git a/crates/bevy_pbr/src/material.rs b/crates/bevy_pbr/src/material.rs index a4a8eb8401e8a..5f8bce5dfe7bb 100644 --- a/crates/bevy_pbr/src/material.rs +++ b/crates/bevy_pbr/src/material.rs @@ -31,8 +31,9 @@ use bevy_render::{ Extract, ExtractSchedule, Render, RenderApp, RenderSet, }; use bevy_utils::{tracing::error, HashMap, HashSet}; -use std::hash::Hash; use std::marker::PhantomData; +use std::sync::atomic::{AtomicU32, Ordering}; +use std::{hash::Hash, num::NonZeroU32}; use self::{irradiance_volume::IrradianceVolume, prelude::EnvironmentMapLight}; @@ -236,9 +237,7 @@ where .after(prepare_materials::), queue_material_meshes:: .in_set(RenderSet::QueueMeshes) - .after(prepare_materials::) - // queue_material_meshes only writes to `material_bind_group_id`, which `queue_shadows` doesn't read - .ambiguous_with(render::queue_shadows::), + .after(prepare_materials::), ), ); } @@ -466,7 +465,7 @@ pub fn queue_material_meshes( msaa: Res, render_meshes: Res>, render_materials: Res>, - mut render_mesh_instances: ResMut, + render_mesh_instances: Res, render_material_instances: Res>, render_lightmaps: Res, mut views: Query<( @@ -592,7 +591,7 @@ pub fn queue_material_meshes( let Some(material_asset_id) = render_material_instances.get(visible_entity) else { continue; }; - let Some(mesh_instance) = render_mesh_instances.get_mut(visible_entity) else { + let Some(mesh_instance) = render_mesh_instances.get(visible_entity) else { continue; }; let Some(mesh) = render_meshes.get(mesh_instance.mesh_asset_id) else { @@ -646,7 +645,9 @@ pub fn queue_material_meshes( } }; - mesh_instance.material_bind_group_id = material.get_bind_group_id(); + mesh_instance + .material_bind_group_id + .set(material.get_bind_group_id()); match material.properties.alpha_mode { AlphaMode::Opaque => { @@ -795,6 +796,34 @@ pub struct PreparedMaterial { #[derive(Component, Clone, Copy, Default, PartialEq, Eq, Deref, DerefMut)] pub struct MaterialBindGroupId(Option); +/// An atomic version of [`MaterialBindGroupId`] that can be read from and written to +/// safely from multiple threads. +#[derive(Default)] +pub struct AtomicMaterialBindGroupId(AtomicU32); + +impl AtomicMaterialBindGroupId { + /// Stores a value atomically. Uses [`Ordering::Relaxed`] so there is zero guarentee of ordering + /// relative to other operations. + /// + /// See also: [`AtomicU32::store`]. + pub fn set(&self, id: MaterialBindGroupId) { + let id = if let Some(id) = id.0 { + NonZeroU32::from(id).get() + } else { + 0 + }; + self.0.store(id, Ordering::Relaxed); + } + + /// Loads a value atomically. Uses [`Ordering::Relaxed`] so there is zero guarentee of ordering + /// relative to other operations. + /// + /// See also: [`AtomicU32::load`]. + pub fn get(&self) -> MaterialBindGroupId { + MaterialBindGroupId(NonZeroU32::new(self.0.load(Ordering::Relaxed)).map(BindGroupId::from)) + } +} + impl PreparedMaterial { pub fn get_bind_group_id(&self) -> MaterialBindGroupId { MaterialBindGroupId(Some(self.bind_group.id())) diff --git a/crates/bevy_pbr/src/render/light.rs b/crates/bevy_pbr/src/render/light.rs index e57dd38f4a956..5a923e73b17ad 100644 --- a/crates/bevy_pbr/src/render/light.rs +++ b/crates/bevy_pbr/src/render/light.rs @@ -1591,7 +1591,7 @@ pub fn queue_shadows( shadow_draw_functions: Res>, prepass_pipeline: Res>, render_meshes: Res>, - mut render_mesh_instances: ResMut, + render_mesh_instances: Res, render_materials: Res>, render_material_instances: Res>, mut pipelines: ResMut>>, @@ -1637,7 +1637,7 @@ pub fn queue_shadows( // NOTE: Lights with shadow mapping disabled will have no visible entities // so no meshes will be queued for entity in visible_entities.iter().copied() { - let Some(mesh_instance) = render_mesh_instances.get_mut(&entity) else { + let Some(mesh_instance) = render_mesh_instances.get(&entity) else { continue; }; if !mesh_instance.shadow_caster { @@ -1697,7 +1697,9 @@ pub fn queue_shadows( } }; - mesh_instance.material_bind_group_id = material.get_bind_group_id(); + mesh_instance + .material_bind_group_id + .set(material.get_bind_group_id()); shadow_phase.add(Shadow { draw_function: draw_shadow_mesh, diff --git a/crates/bevy_pbr/src/render/mesh.rs b/crates/bevy_pbr/src/render/mesh.rs index d34121c919e89..7ea546749f744 100644 --- a/crates/bevy_pbr/src/render/mesh.rs +++ b/crates/bevy_pbr/src/render/mesh.rs @@ -1,7 +1,8 @@ use crate::{ - MaterialBindGroupId, NotShadowCaster, NotShadowReceiver, PreviousGlobalTransform, Shadow, - ViewFogUniformOffset, ViewLightProbesUniformOffset, ViewLightsUniformOffset, - CLUSTERED_FORWARD_STORAGE_BUFFER_COUNT, MAX_CASCADES_PER_LIGHT, MAX_DIRECTIONAL_LIGHTS, + AtomicMaterialBindGroupId, MaterialBindGroupId, NotShadowCaster, NotShadowReceiver, + PreviousGlobalTransform, Shadow, ViewFogUniformOffset, ViewLightProbesUniformOffset, + ViewLightsUniformOffset, CLUSTERED_FORWARD_STORAGE_BUFFER_COUNT, MAX_CASCADES_PER_LIGHT, + MAX_DIRECTIONAL_LIGHTS, }; use bevy_app::{Plugin, PostUpdate}; use bevy_asset::{load_internal_asset, AssetId, Handle}; @@ -251,14 +252,14 @@ bitflags::bitflags! { pub struct RenderMeshInstance { pub transforms: MeshTransforms, pub mesh_asset_id: AssetId, - pub material_bind_group_id: MaterialBindGroupId, + pub material_bind_group_id: AtomicMaterialBindGroupId, pub shadow_caster: bool, pub automatic_batching: bool, } impl RenderMeshInstance { pub fn should_batch(&self) -> bool { - self.automatic_batching && self.material_bind_group_id.is_some() + self.automatic_batching && self.material_bind_group_id.get().is_some() } } @@ -323,7 +324,7 @@ pub fn extract_meshes( mesh_asset_id: handle.id(), transforms, shadow_caster: !not_shadow_caster, - material_bind_group_id: MaterialBindGroupId::default(), + material_bind_group_id: AtomicMaterialBindGroupId::default(), automatic_batching: !no_automatic_batching, }, )); @@ -475,7 +476,7 @@ impl GetBatchData for MeshPipeline { maybe_lightmap.map(|lightmap| lightmap.uv_rect), ), mesh_instance.should_batch().then_some(( - mesh_instance.material_bind_group_id, + mesh_instance.material_bind_group_id.get(), mesh_instance.mesh_asset_id, maybe_lightmap.map(|lightmap| lightmap.image), )), diff --git a/crates/bevy_render/src/render_resource/resource_macros.rs b/crates/bevy_render/src/render_resource/resource_macros.rs index ab5ce2cbebe19..de2ea0ec00e58 100644 --- a/crates/bevy_render/src/render_resource/resource_macros.rs +++ b/crates/bevy_render/src/render_resource/resource_macros.rs @@ -136,6 +136,18 @@ macro_rules! define_atomic_id { })) } } + + impl From<$atomic_id_type> for core::num::NonZeroU32 { + fn from(value: $atomic_id_type) -> Self { + value.0 + } + } + + impl From for $atomic_id_type { + fn from(value: core::num::NonZeroU32) -> Self { + Self(value) + } + } }; }