geam 0.1.1

Experimental Rust-embedded execution runtime for typed Gleam programs
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
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126

use super::super::Edge;
use crate::plan::execution::explain::{Explain, ExplainContext};
use crate::plan::execution::graph::FloatLocalId;

pub(crate) struct FloatSwitch {
    subject: FloatLocalId,
    clauses: Box<[(f64, Edge)]>,
    fallback: Edge,
}

impl FloatSwitch {
    pub(in crate::plan::execution) fn new(
        subject: FloatLocalId,
        clauses: Box<[(f64, Edge)]>,
        fallback: Edge,
    ) -> Self {
        Self {
            subject,
            clauses,
            fallback,
        }
    }

    pub(crate) fn subject(&self) -> FloatLocalId {
        self.subject
    }

    pub(crate) fn clauses(&self) -> &[(f64, Edge)] {
        &self.clauses
    }

    pub(crate) fn fallback(&self) -> &Edge {
        &self.fallback
    }
}

impl Explain for FloatSwitch {
    fn write_explanation(&self, context: &mut ExplainContext<'_, '_>) {
        context.push_str("switch.float ");
        context.write(&self.subject());
        super::write_clauses(context, self.clauses(), |context, pattern| {
            context.push_str(&format!("{pattern:?}"));
        });
        context.push_str(" fallback=");
        context.write(self.fallback());
    }
}

#[cfg(test)]
mod explain_tests {
    use super::super::super::Terminator;
    use super::FloatSwitch;
    use crate::plan::execution::explain;
    use crate::plan::execution::function::IntFunctionId;

    #[test]
    fn writes_float_switch() {
        let source = r#"
fn identity(value: Float) { value }
pub fn main() { case identity(1.0) { 1.0 -> 1 _ -> 0 } }
"#;
        let expected = "switch.float %float#1 clauses=[1.0->b1()] fallback=b2()";

        assert_explanation(source, expected);
    }

    #[test]
    #[should_panic(expected = "source should lower one Float switch")]
    fn float_switch_shape_guard_is_visible() {
        explain::with_execution_plan("pub fn main() { 1 }", |plan| {
            float_switch(&terminators(plan));
        });
    }

    #[test]
    #[should_panic(expected = "source should lower one Float switch")]
    fn float_switch_uniqueness_guard_is_visible() {
        let source = r#"
fn identity(value: Float) { value }
pub fn main() { case identity(1.0) { 1.0 -> 1 _ -> 0 } }
"#;
        explain::with_execution_plan(source, |plan| {
            let switch = float_switch(&terminators(plan));
            float_switch_from_nodes(&[switch, switch]);
        });
    }

    fn terminators(
        plan: &crate::plan::execution::ExecutionPlan,
    ) -> Vec<&crate::plan::execution::graph::Terminator> {
        plan.int_function(IntFunctionId(0))
            .body()
            .block_graph()
            .blocks()
            .iter()
            .map(|block| block.terminator())
            .collect()
    }

    fn float_switch<'a>(terminators: &[&'a Terminator]) -> &'a FloatSwitch {
        let switches = terminators
            .iter()
            .copied()
            .filter_map(|terminator| match terminator {
                Terminator::FloatSwitch(switch) => Some(switch),
                _ => None,
            })
            .collect::<Vec<_>>();
        float_switch_from_nodes(&switches)
    }

    fn float_switch_from_nodes<'a>(switches: &[&'a FloatSwitch]) -> &'a FloatSwitch {
        let [switch] = switches else {
            panic!("source should lower one Float switch");
        };
        switch
    }

    fn assert_explanation(source: &str, expected: &str) {
        explain::assert_rendered(source, expected, |plan, output| {
            let switch = float_switch(&terminators(plan));
            let mut context = explain::ExplainContext::new(plan, output);
            context.write(switch);
        });
    }
}