Back to Home

ESO Lua File v101041

ingame/utilitywheel/keyboard/assignableutilitywheel_keyboard.lua

[◄ back to folders ]
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
ZO_AssignableUtilityWheel_Keyboard = ZO_AssignableUtilityWheel_Shared:Subclass()
function ZO_AssignableUtilityWheel_Keyboard:RegisterForEvents()
    ZO_AssignableUtilityWheel_Shared.RegisterForEvents(self)
    local function HandleInventorySlotPickup(bagId, slotIndex)
        local _, _, _, meetsUsageRequirement = GetItemInfo(bagId, slotIndex)
        local hotbarCategory = self:GetHotbarCategory()
        for slotNum, slot in pairs(self.slots) do
            local validInSlot = IsValidItemForSlot(bagId, slotIndex, slotNum, hotbarCategory)
            if validInSlot then
                self:ShowSlotDropCallout(slot:GetNamedChild("DropCallout"), meetsUsageRequirement)
            end
        end
    end
    
    local function HandleActionSlotPickup(slotType, sourceSlot, itemId)
        local MET_EQUIP_REQUIREMENTS = true -- This was already in a slot, chances are you're not going to fail equip requirements
        if slotType == ACTION_TYPE_ITEM then
            local hotbarCategory = self:GetHotbarCategory()
            for slotNum, slot in pairs(self.slots) do
                local validInSlot = IsValidItemForSlotByItemId(itemId, slotNum, hotbarCategory)
                if validInSlot then
                    self:ShowSlotDropCallout(slot:GetNamedChild("DropCallout"), MET_EQUIP_REQUIREMENTS)
                end
            end
        end
    end
    local function HandleCollectibleSlotPickup(collectibleId)
        local hotbarCategory = self:GetHotbarCategory()
        for slotNum, slot in pairs(self.slots) do
            local validInSlot = IsValidCollectibleForSlot(collectibleId, slotNum, hotbarCategory)
            if validInSlot then
                self:ShowSlotDropCallout(slot:GetNamedChild("DropCallout"), true)
            end
        end
    end
    local function HandleQuestItemSlotPickup(questIndex, stepIndex, conditionIndex, toolIndex, questItemId)
        local hotbarCategory = self:GetHotbarCategory()
        for slotNum, slot in pairs(self.slots) do
            local validInSlot = IsValidQuestItemForSlot(questItemId, slotNum, hotbarCategory)
            if validInSlot then
                self:ShowSlotDropCallout(slot:GetNamedChild("DropCallout"), true)
            end
        end
    end
    local function HandleEmoteSlotPickup(emoteId)
        local hotbarCategory = self:GetHotbarCategory()
        for slotNum, slot in pairs(self.slots) do
            local validInSlot = IsValidEmoteForSlot(emoteId, slotNum, hotbarCategory)
            if validInSlot then
                self:ShowSlotDropCallout(slot:GetNamedChild("DropCallout"), true)
            end
        end
    end
    local function HandleQuickChatSlotPickup(quickChatId)
        local hotbarCategory = self:GetHotbarCategory()
        for slotNum, slot in pairs(self.slots) do
            local validInSlot = IsValidQuickChatForSlot(quickChatId, slotNum, hotbarCategory)
            if validInSlot then
                self:ShowSlotDropCallout(slot:GetNamedChild("DropCallout"), true)
            end
        end
    end
    local function HandleCursorPickup(eventCode, cursorType, ...)
        if self:GetHotbarCategory() ~= ZO_UTILITY_WHEEL_HOTBAR_CATEGORY_HIDDEN then
            if cursorType == MOUSE_CONTENT_INVENTORY_ITEM then
                HandleInventorySlotPickup(...)
            elseif cursorType == MOUSE_CONTENT_ACTION then
                HandleActionSlotPickup(...)
            elseif cursorType == MOUSE_CONTENT_COLLECTIBLE then
                HandleCollectibleSlotPickup(...)
            elseif cursorType == MOUSE_CONTENT_QUEST_ITEM then
                HandleQuestItemSlotPickup(...)
            elseif cursorType == MOUSE_CONTENT_EMOTE then
                HandleEmoteSlotPickup(...)
            elseif cursorType == MOUSE_CONTENT_QUICK_CHAT then
                HandleQuickChatSlotPickup(...)
            end
        end
    end
    local function HandleCursorCleared()
        self:HideAllSlotDropCallouts()
    end
    local function HandleInventoryChanged()
        --This is only relevant if the wheel supports inventory items
        if not self.control:IsHidden() and self:IsActionTypeSupported(ACTION_TYPE_ITEM) then
            local hotbarCategory = self:GetHotbarCategory()
            for slotNum, slot in pairs(self.slots) do
                local slotType = GetSlotType(slotNum, hotbarCategory)
                if slotType == ACTION_TYPE_ITEM then
                    local itemCount = GetSlotItemCount(slotNum, hotbarCategory)
                    self:SetSlotItemCount(slot, itemCount)
                end
            end
        end
    end
    self.control:RegisterForEvent(EVENT_CURSOR_PICKUP, HandleCursorPickup)
    self.control:RegisterForEvent(EVENT_CURSOR_DROPPED, HandleCursorCleared)
    self.control:RegisterForEvent(EVENT_INVENTORY_FULL_UPDATE, HandleInventoryChanged)
    self.control:RegisterForEvent(EVENT_INVENTORY_SINGLE_SLOT_UPDATE, HandleInventoryChanged)
