warscript 0.0.1-dev.5a50e5d → 0.0.1-dev.5bc80ec

package/engine/object-data/entry/ability-type.lua

@@ -3,6 +3,7 @@ local __TS__Class = ____lualib.__TS__Class
 local __TS__ClassExtends = ____lualib.__TS__ClassExtends
 local __TS__SetDescriptor = ____lualib.__TS__SetDescriptor
 local __TS__InstanceOf = ____lualib.__TS__InstanceOf
+local __TS__ArraySome = ____lualib.__TS__ArraySome
 local ____exports = {}
 local ____unit = require("engine.internal.unit")
 local Unit = ____unit.Unit
@@ -18,9 +19,11 @@ local array = ____arrays.array
 local map = ____arrays.map
 local mapIndexed = ____arrays.mapIndexed
 local ____attachment_2Dpreset = require("engine.object-data.auxiliary.attachment-preset")
+local attachmentPointToNodeFQN = ____attachment_2Dpreset.attachmentPointToNodeFQN
 local extractAttachmentPresetInputModelPath = ____attachment_2Dpreset.extractAttachmentPresetInputModelPath
 local extractAttachmentPresetInputNodeFQN = ____attachment_2Dpreset.extractAttachmentPresetInputNodeFQN
-local toAttachmentPreset = ____attachment_2Dpreset.toAttachmentPreset
+local splitAttachmentNodeFQN = ____attachment_2Dpreset.splitAttachmentNodeFQN
+local toEffectPreset = ____attachment_2Dpreset.toEffectPreset
 local ____combat_2Dclassification = require("engine.object-data.auxiliary.combat-classification")
 local combatClassificationsToStringArray = ____combat_2Dclassification.combatClassificationsToStringArray
 local stringArrayToCombatClassifications = ____combat_2Dclassification.stringArrayToCombatClassifications
@@ -35,13 +38,22 @@ local abilityTypeIdGenerator = ____object_2Ddata_2Dentry_2Did_2Dgenerator.abilit
 local ____upgrade = require("engine.object-data.entry.upgrade")
 local Upgrade = ____upgrade.Upgrade
 local ____sound = require("core.types.sound")
+local isSoundLabelCustom = ____sound.isSoundLabelCustom
 local Sound3D = ____sound.Sound3D
-local SoundPreset = ____sound.SoundPreset
+local SoundSettings = ____sound.SoundSettings
+local ____lua_2Dsets = require("utility.lua-sets")
+local luaSetOf = ____lua_2Dsets.luaSetOf
+local ____local_2Dclient = require("engine.local-client")
+local LocalClient = ____local_2Dclient.LocalClient
 local castAnimationFQNByAbilityTypeId = {}
 local isButtonVisibleFalseAbilityTypes = {}
 local casterCastingEffectPresetsByAbilityTypeId = {}
 local casterChannelingEffectPresetsByAbilityTypeId = {}
 local targetCastingEffectPresetsByAbilityTypeId = {}
+local targetChannelingEffectPresetsByAbilityTypeId = {}
+local targetChannelingEffectPresetsSDByAbilityTypeId = {}
+local targetChannelingEffectPresetsHDByAbilityTypeId = {}
+local targetChannelingEffectPresetsDEByAbilityTypeId = {}
 local targetEffectSoundPresetByAbilityTypeId = {}
 ____exports.AbilityType = __TS__Class()
 local AbilityType = ____exports.AbilityType
@@ -118,7 +130,7 @@ __TS__SetDescriptor(
             return casterCastingEffectPresetsByAbilityTypeId[self.id] or ({})
         end,
         set = function(self, casterCastingEffectPresets)
-            casterCastingEffectPresetsByAbilityTypeId[self.id] = map(casterCastingEffectPresets, toAttachmentPreset)
+            casterCastingEffectPresetsByAbilityTypeId[self.id] = map(casterCastingEffectPresets, toEffectPreset)
         end
     },
     true
