Skip to content

Commit

Permalink
Build only needed shaders with --cpu
Browse files Browse the repository at this point in the history
  • Loading branch information
@Zoxc
Zoxc committed Oct 12, 2023
1 parent 4edf786 commit 84103ac
Show file tree
Hide file tree
Showing 3 changed files with 179 additions and 131 deletions.
14 changes: 4 additions & 10 deletions src/lib.rs
View file
Original file line number Diff line number Diff line change
Expand Up @@ -116,11 +116,8 @@ pub struct RendererOptions {
impl Renderer {
/// Creates a new renderer for the specified device.
pub fn new(device: &Device, render_options: &RendererOptions) -> Result<Self> {
let mut engine = WgpuEngine::new();
let mut shaders = shaders::full_shaders(device, &mut engine)?;
if render_options.use_cpu {
shaders.install_cpu_shaders(&mut engine);
}
let mut engine = WgpuEngine::new(render_options.use_cpu);
let shaders = shaders::full_shaders(device, &mut engine)?;
let blit = render_options
.surface_format
.map(|surface_format| BlitPipeline::new(device, surface_format));
Expand Down Expand Up @@ -240,11 +237,8 @@ impl Renderer {
#[cfg(feature = "hot_reload")]
pub async fn reload_shaders(&mut self, device: &Device) -> Result<()> {
device.push_error_scope(wgpu::ErrorFilter::Validation);
let mut engine = WgpuEngine::new();
let mut shaders = shaders::full_shaders(device, &mut engine)?;
if self.use_cpu {
shaders.install_cpu_shaders(&mut engine);
}
let mut engine = WgpuEngine::new(self.use_cpu);
let shaders = shaders::full_shaders(device, &mut engine)?;
let error = device.pop_error_scope().await;
if let Some(error) = error {
return Err(error.into());
Expand Down
56 changes: 23 additions & 33 deletions src/shaders.rs
View file
Original file line number Diff line number Diff line change
Expand Up @@ -86,6 +86,7 @@ pub struct FullShaders {

#[cfg(feature = "wgpu")]
pub fn full_shaders(device: &Device, engine: &mut WgpuEngine) -> Result<FullShaders, Error> {
use crate::wgpu_engine::CpuShaderType;
use crate::ANTIALIASING;

let imports = SHARED_SHADERS
Expand Down Expand Up @@ -114,12 +115,14 @@ pub fn full_shaders(device: &Device, engine: &mut WgpuEngine) -> Result<FullShad
"pathtag_reduce",
preprocess::preprocess(shader!("pathtag_reduce"), &full_config, &imports).into(),
&[BindType::Uniform, BindType::BufReadOnly, BindType::Buffer],
CpuShaderType::Present(cpu_shader::pathtag_reduce),
)?;
let pathtag_reduce2 = engine.add_shader(
device,
"pathtag_reduce2",
preprocess::preprocess(shader!("pathtag_reduce2"), &full_config, &imports).into(),
&[BindType::BufReadOnly, BindType::Buffer],
CpuShaderType::Ununsed,
)?;
let pathtag_scan1 = engine.add_shader(
device,
Expand All @@ -130,6 +133,7 @@ pub fn full_shaders(device: &Device, engine: &mut WgpuEngine) -> Result<FullShad
BindType::BufReadOnly,
BindType::Buffer,
],
CpuShaderType::Ununsed,
)?;
let pathtag_scan = engine.add_shader(
device,
Expand All @@ -141,6 +145,7 @@ pub fn full_shaders(device: &Device, engine: &mut WgpuEngine) -> Result<FullShad
BindType::BufReadOnly,
BindType::Buffer,
],
CpuShaderType::Present(cpu_shader::pathtag_scan),
)?;
let pathtag_scan_large = engine.add_shader(
device,
Expand All @@ -152,12 +157,14 @@ pub fn full_shaders(device: &Device, engine: &mut WgpuEngine) -> Result<FullShad
BindType::BufReadOnly,
BindType::Buffer,
],
CpuShaderType::Ununsed,
)?;
let bbox_clear = engine.add_shader(
device,
"bbox_clear",
preprocess::preprocess(shader!("bbox_clear"), &empty, &imports).into(),
&[BindType::Uniform, BindType::Buffer],
CpuShaderType::Present(cpu_shader::bbox_clear),
)?;
let flatten = engine.add_shader(
device,
Expand All @@ -171,12 +178,14 @@ pub fn full_shaders(device: &Device, engine: &mut WgpuEngine) -> Result<FullShad
BindType::Buffer,
BindType::Buffer,
],
CpuShaderType::Present(cpu_shader::flatten),
)?;
let draw_reduce = engine.add_shader(
device,
"draw_reduce",
preprocess::preprocess(shader!("draw_reduce"), &empty, &imports).into(),
&[BindType::Uniform, BindType::BufReadOnly, BindType::Buffer],
CpuShaderType::Present(cpu_shader::draw_reduce),
)?;
let draw_leaf = engine.add_shader(
device,
Expand All @@ -191,6 +200,7 @@ pub fn full_shaders(device: &Device, engine: &mut WgpuEngine) -> Result<FullShad
BindType::Buffer,
BindType::Buffer,
],
CpuShaderType::Present(cpu_shader::draw_leaf),
)?;
let clip_reduce = engine.add_shader(
device,
Expand All @@ -202,6 +212,7 @@ pub fn full_shaders(device: &Device, engine: &mut WgpuEngine) -> Result<FullShad
BindType::Buffer,
BindType::Buffer,
],
CpuShaderType::Present(cpu_shader::clip_reduce),
)?;
let clip_leaf = engine.add_shader(
device,
Expand All @@ -216,6 +227,7 @@ pub fn full_shaders(device: &Device, engine: &mut WgpuEngine) -> Result<FullShad
BindType::Buffer,
BindType::Buffer,
],
CpuShaderType::Present(cpu_shader::clip_leaf),
)?;
let binning = engine.add_shader(
device,
Expand All @@ -231,6 +243,7 @@ pub fn full_shaders(device: &Device, engine: &mut WgpuEngine) -> Result<FullShad
BindType::Buffer,
BindType::Buffer,
],
CpuShaderType::Present(cpu_shader::binning),
)?;
let tile_alloc = engine.add_shader(
device,
Expand All @@ -244,12 +257,14 @@ pub fn full_shaders(device: &Device, engine: &mut WgpuEngine) -> Result<FullShad
BindType::Buffer,
BindType::Buffer,
],
CpuShaderType::Present(cpu_shader::tile_alloc),
)?;
let path_count_setup = engine.add_shader(
device,
"path_count_setup",
preprocess::preprocess(shader!("path_count_setup"), &empty, &imports).into(),
&[BindType::BufReadOnly, BindType::Buffer],
CpuShaderType::Present(cpu_shader::path_count_setup),
)?;
let path_count = engine.add_shader(
device,
Expand All @@ -263,12 +278,14 @@ pub fn full_shaders(device: &Device, engine: &mut WgpuEngine) -> Result<FullShad
BindType::Buffer,
BindType::Buffer,
],
CpuShaderType::Present(cpu_shader::path_count),
)?;
let backdrop = engine.add_shader(
device,
"backdrop_dyn",
preprocess::preprocess(shader!("backdrop_dyn"), &empty, &imports).into(),
&[BindType::Uniform, BindType::BufReadOnly, BindType::Buffer],
CpuShaderType::Present(cpu_shader::backdrop),
)?;
let coarse = engine.add_shader(
device,
Expand All @@ -285,12 +302,14 @@ pub fn full_shaders(device: &Device, engine: &mut WgpuEngine) -> Result<FullShad
BindType::Buffer,
BindType::Buffer,
],
CpuShaderType::Present(cpu_shader::coarse),
)?;
let path_tiling_setup = engine.add_shader(
device,
"path_tiling_setup",
preprocess::preprocess(shader!("path_tiling_setup"), &empty, &imports).into(),
&[BindType::BufReadOnly, BindType::Buffer],
CpuShaderType::Present(cpu_shader::path_tiling_setup),
)?;
let path_tiling = engine.add_shader(
device,
Expand All @@ -304,6 +323,7 @@ pub fn full_shaders(device: &Device, engine: &mut WgpuEngine) -> Result<FullShad
BindType::BufReadOnly,
BindType::Buffer,
],
CpuShaderType::Present(cpu_shader::path_tiling),
)?;
let fine = match ANTIALIASING {
crate::AaConfig::Area => engine.add_shader(
Expand All @@ -319,6 +339,7 @@ pub fn full_shaders(device: &Device, engine: &mut WgpuEngine) -> Result<FullShad
BindType::ImageRead(ImageFormat::Rgba8),
BindType::ImageRead(ImageFormat::Rgba8),
],
CpuShaderType::Missing,
)?,
_ => {
engine.add_shader(
Expand All @@ -335,6 +356,7 @@ pub fn full_shaders(device: &Device, engine: &mut WgpuEngine) -> Result<FullShad
BindType::ImageRead(ImageFormat::Rgba8),
BindType::BufReadOnly, // mask buffer
],
CpuShaderType::Missing,
)?
}
};
Expand All @@ -359,42 +381,10 @@ pub fn full_shaders(device: &Device, engine: &mut WgpuEngine) -> Result<FullShad
path_tiling_setup,
path_tiling,
fine,
pathtag_is_cpu: false,
pathtag_is_cpu: engine.use_cpu,
})
}

