Andrii Danylchenko преди 6 години
родител
ревизия
0bd5782470
променени са 1 файла, в които са добавени 7 реда и са изтрити 0 реда
  1. 7 0
      AI/VCAI/Goals/GatherTroops.cpp

+ 7 - 0
AI/VCAI/Goals/GatherTroops.cpp

@@ -115,6 +115,7 @@ TGoalVec GatherTroops::getAllPossibleSubgoals()
 			}*/
 		}
 	}
+
 	for(auto obj : ai->visitableObjs)
 	{
 		auto d = dynamic_cast<const CGDwelling *>(obj);
@@ -135,6 +136,12 @@ TGoalVec GatherTroops::getAllPossibleSubgoals()
 		}
 	}
 
+	CreatureID creID = CreatureID(objid);
+
+	vstd::erase_if(solutions, [&](TSubgoal goal)->bool{
+		return goal->hero && !goal->hero->getSlotFor(creID).validSlot() && !goal->hero->getFreeSlot().validSlot();
+	});
+
 	return solutions;
 	//TODO: exchange troops between heroes
 }