1#import bevy_pbr::{
2    forward_io::VertexOutput,
3    mesh_functions,
4    mesh_view_bindings::view,
5    pbr_types::{STANDARD_MATERIAL_FLAGS_DOUBLE_SIDED_BIT, PbrInput, pbr_input_new},
6    pbr_functions as fns,
7    pbr_bindings,
8}
9#import bevy_core_pipeline::tonemapping::tone_mapping
10
11@group(#{MATERIAL_BIND_GROUP}) @binding(0) var my_array_texture: texture_2d_array<f32>;
12@group(#{MATERIAL_BIND_GROUP}) @binding(1) var my_array_texture_sampler: sampler;
13
14@fragment
15fn fragment(
16    @builtin(front_facing) is_front: bool,
17    mesh: VertexOutput,
18) -> @location(0) vec4<f32> {
19    // Determine which layer of the array texture to sample from based on the
20    // mesh tag which originates from the MeshTag component on the entity.
21    let layer = mesh_functions::get_tag(mesh.instance_index);
22
23    // Prepare a 'processed' StandardMaterial by sampling all textures to resolve
24    // the material members
25    var pbr_input: PbrInput = pbr_input_new();
26
27    pbr_input.material.base_color = textureSample(my_array_texture, my_array_texture_sampler, mesh.uv, layer);
28#ifdef VERTEX_COLORS
29    pbr_input.material.base_color = pbr_input.material.base_color * mesh.color;
30#endif
31
32    let double_sided = (pbr_input.material.flags & STANDARD_MATERIAL_FLAGS_DOUBLE_SIDED_BIT) != 0u;
33
34    pbr_input.frag_coord = mesh.position;
35    pbr_input.world_position = mesh.world_position;
36    pbr_input.world_normal = fns::prepare_world_normal(
37        mesh.world_normal,
38        double_sided,
39        is_front,
40    );
41
42    pbr_input.is_orthographic = view.clip_from_view[3].w == 1.0;
43
44    pbr_input.N = normalize(pbr_input.world_normal);
45
46#ifdef VERTEX_TANGENTS
47    let Nt = textureSampleBias(pbr_bindings::normal_map_texture, pbr_bindings::normal_map_sampler, mesh.uv, view.mip_bias).rgb;
48    let TBN = fns::calculate_tbn_mikktspace(mesh.world_normal, mesh.world_tangent);
49    pbr_input.N = fns::apply_normal_mapping(
50        pbr_input.material.flags,
51        TBN,
52        double_sided,
53        is_front,
54        Nt,
55    );
56#endif
57
58    pbr_input.V = fns::calculate_view(mesh.world_position, pbr_input.is_orthographic);
59
60    return tone_mapping(fns::apply_pbr_lighting(pbr_input), view.color_grading);
61}