1#define_import_path bevy_pbr::pbr_types
2
3// Since this is a hot path, try to keep the alignment and size of the struct members in mind.
4// You can find the alignment and sizes at <https://www.w3.org/TR/WGSL/#alignment-and-size>.
5struct StandardMaterial {
6    base_color: vec4<f32>,
7    emissive: vec4<f32>,
8    attenuation_color: vec4<f32>,
9    uv_transform: mat3x3<f32>,
10    reflectance: vec3<f32>,
11    perceptual_roughness: f32,
12    metallic: f32,
13    diffuse_transmission: f32,
14    specular_transmission: f32,
15    thickness: f32,
16    ior: f32,
17    attenuation_distance: f32,
18    clearcoat: f32,
19    clearcoat_perceptual_roughness: f32,
20    anisotropy_strength: f32,
21    anisotropy_rotation: vec2<f32>,
22    // 'flags' is a bit field indicating various options. u32 is 32 bits so we have up to 32 options.
23    flags: u32,
24    alpha_cutoff: f32,
25    parallax_depth_scale: f32,
26    max_parallax_layer_count: f32,
27    lightmap_exposure: f32,
28    max_relief_mapping_search_steps: u32,
29    /// ID for specifying which deferred lighting pass should be used for rendering this material, if any.
30    deferred_lighting_pass_id: u32,
31};
32
33// !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
34// NOTE: if these flags are updated or changed. Be sure to also update
35// deferred_flags_from_mesh_material_flags and mesh_material_flags_from_deferred_flags
36// !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
37const STANDARD_MATERIAL_FLAGS_BASE_COLOR_TEXTURE_BIT: u32            = 1u << 0u;
38const STANDARD_MATERIAL_FLAGS_EMISSIVE_TEXTURE_BIT: u32              = 1u << 1u;
39const STANDARD_MATERIAL_FLAGS_METALLIC_ROUGHNESS_TEXTURE_BIT: u32    = 1u << 2u;
40const STANDARD_MATERIAL_FLAGS_OCCLUSION_TEXTURE_BIT: u32             = 1u << 3u;
41const STANDARD_MATERIAL_FLAGS_DOUBLE_SIDED_BIT: u32                  = 1u << 4u;
42const STANDARD_MATERIAL_FLAGS_UNLIT_BIT: u32                         = 1u << 5u;
43const STANDARD_MATERIAL_FLAGS_TWO_COMPONENT_NORMAL_MAP: u32          = 1u << 6u;
44const STANDARD_MATERIAL_FLAGS_FLIP_NORMAL_MAP_Y: u32                 = 1u << 7u;
45const STANDARD_MATERIAL_FLAGS_FOG_ENABLED_BIT: u32                   = 1u << 8u;
46const STANDARD_MATERIAL_FLAGS_DEPTH_MAP_BIT: u32                     = 1u << 9u;
47const STANDARD_MATERIAL_FLAGS_SPECULAR_TRANSMISSION_TEXTURE_BIT: u32 = 1u << 10u;
48const STANDARD_MATERIAL_FLAGS_THICKNESS_TEXTURE_BIT: u32             = 1u << 11u;
49const STANDARD_MATERIAL_FLAGS_DIFFUSE_TRANSMISSION_TEXTURE_BIT: u32  = 1u << 12u;
50const STANDARD_MATERIAL_FLAGS_ATTENUATION_ENABLED_BIT: u32           = 1u << 13u;
51const STANDARD_MATERIAL_FLAGS_CLEARCOAT_TEXTURE_BIT: u32             = 1u << 14u;
52const STANDARD_MATERIAL_FLAGS_CLEARCOAT_ROUGHNESS_TEXTURE_BIT: u32   = 1u << 15u;
53const STANDARD_MATERIAL_FLAGS_CLEARCOAT_NORMAL_TEXTURE_BIT: u32      = 1u << 16u;
54const STANDARD_MATERIAL_FLAGS_ANISOTROPY_TEXTURE_BIT: u32            = 1u << 17u;
55const STANDARD_MATERIAL_FLAGS_SPECULAR_TEXTURE_BIT: u32              = 1u << 18u;
56const STANDARD_MATERIAL_FLAGS_SPECULAR_TINT_TEXTURE_BIT: u32         = 1u << 19u;
57const STANDARD_MATERIAL_FLAGS_ALPHA_MODE_RESERVED_BITS: u32          = 7u << 29u; // (0b111u << 29u)
58const STANDARD_MATERIAL_FLAGS_ALPHA_MODE_OPAQUE: u32                 = 0u << 29u;
59const STANDARD_MATERIAL_FLAGS_ALPHA_MODE_MASK: u32                   = 1u << 29u;
60const STANDARD_MATERIAL_FLAGS_ALPHA_MODE_BLEND: u32                  = 2u << 29u;
61const STANDARD_MATERIAL_FLAGS_ALPHA_MODE_PREMULTIPLIED: u32          = 3u << 29u;
62const STANDARD_MATERIAL_FLAGS_ALPHA_MODE_ADD: u32                    = 4u << 29u;
63const STANDARD_MATERIAL_FLAGS_ALPHA_MODE_MULTIPLY: u32               = 5u << 29u;
64const STANDARD_MATERIAL_FLAGS_ALPHA_MODE_ALPHA_TO_COVERAGE: u32      = 6u << 29u;
65
66
67// Creates a StandardMaterial with default values
68fn standard_material_new() -> StandardMaterial {
69    var material: StandardMaterial;
70
71    // NOTE: Keep in-sync with src/pbr_material.rs!
72    material.base_color = vec4<f32>(1.0, 1.0, 1.0, 1.0);
73    material.emissive = vec4<f32>(0.0, 0.0, 0.0, 1.0);
74    material.perceptual_roughness = 0.5;
75    material.metallic = 0.00;
76    material.reflectance = vec3<f32>(0.5);
77    material.diffuse_transmission = 0.0;
78    material.specular_transmission = 0.0;
79    material.thickness = 0.0;
80    material.ior = 1.5;
81    material.attenuation_distance = 1.0;
82    material.attenuation_color = vec4<f32>(1.0, 1.0, 1.0, 1.0);
83    material.clearcoat = 0.0;
84    material.clearcoat_perceptual_roughness = 0.0;
85    material.flags = STANDARD_MATERIAL_FLAGS_ALPHA_MODE_OPAQUE;
86    material.alpha_cutoff = 0.5;
87    material.parallax_depth_scale = 0.1;
88    material.max_parallax_layer_count = 16.0;
89    material.max_relief_mapping_search_steps = 5u;
90    material.deferred_lighting_pass_id = 1u;
91    // scale 1, translation 0, rotation 0
92    material.uv_transform = mat3x3<f32>(1.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 1.0);
93
94    return material;
95}
96
97struct PbrInput {
98    material: StandardMaterial,
99    // Note: this gets monochromized upon deferred PbrInput reconstruction.
100    diffuse_occlusion: vec3<f32>,
101    // Note: this is 1.0 (entirely unoccluded) when SSAO and SSR are off.
102    specular_occlusion: f32,
103    frag_coord: vec4<f32>,
104    world_position: vec4<f32>,
105    // Normalized world normal used for shadow mapping as normal-mapping is not used for shadow
106    // mapping
107    world_normal: vec3<f32>,
108    // Normalized normal-mapped world normal used for lighting
109    N: vec3<f32>,
110    // Normalized view vector in world space, pointing from the fragment world position toward the
111    // view world position
112    V: vec3<f32>,
113    lightmap_light: vec3<f32>,
114    clearcoat_N: vec3<f32>,
115    anisotropy_strength: f32,
116    // These two aren't specific to anisotropy, but we only fill them in if
117    // we're doing anisotropy, so they're prefixed with `anisotropy_`.
118    anisotropy_T: vec3<f32>,
119    anisotropy_B: vec3<f32>,
120    is_orthographic: bool,
121    flags: u32,
122};
123
124// Creates a PbrInput with default values
125fn pbr_input_new() -> PbrInput {
126    var pbr_input: PbrInput;
127
128    pbr_input.material = standard_material_new();
129    pbr_input.diffuse_occlusion = vec3<f32>(1.0);
130    // If SSAO is enabled, then this gets overwritten with proper specular occlusion. If its not, then we get specular environment map unoccluded (we have no data with which to occlude it with).
131    pbr_input.specular_occlusion = 1.0;
132
133    pbr_input.frag_coord = vec4<f32>(0.0, 0.0, 0.0, 1.0);
134    pbr_input.world_position = vec4<f32>(0.0, 0.0, 0.0, 1.0);
135    pbr_input.world_normal = vec3<f32>(0.0, 0.0, 1.0);
136
137    pbr_input.is_orthographic = false;
138
139    pbr_input.N = vec3<f32>(0.0, 0.0, 1.0);
140    pbr_input.V = vec3<f32>(1.0, 0.0, 0.0);
141
142    pbr_input.clearcoat_N = vec3<f32>(0.0);
143    pbr_input.anisotropy_T = vec3<f32>(0.0);
144    pbr_input.anisotropy_B = vec3<f32>(0.0);
145
146    pbr_input.lightmap_light = vec3<f32>(0.0);
147
148    pbr_input.flags = 0u;
149
150    return pbr_input;
151}