end
function ZO_AssignableUtilityWheel_Keyboard:InitializeKeybindStripDescriptors()
    ZO_AssignableUtilityWheel_Shared.InitializeKeybindStripDescriptors(self)
    self.mouseOverKeybindStripDescriptor =
    {
        alignment = KEYBIND_STRIP_ALIGN_RIGHT,
        -- Remove
        {
            name = GetString(SI_ABILITY_ACTION_CLEAR_SLOT),
            keybind = "UI_SHORTCUT_PRIMARY",
            callback = function()
                local slotId = self.mouseOverSlot.slotNum
                ClearSlot(slotId, self:GetHotbarCategory())
                PlaySound(SOUNDS.QUICKSLOT_CLEAR)
            end,
        },
    }
end
function ZO_AssignableUtilityWheel_Keyboard:Deactivate()
    ZO_AssignableUtilityWheel_Shared.Deactivate(self)
    KEYBIND_STRIP:RemoveKeybindButtonGroup(self.mouseOverKeybindStripDescriptor)
end
function ZO_AssignableUtilityWheel_Keyboard:PerformSlotLayout()
    local width, height = self.control:GetDimensions()
    local scale = self.control:GetScale()
    local halfWidth, halfHeight = width * scale * 0.5, height * scale * 0.5
    local numSlots = self.data.numSlots
    local actionBarOffset = self.data.startSlotIndex or 0
    local arcAnglePerSlotRadians, startingOffsetAngleRadians = self:GetArcAnglePerSlotAndStartingOffsetAngle()
    for i = 1, numSlots do
        local control = self.slots[i + actionBarOffset]
        local centerAngle = startingOffsetAngleRadians + (i - 1) * arcAnglePerSlotRadians
        local x = math.sin(centerAngle)
        local y = math.cos(centerAngle)
        if math.abs(x) < 0.01 then
            x = 0
        end
        if control.nameLabel then
            control.nameLabel:ClearAnchors()
            if x > 0 then
                control.nameLabel:SetAnchor(LEFT, control.icon, RIGHT, 15, 0)
            elseif x < 0 then
                control.nameLabel:SetAnchor(RIGHT, control.icon, LEFT, -15, 0)
            elseif y > 0 then
                control.nameLabel:SetAnchor(TOP, control.icon, BOTTOM, 0, 0)
            else
                control.nameLabel:SetAnchor(BOTTOM, control.icon, TOP, 0, -5)
            end
        end
        control:SetAnchor(CENTER, nil, CENTER, x * halfWidth, y * halfHeight)
        control:SetHidden(false)
    end
    --We need to do this *after* the rest of the layout has been completed so we can correctly calculate the ordinal indices
    for ordinalIndex, slot in self:OrdinalSlotIterator() do
    end
end
function ZO_AssignableUtilityWheel_Keyboard:InitializeSlots()
    local numSlots = self.data.numSlots
    local actionBarOffset = self.data.startSlotIndex or 0
    local showKeybinds = self:ShouldShowKeybinds()
    for i = actionBarOffset + 1, actionBarOffset + numSlots do
        local slot = CreateControlFromVirtual("$(parent)WheelSlot" .. i, self.control, "ZO_AssignableUtilityWheelSlot_Keyboard_Template")
        self.slots[i] = slot
        slot.button = slot:GetNamedChild("Button")
        slot.button.slotNum = i
        slot.button.object = self
        slot.icon = slot:GetNamedChild("Icon")
        slot.countText = slot:GetNamedChild("CountText")
        slot.nameLabel = slot:GetNamedChild("Label")
        slot.keybindLabel = slot:GetNamedChild("KeyLabel")
        ZO_KeyMarkupLabel_SetCustomOffsets(slot.keybindLabel, -5, 5, -2, 3)
        --If we are showing keybinds, do not show name labels in any circumstance
        if self.data.overrideShowNameLabels ~= nil or showKeybinds then
            slot.nameLabel:SetHidden(showKeybinds or not self.data.overrideShowNameLabels)
        end
        slot.keybindLabel:SetHidden(not showKeybinds)
        ZO_ActionSlot_SetupSlot(slot.icon, slot.button, ZO_UTILITY_SLOT_EMPTY_TEXTURE)
        ZO_CreateSparkleAnimation(slot)
    end
