Преглед изворни кода

Fix creating buttons for unavailable upgrades

MichalZr6 пре 10 месеци
родитељ
комит
358189d9c3
1 измењених фајлова са 6 додато и 2 уклоњено
  1. 6 2
      client/windows/CCreatureWindow.cpp

+ 6 - 2
client/windows/CCreatureWindow.cpp

@@ -769,9 +769,13 @@ CStackWindow::CStackWindow(const CStackInstance * stack, std::function<void()> d
 	info->creature = stack->getCreature();
 	info->creatureCount = stack->count;
 
-	info->upgradeInfo = std::make_optional(UnitView::StackUpgradeInfo(upgradeInfo));
+	if(upgradeInfo.canUpgrade())
+	{
+		info->upgradeInfo = std::make_optional(UnitView::StackUpgradeInfo(upgradeInfo));
+		info->upgradeInfo->callback = callback;
+	}
+	
 	info->dismissInfo = std::make_optional(UnitView::StackDismissInfo());
-	info->upgradeInfo->callback = callback;
 	info->dismissInfo->callback = dismiss;
 	info->owner = dynamic_cast<const CGHeroInstance *> (stack->armyObj);
 	init();