aboutsummaryrefslogtreecommitdiff
path: root/src/render_helpers/shaders/mod.rs
blob: 1adbaf825a1c4cc88508befdf12f7f8643fd2fe8 (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
use smithay::backend::renderer::gles::{GlesPixelProgram, GlesRenderer, UniformName, UniformType};

use super::renderer::NiriRenderer;

pub struct Shaders {
    pub gradient_border: Option<GlesPixelProgram>,
}

impl Shaders {
    fn compile(renderer: &mut GlesRenderer) -> Self {
        let _span = tracy_client::span!("Shaders::compile");

        let gradient_border = renderer
            .compile_custom_pixel_shader(
                include_str!("gradient_border.frag"),
                &[
                    UniformName::new("color_from", UniformType::_4f),
                    UniformName::new("color_to", UniformType::_4f),
                    UniformName::new("grad_offset", UniformType::_2f),
                    UniformName::new("grad_width", UniformType::_1f),
                    UniformName::new("grad_vec", UniformType::_2f),
                ],
            )
            .map_err(|err| {
                warn!("error compiling gradient border shader: {err:?}");
            })
            .ok();

        Self { gradient_border }
    }

    pub fn get(renderer: &mut impl NiriRenderer) -> &Self {
        let renderer = renderer.as_gles_renderer();
        let data = renderer.egl_context().user_data();
        data.get()
            .expect("shaders::init() must be called when creating the renderer")
    }
}

pub fn init(renderer: &mut GlesRenderer) {
    let shaders = Shaders::compile(renderer);
    let data = renderer.egl_context().user_data();
    if !data.insert_if_missing(|| shaders) {
        error!("shaders were already compiled");
    }
}