mun_codegen 0.4.0

LLVM IR code generation for Mun
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

use super::{AsValue, BytesOrPtr, Global, IrTypeContext, IrValueContext, TransparentValue, Value};
use std::ffi::{CStr, CString};

/// Enables internalizing certain data structures like strings.
pub trait CanInternalize {
    type Type;

    /// Internalizes the instance into a global value.
    fn intern<'ink, S: AsRef<str>>(
        &self,
        name: S,
        context: &IrValueContext<'ink, '_, '_>,
    ) -> Global<'ink, Self::Type>;
}

impl CanInternalize for str {
    type Type = String;

    fn intern<'ink, S: AsRef<str>>(
        &self,
        name: S,
        context: &IrValueContext<'ink, '_, '_>,
    ) -> Global<'ink, Self::Type> {
        unsafe {
            Global::from_raw(
                self.as_bytes()
                    .as_value(context)
                    .into_const_private_global(name, context)
                    .value,
            )
        }
    }
}

impl CanInternalize for CStr {
    type Type = CString;

    fn intern<'ink, S: AsRef<str>>(
        &self,
        name: S,
        context: &IrValueContext<'ink, '_, '_>,
    ) -> Global<'ink, Self::Type> {
        unsafe {
            Global::from_raw(
                self.to_bytes_with_nul()
                    .as_value(context)
                    .into_const_private_global(name, context)
                    .value,
            )
        }
    }
}

impl<'ink> TransparentValue<'ink> for CString {
    type Target = [u8];

    fn as_target_value(&self, context: &IrValueContext<'ink, '_, '_>) -> Value<'ink, Self::Target> {
        self.as_bytes_with_nul().as_value(context)
    }

    fn as_bytes_and_ptrs(&self, _: &IrTypeContext<'ink, '_>) -> Vec<BytesOrPtr<'ink>> {
        unreachable!(
            "`as_bytes_and_ptrs` should never be called on a `String`, as it cannot be a member of an enum."
        )
    }
}

impl<'ink> TransparentValue<'ink> for String {
    type Target = [u8];

    fn as_target_value(&self, context: &IrValueContext<'ink, '_, '_>) -> Value<'ink, Self::Target> {
        self.as_bytes().as_value(context)
    }

    fn as_bytes_and_ptrs(&self, _: &IrTypeContext<'ink, '_>) -> Vec<BytesOrPtr<'ink>> {
        unreachable!(
            "`as_bytes_and_ptrs` should never be called on a `String`, as it cannot be a member of an enum."
        );
    }
}