lamina-codegen 0.1.1

Codegen utilities for Lamina (register allocation, ABI, frame management)
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
44
45

/// RISC-V stack frame management utilities
pub struct RiscVFrame;

impl RiscVFrame {
    /// Generate function prologue
    pub fn generate_prologue<W: std::io::Write>(
        writer: &mut W,
        stack_size: usize,
    ) -> Result<(), std::io::Error> {
        // Save return address and frame pointer
        writeln!(writer, "    addi sp, sp, -16")?;
        writeln!(writer, "    sd ra, 8(sp)")?;
        writeln!(writer, "    sd fp, 0(sp)")?;
        writeln!(writer, "    addi fp, sp, 16")?;

        // Allocate stack space for local variables if needed
        if stack_size > 0 {
            writeln!(writer, "    addi sp, sp, -{}", stack_size)?;
        }
        Ok(())
    }

    /// Generate function epilogue
    pub fn generate_epilogue<W: std::io::Write>(
        writer: &mut W,
        stack_size: usize,
    ) -> Result<(), std::io::Error> {
        // Deallocate stack space for local variables if needed
        if stack_size > 0 {
            writeln!(writer, "    addi sp, sp, {}", stack_size)?;
        }

        // Restore return address and frame pointer
        writeln!(writer, "    ld ra, -8(fp)")?;
        writeln!(writer, "    ld fp, -16(fp)")?;
        writeln!(writer, "    addi sp, sp, 16")?;
        writeln!(writer, "    ret")?;
        Ok(())
    }

    /// Calculate stack slot offset from frame pointer (fp)
    pub fn calculate_stack_offset(slot_index: usize) -> i32 {
        -((slot_index as i32 + 1) * 8)
    }
}