jit 0.9.1

Just-In-Time Compilation in Rust using LibJIT bindings
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

use raw::*;
use alloc::oom;
use function::Func;
use util::{from_ptr, from_ptr_opt};
use std::marker::PhantomData;
use std::{mem, ptr};
use std::ops::{Index, IndexMut};
use std::iter::IntoIterator;
/// Holds all of the functions you have built and compiled. There can be
/// multiple, but normally there is only one.
///
/// The type parameter `T` represents the type of the tagged data on the
/// context, which can be indexed to get this data. If you do not want
/// to tag the context with data, make sure to instantiate it with a
/// unit (`()`) for `T`, like so:
///
/// ```rust
/// use jit::Context;
/// let ctx = Context::<()>::new();
/// ```
/// However, if you do want to set tagged data on it, simply put the type
/// of the data as the `T` parameter when you instantiate it, like so:
///
/// ```rust
/// use jit::Context;
/// let mut ctx = Context::<usize>::new();
/// ctx[0] = 42;
/// ctx[1] = 21;
/// assert_eq!(ctx[0], 42);
/// assert_eq!(ctx[1], 21);
/// ```
pub struct Context<T = ()> {
    _context: jit_context_t,
    marker: PhantomData<T>
}
native_ref!(Context<T>, _context: jit_context_t, marker = PhantomData);

impl<T = ()> Index<i32> for Context<T> {
    type Output = T;
    fn index(&self, index: i32) -> &T {
        unsafe {
            let meta = jit_context_get_meta(self.into(), index);
            if meta.is_null() {
                panic!("No such index {} on Context", index)
            }
            mem::transmute(meta)
        }
    }
}
impl<T = ()> IndexMut<i32> for Context<T> {
    fn index_mut(&mut self, index: i32) -> &mut T {
        unsafe {
            let meta = jit_context_get_meta(self.into(), index);
            if meta.is_null() {
                let boxed = Box::new(mem::uninitialized::<T>());
                if jit_context_set_meta(self.into(), index, mem::transmute(boxed), Some(::free_data::<T>)) == 0 {
                    oom()
                } else {
                    mem::transmute(jit_context_get_meta(self.into(), index))
                }
            } else {
                mem::transmute(meta)
            }
        }
    }
}
impl<T = ()> Context<T> {
    #[inline(always)]
    /// Create a new JIT Context
    pub fn new() -> Context<T> {
        unsafe {
            from_ptr(jit_context_create())
        }
    }
    /// Iterate through the functions contained inside this context
    pub fn functions(&self) -> Functions {
        Functions {
            context: self.into(),
            last: ptr::null_mut(),
            lifetime: PhantomData,
        }
    }
}
impl !Send for Context {

}
impl<'a, T> IntoIterator for &'a Context<T> {
    type IntoIter = Functions<'a>;
    type Item = &'a Func;
    fn into_iter(self) -> Functions<'a> {
        self.functions()
    }
}
impl<T> Drop for Context<T> {
    #[inline(always)]
    fn drop(&mut self) {
        unsafe {
            jit_context_destroy(self.into());
        }
    }
}

pub struct Functions<'a> {
    context: jit_context_t,
    last: jit_function_t,
    lifetime: PhantomData<&'a ()>
}
impl<'a> Iterator for Functions<'a> {
    type Item = &'a Func;
    fn next(&mut self) -> Option<&'a Func> {
        unsafe {
            self.last = jit_function_next(self.context, self.last);
            from_ptr_opt(self.last)
        }
    }
}