vicis 0.1.0

Manipulate LLVM-IR in Pure Rust
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

use crate::codegen::isa::TargetIsa;
use crate::ir::types::TypeId;
use id_arena::{Arena, Id};

pub type SlotId = Id<Slot>;

pub struct Slots<T: TargetIsa> {
    pub isa: T,
    arena: Arena<Slot>,
}

#[derive(Debug, Clone)]
pub struct Slot {
    pub(crate) size: u32,
    pub(crate) ty: TypeId,
    pub(crate) num_elements: u32,
    pub(crate) align: u32,
}

impl<T: TargetIsa> Slots<T> {
    pub fn new(isa: T) -> Self {
        Self {
            isa,
            arena: Arena::new(),
        }
    }

    pub fn add_slot(&mut self, ty: TypeId, size: u32) -> SlotId {
        self.arena.alloc(Slot {
            size,
            ty,
            num_elements: 0,
            align: 0,
        })
    }

    pub fn get(&self, id: SlotId) -> &Slot {
        &self.arena[id]
    }

    pub fn unaligned_size(&self) -> u32 {
        let mut total = 0;
        for (_, slot) in &self.arena {
            total += slot.size;
        }
        total
    }
}