warscript 0.0.1-dev.a4deb44 → 0.0.1-dev.a5a2f5e

package/engine/behaviour/unit.lua

@@ -1,24 +1,170 @@
 local ____lualib = require("lualib_bundle")
 local __TS__Class = ____lualib.__TS__Class
 local __TS__ClassExtends = ____lualib.__TS__ClassExtends
+local __TS__New = ____lualib.__TS__New
 local __TS__SetDescriptor = ____lualib.__TS__SetDescriptor
 local ____exports = {}
 local ____behavior = require("engine.behavior")
 local Behavior = ____behavior.Behavior
 local ____unit = require("engine.internal.unit")
 local Unit = ____unit.Unit
+require("engine.internal.unit+ability")
 require("engine.internal.unit-missile-launch")
+local ____linked_2Dset = require("utility.linked-set")
+local LinkedSet = ____linked_2Dset.LinkedSet
+local ____lua_2Dmaps = require("utility.lua-maps")
+local getOrPut = ____lua_2Dmaps.getOrPut
+local mutableLuaMap = ____lua_2Dmaps.mutableLuaMap
+local ____lua_2Dsets = require("utility.lua-sets")
+local mutableLuaSet = ____lua_2Dsets.mutableLuaSet
+local ____bonus = require("engine.internal.unit.bonus")
+local addOrUpdateOrRemoveUnitBonus = ____bonus.addOrUpdateOrRemoveUnitBonus
+local getUnitBonus = ____bonus.getUnitBonus
+local removeUnitBonus = ____bonus.removeUnitBonus
+local ____linked_2Dmap = require("utility.linked-map")
+local LinkedMap = ____linked_2Dmap.LinkedMap
+local safeCall = warpack.safeCall
+local createBehaviorFunctionsByUnitTypeId = {}
+local behaviorsByOwningPlayerEvent = {}
+local listenerByBehaviorByOwningPlayerEvent = {}
+local owningPlayerEventsByBehavior = {}
+local behaviorsByInRangeUnitEvent = {}
+local rangeByBehaviorByInRangeUnitEvent = {}
+local listenerByBehaviorByInRangeUnitEvent = {}
+local inRangeUnitEventsByBehavior = {}
 ____exports.UnitBehavior = __TS__Class()
 local UnitBehavior = ____exports.UnitBehavior
 UnitBehavior.name = "UnitBehavior"
 __TS__ClassExtends(UnitBehavior, Behavior)
