X-Git-Url: https://ruin.nu/git/index.pl?a=blobdiff_plain;f=bs%2Ffleet.cpp;h=7361fe2122b486a4b32b11f7d69e44a3d27a857e;hb=33b6dc7ca95a05c26a78879413395c1588066b77;hp=cb15df528c6bce726a0f7dd52db026f7d57f5a31;hpb=5940cfda3be2c7b4ecbd64c814948b29139177c0;p=hbs.git

diff --git a/bs/fleet.cpp b/bs/fleet.cpp
index cb15df5..7361fe2 100644
--- a/bs/fleet.cpp
+++ b/bs/fleet.cpp
@@ -29,6 +29,7 @@ Fleet::Fleet()
 	m_iETA = 0;
 	m_sRace = "Cathaar";
 }
+
 Fleet::~Fleet(){
 }
 
@@ -174,3 +175,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];
+}
+
+