mirror of
https://github.com/servo/servo.git
synced 2025-08-05 21:50:18 +01:00
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:
parent
0ba9c9e3ed
commit
d6002a0a50
2 changed files with 141 additions and 65 deletions
|
@ -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(
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue