From 195a251ff3b71c2f6bbd6ee3a744d460d8c1e458 Mon Sep 17 00:00:00 2001
From: yairm210 <yairm210@hotmail.com>
Date: Thu, 27 Jun 2024 15:16:58 +0300
Subject: [PATCH] Resolved #11865 - unified unit availability checks to one
 function

---
 .../com/unciv/models/ruleset/unit/BaseUnit.kt | 68 +++++++++----------
 1 file changed, 33 insertions(+), 35 deletions(-)

diff --git a/core/src/com/unciv/models/ruleset/unit/BaseUnit.kt b/core/src/com/unciv/models/ruleset/unit/BaseUnit.kt
index ef49fe0b43..b5a972fbbb 100644
--- a/core/src/com/unciv/models/ruleset/unit/BaseUnit.kt
+++ b/core/src/com/unciv/models/ruleset/unit/BaseUnit.kt
@@ -171,42 +171,34 @@ class BaseUnit : RulesetObject(), INonPerpetualConstruction {
         return false
     }
 
-    override fun getRejectionReasons(cityConstructions: CityConstructions): Sequence<RejectionReason> = sequence {
-        if (isWaterUnit() && !cityConstructions.city.isCoastal())
-            yield(RejectionReasonType.WaterUnitsInCoastalCities.toInstance())
-
-        val civInfo = cityConstructions.city.civ
-
-        for (unique in getMatchingUniques(UniqueType.OnlyAvailable, StateForConditionals.IgnoreConditionals))
-            yieldAll(notMetRejections(unique, cityConstructions))
-
-        for (unique in getMatchingUniques(UniqueType.CanOnlyBeBuiltWhen, StateForConditionals.IgnoreConditionals))
-            yieldAll(notMetRejections(unique, cityConstructions, true))
-
-        for (unique in getMatchingUniques(UniqueType.Unavailable, StateForConditionals(civInfo, cityConstructions.city)))
-            yield(RejectionReasonType.ShouldNotBeDisplayed.toInstance())
-
-        for (unique in getMatchingUniques(UniqueType.RequiresPopulation))
-            if (unique.params[0].toInt() > cityConstructions.city.population.population)
-                yield(RejectionReasonType.PopulationRequirement.toInstance(unique.getDisplayText()))
-
-        yieldAll(getRejectionReasons(civInfo, cityConstructions.city))
-
-        // Expensive, since adding and removing the fake unit causes side-effects
-        if (isAirUnit()) {
-            // Not actually added to civ so doesn't require destroy
-            val fakeUnit = getMapUnit(cityConstructions.city.civ, Constants.NO_ID)
-            val canUnitEnterTile = fakeUnit.movement.canMoveTo(cityConstructions.city.getCenterTile())
-            if (!canUnitEnterTile)
-                yield(RejectionReasonType.NoPlaceToPutUnit.toInstance())
-        }
-    }
+    override fun getRejectionReasons(cityConstructions: CityConstructions): Sequence<RejectionReason> =
+        getRejectionReasons(cityConstructions.city.civ, cityConstructions.city)
 
     fun getRejectionReasons(
         civ: Civilization,
         city: City? = null,
         additionalResources: Counter<String> = Counter.ZERO
     ): Sequence<RejectionReason> = sequence {
+
+        val stateForConditionals = StateForConditionals(civ, city)
+
+        if (city != null && isWaterUnit() && !city.isCoastal())
+            yield(RejectionReasonType.WaterUnitsInCoastalCities.toInstance())
+
+        for (unique in getMatchingUniques(UniqueType.OnlyAvailable, StateForConditionals.IgnoreConditionals))
+            yieldAll(notMetRejections(unique, civ, city))
+
+        for (unique in getMatchingUniques(UniqueType.CanOnlyBeBuiltWhen, StateForConditionals.IgnoreConditionals))
+            yieldAll(notMetRejections(unique, civ, city, true))
+
+        for (unique in getMatchingUniques(UniqueType.Unavailable, stateForConditionals))
+            yield(RejectionReasonType.ShouldNotBeDisplayed.toInstance())
+
+        if (city != null)
+            for (unique in getMatchingUniques(UniqueType.RequiresPopulation))
+                if (unique.params[0].toInt() > city.population.population)
+                    yield(RejectionReasonType.PopulationRequirement.toInstance(unique.getDisplayText()))
+
         for (requiredTech: String in requiredTechs())
             if (!civ.tech.isResearched(requiredTech))
                 yield(RejectionReasonType.RequiresTech.toInstance("$requiredTech not researched"))
@@ -222,8 +214,6 @@ class BaseUnit : RulesetObject(), INonPerpetualConstruction {
         if (isHiddenBySettings(civ.gameInfo))
             yield(RejectionReasonType.DisabledBySetting.toInstance())
 
-        val stateForConditionals = StateForConditionals(civ, city)
-
         if (hasUnique(UniqueType.Unbuildable, stateForConditionals))
             yield(RejectionReasonType.Unbuildable.toInstance())
 
@@ -255,6 +245,15 @@ class BaseUnit : RulesetObject(), INonPerpetualConstruction {
                     yield(RejectionReasonType.CannotBeBuiltUnhappiness.toInstance(unique.getDisplayText()))
                 else yield(RejectionReasonType.CannotBeBuilt.toInstance())
             }
+
+        // Expensive, since adding and removing the fake unit causes side-effects
+        if (city != null && isAirUnit()) {
+            // Not actually added to civ so doesn't require destroy
+            val fakeUnit = getMapUnit(civ, Constants.NO_ID)
+            val canUnitEnterTile = fakeUnit.movement.canMoveTo(city.getCenterTile())
+            if (!canUnitEnterTile)
+                yield(RejectionReasonType.NoPlaceToPutUnit.toInstance())
+        }
     }
 
     /**
@@ -262,11 +261,10 @@ class BaseUnit : RulesetObject(), INonPerpetualConstruction {
      * Also custom handles [UniqueType.ConditionalBuildingBuiltAmount], and
      * [UniqueType.ConditionalBuildingBuiltAll]
      */
-    private fun notMetRejections(unique: Unique, cityConstructions: CityConstructions, built: Boolean=false): Sequence<RejectionReason> = sequence {
-        val civ = cityConstructions.city.civ
+    private fun notMetRejections(unique: Unique, civ: Civilization, city: City?, built: Boolean=false): Sequence<RejectionReason> = sequence {
         for (conditional in unique.conditionals) {
             // We yield a rejection only when conditionals are NOT met
-            if (Conditionals.conditionalApplies(unique, conditional, StateForConditionals(civ, cityConstructions.city)))
+            if (Conditionals.conditionalApplies(unique, conditional, StateForConditionals(civ, city)))
                 continue
             when (conditional.type) {
                 UniqueType.ConditionalBuildingBuiltAmount -> {