aboutsummaryrefslogtreecommitdiff
path: root/stockton-skeleton/src/lib.rs
blob: 1e0d108f687c273c68a2d420da5874d0ddd238a7 (plain)
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
#[cfg(feature = "vulkan")]
extern crate gfx_backend_vulkan as back;
extern crate gfx_hal as hal;
extern crate nalgebra_glm as na;

#[macro_use]
extern crate derive_builder;

pub mod buffers;
pub mod builders;
pub mod components;
pub mod context;
pub mod draw_passes;
pub mod error;
pub mod mem;
pub mod queue_negotiator;
pub mod session;
mod target;
pub mod texture;
pub mod types;
pub mod utils;

pub use anyhow::Result;
pub use context::RenderingContext;
pub use draw_passes::{DrawPass, IntoDrawPass, PassPosition};
pub use session::Session;

use anyhow::Context;
use draw_passes::Singular;
use std::mem::ManuallyDrop;
use winit::window::Window;

/// Renders a world to a window when you tell it to.
/// Also takes ownership of the window and channels window events to be processed outside winit's event loop.
pub struct Renderer<DP> {
    /// All the vulkan stuff
    context: ManuallyDrop<RenderingContext>,

    /// The draw pass we're using
    draw_pass: ManuallyDrop<DP>,
}

impl<DP: DrawPass<Singular>> Renderer<DP> {
    /// Create a new Renderer.
    pub fn new<IDP: IntoDrawPass<DP, Singular>>(
        window: &Window,
        session: &mut Session,
        idp: IDP,
    ) -> Result<Self> {
        let mut context = RenderingContext::new::<IDP, DP>(window)?;

        // Draw pass
        let draw_pass = idp
            .init(session, &mut context)
            .context("Error initialising draw pass")?;

        Ok(Renderer {
            context: ManuallyDrop::new(context),
            draw_pass: ManuallyDrop::new(draw_pass),
        })
    }

    /// Render a single frame of the given session.
    /// If this returns an error, the whole renderer is dead, hence it takes ownership to ensure it can't be called in that case.
    pub fn render(mut self, session: &Session) -> Result<Renderer<DP>> {
        // Safety: If this fails at any point, the ManuallyDrop won't be touched again, as Renderer will be dropped.
        // Hence, we can always take from the ManuallyDrop
        unsafe {
            match ManuallyDrop::take(&mut self.context)
                .draw_next_frame(session, &mut *self.draw_pass)
            {
                Ok(c) => {
                    self.context = ManuallyDrop::new(c);
                    Ok(self)
                }
                Err((_e, c)) => {
                    // TODO: Try to detect if the error is actually surface related.

                    let c = c.attempt_recovery()?;
                    match c.draw_next_frame(session, &mut *self.draw_pass) {
                        Ok(c) => {
                            self.context = ManuallyDrop::new(c);
                            Ok(self)
                        }
                        Err((e, _c)) => Err(e),
                    }
                }
            }
        }
    }

    /// Recreate the surface, and other derived components.
    /// This should be called when the window is resized.
    pub fn recreate_surface(mut self, session: &Session) -> Result<Renderer<DP>> {
        // Safety: If this fails at any point, the ManuallyDrop won't be touched again, as Renderer will be dropped.
        // Hence, we can always take from the ManuallyDrop
        unsafe {
            let ctx = ManuallyDrop::take(&mut self.context);
            log::debug!("ctx");
            let ctx = ctx.recreate_surface()?;
            self.context = ManuallyDrop::new(ctx);
            log::debug!("Finished resizing ctx");
            let dp = ManuallyDrop::take(&mut self.draw_pass)
                .handle_surface_change(session, &mut self.context)?;
            self.draw_pass = ManuallyDrop::new(dp);
        }

        Ok(self)
    }

    pub fn get_aspect_ratio(&self) -> f32 {
        let e = self.context.properties().extent;
        e.width as f32 / e.height as f32
    }

    /// Get a reference to the renderer's context.
    pub fn context(&self) -> &RenderingContext {
        &self.context
    }
}