some

[hbs.git] / bs / unittype.cpp

diff --git a/bs/unittype.cpp b/bs/unittype.cpp
index 69b2d2d9c09b3f9b20cb2b36c7223437fc6c43ee..836ddb95056f0d79978e78a48d4d5f9735762006 100644 (file)
--- a/bs/unittype.cpp
+++ b/bs/unittype.cpp
@@ -40,7 +40,7 @@ void UnitType::setName(string sName)
 
 //////////////////////////////////////////////////////////////////////////
 //
-string UnitType::Name()
+string UnitType::Name() const
 {
        return m_sName;
 }
@@ -54,37 +54,37 @@ void UnitType::setRace(int iRace)
 
 //////////////////////////////////////////////////////////////////////////
 //
-int UnitType::Race()
+int UnitType::race() const
 {
        return m_iRace;
 }
 
 //////////////////////////////////////////////////////////////////////////
 //
-void UnitType::setClass(int iClass)
+void UnitType::setUnitClass(string sClass)
 {
-       m_iClass = iClass;
+       m_sClass = sClass;
 }
 
 //////////////////////////////////////////////////////////////////////////
 //
-int UnitType::iClass()
+string UnitType::unitClass() const
 {
-       return m_iClass;
+       return m_sClass;
 }
 
 //////////////////////////////////////////////////////////////////////////
 //
-void UnitType::setTarget(vector<int> Target)
+void UnitType::setTarget(vector<string> Target)
 {
        m_vTarget = Target;
 }
 
 //////////////////////////////////////////////////////////////////////////
 //
-void UnitType::addTarget(int iTarget)
+void UnitType::addTarget(string target)
 {
-       m_vTarget.push_back(iTarget);
+       m_vTarget.push_back(target);
 }
 
 //////////////////////////////////////////////////////////////////////////
@@ -93,33 +93,33 @@ void UnitType::addTarget(int iTarget)
  * If it finds the position it inserts the target there, if it reaches the end 
  * before it finds the correct possition it adds the target to the end. 
  */
-void UnitType::insTarget(int iTarget, int index = 0)
+void UnitType::insTarget(string target, int index)
 {
 
-       vector<int>::iterator i = m_vTarget.begin();    
+       vector<string>::iterator i = m_vTarget.begin(); 
 
        for (int j = 0; j < index; j++, i++)
        {
                if (i == m_vTarget.end())
                {
-                       m_vTarget.push_back(iTarget);
+                       m_vTarget.push_back(target);
                        return;
                }
        }
 
-       m_vTarget.insert(i, iTarget);
+       m_vTarget.insert(i, target);
 }
 
 //////////////////////////////////////////////////////////////////////////
 //
-vector<int> UnitType::Target()
+vector<string> UnitType::target() const
 {
        return m_vTarget;
 }
 
 //////////////////////////////////////////////////////////////////////////
 //
-int    UnitType::Target(int index)
+string UnitType::target(int index) const
 {
        return m_vTarget[index];
 }
@@ -133,7 +133,7 @@ void UnitType::setInitiative(int iInit)
 
 //////////////////////////////////////////////////////////////////////////
 //
-int     UnitType::Initiative()
+int     UnitType::initiative() const
 {
        return m_iInitiative;
 }
@@ -147,7 +147,7 @@ void UnitType::setAgility (int iAgil)
 
 //////////////////////////////////////////////////////////////////////////
 //
-int  UnitType::Agility()
+int  UnitType::agility() const
 {
        return m_iAgility;
 }
@@ -161,7 +161,7 @@ void UnitType::setWeaponSpeed(int iWPSP)
 
 //////////////////////////////////////////////////////////////////////////
 //
-int  UnitType::WeaponSpeed()
+int  UnitType::weaponSpeed() const
 {
        return m_iWeaponSpeed;
 }
@@ -175,7 +175,7 @@ void UnitType::setGuns(int iGuns)
 
 //////////////////////////////////////////////////////////////////////////
 //
-int UnitType::Guns()
+int UnitType::guns() const
 {
        return m_iGuns;
 }
@@ -189,7 +189,7 @@ void UnitType::setPower(int iPower)
 
 //////////////////////////////////////////////////////////////////////////
 //
-int  UnitType::Power()
+int  UnitType::power() const
 {
        return m_iPower;
 }
@@ -203,7 +203,7 @@ void UnitType::setArmor(int iArmor)
 
 //////////////////////////////////////////////////////////////////////////
 //
-int  UnitType::Armor()
+int  UnitType::armor() const
 {
        return m_iArmor;
 }
@@ -217,23 +217,40 @@ void UnitType::setEMP(int iEMP)
 
 //////////////////////////////////////////////////////////////////////////
 //
-int  UnitType::EMP()
+int  UnitType::EMP() const
 {
        return m_iEMP;
 }
 
 //////////////////////////////////////////////////////////////////////////
 //
-void UnitType::setTotalResources(int iTR)
+void UnitType::setResources(std::string type, int i)
 {
-       m_iTotalResources = iTR;
+       m_Resources[type] = i;
 }
 
 //////////////////////////////////////////////////////////////////////////
 //
-int UnitType::TotRes()
+void UnitType::setResources(map<std::string, int> res)
 {
-       return m_iTotalResources;
+       m_Resources = res;
+}
+
+//////////////////////////////////////////////////////////////////////////
+//
+std::map<std::string, int> UnitType::resources()
+{
+       return m_Resources;
+}
+
+//////////////////////////////////////////////////////////////////////////
+//
+int UnitType::totRes() const
+{
+       int totres = 0;
+       for (map<string, int>::const_iterator i = m_Resources.begin(); i != m_Resources.end(); ++i)
+               totres += i->second;
+       return totres;
 }
 
 //////////////////////////////////////////////////////////////////////////
@@ -245,7 +262,7 @@ void UnitType::setFuel(int iFuel)
 
 //////////////////////////////////////////////////////////////////////////
 //
-int  UnitType::Fuel()
+int  UnitType::fuel() const
 {
        return m_iFuel;
 }
@@ -259,22 +276,22 @@ void UnitType::setETA(int iETA)
 
 //////////////////////////////////////////////////////////////////////////
 //
-int  UnitType::ETA()
+int  UnitType::ETA() const
 {
        return m_iETA;
 }
 
 //////////////////////////////////////////////////////////////////////////
 //
-void UnitType::setType(int iType)
+void UnitType::setType(string type)
 {
-       m_iType = iType;
+       m_sType = type;
 }
 
 //////////////////////////////////////////////////////////////////////////
 //
-int  UnitType::Type()
+string  UnitType::type() const
 {
-       return m_iType;
+       return m_sType;
 }