X-Git-Url: https://ruin.nu/git/index.pl?a=blobdiff_plain;f=bs%2Ffleet.cpp;h=2bd245ba29af67dfc1abadb6a4bdd33f12e35528;hb=ba016135c4c232c79cb35c92f239579ca3d84df3;hp=a46a50a2ad51944c28a979e45992f99748deecfa;hpb=56b09bf912d2965744e2e78a30fc68d73ebe401c;p=hbs.git

diff --git a/bs/fleet.cpp b/bs/fleet.cpp
index a46a50a..2bd245b 100644
--- a/bs/fleet.cpp
+++ b/bs/fleet.cpp
@@ -22,12 +22,12 @@ using namespace std;
 
 //Static variables
 map<string, vector<int> > Fleet::s_Races;
-map<string, UnitType > Fleet::s_Units;
+UnitList Fleet::s_Units;
 
 Fleet::Fleet()
 {
 	m_iETA = 0;
-	m_sRace = "Generic";
+	m_sRace = "Cathaar";
 }
 Fleet::~Fleet(){
 }
@@ -110,11 +110,11 @@ void Fleet::setRaces(map<string, vector<int> >& races)
 
 //////////////////////////////////////////////////////////////////////////
 //
-void Fleet::setUnits(map<string, UnitType>& units)
+void Fleet::setUnits(UnitList& units)
 {
 	s_Units = units;
 
-	for (map<string, UnitType >::iterator i = s_Units.begin(); i != s_Units.end(); i++)
+	for (UnitList::iterator i = s_Units.begin(); i != s_Units.end(); i++)
 	{
 		cerr << s_Units[(*i).first].Name() << "\t\t"
 			<< s_Units[(*i).first].Race() <<"\t"
@@ -145,10 +145,15 @@ const map<string, vector<int> >& Fleet::Races()
 
 //////////////////////////////////////////////////////////////////////////
 //
-const map<string, UnitType>& Fleet::Units()
+const UnitList& Fleet::Units()
 {
 	return s_Units;
 }
 
-
+//////////////////////////////////////////////////////////////////////////
+//
+vector<int> Fleet::RacesAllowed() const
+{
+	return s_Races[m_sRace];
+}