Add QueryState related operations

Cargo.toml

@@ -54,6 +54,30 @@ crate-type = ["lib"]
 name = "query_get_sparse"
 crate-type = ["lib"]
 
+[[example]]
+name = "query_state_iter"
+crate-type = ["lib"]
+
+[[example]]
+name = "query_state_iter_sparse"
+crate-type = ["lib"]
+
+[[example]]
+name = "query_state_for_each"
+crate-type = ["lib"]
+
+[[example]]
+name = "query_state_for_each_sparse"
+crate-type = ["lib"]
+
+[[example]]
+name = "query_state_get"
+crate-type = ["lib"]
+
+[[example]]
+name = "query_state_get_sparse"
+crate-type = ["lib"]
+
 [[example]]
 name = "world_get"
 crate-type = ["lib"]

examples/query_state_for_each.rs

@@ -0,0 +1,14 @@
+use bevy_ecs::prelude::*;
+
+#[derive(Component)]
+pub struct A(f32);
+
+#[derive(Component)]
+pub struct B(f32);
+
+#[no_mangle]
+fn query_state_for_each(mut query: QueryState<(&mut A, &B)>, world: &mut World) {
+    query.for_each_mut(world, |(mut a, b)| {
+        a.0 += b.0;
+    })
+}

examples/query_state_for_each_sparse.rs

@@ -0,0 +1,16 @@
+use bevy_ecs::prelude::*;
+
+#[derive(Component)]
+#[component(storage = "SparseSet")]
+pub struct A(f32);
+
+#[derive(Component)]
+#[component(storage = "SparseSet")]
+pub struct B(f32);
+
+#[no_mangle]
+fn query_for_each(mut query: QueryState<(&mut A, &B)>, world: &mut World) {
+    query.for_each_mut(world, |(mut a, b)| {
+        a.0 += b.0;
+    })
+}

examples/query_state_get.rs

@@ -0,0 +1,13 @@
+use bevy_ecs::prelude::*;
+use std::hint::black_box;
+
+#[derive(Component)]
+pub struct A(f32);
+
+#[derive(Component)]
+pub struct B(f32);
+
+#[no_mangle]
+fn query_get(mut query: QueryState<(&mut A, &B)>, entity: Entity, world: &mut World) {
+    let _ = black_box(query.get_mut(world, entity));
+}

examples/query_state_get_sparse.rs

@@ -0,0 +1,15 @@
+use bevy_ecs::prelude::*;
+use std::hint::black_box;
+
+#[derive(Component)]
+#[component(storage = "SparseSet")]
+pub struct A(f32);
+
+#[derive(Component)]
+#[component(storage = "SparseSet")]
+pub struct B(f32);
+
+#[no_mangle]
+fn query_get(mut query: QueryState<(&mut A, &B)>, entity: Entity, world: &mut World) {
+    let _ = black_box(query.get_mut(world, entity));
+}

examples/query_state_iter.rs

@@ -0,0 +1,14 @@
+use bevy_ecs::prelude::*;
+
+#[derive(Component)]
+pub struct A(f32);
+
+#[derive(Component)]
+pub struct B(f32);
+
+#[no_mangle]
+fn query_iter(mut query: QueryState<(&mut A, &B)>, world: &mut World) {
+    for (mut a, b) in query.iter_mut(world) {
+        a.0 += b.0;
+    }
+}

examples/query_state_iter_sparse.rs

@@ -0,0 +1,16 @@
+use bevy_ecs::prelude::*;
+
+#[derive(Component)]
+#[component(storage = "SparseSet")]
+pub struct A(f32);
+
+#[derive(Component)]
+#[component(storage = "SparseSet")]
+pub struct B(f32);
+
+#[no_mangle]
+fn query_iter(mut query: QueryState<(&mut A, &B)>, world: &mut World) {
+    for (mut a, b) in query.iter_mut(world) {
+        a.0 += b.0;
+    }
+}