@@ -131,20 +143,20 @@ __TS__SetDescriptor(
             return casterChannelingEffectPresetsByAbilityTypeId[self.id] or ({})
         end,
         set = function(self, casterChannelingEffectPresets)
-            casterChannelingEffectPresetsByAbilityTypeId[self.id] = map(casterChannelingEffectPresets, toAttachmentPreset)
+            casterChannelingEffectPresetsByAbilityTypeId[self.id] = map(casterChannelingEffectPresets, toEffectPreset)
         end
     },
     true
 )
 __TS__SetDescriptor(
     AbilityType.prototype,
-    "casterAttachmentPresets",
+    "casterEffectPresets",
     {
         get = function(self)
             return self:getAttachmentPresetListField("acat", {"acap", "aca1"})
         end,
-        set = function(self, casterAttachmentPresets)
-            self:setAttachmentPresetListField("acat", {"acap", "aca1"}, "acac", casterAttachmentPresets)
+        set = function(self, casterEffectPresets)
+            self:setAttachmentPresetListField("acat", {"acap", "aca1"}, "acac", casterEffectPresets)
         end
     },
     true
@@ -162,6 +174,45 @@ __TS__SetDescriptor(
     },
     true
 )
+__TS__SetDescriptor(
+    AbilityType.prototype,
+    "effectModelPathsSD",
+    {
+        get = function(self)
+            return self:getStringsField("aeat:sd")
+        end,
+        set = function(self, effectModelPathsSD)
+            self:setStringsField("aeat:sd", effectModelPathsSD)
+        end
+    },
+    true
+)
+__TS__SetDescriptor(
+    AbilityType.prototype,
+    "effectModelPathsHD",
+    {
+        get = function(self)
+            return self:getStringsField("aeat:hd")
+        end,
+        set = function(self, effectModelPathsHD)
+            self:setStringsField("aeat:hd", effectModelPathsHD)
+        end
+    },
+    true
+)
+__TS__SetDescriptor(
+    AbilityType.prototype,
+    "effectModelPathsDE",
+    {
+        get = function(self)
+            return self:getStringsField("aeat:de")
+        end,
+        set = function(self, effectModelPathsDE)
+            self:setStringsField("aeat:de", effectModelPathsDE)
+        end
+    },
+    true
+)
 __TS__SetDescriptor(
     AbilityType.prototype,
     "iconPath",
@@ -201,6 +252,19 @@ __TS__SetDescriptor(
     },
     true
 )
+__TS__SetDescriptor(
+    AbilityType.prototype,
+    "iconPathDE",
+    {
+        get = function(self)
+            return self:getStringField("aart:de")
+        end,
+        set = function(self, iconPathDE)
+            self:setStringField("aart:de", iconPathDE)
+        end
+    },
+    true
+)
 __TS__SetDescriptor(
     AbilityType.prototype,
     "isButtonVisible",
@@ -323,6 +387,19 @@ __TS__SetDescriptor(
     },
     true
 )
