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

use proc_macro::TokenStream;
use proc_macro_error::proc_macro_error;

mod bitsize;
mod bitsize_internal;
mod try_from_bits;
mod from_bits;
mod debug_bits;

mod shared;

/// Defines the bitsize of a struct or an enum.
/// 
/// e.g. `#[bitsize(4)]` represents the item as a u4, which is UInt<u8, 4> underneath.
/// The size of structs is currently limited to 128 bits.
/// The size of enums is limited to 64 bits.
/// Please open an issue if you have a usecase for bigger bitfields.
#[proc_macro_error]
#[proc_macro_attribute]
pub fn bitsize(args: TokenStream, item: TokenStream) -> TokenStream {
    bitsize::bitsize(args.into(), item.into()).into()
}

/// This is internally used, not to be used by anything besides `bitsize`.
/// No guarantees are given.
#[proc_macro_error]
#[proc_macro_attribute]
pub fn bitsize_internal(args: TokenStream, item: TokenStream) -> TokenStream {
    bitsize_internal::bitsize_internal(args.into(), item.into()).into()
}

/// Generate an `impl TryFrom<uN>` for unfilled bitfields.
/// 
/// This should be used when your enum or enums nested in
/// a struct don't fill their given `bitsize`.
#[proc_macro_error]
#[proc_macro_derive(TryFromBits, attributes(bitsize_internal))]
pub fn derive_try_from_bits(item: TokenStream) -> TokenStream {
    try_from_bits::try_from_bits(item.into()).into()
}

/// Generate an `impl From<uN>` for filled bitfields.
/// 
/// This should be used when your enum or enums nested in
/// a struct fill their given `bitsize` or if you're not
/// using enums.
#[proc_macro_error]
#[proc_macro_derive(FromBits, attributes(bitsize_internal))]
pub fn derive_from_bits(item: TokenStream) -> TokenStream {
    from_bits::from_bits(item.into()).into()
}

/// Generate an `impl Debug` for bitfield structs.
/// 
/// Please use normal #[derive(Debug)] for enums.
#[proc_macro_error]
#[proc_macro_derive(DebugBits, attributes(bitsize_internal))]
pub fn debug_bits(item: TokenStream) -> TokenStream {
    debug_bits::debug_bits(item.into()).into()
}