Move PropertyBitField to a syntax extension.

This is a first step towards moving away the Mako template in the style crate.
This commit is contained in:
Simon Sapin 2014-08-23 23:31:44 +01:00
parent 0ba9c9e3ed
commit d6002a0a50
2 changed files with 141 additions and 65 deletions

View file

@ -4,14 +4,133 @@
#![crate_name = "macros"]
#![crate_type = "rlib"]
#![crate_type = "dylib"]
#![feature(macro_rules)]
#![feature(macro_rules, plugin_registrar, quote)]
//! Exports macros for use in other Servo crates.
#[cfg(test)]
extern crate sync;
extern crate rustc;
extern crate syntax;
use syntax::ast;
use syntax::codemap::Span;
use syntax::ext::base;
use syntax::ext::base::{ExtCtxt, MacResult};
use syntax::parse::token;
use syntax::util::small_vector::SmallVector;
use rustc::plugin::Registry;
use std::gc::{Gc, GC};
#[plugin_registrar]
pub fn plugin_registrar(reg: &mut Registry) {
reg.register_syntax_extension(
token::intern("bit_struct"),
base::IdentTT(box base::BasicIdentMacroExpander {
expander: expand_bit_struct,
span: None,
}, None));
}
fn expand_bit_struct(cx: &mut ExtCtxt, sp: Span, name: ast::Ident, tts: Vec<ast::TokenTree>)
-> Box<base::MacResult> {
let mut fields = Vec::new();
for (i, e) in tts.iter().enumerate() {
if i & 1 == 1 {
match *e {
ast::TTTok(_, token::COMMA) => (),
_ => {
cx.span_err(sp, "bit_struct! expecting comma.");
return base::DummyResult::any(sp);
}
}
} else {
match *e {
ast::TTTok(_, token::IDENT(ident, _)) => {
fields.push(ident)
}
_ => {
cx.span_err(sp, "bit_struct! requires ident args.");
return base::DummyResult::any(sp);
}
}
}
}
let bits_per_word =
if cfg!(target_word_size = "64") { 64 }
else if cfg!(target_word_size = "32") { 32 }
else { fail!("Unexpected target word size") };
let nb_words = (fields.len() - 1 + bits_per_word) / bits_per_word;
let struct_def = quote_item!(&*cx,
pub struct $name {
storage: [uint, ..$nb_words]
}
).unwrap();
let impl_def = quote_item!(&*cx,
impl $name {
#[inline]
pub fn new() -> $name {
$name { storage: [0, ..$nb_words] }
}
}
).unwrap();
// Unwrap from Gc<T>, which does not implement DerefMut
let mut impl_def = (*impl_def).clone();
match impl_def.node {
ast::ItemImpl(_, _, _, ref mut methods) => {
for (i, field) in fields.iter().enumerate() {
let setter_name = "set_".to_string() + field.as_str();
let setter = token::str_to_ident(setter_name.as_slice());
let word = i / bits_per_word;
let bit = i % bits_per_word;
let additional_impl = quote_item!(&*cx,
impl $name {
#[allow(non_snake_case_functions)]
pub fn $field(&self) -> bool {
(self.storage[$word] & (1 << $bit)) != 0
}
#[allow(non_snake_case_functions)]
pub fn $setter(&mut self, new_value: bool) {
if new_value {
self.storage[$word] |= 1 << $bit
} else {
self.storage[$word] &= !(1 << $bit)
}
}
}
).unwrap();
match additional_impl.node {
ast::ItemImpl(_, _, _, ref additional_methods) => {
methods.push_all(additional_methods.as_slice());
}
_ => fail!()
}
}
}
_ => fail!()
}
// Re-wrap.
let impl_def = box(GC) impl_def;
struct MacItems {
items: Vec<Gc<ast::Item>>
}
impl MacResult for MacItems {
fn make_items(&self) -> Option<SmallVector<Gc<ast::Item>>> {
Some(SmallVector::many(self.items.clone()))
}
}
box MacItems { items: vec![struct_def, impl_def] } as Box<MacResult>
}
#[macro_export]
macro_rules! bitfield(