SolarEngine/solar_engine/src/shader.wgsl

struct VertexInput {
    @location(0) position: vec3<f32>,
    @location(1) color: vec3<f32>,
    @location(2) normal: vec3<f32>,
};

struct InstanceInput {
    @location(5) model_row0: vec4<f32>,
    @location(6) model_row1: vec4<f32>,
    @location(7) model_row2: vec4<f32>,
    @location(8) model_row3: vec4<f32>,
    @location(9) color: vec3<f32>,
    @location(10) flags: u32,
};

struct VSOutput {
    @builtin(position) position: vec4<f32>,
    @location(0) frag_color: vec3<f32>,
    @location(1) world_pos: vec3<f32>,
    @location(2) normal: vec3<f32>,
    @location(3) flags: u32,
};

struct Globals {
    view_proj: mat4x4<f32>,
}

@group(0) @binding(0)
var<uniform> globals: Globals;

struct GpuLight {
    position: vec3<f32>,
    color: vec3<f32>,
    intensity: f32,
};

@group(1) @binding(0)
var<uniform> lights: array<GpuLight, 10>;

@vertex
fn vs_main(vertex: VertexInput, instance: InstanceInput) -> VSOutput {
    var out: VSOutput;

    let model = mat4x4<f32>(
        instance.model_row0,
        instance.model_row1,
        instance.model_row2,
        instance.model_row3
    );

    let world_position = (model * vec4<f32>(vertex.position, 1.0)).xyz;
    let normal_matrix = mat3x3<f32>(
        instance.model_row0.xyz,
        instance.model_row1.xyz,
        instance.model_row2.xyz
    );

    out.position = globals.view_proj * vec4<f32>(world_position, 1.0);
    out.frag_color = instance.color * vertex.color;
    out.world_pos = world_position;
    out.normal = normalize(normal_matrix * vertex.normal);
    out.flags = instance.flags;

    return out;
}

@fragment
fn fs_main(input: VSOutput) -> @location(0) vec4<f32> {
    var lighting: vec3<f32>;

    let always_lit = (input.flags & 0x1u) != 0u;

    if (always_lit) {
        return vec4<f32>(input.frag_color * 2.0, 1.0);
    } else {
        lighting = vec3<f32>(0.0);
        for (var i = 0u; i < 10u; i = i + 1u) {
            let light = lights[i];
            let light_dir = normalize(light.position - input.world_pos);
            let diff = max(dot(input.normal, light_dir), 0.0);
            lighting += light.color * light.intensity * diff;
        }
    }

    let final_color = input.frag_color * lighting;
    return vec4<f32>(final_color, 1.0);
}