added the possibility to edit fleets

[hbs.git] / bs / fleet.cpp

diff --git a/bs/fleet.cpp b/bs/fleet.cpp
index cb15df528c6bce726a0f7dd52db026f7d57f5a31..11334172decbda3f477e0c93f2db55320876fb7e 100644 (file)
--- a/bs/fleet.cpp
+++ b/bs/fleet.cpp
@@ -174,3 +174,26 @@ unsigned Fleet::score(int tick = 0) const
        return tot_score;
 }
 
+//////////////////////////////////////////////////////////////////////////
+//
+void Fleet::setFleet(string unittype, int number)
+{
+       if (m_Fleet[unittype].size() == 0)
+       {
+               m_Fleet[unittype].push_back(number);
+               return;
+       }
+       m_Fleet[unittype][0] = number;
+}
+
+//////////////////////////////////////////////////////////////////////////
+//
+int     Fleet::fleet(string unittype, int tick = 0)
+{
+       if (m_Fleet[unittype].size() == 0)
+               return 0;
+
+       return m_Fleet[unittype][tick];
+}
+
+