msl-assembler 0.1.1

Metal Shading Language (MSL) assembler for Gaia project
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
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43

#![warn(missing_docs)]

//! MSL (Metal Shading Language) assembler for Apple platforms.
//!
//! This crate provides utilities for generating MSL code and metallib files.

/// MSL code generator
pub struct MslGenerator;

impl MslGenerator {
    /// Get the name of the generator.
    pub fn name(&self) -> &'static str {
        "msl"
    }

    /// Generate MSL files from a program.
    ///
    /// # Arguments
    /// * `program` - The program to generate MSL from
    ///
    /// # Returns
    /// A hash map of filename to MSL bytecode
    pub fn generate(&self, _program: &serde_json::Value) -> gaia_types::Result<std::collections::HashMap<String, Vec<u8>>> {
        let mut files = std::collections::HashMap::new();

        let mut module_source = self.msl_header();
        module_source.push_str("\n");

        files.insert("module.metal".to_string(), Vec::from(module_source.as_bytes()));
        // Simulate metallib generation
        files.insert("default.metallib".to_string(), vec![0x4d, 0x54, 0x4c, 0x42]); // MTLB magic

        Ok(files)
    }

    /// Generate MSL header.
    ///
    /// # Returns
    /// MSL header string
    fn msl_header(&self) -> String {
        "#include <metal_stdlib>\nusing namespace metal;".to_string()
    }
}