#[macro_export]
macro_rules! auto_bitflags_h {
(u16, $shift_num:expr, $flag_name:ident) => {
const $flag_name: u16 = 1 << $shift_num;
};
(u32, $shift_num:expr, $flag_name:ident) => {
const $flag_name: u32 = 1 << $shift_num;
};
(u64, $shift_num:expr, $flag_name:ident) => {
const $flag_name: u64 = 1 << $shift_num;
};
}
#[macro_export]
macro_rules! count_idents {
() => {0usize};
($one:ident) => {1usize};
($($pairs:ident, $_p:ident), *) => {
count_idents!($($pairs), *) << 1usize
};
($odd:ident, $($rest:ident), *) => {
count_idents!($($rest), *) | 1usize
};
}
#[macro_export]
macro_rules! auto_bitflags {
(u16, $flag:ident, $($flagrest:ident), *) => {
auto_bitflags_h!(u16, count_idents!($($flagrest),*), $flag);
auto_bitflags!(u16, $($flagrest), *);
};
(u16, $flag:ident) => {
auto_bitflags_h!(u16, 0, $flag);
};
(u32, $flag:ident, $($flagrest:ident), *) => {
auto_bitflags_h!(u32, count_idents!($($flagrest),*), $flag);
auto_bitflags!(u32, $($flagrest), *);
};
(u32, $flag:ident) => {
auto_bitflags_h!(u32, 0, $flag);
};
(u64, $flag:ident, $($flagrest:ident), *) => {
auto_bitflags_h!(u64, count_idents!($($flagrest),*), $flag);
auto_bitflags!(u64, $($flagrest), *);
};
(u64, $flag:ident) => {
auto_bitflags_h!(u64, 0, $flag);
};
}
#[cfg(test)]
mod tests {
auto_bitflags!(u16, FLAG_T1_16, FLAG_T2_16);
#[test]
fn test_u16() {
assert_eq!(0b11, FLAG_T1_16 | FLAG_T2_16);
}
auto_bitflags!(u32, FLAG_T1_32, FLAG_T2_32);
#[test]
fn test_u32() {
assert_eq!(0b11, FLAG_T1_32 | FLAG_T2_32);
}
auto_bitflags!(u64, FLAG_T1_64, FLAG_T2_64);
#[test]
fn test_u64() {
assert_eq!(0b11, FLAG_T1_64 | FLAG_T2_64);
}
}