Fix air wrecks

FAForever · Dec 13, 2014 · 4210bd3 · 4210bd3 · aeoncleanse · Dec 13, 2014
1 parent 44d76ae
commit 4210bd3
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/lua/sim/Unit.lua b/lua/sim/Unit.lua
@@ -1424,11 +1424,15 @@ Unit = Class(moho.unit_methods) {
             mass = mass * 0.5
             energy = energy * 0.5
         end
+
+        if self:GetCurrentLayer() == 'Air' then
+            pos[2] = GetSurfaceHeight(pos[1], pos[3])
+        elseif self:GetCurrentLayer() == 'Seabed' or self:GetCurrentLayer() == 'Land' then
+            pos[2] = GetTerrainHeight(pos[1], pos[3]) + GetTerrainTypeOffset(pos[1], pos[3])
+        else
+            pos[2] = GetSurfaceHeight(pos[1], pos[3]) + GetTerrainTypeOffset(pos[1], pos[3])
+        end
 
-        -- Stop the Y position of the wreck being on the seabed for Naval Factories
-        if not (EntityCategoryContains(categories.FACTORY, self) and EntityCategoryContains(categories.STRUCTURE, self) and EntityCategoryContains(categories.NAVAL, self)) then
-            pos[2] = GetTerrainHeight(pos[1], pos[3]) + GetTerrainTypeOffset(pos[1], pos[3])
-        end
 
         local prop = CreateProp( pos, wreck )