From 27c0795fb70739ce5609a0f424d80491d4a8c5a1 Mon Sep 17 00:00:00 2001
From: troido <troido@protonmail.com>
Date: Mon, 2 Mar 2020 12:02:54 +0100
Subject: added monster ai

---
 src/systems/updatecooldowns.rs | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

(limited to 'src/systems/updatecooldowns.rs')

diff --git a/src/systems/updatecooldowns.rs b/src/systems/updatecooldowns.rs
index 73ca770..705561d 100644
--- a/src/systems/updatecooldowns.rs
+++ b/src/systems/updatecooldowns.rs
@@ -6,16 +6,18 @@ use specs::{
 	Join
 };
 
-use crate::components::ControlCooldown;
+use crate::components::{Controller, ControlCooldown};
 
 
 pub struct UpdateCooldowns;
 impl <'a> System<'a> for UpdateCooldowns {
 	type SystemData = (
 		Entities<'a>,
-		WriteStorage<'a, ControlCooldown>
+		WriteStorage<'a, ControlCooldown>,
+		WriteStorage<'a, Controller>
 	);
-	fn run(&mut self, (entities, mut cooldowns): Self::SystemData) {
+	fn run(&mut self, (entities, mut cooldowns, mut controllers): Self::SystemData) {
+		controllers.clear();
 		let mut to_remove = Vec::new();
 		for (entity, cooldown) in (&entities, &mut cooldowns).join() {
 			if cooldown.amount > 0 {
-- 
cgit