public bool HasEnoughFreeHousingSpace()
{
LogicCalendar calendar = this.m_level.GetCalendar();
LogicAvatar homeOwnerAvatar = this.m_level.GetHomeOwnerAvatar();
LogicUnitProduction unitProduction = this.m_level.GetGameObjectManagerAt(0).GetUnitProduction();
LogicDataTable characterTable = LogicDataTables.GetTable(LogicDataType.CHARACTER);
int freeHousingSpace = unitProduction.GetMaxTrainCount() - (homeOwnerAvatar.GetUnitsTotalCapacity() - unitProduction.GetTotalCount());
int requiredHousingSpace = 0;
for (int i = 0; i < characterTable.GetItemCount(); i++)
{
LogicCharacterData data = (LogicCharacterData)characterTable.GetItemAt(i);
if (calendar.IsProductionEnabled(data) && !data.IsSecondaryTroop())
{
int count = homeOwnerAvatar.GetUnitPresetCount(data, this.m_presetId);
if (count > 0)
{
requiredHousingSpace += data.GetHousingSpace() * count;
}
}
}
if (requiredHousingSpace <= freeHousingSpace)
{
LogicUnitProduction spellProduction = this.m_level.GetGameObjectManagerAt(0).GetSpellProduction();
LogicDataTable spellTable = LogicDataTables.GetTable(LogicDataType.SPELL);
int freeSpellHousingSpace = spellProduction.GetMaxTrainCount() - (homeOwnerAvatar.GetSpellsTotalCapacity() - spellProduction.GetTotalCount());
int requiredSpellHousingSpace = 0;
for (int i = 0; i < spellTable.GetItemCount(); i++)
{
LogicSpellData data = (LogicSpellData)spellTable.GetItemAt(i);
if (calendar.IsProductionEnabled(data))
{
int count = homeOwnerAvatar.GetUnitPresetCount(data, this.m_presetId);
if (count > 0)
{
requiredSpellHousingSpace += data.GetHousingSpace() * count;
}
}
}
return(requiredSpellHousingSpace <= freeSpellHousingSpace);
}
return(false);
}
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);
}