adding, removing and chaning fleets now works.

[hbs.git] / bs / planet.h

diff --git a/bs/planet.h b/bs/planet.h
index f98d7234c74e5bdf30c07ae646f68be5c1beb268..e59007ceebbc46c307d78cc84d96122deb092203 100644 (file)
--- a/bs/planet.h
+++ b/bs/planet.h
@@ -20,6 +20,8 @@
 
 #include "fleet.h"
 
+typedef std::map<std::string, std::vector<int> > RoidList;
+
 /**This class is the implementation of a planet.
  *
  * Yes I know.. In real life you wouldn't call a planet a fleet, but this isn't
@@ -32,8 +34,20 @@
 class Planet : public Fleet  {
 public: 
        Planet();
-       ~Planet();
-       unsigned score();
+       virtual ~Planet();
+
+       unsigned planetScore() const;
+       void setPlanetScore(unsigned i);
+
+       int roids(std::string type, int tick = 0) const;
+       void setRoids(std::string type, int number);
+
+       void runBattle(std::vector<Fleet*> friendly, std::vector<Fleet*> hostile);
+
+protected:
+       unsigned m_iScore;
+       RoidList m_Roids;
+       
 };
 
 #endif