- if (heroes[i].AttackAnimation() && heroes[i].AttackAnimation()->Running()) {
- heroes[i].AttackAnimation()->DrawCenter(screen, heroesPositions[i] + offset);
- } else if (heroes[i].SpellAnimation() && heroes[i].SpellAnimation()->Running()) {
- heroes[i].SpellAnimation()->DrawCenter(screen, heroesPositions[i] + offset);
+ if (heroes[i].GetAnimation().Running()) {
+ heroes[i].GetAnimation().DrawCenter(screen, heroesPositions[i] + offset);