Some changes..

[hbs.git] / bs / fleet.cpp

diff --git a/bs/fleet.cpp b/bs/fleet.cpp
index a57d25e86db06c0a359928fee7b3a70e50729d6b..35de49b849adc58016de508afd2217036f50da2d 100644 (file)
--- a/bs/fleet.cpp
+++ b/bs/fleet.cpp
@@ -17,7 +17,51 @@
 
 #include "fleet.h"
 
-Fleet::Fleet(){
+using namespace std;
+
+Fleet::Fleet()
+{
 }
 Fleet::~Fleet(){
 }
+
+//////////////////////////////////////////////////////////////////////////
+//
+void Fleet::setName(string sName)
+{
+       m_sName = sName;
+}
+
+//////////////////////////////////////////////////////////////////////////
+//
+string Fleet::Name()
+{
+       return m_sName;
+}
+
+//////////////////////////////////////////////////////////////////////////
+//
+/** This function first sets the race, then it iterates through the the 
+ * s_Races and checks if it finds the race it returns true, if it reaches
+ * the end without finding it it returns false.
+ */
+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)
+                       return true;
+       }
+       return false;
+}
+
+//////////////////////////////////////////////////////////////////////////
+//
+string Fleet::Race()
+{
+       return m_sRace;
+}
+
+