-function UnitBehavior.prototype.____constructor(self, unit)
-    Behavior.prototype.____constructor(self, unit)
+function UnitBehavior.prototype.____constructor(self, unit, priority)
+    Behavior.prototype.____constructor(self, unit, priority)
+end
+function UnitBehavior.prototype.onDestroy(self)
+    local owningPlayerEvents = owningPlayerEventsByBehavior[self]
+    if owningPlayerEvents ~= nil then
+        for event in pairs(owningPlayerEvents) do
+            local ____opt_0 = behaviorsByOwningPlayerEvent[event]
+            if ____opt_0 ~= nil then
+                ____opt_0:remove(self)
+            end
+            local ____opt_2 = listenerByBehaviorByOwningPlayerEvent[event]
+            if ____opt_2 ~= nil then
+                ____opt_2[self] = nil
+            end
+        end
+        owningPlayerEventsByBehavior[self] = nil
+    end
+    local inRangeUnitEvents = inRangeUnitEventsByBehavior[self]
+    if inRangeUnitEvents ~= nil then
+        for event in pairs(inRangeUnitEvents) do
+            local ____opt_4 = behaviorsByInRangeUnitEvent[event]
+            if ____opt_4 ~= nil then
+                ____opt_4:remove(self)
+            end
+            local ____opt_6 = rangeByBehaviorByInRangeUnitEvent[event]
+            if ____opt_6 ~= nil then
+                ____opt_6[self] = nil
+            end
+            local ____opt_8 = listenerByBehaviorByInRangeUnitEvent[event]
+            if ____opt_8 ~= nil then
+                ____opt_8[self] = nil
+            end
+        end
+        inRangeUnitEventsByBehavior[self] = nil
+    end
+    if self._bonusIdByBonusType ~= nil then
+        for bonusType, bonusId in pairs(self._bonusIdByBonusType) do
+            removeUnitBonus(self.object, bonusType, bonusId)
+        end
+    end
+    return Behavior.prototype.onDestroy(self)
+end
+function UnitBehavior.prototype.getUnitBonus(self, bonusType)
+    local ____opt_10 = self._bonusIdByBonusType
+    local bonusId = ____opt_10 and ____opt_10:get(bonusType)
+    return bonusId == nil and 0 or getUnitBonus(self.object, bonusType, bonusId)
+end
+function UnitBehavior.prototype.addOrUpdateOrRemoveUnitBonus(self, bonusType, value)
+    local bonusIdByBonusType = self._bonusIdByBonusType
+    if bonusIdByBonusType == nil then
+        bonusIdByBonusType = __TS__New(LinkedMap)
+        self._bonusIdByBonusType = bonusIdByBonusType
+    end
+    bonusIdByBonusType:put(
+        bonusType,
+        addOrUpdateOrRemoveUnitBonus(
+            self.object,
+            bonusType,
+            bonusIdByBonusType:get(bonusType),
+            value
+        )
+    )
+end
+function UnitBehavior.prototype.registerOwningPlayerEvent(self, event, extractPlayer, listener)
+    local listenerByBehavior = getOrPut(listenerByBehaviorByOwningPlayerEvent, event, mutableLuaMap)
+    listenerByBehavior[self] = listener
+    getOrPut(inRangeUnitEventsByBehavior, self, mutableLuaSet)[event] = true
+    local behaviors = behaviorsByOwningPlayerEvent[event]
+    if behaviors == nil then
+        event:addListener(function(...)
+            local behaviors = behaviorsByOwningPlayerEvent[event]
+            if behaviors ~= nil then
+                local player = extractPlayer(...)
+                if player ~= nil then
+                    for behavior in pairs(behaviors) do
+                        if behavior.unit.owner == player then
+                            safeCall(behavior[listenerByBehavior[behavior]], behavior, ...)
+                        end
+                    end
+                end
+            end
+        end)
+        behaviors = __TS__New(LinkedSet)
+        behaviorsByOwningPlayerEvent[event] = behaviors
+    end
+    behaviors:add(self)
+end
+function UnitBehavior.prototype.registerInRangeUnitEvent(self, event, extractUnit, range, listener)
+    local rangeByBehavior = getOrPut(rangeByBehaviorByInRangeUnitEvent, event, mutableLuaMap)
+    rangeByBehavior[self] = range
+    local listenerByBehavior = getOrPut(listenerByBehaviorByInRangeUnitEvent, event, mutableLuaMap)
+    listenerByBehavior[self] = listener
+    getOrPut(inRangeUnitEventsByBehavior, self, mutableLuaSet)[event] = true
+    local behaviors = behaviorsByInRangeUnitEvent[event]
+    if behaviors == nil then
+        event:addListener(function(...)
+            local behaviors = behaviorsByInRangeUnitEvent[event]
+            if behaviors ~= nil then
+                local unit = extractUnit(...)
+                if unit ~= nil then
+                    for behavior in pairs(behaviors) do
+                        local range = rangeByBehavior[behavior]
+                        if range ~= nil and unit:getCollisionDistanceTo(behavior.unit) <= range then
+                            safeCall(behavior[listenerByBehavior[behavior]], behavior, ...)
+                        end
+                    end
+                end
+            end
+        end)
+        behaviors = __TS__New(LinkedSet)
+        behaviorsByInRangeUnitEvent[event] = behaviors
+    end
+    behaviors:add(self)
+end
+function UnitBehavior.prototype.onImmediateOrder(self, orderId)
+end
+function UnitBehavior.prototype.onTargetOrder(self, orderId, target)
+end
+function UnitBehavior.prototype.onPointOrder(self, orderId, x, y)
 end
 function UnitBehavior.prototype.onAutoAttackStart(self, target)
 end
 function UnitBehavior.prototype.onAutoAttackFinish(self, target)
 end
+function UnitBehavior.prototype.onTargetingAutoAttackStart(self, source)
+end
+function UnitBehavior.prototype.onTargetingAutoAttackFinish(self, source)
+end
 function UnitBehavior.prototype.onDamageDealing(self, target, event)
 end
 function UnitBehavior.prototype.onDamageDealt(self, target, event)
