//////////////////////////////////////////////////////////////////////////
//
-int UnitType::Race() const
+int UnitType::race() const
{
return m_iRace;
}
//////////////////////////////////////////////////////////////////////////
//
-void UnitType::setClass(string sClass)
+void UnitType::setUnitClass(string sClass)
{
m_sClass = sClass;
}
//////////////////////////////////////////////////////////////////////////
//
-string UnitType::Class() const
+string UnitType::unitClass() const
{
return m_sClass;
}
//////////////////////////////////////////////////////////////////////////
//
-vector<string> UnitType::Target() const
+vector<string> UnitType::target() const
{
return m_vTarget;
}
//////////////////////////////////////////////////////////////////////////
//
-string UnitType::Target(int index) const
+string UnitType::target(int index) const
{
return m_vTarget[index];
}
//////////////////////////////////////////////////////////////////////////
//
-int UnitType::Initiative() const
+int UnitType::initiative() const
{
return m_iInitiative;
}
//////////////////////////////////////////////////////////////////////////
//
-int UnitType::Agility() const
+int UnitType::agility() const
{
return m_iAgility;
}
//////////////////////////////////////////////////////////////////////////
//
-int UnitType::WeaponSpeed() const
+int UnitType::weaponSpeed() const
{
return m_iWeaponSpeed;
}
//////////////////////////////////////////////////////////////////////////
//
-int UnitType::Guns() const
+int UnitType::guns() const
{
return m_iGuns;
}
//////////////////////////////////////////////////////////////////////////
//
-int UnitType::Power() const
+int UnitType::power() const
{
return m_iPower;
}
//////////////////////////////////////////////////////////////////////////
//
-int UnitType::Armor() const
+int UnitType::armor() const
{
return m_iArmor;
}
//////////////////////////////////////////////////////////////////////////
//
-void UnitType::setTotalResources(int iTR)
+void UnitType::setResources(std::string type, int i)
{
- m_iTotalResources = iTR;
+ m_Resources[type] = i;
}
//////////////////////////////////////////////////////////////////////////
//
-int UnitType::TotRes() const
+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;
}
//////////////////////////////////////////////////////////////////////////
//////////////////////////////////////////////////////////////////////////
//
-int UnitType::Fuel() const
+int UnitType::fuel() const
{
return m_iFuel;
}
//////////////////////////////////////////////////////////////////////////
//
-string UnitType::Type() const
+string UnitType::type() const
{
return m_sType;
}