Added the possibility to add new battles.

[hbs.git] / bs / fleet.cpp

diff --git a/bs/fleet.cpp b/bs/fleet.cpp
index 35de49b849adc58016de508afd2217036f50da2d..4989573d87f069b3e3bd6e27c4b009e9a0e7268c 100644 (file)
--- a/bs/fleet.cpp
+++ b/bs/fleet.cpp
@@ -19,6 +19,10 @@
 
 using namespace std;
 
+//Static variables
+map<string, vector<int> > Fleet::s_Races;
+map<string, UnitType > Fleet::s_Units;
+
 Fleet::Fleet()
 {
 }
@@ -48,7 +52,6 @@ string Fleet::Name()
 bool Fleet::setRace(std::string sRace)
 {
        m_sRace = sRace;
-  s_Units.begin();
        for (map<string, vector<int> >::iterator i = s_Races.begin(); i != s_Races.end(); i++)
        {
                if (m_sRace == (*i).first)
@@ -64,4 +67,20 @@ string Fleet::Race()
        return m_sRace;
 }
 
+//////////////////////////////////////////////////////////////////////////
+//
+/** This function iterates through m_Fleet and adds all numbers together to
+ * produce a total.
+ */
+int Fleet::NumberOfShips()
+{
+       int total = 0;
+
+       for (map<string, vector<int> >::iterator i = m_Fleet.begin(); i != m_Fleet.end(); i++)
+       {
+               total += m_Fleet[(*i).first][0];
+       }
+
+       return total;
+}