X-Git-Url: https://ruin.nu/git/?p=hbs.git;a=blobdiff_plain;f=bs%2Funittype.h;h=ccff1d1a958a34a7162ca2df70b6f2a31c0b39b5;hp=b01de80ef3c247107bfd3fb6b7c601aeef1a3f6f;hb=924d4cd0f8c9c340c8ba6f2e5759fed8f0c83573;hpb=6ab17155b82923f4263f1c536133e657970239a4

diff --git a/bs/unittype.h b/bs/unittype.h
index b01de80..ccff1d1 100644
--- a/bs/unittype.h
+++ b/bs/unittype.h
@@ -134,10 +134,13 @@ public:
 	int  EMP() const;
 
 	/** Sets the resource cost for this unittype. Used for example for score calculation and so on. */
-	void setTotalResources(int iTR);
+	void setResources(std::string type, int i);
+	void setResources(std::map<std::string, int> res);
 	/** Returns the number of total resources this unittype cost. */
 	int totRes() const;
 
+	std::map<std::string, int> resources();
+
 	/** Sets the fuelcost for this unittype */
 	void setFuel(int iFuel);
 	/** Returns the fuelcost */
@@ -158,17 +161,17 @@ protected:
 	int			m_iRace;            //!< Not really the race, but an indiaction on what race can use it.. 
 	std::string	m_sClass;
 	std::vector<std::string>	m_vTarget;
-	int			m_iInitiative;
-	int			m_iAgility;
-	int			m_iWeaponSpeed;
-	int			m_iGuns;
-	int			m_iPower;
-	int			m_iArmor;
-	int			m_iEMP;
-	int			m_iTotalResources;
-	std::string	m_sType;            //!< normal,emp,cloak,steal,pod
-	int			m_iETA;
-	int			m_iFuel;
+	int			  m_iInitiative;
+	int			  m_iAgility;
+	int			  m_iWeaponSpeed;
+	int			  m_iGuns;
+	int			  m_iPower;
+	int			  m_iArmor;
+	int		      m_iEMP;
+	std::map<std::string, int> m_Resources;
+	std::string   m_sType;            //!< normal,emp,cloak,steal,pod
+	int			  m_iETA;
+	int			  m_iFuel;
 };
 
 #endif