Added knowledge of weapons

Next step: knowledge of weapon's effects.

1 files changed, 4 insertions, 2 deletions

diff --git a/src/shooting.rs b/src/shooting.rs
index 4d87d86..bcdf783 100644
--- a/src/shooting.rs
+++ b/src/shooting.rs
@@ -4,6 +4,7 @@ use rand::distributions::{IndependentSample, Range};
 use actions::*;
 use math::*;
 use knowledge::*;
+use ships::*;
 
 pub fn shoot_smartly(knowledge: &Knowledge) -> Action {
     let shot = if knowledge.has_unknown_hits() {
@@ -12,8 +13,9 @@ pub fn shoot_smartly(knowledge: &Knowledge) -> Action {
     else {
         seek_shoot(&knowledge)
     };
-    
-    Action::Shoot(shot)
+
+    //TODO shoot other weapons
+    Action::Shoot(Weapon::SingleShot, shot)
 }
 
 fn seek_shoot(knowledge: &Knowledge) -> Point {