@@ -31,10 +177,65 @@ function UnitBehavior.prototype.onAbilityGained(self, ability)
 end
 function UnitBehavior.prototype.onAbilityLost(self, ability)
 end
+function UnitBehavior.prototype.onAbilityChannelingStart(self, ability)
+end
+function UnitBehavior.prototype.onAbilityImpact(self, ability)
+end
+function UnitBehavior.prototype.onAbilityWidgetTargetImpact(self, ability, target)
+end
+function UnitBehavior.prototype.onAbilityUnitTargetImpact(self, ability, target)
+end
+function UnitBehavior.prototype.onAbilityItemTargetImpact(self, ability, target)
+end
+function UnitBehavior.prototype.onAbilityDestructibleTargetImpact(self, ability, target)
+end
+function UnitBehavior.prototype.onAbilityPointTargetImpact(self, ability, x, y)
+end
+function UnitBehavior.prototype.onAbilityNoTargetImpact(self, ability)
+end
+function UnitBehavior.prototype.onAbilityChannelingFinish(self, ability)
+end
+function UnitBehavior.prototype.onAbilityStop(self, ability)
+end
+function UnitBehavior.prototype.onTargetingAbilityChannelingStart(self, ability, source)
+end
+function UnitBehavior.prototype.onTargetingAbilityImpact(self, ability, source)
+end
+function UnitBehavior.prototype.onBuffGained(self, buff)
+end
+function UnitBehavior.prototype.onBuffLost(self, buff)
+end
+function UnitBehavior.prototype.onItemDropped(self, item)
+end
+function UnitBehavior.prototype.onItemPickedUp(self, item)
+end
+function UnitBehavior.prototype.onItemUsed(self, item)
+end
+function UnitBehavior.prototype.onItemStacked(self, item)
+end
+function UnitBehavior.prototype.onItemChargesChanged(self, item)
+end
 function UnitBehavior.prototype.onKill(self, target)
 end
 function UnitBehavior.prototype.onDeath(self, source)
 end
