rust-gpu/tests/compiletests/ui/image/sample_gradient.rs at 972d0457299b16cefed4ff352ce4185b08b9aaf3 · Rust-GPU/rust-gpu · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
// Test `OpImageSampleExplicitLod` Grad
// build-pass

use spirv_std::spirv;
use spirv_std::{Image, Sampler, arch};

#[spirv(fragment)]
pub fn main(
    #[spirv(descriptor_set = 0, binding = 0)] image2d: &Image!(2D, type=f32, sampled),
    #[spirv(descriptor_set = 1, binding = 1)] image2d_array: &Image!(2D, type=f32, arrayed, sampled),
    #[spirv(descriptor_set = 2, binding = 2)] cubemap: &Image!(3D, type=f32, sampled),
    #[spirv(descriptor_set = 3, binding = 3)] sampler: &Sampler,
    output: &mut glam::Vec4,
) {
    let v2 = glam::Vec2::new(0.0, 1.0);
    let v3 = glam::Vec3::new(0.0, 1.0, 0.5);
    let r1: glam::Vec4 = image2d.sample_by_gradient(*sampler, v2, v2, v2);
    let r2: glam::Vec4 = image2d_array.sample_by_gradient(*sampler, v3, v2, v2);
    let r3: glam::Vec4 = cubemap.sample_by_gradient(*sampler, v3, v3, v3);
    *output = r1 + r2 + r3;
}