add interpolation setting to pbr based on 0.16.1 release

This commit is contained in:
Arkitu 2025-07-20 10:14:32 +02:00
parent 383b351045
commit 922b1c9fd1
4 changed files with 58 additions and 5 deletions

View File

@ -0,0 +1,9 @@
use bevy_reflect::{std_traits::ReflectDefault, Reflect};
#[derive(Debug, Copy, Clone, PartialEq, Eq, Default, Reflect)]
#[reflect(Default, Clone, PartialEq)]
pub enum InterpolationMethod {
#[default]
Linear,
Flat,
}

View File

@ -31,6 +31,7 @@ pub mod decal;
pub mod deferred; pub mod deferred;
mod extended_material; mod extended_material;
mod fog; mod fog;
mod interpolation;
mod light; mod light;
mod light_probe; mod light_probe;
mod lightmap; mod lightmap;
@ -53,6 +54,7 @@ pub use components::*;
pub use decal::clustered::ClusteredDecalPlugin; pub use decal::clustered::ClusteredDecalPlugin;
pub use extended_material::*; pub use extended_material::*;
pub use fog::*; pub use fog::*;
pub use interpolation::*;
pub use light::*; pub use light::*;
pub use light_probe::*; pub use light_probe::*;
pub use lightmap::*; pub use lightmap::*;

View File

@ -386,6 +386,23 @@ pub struct StandardMaterial {
/// ///
/// [`Mesh::generate_tangents`]: bevy_render::mesh::Mesh::generate_tangents /// [`Mesh::generate_tangents`]: bevy_render::mesh::Mesh::generate_tangents
/// [`Mesh::with_generated_tangents`]: bevy_render::mesh::Mesh::with_generated_tangents /// [`Mesh::with_generated_tangents`]: bevy_render::mesh::Mesh::with_generated_tangents
///
/// # Usage
///
/// ```
/// # use bevy_asset::{AssetServer, Handle};
/// # use bevy_ecs::change_detection::Res;
/// # use bevy_image::{Image, ImageLoaderSettings};
/// #
/// fn load_normal_map(asset_server: Res<AssetServer>) {
/// let normal_handle: Handle<Image> = asset_server.load_with_settings(
/// "textures/parallax_example/cube_normal.png",
/// // The normal map texture is in linear color space. Lighting won't look correct
/// // if `is_srgb` is `true`, which is the default.
/// |settings: &mut ImageLoaderSettings| settings.is_srgb = false,
/// );
/// }
/// ```
#[texture(9)] #[texture(9)]
#[sampler(10)] #[sampler(10)]
#[dependency] #[dependency]
@ -752,6 +769,8 @@ pub struct StandardMaterial {
/// The transform applied to the UVs corresponding to `ATTRIBUTE_UV_0` on the mesh before sampling. Default is identity. /// The transform applied to the UVs corresponding to `ATTRIBUTE_UV_0` on the mesh before sampling. Default is identity.
pub uv_transform: Affine2, pub uv_transform: Affine2,
pub interpolation_method: InterpolationMethod,
} }
impl StandardMaterial { impl StandardMaterial {
@ -907,6 +926,7 @@ impl Default for StandardMaterial {
opaque_render_method: OpaqueRendererMethod::Auto, opaque_render_method: OpaqueRendererMethod::Auto,
deferred_lighting_pass_id: DEFAULT_PBR_DEFERRED_LIGHTING_PASS_ID, deferred_lighting_pass_id: DEFAULT_PBR_DEFERRED_LIGHTING_PASS_ID,
uv_transform: Affine2::IDENTITY, uv_transform: Affine2::IDENTITY,
interpolation_method: InterpolationMethod::Linear,
} }
} }
} }
@ -1185,7 +1205,8 @@ impl AsBindGroupShaderType<StandardMaterialUniform> for StandardMaterial {
bitflags! { bitflags! {
/// The pipeline key for `StandardMaterial`, packed into 64 bits. /// The pipeline key for `StandardMaterial`, packed into 64 bits.
#[derive(Clone, Copy, PartialEq, Eq, Hash)] #[repr(C)]
#[derive(Clone, Copy, PartialEq, Eq, Hash, bytemuck::Pod, bytemuck::Zeroable)]
pub struct StandardMaterialKey: u64 { pub struct StandardMaterialKey: u64 {
const CULL_FRONT = 0x000001; const CULL_FRONT = 0x000001;
const CULL_BACK = 0x000002; const CULL_BACK = 0x000002;
@ -1210,6 +1231,7 @@ bitflags! {
const CLEARCOAT_NORMAL_UV = 0x100000; const CLEARCOAT_NORMAL_UV = 0x100000;
const SPECULAR_UV = 0x200000; const SPECULAR_UV = 0x200000;
const SPECULAR_TINT_UV = 0x400000; const SPECULAR_TINT_UV = 0x400000;
const FLAT_INTERPOLATION = 0x800000;
const DEPTH_BIAS = 0xffffffff_00000000; const DEPTH_BIAS = 0xffffffff_00000000;
} }
} }
@ -1334,18 +1356,24 @@ impl From<&StandardMaterial> for StandardMaterialKey {
); );
} }
key.set(
StandardMaterialKey::FLAT_INTERPOLATION,
material.interpolation_method == InterpolationMethod::Flat,
);
key.insert(StandardMaterialKey::from_bits_retain( key.insert(StandardMaterialKey::from_bits_retain(
// Casting to i32 first to ensure the full i32 range is preserved. // Casting to i32 first to ensure the full i32 range is preserved.
// (wgpu expects the depth_bias as an i32 when this is extracted in a later step) // (wgpu expects the depth_bias as an i32 when this is extracted in a later step)
(material.depth_bias as i32 as u64) << STANDARD_MATERIAL_KEY_DEPTH_BIAS_SHIFT, (material.depth_bias as i32 as u64) << STANDARD_MATERIAL_KEY_DEPTH_BIAS_SHIFT,
)); ));
key key
} }
} }
impl Material for StandardMaterial { impl Material for StandardMaterial {
fn fragment_shader() -> ShaderRef { fn fragment_shader() -> ShaderRef {
PBR_SHADER_HANDLE.into() shader_ref(bevy_asset::embedded_path!("render/pbr.wgsl"))
} }
#[inline] #[inline]
@ -1381,11 +1409,11 @@ impl Material for StandardMaterial {
} }
fn prepass_fragment_shader() -> ShaderRef { fn prepass_fragment_shader() -> ShaderRef {
PBR_PREPASS_SHADER_HANDLE.into() shader_ref(bevy_asset::embedded_path!("render/pbr_prepass.wgsl"))
} }
fn deferred_fragment_shader() -> ShaderRef { fn deferred_fragment_shader() -> ShaderRef {
PBR_SHADER_HANDLE.into() shader_ref(bevy_asset::embedded_path!("render/pbr.wgsl"))
} }
#[cfg(feature = "meshlet")] #[cfg(feature = "meshlet")]
@ -1404,7 +1432,7 @@ impl Material for StandardMaterial {
} }
fn specialize( fn specialize(
_pipeline: &MaterialPipeline<Self>, _pipeline: &MaterialPipeline,
descriptor: &mut RenderPipelineDescriptor, descriptor: &mut RenderPipelineDescriptor,
_layout: &MeshVertexBufferLayoutRef, _layout: &MeshVertexBufferLayoutRef,
key: MaterialPipelineKey<Self>, key: MaterialPipelineKey<Self>,
@ -1499,9 +1527,14 @@ impl Material for StandardMaterial {
StandardMaterialKey::SPECULAR_TINT_UV, StandardMaterialKey::SPECULAR_TINT_UV,
"STANDARD_MATERIAL_SPECULAR_TINT_UV_B", "STANDARD_MATERIAL_SPECULAR_TINT_UV_B",
), ),
(
StandardMaterialKey::FLAT_INTERPOLATION,
"STANDARD_MATERIAL_FLAT_INTERPOLATION",
),
] { ] {
if key.bind_group_data.intersects(flags) { if key.bind_group_data.intersects(flags) {
shader_defs.push(shader_def.into()); shader_defs.push(shader_def.into());
descriptor.vertex.shader_defs.push(shader_def.into());
} }
} }
} }
@ -1524,6 +1557,7 @@ impl Material for StandardMaterial {
depth_stencil.bias.constant = depth_stencil.bias.constant =
(key.bind_group_data.bits() >> STANDARD_MATERIAL_KEY_DEPTH_BIAS_SHIFT) as i32; (key.bind_group_data.bits() >> STANDARD_MATERIAL_KEY_DEPTH_BIAS_SHIFT) as i32;
} }
Ok(()) Ok(())
} }
} }

