ecs: ergonomic query.iter(), remove locks, add QuerySets

bevyengine · Oct 28, 2020 · 9a08483 · 9a08483
1 parent a7565e9
commit 9a08483
Show file tree

Hide file tree

Showing 51 changed files with 1,311 additions and 1,485 deletions.
diff --git a/crates/bevy_core/src/label.rs b/crates/bevy_core/src/label.rs
@@ -70,10 +70,10 @@ pub(crate) fn entity_labels_system(
     mut entity_labels: ResMut<EntityLabels>,
     // TODO: use change tracking when add/remove events are added
     // mut query: Query<(Entity, Changed<Labels>)>,
-    mut query: Query<(Entity, &Labels)>,
+    query: Query<(Entity, &Labels)>,
 ) {
     let entity_labels = entity_labels.deref_mut();
-    for (entity, labels) in &mut query.iter() {
+    for (entity, labels) in query.iter() {
         let current_labels = entity_labels
             .entity_labels
             .entry(entity)

diff --git a/crates/bevy_core/src/time/timer.rs b/crates/bevy_core/src/time/timer.rs
@@ -57,7 +57,7 @@ impl Timer {
 }
 
 pub(crate) fn timer_system(time: Res<Time>, mut query: Query<&mut Timer>) {
-    for mut timer in &mut query.iter() {
+    for mut timer in query.iter_mut() {
         timer.tick(time.delta_seconds);
     }
 }
diff --git a/crates/bevy_ecs/hecs/macros/src/lib.rs b/crates/bevy_ecs/hecs/macros/src/lib.rs
@@ -20,7 +20,7 @@ use proc_macro::TokenStream;
 use proc_macro2::Span;
 use proc_macro_crate::crate_name;
 use quote::quote;
-use syn::{parse_macro_input, DeriveInput, Path};
+use syn::{parse_macro_input, DeriveInput, Ident, Index, Lifetime, Path};
 
 /// Implement `Bundle` for a monomorphic struct
 ///
@@ -141,3 +141,76 @@ fn struct_fields(fields: &syn::Fields) -> (Vec<&syn::Type>, Vec<syn::Ident>) {
         syn::Fields::Unit => (Vec::new(), Vec::new()),
     }
 }
+
+fn get_idents(fmt_string: fn(usize) -> String, count: usize) -> Vec<Ident> {
+    (0..count)
+        .map(|i| Ident::new(&fmt_string(i), Span::call_site()))
+        .collect::<Vec<Ident>>()
+}
+
+fn get_lifetimes(fmt_string: fn(usize) -> String, count: usize) -> Vec<Lifetime> {
+    (0..count)
+        .map(|i| Lifetime::new(&fmt_string(i), Span::call_site()))
+        .collect::<Vec<Lifetime>>()
+}
+
+#[proc_macro]
+pub fn impl_query_set(_input: TokenStream) -> TokenStream {
+    let mut tokens = TokenStream::new();
+    let max_queries = 4;
+    let queries = get_idents(|i| format!("Q{}", i), max_queries);
+    let lifetimes = get_lifetimes(|i| format!("'q{}", i), max_queries);
+    let mut query_fns = Vec::new();
+    let mut query_fn_muts = Vec::new();
+    for i in 0..max_queries {
+        let query = &queries[i];
+        let lifetime = &lifetimes[i];
+        let fn_name = Ident::new(&format!("q{}", i), Span::call_site());
+        let fn_name_mut = Ident::new(&format!("q{}_mut", i), Span::call_site());
+        let index = Index::from(i);
+        query_fns.push(quote! {
+            pub fn #fn_name(&self) -> &Query<#lifetime, #query> {
+                &self.value.#index
+            }
+        });
+        query_fn_muts.push(quote! {
+            pub fn #fn_name_mut(&mut self) -> &mut Query<#lifetime, #query> {
+                &mut self.value.#index
+            }
+        });
+    }
+
+    for query_count in 1..=max_queries {
+        let query = &queries[0..query_count];
+        let lifetime = &lifetimes[0..query_count];
+        let query_fn = &query_fns[0..query_count];
+        let query_fn_mut = &query_fn_muts[0..query_count];
+        tokens.extend(TokenStream::from(quote! {
+            impl<#(#lifetime,)* #(#query: HecsQuery,)*> QueryTuple for (#(Query<#lifetime, #query>,)*) {
+                unsafe fn new(world: &World, component_access: &TypeAccess<ArchetypeComponent>) -> Self {
+                    (
+                        #(
+                            Query::<#query>::new(
+                                std::mem::transmute(world),
+                                std::mem::transmute(component_access),
+                            ),
+                        )*
+                    )
+                }
+
+                fn get_accesses() -> Vec<QueryAccess> {
+                    vec![
+                        #(<#query::Fetch as Fetch>::access(),)*
+                    ]
+                }
+            }
+
+            impl<#(#lifetime,)* #(#query: HecsQuery,)*> QuerySet<(#(Query<#lifetime, #query>,)*)> {
+                #(#query_fn)*
+                #(#query_fn_mut)*
+            }
+        }));
+    }
+
+    tokens
+}