mirror of
https://github.com/italicsjenga/vello.git
synced 2025-02-24 00:27:43 +11:00
1dea6c0ef0
Fixed several other shader validation errors caught when running vello_shaders natively on Metal. These were primarily caused by reading an invalid drawtag while accessing the scene buffer. Scene buffer access in the offending pipelines now initialize the draw tag to DRAWTAG_NOP if an invocation ID would land beyond the valid index range of encoded draw objects.
323 lines
9.4 KiB
Rust
323 lines
9.4 KiB
Rust
// Copyright 2022 Google LLC
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// https://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
//
|
|
// Also licensed under MIT license, at your choice.
|
|
|
|
//! Load rendering shaders.
|
|
|
|
mod preprocess;
|
|
|
|
use std::collections::HashSet;
|
|
|
|
use wgpu::Device;
|
|
|
|
use crate::engine::{BindType, Engine, Error, ImageFormat, ShaderId};
|
|
|
|
macro_rules! shader {
|
|
($name:expr) => {&{
|
|
let shader = include_str!(concat!(
|
|
env!("CARGO_MANIFEST_DIR"),
|
|
"/shader/",
|
|
$name,
|
|
".wgsl"
|
|
));
|
|
#[cfg(feature = "hot_reload")]
|
|
let shader = std::fs::read_to_string(concat!(
|
|
env!("CARGO_MANIFEST_DIR"),
|
|
"/shader/",
|
|
$name,
|
|
".wgsl"
|
|
))
|
|
.unwrap_or_else(|e| {
|
|
eprintln!(
|
|
"Failed to read shader {name}, error falling back to version at compilation time. Error: {e:?}",
|
|
name = $name
|
|
);
|
|
shader.to_string()
|
|
});
|
|
shader
|
|
}};
|
|
}
|
|
|
|
// Shaders for the full pipeline
|
|
pub struct FullShaders {
|
|
pub pathtag_reduce: ShaderId,
|
|
pub pathtag_reduce2: ShaderId,
|
|
pub pathtag_scan1: ShaderId,
|
|
pub pathtag_scan: ShaderId,
|
|
pub pathtag_scan_large: ShaderId,
|
|
pub bbox_clear: ShaderId,
|
|
pub pathseg: ShaderId,
|
|
pub draw_reduce: ShaderId,
|
|
pub draw_leaf: ShaderId,
|
|
pub clip_reduce: ShaderId,
|
|
pub clip_leaf: ShaderId,
|
|
pub binning: ShaderId,
|
|
pub tile_alloc: ShaderId,
|
|
pub path_coarse: ShaderId,
|
|
pub backdrop: ShaderId,
|
|
pub coarse: ShaderId,
|
|
pub fine: ShaderId,
|
|
}
|
|
|
|
pub fn full_shaders(device: &Device, engine: &mut Engine) -> Result<FullShaders, Error> {
|
|
let imports = SHARED_SHADERS
|
|
.iter()
|
|
.copied()
|
|
.collect::<std::collections::HashMap<_, _>>();
|
|
let empty = HashSet::new();
|
|
let mut full_config = HashSet::new();
|
|
full_config.insert("full".into());
|
|
let mut small_config = HashSet::new();
|
|
small_config.insert("full".into());
|
|
small_config.insert("small".into());
|
|
// TODO: remove this workaround when workgroupUniformLoad lands in naga
|
|
#[allow(unused_mut)]
|
|
let mut uniform = HashSet::new();
|
|
#[cfg(target_arch = "wasm32")]
|
|
uniform.insert("have_uniform".into());
|
|
let pathtag_reduce = engine.add_shader(
|
|
device,
|
|
"pathtag_reduce",
|
|
preprocess::preprocess(shader!("pathtag_reduce"), &full_config, &imports).into(),
|
|
&[BindType::Uniform, BindType::BufReadOnly, BindType::Buffer],
|
|
)?;
|
|
let pathtag_reduce2 = engine.add_shader(
|
|
device,
|
|
"pathtag_reduce2",
|
|
preprocess::preprocess(shader!("pathtag_reduce2"), &full_config, &imports).into(),
|
|
&[BindType::BufReadOnly, BindType::Buffer],
|
|
)?;
|
|
let pathtag_scan1 = engine.add_shader(
|
|
device,
|
|
"pathtag_scan1",
|
|
preprocess::preprocess(shader!("pathtag_scan1"), &full_config, &imports).into(),
|
|
&[
|
|
BindType::BufReadOnly,
|
|
BindType::BufReadOnly,
|
|
BindType::Buffer,
|
|
],
|
|
)?;
|
|
let pathtag_scan = engine.add_shader(
|
|
device,
|
|
"pathtag_scan",
|
|
preprocess::preprocess(shader!("pathtag_scan"), &small_config, &imports).into(),
|
|
&[
|
|
BindType::Uniform,
|
|
BindType::BufReadOnly,
|
|
BindType::BufReadOnly,
|
|
BindType::Buffer,
|
|
],
|
|
)?;
|
|
let pathtag_scan_large = engine.add_shader(
|
|
device,
|
|
"pathtag_scan",
|
|
preprocess::preprocess(shader!("pathtag_scan"), &full_config, &imports).into(),
|
|
&[
|
|
BindType::Uniform,
|
|
BindType::BufReadOnly,
|
|
BindType::BufReadOnly,
|
|
BindType::Buffer,
|
|
],
|
|
)?;
|
|
let bbox_clear = engine.add_shader(
|
|
device,
|
|
"bbox_clear",
|
|
preprocess::preprocess(shader!("bbox_clear"), &empty, &imports).into(),
|
|
&[BindType::Uniform, BindType::Buffer],
|
|
)?;
|
|
let pathseg = engine.add_shader(
|
|
device,
|
|
"pathseg",
|
|
preprocess::preprocess(shader!("pathseg"), &full_config, &imports).into(),
|
|
&[
|
|
BindType::Uniform,
|
|
BindType::BufReadOnly,
|
|
BindType::BufReadOnly,
|
|
BindType::Buffer,
|
|
BindType::Buffer,
|
|
],
|
|
)?;
|
|
let draw_reduce = engine.add_shader(
|
|
device,
|
|
"draw_reduce",
|
|
preprocess::preprocess(shader!("draw_reduce"), &empty, &imports).into(),
|
|
&[BindType::Uniform, BindType::BufReadOnly, BindType::Buffer],
|
|
)?;
|
|
let draw_leaf = engine.add_shader(
|
|
device,
|
|
"draw_leaf",
|
|
preprocess::preprocess(shader!("draw_leaf"), &empty, &imports).into(),
|
|
&[
|
|
BindType::Uniform,
|
|
BindType::BufReadOnly,
|
|
BindType::BufReadOnly,
|
|
BindType::BufReadOnly,
|
|
BindType::Buffer,
|
|
BindType::Buffer,
|
|
BindType::Buffer,
|
|
],
|
|
)?;
|
|
let clip_reduce = engine.add_shader(
|
|
device,
|
|
"clip_reduce",
|
|
preprocess::preprocess(shader!("clip_reduce"), &empty, &imports).into(),
|
|
&[
|
|
BindType::Uniform,
|
|
BindType::BufReadOnly,
|
|
BindType::BufReadOnly,
|
|
BindType::Buffer,
|
|
BindType::Buffer,
|
|
],
|
|
)?;
|
|
let clip_leaf = engine.add_shader(
|
|
device,
|
|
"clip_leaf",
|
|
preprocess::preprocess(shader!("clip_leaf"), &empty, &imports).into(),
|
|
&[
|
|
BindType::Uniform,
|
|
BindType::BufReadOnly,
|
|
BindType::BufReadOnly,
|
|
BindType::BufReadOnly,
|
|
BindType::BufReadOnly,
|
|
BindType::Buffer,
|
|
BindType::Buffer,
|
|
],
|
|
)?;
|
|
let binning = engine.add_shader(
|
|
device,
|
|
"binning",
|
|
preprocess::preprocess(shader!("binning"), &empty, &imports).into(),
|
|
&[
|
|
BindType::Uniform,
|
|
BindType::BufReadOnly,
|
|
BindType::BufReadOnly,
|
|
BindType::BufReadOnly,
|
|
BindType::Buffer,
|
|
BindType::Buffer,
|
|
BindType::Buffer,
|
|
BindType::Buffer,
|
|
],
|
|
)?;
|
|
let tile_alloc = engine.add_shader(
|
|
device,
|
|
"tile_alloc",
|
|
preprocess::preprocess(shader!("tile_alloc"), &uniform, &imports).into(),
|
|
&[
|
|
BindType::Uniform,
|
|
BindType::BufReadOnly,
|
|
BindType::BufReadOnly,
|
|
BindType::Buffer,
|
|
BindType::Buffer,
|
|
BindType::Buffer,
|
|
],
|
|
)?;
|
|
|
|
let path_coarse = engine.add_shader(
|
|
device,
|
|
"path_coarse_full",
|
|
preprocess::preprocess(shader!("path_coarse_full"), &full_config, &imports).into(),
|
|
&[
|
|
BindType::Uniform,
|
|
BindType::BufReadOnly,
|
|
BindType::BufReadOnly,
|
|
BindType::BufReadOnly,
|
|
BindType::BufReadOnly,
|
|
BindType::Buffer,
|
|
BindType::Buffer,
|
|
BindType::Buffer,
|
|
],
|
|
)?;
|
|
let backdrop = engine.add_shader(
|
|
device,
|
|
"backdrop_dyn",
|
|
preprocess::preprocess(shader!("backdrop_dyn"), &empty, &imports).into(),
|
|
&[BindType::Uniform, BindType::BufReadOnly, BindType::Buffer],
|
|
)?;
|
|
let coarse = engine.add_shader(
|
|
device,
|
|
"coarse",
|
|
preprocess::preprocess(shader!("coarse"), &uniform, &imports).into(),
|
|
&[
|
|
BindType::Uniform,
|
|
BindType::BufReadOnly,
|
|
BindType::BufReadOnly,
|
|
BindType::BufReadOnly,
|
|
BindType::BufReadOnly,
|
|
BindType::BufReadOnly,
|
|
BindType::BufReadOnly,
|
|
BindType::Buffer,
|
|
BindType::Buffer,
|
|
],
|
|
)?;
|
|
let fine = engine.add_shader(
|
|
device,
|
|
"fine",
|
|
preprocess::preprocess(shader!("fine"), &full_config, &imports).into(),
|
|
&[
|
|
BindType::Uniform,
|
|
BindType::BufReadOnly,
|
|
BindType::BufReadOnly,
|
|
BindType::Image(ImageFormat::Rgba8),
|
|
BindType::BufReadOnly,
|
|
BindType::ImageRead(ImageFormat::Rgba8),
|
|
BindType::BufReadOnly,
|
|
BindType::ImageRead(ImageFormat::Rgba8),
|
|
],
|
|
)?;
|
|
Ok(FullShaders {
|
|
pathtag_reduce,
|
|
pathtag_reduce2,
|
|
pathtag_scan,
|
|
pathtag_scan1,
|
|
pathtag_scan_large,
|
|
bbox_clear,
|
|
pathseg,
|
|
draw_reduce,
|
|
draw_leaf,
|
|
clip_reduce,
|
|
clip_leaf,
|
|
binning,
|
|
tile_alloc,
|
|
path_coarse,
|
|
backdrop,
|
|
coarse,
|
|
fine,
|
|
})
|
|
}
|
|
|
|
macro_rules! shared_shader {
|
|
($name:expr) => {
|
|
(
|
|
$name,
|
|
include_str!(concat!("../shader/shared/", $name, ".wgsl")),
|
|
)
|
|
};
|
|
}
|
|
|
|
const SHARED_SHADERS: &[(&str, &str)] = &[
|
|
shared_shader!("bbox"),
|
|
shared_shader!("blend"),
|
|
shared_shader!("bump"),
|
|
shared_shader!("clip"),
|
|
shared_shader!("config"),
|
|
shared_shader!("cubic"),
|
|
shared_shader!("drawtag"),
|
|
shared_shader!("pathtag"),
|
|
shared_shader!("ptcl"),
|
|
shared_shader!("segment"),
|
|
shared_shader!("tile"),
|
|
shared_shader!("transform"),
|
|
shared_shader!("util"),
|
|
];
|