X-Git-Url: https://ruin.nu/git/?p=hbs.git;a=blobdiff_plain;f=bs%2Ffleet.cpp;h=11334172decbda3f477e0c93f2db55320876fb7e;hp=cb15df528c6bce726a0f7dd52db026f7d57f5a31;hb=07acf9a898e295587b9034bee6b8682e11a97086;hpb=0fcfaff680c6974a9b3f2a63704a8eb788d3fe94

diff --git a/bs/fleet.cpp b/bs/fleet.cpp
index cb15df5..1133417 100644
--- 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];
+}
+
+