un

jonylu7 · Jun 16, 2024 · 8500298 · 8500298
1 parent 2cba26b
commit 8500298
Showing 1 changed file with 19 additions and 15 deletions.
diff --git a/src/Mechanics/UnitManager.cpp b/src/Mechanics/UnitManager.cpp
@@ -4,7 +4,9 @@
 #include "Mechanics/UnitManager.hpp"
 
 void UnitManager::spawnToWayPoint(UnitType unit, HouseType house) {
-
+    if (!m_StructureManager->getStructureArray()->ifBarrackBuilt()) {
+        return;
+    }
     switch (unit) {
     case UnitType::INFANTRY: {
         auto avatar = std::make_shared<Infantry>(house);
@@ -41,42 +43,44 @@ void UnitManager::spawn(UnitType unit, HouseType house, glm::vec2 cellPos) {
         auto structure = std::make_shared<Barracks>(house);
         auto globalPos = MapUtil::CellCoordToGlobal(cellPos);
         structure->Start(globalPos);
-        m_StructureManager->getStructureArray()->buildNewStructure(
-            structure, true);
-        structure->setWaypointLocationByCellCoord({cellPos.x+2,cellPos.y+2});
+        m_StructureManager->getStructureArray()->buildNewStructure(structure,
+                                                                   true);
+        structure->setWaypointLocationByCellCoord(
+            {cellPos.x + 2, cellPos.y + 2});
         break;
     }
     case UnitType::ORE_REF: {
         auto structure = std::make_shared<OreRefinery>(house);
         auto globalPos = MapUtil::CellCoordToGlobal(cellPos);
         structure->Start(globalPos);
-        m_StructureManager->getStructureArray()->buildNewStructure(
-            structure, true);
+        m_StructureManager->getStructureArray()->buildNewStructure(structure,
+                                                                   true);
         break;
     }
     case UnitType::POWER_PLANT: {
         auto structure = std::make_shared<PowerPlants>(house);
         auto globalPos = MapUtil::CellCoordToGlobal(cellPos);
         structure->Start(globalPos);
-        m_StructureManager->getStructureArray()->buildNewStructure(
-            structure, true);
+        m_StructureManager->getStructureArray()->buildNewStructure(structure,
+                                                                   true);
         break;
     }
     case UnitType::WAR_FACT: {
         auto structure = std::make_shared<WarFactory>(house);
         auto globalPos = MapUtil::CellCoordToGlobal(cellPos);
         structure->Start(globalPos);
-        structure->setWaypointLocationByCellCoord({cellPos.x+2,cellPos.y-2});
-        m_StructureManager->getStructureArray()->buildNewStructure(
-            structure, true);
+        structure->setWaypointLocationByCellCoord(
+            {cellPos.x + 2, cellPos.y - 2});
+        m_StructureManager->getStructureArray()->buildNewStructure(structure,
+                                                                   true);
         break;
     }
     case UnitType::ADV_POWER_PLANT: {
         auto structure = std::make_shared<ADVPowerPlants>(house);
         auto globalPos = MapUtil::CellCoordToGlobal(cellPos);
         structure->Start(globalPos);
-        m_StructureManager->getStructureArray()->buildNewStructure(
-            structure, true);
+        m_StructureManager->getStructureArray()->buildNewStructure(structure,
+                                                                   true);
         break;
     }
     case UnitType::INFANTRY: {
@@ -94,8 +98,8 @@ void UnitManager::spawn(UnitType unit, HouseType house, glm::vec2 cellPos) {
         break;
     }
     }
-    if(unit!=UnitType::NONE){
-        addUnitConstructCount(unit,1);
+    if (unit != UnitType::NONE) {
+        addUnitConstructCount(unit, 1);
     }
 }