struct ModelMatrix {
    model_matrix: mat4x4<f32>,
}

struct Material {
    albedo: vec4<f32>,
    roughness: f32,
    reflectance: f32,
    ambient_ratio: f32,
    background_ratio: f32,
}

@group(1)
@binding(0)
var<uniform> model_matrix: ModelMatrix;

@group(1)
@binding(1)
var<uniform> material: Material;

@group(1)
@binding(2)
var r_color: texture_2d<f32>;

@group(1)
@binding(3)
var r_sampler: sampler;

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

struct VertexOutput {
    @builtin(position) position: vec4<f32>,
    @location(0) uv: vec2<f32>,
}

const EPS: f32 = 1.0e-6;
const e: vec2<f32> = vec2<f32>(1.0, 0.0);

@vertex
fn vs_main(in: VertexInput) -> VertexOutput {
    var out: VertexOutput;
    out.uv = vec2<f32>(0.1);
    if (distance(in.position, vec3<f32>(in.uv.x, 2.0, in.uv.y)) > EPS) {
        out.position = vec4<f32>(0.0, 0.0, 0.0, 1.0);
    } else if (distance(in.normal, vec3<f32>(in.uv.y, 0.2, in.uv.x)) > EPS) {
        out.position = vec4<f32>(0.0, 0.0, 0.0, 1.0);
    } else if (distance(model_matrix.model_matrix * e.xyyy, vec4<f32>(1.0, 2.0, 3.0, 4.0)) > EPS) {
        out.position = vec4<f32>(0.0, 0.0, 0.0, 1.0);
    } else if (distance(model_matrix.model_matrix * e.yxyy, vec4<f32>(5.0, 6.0, 7.0, 8.0)) > EPS) {
        out.position = vec4<f32>(0.0, 0.0, 0.0, 1.0);
    } else if (distance(model_matrix.model_matrix * e.yyxy, vec4<f32>(9.0, 10.0, 11.0, 12.0)) > EPS) {
        out.position = vec4<f32>(0.0, 0.0, 0.0, 1.0);
    } else if (distance(model_matrix.model_matrix * e.yyyx, vec4<f32>(13.0, 14.0, 15.0, 16.0)) > EPS) {
        out.position = vec4<f32>(0.0, 0.0, 0.0, 1.0);  
    } else {
        out.position = vec4<f32>(in.uv, 0.0, 1.0);
        out.uv = in.uv;
    }
    return out;
}

@fragment
fn nontex_main(in: VertexOutput) -> @location(0) vec4<f32> {
    if (distance(model_matrix.model_matrix * e.xyyy, vec4<f32>(1.0, 2.0, 3.0, 4.0)) > EPS) {
        return vec4<f32>(0.0, 0.0, 1.0, 1.0);
    } else if (distance(model_matrix.model_matrix * e.yxyy, vec4<f32>(5.0, 6.0, 7.0, 8.0)) > EPS) {
        return vec4<f32>(0.0, 0.0, 1.0, 1.0);
    } else if (distance(model_matrix.model_matrix * e.yyxy, vec4<f32>(9.0, 10.0, 11.0, 12.0)) > EPS) {
        return vec4<f32>(0.0, 0.0, 1.0, 1.0);
    } else if (distance(model_matrix.model_matrix * e.yyyx, vec4<f32>(13.0, 14.0, 15.0, 16.0)) > EPS) {
        return vec4<f32>(0.0, 0.0, 1.0, 1.0); 
    } else if (distance(material.albedo, vec4<f32>(0.2, 0.4, 0.6, 1.0)) > EPS) {
        return vec4<f32>(1.0, 1.0, 0.0, 1.0);
    } else if (abs(material.roughness - 0.31415) > EPS) {
        return vec4<f32>(1.0, 0.0, 1.0, 1.0);
    } else if (abs(material.reflectance - 0.29613) > EPS) {
        return vec4<f32>(0.0, 1.0, 1.0, 1.0);
    } else if (abs(material.ambient_ratio - 0.92) > EPS) {
        return vec4<f32>(0.25, 0.25, 0.25, 1.0);
    } else if (abs(material.background_ratio - 0.32) > EPS) {
        return vec4<f32>(0.25, 0.25, 0.25, 1.0);
    } else {
        return vec4<f32>(0.2, 0.4, 0.6, 0.8);
    }
}