View File

@ -34,7 +34,11 @@ struct VertexOutput {
// and `frag coord` when used as a fragment stage input // and `frag coord` when used as a fragment stage input
@builtin(position) position: vec4<f32>, @builtin(position) position: vec4<f32>,
@location(0) world_position: vec4<f32>, @location(0) world_position: vec4<f32>,
#ifdef STANDARD_MATERIAL_FLAT_INTERPOLATION
@location(1) @interpolate(flat) world_normal: vec3<f32>,
#else
@location(1) world_normal: vec3<f32>, @location(1) world_normal: vec3<f32>,
#endif
#ifdef VERTEX_UVS_A #ifdef VERTEX_UVS_A
@location(2) uv: vec2<f32>, @location(2) uv: vec2<f32>,
#endif #endif
@ -45,8 +49,12 @@ struct VertexOutput {
@location(4) world_tangent: vec4<f32>, @location(4) world_tangent: vec4<f32>,
#endif #endif
#ifdef VERTEX_COLORS #ifdef VERTEX_COLORS
#ifdef STANDARD_MATERIAL_FLAT_INTERPOLATION
@location(5) @interpolate(flat) color: vec4<f32>,
#else
@location(5) color: vec4<f32>, @location(5) color: vec4<f32>,
#endif #endif
#endif
#ifdef VERTEX_OUTPUT_INSTANCE_INDEX #ifdef VERTEX_OUTPUT_INSTANCE_INDEX
@location(6) @interpolate(flat) instance_index: u32, @location(6) @interpolate(flat) instance_index: u32,
#endif #endif