Make Drawer work with background pipeline creation

This commit is contained in:
Imbris 2021-04-25 11:25:32 -04:00 committed by Avi Weinstock
parent 71a561fc70
commit 604ea8ff18
5 changed files with 164 additions and 76 deletions

View File

@ -247,19 +247,21 @@ impl PlayState for CharSelectionState {
let (humanoid_body, loadout) = let (humanoid_body, loadout) =
Self::get_humanoid_body_inventory(&self.char_selection_ui, &client); Self::get_humanoid_body_inventory(&self.char_selection_ui, &client);
self.scene.render( if let Some(mut first_pass) = drawer.first_pass() {
&mut drawer.first_pass(), self.scene
client.get_tick(), .render(&mut first_pass, client.get_tick(), humanoid_body, loadout);
humanoid_body, }
loadout,
);
// Clouds // Clouds
drawer.second_pass().draw_clouds(); if let Some(mut second_pass) = drawer.second_pass() {
second_pass.draw_clouds();
}
// PostProcess and UI // PostProcess and UI
let mut third_pass = drawer.third_pass(); let mut third_pass = drawer.third_pass();
third_pass.draw_post_process(); third_pass.draw_postprocess();
// Draw the UI to the screen. // Draw the UI to the screen.
self.char_selection_ui.render(&mut third_pass.draw_ui()); if let Some(mut ui_drawer) = third_pass.draw_ui() {
self.char_selection_ui.render(&mut ui_drawer);
};
} }
} }

View File

@ -356,7 +356,9 @@ impl PlayState for MainMenuState {
}; };
// Draw the UI to the screen. // Draw the UI to the screen.
self.main_menu_ui.render(&mut drawer.third_pass().draw_ui()); if let Some(mut ui_drawer) = drawer.third_pass().draw_ui() {
self.main_menu_ui.render(&mut ui_drawer);
};
} }
} }

View File