@fragment
fn nontex_main_anti(in: VertexOutput) -> @location(0) vec4<f32> {
    if (distance(model_matrix.model_matrix * e.xyyy, vec4<f32>(1.0, 2.0, 3.0, 4.0)) > EPS) {
        return vec4<f32>(0.0, 0.0, 1.0, 1.0);
    } else if (distance(model_matrix.model_matrix * e.yxyy, vec4<f32>(5.0, 6.0, 7.0, 8.0)) > EPS) {
        return vec4<f32>(0.0, 0.0, 1.0, 1.0);
    } else if (distance(model_matrix.model_matrix * e.yyxy, vec4<f32>(9.0, 10.0, 11.0, 12.0)) > EPS) {
        return vec4<f32>(0.0, 0.0, 1.0, 1.0);
    } else if (distance(model_matrix.model_matrix * e.yyyx, vec4<f32>(13.0, 14.0, 15.0, 16.0)) > EPS) {
        return vec4<f32>(0.0, 0.0, 1.0, 1.0); 
    } else if (distance(material.albedo, vec4<f32>(0.2, 0.4, 0.6, 1.0)) > EPS) {
        return vec4<f32>(1.0, 1.0, 0.0, 1.0);
    } else if (abs(material.roughness - 0.31415) > EPS) {
        return vec4<f32>(1.0, 0.0, 1.0, 1.0);
    } else if (abs(material.reflectance - 0.29613) < EPS) {
        return vec4<f32>(0.0, 1.0, 1.0, 1.0);
    } else if (abs(material.ambient_ratio - 0.92) > EPS) {
        return vec4<f32>(0.25, 0.25, 0.25, 1.0);
    } else if (abs(material.background_ratio - 0.32) > EPS) {
        return vec4<f32>(0.25, 0.25, 0.25, 1.0);
    } else {
        return vec4<f32>(0.2, 0.4, 0.6, 0.8);
    }
}

@fragment
fn tex_main(in: VertexOutput) -> @location(0) vec4<f32> {
    if (distance(model_matrix.model_matrix * e.xyyy, vec4<f32>(1.0, 2.0, 3.0, 4.0)) > EPS) {
        return vec4<f32>(0.0, 0.0, 1.0, 1.0);
    } else if (distance(model_matrix.model_matrix * e.yxyy, vec4<f32>(5.0, 6.0, 7.0, 8.0)) > EPS) {
        return vec4<f32>(0.0, 0.0, 1.0, 1.0);
    } else if (distance(model_matrix.model_matrix * e.yyxy, vec4<f32>(9.0, 10.0, 11.0, 12.0)) > EPS) {
        return vec4<f32>(0.0, 0.0, 1.0, 1.0);
    } else if (distance(model_matrix.model_matrix * e.yyyx, vec4<f32>(13.0, 14.0, 15.0, 16.0)) > EPS) {
        return vec4<f32>(0.0, 0.0, 1.0, 1.0); 
    } else if (distance(material.albedo, vec4<f32>(0.2, 0.4, 0.6, 1.0)) > EPS) {
        return vec4<f32>(1.0, 1.0, 0.0, 1.0);
    } else if (abs(material.roughness - 0.31415) > EPS) {
        return vec4<f32>(1.0, 0.0, 1.0, 1.0);
    } else if (abs(material.reflectance - 0.29613) > EPS) {
        return vec4<f32>(0.0, 1.0, 1.0, 1.0);
    } else if (abs(material.ambient_ratio - 0.92) > EPS) {
        return vec4<f32>(0.25, 0.25, 0.25, 1.0);
    } else if (abs(material.background_ratio - 0.32) > EPS) {
        return vec4<f32>(0.25, 0.25, 0.25, 1.0);
    } else {
        let uv = vec2<f32>(1.0 + in.uv.x, 1.0 - in.uv.y) / 2.0;
        return textureSample(r_color, r_sampler, uv);
    }
}

@fragment
fn tex_main_anti(in: VertexOutput) -> @location(0) vec4<f32> {
    if (distance(model_matrix.model_matrix * e.xyyy, vec4<f32>(1.0, 2.0, 3.0, 4.0)) > EPS) {
        return vec4<f32>(0.0, 0.0, 1.0, 1.0);
    } else if (distance(model_matrix.model_matrix * e.yxyy, vec4<f32>(5.0, 6.0, 7.0, 8.0)) > EPS) {
        return vec4<f32>(0.0, 0.0, 1.0, 1.0);
    } else if (distance(model_matrix.model_matrix * e.yyxy, vec4<f32>(9.0, 10.0, 11.0, 12.0)) > EPS) {
        return vec4<f32>(0.0, 0.0, 1.0, 1.0);
    } else if (distance(model_matrix.model_matrix * e.yyyx, vec4<f32>(13.0, 14.0, 15.0, 16.0)) > EPS) {
        return vec4<f32>(0.0, 0.0, 1.0, 1.0); 
    } else if (distance(material.albedo, vec4<f32>(0.2, 0.4, 0.6, 1.0)) > EPS) {
        return vec4<f32>(1.0, 1.0, 0.0, 1.0);
    } else if (abs(material.roughness - 0.31415) > EPS) {
        return vec4<f32>(1.0, 0.0, 1.0, 1.0);
    } else if (abs(material.reflectance - 0.29613) > EPS) {
        return vec4<f32>(0.0, 1.0, 1.0, 1.0);
    } else if (abs(material.ambient_ratio - 0.92) > EPS) {
        return vec4<f32>(0.25, 0.25, 0.25, 1.0);
    } else if (abs(material.background_ratio - 0.32) > EPS) {
        return vec4<f32>(0.25, 0.25, 0.25, 1.0);
    } else {
        let uv = vec2<f32>(1.0 + in.uv.x, 1.0 + in.uv.y) / 2.0;
        return textureSample(r_color, r_sampler, uv);
    }
}