Various crash fixes.

This commit is contained in:
Nolan Darilek 2022-01-20 18:06:20 -06:00
parent 488037d074
commit e4ee87ee27
2 changed files with 23 additions and 15 deletions

View File

@ -1,6 +1,7 @@
use std::collections::HashMap;
use bevy::{
ecs::entity::Entities,
prelude::*,
tasks::{prelude::*, Task},
};
@ -258,10 +259,16 @@ fn poll_tasks(mut commands: Commands, mut query: Query<(Entity, &mut Calculating
}
}
fn remove_destination(mut commands: Commands, removed: RemovedComponents<Destination>) {
fn remove_destination(
mut commands: Commands,
entities: &Entities,
removed: RemovedComponents<Destination>,
) {
for entity in removed.iter() {
if entities.contains(entity) {
commands.entity(entity).remove::<Calculating>();
}
}
}
fn negotiate_path(

View File

@ -314,7 +314,7 @@ fn remove_visible(
continue;
}
visible_entities.remove(&removed);
let map = map.single();
if let Ok(map) = map.get_single() {
let mut cache = HashMap::new();
viewshed.update(
&viewer_entity,
@ -329,6 +329,7 @@ fn remove_visible(
);
}
}
}
}
fn update_revealed_tiles(