@ -16,13 +16,47 @@ use std::sync::Arc;
use vek::Aabr; use vek::Aabr;
use wgpu_profiler::scope::{ManualOwningScope, OwningScope, Scope}; use wgpu_profiler::scope::{ManualOwningScope, OwningScope, Scope};
// Currently available pipelines
// #[derive(Clone, Copy)]
enum Pipelines<'frame> {
Interface(&'frame super::InterfacePipelines),
All(&'frame super::Pipelines),
// Should never be in this state for now but we need this to accound for super::State::Nothing
None,
}
impl<'frame> Pipelines<'frame> {
fn ui(&self) -> Option<&ui::UiPipeline> {
match self {
Pipelines::Interface(pipelines) => Some(&pipelines.ui),
Pipelines::All(pipelines) => Some(&pipelines.ui),
Pipelines::None => None,
}
}
fn blit(&self) -> Option<&blit::BlitPipeline> {
match self {
Pipelines::Interface(pipelines) => Some(&pipelines.blit),
Pipelines::All(pipelines) => Some(&pipelines.blit),
Pipelines::None => None,
}
}
fn all(&self) -> Option<&super::Pipelines> {
match self {
Pipelines::All(pipelines) => Some(pipelines),
Pipelines::Interface(_) | Pipelines::None => None,
}
}
}
// Borrow the fields we need from the renderer so that the GpuProfiler can be // Borrow the fields we need from the renderer so that the GpuProfiler can be
// dijointly borrowed mutably // dijointly borrowed mutably
struct RendererBorrow<'frame> { struct RendererBorrow<'frame> {
queue: &'frame wgpu::Queue, queue: &'frame wgpu::Queue,
device: &'frame wgpu::Device, device: &'frame wgpu::Device,
shadow: &'frame super::Shadow, shadow: Option<&'frame super::Shadow>,
pipelines: &'frame super::Pipelines, pipelines: Pipelines<'frame>,
locals: &'frame super::locals::Locals, locals: &'frame super::locals::Locals,
views: &'frame super::Views, views: &'frame super::Views,
mode: &'frame super::super::RenderMode, mode: &'frame super::super::RenderMode,
@ -57,11 +91,19 @@ impl<'frame> Drawer<'frame> {
) )
}); });
let (pipelines, shadow) = match &renderer.state {
super::State::Interface { pipelines, .. } => (Pipelines::Interface(pipelines), None),
super::State::Complete {
pipelines, shadow, ..
} => (Pipelines::All(pipelines), Some(shadow)),
super::State::Nothing => (Pipelines::None, None),
};
let borrow = RendererBorrow { let borrow = RendererBorrow {
queue: &renderer.queue, queue: &renderer.queue,
device: &renderer.device, device: &renderer.device,
shadow: todo!(), //&renderer.shadow, shadow,
pipelines: todo!(), //&renderer.pipelines, pipelines,
locals: &renderer.locals, locals: &renderer.locals,
views: &renderer.views, views: &renderer.views,
mode: &renderer.mode, mode: &renderer.mode,
@ -84,8 +126,10 @@ impl<'frame> Drawer<'frame> {
/// Get the render mode. /// Get the render mode.
pub fn render_mode(&self) -> &super::super::RenderMode { self.borrow.mode } pub fn render_mode(&self) -> &super::super::RenderMode { self.borrow.mode }
/// Returns None if the shadow renderer is not enabled or the pipelines are
/// not available yet
pub fn shadow_pass(&mut self) -> Option<ShadowPassDrawer> { pub fn shadow_pass(&mut self) -> Option<ShadowPassDrawer> {
if let ShadowMap::Enabled(ref shadow_renderer) = self.borrow.shadow.map { if let ShadowMap::Enabled(ref shadow_renderer) = self.borrow.shadow?.map {
let encoder = self.encoder.as_mut().unwrap(); let encoder = self.encoder.as_mut().unwrap();
let device = self.borrow.device; let device = self.borrow.device;
let mut render_pass = let mut render_pass =
@ -114,7 +158,13 @@ impl<'frame> Drawer<'frame> {
} }
} }
pub fn first_pass(&mut self) -> FirstPassDrawer { /// Returns None if all the pipelines are not available
pub fn first_pass(&mut self) -> Option<FirstPassDrawer> {
let pipelines = self.borrow.pipelines.all()?;
// Note: this becomes Some once pipeline creation is complete even if shadows
// are not enabled
let shadow = self.borrow.shadow?;
let encoder = self.encoder.as_mut().unwrap(); let encoder = self.encoder.as_mut().unwrap();
let device = self.borrow.device; let device = self.borrow.device;
let mut render_pass = let mut render_pass =
@ -139,16 +189,23 @@ impl<'frame> Drawer<'frame> {
}); });
render_pass.set_bind_group(0, &self.globals.bind_group, &[]); render_pass.set_bind_group(0, &self.globals.bind_group, &[]);
render_pass.set_bind_group(1, &self.borrow.shadow.bind.bind_group, &[]); render_pass.set_bind_group(1, &shadow.bind.bind_group, &[]);
FirstPassDrawer { Some(FirstPassDrawer {
render_pass, render_pass,
borrow: &self.borrow, borrow: &self.borrow,
pipelines,
globals: self.globals, globals: self.globals,
} })
} }
pub fn second_pass(&mut self) -> SecondPassDrawer { /// Returns None if the clouds pipeline is not available
pub fn second_pass(&mut self) -> Option<SecondPassDrawer> {
let pipeline = &self.borrow.pipelines.all()?.clouds;
// Note: this becomes Some once pipeline creation is complete even if shadows
// are not enabled
let shadow = self.borrow.shadow?;
let encoder = self.encoder.as_mut().unwrap(); let encoder = self.encoder.as_mut().unwrap();
let device = self.borrow.device; let device = self.borrow.device;
let mut render_pass = let mut render_pass =
@ -166,12 +223,14 @@ impl<'frame> Drawer<'frame> {
}); });
render_pass.set_bind_group(0, &self.globals.bind_group, &[]); render_pass.set_bind_group(0, &self.globals.bind_group, &[]);
render_pass.set_bind_group(1, &self.borrow.shadow.bind.bind_group, &[]); // TODO: what are shadows used for here????
render_pass.set_bind_group(1, &shadow.bind.bind_group, &[]);
SecondPassDrawer { Some(SecondPassDrawer {
render_pass, render_pass,
borrow: &self.borrow, borrow: &self.borrow,
} pipeline,
})
} }
pub fn third_pass(&mut self) -> ThirdPassDrawer { pub fn third_pass(&mut self) -> ThirdPassDrawer {
@ -204,13 +263,14 @@ impl<'frame> Drawer<'frame> {
} }
} }
/// Does nothing if the shadow pipelines are not available
pub fn draw_point_shadows<'data: 'frame>( pub fn draw_point_shadows<'data: 'frame>(
&mut self, &mut self,
matrices: &[shadow::PointLightMatrix; 126], matrices: &[shadow::PointLightMatrix; 126],
chunks: impl Clone chunks: impl Clone
+ Iterator<Item = (&'data Model<terrain::Vertex>, &'data terrain::BoundLocals)>, + Iterator<Item = (&'data Model<terrain::Vertex>, &'data terrain::BoundLocals)>,
) { ) {
if let ShadowMap::Enabled(ref shadow_renderer) = self.borrow.shadow.map { if let Some(ShadowMap::Enabled(ref shadow_renderer)) = self.borrow.shadow.map(|s| &s.map) {
let device = self.borrow.device; let device = self.borrow.device;
let mut encoder = self let mut encoder = self
.encoder .encoder
@ -280,8 +340,11 @@ impl<'frame> Drawer<'frame> {
/// NOTE: could simply use the above passes except `draw_point_shadows` /// NOTE: could simply use the above passes except `draw_point_shadows`
/// requires an array of matrices that could be a pain to construct /// requires an array of matrices that could be a pain to construct
/// simply for clearing /// simply for clearing
///
/// Does nothing if the shadow pipelines are not available (although they
/// aren't used here they are needed for the ShadowMap to exist)
pub fn clear_shadows(&mut self) { pub fn clear_shadows(&mut self) {
if let ShadowMap::Enabled(ref shadow_renderer) = self.borrow.shadow.map { if let Some(ShadowMap::Enabled(ref shadow_renderer)) = self.borrow.shadow.map(|s| &s.map) {
let device = self.borrow.device; let device = self.borrow.device;
let encoder = self.encoder.as_mut().unwrap(); let encoder = self.encoder.as_mut().unwrap();
let _ = encoder.scoped_render_pass( let _ = encoder.scoped_render_pass(
@ -342,8 +405,14 @@ impl<'frame> Drop for Drawer<'frame> {
// maybe we should submit each render pass to the queue as they are produced? // maybe we should submit each render pass to the queue as they are produced?
let mut encoder = self.encoder.take().unwrap(); let mut encoder = self.encoder.take().unwrap();
// If taking a screenshot // If taking a screenshota and the blit pipeline is available
if let Some(screenshot) = self.taking_screenshot.take() { // NOTE: blit pipeline should always be available for now so we don't report an
// error if it isn't
if let Some((screenshot, blit)) = self
.taking_screenshot
.take()
.zip(self.borrow.pipelines.blit())
{
// Image needs to be copied from the screenshot texture to the swapchain texture // Image needs to be copied from the screenshot texture to the swapchain texture
let mut render_pass = encoder.scoped_render_pass( let mut render_pass = encoder.scoped_render_pass(
"screenshot blit", "screenshot blit",
@ -361,7 +430,7 @@ impl<'frame> Drop for Drawer<'frame> {
depth_stencil_attachment: None, depth_stencil_attachment: None,
}, },
); );
render_pass.set_pipeline(&self.borrow.pipelines.blit.pipeline); render_pass.set_pipeline(&blit.pipeline);
render_pass.set_bind_group(0, &screenshot.bind_group(), &[]); render_pass.set_bind_group(0, &screenshot.bind_group(), &[]);
render_pass.draw(0..3, 0..1); render_pass.draw(0..3, 0..1);
drop(render_pass); drop(render_pass);
@ -450,6 +519,7 @@ impl<'pass_ref, 'pass: 'pass_ref> TerrainShadowDrawer<'pass_ref, 'pass> {
pub struct FirstPassDrawer<'pass> { pub struct FirstPassDrawer<'pass> {
pub(super) render_pass: OwningScope<'pass, wgpu::RenderPass<'pass>>, pub(super) render_pass: OwningScope<'pass, wgpu::RenderPass<'pass>>,
borrow: &'pass RendererBorrow<'pass>, borrow: &'pass RendererBorrow<'pass>,
pipelines: &'pass super::Pipelines,
globals: &'pass GlobalsBindGroup, globals: &'pass GlobalsBindGroup,
} }
@ -457,7 +527,7 @@ impl<'pass> FirstPassDrawer<'pass> {
pub fn draw_skybox<'data: 'pass>(&mut self, model: &'data Model<skybox::Vertex>) { pub fn draw_skybox<'data: 'pass>(&mut self, model: &'data Model<skybox::Vertex>) {
let mut render_pass = self.render_pass.scope("skybox", self.borrow.device); let mut render_pass = self.render_pass.scope("skybox", self.borrow.device);
render_pass.set_pipeline(&self.borrow.pipelines.skybox.pipeline); render_pass.set_pipeline(&self.pipelines.skybox.pipeline);
set_quad_index_buffer::<skybox::Vertex>(&mut render_pass, &self.borrow); set_quad_index_buffer::<skybox::Vertex>(&mut render_pass, &self.borrow);
render_pass.set_vertex_buffer(0, model.buf().slice(..)); render_pass.set_vertex_buffer(0, model.buf().slice(..));
render_pass.draw(0..model.len() as u32, 0..1); render_pass.draw(0..model.len() as u32, 0..1);
@ -466,7 +536,7 @@ impl<'pass> FirstPassDrawer<'pass> {
pub fn draw_lod_terrain<'data: 'pass>(&mut self, model: &'data Model<lod_terrain::Vertex>) { pub fn draw_lod_terrain<'data: 'pass>(&mut self, model: &'data Model<lod_terrain::Vertex>) {
let mut render_pass = self.render_pass.scope("lod_terrain", self.borrow.device); let mut render_pass = self.render_pass.scope("lod_terrain", self.borrow.device);
render_pass.set_pipeline(&self.borrow.pipelines.lod_terrain.pipeline); render_pass.set_pipeline(&self.pipelines.lod_terrain.pipeline);
set_quad_index_buffer::<lod_terrain::Vertex>(&mut render_pass, &self.borrow); set_quad_index_buffer::<lod_terrain::Vertex>(&mut render_pass, &self.borrow);
render_pass.set_vertex_buffer(0, model.buf().slice(..)); render_pass.set_vertex_buffer(0, model.buf().slice(..));
render_pass.draw_indexed(0..model.len() as u32 / 4 * 6, 0, 0..1); render_pass.draw_indexed(0..model.len() as u32 / 4 * 6, 0, 0..1);
@ -475,7 +545,7 @@ impl<'pass> FirstPassDrawer<'pass> {
pub fn draw_figures(&mut self) -> FigureDrawer<'_, 'pass> { pub fn draw_figures(&mut self) -> FigureDrawer<'_, 'pass> {
let mut render_pass = self.render_pass.scope("figures", self.borrow.device); let mut render_pass = self.render_pass.scope("figures", self.borrow.device);
render_pass.set_pipeline(&self.borrow.pipelines.figure.pipeline); render_pass.set_pipeline(&self.pipelines.figure.pipeline);
set_quad_index_buffer::<terrain::Vertex>(&mut render_pass, &self.borrow); set_quad_index_buffer::<terrain::Vertex>(&mut render_pass, &self.borrow);
FigureDrawer { render_pass } FigureDrawer { render_pass }
@ -484,7 +554,7 @@ impl<'pass> FirstPassDrawer<'pass> {
pub fn draw_terrain<'data: 'pass>(&mut self) -> TerrainDrawer<'_, 'pass> { pub fn draw_terrain<'data: 'pass>(&mut self) -> TerrainDrawer<'_, 'pass> {
let mut render_pass = self.render_pass.scope("terrain", self.borrow.device); let mut render_pass = self.render_pass.scope("terrain", self.borrow.device);
render_pass.set_pipeline(&self.borrow.pipelines.terrain.pipeline); render_pass.set_pipeline(&self.pipelines.terrain.pipeline);
set_quad_index_buffer::<terrain::Vertex>(&mut render_pass, &self.borrow); set_quad_index_buffer::<terrain::Vertex>(&mut render_pass, &self.borrow);
TerrainDrawer { TerrainDrawer {
@ -496,7 +566,7 @@ impl<'pass> FirstPassDrawer<'pass> {
pub fn draw_particles(&mut self) -> ParticleDrawer<'_, 'pass> { pub fn draw_particles(&mut self) -> ParticleDrawer<'_, 'pass> {
let mut render_pass = self.render_pass.scope("particles", self.borrow.device); let mut render_pass = self.render_pass.scope("particles", self.borrow.device);
render_pass.set_pipeline(&self.borrow.pipelines.particle.pipeline); render_pass.set_pipeline(&self.pipelines.particle.pipeline);
set_quad_index_buffer::<particle::Vertex>(&mut render_pass, &self.borrow); set_quad_index_buffer::<particle::Vertex>(&mut render_pass, &self.borrow);
ParticleDrawer { render_pass } ParticleDrawer { render_pass }
@ -509,7 +579,7 @@ impl<'pass> FirstPassDrawer<'pass> {
) -> SpriteDrawer<'_, 'pass> { ) -> SpriteDrawer<'_, 'pass> {
let mut render_pass = self.render_pass.scope("sprites", self.borrow.device); let mut render_pass = self.render_pass.scope("sprites", self.borrow.device);
render_pass.set_pipeline(&self.borrow.pipelines.sprite.pipeline); render_pass.set_pipeline(&self.pipelines.sprite.pipeline);
set_quad_index_buffer::<particle::Vertex>(&mut render_pass, &self.borrow); set_quad_index_buffer::<particle::Vertex>(&mut render_pass, &self.borrow);
render_pass.set_bind_group(0, &globals.bind_group, &[]); render_pass.set_bind_group(0, &globals.bind_group, &[]);
render_pass.set_bind_group(3, &col_lights.bind_group, &[]); render_pass.set_bind_group(3, &col_lights.bind_group, &[]);
@ -526,7 +596,7 @@ impl<'pass> FirstPassDrawer<'pass> {
) -> FluidDrawer<'_, 'pass> { ) -> FluidDrawer<'_, 'pass> {
let mut render_pass = self.render_pass.scope("fluid", self.borrow.device); let mut render_pass = self.render_pass.scope("fluid", self.borrow.device);
render_pass.set_pipeline(&self.borrow.pipelines.fluid.pipeline); render_pass.set_pipeline(&self.pipelines.fluid.pipeline);
set_quad_index_buffer::<fluid::Vertex>(&mut render_pass, &self.borrow); set_quad_index_buffer::<fluid::Vertex>(&mut render_pass, &self.borrow);
render_pass.set_bind_group(2, &waves.bind_group, &[]); render_pass.set_bind_group(2, &waves.bind_group, &[]);
@ -666,38 +736,48 @@ impl<'pass_ref, 'pass: 'pass_ref> FluidDrawer<'pass_ref, 'pass> {
pub struct SecondPassDrawer<'pass> { pub struct SecondPassDrawer<'pass> {
render_pass: OwningScope<'pass, wgpu::RenderPass<'pass>>, render_pass: OwningScope<'pass, wgpu::RenderPass<'pass>>,
borrow: &'pass RendererBorrow<'pass>, borrow: &'pass RendererBorrow<'pass>,
pipeline: &'pass clouds::CloudsPipeline,
} }
impl<'pass> SecondPassDrawer<'pass> { impl<'pass> SecondPassDrawer<'pass> {
pub fn draw_clouds(&mut self) { pub fn draw_clouds(&mut self) {
self.render_pass self.render_pass.set_pipeline(&self.pipeline.pipeline);
.set_pipeline(&self.borrow.pipelines.clouds.pipeline);
self.render_pass self.render_pass
.set_bind_group(2, &self.borrow.locals.clouds_bind.bind_group, &[]); .set_bind_group(2, &self.borrow.locals.clouds_bind.bind_group, &[]);
self.render_pass.draw(0..3, 0..1); self.render_pass.draw(0..3, 0..1);
} }
} }
// Third pass: postprocess + ui /// Third pass: postprocess + ui
pub struct ThirdPassDrawer<'pass> { pub struct ThirdPassDrawer<'pass> {
render_pass: OwningScope<'pass, wgpu::RenderPass<'pass>>, render_pass: OwningScope<'pass, wgpu::RenderPass<'pass>>,
borrow: &'pass RendererBorrow<'pass>, borrow: &'pass RendererBorrow<'pass>,
} }
impl<'pass> ThirdPassDrawer<'pass> { impl<'pass> ThirdPassDrawer<'pass> {
pub fn draw_post_process(&mut self) { /// Does nothing if the postprocess pipeline is not available
pub fn draw_postprocess(&mut self) {
let postprocess = match self.borrow.pipelines.all() {
Some(p) => &p.postprocess,
None => return,
};
let mut render_pass = self.render_pass.scope("postprocess", self.borrow.device); let mut render_pass = self.render_pass.scope("postprocess", self.borrow.device);
render_pass.set_pipeline(&self.borrow.pipelines.postprocess.pipeline); render_pass.set_pipeline(&postprocess.pipeline);
render_pass.set_bind_group(1, &self.borrow.locals.postprocess_bind.bind_group, &[]); render_pass.set_bind_group(1, &self.borrow.locals.postprocess_bind.bind_group, &[]);
render_pass.draw(0..3, 0..1); render_pass.draw(0..3, 0..1);
} }
pub fn draw_ui(&mut self) -> UiDrawer<'_, 'pass> { /// Returns None if the UI pipeline is not available (note: this should
/// never be the case for now)
pub fn draw_ui(&mut self) -> Option<UiDrawer<'_, 'pass>> {
let ui = self.borrow.pipelines.ui()?;
let mut render_pass = self.render_pass.scope("ui", self.borrow.device); let mut render_pass = self.render_pass.scope("ui", self.borrow.device);
render_pass.set_pipeline(&self.borrow.pipelines.ui.pipeline); render_pass.set_pipeline(&ui.pipeline);
set_quad_index_buffer::<ui::Vertex>(&mut render_pass, &self.borrow); set_quad_index_buffer::<ui::Vertex>(&mut render_pass, &self.borrow);
UiDrawer { render_pass } Some(UiDrawer { render_pass })
} }
} }

View File

@ -1074,41 +1074,41 @@ impl Scene {
) )
} }
let mut first_pass = drawer.first_pass(); if let Some(mut first_pass) = drawer.first_pass() {
self.figure_mgr.render_player(
&mut first_pass.draw_figures(),
state,
player_entity,
tick,
camera_data,
);
self.figure_mgr.render_player( self.terrain.render(&mut first_pass, focus_pos);
&mut first_pass.draw_figures(),
state,
player_entity,
tick,
camera_data,
);
self.terrain.render(&mut first_pass, focus_pos); self.figure_mgr.render(
&mut first_pass.draw_figures(),
state,
player_entity,
tick,
camera_data,
);
self.figure_mgr.render( self.lod.render(&mut first_pass);
&mut first_pass.draw_figures(),
state,
player_entity,
tick,
camera_data,
);
self.lod.render(&mut first_pass); // Render the skybox.
first_pass.draw_skybox(&self.skybox.model);
// Render the skybox. // Draws translucent terrain and sprites
first_pass.draw_skybox(&self.skybox.model); self.terrain.render_translucent(
&mut first_pass,
focus_pos,
cam_pos,
scene_data.sprite_render_distance,
);
// Draws translucent terrain and sprites // Render particle effects.
self.terrain.render_translucent( self.particle_mgr
&mut first_pass, .render(&mut first_pass.draw_particles(), scene_data);
focus_pos, }
cam_pos,
scene_data.sprite_render_distance,
);
// Render particle effects.
self.particle_mgr
.render(&mut first_pass.draw_particles(), scene_data);
} }
} }

View File

@ -1429,12 +1429,16 @@ impl PlayState for SessionState {
} }
// Clouds // Clouds
drawer.second_pass().draw_clouds(); if let Some(mut second_pass) = drawer.second_pass() {
second_pass.draw_clouds();
}
// PostProcess and UI // PostProcess and UI
let mut third_pass = drawer.third_pass(); let mut third_pass = drawer.third_pass();
third_pass.draw_post_process(); third_pass.draw_postprocess();
// Draw the UI to the screen // Draw the UI to the screen
self.hud.render(&mut third_pass.draw_ui()); if let Some(mut ui_drawer) = third_pass.draw_ui() {
self.hud.render(&mut ui_drawer);
}; // Note: this semicolon is needed for the third_pass borrow to be dropped before it's lifetime ends
} }
} }