1#import bevy_pbr::{
2    mesh_functions,
3    view_transformations::position_world_to_clip
4}
5
6@group(2) @binding(0) var<storage, read> colors: array<vec4<f32>, 5>;
7
8struct Vertex {
9    @builtin(instance_index) instance_index: u32,
10    @location(0) position: vec3<f32>,
11};
12
13struct VertexOutput {
14    @builtin(position) clip_position: vec4<f32>,
15    @location(0) world_position: vec4<f32>,
16    @location(1) color: vec4<f32>,
17};
18
19@vertex
20fn vertex(vertex: Vertex) -> VertexOutput {
21    var out: VertexOutput;
22    let tag = mesh_functions::get_tag(vertex.instance_index);
23    var world_from_local = mesh_functions::get_world_from_local(vertex.instance_index);
24    out.world_position = mesh_functions::mesh_position_local_to_world(world_from_local, vec4(vertex.position, 1.0));
25    out.clip_position = position_world_to_clip(out.world_position.xyz);
26
27    out.color = colors[tag];
28    return out;
29}
30
31@fragment
32fn fragment(
33    mesh: VertexOutput,
34) -> @location(0) vec4<f32> {
35    return mesh.color;
36}