bevy_gaussian_splatting 2.4.2

bevy gaussian splatting render pipeline plugin
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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111

#define_import_path bevy_gaussian_splatting::bindings

#import bevy_render::globals::Globals
#import bevy_render::view::View


@group(0) @binding(0) var<uniform> view: View;
@group(0) @binding(1) var<uniform> globals: Globals;

struct GaussianUniforms {
    transform: mat4x4<f32>,
    global_scale: f32,
    count: u32,
    count_root_ceil: u32,
};
@group(1) @binding(0) var<uniform> gaussian_uniforms: GaussianUniforms;


#ifdef PACKED_F32
struct Gaussian {
    @location(0) rotation: vec4<f32>,
    @location(1) position_visibility: vec4<f32>,
    @location(2) scale_opacity: vec4<f32>,
    sh: array<f32, #{SH_COEFF_COUNT}>,
};

#ifdef READ_WRITE_POINTS
@group(2) @binding(0) var<storage, read_write> points: array<Gaussian>;
#else
@group(2) @binding(0) var<storage, read> points: array<Gaussian>;
#endif
#endif


#ifdef PLANAR_F32
#ifdef READ_WRITE_POINTS
@group(2) @binding(0) var<storage, read_write> position_visibility: array<vec4<f32>>;
#else
@group(2) @binding(0) var<storage, read> position_visibility: array<vec4<f32>>;
#endif

@group(2) @binding(1) var<storage, read> spherical_harmonics: array<array<f32, #{SH_COEFF_COUNT}>>;

#ifdef PRECOMPUTE_COVARIANCE_3D
@group(2) @binding(2) var<storage, read> covariance_3d_opacity: array<array<f32, 8>>;
#else
@group(2) @binding(2) var<storage, read> rotation: array<vec4<f32>>;
@group(2) @binding(3) var<storage, read> scale_opacity: array<vec4<f32>>;
#endif
#endif


#ifdef PLANAR_F16
#ifdef READ_WRITE_POINTS
@group(2) @binding(0) var<storage, read_write> position_visibility: array<vec4<f32>>;
#else
@group(2) @binding(0) var<storage, read> position_visibility: array<vec4<f32>>;
#endif
@group(2) @binding(1) var<storage, read> spherical_harmonics: array<array<u32, #{HALF_SH_COEFF_COUNT}>>;

#ifdef PRECOMPUTE_COVARIANCE_3D
@group(2) @binding(2) var<storage, read> covariance_3d_opacity: array<vec4<u32>>;
#else
@group(2) @binding(2) var<storage, read> rotation_scale_opacity: array<vec4<u32>>;
#endif
#endif


#ifdef PLANAR_TEXTURE_F16
@group(2) @binding(0) var position_visibility: texture_2d<f32>;

#if SH_VEC4_PLANES == 1
@group(2) @binding(1) var spherical_harmonics: texture_2d<u32>;
#else
@group(2) @binding(1) var spherical_harmonics: texture_2d_array<u32>;
#endif

#ifdef PRECOMPUTE_COVARIANCE_3D
@group(2) @binding(2) var covariance_3d_opacity: texture_2d<u32>;
#else
@group(2) @binding(2) var rotation_scale_opacity: texture_2d<u32>;
#endif
#endif


#ifdef PLANAR_TEXTURE_F32
@group(2) @binding(0) var position_visibility: texture_2d<f32>;

#if SH_VEC4_PLANES == 1
@group(2) @binding(1) var spherical_harmonics: texture_2d<f32>;
#else
@group(2) @binding(1) var spherical_harmonics: texture_2d_array<f32>;
#endif

// TODO: support f32_cov3d_opacity texture

@group(2) @binding(2) var rotation_scale_opacity: texture_2d<f32>;
#endif


struct DrawIndirect {
    vertex_count: u32,
    instance_count: atomic<u32>,
    base_vertex: u32,
    base_instance: u32,
}

struct Entry {
    key: u32,
    value: u32,
}