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 std::collections::HashMap;
use bevy::{ use bevy::{
ecs::entity::Entities,
prelude::*, prelude::*,
tasks::{prelude::*, Task}, 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() { for entity in removed.iter() {
if entities.contains(entity) {
commands.entity(entity).remove::<Calculating>(); commands.entity(entity).remove::<Calculating>();
} }
}
} }
fn negotiate_path( fn negotiate_path(

View File

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