rune-ssa 0.9.1

Intermediate mode representation for Rune.
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
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87

use crate::global::Global;
use crate::{Block, Error, Var};
use std::fmt;

/// The central state machine assembler.
pub struct Program {
    global: Global,
    blocks: Vec<Block>,
}

impl Program {
    /// Construct a new empty state machine.
    pub fn new() -> Self {
        Self {
            global: Global::default(),
            blocks: Vec::new(),
        }
    }

    /// Seal the given program.
    pub fn seal(&self) -> Result<(), Error> {
        for block in &self.blocks {
            block.seal()?;
        }

        Ok(())
    }

    /// Allocate a new value.
    pub fn var(&self) -> Var {
        self.global.var()
    }

    /// Construct a new block associated with the state machine.
    pub fn block(&mut self) -> Block {
        let block = self.global.block(None);
        self.blocks.push(block.clone());
        block
    }

    /// Construct a block with a name.
    pub fn named(&mut self, name: &str) -> Block {
        let block = self.global.block(Some(name.into()));
        self.blocks.push(block.clone());
        block
    }

    /// Dump the current state of the program.
    ///
    /// This is useful for diagnostics.
    pub fn dump(&self) -> ProgramDump<'_> {
        ProgramDump(self)
    }
}

pub struct ProgramDump<'a>(&'a Program);

impl fmt::Display for ProgramDump<'_> {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        let constants = self.0.global.constants();

        if !constants.is_empty() {
            writeln!(f, "constants:")?;

            for (id, c) in constants.iter().enumerate() {
                writeln!(f, "  C{} <- {:?}", id, c)?;
            }

            if !self.0.blocks.is_empty() {
                writeln!(f)?;
            }
        }

        let mut it = self.0.blocks.iter();
        let last = it.next_back();

        for b in it {
            writeln!(f, "{}", b.dump())?;
        }

        if let Some(b) = last {
            write!(f, "{}", b.dump())?;
        }

        Ok(())
    }
}