specs-system-macro/src/lib.rs

pub struct SpecsUnit;
impl<'a> specs::SystemData<'a> for SpecsUnit {
    fn setup(_world: &mut specs::World) {
    }

    fn fetch(_world: &'a specs::World) -> Self {
        SpecsUnit
    }

    fn reads() -> Vec<specs::prelude::ResourceId> {
        Vec::new()
    }

    fn writes() -> Vec<specs::prelude::ResourceId> {
        Vec::new()
    }
}

#[macro_export]
macro_rules! system {
    ($name:ident ( $($pat:tt)* ) { $($rest:tt)* } ) => {
        pub struct $name;
        impl<'a> specs::System<'a> for $name {
            type SystemData = args_to_systemdata!(($($pat)*,));
            fn run(&mut self, args_to_fn_pat!(($($pat)*)): Self::SystemData) {
                for args_to_join_pat!(($($pat)*,)) in args_to_join!(($($pat)*,)).join() {
                    $($rest)*
                }
            }
        }
    };
    ($name:ident $pat:tt { $($rest:tt)* } finally { $($finally:tt)* }) => {
        pub struct $name;
        impl<'a> specs::System<'a> for $name {
            type SystemData = args_to_systemdata!($pat);
            fn run(&mut self, args_to_fn_pat!($pat): Self::SystemData) {
                for args_to_join_pat!($pat) in args_to_join!($pat).join() {
                    $($rest)*
                }
                $($finally)*
            }
        }
    };
}

#[macro_export]
macro_rules! args_to_systemdata {
    ( ( ) ) => { crate::SpecsUnit };
    ( ( $name:ident : $ty:ty $(,)? ) ) => {
        ( specs::ReadStorage<'a, $ty> ,)
    };
    ( ( mut $name:ident : $ty:ty $(,)? ) ) => {
        ( specs::WriteStorage<'a, $ty> ,)
    };
    ( ( $name:ident : $ty:ty , $($tok:tt)* ) ) => {
        ( specs::ReadStorage<'a, $ty>, args_to_systemdata!( ( $( $tok )* ) ) )
    };
    ( ( mut $name:ident : $ty:ty , $( $tok:tt )* ) ) => {
        ( specs::WriteStorage<'a, $ty>, args_to_systemdata!( ( $( $tok )* ) ) )
    };
    ( ( resource $name:ident : $ty:ty , $( $tok:tt )* ) ) => {
        ( specs::ReadExpect<'a, $ty>, args_to_systemdata!( ( $( $tok )* ) ) )
    };
    ( ( resource mut $name:ident : $ty:ty , $( $tok:tt )* ) ) => {
        ( specs::WriteExpect<'a, $ty>, args_to_systemdata!( ( $( $tok )* ) ) )
    };
}

#[macro_export]
macro_rules! args_to_fn_pat {
    ( ( ) ) => { _ };
    ( ( $name:ident : $ty:ty $(,)? ) ) => {
        ( $name ,)
    };
    ( ( mut $name:ident : $ty:ty $(,)? ) ) => {
        ( mut $name ,)
    };
    ( ( $name:ident : $ty:ty , $($tok:tt)* ) ) => {
        ( $name, args_to_fn_pat!( ( $($tok)* ) ) )
    };
    ( ( mut $name:ident : $ty:ty , $($tok:tt)* ) ) => {
        ( mut $name, args_to_fn_pat!( ( $($tok)* ) ) )
    };
    ( ( resource $name:ident : $ty:ty , $($tok:tt)* ) ) => {
        ( $name, args_to_fn_pat!( ( $($tok)* ) ) )
    };
    ( ( resource mut $name:ident : $ty:ty , $($tok:tt)* ) ) => {
        ( mut $name, args_to_fn_pat!( ( $($tok)* ) ) )
    };
}

#[macro_export]
macro_rules! args_to_join_pat {
    ( ( ) ) => { _ };
    ( ( $name:ident : $ty:ty $(,)? ) ) => {
        ( $name ,)
    };
    ( ( mut $name:ident : $ty:ty $(,)? ) ) => {
        ( mut $name ,)
    };
    ( ( $name:ident : $ty:ty , $($tok:tt)* ) ) => {
        ( $name, args_to_join_pat!( ( $($tok)* ) ) )
    };
    ( ( mut $name:ident : $ty:ty , $($tok:tt)* ) ) => {
        ( mut $name, args_to_join_pat!( ( $($tok)* ) ) )
    };
    ( ( resource $name:ident : $ty:ty , $($tok:tt)* ) ) => {
        args_to_join_pat!( ( $($tok)* ) )
    };
    ( ( resource mut $name:ident : $ty:ty , $($tok:tt)* ) ) => {
        args_to_join_pat!( ( $($tok)* ) )
    };
}

#[macro_export]
macro_rules! args_to_join {
    ( ( ) ) => { crate::SpecsUnit };
    ( ( $name:ident : $ty:ty $(,)? ) ) => {
        ( & $name ,)
    };
    ( ( mut $name:ident : $ty:ty $(,)? ) ) => {
        ( &mut $name ,)
    };
    ( ( $name:ident : $ty:ty , $($tok:tt)* ) ) => {
        ( & $name, args_to_join!( ( $($tok)* ) ) )
    };
    ( ( mut $name:ident : $ty:ty , $($tok:tt)* ) ) => {
        ( &mut $name, args_to_join!( ( $($tok)* ) ) )
    };
    ( ( resource $name:ident : $ty:ty , $($tok:tt)* ) ) => {
        args_to_join!( ( $($tok)* ) )
    };
    ( ( resource mut $name:ident : $ty:ty , $($tok:tt)* ) ) => {
        args_to_join!( ( $($tok)* ) )
    };
}


#[cfg(test)]
mod tests {
    use specs::join::Join;

    pub struct Pos {
        x: usize,
    }
    impl specs::Component for Pos {
        type Storage = specs::VecStorage<Pos>;
    }

    pub struct Mov;
    impl specs::Component for Mov {
        type Storage = specs::VecStorage<Mov>;
    }

    system! {
        Foo (_x: Mov, mut y: Pos) {
            y.x += 1;
        }
    }

    system! {
        Bar (_x: Mov, mut y: Pos) {
            y.x += 1;
        } finally {
            println!("Done!");
        }
    }

    system! {
        Baz (resource n: usize, _x: Mov, mut y: Pos) {
            y.x += *n;
        }
    }

    system! {
        Quux (resource mut n: usize, _x: Mov, mut y: Pos) {
            y.x += *n;
            *n += 1
        }
    }

}