+__TS__SetDescriptor(
+    AbilityType.prototype,
+    "missileModelPathDE",
+    {
+        get = function(self)
+            return self:getStringField("amat:de")
+        end,
+        set = function(self, missileModelPathDE)
+            self:setStringField("amat:de", missileModelPathDE)
+        end
+    },
+    true
+)
 __TS__SetDescriptor(
     AbilityType.prototype,
     "missileMovementArc",
@@ -351,13 +428,30 @@ __TS__SetDescriptor(
 )
 __TS__SetDescriptor(
     AbilityType.prototype,
-    "specialAttachmentPreset",
+    "specialEffectModelPaths",
     {
         get = function(self)
-            return self:getAttachmentPresetField("asat", "aspt")
+            return self:getStringsField("asat")
         end,
-        set = function(self, specialAttachmentPreset)
-            self:setAttachmentPresetField("asat", "aspt", specialAttachmentPreset)
+        set = function(self, specialEffectModelPaths)
+            self:setStringsField("asat", specialEffectModelPaths)
+        end
+    },
+    true
+)
+__TS__SetDescriptor(
+    AbilityType.prototype,
+    "specialEffectAttachmentPoint",
+    {
+        get = function(self)
+            local nodeName, nodeQualifiers = splitAttachmentNodeFQN(self:getStringField("aspt"))
+            return {nodeName = nodeName, nodeQualifiers = nodeQualifiers}
+        end,
+        set = function(self, specialEffectAttachmentPoint)
+            self:setStringField(
+                "aspt",
+                attachmentPointToNodeFQN(specialEffectAttachmentPoint)
+            )
         end
     },
     true
@@ -370,7 +464,59 @@ __TS__SetDescriptor(
             return targetCastingEffectPresetsByAbilityTypeId[self.id] or ({})
         end,
         set = function(self, targetCastingEffectPresets)
-            targetCastingEffectPresetsByAbilityTypeId[self.id] = map(targetCastingEffectPresets, toAttachmentPreset)
+            targetCastingEffectPresetsByAbilityTypeId[self.id] = map(targetCastingEffectPresets, toEffectPreset)
+        end
+    },
+    true
+)
+__TS__SetDescriptor(
+    AbilityType.prototype,
+    "targetChannelingEffectPresets",
+    {
+        get = function(self)
+            return targetChannelingEffectPresetsByAbilityTypeId[self.id] or ({})
+        end,
+        set = function(self, targetChannelingEffectPresets)
+            targetChannelingEffectPresetsByAbilityTypeId[self.id] = map(targetChannelingEffectPresets, toEffectPreset)
+        end
+    },
+    true
+)
+__TS__SetDescriptor(
+    AbilityType.prototype,
+    "targetChannelingEffectPresetsSD",
+    {
+        get = function(self)
+            return targetChannelingEffectPresetsSDByAbilityTypeId[self.id] or self.targetChannelingEffectPresets
+        end,
+        set = function(self, targetChannelingEffectPresetsSD)
+            targetChannelingEffectPresetsSDByAbilityTypeId[self.id] = map(targetChannelingEffectPresetsSD, toEffectPreset)
+        end
+    },
+    true
+)
+__TS__SetDescriptor(
+    AbilityType.prototype,
+    "targetChannelingEffectPresetsHD",
+    {
+        get = function(self)
+            return targetChannelingEffectPresetsHDByAbilityTypeId[self.id] or self.targetChannelingEffectPresets
+        end,
+        set = function(self, targetChannelingEffectPresetsHD)
+            targetChannelingEffectPresetsHDByAbilityTypeId[self.id] = map(targetChannelingEffectPresetsHD, toEffectPreset)
+        end
+    },
+    true
+)
+__TS__SetDescriptor(
+    AbilityType.prototype,
+    "targetChannelingEffectPresetsDE",
+    {
+        get = function(self)
+            return targetChannelingEffectPresetsDEByAbilityTypeId[self.id] or self.targetChannelingEffectPresets
+        end,
+        set = function(self, targetChannelingEffectPresetsDE)
+            targetChannelingEffectPresetsDEByAbilityTypeId[self.id] = map(targetChannelingEffectPresetsDE, toEffectPreset)
         end
     },
     true
@@ -456,6 +602,33 @@ __TS__SetDescriptor(
     },
     true
 )
+__TS__SetDescriptor(
+    AbilityType.prototype,
+    "targetEffectPresetsDE",
+    {
+        get = function(self)
+            return self:getAttachmentPresetListField(
+                "atat:de",
+                array(
+                    6,
+                    function(i) return ("ata" .. tostring(i)) .. ":de" end
+                )
+            )
+        end,
+        set = function(self, targetEffectPresetsDE)
+            self:setAttachmentPresetListField(
+                "atat:de",
+                array(
+                    6,
+                    function(i) return ("ata" .. tostring(i)) .. ":de" end
+                ),
+                "atac:de",
+                targetEffectPresetsDE
+            )
+        end
+    },
+    true
+)
 __TS__SetDescriptor(
     AbilityType.prototype,
     "turnOffButtonPositionX",
@@ -888,6 +1061,19 @@ __TS__SetDescriptor(
     },
     true
 )
+__TS__SetDescriptor(
+    AbilityType.prototype,
+    "isHeroAbility",
+    {
+        get = function(self)
+            return self:getBooleanField("aher")
+        end,
+        set = function(self, isHeroAbility)
+            self:setBooleanField("aher", isHeroAbility)
+        end
+    },
+    true
+)
 __TS__SetDescriptor(
     AbilityType.prototype,
     "learnLevelMinimum",
@@ -978,9 +1164,7 @@ for abilityTypeId, animationFQN in pairs(postcompile(function() return castAnima
         4,
         function(caster, ability)
             if ability:getField(ABILITY_RLF_CASTING_TIME) ~= 0 then
-                Timer:run(function()
-                    caster:playAnimation(animationFQN)
-                end)
+                Timer:run(caster, "playAnimation", animationFQN)
             end
         end
     )
@@ -991,20 +1175,27 @@ for abilityTypeId, soundPresetId in pairs(postcompile(function() return targetEf
             4,
             function(caster, ability, target)
                 if __TS__InstanceOf(target, Unit) then
-                    Sound3D:playFromLabel(soundPresetId, SoundPreset.Ability, target)
+                    Sound3D:playFromLabel(soundPresetId, SoundSettings.Ability, target)
                 else
-                    Sound3D:playFromLabel(soundPresetId, SoundPreset.Ability, target.x, target.y)
+                    Sound3D:playFromLabel(soundPresetId, SoundSettings.Ability, target.x, target.y)
                 end
             end
         )
         Unit.abilityPointTargetChannelingStartEvent[abilityTypeId]:addListener(
             4,
             function(caster, ability, x, y)
-                Sound3D:playFromLabel(soundPresetId, SoundPreset.Ability, x, y)
+                Sound3D:playFromLabel(soundPresetId, SoundSettings.Ability, x, y)
             end
         )
     end
 end
+local unsupportedEffectSoundAbilityTypeIds = luaSetOf(fourCC("AAns"))
+Unit.abilityChannelingStartEvent:addListener(function(caster, ability)
+    local soundPresetId = ability:getField(ABILITY_SF_EFFECT_SOUND)
+    if isSoundLabelCustom(soundPresetId) or soundPresetId ~= "" and unsupportedEffectSoundAbilityTypeIds[ability.parentTypeId] ~= nil then
+        Sound3D:playFromLabel(soundPresetId, SoundSettings.Ability, caster)
+    end
+end)
 local casterCastingEffectModelPathsByAbilityTypeId = postcompile(function()
     return mapValues(
         casterCastingEffectPresetsByAbilityTypeId,
@@ -1017,10 +1208,17 @@ local casterCastingEffectAttachmentPointsByAbilityTypeId = postcompile(function(
         function(casterCastingEffectPresets) return map(casterCastingEffectPresets, extractAttachmentPresetInputNodeFQN) end
     )
 end)
+local casterCastingEffectParametersByAbilityTypeId = postcompile(function()
+    return mapValues(
+        casterCastingEffectPresetsByAbilityTypeId,
+        function(casterCastingEffectPresets) return map(casterCastingEffectPresets, "parameters") end
+    )
+end)
 local casterCastingEffectsByCaster = {}
 local function handleAbilityCastingStartEvent(caster, ability)
     local effectModelPaths = casterCastingEffectModelPathsByAbilityTypeId[ability.typeId]
     local attachmentPoints = casterCastingEffectAttachmentPointsByAbilityTypeId[ability.typeId]
+    local parameters = casterCastingEffectParametersByAbilityTypeId[ability.typeId]
     local effects = {}
     if effectModelPaths ~= nil then
         for i = 1, #effectModelPaths do
@@ -1029,7 +1227,7 @@ local function handleAbilityCastingStartEvent(caster, ability)
             if attachmentPoint == nil or attachmentPoint == "" then
                 attachmentPoint = "origin"
             end
-            effects[i] = Effect:createTarget(effectModelPath, caster, attachmentPoint)
+            effects[i] = Effect:create(effectModelPath, caster, attachmentPoint, parameters and parameters[i])
         end
     end
     casterCastingEffectsByCaster[caster] = effects
@@ -1060,10 +1258,17 @@ local casterChannelingEffectAttachmentPointsByAbilityTypeId = postcompile(functi
         function(casterChannelingEffectPresets) return map(casterChannelingEffectPresets, extractAttachmentPresetInputNodeFQN) end
     )
 end)
+local casterChannelingEffectParametersByAbilityTypeId = postcompile(function()
+    return mapValues(
+        casterChannelingEffectPresetsByAbilityTypeId,
+        function(casterChannelingEffectPresets) return map(casterChannelingEffectPresets, "parameters") end
+    )
+end)
 local casterChannelingEffectsByCaster = {}
 local function handleAbilityChannelingStartEvent(caster, ability)
     local effectModelPaths = casterChannelingEffectModelPathsByAbilityTypeId[ability.typeId]
     local attachmentPoints = casterChannelingEffectAttachmentPointsByAbilityTypeId[ability.typeId]
+    local parameters = casterChannelingEffectParametersByAbilityTypeId[ability.typeId]
     local effects = {}
     if effectModelPaths ~= nil then
         for i = 1, #effectModelPaths do
@@ -1072,23 +1277,128 @@ local function handleAbilityChannelingStartEvent(caster, ability)
             if attachmentPoint == nil or attachmentPoint == "" then
                 attachmentPoint = "origin"
             end
-            effects[i] = Effect:createTarget(effectModelPath, caster, attachmentPoint)
+            effects[i] = Effect:create(effectModelPath, caster, attachmentPoint, parameters and parameters[i])
         end
     end
     casterChannelingEffectsByCaster[caster] = effects
 end
+local targetChannelingEffectsDataByAbilityTypeId = postcompile(function()
+    local function toData(presets)
+        if presets == nil then
+            return nil
+        end
+        local attachmentPoints = map(presets, extractAttachmentPresetInputNodeFQN)
+        local parameters = map(presets, "parameters")
+        return {
+            modelPaths = map(presets, extractAttachmentPresetInputModelPath),
+            attachmentPoints = __TS__ArraySome(
+                attachmentPoints,
+                function(____, attachmentPoint) return attachmentPoint ~= "" end
+            ) and attachmentPoints or nil,
+            parameters = __TS__ArraySome(
+                presets,
+                function(____, preset) return preset.parameters ~= nil end
+            ) and parameters or nil
+        }
+    end
+    local abilityTypeIds = {}
+    for ____, presetsByAbilityTypeId in ipairs({targetChannelingEffectPresetsByAbilityTypeId, targetChannelingEffectPresetsSDByAbilityTypeId, targetChannelingEffectPresetsHDByAbilityTypeId, targetChannelingEffectPresetsDEByAbilityTypeId}) do
+        for abilityTypeId in pairs(presetsByAbilityTypeId) do
+            abilityTypeIds[abilityTypeId] = true
+        end
+    end
+    local result = {}
+    for abilityTypeId in pairs(abilityTypeIds) do
+        local presets = targetChannelingEffectPresetsByAbilityTypeId[abilityTypeId]
+        local presetsSD = targetChannelingEffectPresetsSDByAbilityTypeId[abilityTypeId]
+        local presetsHD = targetChannelingEffectPresetsHDByAbilityTypeId[abilityTypeId]
+        local presetsDE = targetChannelingEffectPresetsDEByAbilityTypeId[abilityTypeId]
+        local ____math_max_8 = math.max
+        local ____opt_0 = presetsSD or presets
+        local ____temp_6 = ____opt_0 and #____opt_0 or 0
+        local ____opt_2 = presetsHD or presets
+        local ____temp_7 = ____opt_2 and #____opt_2 or 0
+        local ____opt_4 = presetsDE or presets
+        result[abilityTypeId] = {
+            effectCount = ____math_max_8(____temp_6, ____temp_7, ____opt_4 and #____opt_4 or 0),
+            data = toData(presets),
+            dataSD = toData(presetsSD),
+            dataHD = toData(presetsHD),
+            dataDE = toData(presetsDE)
+        }
+    end
+    return result
+end)
+local targetChannelingEffectsByCaster = {}
+local function createTargetChannelingEffects(caster, abilityTypeId, xOrWidget, y)
+    local effectsData = targetChannelingEffectsDataByAbilityTypeId[abilityTypeId]
+    if effectsData == nil then
+        return
+    end
+    local graphicsMode = LocalClient.graphicsMode
+    local ____temp_10
+    if graphicsMode == 0 then
+        ____temp_10 = effectsData.dataSD
+    else
+        local ____temp_9
+        if graphicsMode == 1 then
+            ____temp_9 = effectsData.dataHD
+        else
+            ____temp_9 = effectsData.dataDE
+        end
+        ____temp_10 = ____temp_9
+    end
+    local data = ____temp_10 or effectsData.data
+    local modelPaths = data and data.modelPaths
+    local attachmentPoints = data and data.attachmentPoints
+    local parameters = data and data.parameters
+    local effects = {}
+    for i = 1, effectsData.effectCount do
+        local effectModelPath = modelPaths and modelPaths[i] or ""
+        local effectParameters = parameters and parameters[i]
+        if type(xOrWidget) == "number" then
+            effects[i] = Effect:create(effectModelPath, xOrWidget, y, effectParameters)
+        else
+            local attachmentPoint = attachmentPoints and attachmentPoints[i]
+            if attachmentPoint == nil or attachmentPoint == "" then
+                attachmentPoint = "origin"
+            end
+            effects[i] = Effect:create(effectModelPath, xOrWidget, attachmentPoint, effectParameters)
+        end
+    end
+    targetChannelingEffectsByCaster[caster] = effects
+end
+local function handleAbilityWidgetTargetChannelingStartEvent(caster, ability, target)
+    createTargetChannelingEffects(caster, ability.typeId, target)
+end
+local function handleAbilityPointTargetChannelingStartEvent(caster, ability, x, y)
+    createTargetChannelingEffects(caster, ability.typeId, x, y)
+end
 local function handleAbilityStopChannelingEvent(caster)
-    local effects = casterChannelingEffectsByCaster[caster]
-    if effects ~= nil then
-        for i = 1, #effects do
-            effects[i]:destroy()
+    local casterEffects = casterChannelingEffectsByCaster[caster]
+    if casterEffects ~= nil then
+        for i = 1, #casterEffects do
+            casterEffects[i]:destroy()
         end
         casterChannelingEffectsByCaster[caster] = nil
     end
+    local targetEffects = targetChannelingEffectsByCaster[caster]
+    if targetEffects ~= nil then
+        for i = 1, #targetEffects do
+            targetEffects[i]:destroy()
+        end
+        targetChannelingEffectsByCaster[caster] = nil
+    end
 end
 for abilityTypeId in pairs(casterChannelingEffectModelPathsByAbilityTypeId) do
     Unit.abilityChannelingStartEvent[abilityTypeId]:addListener(4, handleAbilityChannelingStartEvent)
     Unit.abilityChannelingFinishEvent[abilityTypeId]:addListener(4, handleAbilityStopChannelingEvent)
     Unit.abilityStopEvent[abilityTypeId]:addListener(4, handleAbilityStopChannelingEvent)
 end
+for abilityTypeId in pairs(targetChannelingEffectsDataByAbilityTypeId) do
+    Unit.abilityWidgetTargetChannelingStartEvent[abilityTypeId]:addListener(4, handleAbilityWidgetTargetChannelingStartEvent)
+    Unit.abilityPointTargetChannelingStartEvent[abilityTypeId]:addListener(4, handleAbilityPointTargetChannelingStartEvent)
+    Unit.abilityChannelingFinishEvent[abilityTypeId]:addListener(4, handleAbilityStopChannelingEvent)
+    Unit.abilityStopEvent[abilityTypeId]:addListener(4, handleAbilityStopChannelingEvent)
+end
 return ____exports