X-Git-Url: https://ruin.nu/git/?p=hbs.git;a=blobdiff_plain;f=bs%2Fplanet.cpp;h=c72fb4e61c39e5b9450cac19f3c97a82ac2b723f;hp=4fa69786a009e426ae8dd666724965d3aa89465c;hb=f7ce9d1a41bdeab4f12110c6911f662dc6e06ca3;hpb=595e711e719a6e6b675b1f40e3e141cd9b770fd9

diff --git a/bs/planet.cpp b/bs/planet.cpp
index 4fa6978..c72fb4e 100644
--- a/bs/planet.cpp
+++ b/bs/planet.cpp
@@ -17,6 +17,8 @@
 
 #include "planet.h"
 
+using namespace std;
+
 Planet::Planet()
 {
 	m_sRace = "Planet";
@@ -42,22 +44,37 @@ void Planet::setPlanetScore(unsigned i)
 
 //////////////////////////////////////////////////////////////////////////
 //
-int roids(std::string type, int tick = 0) const
+int Planet::roids(std::string type, int tick = 0) const
 {
-	ticks = m_Roids[type].size();
+	// const... I would like [] as for const types: int ticks = m_Roids[type].size();
+	
+	vector<int>const* roids = 0;
+	for (RoidList::const_iterator i = m_Roids.begin(); i != m_Roids.end(); ++i)
+	{
+		if (i->first == type)
+		{
+			roids = &i->second;
+			break;
+		}
+	}
+	if (roids == 0)
+		return 0;
+
+	int ticks = roids->size();
+
 	if( ticks == 0)
 		return 0;
 	if (ticks < tick)
-		return m_Roids[type][ticks];
-	return m_Roids[type][tick];
+		return roids->at(ticks);
+	return roids->at(tick);
 }
 
 //////////////////////////////////////////////////////////////////////////
 //
-void setRoids(std::string type, int number)
+void Planet::setRoids(std::string type, int number)
 {
 	if (m_Roids[type].size() == 0)
-		m_Roids.push_back(number);
+		m_Roids[type].push_back(number);
 	m_Roids[type][0] = number;
 }