custos 0.7.0

A minimal OpenCL, WGPU, CUDA and host CPU array manipulation 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

use std::borrow::Cow;
use std::collections::HashMap;
use wgpu::ShaderModule;

#[derive(Debug, Default)]
pub struct ShaderCache {
    shaders: HashMap<String, ShaderModule>,
}

impl ShaderCache {
    pub fn shader(&mut self, device: &wgpu::Device, src: &str) -> &ShaderModule {
        self.add_shader(device, src);
        self.shaders.get(src).unwrap()
    }

    fn add_shader(&mut self, device: &wgpu::Device, src: &str) {
        if self.shaders.get(src).is_some() {
            return;
        }

        let cs_module = device.create_shader_module(wgpu::ShaderModuleDescriptor {
            label: None,
            source: wgpu::ShaderSource::Wgsl(Cow::Borrowed(src)),
        });

        self.shaders.insert(src.to_string(), cs_module);
    }
}