Changing how gamestate is interpreted into network

1 files changed, 6 insertions, 2 deletions

diff --git a/include/game_state.h b/include/game_state.h
index 482dd7d..880557e 100644
--- a/include/game_state.h
+++ b/include/game_state.h
@@ -8,6 +8,7 @@
 #include <vector>
 #include <string>
 #include <istream>
+#include <memory>
 
 class GameState
 {
@@ -19,7 +20,9 @@ public:
     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<Spaceship>& spaceships() const { return _spaceships; }
+    
+    const std::unique_ptr<Spaceship>& playerSpaceship() const { return _playerSpaceship; }
+    const std::unique_ptr<Spaceship>& enemySpaceship() const { return _enemySpaceship; }
 
     std::vector<bool> toBitArray() const;
     
@@ -28,7 +31,8 @@ private:
     std::vector<EnemyBullet> _bullets;
     std::vector<PlayerMissile> _missiles;
     std::vector<Shield> _shields;
-    std::vector<Spaceship> _spaceships;
+    std::unique_ptr<Spaceship> _playerSpaceship;
+    std::unique_ptr<Spaceship> _enemySpaceship;
 
     int addEntity(int x, int y, char type);
 };