+function UnitBehavior.prototype.onOwnerChange(self, previousOwner)
+end
+function UnitBehavior.bindUnitType(self, unitTypeId, ...)
+    local args = {...}
+    local createBehaviorFunctions = createBehaviorFunctionsByUnitTypeId[unitTypeId]
+    if createBehaviorFunctions == nil then
+        createBehaviorFunctions = {}
+        createBehaviorFunctionsByUnitTypeId[unitTypeId] = createBehaviorFunctions
+    end
+    createBehaviorFunctions[#createBehaviorFunctions + 1] = function(unit)
+        return __TS__New(
+            self,
+            unit,
+            table.unpack(args)
+        )
+    end
+end
 __TS__SetDescriptor(
     UnitBehavior.prototype,
     "unit",
@@ -44,11 +245,28 @@ __TS__SetDescriptor(
     true
 );
 (function(self)
+    Unit.onImmediateOrder:addListener(function(source, orderId)
+        ____exports.UnitBehavior:forAll(source, "onImmediateOrder", orderId)
+    end)
+    Unit.onTargetOrder:addListener(function(source, orderId, target)
+        ____exports.UnitBehavior:forAll(source, "onTargetOrder", orderId, target)
+    end)
+    Unit.onPointOrder:addListener(function(source, orderId, x, y)
+        ____exports.UnitBehavior:forAll(
+            source,
+            "onPointOrder",
+            orderId,
+            x,
+            y
+        )
+    end)
     Unit.autoAttackStartEvent:addListener(function(source, target)
         ____exports.UnitBehavior:forAll(source, "onAutoAttackStart", target)
+        ____exports.UnitBehavior:forAll(target, "onTargetingAutoAttackStart", source)
     end)
     Unit.autoAttackFinishEvent:addListener(function(source, target)
         ____exports.UnitBehavior:forAll(source, "onAutoAttackFinish", target)
+        ____exports.UnitBehavior:forAll(target, "onTargetingAutoAttackFinish", source)
     end)
     Unit.onDamaging:addListener(function(source, target, event)
         if source ~= nil then
@@ -62,13 +280,85 @@ __TS__SetDescriptor(
         end
         ____exports.UnitBehavior:forAll(target, "onDamageReceived", source, event)
     end)
+    Unit.abilityGainedEvent:addListener(function(source, ability)
+        ____exports.UnitBehavior:forAll(source, "onAbilityGained", ability)
+    end)
+    Unit.abilityLostEvent:addListener(function(source, ability)
+        ____exports.UnitBehavior:forAll(source, "onAbilityLost", ability)
+    end)
+    Unit.abilityChannelingStartEvent:addListener(function(source, ability)
+        ____exports.UnitBehavior:forAll(source, "onAbilityChannelingStart", ability)
+    end)
+    Unit.abilityUnitTargetChannelingStartEvent:addListener(function(source, ability, target)
+        ____exports.UnitBehavior:forAll(target, "onTargetingAbilityChannelingStart", ability, source)
+    end)
+    Unit.abilityImpactEvent:addListener(function(source, ability)
+        ____exports.UnitBehavior:forAll(source, "onAbilityImpact", ability)
+    end)
+    Unit.abilityWidgetTargetImpactEvent:addListener(function(source, ability, target)
+        ____exports.UnitBehavior:forAll(source, "onAbilityWidgetTargetImpact", ability, target)
+    end)
+    Unit.abilityUnitTargetImpactEvent:addListener(function(source, ability, target)
+        ____exports.UnitBehavior:forAll(source, "onAbilityUnitTargetImpact", ability, target)
+        ____exports.UnitBehavior:forAll(target, "onTargetingAbilityImpact", ability, source)
+    end)
+    Unit.abilityItemTargetImpactEvent:addListener(function(source, ability, target)
+        ____exports.UnitBehavior:forAll(source, "onAbilityItemTargetImpact", ability, target)
+    end)
+    Unit.abilityDestructibleTargetImpactEvent:addListener(function(source, ability, target)
+        ____exports.UnitBehavior:forAll(source, "onAbilityDestructibleTargetImpact", ability, target)
+    end)
+    Unit.abilityPointTargetImpactEvent:addListener(function(source, ability, x, y)
+        ____exports.UnitBehavior:forAll(
+            source,
+            "onAbilityPointTargetImpact",
+            ability,
+            x,
+            y
+        )
+    end)
+    Unit.abilityNoTargetImpactEvent:addListener(function(source, ability)
+        ____exports.UnitBehavior:forAll(source, "onAbilityNoTargetImpact", ability)
+    end)
+    Unit.abilityChannelingFinishEvent:addListener(function(source, ability)
+        ____exports.UnitBehavior:forAll(source, "onAbilityChannelingFinish", ability)
+    end)
+    Unit.abilityStopEvent:addListener(function(source, ability)
+        ____exports.UnitBehavior:forAll(source, "onAbilityStop", ability)
+    end)
     Unit.deathEvent:addListener(function(target, source)
         if source ~= nil then
             ____exports.UnitBehavior:forAll(source, "onKill", target)
         end
         ____exports.UnitBehavior:forAll(target, "onDeath", source)
     end)
+    Unit.itemDroppedEvent:addListener(function(unit, item)
+        ____exports.UnitBehavior:forAll(unit, "onItemDropped", item)
+    end)
+    Unit.itemPickedUpEvent:addListener(function(unit, item)
+        ____exports.UnitBehavior:forAll(unit, "onItemPickedUp", item)
+    end)
+    Unit.itemUsedEvent:addListener(function(unit, item)
+        ____exports.UnitBehavior:forAll(unit, "onItemUsed", item)
+    end)
+    Unit.itemStackedEvent:addListener(function(unit, item)
+        ____exports.UnitBehavior:forAll(unit, "onItemStacked", item)
+    end)
+    Unit.itemChargesChangedEvent:addListener(function(unit, item)
+        ____exports.UnitBehavior:forAll(unit, "onItemChargesChanged", item)
+    end)
+    Unit.onOwnerChange:addListener(function(unit, previousOwner)
+        ____exports.UnitBehavior:forAll(unit, "onOwnerChange", previousOwner)
+    end)
 end)(UnitBehavior)
+Unit.onCreate:addListener(function(unit)
+    local createBehaviorFunctions = createBehaviorFunctionsByUnitTypeId[unit.typeId]
+    if createBehaviorFunctions ~= nil then
+        for ____, createBehavior in ipairs(createBehaviorFunctions) do
+            createBehavior(unit)
+        end
+    end
+end)
 Unit.destroyEvent:addListener(function(unit)
     ____exports.UnitBehavior:forAll(unit, "destroy")
 end)