public int GetResourceCost(LogicResourceData resourceData)
{
int cost = 0;
LogicAvatar homeOwnerAvatar = this.m_level.GetHomeOwnerAvatar();
LogicCalendar calendar = this.m_level.GetCalendar();
LogicDataTable table = LogicDataTables.GetTable(LogicDataType.CHARACTER);
for (int i = 0; i < table.GetItemCount(); i++)
{
LogicCharacterData data = (LogicCharacterData)table.GetItemAt(i);
if (calendar.IsProductionEnabled(data) && !data.IsSecondaryTroop())
{
int count = homeOwnerAvatar.GetUnitPresetCount(data, this.m_presetId);
if (count > 0)
{
if (data.GetTrainingResource() == resourceData)
{
cost += count * calendar.GetTrainingCost(data, homeOwnerAvatar.GetUnitUpgradeLevel(data));
}
}
}
}
table = LogicDataTables.GetTable(LogicDataType.SPELL);
for (int i = 0; i < table.GetItemCount(); i++)
{
LogicSpellData data = (LogicSpellData)table.GetItemAt(i);
if (calendar.IsProductionEnabled(data))
{
int count = homeOwnerAvatar.GetUnitPresetCount(data, this.m_presetId);
if (count > 0)
{
if (data.GetTrainingResource() == resourceData)
{
cost += count * calendar.GetTrainingCost(data, homeOwnerAvatar.GetUnitUpgradeLevel(data));
}
}
}
}
return(cost);
}
public void AddUnitsToQueue(LogicCombatItemData data, int count)
{
LogicCalendar calendar = this.m_level.GetCalendar();
LogicAvatar homeOwnerAvatar = this.m_level.GetHomeOwnerAvatar();
LogicClientAvatar playerAvatar = this.m_level.GetPlayerAvatar();
LogicGameObjectManager gameObjectManager = this.m_level.GetGameObjectManagerAt(0);
LogicUnitProduction production = gameObjectManager.GetUnitProduction();
if (data.GetCombatItemType() != LogicCombatItemData.COMBAT_ITEM_TYPE_CHARACTER)
{
if (data.GetCombatItemType() != LogicCombatItemData.COMBAT_ITEM_TYPE_SPELL)
{
return;
}
production = gameObjectManager.GetSpellProduction();
}
if (production != null)
{
int trainCost = calendar.GetTrainingCost(data, homeOwnerAvatar.GetUnitUpgradeLevel(data));
for (int i = 0; i < count; i++)
{
if (production.CanAddUnitToQueue(data, true) &&
playerAvatar.HasEnoughResources(data.GetTrainingResource(), trainCost, false, null, false))
{
playerAvatar.CommodityCountChangeHelper(0, data.GetTrainingResource(), -trainCost);
production.AddUnitToQueue(data, production.GetSlotCount(), true);
}
}
}
}