wgpu-3dgs-viewer 0.7.0

A 3D Gaussian splatting viewing library written in Rust using wgpu.
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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65

use crate::{
    core::{self, ComputeBundle, ComputeBundleBuilder, GaussianPod},
    editor::SelectionBundle,
    shader,
};

/// The viewport selection bind group layout descriptor.
pub const VIEWPORT_BIND_GROUP_LAYOUT_DESCRIPTOR: wgpu::BindGroupLayoutDescriptor<'static> =
    wgpu::BindGroupLayoutDescriptor {
        label: Some("Viewport Selection Bind Group Layout"),
        entries: &[
            // Camera uniform buffer
            wgpu::BindGroupLayoutEntry {
                binding: 0,
                visibility: wgpu::ShaderStages::COMPUTE,
                ty: wgpu::BindingType::Buffer {
                    ty: wgpu::BufferBindingType::Uniform,
                    has_dynamic_offset: false,
                    min_binding_size: None,
                },
                count: None,
            },
            // Viewport selection texture
            wgpu::BindGroupLayoutEntry {
                binding: 1,
                visibility: wgpu::ShaderStages::COMPUTE,
                ty: wgpu::BindingType::Texture {
                    sample_type: wgpu::TextureSampleType::Float { filterable: false },
                    view_dimension: wgpu::TextureViewDimension::D2,
                    multisampled: false,
                },
                count: None,
            },
        ],
    };

/// Create a viewport selection operation.
///
/// - Bind group 0 is [`SelectionBundle::GAUSSIANS_BIND_GROUP_LAYOUT_DESCRIPTOR`].
/// - Bind group 1 is [`VIEWPORT_BIND_GROUP_LAYOUT_DESCRIPTOR`].
pub fn create_viewport_bundle<G: GaussianPod>(device: &wgpu::Device) -> ComputeBundle<()> {
    let mut resolver = wesl::PkgResolver::new();
    resolver.add_package(&core::shader::PACKAGE);
    resolver.add_package(&shader::PACKAGE);

    ComputeBundleBuilder::new()
        .label("Viewport Selection")
        .bind_group_layouts([
            &SelectionBundle::<G>::GAUSSIANS_BIND_GROUP_LAYOUT_DESCRIPTOR,
            &VIEWPORT_BIND_GROUP_LAYOUT_DESCRIPTOR,
        ])
        .main_shader(
            "wgpu_3dgs_viewer::selection::viewport"
                .parse()
                .expect("selection::viewport module path"),
        )
        .entry_point("main")
        .wesl_compile_options(wesl::CompileOptions {
            features: G::wesl_features(),
            ..Default::default()
        })
        .resolver(resolver)
        .build_without_bind_groups(device)
        .expect("viewport selection compute bundle")
}