diff options
Diffstat (limited to 'components/macros')
-rw-r--r-- | components/macros/Cargo.toml | 9 | ||||
-rw-r--r-- | components/macros/lib.rs | 93 |
2 files changed, 102 insertions, 0 deletions
diff --git a/components/macros/Cargo.toml b/components/macros/Cargo.toml new file mode 100644 index 00000000000..c187d554829 --- /dev/null +++ b/components/macros/Cargo.toml @@ -0,0 +1,9 @@ +[package] +name = "macros" +version = "0.0.1" +authors = ["The Servo Project Developers"] + +[lib] +name = "macros" +path = "lib.rs" +plugin = true diff --git a/components/macros/lib.rs b/components/macros/lib.rs new file mode 100644 index 00000000000..20304ffe806 --- /dev/null +++ b/components/macros/lib.rs @@ -0,0 +1,93 @@ +/* This Source Code Form is subject to the terms of the Mozilla Public + * License, v. 2.0. If a copy of the MPL was not distributed with this + * file, You can obtain one at http://mozilla.org/MPL/2.0/. */ + +#![feature(macro_rules, plugin_registrar, quote, phase)] + +//! Exports macros for use in other Servo crates. + +#[cfg(test)] +extern crate sync; + + +#[macro_export] +macro_rules! bitfield( + ($bitfieldname:ident, $getter:ident, $setter:ident, $value:expr) => ( + impl $bitfieldname { + #[inline] + pub fn $getter(self) -> bool { + let $bitfieldname(this) = self; + (this & $value) != 0 + } + + #[inline] + pub fn $setter(&mut self, value: bool) { + let $bitfieldname(this) = *self; + *self = $bitfieldname((this & !$value) | (if value { $value } else { 0 })) + } + } + ) +) + + +#[macro_export] +macro_rules! lazy_init( + ($(static ref $N:ident : $T:ty = $e:expr;)*) => ( + $( + #[allow(non_camel_case_types)] + struct $N {__unit__: ()} + static $N: $N = $N {__unit__: ()}; + impl Deref<$T> for $N { + fn deref<'a>(&'a self) -> &'a $T { + unsafe { + static mut s: *const $T = 0 as *const $T; + static mut ONCE: ::sync::one::Once = ::sync::one::ONCE_INIT; + ONCE.doit(|| { + s = ::std::mem::transmute::<Box<$T>, *const $T>(box () ($e)); + }); + &*s + } + } + } + + )* + ) +) + + +#[cfg(test)] +mod tests { + use std::collections::hashmap::HashMap; + lazy_init! { + static ref NUMBER: uint = times_two(3); + static ref VEC: [Box<uint>, ..3] = [box 1, box 2, box 3]; + static ref OWNED_STRING: String = "hello".to_string(); + static ref HASHMAP: HashMap<uint, &'static str> = { + let mut m = HashMap::new(); + m.insert(0u, "abc"); + m.insert(1, "def"); + m.insert(2, "ghi"); + m + }; + } + + fn times_two(n: uint) -> uint { + n * 2 + } + + #[test] + fn test_basic() { + assert_eq!(*OWNED_STRING, "hello".to_string()); + assert_eq!(*NUMBER, 6); + assert!(HASHMAP.find(&1).is_some()); + assert!(HASHMAP.find(&3).is_none()); + assert_eq!(VEC.as_slice(), &[box 1, box 2, box 3]); + } + + #[test] + fn test_repeat() { + assert_eq!(*NUMBER, 6); + assert_eq!(*NUMBER, 6); + assert_eq!(*NUMBER, 6); + } +} |