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
use cranelift_codegen::ir::{Function, Value as ValueRef};
use cranelift_reader::DataValue;
use log::trace;
use std::collections::HashMap;
#[derive(Debug)]
pub struct Frame<'a> {
pub function: &'a Function,
registers: HashMap<ValueRef, DataValue>,
}
impl<'a> Frame<'a> {
pub fn new(function: &'a Function) -> Self {
trace!("Create new frame for function: {}", function.signature);
Self {
function,
registers: HashMap::with_capacity(function.dfg.num_values()),
}
}
#[inline]
pub fn get(&self, name: &ValueRef) -> &DataValue {
trace!("Get {}", name);
self.registers
.get(name)
.unwrap_or_else(|| panic!("unknown value: {}", name))
}
pub fn get_all(&self, names: &[ValueRef]) -> Vec<DataValue> {
names.iter().map(|r| self.get(r)).cloned().collect()
}
#[inline]
pub fn set(&mut self, name: ValueRef, value: DataValue) -> Option<DataValue> {
trace!("Set {} -> {}", name, value);
self.registers.insert(name, value)
}
pub fn set_all(&mut self, names: &[ValueRef], values: Vec<DataValue>) {
assert_eq!(names.len(), values.len());
for (n, v) in names.iter().zip(values) {
self.set(*n, v);
}
}
pub fn rename(&mut self, old_names: &[ValueRef], new_names: &[ValueRef]) {
trace!("Renaming {:?} -> {:?}", old_names, new_names);
assert_eq!(old_names.len(), new_names.len());
let mut registers = HashMap::with_capacity(self.registers.len());
for (on, nn) in old_names.iter().zip(new_names) {
let v = self.registers.get(on).unwrap().clone();
registers.insert(*nn, v);
}
self.registers = registers;
}
}
#[cfg(test)]
mod tests {
use super::*;
use cranelift_codegen::ir::InstBuilder;
use cranelift_frontend::{FunctionBuilder, FunctionBuilderContext};
use cranelift_reader::DataValue;
fn empty_function() -> Function {
let mut func = Function::new();
let mut context = FunctionBuilderContext::new();
let mut builder = FunctionBuilder::new(&mut func, &mut context);
let block = builder.create_block();
builder.switch_to_block(block);
builder.ins().return_(&[]);
func
}
#[test]
fn construction() {
let func = empty_function();
Frame::new(&func);
}
#[test]
fn assignment() {
let func = empty_function();
let mut frame = Frame::new(&func);
let a = ValueRef::with_number(1).unwrap();
let fortytwo = DataValue::I32(42);
frame.set(a, fortytwo.clone());
assert_eq!(frame.get(&a), &fortytwo);
}
#[test]
#[should_panic]
fn no_existing_value() {
let func = empty_function();
let frame = Frame::new(&func);
let a = ValueRef::with_number(1).unwrap();
frame.get(&a);
}
}