projects/blackout
1
0
Fork 0
Code Pull Requests Projects Releases Wiki Activity

Even more cleanup.

Browse Source
This commit is contained in:
Nolan Darilek 2021-07-27 20:26:16 -05:00
parent d28dd051e4
commit 7f61eb3a90
1 changed files with 15 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
src/pathfinding.rs
View File

@ -226,31 +226,28 @@ fn calculate_path(
}
}
fn poll_tasks(
mut commands: Commands,
mut query: Query<(Entity, &mut Calculating)>,
destinations: Query<&Destination>,
) {
fn poll_tasks(mut commands: Commands, mut query: Query<(Entity, &mut Calculating)>) {
for (entity, mut calculating) in query.iter_mut() {
println!("Got a calculating task for {:?}", entity);
if destinations.get(entity).is_ok() {
if let Some(result) = future::block_on(future::poll_once(&mut **calculating)) {
println!("Got a result, setting and returning");
if let Some(path) = result {
commands.entity(entity).insert(path);
} else {
commands.entity(entity).insert(NoPath);
}
println!("Removing calculation task");
commands.entity(entity).remove::<Calculating>();
if let Some(result) = future::block_on(future::poll_once(&mut **calculating)) {
println!("Got a result, setting and returning");
if let Some(path) = result {
commands.entity(entity).insert(path);
} else {
commands.entity(entity).insert(NoPath);
}
} else {
println!("Destination was removed, clearing task");
println!("Removing calculation task");
commands.entity(entity).remove::<Calculating>();
}
}
}
fn remove_destination(mut commands: Commands, removed: RemovedComponents<Destination>) {
for entity in removed.iter() {
commands.entity(entity).remove::<Calculating>();
}
}
fn negotiate_path(
mut commands: Commands,
mut query: Query<(
@ -336,6 +333,7 @@ pub struct PathfindingPlugin;
impl Plugin for PathfindingPlugin {
fn build(&self, app: &mut AppBuilder) {
app.add_system(calculate_path.system())
.add_system_to_stage(CoreStage::PostUpdate, remove_destination.system())
.add_system(poll_tasks.system())
.add_system(negotiate_path.system())
.add_system_to_stage(CoreStage::PostUpdate, remove_calculating.system());