anvilkit-render 0.1.0

Cross-platform rendering system built on wgpu and winit for AnvilKit game engine
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
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132

//! # 渲染状态共享资源
//!
//! 在 GPU 初始化后由 RenderApp 插入到 ECS World，
//! 供渲染系统读取表面信息和场景 Uniform。

use bevy_ecs::prelude::*;

/// GPU 端单个光源数据 (64 字节)
///
/// | 字段 | 含义 |
/// |------|------|
/// | position_type | xyz=位置(点光/聚光), w=类型 (0=方向光, 1=点光, 2=聚光) |
/// | direction_range | xyz=方向, w=衰减距离 |
/// | color_intensity | rgb=颜色(linear), w=强度 |
/// | params | x=inner_cone_cos, y=outer_cone_cos, z=0, w=0 |
#[repr(C)]
#[derive(Copy, Clone, Debug, bytemuck::Pod, bytemuck::Zeroable)]
pub struct GpuLight {
    pub position_type: [f32; 4],
    pub direction_range: [f32; 4],
    pub color_intensity: [f32; 4],
    pub params: [f32; 4],
}

impl Default for GpuLight {
    fn default() -> Self {
        Self {
            position_type: [0.0; 4],
            direction_range: [0.0, -1.0, 0.0, 0.0],
            color_intensity: [0.0; 4],
            params: [0.0; 4],
        }
    }
}

/// 最大光源数量
pub const MAX_LIGHTS: usize = 8;

/// PBR 场景 Uniform (848 字节)
///
/// 包含 per-object 变换、材质参数和多光源数据。
/// 前 256 字节与旧布局兼容（light_dir/light_color 保留但多光源路径不使用）。
#[repr(C)]
#[derive(Copy, Clone, bytemuck::Pod, bytemuck::Zeroable)]
pub struct PbrSceneUniform {
    pub model: [[f32; 4]; 4],           // 64 bytes
    pub view_proj: [[f32; 4]; 4],       // 64 bytes
    pub normal_matrix: [[f32; 4]; 4],   // 64 bytes
    pub camera_pos: [f32; 4],           // 16 bytes
    pub light_dir: [f32; 4],            // 16 bytes (legacy / lights[0] shortcut)
    pub light_color: [f32; 4],          // 16 bytes (legacy / lights[0] shortcut)
    pub material_params: [f32; 4],      // 16 bytes (metallic, roughness, normal_scale, light_count)
    // Multi-light array
    pub lights: [GpuLight; MAX_LIGHTS], // 512 bytes (8 * 64)
    // Shadow mapping
    pub shadow_view_proj: [[f32; 4]; 4], // 64 bytes
    // Emissive
    pub emissive_factor: [f32; 4],       // 16 bytes (rgb + 0)
}

impl Default for PbrSceneUniform {
    fn default() -> Self {
        Self {
            model: glam::Mat4::IDENTITY.to_cols_array_2d(),
            view_proj: glam::Mat4::IDENTITY.to_cols_array_2d(),
            normal_matrix: glam::Mat4::IDENTITY.to_cols_array_2d(),
            camera_pos: [0.0; 4],
            light_dir: [0.0, -1.0, 0.0, 0.0],
            light_color: [1.0, 1.0, 1.0, 3.0],
            material_params: [0.0, 0.5, 1.0, 0.0],
            lights: [GpuLight::default(); MAX_LIGHTS],
            shadow_view_proj: glam::Mat4::IDENTITY.to_cols_array_2d(),
            emissive_factor: [0.0; 4],
        }
    }
}

/// 共享渲染状态
///
/// 持有与 GPU 表面和场景 Uniform 相关的资源。
/// RenderApp 在 GPU 初始化后将其插入 World。
#[derive(Resource)]
pub struct RenderState {
    pub surface_format: wgpu::TextureFormat,
    pub surface_size: (u32, u32),
    pub scene_uniform_buffer: wgpu::Buffer,
    pub scene_bind_group: wgpu::BindGroup,
    pub scene_bind_group_layout: wgpu::BindGroupLayout,
    pub depth_texture_view: wgpu::TextureView,
    // HDR multi-pass rendering
    pub hdr_texture_view: wgpu::TextureView,
    pub tonemap_pipeline: wgpu::RenderPipeline,
    pub tonemap_bind_group: wgpu::BindGroup,
    pub tonemap_bind_group_layout: wgpu::BindGroupLayout,
    // IBL + Shadow (group 2)
    pub ibl_shadow_bind_group: wgpu::BindGroup,
    pub ibl_shadow_bind_group_layout: wgpu::BindGroupLayout,
    // Shadow pass
    pub shadow_pipeline: wgpu::RenderPipeline,
    pub shadow_map_view: wgpu::TextureView,
    // MSAA
    pub hdr_msaa_texture_view: wgpu::TextureView,
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_pbr_scene_uniform_size() {
        assert_eq!(std::mem::size_of::<PbrSceneUniform>(), 848);
    }

    #[test]
    fn test_gpu_light_size() {
        assert_eq!(std::mem::size_of::<GpuLight>(), 64);
    }

    #[test]
    fn test_pbr_scene_uniform_default() {
        let u = PbrSceneUniform::default();
        // model should be identity
        assert_eq!(u.model[0][0], 1.0);
        assert_eq!(u.model[1][1], 1.0);
        // default light direction points down
        assert_eq!(u.light_dir[1], -1.0);
        // default roughness = 0.5
        assert_eq!(u.material_params[1], 0.5);
        // default normal_scale = 1.0
        assert_eq!(u.material_params[2], 1.0);
    }
}