aluminium 0.1.0

vulkan-based graphics engine
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

use ash::vk;
use bytemuck::{Pod, Zeroable};

use crate::core::{Device, GpuBuffer, GpuBufferBuilder, VulkanResult};
use crate::per_frame::{PerFrameBuffer, PerFrameBufferBuilder};

#[repr(C)]
#[derive(Clone, Copy, Pod, Zeroable)]
pub struct CameraData {
    view: [[f32; 4]; 4],
    proj: [[f32; 4]; 4],
    view_proj: [[f32; 4]; 4],
    inv_view: [[f32; 4]; 4],
    inv_proj: [[f32; 4]; 4],
    inv_view_proj: [[f32; 4]; 4],
}

impl CameraData {
    pub fn identity() -> Self {
        Self {
            view: [[1.0, 0.0, 0.0, 0.0], [0.0, 1.0, 0.0, 0.0], [0.0, 0.0, 1.0, 0.0], [0.0, 0.0, 0.0, 1.0]],
            proj: [[1.0, 0.0, 0.0, 0.0], [0.0, 1.0, 0.0, 0.0], [0.0, 0.0, 1.0, 0.0], [0.0, 0.0, 0.0, 1.0]],
            view_proj: [[1.0, 0.0, 0.0, 0.0], [0.0, 1.0, 0.0, 0.0], [0.0, 0.0, 1.0, 0.0], [0.0, 0.0, 0.0, 1.0]],
            inv_view: [[1.0, 0.0, 0.0, 0.0], [0.0, 1.0, 0.0, 0.0], [0.0, 0.0, 1.0, 0.0], [0.0, 0.0, 0.0, 1.0]],
            inv_proj: [[1.0, 0.0, 0.0, 0.0], [0.0, 1.0, 0.0, 0.0], [0.0, 0.0, 1.0, 0.0], [0.0, 0.0, 0.0, 1.0]],
            inv_view_proj: [[1.0, 0.0, 0.0, 0.0], [0.0, 1.0, 0.0, 0.0], [0.0, 0.0, 1.0, 0.0], [0.0, 0.0, 0.0, 1.0]],
        }
    }
}

pub struct Camera {
    pub(crate) is_dirty: bool,
    pub(crate) buffer: PerFrameBuffer,
    pub(crate) data: CameraData,
}

impl Camera {

    pub fn proj(&mut self) -> &[[f32; 4]; 4] {
        &self.data.proj
    }

    pub fn view(&mut self) -> &[[f32; 4]; 4] {
        &self.data.view
    }

    pub fn view_mut(&mut self) -> &mut [[f32; 4]; 4] {
        self.is_dirty = true;
        &mut self.data.view
    }

    pub fn proj_mut(&mut self) -> &mut [[f32; 4]; 4] {
        self.is_dirty = true;
        &mut self.data.proj
    }

    pub fn new(device: &Device, frame_count: usize) -> VulkanResult<Self> {
        let size = size_of::<CameraData>() as u64;

        let mut buffer = PerFrameBufferBuilder::new(device)
            .buffer_size(size)
            .frame_count(frame_count)
            .usage(vk::BufferUsageFlags::UNIFORM_BUFFER)
            .build()?;

        let data = CameraData::identity();

        for i in 0..frame_count {
            let buffer = buffer.get_mut(i as u32);
            buffer.upload_data(device, &[data])?;
        }

        Ok(Self { 
            is_dirty: false, 
            buffer, 
            data 
        })
    }
}