end
function ZO_AssignableUtilityWheel_Keyboard:DoSlotUpdate(physicalSlot, playAnimation)
    local slot = self.slots[physicalSlot]
    if slot then
        local hotbarCategory = self:GetHotbarCategory()
        local physicalSlotType = GetSlotType(physicalSlot, hotbarCategory)
        slot.nameLabel:SetText("")
        local showKeybinds = self:ShouldShowKeybinds()
        --If we are showing keybinds, do not show name labels in any circumstance
        if self.data.overrideShowNameLabels ~= nil or showKeybinds then
            slot.nameLabel:SetHidden(showKeybinds or not self.data.overrideShowNameLabels)
        else
            --Only the emote wheel shows name labels by default
            slot.nameLabel:SetHidden(hotbarCategory ~= HOTBAR_CATEGORY_EMOTE_WHEEL)
        end
        slot.keybindLabel:SetHidden(not showKeybinds)
        local slotIcon = GetSlotTexture(physicalSlot, hotbarCategory)
        if physicalSlotType == ACTION_TYPE_NOTHING then
            ZO_ActionSlot_SetupSlot(slot.icon, slot.button, ZO_UTILITY_SLOT_EMPTY_TEXTURE)
            slot.nameLabel:SetText(ZO_UTILITY_SLOT_EMPTY_STRING)
        elseif physicalSlotType == ACTION_TYPE_EMOTE then
            local name = ZO_UTILITY_SLOT_EMPTY_STRING
            local emoteId = GetSlotBoundId(physicalSlot, hotbarCategory)
            local emoteInfo = PLAYER_EMOTE_MANAGER:GetEmoteItemInfo(emoteId)
            local found = emoteInfo ~= nil
            if found then
                if emoteInfo.isOverriddenByPersonality then
                    name = ZO_PERSONALITY_EMOTES_COLOR:Colorize(emoteInfo.displayName)
                else
                    name = emoteInfo.displayName
                end
            end
            ZO_ActionSlot_SetupSlot(slot.icon, slot.button, slotIcon)
            slot.nameLabel:SetText(name)
        elseif physicalSlotType == ACTION_TYPE_QUICK_CHAT then
            local name = ZO_UTILITY_SLOT_EMPTY_STRING
            local quickChatId = GetSlotBoundId(physicalSlot, hotbarCategory)
            if QUICK_CHAT_MANAGER:HasQuickChat(quickChatId) then
                name = QUICK_CHAT_MANAGER:GetFormattedQuickChatName(quickChatId)
            end
            ZO_ActionSlot_SetupSlot(slot.icon, slot.button, slotIcon)
            slot.nameLabel:SetText(name)
        else
            local name = zo_strformat(SI_TOOLTIP_ITEM_NAME, GetSlotName(physicalSlot, hotbarCategory))
            slot.nameLabel:SetText(name)
            ZO_ActionSlot_SetupSlot(slot.icon, slot.button, slotIcon)
        end
        local itemCount = GetSlotItemCount(physicalSlot, hotbarCategory)
        self:SetSlotItemCount(slot, itemCount)
        local mouseOverControl = WINDOW_MANAGER:GetMouseOverControl()
        if mouseOverControl == slot.button then
            ZO_AbilitySlot_OnMouseEnter(slot.button)
        end
        if physicalSlotType ~= ACTION_TYPE_NOTHING and playAnimation and not slot:IsHidden() then
            ZO_PlaySparkleAnimation(slot)
            if hotbarCategory == HOTBAR_CATEGORY_QUICKSLOT_WHEEL and self:GetNumSlotted() == 1 then
                SetCurrentQuickslot(physicalSlot)
            end
        end
    end
end
function ZO_AssignableUtilityWheel_Keyboard:SetSlotItemCount(slotControl, itemCount)
    if itemCount then
        slotControl.icon:SetDesaturation(itemCount == 0 and 1 or 0)
        local USE_LOWERCASE_NUMBER_SUFFIXES = false
        local abbreviatedItemCount = ZO_AbbreviateAndLocalizeRadialMenuEntryCount(itemCount, USE_LOWERCASE_NUMBER_SUFFIXES)
        slotControl.countText:SetText(abbreviatedItemCount)
        slotControl.countText:SetHidden(false)
    else
        slotControl.icon:SetDesaturation(0)
        slotControl.countText:SetHidden(true)
    end
