- if (GetHero().HasWeapon()) {
- equipmentMenu.Add(GetHero().Weapon()->Name(), GetHero().Weapon(), true, GetHero().Weapon()->MenuIcon());
- } else {
- equipmentMenu.Add(parent->Res().noEquipmentText, 0, false);
- }
- if (GetHero().HasArmor()) {
- equipmentMenu.Add(GetHero().Armor()->Name(), GetHero().Armor(), true, GetHero().Armor()->MenuIcon());
- } else {
- equipmentMenu.Add(parent->Res().noEquipmentText, 0, false);
- }
- if (GetHero().HasShield()) {
- equipmentMenu.Add(GetHero().Shield()->Name(), GetHero().Shield(), true, GetHero().Shield()->MenuIcon());
- } else {
- equipmentMenu.Add(parent->Res().noEquipmentText, 0, false);
- }
- if (GetHero().HasHelmet()) {
- equipmentMenu.Add(GetHero().Helmet()->Name(), GetHero().Helmet(), true, GetHero().Helmet()->MenuIcon());
- } else {
- equipmentMenu.Add(parent->Res().noEquipmentText, 0, false);
- }
- if (GetHero().HasRing()) {
- equipmentMenu.Add(GetHero().Ring()->Name(), GetHero().Ring(), true, GetHero().Ring()->MenuIcon());
- } else {
- equipmentMenu.Add(parent->Res().noEquipmentText, 0, false);
- }
- if (GetHero().HasJewel()) {
- equipmentMenu.Add(GetHero().Jewel()->Name(), GetHero().Jewel(), true, GetHero().Jewel()->MenuIcon());
- } else {
- equipmentMenu.Add(parent->Res().noEquipmentText, 0, false);
+ for (int i = 0; i < Hero::EQUIP_COUNT; ++i) {
+ if (GetHero().Equipped(Hero::EquipSlot(i))) {
+ const Item *item(GetHero().Equipment(Hero::EquipSlot(i)));
+ equipmentMenu.Add(item->Name(), item, true, item->MenuIcon());
+ } else {
+ equipmentMenu.Add(parent->Res().noEquipmentText, 0);
+ }