X-Git-Url: https://ruin.nu/git/?a=blobdiff_plain;f=bs%2Ffleet.h;h=0d81fd2b5de4637c15d5970385ee77d29a10f495;hb=e07f49323a1936c90605d06f6f63e4c39b4c4aca;hp=16422dd7b80b163ed2dec9913ec0b66f1c46e058;hpb=a779cd198be469897a9dec45c9b2ba38b183de2f;p=hbs.git

diff --git a/bs/fleet.h b/bs/fleet.h
index 16422dd..0d81fd2 100644
--- a/bs/fleet.h
+++ b/bs/fleet.h
@@ -18,17 +18,60 @@
 #ifndef FLEET_H
 #define FLEET_H
 
+#include <map>
+#include <string>
+#include <vector>
 
-/**
+#include "unittype.h"
+
+//!This is the engine for the whole battlesystem.
+/**This is the engine for the whole battlesystem.
+  *	One of the few parts that I plan make fully portable.
+  *\todo LOTS (I think ;)
   *@author Michael Andreen
-  *This is the engine for the whole battlesystem.
   */
-
 class Fleet 
 {
 public: 
 	Fleet();
 	~Fleet();
+
+	/**Sets the name that represents this fleet. Might be different a name
+	 * like foobar or some coordinates like 1:1:1. The name doesn't have to be unique,
+	 * since it's up to the rest of the program to handle that part.
+	 * \see Name
+	 */
+	void   setName(std::string sName);
+	/**Returns the name of this fleet.
+	 * \see setName
+	 */
+	std::string Name();
+
+	/**The race string decides what type of ships this fleet can have.
+	 * The values must be feeded into this class.
+	 * \param sRace This is just the name of the race. It's case-sensitive.
+	 * \return If the race is available this function returns true, if not false is returned. The race is set in both cases though.
+	 * \see Race
+	 */
+	bool   setRace(std::string sRace);
+	/**Just returns what race this fleet belongs to.. 
+	 * \return The race, represented as a string.
+	 * \see setRace
+	 */
+	std::string Race();
+
+	/**Returns the total number of ships in this fleet
+	 */
+	int NumberOfShips();
+
+
+protected:
+	std::string	m_sName;
+	std::string	m_sRace;
+	std::map<std::string, std::vector<int> >	m_Fleet;
+
+	static std::map<std::string, UnitType>	s_Units;
+	static std::map<std::string, std::vector<int> > s_Races;
 };
 
 #endif