end
do
    local INITIAL_ROTATION = 0
    function ZO_AssignableUtilityWheel_Keyboard:GetArcAnglePerSlotAndStartingOffsetAngle()
        local numSlots = self.data.numSlots
        local arcAnglePerSlotRadians = ZO_TWO_PI / numSlots
        local startingOffsetAngleRadians = INITIAL_ROTATION + arcAnglePerSlotRadians
        return arcAnglePerSlotRadians, startingOffsetAngleRadians
    end
end
-- Returns the Slot at the ordinal position [1..n] beginning with the slot nearest to
-- ZO_RADIAL_MENU_PREFERRED_ORDINAL_STARTING_ANGLE_RADIANS and progressing clockwise or
-- counter-clockwise as defined by ZO_RADIAL_MENU_ORDINAL_DIRECTION.
function ZO_AssignableUtilityWheel_Keyboard:GetOrdinalSlot(ordinalIndex)
    local numSlots = self.data.numSlots
    if ordinalIndex > numSlots then
        -- Invalid ordinal index and/or no slots exist.
        return nil
    end
    -- Determine the shortest arc distance between the positional angle
    -- of the first slot and the preferred starting ordinal angle.
    local arcAnglePerSlotRadians, startingOffsetAngleRadians = self:GetArcAnglePerSlotAndStartingOffsetAngle()
    local preferredStartingAngleDistanceRadians = (ZO_RADIAL_MENU_PREFERRED_ORDINAL_STARTING_ANGLE_RADIANS - startingOffsetAngleRadians) % ZO_TWO_PI
    -- Determine the slot index offset of the starting ordinal index.
    local startingOrdinalSlotIndexOffset = zo_round(preferredStartingAngleDistanceRadians / arcAnglePerSlotRadians)
    -- Determine the final slot index associated with the requested ordinal
    -- index and clamp within the range [1..n] for n slots.
    local ordinalSlotIndex = startingOrdinalSlotIndexOffset + ((ordinalIndex - 1) * ZO_RADIAL_MENU_ORDINAL_DIRECTION)
    ordinalSlotIndex = (ordinalSlotIndex % numSlots) + 1
    local actionBarOffset = self.data.startSlotIndex or 0
    return self.slots[ordinalSlotIndex + actionBarOffset]
end
-- Returns an iterator that iterates over each Slot in the ordinal
-- order defined by ZO_AssignableUtilityWheel_Keyboard:GetOrdinalSlot.
function ZO_AssignableUtilityWheel_Keyboard:OrdinalSlotIterator()
    local ordinalIndex = 0
    return function()
        ordinalIndex = ordinalIndex + 1
        local slot = self:GetOrdinalSlot(ordinalIndex)
        if slot then
            return ordinalIndex, slot
        end
        return nil, nil
    end
end
function ZO_AssignableUtilityWheel_Keyboard:HideAllSlotDropCallouts()
    for _, slot in pairs(self.slots) do
       slot:GetNamedChild("DropCallout"):SetHidden(true)
    end
end
function ZO_AssignableUtilityWheel_Keyboard:ShowSlotDropCallout(calloutControl, meetsUsageRequirement)
    calloutControl:SetHidden(false)
    if meetsUsageRequirement then
        calloutControl:SetColor(1, 1, 1, 1)
    else
        calloutControl:SetColor(1, 0, 0, 1)
    end
end
function ZO_AssignableUtilityWheel_Keyboard:OnMouseOverUtilitySlot(slotControl)
    if slotControl.animation then
        slotControl.animation:PlayForward()
    end
    if self.mouseOverSlot ~= slotControl then
        PlaySound(SOUNDS.QUICKSLOT_MOUSEOVER)
        self.mouseOverSlot = slotControl
    end
    if IsSlotUsed(slotControl.slotNum, self:GetHotbarCategory()) then
        KEYBIND_STRIP:AddKeybindButtonGroup(self.mouseOverKeybindStripDescriptor)
    else
        KEYBIND_STRIP:RemoveKeybindButtonGroup(self.mouseOverKeybindStripDescriptor)
    end
end
function ZO_AssignableUtilityWheel_Keyboard:OnMouseExitUtilitySlot(slotControl)
    if slotControl.animation then
        slotControl.animation:PlayBackward()
    end
    self.mouseOverSlot = nil
    KEYBIND_STRIP:RemoveKeybindButtonGroup(self.mouseOverKeybindStripDescriptor)
end
    local button = control:GetNamedChild("Button")
    button:SetDrawTier(DT_MEDIUM)
    button:SetDrawLayer(DL_BACKGROUND)
    button.slotType = ABILITY_SLOT_TYPE_UTILITY
    local glow = control:GetNamedChild("Glow")
    button.animation = ANIMATION_MANAGER:CreateTimelineFromVirtual("UtilitySlotGlowAlphaAnimation", glow)
    local icon = control:GetNamedChild("Icon")
    icon:SetDrawTier(DT_MEDIUM)
    icon:SetDrawLayer(DL_BACKGROUND)
end