Make Production.SetRallyPoint and .SetPrimaryBuilding more robust

This commit is contained in:
Oliver Brakmann
2014-03-12 22:26:33 +01:00
parent 800665579b
commit 00bc3f71fe

View File

@@ -17,9 +17,15 @@ Production.PerFactoryQueueIsBusy = function(factory)
end end
Production.SetRallyPoint = function(factory, location) Production.SetRallyPoint = function(factory, location)
Actor.Trait(factory, "RallyPoint").rallyPoint = location.Location local srp = Actor.Trait(factory, "RallyPoint")
if srp ~= nil then
srp.rallyPoint = location.Location
end
end end
Production.SetPrimaryBuilding = function(factory) Production.SetPrimaryBuilding = function(factory)
Actor.Trait(factory, "PrimaryBuilding"):SetPrimaryProducer(factory, true) local pb = Actor.TraitOrDefault(factory, "PrimaryBuilding")
if pb ~= nil then
pb:SetPrimaryProducer(factory, true)
end
end end