Added building:IsUnderConstruction() call

MNoya · May 21, 2016 · 4d1214a · 4d1214a
1 parent 05bb423
commit 4d1214a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/scripts/vscripts/libraries/buildinghelper.lua b/scripts/vscripts/libraries/buildinghelper.lua
@@ -945,6 +945,7 @@ function BuildingHelper:StartBuilding(builder)
     building.construction_size = construction_size
     building.buildingTable = buildingTable
     building.state = "building"
+    function building:IsUnderConstruction() return true end
 
     -- Adjust the Model Orientation
     local yaw = buildingTable:GetVal("ModelRotation", "float")
@@ -2016,6 +2017,7 @@ function BuildingHelper:AddBuildingToPlayerTable(playerID, building)
     local buildingName = building:GetUnitName()
     table.insert(BuildingHelper:GetBuildings(playerID), building)
     BuildingHelper:SetBuildingCount(playerID, buildingName, BuildingHelper:GetBuildingCount(playerID, buildingName)+1)
+    function building:IsUnderConstruction() return false end
 end
 
 -- Returns "ConstructionSize" value of a unit handle or unit name