#[cfg(feature = "wgpu")]
impl FullShaders {
/// Install the CPU shaders.
///
/// There are a couple things to note here. The granularity provided by
/// this method is coarse; it installs all the shaders. There are many
/// use cases (including debugging), where a mix is desired, or the
/// choice between GPU and CPU dispatch might be dynamic.
///
/// Second, the actual mapping to CPU shaders is not really specific to
/// the engine, and should be split out into a back-end agnostic struct.
pub fn install_cpu_shaders(&mut self, engine: &mut WgpuEngine) {
engine.set_cpu_shader(self.pathtag_reduce, cpu_shader::pathtag_reduce);
engine.set_cpu_shader(self.pathtag_scan, cpu_shader::pathtag_scan);
engine.set_cpu_shader(self.bbox_clear, cpu_shader::bbox_clear);
engine.set_cpu_shader(self.flatten, cpu_shader::flatten);
engine.set_cpu_shader(self.draw_reduce, cpu_shader::draw_reduce);
engine.set_cpu_shader(self.draw_leaf, cpu_shader::draw_leaf);
engine.set_cpu_shader(self.clip_reduce, cpu_shader::clip_reduce);
engine.set_cpu_shader(self.clip_leaf, cpu_shader::clip_leaf);
engine.set_cpu_shader(self.binning, cpu_shader::binning);
engine.set_cpu_shader(self.tile_alloc, cpu_shader::tile_alloc);
engine.set_cpu_shader(self.path_count_setup, cpu_shader::path_count_setup);
engine.set_cpu_shader(self.path_count, cpu_shader::path_count);
engine.set_cpu_shader(self.backdrop, cpu_shader::backdrop);
engine.set_cpu_shader(self.coarse, cpu_shader::coarse);
engine.set_cpu_shader(self.path_tiling_setup, cpu_shader::path_tiling_setup);
engine.set_cpu_shader(self.path_tiling, cpu_shader::path_tiling);
self.pathtag_is_cpu = true;
}
}

macro_rules! shared_shader {
($name:expr) => {
(
Expand Down
Loading

0 comments on commit 84103ac

Please sign in to comment.