sass_assembler/
lib.rs

1#![warn(missing_docs)]
2
3pub mod instructions;
4pub mod program;
5
6use crate::{instructions::SassInstruction, program::SassProgram};
7use gaia_types::Result;
8
9/// SASS 汇编器/写入器
10pub struct SassWriter {}
11
12impl SassWriter {
13    pub fn new() -> Self {
14        Self {}
15    }
16
17    pub fn write(&self, program: &SassProgram) -> Result<Vec<u8>> {
18        let mut code = String::new();
19        for kernel in &program.kernels {
20            code.push_str(&format!(".section .text.{}\n.global {}\n{}:\n", kernel.name, kernel.name, kernel.name));
21            for inst in &kernel.instructions {
22                code.push_str("    ");
23                code.push_str(&self.format_instruction(inst));
24                code.push('\n');
25            }
26        }
27
28        // 模拟生成二进制 (ELF Magic + 汇编内容)
29        let mut binary = Vec::from("\x7fELF".as_bytes());
30        binary.extend_from_slice(code.as_bytes());
31        Ok(binary)
32    }
33
34    fn format_instruction(&self, inst: &SassInstruction) -> String {
35        match inst {
36            SassInstruction::FAdd { dst, src0, src1 } => format!("FADD {}, {}, {}", dst, src0, src1),
37            SassInstruction::FMul { dst, src0, src1 } => format!("FMUL {}, {}, {}", dst, src0, src1),
38            SassInstruction::Imma { dst, src0, src1, src2 } => {
39                format!("IMMA.16.16.16.F32 {}, {}, {}, {}", dst, src0, src1, src2)
40            }
41            SassInstruction::Ldg { dst, addr } => format!("LDG.E {}, [{}]", dst, addr),
42            SassInstruction::Stg { addr, src } => format!("STG.E [{}], {}", addr, src),
43            SassInstruction::Exit => "EXIT".to_string(),
44            SassInstruction::Nop => "NOP".to_string(),
45        }
46    }
47}
sass_assembler/lib.rs

sass_assembler/
lib.rs
