Moved ship placement knowledge out to be one per ship

1 files changed, 12 insertions, 0 deletions

diff --git a/src/ships.rs b/src/ships.rs
index 344f9ed..ef29fe9 100644
--- a/src/ships.rs
+++ b/src/ships.rs
@@ -54,4 +54,16 @@ impl Ship {
             &Submarine => 3
         }
     }
+
+    pub fn all_types() -> Vec<Ship> {
+        use Ship::*;
+        
+        vec!(
+            Battleship,
+            Carrier,
+            Cruiser,
+            Destroyer,
+            Submarine
+        )
+    }
 }