molgfx-render 0.1.0

The render graph, passes and the engine that drives a frame.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41

//! Fixed-step visual advection for caller-supplied particle samples.

use crate::error::RenderError;
use molgfx_gpu::{
    CommandEncoder as _, ComputePassDesc, ComputePassEncoder as _, ComputePipelineDesc, Device,
    ShaderModuleDesc,
};

const WORKGROUP_SIZE: u32 = 64;

#[derive(Debug)]
pub(crate) struct ParticleMotionPass<D: Device> {
    pipeline: D::Pipeline,
}

impl<D: Device> ParticleMotionPass<D> {
    pub(crate) fn new(device: &D, primitive: &D::BindGroupLayout) -> Result<Self, RenderError> {
        let shader = device.create_shader_module(&ShaderModuleDesc {
            label: "fixed-step particle advection",
            wgsl: molgfx_shaders::PARTICLE_ADVECTION,
        })?;
        Ok(Self {
            pipeline: device.create_compute_pipeline(&ComputePipelineDesc {
                label: "fixed-step particle advection",
                layouts: &[None, None, Some(primitive)],
                shader: &shader,
                entry: "advect_particles",
            })?,
        })
    }

    pub(crate) fn record(&self, encoder: &mut D::CommandEncoder, group: &D::BindGroup, count: u32) {
        let mut pass = encoder.begin_compute_pass(&ComputePassDesc {
            label: "fixed-step particle advection",
            timestamps: None,
        });
        pass.set_pipeline(&self.pipeline);
        pass.set_bind_group(2, group, &[]);
        pass.dispatch(count.div_ceil(WORKGROUP_SIZE), 1, 1);
    }
}