From 70ec00285128f8f9f5fa0e848950212e6a235d43 Mon Sep 17 00:00:00 2001
From: Justin Worthe <justin.worthe@gmail.com>
Date: Fri, 14 Aug 2015 22:12:56 +0200
Subject: Added missing sensors

---
 include/game_state.h | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

(limited to 'include/game_state.h')

diff --git a/include/game_state.h b/include/game_state.h
index 880557e..c129d28 100644
--- a/include/game_state.h
+++ b/include/game_state.h
@@ -5,6 +5,7 @@
 #include "player_missile.h"
 #include "shield.h"
 #include "spaceship.h"
+#include "building.h"
 #include <vector>
 #include <string>
 #include <istream>
@@ -14,12 +15,14 @@ class GameState
 {
 public:
     GameState(std::istream &&mapFile);
-    void logState();
+    void logState() const;
 
     const std::vector<Alien>& aliens() const { return _aliens; }
     const std::vector<EnemyBullet>& bullets() const { return _bullets; }
     const std::vector<PlayerMissile>& missiles() const { return _missiles; }
     const std::vector<Shield>& shields() const { return _shields; }
+    const std::vector<Building>& missileControllers() const { return _missileControllers; }
+    const std::vector<Building>& alienFactories() const { return _alienFactories; }
     
     const std::unique_ptr<Spaceship>& playerSpaceship() const { return _playerSpaceship; }
     const std::unique_ptr<Spaceship>& enemySpaceship() const { return _enemySpaceship; }
@@ -31,8 +34,11 @@ private:
     std::vector<EnemyBullet> _bullets;
     std::vector<PlayerMissile> _missiles;
     std::vector<Shield> _shields;
+    std::vector<Building> _missileControllers;
+    std::vector<Building> _alienFactories;
+
     std::unique_ptr<Spaceship> _playerSpaceship;
     std::unique_ptr<Spaceship> _enemySpaceship;
-
+    
     int addEntity(int x, int y, char type);
 };
-- 
cgit v1.2.3