Small tweaks
This commit is contained in:
parent
d037e22164
commit
f3922ff84d
7 changed files with 182 additions and 23 deletions
1
.gitignore
vendored
1
.gitignore
vendored
|
|
@ -1,2 +1,3 @@
|
||||||
target
|
target
|
||||||
.idea
|
.idea
|
||||||
|
Cargo.lock
|
||||||
|
|
|
||||||
53
Cargo.lock
generated
53
Cargo.lock
generated
|
|
@ -5,3 +5,56 @@ version = 4
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "deimos"
|
name = "deimos"
|
||||||
version = "0.1.0"
|
version = "0.1.0"
|
||||||
|
dependencies = [
|
||||||
|
"macros",
|
||||||
|
"map",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "macros"
|
||||||
|
version = "0.0.0"
|
||||||
|
dependencies = [
|
||||||
|
"map",
|
||||||
|
"proc-macro2",
|
||||||
|
"quote",
|
||||||
|
"syn",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "map"
|
||||||
|
version = "0.1.0"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "proc-macro2"
|
||||||
|
version = "1.0.107"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "985e7ec9bb745e6ce6535b544d84d6cd6f7ad8bd711c398938ae983b91a766d9"
|
||||||
|
dependencies = [
|
||||||
|
"unicode-ident",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "quote"
|
||||||
|
version = "1.0.47"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "1fbf4db142a473a8d80c26bbf18454ed458bf8d26c8219c331daecfdbd079001"
|
||||||
|
dependencies = [
|
||||||
|
"proc-macro2",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "syn"
|
||||||
|
version = "1.0.109"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "72b64191b275b66ffe2469e8af2c1cfe3bafa67b529ead792a6d0160888b4237"
|
||||||
|
dependencies = [
|
||||||
|
"proc-macro2",
|
||||||
|
"quote",
|
||||||
|
"unicode-ident",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "unicode-ident"
|
||||||
|
version = "1.0.26"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "d245f478577f809a851594d02313b640fb437e0bb33866753cff937863096954"
|
||||||
|
|
|
||||||
|
|
@ -3,11 +3,6 @@ name = "deimos"
|
||||||
version = "0.1.0"
|
version = "0.1.0"
|
||||||
edition = "2024"
|
edition = "2024"
|
||||||
|
|
||||||
[features]
|
|
||||||
bighash = []
|
|
||||||
assertions = []
|
|
||||||
messages = []
|
|
||||||
|
|
||||||
[dependencies]
|
[dependencies]
|
||||||
macros = { path = "macros" }
|
macros = { path = "macros" }
|
||||||
map = { path = "map" }
|
map = { path = "map" }
|
||||||
|
|
@ -1,19 +1,108 @@
|
||||||
extern crate proc_macro;
|
extern crate proc_macro;
|
||||||
use proc_macro::TokenStream;
|
use proc_macro::{Delimiter, Group, Punct, Ident, TokenStream, TokenTree, Literal, Spacing, Span};
|
||||||
|
use proc_macro::token_stream::IntoIter;
|
||||||
|
use std::iter::Peekable;
|
||||||
|
use map::{DisplacementSet, Equivalent, Hashed, Hashes, MapEntry, TableEntry};
|
||||||
|
|
||||||
use syn::{parse_macro_input, DeriveInput};
|
type Tokens = Peekable<IntoIter>;
|
||||||
use quote::quote;
|
|
||||||
|
fn brackets(stream: TokenStream) -> TokenStream {
|
||||||
|
TokenStream::from(TokenTree::Group(Group::new(Delimiter::Brace,stream)))
|
||||||
|
}
|
||||||
|
|
||||||
|
fn entry(input: &mut Tokens) -> Option<MapEntry<String,TokenStream>> {
|
||||||
|
if let Some(token) = input.peek() {
|
||||||
|
let index = match input.next() { // Take one literal as an index
|
||||||
|
Some(TokenTree::Literal(literal)) => literal.to_string(),
|
||||||
|
Some(_) => panic!("Expected literal index"),
|
||||||
|
None => return None,
|
||||||
|
};
|
||||||
|
match input.next() { // Look for '=' in "=>"
|
||||||
|
Some(TokenTree::Punct(punct)) if punct.as_char() == '=' => {},
|
||||||
|
_ => panic!("Expected '=>'"),
|
||||||
|
};
|
||||||
|
match input.next() { // Look for '>' in "=>"
|
||||||
|
Some(TokenTree::Punct(punct)) if punct.as_char() == '>' => {},
|
||||||
|
_ => panic!("Expected '=>'"),
|
||||||
|
};
|
||||||
|
let mut value = Vec::new();
|
||||||
|
let mut closure = false;
|
||||||
|
loop { // Accumulate value token stream until ',' is found
|
||||||
|
match input.next() {
|
||||||
|
Some(TokenTree::Punct(punct)) if punct.as_char() == ',' && !closure => break,
|
||||||
|
Some(ref it@TokenTree::Punct(ref punct)) if punct.as_char() == '|' => {
|
||||||
|
value.push(it.clone());
|
||||||
|
closure = !closure
|
||||||
|
},
|
||||||
|
Some(it) => value.push(it),
|
||||||
|
None => break,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
let mut value_stream = TokenStream::new();
|
||||||
|
value_stream.extend(value);
|
||||||
|
Some(MapEntry(Some((
|
||||||
|
index,
|
||||||
|
value_stream,
|
||||||
|
))))
|
||||||
|
} else {
|
||||||
|
None
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
/// Example of [function-like procedural macro][1].
|
|
||||||
///
|
|
||||||
/// [1]: https://doc.rust-lang.org/reference/procedural-macros.html#function-like-procedural-macros
|
|
||||||
#[proc_macro]
|
#[proc_macro]
|
||||||
pub fn map(input: TokenStream) -> TokenStream {
|
pub fn map(input: TokenStream) -> TokenStream { // todo: only works for string literals right now
|
||||||
let input = parse_macro_input!(input as DeriveInput);
|
let mut tokens = input.into_iter().peekable();
|
||||||
|
let mut entries = DisplacementSet::new();
|
||||||
let tokens = quote! {
|
while let Some(entry) = entry(&mut tokens) {
|
||||||
const
|
entries.set(entry);
|
||||||
};
|
}
|
||||||
|
let mut elements = TokenStream::new();
|
||||||
tokens.into()
|
for entry in entries.map.iter() {
|
||||||
|
let option = if let Some(entry) = entry.entry.clone().0 {
|
||||||
|
let mut inner = TokenStream::new();
|
||||||
|
inner.extend(vec![
|
||||||
|
TokenTree::Literal(Literal::string(entry.0.as_str())),
|
||||||
|
TokenTree::Punct(Punct::new(',',Spacing::Alone)),
|
||||||
|
]);
|
||||||
|
inner.extend(entry.1);
|
||||||
|
let inner = TokenStream::from(TokenTree::Group(Group::new(Delimiter::Parenthesis,inner)));
|
||||||
|
let mut option = TokenStream::new();
|
||||||
|
option.extend(vec![
|
||||||
|
TokenTree::Ident(Ident::new("Some",Span::mixed_site())),
|
||||||
|
TokenTree::Group(Group::new(Delimiter::Parenthesis,inner))
|
||||||
|
]);
|
||||||
|
option
|
||||||
|
} else {
|
||||||
|
TokenStream::from(TokenTree::Ident(Ident::new("None",Span::mixed_site())))
|
||||||
|
};
|
||||||
|
let mut set_entry = TokenStream::new();
|
||||||
|
set_entry.extend(vec![
|
||||||
|
TokenTree::Ident(Ident::new("entry",Span::mixed_site())),
|
||||||
|
TokenTree::Punct(Punct::new(':',Spacing::Alone)),
|
||||||
|
TokenTree::Ident(Ident::new("MapEntry",Span::mixed_site())),
|
||||||
|
TokenTree::Group(Group::new(Delimiter::Parenthesis,option)),
|
||||||
|
TokenTree::Punct(Punct::new(',',Spacing::Alone)),
|
||||||
|
TokenTree::Ident(Ident::new("home",Span::mixed_site())),
|
||||||
|
TokenTree::Punct(Punct::new(':',Spacing::Alone)),
|
||||||
|
TokenTree::Literal(Literal::usize_unsuffixed(entry.home)),
|
||||||
|
TokenTree::Punct(Punct::new(',',Spacing::Alone)),
|
||||||
|
TokenTree::Ident(Ident::new("displacement",Span::mixed_site())),
|
||||||
|
TokenTree::Punct(Punct::new(':',Spacing::Alone)),
|
||||||
|
TokenTree::Literal(Literal::usize_unsuffixed(entry.displacement as usize)),
|
||||||
|
TokenTree::Punct(Punct::new(',',Spacing::Alone)),
|
||||||
|
]);
|
||||||
|
elements.extend(vec![
|
||||||
|
/*TokenTree::Ident(Ident::new("deimos",Span::mixed_site())),
|
||||||
|
TokenTree::Punct(Punct::new(':',Spacing::Joint)),
|
||||||
|
TokenTree::Punct(Punct::new(':',Spacing::Joint)),
|
||||||
|
TokenTree::Ident(Ident::new("map",Span::mixed_site())),
|
||||||
|
TokenTree::Punct(Punct::new(':',Spacing::Joint)),
|
||||||
|
TokenTree::Punct(Punct::new(':',Spacing::Joint)),*/
|
||||||
|
TokenTree::Ident(Ident::new("Entry",Span::mixed_site())),
|
||||||
|
TokenTree::Group(Group::new(Delimiter::Brace,set_entry)),
|
||||||
|
TokenTree::Punct(Punct::new(',',Spacing::Alone)),
|
||||||
|
])
|
||||||
|
}
|
||||||
|
let mut array = TokenStream::from(TokenTree::Group(Group::new(Delimiter::Bracket,elements)));
|
||||||
|
array
|
||||||
}
|
}
|
||||||
|
|
@ -3,4 +3,9 @@ name = "map"
|
||||||
version = "0.1.0"
|
version = "0.1.0"
|
||||||
edition = "2024"
|
edition = "2024"
|
||||||
|
|
||||||
|
[features]
|
||||||
|
bighash = []
|
||||||
|
assertions = []
|
||||||
|
messages = []
|
||||||
|
|
||||||
[dependencies]
|
[dependencies]
|
||||||
|
|
|
||||||
|
|
@ -45,11 +45,18 @@ pub trait Hashes {
|
||||||
impl Hashes for &str {
|
impl Hashes for &str {
|
||||||
fn hashed(&self) -> Hashed {
|
fn hashed(&self) -> Hashed {
|
||||||
let mut hasher = DefaultHasher::new();
|
let mut hasher = DefaultHasher::new();
|
||||||
self[..self.len().min(6)].hash(&mut hasher);
|
self[..self.len().min(8)].hash(&mut hasher);
|
||||||
hasher.finish() as Hashed
|
hasher.finish() as Hashed
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
impl Hashes for String {
|
||||||
|
fn hashed(&self) -> Hashed {
|
||||||
|
self.as_str().hashed()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
pub trait TableEntry: Equivalent<Self> + Hashes {
|
pub trait TableEntry: Equivalent<Self> + Hashes {
|
||||||
const VACANT: Self;
|
const VACANT: Self;
|
||||||
fn is_vacant(&self) -> bool;
|
fn is_vacant(&self) -> bool;
|
||||||
|
|
@ -58,8 +65,8 @@ pub trait TableEntry: Equivalent<Self> + Hashes {
|
||||||
#[derive(Debug)]
|
#[derive(Debug)]
|
||||||
pub struct Entry<E: TableEntry> {
|
pub struct Entry<E: TableEntry> {
|
||||||
pub entry: E,
|
pub entry: E,
|
||||||
home: usize,
|
pub home: usize,
|
||||||
displacement: Displacement,
|
pub displacement: Displacement,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl<E: TableEntry> Entry<E> {
|
impl<E: TableEntry> Entry<E> {
|
||||||
|
|
|
||||||
11
src/lib.rs
11
src/lib.rs
|
|
@ -1,9 +1,18 @@
|
||||||
pub mod gc;
|
pub mod gc;
|
||||||
pub use macros;
|
pub use macros;
|
||||||
|
pub use map;
|
||||||
|
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod tests {
|
mod tests {
|
||||||
|
use map::{TableEntry,MapEntry,Entry};
|
||||||
|
extern crate self as deimos;
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn simple() {
|
fn simple() {
|
||||||
|
static MAP: &[Entry<deimos::map::MapEntry<&'static str,fn()>>] = &deimos::macros::map! {
|
||||||
|
"hello" => || println!("hello there"),
|
||||||
|
"hi" => || println!("hi friend"),
|
||||||
|
"greetings" => || println!("salutations"),
|
||||||
|
};
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
Loading…
Add table
Add a link
Reference in a new issue