rasen 0.12.0

Build a SPIR-V module from a data flow graph
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

//! A Module is the advanced entry point for the Rasen compiler. It holds the graph for the main function,
//! as well as the subgraphs for all the user-defined functions

use std::ops::{Index, IndexMut};
use graph::Graph;

/// An opaque pointer struct to a function
#[derive(Copy, Clone, Debug)]
pub struct FunctionRef(pub(crate) usize);

/// A container for complex shader programs with multiple functions
#[derive(Debug, Default)]
pub struct Module {
    pub main: Graph,
    pub(crate) functions: Vec<Graph>,
}

impl Module {
    /// Add a function to the graph
    pub fn add_function(&mut self) -> FunctionRef {
        let index = self.functions.len();
        self.functions.push(Graph::default());
        FunctionRef(index)
    }

    /// Get a reference to a function's graph from its index
    pub fn function<'a>(&'a mut self, index: FunctionRef) -> Option<&'a mut Graph> {
        self.functions.get_mut(index.0)
    }
}

impl Index<FunctionRef> for Module {
    type Output = Graph;

    fn index(&self, index: FunctionRef) -> &Graph {
        &self.functions[index.0]
    }
}

impl IndexMut<FunctionRef> for Module {
    fn index_mut(&mut self, index: FunctionRef) -> &mut Graph {
        &mut self.functions[index.0]
    }
}