annotate Gear.lua @ 145:df1596b1a744 v68

Tweak for handling duplicate azerite items.
author yellowfive
date Sun, 09 Dec 2018 22:53:08 -0800
parents 55823e37403b
children cfdb5e5c828e
rev   line source
yellowfive@57 1 local Amr = LibStub("AceAddon-3.0"):GetAddon("AskMrRobot")
yellowfive@57 2 local L = LibStub("AceLocale-3.0"):GetLocale("AskMrRobot", true)
yellowfive@57 3 local AceGUI = LibStub("AceGUI-3.0")
yellowfive@57 4
yellowfive@139 5 local _cboSetups
yellowfive@139 6 local _panelGear
yellowfive@139 7 local _activeSetupId
yellowfive@139 8
yellowfive@139 9 local function getSetupById(id)
yellowfive@139 10 if not id then
yellowfive@139 11 id = _activeSetupId
yellowfive@139 12 end
yellowfive@139 13 local setup
yellowfive@139 14 for i,s in ipairs(Amr.db.char.GearSetups) do
yellowfive@139 15 if s.Id == id then
yellowfive@139 16 setup = s
yellowfive@139 17 break
yellowfive@139 18 end
yellowfive@139 19 end
yellowfive@139 20 return setup
yellowfive@139 21 end
yellowfive@57 22
yellowfive@57 23 -- Returns a number indicating how different two items are (0 means the same, higher means more different)
yellowfive@57 24 local function countItemDifferences(item1, item2)
yellowfive@124 25 -- both nil, the same
yellowfive@124 26 if not item1 and not item2 then
yellowfive@124 27 return 0
yellowfive@124 28 end
yellowfive@124 29
yellowfive@124 30 -- one nil and other not, or different id, totally different
yellowfive@124 31 if (not item1 and item2) or (item1 and not item2) or item1.id ~= item2.id then
yellowfive@124 32 return 10000
yellowfive@124 33 end
yellowfive@124 34
yellowfive@124 35 -- different versions of same item (id + bonus ids + suffix + drop level, constitutes a different physical drop)
yellowfive@135 36 if Amr.GetItemUniqueId(item1, true, true) ~= Amr.GetItemUniqueId(item2, true, true) then
yellowfive@57 37 return 1000
yellowfive@57 38 end
yellowfive@57 39
yellowfive@81 40 -- different upgrade levels of the same item
yellowfive@57 41 if item1.upgradeId ~= item2.upgradeId then
yellowfive@57 42 return 100
yellowfive@124 43 end
yellowfive@124 44
yellowfive@124 45 -- different azerite powers
yellowfive@124 46 local aztDiffs = 0
yellowfive@124 47 if item1.azerite or item2.azerite then
yellowfive@124 48 if item1.azerite and not item2.azerite then
yellowfive@124 49 aztDiffs = #item1.azerite * 10
yellowfive@124 50 elseif item2.azerite and not item1.azerite then
yellowfive@124 51 aztDiffs = #item2.azerite * 10
yellowfive@124 52 else
yellowfive@124 53 -- count up number in item1 but missing from item2
yellowfive@145 54 for i = 1, #item1.azerite do
yellowfive@145 55 local missing = true
yellowfive@124 56 for j = 1, #item2.azerite do
yellowfive@145 57 if item2.azerite[j] == item1.azerite[i] then
yellowfive@124 58 missing = false
yellowfive@124 59 end
yellowfive@124 60 end
yellowfive@124 61 if missing then
yellowfive@124 62 aztDiffs = aztDiffs + 10
yellowfive@124 63 end
yellowfive@124 64 end
yellowfive@124 65 -- count up number in item2 but missing from item1
yellowfive@124 66 for i = 1, #item2.azerite do
yellowfive@145 67 local missing = true
yellowfive@124 68 for j = 1, #item1.azerite do
yellowfive@145 69 if item1.azerite[j] == item2.azerite[i] then
yellowfive@124 70 missing = false
yellowfive@124 71 end
yellowfive@124 72 end
yellowfive@124 73 if missing then
yellowfive@124 74 aztDiffs = aztDiffs + 10
yellowfive@124 75 end
yellowfive@124 76 end
yellowfive@124 77 end
yellowfive@124 78 end
yellowfive@57 79
yellowfive@57 80 -- different gems
yellowfive@57 81 local gemDiffs = 0
yellowfive@57 82 for i = 1, 3 do
yellowfive@57 83 if item1.gemIds[i] ~= item2.gemIds[i] then
yellowfive@57 84 gemDiffs = gemDiffs + 1
yellowfive@57 85 end
yellowfive@57 86 end
yellowfive@57 87
yellowfive@57 88 -- different enchants
yellowfive@57 89 local enchantDiff = 0
yellowfive@57 90 if item1.enchantId ~= item2.enchantId then
yellowfive@57 91 enchantDiff = 1
yellowfive@57 92 end
yellowfive@57 93
yellowfive@124 94 return aztDiffs + gemDiffs + enchantDiff
yellowfive@57 95 end
yellowfive@57 96
yellowfive@57 97 -- given a table of items (keyed or indexed doesn't matter) find closest match to item, or nil if none are a match
yellowfive@124 98 local function findMatchingItemFromTable(item, list, bestItem, bestDiff, bestLoc, usedItems, tableType)
yellowfive@57 99 if not list then return nil end
yellowfive@57 100
yellowfive@73 101 local found = false
yellowfive@129 102 for k,listItem in pairs(list) do
yellowfive@57 103 if listItem then
yellowfive@57 104 local diff = countItemDifferences(item, listItem)
yellowfive@57 105 if diff < bestDiff then
yellowfive@57 106 -- each physical item can only be used once, the usedItems table has items we can't use in this search
yellowfive@57 107 local key = string.format("%s_%s", tableType, k)
yellowfive@57 108 if not usedItems[key] then
yellowfive@57 109 bestItem = listItem
yellowfive@57 110 bestDiff = diff
yellowfive@124 111 bestLoc = key
yellowfive@73 112 found = true
yellowfive@57 113 end
yellowfive@57 114 end
yellowfive@73 115 if found then break end
yellowfive@57 116 end
yellowfive@57 117 end
yellowfive@57 118
yellowfive@124 119 return bestItem, bestDiff, bestLoc
yellowfive@57 120 end
yellowfive@57 121
yellowfive@124 122 -- search the player's equipped gear, bag, and bank for an item that best matches the specified item
yellowfive@57 123 function Amr:FindMatchingItem(item, player, usedItems)
yellowfive@57 124 if not item then return nil end
yellowfive@57 125
yellowfive@57 126 local equipped = player.Equipped and player.Equipped[player.ActiveSpec] or nil
yellowfive@124 127 local bestItem, bestDiff, bestLoc = findMatchingItemFromTable(item, equipped, nil, 10000, nil, usedItems, "equip")
yellowfive@124 128 bestItem, bestDiff, bestLoc = findMatchingItemFromTable(item, player.BagItems, bestItem, bestDiff, bestLoc, usedItems, "bag")
yellowfive@124 129 if player.BankItems then
yellowfive@129 130 bestItem, bestDiff, bestLoc = findMatchingItemFromTable(item, player.BankItems, bestItem, bestDiff, bestLoc, usedItems, "bank")
yellowfive@124 131 end
yellowfive@57 132
yellowfive@124 133 if bestDiff >= 10000 then
yellowfive@124 134 return nil, 10000
yellowfive@57 135 else
yellowfive@57 136 usedItems[bestLoc] = true
yellowfive@124 137 return bestItem, bestDiff
yellowfive@57 138 end
yellowfive@57 139 end
yellowfive@57 140
yellowfive@57 141 local function renderEmptyGear(container)
yellowfive@57 142
yellowfive@57 143 local panelBlank = AceGUI:Create("AmrUiPanel")
yellowfive@57 144 panelBlank:SetLayout("None")
yellowfive@57 145 panelBlank:SetBackgroundColor(Amr.Colors.Black, 0.4)
yellowfive@124 146 container:AddChild(panelBlank)
yellowfive@57 147 panelBlank:SetPoint("TOPLEFT", container.content, "TOPLEFT", 6, 0)
yellowfive@57 148 panelBlank:SetPoint("BOTTOMRIGHT", container.content, "BOTTOMRIGHT")
yellowfive@57 149
yellowfive@57 150 local lbl = AceGUI:Create("AmrUiLabel")
yellowfive@124 151 panelBlank:AddChild(lbl)
yellowfive@57 152 lbl:SetText(L.GearBlank)
yellowfive@57 153 lbl:SetWidth(700)
yellowfive@57 154 lbl:SetJustifyH("MIDDLE")
yellowfive@57 155 lbl:SetFont(Amr.CreateFont("Italic", 16, Amr.Colors.TextTan))
yellowfive@57 156 lbl:SetPoint("BOTTOM", panelBlank.content, "CENTER", 0, 20)
yellowfive@57 157
yellowfive@57 158 local lbl2 = AceGUI:Create("AmrUiLabel")
yellowfive@124 159 panelBlank:AddChild(lbl2)
yellowfive@57 160 lbl2:SetText(L.GearBlank2)
yellowfive@57 161 lbl2:SetWidth(700)
yellowfive@57 162 lbl2:SetJustifyH("MIDDLE")
yellowfive@57 163 lbl2:SetFont(Amr.CreateFont("Italic", 16, Amr.Colors.TextTan))
yellowfive@57 164 lbl2:SetPoint("TOP", lbl.frame, "CENTER", 0, -20)
yellowfive@124 165 end
yellowfive@124 166
yellowfive@124 167 -- helper to create a widget for showing a socket or azerite power
yellowfive@124 168 local function createSocketWidget(panelMods, prevWidget, prevIsSocket, isEquipped)
yellowfive@124 169
yellowfive@124 170 -- highlight for socket that doesn't match
yellowfive@124 171 local socketBorder = AceGUI:Create("AmrUiPanel")
yellowfive@124 172 panelMods:AddChild(socketBorder)
yellowfive@124 173 if not prevIsSocket then
yellowfive@124 174 socketBorder:SetPoint("LEFT", prevWidget.frame, "RIGHT", 30, 0)
yellowfive@124 175 else
yellowfive@124 176 socketBorder:SetPoint("LEFT", prevWidget.frame, "RIGHT", 2, 0)
yellowfive@124 177 end
yellowfive@124 178 socketBorder:SetLayout("None")
yellowfive@124 179 socketBorder:SetBackgroundColor(Amr.Colors.Black, isEquipped and 0 or 1)
yellowfive@124 180 socketBorder:SetWidth(26)
yellowfive@124 181 socketBorder:SetHeight(26)
yellowfive@124 182 if isEquipped then
yellowfive@124 183 socketBorder:SetAlpha(0.3)
yellowfive@124 184 end
yellowfive@124 185
yellowfive@124 186 local socketBg = AceGUI:Create("AmrUiIcon")
yellowfive@124 187 socketBorder:AddChild(socketBg)
yellowfive@124 188 socketBg:SetPoint("TOPLEFT", socketBorder.content, "TOPLEFT", 1, -1)
yellowfive@124 189 socketBg:SetLayout("None")
yellowfive@124 190 socketBg:SetBorderWidth(2)
yellowfive@124 191 socketBg:SetIconBorderColor(Amr.Colors.Green, isEquipped and 0 or 1)
yellowfive@124 192 socketBg:SetWidth(24)
yellowfive@124 193 socketBg:SetHeight(24)
yellowfive@124 194
yellowfive@124 195 local socketIcon = AceGUI:Create("AmrUiIcon")
yellowfive@124 196 socketBg:AddChild(socketIcon)
yellowfive@124 197 socketIcon:SetPoint("CENTER", socketBg.content, "CENTER")
yellowfive@124 198 socketIcon:SetBorderWidth(1)
yellowfive@124 199 socketIcon:SetIconBorderColor(Amr.Colors.White)
yellowfive@124 200 socketIcon:SetWidth(18)
yellowfive@124 201 socketIcon:SetHeight(18)
yellowfive@124 202
yellowfive@124 203 return socketBorder, socketIcon
yellowfive@57 204 end
yellowfive@57 205
yellowfive@139 206 local function renderGear(setupId, container)
yellowfive@139 207
yellowfive@139 208 -- release all children that were previously rendered, we gonna redo it now
yellowfive@139 209 container:ReleaseChildren()
yellowfive@57 210
yellowfive@57 211 local player = Amr:ExportCharacter()
yellowfive@139 212
yellowfive@139 213 local gear
yellowfive@139 214 local spec
yellowfive@139 215 local setupIndex
yellowfive@139 216 for i, setup in ipairs(Amr.db.char.GearSetups) do
yellowfive@139 217 if setup.Id == setupId then
yellowfive@139 218 setupIndex = i
yellowfive@139 219 gear = setup.Gear
yellowfive@139 220 spec = setup.SpecSlot
yellowfive@139 221 break
yellowfive@139 222 end
yellowfive@139 223 end
yellowfive@139 224
yellowfive@57 225 local equipped = player.Equipped[player.ActiveSpec]
yellowfive@57 226
yellowfive@57 227 if not gear then
yellowfive@57 228 -- no gear has been imported for this spec so show a message
yellowfive@57 229 renderEmptyGear(container)
yellowfive@57 230 else
yellowfive@57 231 local panelGear = AceGUI:Create("AmrUiPanel")
yellowfive@57 232 panelGear:SetLayout("None")
yellowfive@57 233 panelGear:SetBackgroundColor(Amr.Colors.Black, 0.3)
yellowfive@124 234 container:AddChild(panelGear)
yellowfive@57 235 panelGear:SetPoint("TOPLEFT", container.content, "TOPLEFT", 6, 0)
yellowfive@57 236 panelGear:SetPoint("BOTTOMRIGHT", container.content, "BOTTOMRIGHT", -300, 0)
yellowfive@57 237
yellowfive@57 238 local panelMods = AceGUI:Create("AmrUiPanel")
yellowfive@57 239 panelMods:SetLayout("None")
yellowfive@124 240 panelMods:SetBackgroundColor(Amr.Colors.Black, 0.3)
yellowfive@124 241 container:AddChild(panelMods)
yellowfive@57 242 panelMods:SetPoint("TOPLEFT", panelGear.frame, "TOPRIGHT", 15, 0)
yellowfive@57 243 panelMods:SetPoint("BOTTOMRIGHT", container.content, "BOTTOMRIGHT")
yellowfive@57 244
yellowfive@57 245 -- spec icon
yellowfive@57 246 local icon = AceGUI:Create("AmrUiIcon")
yellowfive@57 247 icon:SetIconBorderColor(Amr.Colors.Classes[player.Class])
yellowfive@57 248 icon:SetWidth(48)
yellowfive@57 249 icon:SetHeight(48)
yellowfive@57 250
yellowfive@57 251 local iconSpec
yellowfive@81 252 if player.SubSpecs and player.SubSpecs[spec] then
yellowfive@57 253 iconSpec = player.SubSpecs[spec]
yellowfive@57 254 else
yellowfive@57 255 iconSpec = player.Specs[spec]
yellowfive@57 256 end
yellowfive@57 257
yellowfive@57 258 icon:SetIcon("Interface\\Icons\\" .. Amr.SpecIcons[iconSpec])
yellowfive@124 259 panelGear:AddChild(icon)
yellowfive@57 260 icon:SetPoint("TOPLEFT", panelGear.content, "TOPLEFT", 10, -10)
yellowfive@57 261
yellowfive@57 262 local btnEquip = AceGUI:Create("AmrUiButton")
yellowfive@81 263 btnEquip:SetText(L.GearButtonEquip(L.SpecsShort[player.Specs[spec]]))
yellowfive@57 264 btnEquip:SetBackgroundColor(Amr.Colors.Green)
yellowfive@57 265 btnEquip:SetFont(Amr.CreateFont("Regular", 14, Amr.Colors.White))
yellowfive@57 266 btnEquip:SetWidth(300)
yellowfive@57 267 btnEquip:SetHeight(26)
yellowfive@57 268 btnEquip:SetCallback("OnClick", function(widget)
yellowfive@139 269 Amr:EquipGearSet(setupIndex)
yellowfive@57 270 end)
yellowfive@57 271 panelGear:AddChild(btnEquip)
yellowfive@124 272 btnEquip:SetPoint("LEFT", icon.frame, "RIGHT", 40, 0)
yellowfive@124 273 btnEquip:SetPoint("RIGHT", panelGear.content, "RIGHT", -40, 0)
yellowfive@57 274
yellowfive@57 275 -- each physical item can only be used once, this tracks ones we have already used
yellowfive@57 276 local usedItems = {}
yellowfive@57 277
yellowfive@57 278 -- gear list
yellowfive@57 279 local prevElem = icon
yellowfive@57 280 for slotNum = 1, #Amr.SlotIds do
yellowfive@57 281 local slotId = Amr.SlotIds[slotNum]
yellowfive@57 282
yellowfive@124 283 local equippedItem = equipped and equipped[slotId] or nil
yellowfive@133 284 --local equippedItemLink = equipped and equipped.link or nil
yellowfive@57 285 local optimalItem = gear[slotId]
yellowfive@57 286 local optimalItemLink = Amr.CreateItemLink(optimalItem)
yellowfive@57 287
yellowfive@57 288 -- see if item is currently equipped, is false if don't have any item for that slot (e.g. OH for a 2-hander)
yellowfive@57 289 local isEquipped = false
yellowfive@135 290 if equippedItem and optimalItem and Amr.GetItemUniqueId(equippedItem, false, true) == Amr.GetItemUniqueId(optimalItem, false, true) then
yellowfive@145 291
yellowfive@145 292 if slotId == 1 or slotId == 3 or slotId == 5 then
yellowfive@145 293 -- show the item as not equipped if azerite doesn't match... might mean they have to switch to another version of same item
yellowfive@145 294 local aztDiff = countItemDifferences(equippedItem, optimalItem)
yellowfive@145 295 if aztDiff == 0 then
yellowfive@145 296 isEquipped = true
yellowfive@145 297 end
yellowfive@145 298 else
yellowfive@145 299 isEquipped = true
yellowfive@145 300 end
yellowfive@57 301 end
yellowfive@124 302
yellowfive@124 303 local isAzerite = optimalItem and C_AzeriteEmpoweredItem.IsAzeriteEmpoweredItemByID(optimalItem.id)
yellowfive@57 304
yellowfive@57 305 -- find the item in the player's inventory that best matches what the optimization wants to use
yellowfive@124 306 local matchItem = Amr:FindMatchingItem(optimalItem, player, usedItems)
yellowfive@57 307
yellowfive@57 308 -- slot label
yellowfive@57 309 local lbl = AceGUI:Create("AmrUiLabel")
yellowfive@124 310 panelGear:AddChild(lbl)
yellowfive@124 311 lbl:SetPoint("TOPLEFT", prevElem.frame, "BOTTOMLEFT", 0, -12)
yellowfive@57 312 lbl:SetText(Amr.SlotDisplayText[slotId])
yellowfive@57 313 lbl:SetWidth(85)
yellowfive@57 314 lbl:SetFont(Amr.CreateFont("Regular", 14, Amr.Colors.White))
yellowfive@57 315 prevElem = lbl
yellowfive@57 316
yellowfive@57 317 -- ilvl label
yellowfive@57 318 local lblIlvl = AceGUI:Create("AmrUiLabel")
yellowfive@124 319 panelGear:AddChild(lblIlvl)
yellowfive@124 320 lblIlvl:SetPoint("TOPLEFT", lbl.frame, "TOPRIGHT", 0, 0)
yellowfive@57 321 lblIlvl:SetWidth(45)
yellowfive@57 322 lblIlvl:SetFont(Amr.CreateFont("Italic", 14, Amr.Colors.TextTan))
yellowfive@57 323
yellowfive@57 324 -- equipped label
yellowfive@57 325 local lblEquipped = AceGUI:Create("AmrUiLabel")
yellowfive@124 326 panelGear:AddChild(lblEquipped)
yellowfive@124 327 lblEquipped:SetPoint("TOPLEFT", lblIlvl.frame, "TOPRIGHT", 0, 0)
yellowfive@57 328 lblEquipped:SetWidth(20)
yellowfive@57 329 lblEquipped:SetFont(Amr.CreateFont("Regular", 14, Amr.Colors.White))
yellowfive@57 330 lblEquipped:SetText(isEquipped and "E" or "")
yellowfive@57 331
yellowfive@57 332 -- item name/link label
yellowfive@57 333 local lblItem = AceGUI:Create("AmrUiLabel")
yellowfive@124 334 panelGear:AddChild(lblItem)
yellowfive@124 335 lblItem:SetPoint("TOPLEFT", lblEquipped.frame, "TOPRIGHT", 0, 0)
yellowfive@57 336 lblItem:SetWordWrap(false)
yellowfive@57 337 lblItem:SetWidth(345)
yellowfive@57 338 lblItem:SetFont(Amr.CreateFont(isEquipped and "Regular" or "Bold", isEquipped and 14 or 15, Amr.Colors.White))
yellowfive@57 339
yellowfive@133 340 -- fill the name/ilvl labels, which may require asynchronous loading of item information
yellowfive@57 341 if optimalItemLink then
yellowfive@133 342 local gameItem = Item:CreateFromItemLink(optimalItemLink)
yellowfive@133 343 if gameItem then
yellowfive@133 344 local q = gameItem:GetItemQuality()
yellowfive@133 345 if q == 6 then
yellowfive@133 346 -- for artifacts, we consider it equipped if the item id alone matches
yellowfive@133 347 if equippedItem and equippedItem.id == optimalItem.id then
yellowfive@133 348 isEquipped = true
yellowfive@133 349 end
yellowfive@133 350 lblEquipped:SetText(isEquipped and "E" or "")
yellowfive@133 351 end
yellowfive@124 352
yellowfive@137 353 lblItem:SetFont(Amr.CreateFont(isEquipped and "Regular" or "Bold", isEquipped and 14 or 15, Amr.Colors.Qualities[q] or Amr.Colors.White))
yellowfive@133 354 lblItem:SetText(gameItem:GetItemName())
yellowfive@133 355 lblIlvl:SetText(gameItem:GetCurrentItemLevel())
yellowfive@133 356 Amr:SetItemTooltip(lblItem, gameItem:GetItemLink(), "ANCHOR_TOPRIGHT")
yellowfive@133 357 end
yellowfive@57 358 end
yellowfive@57 359
yellowfive@57 360 -- modifications
yellowfive@57 361 if optimalItem then
yellowfive@57 362
yellowfive@124 363 -- gems or azerite powers
yellowfive@124 364 local prevSocket = nil
yellowfive@124 365
yellowfive@124 366 if isAzerite then
yellowfive@124 367 local azt = optimalItem.azerite or {}
yellowfive@124 368 for i,spellId in ipairs(azt) do
yellowfive@124 369 if spellId and spellId ~= 0 then
yellowfive@135 370 local equippedAzt = matchItem and matchItem.azerite or {}
yellowfive@124 371 local isPowerActive = Amr.Contains(equippedAzt, spellId)
yellowfive@124 372
yellowfive@124 373 local socketBorder, socketIcon = createSocketWidget(panelMods, prevSocket or lblItem, prevSocket, isPowerActive)
yellowfive@124 374
yellowfive@124 375 -- set icon and tooltip
yellowfive@133 376 local _, _, spellIcon = GetSpellInfo(spellId)
yellowfive@124 377 socketIcon:SetIcon(spellIcon)
yellowfive@124 378 Amr:SetSpellTooltip(socketIcon, spellId, "ANCHOR_TOPRIGHT")
yellowfive@124 379
yellowfive@124 380 prevSocket = socketBorder
yellowfive@124 381 end
yellowfive@124 382 end
yellowfive@124 383 else
yellowfive@124 384 for i = 1, #optimalItem.gemIds do
yellowfive@124 385 -- we rely on the fact that the gear sets coming back from the site will almost always have all sockets filled,
yellowfive@124 386 -- because it's a pain to get the actual number of sockets on an item from within the game
yellowfive@57 387 local g = optimalItem.gemIds[i]
yellowfive@124 388 if g == 0 then break end
yellowfive@124 389
yellowfive@124 390 local isGemEquipped = matchItem and matchItem.gemIds and matchItem.gemIds[i] == g
yellowfive@57 391
yellowfive@124 392 local socketBorder, socketIcon = createSocketWidget(panelMods, prevSocket or lblItem, prevSocket, isGemEquipped)
yellowfive@57 393
yellowfive@57 394 -- get icon for optimized gem
yellowfive@133 395 local gameItem = Item:CreateFromItemID(g)
yellowfive@133 396 if gameItem then
yellowfive@133 397 socketIcon:SetIcon(gameItem:GetItemIcon())
yellowfive@133 398 Amr:SetItemTooltip(socketIcon, gameItem:GetItemLink(), "ANCHOR_TOPRIGHT")
yellowfive@133 399 end
yellowfive@89 400
yellowfive@89 401 prevSocket = socketBorder
yellowfive@57 402 end
yellowfive@57 403 end
yellowfive@124 404
yellowfive@57 405 -- enchant
yellowfive@57 406 if optimalItem.enchantId and optimalItem.enchantId ~= 0 then
yellowfive@57 407 local isEnchantEquipped = matchItem and matchItem.enchantId and matchItem.enchantId == optimalItem.enchantId
yellowfive@135 408
yellowfive@57 409 local lblEnchant = AceGUI:Create("AmrUiLabel")
yellowfive@124 410 panelMods:AddChild(lblEnchant)
yellowfive@124 411 lblEnchant:SetPoint("TOPLEFT", lblItem.frame, "TOPRIGHT", 130, 0)
yellowfive@57 412 lblEnchant:SetWordWrap(false)
yellowfive@57 413 lblEnchant:SetWidth(170)
yellowfive@57 414 lblEnchant:SetFont(Amr.CreateFont(isEnchantEquipped and "Regular" or "Bold", 14, isEnchantEquipped and Amr.Colors.TextGray or Amr.Colors.White))
yellowfive@57 415
yellowfive@124 416 local enchInfo = Amr.db.char.ExtraEnchantData[optimalItem.enchantId]
yellowfive@57 417 if enchInfo then
yellowfive@57 418 lblEnchant:SetText(enchInfo.text)
yellowfive@57 419
yellowfive@133 420 local gameItem = Item:CreateFromItemID(enchInfo.itemId)
yellowfive@133 421 if gameItem then
yellowfive@133 422 Amr:SetItemTooltip(lblEnchant, gameItem:GetItemLink(), "ANCHOR_TOPRIGHT")
yellowfive@133 423 end
yellowfive@57 424 end
yellowfive@124 425
yellowfive@57 426 end
yellowfive@57 427 end
yellowfive@57 428
yellowfive@57 429 prevElem = lbl
yellowfive@57 430 end
yellowfive@57 431 end
yellowfive@57 432 end
yellowfive@57 433
yellowfive@139 434 local function onSetupChange(widget, eventName, value)
yellowfive@139 435 _activeSetupId = value
yellowfive@139 436 renderGear(_activeSetupId, _panelGear)
yellowfive@57 437 end
yellowfive@57 438
yellowfive@57 439 local function onImportClick(widget)
yellowfive@57 440 Amr:ShowImportWindow()
yellowfive@57 441 end
yellowfive@57 442
yellowfive@139 443 function Amr:PickFirstSetupForSpec()
yellowfive@139 444 local specSlot = GetSpecialization()
yellowfive@139 445 for i, setup in ipairs(Amr.db.char.GearSetups) do
yellowfive@139 446 if setup.SpecSlot == specSlot then
yellowfive@139 447 _activeSetupId = setup.Id
yellowfive@139 448 break
yellowfive@139 449 end
yellowfive@139 450 end
yellowfive@139 451 end
yellowfive@139 452
yellowfive@139 453 function Amr:GetActiveSetupId()
yellowfive@139 454 return _activeSetupId
yellowfive@139 455 end
yellowfive@139 456
yellowfive@139 457 function Amr:SetActiveSetupId(setupId)
yellowfive@139 458 _activeSetupId = setupId
yellowfive@139 459 end
yellowfive@139 460
yellowfive@139 461 function Amr:GetActiveSetupLabel()
yellowfive@139 462 if not _activeSetupId then
yellowfive@139 463 return nil
yellowfive@139 464 end
yellowfive@139 465 local setup = getSetupById(_activeSetupId)
yellowfive@139 466 if not setup then
yellowfive@139 467 return nil
yellowfive@139 468 else
yellowfive@139 469 return setup.Label
yellowfive@139 470 end
yellowfive@139 471 end
yellowfive@139 472
yellowfive@57 473 -- renders the main UI for the Gear tab
yellowfive@57 474 function Amr:RenderTabGear(container)
yellowfive@57 475
yellowfive@57 476 local btnImport = AceGUI:Create("AmrUiButton")
yellowfive@57 477 btnImport:SetText(L.GearButtonImportText)
yellowfive@57 478 btnImport:SetBackgroundColor(Amr.Colors.Orange)
yellowfive@57 479 btnImport:SetFont(Amr.CreateFont("Bold", 16, Amr.Colors.White))
yellowfive@57 480 btnImport:SetWidth(120)
yellowfive@57 481 btnImport:SetHeight(26)
yellowfive@57 482 btnImport:SetCallback("OnClick", onImportClick)
yellowfive@57 483 container:AddChild(btnImport)
yellowfive@124 484 btnImport:SetPoint("TOPLEFT", container.content, "TOPLEFT", 0, -81)
yellowfive@57 485
yellowfive@57 486 local lbl = AceGUI:Create("AmrUiLabel")
yellowfive@124 487 container:AddChild(lbl)
yellowfive@57 488 lbl:SetText(L.GearImportNote)
yellowfive@57 489 lbl:SetWidth(100)
yellowfive@57 490 lbl:SetFont(Amr.CreateFont("Italic", 12, Amr.Colors.TextTan))
yellowfive@57 491 lbl:SetJustifyH("MIDDLE")
yellowfive@57 492 lbl:SetPoint("TOP", btnImport.frame, "BOTTOM", 0, -5)
yellowfive@57 493
yellowfive@57 494 local lbl2 = AceGUI:Create("AmrUiLabel")
yellowfive@124 495 container:AddChild(lbl2)
yellowfive@57 496 lbl2:SetText(L.GearTipTitle)
yellowfive@57 497 lbl2:SetWidth(140)
yellowfive@57 498 lbl2:SetFont(Amr.CreateFont("Italic", 20, Amr.Colors.Text))
yellowfive@57 499 lbl2:SetJustifyH("MIDDLE")
yellowfive@57 500 lbl2:SetPoint("TOP", lbl.frame, "BOTTOM", 0, -50)
yellowfive@57 501
yellowfive@57 502 lbl = AceGUI:Create("AmrUiLabel")
yellowfive@124 503 container:AddChild(lbl)
yellowfive@57 504 lbl:SetText(L.GearTipText)
yellowfive@57 505 lbl:SetWidth(140)
yellowfive@57 506 lbl:SetFont(Amr.CreateFont("Italic", 12, Amr.Colors.Text))
yellowfive@57 507 lbl:SetJustifyH("MIDDLE")
yellowfive@57 508 lbl:SetPoint("TOP", lbl2.frame, "BOTTOM", 0, -5)
yellowfive@57 509
yellowfive@57 510 lbl2 = AceGUI:Create("AmrUiLabel")
yellowfive@124 511 container:AddChild(lbl2)
yellowfive@57 512 lbl2:SetText(L.GearTipCommands)
yellowfive@57 513 lbl2:SetWidth(130)
yellowfive@57 514 lbl2:SetFont(Amr.CreateFont("Italic", 12, Amr.Colors.Text))
yellowfive@57 515 lbl2:SetPoint("TOP", lbl.frame, "BOTTOM", 10, -5)
yellowfive@57 516
yellowfive@139 517 _cboSetups = AceGUI:Create("AmrUiDropDown")
yellowfive@139 518 _cboSetups:SetWidth(300)
yellowfive@139 519 container:AddChild(_cboSetups)
yellowfive@139 520 _cboSetups:SetPoint("TOPLEFT", container.content, "TOPLEFT", 150, -27.5)
yellowfive@81 521
yellowfive@139 522 _panelGear = AceGUI:Create("AmrUiPanel")
yellowfive@139 523 _panelGear:SetLayout("None")
yellowfive@139 524 _panelGear:SetBackgroundColor(Amr.Colors.Bg)
yellowfive@139 525 container:AddChild(_panelGear)
yellowfive@139 526 _panelGear:SetPoint("TOPLEFT", container.content, "TOPLEFT", 144, -58)
yellowfive@139 527 _panelGear:SetPoint("BOTTOMRIGHT", container.content, "BOTTOMRIGHT")
yellowfive@57 528
yellowfive@61 529 local btnShop = AceGUI:Create("AmrUiButton")
yellowfive@61 530 btnShop:SetText(L.GearButtonShop)
yellowfive@61 531 btnShop:SetBackgroundColor(Amr.Colors.Blue)
yellowfive@61 532 btnShop:SetFont(Amr.CreateFont("Regular", 14, Amr.Colors.White))
yellowfive@61 533 btnShop:SetWidth(245)
yellowfive@61 534 btnShop:SetHeight(26)
yellowfive@61 535 btnShop:SetCallback("OnClick", function(widget) Amr:ShowShopWindow() end)
yellowfive@61 536 container:AddChild(btnShop)
yellowfive@124 537 btnShop:SetPoint("TOPRIGHT", container.content, "TOPRIGHT", -20, -25)
yellowfive@139 538
yellowfive@139 539 -- pick a default tab based on player's current spec if none is already specified
yellowfive@139 540 if not _activeSetupId then
yellowfive@139 541 Amr:PickFirstSetupForSpec()
yellowfive@57 542 end
yellowfive@139 543
yellowfive@139 544 Amr:RefreshGearDisplay()
yellowfive@139 545
yellowfive@139 546 -- set event on dropdown after UI has been initially rendered
yellowfive@139 547 _cboSetups:SetCallback("OnChange", onSetupChange)
yellowfive@57 548 end
yellowfive@57 549
yellowfive@57 550 function Amr:ReleaseTabGear()
yellowfive@139 551 _cboSetups = nil
yellowfive@139 552 _panelGear = nil
yellowfive@57 553 end
yellowfive@57 554
yellowfive@57 555 -- refresh display of the current gear tab
yellowfive@139 556 function Amr:RefreshGearDisplay()
yellowfive@139 557
yellowfive@139 558 if not _panelGear then
yellowfive@139 559 return
yellowfive@139 560 end
yellowfive@139 561
yellowfive@139 562 -- fill the gear setup picker
yellowfive@139 563 local setupList = {}
yellowfive@139 564 for i, setup in ipairs(Amr.db.char.GearSetups) do
yellowfive@139 565 table.insert(setupList, { text = setup.Label, value = setup.Id })
yellowfive@139 566 end
yellowfive@139 567 _cboSetups:SetItems(setupList)
yellowfive@139 568
yellowfive@139 569 -- set selected value
yellowfive@139 570 local prev = _activeSetupId
yellowfive@139 571 _cboSetups:SelectItem(_activeSetupId)
yellowfive@139 572
yellowfive@139 573 if prev == _activeSetupId then
yellowfive@139 574 -- selecting will trigger the change event if it changed; if it didn't change, do a render now
yellowfive@139 575 renderGear(_activeSetupId, _panelGear)
yellowfive@139 576 end
yellowfive@57 577 end
yellowfive@57 578
yellowfive@57 579
yellowfive@57 580 ------------------------------------------------------------------------------------------------
yellowfive@57 581 -- Gear Set Management
yellowfive@57 582 ------------------------------------------------------------------------------------------------
yellowfive@57 583 local _waitingForSpec = 0
yellowfive@124 584 local _pendingGearOps = nil
yellowfive@124 585 local _currentGearOp = nil
yellowfive@124 586 local _itemLockAction = nil
yellowfive@124 587 local _gearOpPasses = 0
yellowfive@124 588 local _gearOpWaiting = nil
yellowfive@57 589
yellowfive@124 590 local beginEquipGearSet, processCurrentGearOp, nextGearOp
yellowfive@89 591
yellowfive@57 592 -- find the first empty slot in the player's backpack+bags
yellowfive@57 593 local function findFirstEmptyBagSlot()
yellowfive@57 594
yellowfive@57 595 local bagIds = {}
yellowfive@57 596 table.insert(bagIds, BACKPACK_CONTAINER)
yellowfive@57 597 for bagId = 1, NUM_BAG_SLOTS do
yellowfive@57 598 table.insert(bagIds, bagId)
yellowfive@57 599 end
yellowfive@57 600
yellowfive@57 601 for i, bagId in ipairs(bagIds) do
yellowfive@57 602 local numSlots = GetContainerNumSlots(bagId)
yellowfive@57 603 for slotId = 1, numSlots do
yellowfive@57 604 local _, _, _, _, _, _, itemLink = GetContainerItemInfo(bagId, slotId)
yellowfive@57 605 if not itemLink then
yellowfive@57 606 return bagId, slotId
yellowfive@57 607 end
yellowfive@57 608 end
yellowfive@57 609 end
yellowfive@57 610
yellowfive@57 611 return nil, nil
yellowfive@57 612 end
yellowfive@57 613
yellowfive@124 614 -- scan a bag for the best matching item
yellowfive@124 615 local function scanBagForItem(item, bagId, bestItem, bestDiff, bestLink)
yellowfive@124 616 local numSlots = GetContainerNumSlots(bagId)
yellowfive@124 617 for slotId = 1, numSlots do
yellowfive@124 618 local _, _, _, _, _, _, itemLink = GetContainerItemInfo(bagId, slotId)
yellowfive@124 619 -- we skip any stackable item, as far as we know, there is no equippable gear that can be stacked
yellowfive@124 620 if itemLink then
yellowfive@124 621 local bagItem = Amr.ParseItemLink(itemLink)
yellowfive@124 622 if bagItem ~= nil then
yellowfive@124 623 local diff = countItemDifferences(item, bagItem)
yellowfive@124 624 if diff < bestDiff then
yellowfive@124 625 bestItem = { bag = bagId, slot = slotId }
yellowfive@124 626 bestDiff = diff
yellowfive@124 627 bestLink = itemLink
yellowfive@124 628 end
yellowfive@124 629 end
yellowfive@124 630 end
yellowfive@57 631 end
yellowfive@124 632 return bestItem, bestDiff, bestLink
yellowfive@57 633 end
yellowfive@57 634
yellowfive@124 635 -- find the item in the player's inventory that best matches the current gear op item, favoring stuff already equipped, then in bags, then in bank
yellowfive@124 636 local function findCurrentGearOpItem()
yellowfive@124 637
yellowfive@124 638 local item = _currentGearOp.items[_currentGearOp.nextSlot]
yellowfive@124 639
yellowfive@57 640 local bestItem = nil
yellowfive@57 641 local bestLink = nil
yellowfive@124 642 local bestDiff = 10000
yellowfive@57 643
yellowfive@73 644 -- inventory
yellowfive@73 645 bestItem, bestDiff, bestLink = scanBagForItem(item, BACKPACK_CONTAINER, bestItem, bestDiff, bestLink)
yellowfive@73 646 for bagId = 1, NUM_BAG_SLOTS do
yellowfive@73 647 bestItem, bestDiff, bestLink = scanBagForItem(item, bagId, bestItem, bestDiff, bestLink)
yellowfive@73 648 end
yellowfive@73 649
yellowfive@129 650 -- with new approach, the item to use should never be equipped, should be in bags at this point
yellowfive@129 651 --[[
yellowfive@61 652 -- equipped items, but skip slots we have just equipped (to avoid e.g. just moving 2 of the same item back and forth between mh oh weapon slots)
yellowfive@57 653 for slotNum = 1, #Amr.SlotIds do
yellowfive@57 654 local slotId = Amr.SlotIds[slotNum]
yellowfive@124 655 if _currentGearOp.slotsRemaining[slotId] then
yellowfive@61 656 local itemLink = GetInventoryItemLink("player", slotId)
yellowfive@61 657 if itemLink then
yellowfive@61 658 local invItem = Amr.ParseItemLink(itemLink)
yellowfive@124 659 if invItem then
yellowfive@61 660 local diff = countItemDifferences(item, invItem)
yellowfive@61 661 if diff < bestDiff then
yellowfive@61 662 bestItem = { slot = slotId }
yellowfive@61 663 bestDiff = diff
yellowfive@61 664 bestLink = itemLink
yellowfive@61 665 end
yellowfive@57 666 end
yellowfive@57 667 end
yellowfive@57 668 end
yellowfive@57 669 end
yellowfive@129 670 ]]
yellowfive@129 671
yellowfive@57 672 -- bank
yellowfive@124 673 if bestDiff > 0 then
yellowfive@124 674 bestItem, bestDiff, bestLink = scanBagForItem(item, BANK_CONTAINER, bestItem, bestDiff, bestLink)
yellowfive@124 675 for bagId = NUM_BAG_SLOTS + 1, NUM_BAG_SLOTS + NUM_BANKBAGSLOTS do
yellowfive@124 676 bestItem, bestDiff, bestLink = scanBagForItem(item, bagId, bestItem, bestDiff, bestLink)
yellowfive@124 677 end
yellowfive@57 678 end
yellowfive@124 679
yellowfive@124 680 return bestItem, bestDiff, bestLink
yellowfive@124 681 end
yellowfive@124 682
yellowfive@143 683 local function createAmrEquipmentSet()
yellowfive@133 684
yellowfive@143 685 -- clear any currently ignored slots, ignore shirt and tabard
yellowfive@143 686 C_EquipmentSet.ClearIgnoredSlotsForSave()
yellowfive@133 687 C_EquipmentSet.IgnoreSlotForSave(INVSLOT_BODY) -- shirt
yellowfive@133 688 C_EquipmentSet.IgnoreSlotForSave(INVSLOT_TABARD)
yellowfive@133 689
yellowfive@133 690 -- for now use icon of the spec
yellowfive@133 691 local _, specName, _, setIcon = GetSpecializationInfo(GetSpecialization())
yellowfive@57 692
yellowfive@133 693 --[[
yellowfive@124 694 local item = Amr.ParseItemLink(GetInventoryItemLink("player", INVSLOT_MAINHAND))
yellowfive@124 695 if not item then
yellowfive@124 696 item = Amr.ParseItemLink(GetInventoryItemLink("player", INVSLOT_OFFHAND))
yellowfive@124 697 end
yellowfive@124 698 if item then
yellowfive@133 699 local itemObj = Item:CreateFromItemID(item.id)
yellowfive@133 700 if itemObj then
yellowfive@133 701 setIcon = itemObj:GetItemIcon()
yellowfive@133 702 end
yellowfive@133 703 end
yellowfive@133 704 ]]
yellowfive@133 705
yellowfive@139 706 local setup = getSetupById(_activeSetupId)
yellowfive@139 707 local setname = setup.Label -- "AMR " .. specName
yellowfive@133 708 local setid = C_EquipmentSet.GetEquipmentSetID(setname)
yellowfive@133 709 if setid then
yellowfive@145 710 local oldName, oldIcon = C_EquipmentSet.GetEquipmentSetInfo(setid)
yellowfive@145 711 setIcon = oldIcon
yellowfive@133 712 C_EquipmentSet.SaveEquipmentSet(setid, setIcon)
yellowfive@133 713 else
yellowfive@133 714 C_EquipmentSet.CreateEquipmentSet(setname, setIcon)
yellowfive@124 715 end
yellowfive@124 716 end
yellowfive@124 717
yellowfive@143 718 -- on completion, create an equipment manager set if desired
yellowfive@143 719 local function onEquipGearSetComplete()
yellowfive@143 720 if Amr.db.profile.options.disableEm then return end
yellowfive@143 721
yellowfive@143 722 -- create an equipment manager set
yellowfive@143 723 createAmrEquipmentSet()
yellowfive@143 724
yellowfive@143 725 -- need to call it twice because on first load the WoW equipment manager just doesn't work
yellowfive@143 726 Amr.Wait(1, function()
yellowfive@143 727 createAmrEquipmentSet()
yellowfive@143 728 end)
yellowfive@143 729 end
yellowfive@143 730
yellowfive@124 731 -- stop any currently in-progress gear swapping operation and clean up
yellowfive@124 732 local function disposeGearOp()
yellowfive@124 733 _pendingGearOps = nil
yellowfive@124 734 _currentGearOp = nil
yellowfive@124 735 _itemLockAction = nil
yellowfive@124 736 _gearOpPasses = 0
yellowfive@124 737 _gearOpWaiting = nil
yellowfive@124 738
yellowfive@124 739 -- make sure the gear tab is still in sync
yellowfive@139 740 Amr:RefreshGearDisplay()
yellowfive@124 741 end
yellowfive@124 742
yellowfive@124 743 -- initialize a gear op to start running it
yellowfive@139 744 local function initializeGearOp(op, setupId, pos)
yellowfive@124 745 op.pos = pos
yellowfive@139 746 op.setupId = setupId
yellowfive@124 747
yellowfive@124 748 -- fill the remaining slot list and set the starting slot
yellowfive@124 749 op.nextSlot = nil
yellowfive@124 750 op.slotsRemaining = {}
yellowfive@124 751 op.isWaiting = false
yellowfive@124 752 for slotId, item in pairs(op.items) do
yellowfive@124 753 op.slotsRemaining[slotId] = true
yellowfive@124 754 if not op.nextSlot then
yellowfive@124 755 op.nextSlot = slotId
yellowfive@124 756 end
yellowfive@124 757 end
yellowfive@124 758 end
yellowfive@124 759
yellowfive@124 760 function processCurrentGearOp()
yellowfive@124 761 if not _currentGearOp then return end
yellowfive@124 762
yellowfive@124 763 if _currentGearOp.remove then
yellowfive@124 764 -- remove the next item
yellowfive@124 765
yellowfive@124 766 -- check if the slot is already empty
yellowfive@124 767 local itemLink = GetInventoryItemLink("player", _currentGearOp.nextSlot)
yellowfive@124 768 if not itemLink then
yellowfive@124 769 nextGearOp()
yellowfive@124 770 return
yellowfive@124 771 end
yellowfive@124 772
yellowfive@57 773 -- find first empty bag slot
yellowfive@57 774 local invBag, invSlot = findFirstEmptyBagSlot()
yellowfive@57 775 if not invBag then
yellowfive@57 776 -- stop if bags are too full
yellowfive@57 777 Amr:Print(L.GearEquipErrorBagFull)
yellowfive@124 778 disposeGearOp()
yellowfive@57 779 return
yellowfive@57 780 end
yellowfive@57 781
yellowfive@124 782 PickupInventoryItem(_currentGearOp.nextSlot)
yellowfive@57 783 PickupContainerItem(invBag, invSlot)
yellowfive@57 784
yellowfive@124 785 -- set an action to happen on ITEM_UNLOCKED, triggered by ClearCursor
yellowfive@124 786 _itemLockAction = {
yellowfive@57 787 bagId = invBag,
yellowfive@124 788 slotId = invSlot,
yellowfive@124 789 isRemove = true
yellowfive@57 790 }
yellowfive@124 791
yellowfive@124 792 ClearCursor()
yellowfive@124 793 -- wait for remove to complete
yellowfive@124 794 else
yellowfive@124 795 -- equip the next item
yellowfive@57 796
yellowfive@124 797 local bestItem, bestDiff, bestLink = findCurrentGearOpItem()
yellowfive@124 798
yellowfive@124 799 _itemLockAction = nil
yellowfive@57 800 ClearCursor()
yellowfive@124 801
yellowfive@124 802 if not bestItem then
yellowfive@124 803 -- stop if we can't find an item
yellowfive@124 804 Amr:Print(L.GearEquipErrorNotFound)
yellowfive@124 805 Amr:Print(L.GearEquipErrorNotFound2)
yellowfive@124 806 disposeGearOp()
yellowfive@124 807
yellowfive@124 808 elseif bestItem and bestItem.bag and (bestItem.bag == BANK_CONTAINER or bestItem.bag >= NUM_BAG_SLOTS + 1 and bestItem.bag <= NUM_BAG_SLOTS + NUM_BANKBAGSLOTS) then
yellowfive@124 809 -- find first empty bag slot
yellowfive@124 810 local invBag, invSlot = findFirstEmptyBagSlot()
yellowfive@124 811 if not invBag then
yellowfive@124 812 -- stop if bags are too full
yellowfive@124 813 Amr:Print(L.GearEquipErrorBagFull)
yellowfive@124 814 disposeGearOp()
yellowfive@124 815 return
yellowfive@124 816 end
yellowfive@124 817
yellowfive@124 818 -- move from bank to bag
yellowfive@124 819 PickupContainerItem(bestItem.bag, bestItem.slot)
yellowfive@124 820 PickupContainerItem(invBag, invSlot)
yellowfive@124 821
yellowfive@124 822 -- set an action to happen on ITEM_UNLOCKED, triggered by ClearCursor
yellowfive@124 823 _itemLockAction = {
yellowfive@124 824 bagId = invBag,
yellowfive@124 825 slotId = invSlot,
yellowfive@124 826 isBank = true
yellowfive@124 827 }
yellowfive@124 828
yellowfive@124 829 ClearCursor()
yellowfive@124 830 -- now we need to wait for game event to continue and try this item again after it is in our bag and unlocked
yellowfive@124 831
yellowfive@124 832 elseif (bestItem.bag or bestItem.bag == 0) and not Amr:CanEquip(bestItem.bag, bestItem.slot) then
yellowfive@57 833 -- if an item is not soulbound, then warn the user and quit
yellowfive@57 834 Amr:Print(L.GearEquipErrorSoulbound(bestLink))
yellowfive@124 835 disposeGearOp()
yellowfive@124 836
yellowfive@57 837 else
yellowfive@124 838
yellowfive@129 839 --print("equipping " .. bestLink .. " in slot " .. _currentGearOp.nextSlot)
yellowfive@129 840
yellowfive@57 841 -- an item in the player's bags or already equipped, equip it
yellowfive@57 842 if bestItem.bag then
yellowfive@57 843 PickupContainerItem(bestItem.bag, bestItem.slot)
yellowfive@57 844 else
yellowfive@124 845 _gearOpWaiting.inventory[bestItem.slot] = true
yellowfive@57 846 PickupInventoryItem(bestItem.slot)
yellowfive@57 847 end
yellowfive@124 848 _gearOpWaiting.inventory[_currentGearOp.nextSlot] = true
yellowfive@124 849 PickupInventoryItem(_currentGearOp.nextSlot)
yellowfive@124 850
yellowfive@124 851 -- don't wait for now, do all equips at once
yellowfive@124 852 --[[
yellowfive@124 853 -- set an action to happen on ITEM_UNLOCKED, triggered by ClearCursor
yellowfive@124 854 _itemLockAction = {
yellowfive@124 855 bagId = bestItem.bag,
yellowfive@124 856 slotId = bestItem.slot,
yellowfive@124 857 invSlot = _currentGearOp.nextSlot,
yellowfive@124 858 isEquip = true
yellowfive@124 859 }
yellowfive@124 860 ]]
yellowfive@124 861
yellowfive@124 862 ClearCursor()
yellowfive@124 863 nextGearOp()
yellowfive@124 864 end
yellowfive@124 865
yellowfive@124 866 end
yellowfive@124 867 end
yellowfive@124 868
yellowfive@124 869 -- when a gear op completes successfully, this will advance to the next op or finish
yellowfive@124 870 function nextGearOp()
yellowfive@124 871 if not _currentGearOp then return end
yellowfive@124 872
yellowfive@139 873 local setupId = _currentGearOp.setupId
yellowfive@124 874 local pos = _currentGearOp.pos
yellowfive@124 875 local passes = _gearOpPasses
yellowfive@124 876
yellowfive@124 877 -- mark the slot as done and move to the next
yellowfive@124 878 if _currentGearOp.nextSlot then
yellowfive@124 879 _currentGearOp.slotsRemaining[_currentGearOp.nextSlot] = nil
yellowfive@124 880 _currentGearOp.nextSlot = nil
yellowfive@124 881 for slotId, item in pairs(_currentGearOp.items) do
yellowfive@124 882 if _currentGearOp.slotsRemaining[slotId] then
yellowfive@124 883 _currentGearOp.nextSlot = slotId
yellowfive@124 884 break
yellowfive@124 885 end
yellowfive@124 886 end
yellowfive@124 887 end
yellowfive@124 888
yellowfive@124 889 if not _currentGearOp.nextSlot then
yellowfive@124 890 -- see if anything is still in progress and we want to wait for it before continuing
yellowfive@124 891 local inProgress = not Amr.IsEmpty(_gearOpWaiting.inventory)
yellowfive@124 892
yellowfive@124 893 if (_currentGearOp.wait or _currentGearOp.remove) and inProgress then
yellowfive@124 894 -- this will cause the item unlock handler to call nextGearOp again when all in-progress swaps have unlocked related slots
yellowfive@124 895 _currentGearOp.isWaiting = true
yellowfive@124 896 else
yellowfive@124 897 _currentGearOp = _pendingGearOps[pos + 1]
yellowfive@124 898 if _currentGearOp then
yellowfive@124 899 -- we have another op, do it
yellowfive@139 900 initializeGearOp(_currentGearOp, setupId, pos + 1)
yellowfive@124 901 processCurrentGearOp()
yellowfive@124 902 else
yellowfive@124 903 -- we are done
yellowfive@124 904 disposeGearOp()
yellowfive@124 905
yellowfive@124 906 -- this will check if not all items were swapped, and either finish up, try again, or abort if have tried too many times
yellowfive@139 907 beginEquipGearSet(setupId, passes + 1)
yellowfive@124 908 end
yellowfive@124 909 end
yellowfive@124 910 else
yellowfive@124 911 -- do the next item
yellowfive@124 912 processCurrentGearOp()
yellowfive@124 913 end
yellowfive@124 914
yellowfive@124 915 end
yellowfive@124 916
yellowfive@124 917 local function handleItemUnlocked(bagId, slotId)
yellowfive@124 918
yellowfive@124 919 -- mark anything that is waiting as unlocked if it is no longer locked
yellowfive@124 920 if _currentGearOp and _gearOpWaiting then
yellowfive@124 921 for i,s in ipairs(Amr.SlotIds) do
yellowfive@124 922 if not IsInventoryItemLocked(s) then
yellowfive@124 923 _gearOpWaiting.inventory[s] = nil
yellowfive@124 924 end
yellowfive@124 925 end
yellowfive@124 926 end
yellowfive@124 927
yellowfive@124 928 if _itemLockAction then
yellowfive@124 929 if _itemLockAction.isRemove then
yellowfive@124 930 -- waiting for a specific remove op to finish before continuing
yellowfive@124 931 if bagId == _itemLockAction.bagId and slotId == _itemLockAction.slotId then
yellowfive@124 932 _itemLockAction = nil
yellowfive@124 933 nextGearOp()
yellowfive@124 934 end
yellowfive@124 935 elseif _itemLockAction.isBank then
yellowfive@124 936 -- waiting for an item to move from bank into inventory, then reprocess the current op
yellowfive@124 937 if bagId == _itemLockAction.bagId and slotId == _itemLockAction.slotId then
yellowfive@124 938 _itemLockAction = nil
yellowfive@124 939 processCurrentGearOp()
yellowfive@124 940 end
yellowfive@124 941
yellowfive@124 942 elseif _itemLockAction.isEquip then
yellowfive@124 943 -- this is not currently used... we do all equips at once usually, but could go back to this if it causes problems
yellowfive@124 944
yellowfive@124 945 -- waiting for a specific equip op to finish
yellowfive@61 946
yellowfive@124 947 -- inventory slot we're swapping to is still locked, can't continue yet
yellowfive@124 948 if IsInventoryItemLocked(_itemLockAction.invSlot) then return end
yellowfive@124 949
yellowfive@124 950 if _itemLockAction.bagId then
yellowfive@124 951 local _, _, locked = GetContainerItemInfo(_itemLockAction.bagId, _itemLockAction.slotId)
yellowfive@124 952 -- the bag slot we're swapping from is still locked, can't continue yet
yellowfive@124 953 if locked then return end
yellowfive@124 954 else
yellowfive@124 955 -- inventory slot we're swapping from is still locked, can't continue yet
yellowfive@124 956 if IsInventoryItemLocked(_itemLockAction.slotId) then return end
yellowfive@124 957 end
yellowfive@124 958
yellowfive@124 959 _itemLockAction = nil
yellowfive@124 960 nextGearOp()
yellowfive@124 961 else
yellowfive@124 962 -- unknown... shouldn't happen
yellowfive@124 963 _itemLockAction = nil
yellowfive@57 964 end
yellowfive@124 965 else
yellowfive@124 966
yellowfive@124 967 -- not waiting on a specific action, check if we are waiting for all locked slots to open up and they are done
yellowfive@124 968 if _currentGearOp and _gearOpWaiting and _currentGearOp.isWaiting and Amr.IsEmpty(_gearOpWaiting.inventory) then
yellowfive@124 969 nextGearOp()
yellowfive@124 970 end
yellowfive@57 971 end
yellowfive@57 972
yellowfive@57 973 end
yellowfive@57 974
yellowfive@124 975 local function shuffle(tbl)
yellowfive@124 976 local size = #tbl
yellowfive@124 977 for i = size, 1, -1 do
yellowfive@124 978 local rand = math.random(size)
yellowfive@124 979 tbl[i], tbl[rand] = tbl[rand], tbl[i]
yellowfive@89 980 end
yellowfive@124 981 return tbl
yellowfive@89 982 end
yellowfive@89 983
yellowfive@129 984 local _ohFirst = {
yellowfive@129 985 [20] = true, -- PaladinProtection
yellowfive@129 986 [32] = true, -- WarlockDemonology
yellowfive@129 987 [36] = true -- WarriorProtection
yellowfive@129 988 }
yellowfive@129 989
yellowfive@139 990 function beginEquipGearSet(setupId, passes)
yellowfive@57 991
yellowfive@139 992 local setup = getSetupById(setupId)
yellowfive@139 993
yellowfive@139 994 if not setup or not setup.Gear then
yellowfive@57 995 Amr:Print(L.GearEquipErrorEmpty)
yellowfive@57 996 return
yellowfive@57 997 end
yellowfive@124 998
yellowfive@139 999 local gear = setup.Gear
yellowfive@139 1000 local spec = setup.SpecSlot
yellowfive@139 1001
yellowfive@139 1002 -- ensure all our stored data is up to date
yellowfive@57 1003 local player = Amr:ExportCharacter()
yellowfive@129 1004 local doOhFirst = _ohFirst[player.Specs[spec]]
yellowfive@57 1005
yellowfive@124 1006 local itemsToEquip = {
yellowfive@124 1007 legendaries = {},
yellowfive@124 1008 weapons = {},
yellowfive@129 1009 mh = {},
yellowfive@129 1010 oh = {},
yellowfive@124 1011 rings = {},
yellowfive@124 1012 trinkets = {},
yellowfive@124 1013 others = {},
yellowfive@124 1014 blanks = {}
yellowfive@124 1015 }
yellowfive@57 1016 local remaining = 0
yellowfive@129 1017 local usedItems = {}
yellowfive@124 1018
yellowfive@124 1019 -- check for items that need to be equipped, do in a random order to try and defeat any unique constraint issues we might hit
yellowfive@124 1020 local slots = {}
yellowfive@124 1021 for i,s in ipairs(Amr.SlotIds) do
yellowfive@124 1022 table.insert(slots, s)
yellowfive@124 1023 end
yellowfive@124 1024 shuffle(slots)
yellowfive@124 1025
yellowfive@124 1026 for i,slotId in ipairs(slots) do
yellowfive@124 1027
yellowfive@124 1028 -- we do stuff in batches that avoids most unique conflicts
yellowfive@124 1029 local list = itemsToEquip.others
yellowfive@129 1030 if slotId == 16 then
yellowfive@129 1031 list = itemsToEquip.mh
yellowfive@129 1032 elseif slotId == 17 then
yellowfive@129 1033 list = itemsToEquip.oh
yellowfive@124 1034 elseif slotId == 11 or slotId == 12 then
yellowfive@124 1035 list = itemsToEquip.rings
yellowfive@124 1036 elseif slotId == 13 or slotId == 14 then
yellowfive@124 1037 list = itemsToEquip.trinkets
yellowfive@124 1038 end
yellowfive@124 1039
yellowfive@57 1040 local old = player.Equipped[spec][slotId]
yellowfive@57 1041 local new = gear[slotId]
yellowfive@124 1042 local prevRemaining = remaining
yellowfive@69 1043 if new then
yellowfive@124 1044 -- if the new thing is an artifact, only match the item id
yellowfive@124 1045 local newItem = Item:CreateFromItemID(new.id)
yellowfive@124 1046 local quality = newItem and newItem:GetItemQuality() or 0
yellowfive@124 1047 if quality == 6 then
yellowfive@124 1048 if not old or new.id ~= old.id then
yellowfive@124 1049 list[slotId] = new
yellowfive@129 1050 if list == itemsToEquip.mh or list == itemsToEquip.oh then
yellowfive@129 1051 itemsToEquip.weapons[slotId] = {}
yellowfive@129 1052 end
yellowfive@69 1053 remaining = remaining + 1
yellowfive@69 1054 end
yellowfive@129 1055 else
yellowfive@145 1056
yellowfive@129 1057 -- find the best matching item anywhere in the player's gear
yellowfive@129 1058 local bestItem, bestDiff = Amr:FindMatchingItem(new, player, usedItems)
yellowfive@129 1059 new = bestItem
yellowfive@129 1060
yellowfive@145 1061 local diff = countItemDifferences(old, new)
yellowfive@124 1062
yellowfive@129 1063 if diff > 0 then
yellowfive@124 1064 list[slotId] = new
yellowfive@129 1065 if list == itemsToEquip.mh or list == itemsToEquip.oh then
yellowfive@129 1066 itemsToEquip.weapons[slotId] = {}
yellowfive@129 1067 end
yellowfive@124 1068 remaining = remaining + 1
yellowfive@124 1069 end
yellowfive@124 1070 end
yellowfive@129 1071 elseif old then
yellowfive@124 1072 -- need to remove this item
yellowfive@124 1073 itemsToEquip.blanks[slotId] = {}
yellowfive@124 1074 remaining = remaining + 1
yellowfive@124 1075 end
yellowfive@124 1076
yellowfive@124 1077 if remaining > prevRemaining then
yellowfive@124 1078 -- if we need to swap this slot, see if the old item is a legendary, add a step to remove those first to avoid conflicts
yellowfive@124 1079 if old then
yellowfive@124 1080 local oldItem = Item:CreateFromItemID(old.id)
yellowfive@124 1081 if oldItem and oldItem:GetItemQuality() == 5 then
yellowfive@124 1082 itemsToEquip.legendaries[slotId] = {}
yellowfive@124 1083 end
yellowfive@57 1084 end
yellowfive@57 1085 end
yellowfive@57 1086 end
yellowfive@124 1087
yellowfive@124 1088 if remaining > 0 then
yellowfive@57 1089
yellowfive@124 1090 if passes < 5 then
yellowfive@124 1091 _pendingGearOps = {}
yellowfive@124 1092
yellowfive@129 1093 if not Amr.IsEmpty(itemsToEquip.blanks) then
yellowfive@124 1094 -- if gear set wants slots to be blank, do that first
yellowfive@124 1095 table.insert(_pendingGearOps, { items = itemsToEquip.blanks, remove = true, label = "blanks" })
yellowfive@129 1096 end
yellowfive@124 1097 if not Amr.IsEmpty(itemsToEquip.weapons) then
yellowfive@129 1098 -- change weapons first: remove both, wait, then equip each weapon one by one, waiting after each
yellowfive@129 1099 table.insert(_pendingGearOps, { items = itemsToEquip.weapons, remove = true, label = "remove weapons" })
yellowfive@129 1100 local thisWeapon = doOhFirst and itemsToEquip.oh or itemsToEquip.mh
yellowfive@129 1101 if not Amr.IsEmpty(thisWeapon) then
yellowfive@129 1102 table.insert(_pendingGearOps, { items = thisWeapon, wait = true, label = "equip weapon 1" })
yellowfive@129 1103 end
yellowfive@129 1104 thisWeapon = doOhFirst and itemsToEquip.mh or itemsToEquip.oh
yellowfive@129 1105 if not Amr.IsEmpty(thisWeapon) then
yellowfive@129 1106 table.insert(_pendingGearOps, { items = thisWeapon, wait = true, label = "equip weapon 2" })
yellowfive@129 1107 end
yellowfive@124 1108 end
yellowfive@124 1109 if not Amr.IsEmpty(itemsToEquip.legendaries) then
yellowfive@124 1110 -- remove any legendaries, wait
yellowfive@124 1111 table.insert(_pendingGearOps, { items = itemsToEquip.legendaries, remove = true, label = "remove legendaries" })
yellowfive@124 1112 end
yellowfive@124 1113 if not Amr.IsEmpty(itemsToEquip.rings) then
yellowfive@124 1114 -- remove both rings, wait, then equip new ones
yellowfive@124 1115 table.insert(_pendingGearOps, { items = itemsToEquip.rings, remove = true, label = "remove rings" })
yellowfive@124 1116 table.insert(_pendingGearOps, { items = itemsToEquip.rings, wait = true, label = "equip rings" })
yellowfive@124 1117 end
yellowfive@124 1118 if not Amr.IsEmpty(itemsToEquip.trinkets) then
yellowfive@124 1119 -- remove both trinkets, wait, then equip new ones
yellowfive@124 1120 table.insert(_pendingGearOps, { items = itemsToEquip.trinkets, remove = true, label = "remove trinkets" })
yellowfive@124 1121 table.insert(_pendingGearOps, { items = itemsToEquip.trinkets, wait = true, label = "equip trinkets" })
yellowfive@124 1122 end
yellowfive@124 1123 if not Amr.IsEmpty(itemsToEquip.others) then
yellowfive@124 1124 -- equip all other items, wait for completion
yellowfive@124 1125 table.insert(_pendingGearOps, { items = itemsToEquip.others, wait = true, label = "equip others" })
yellowfive@124 1126 end
yellowfive@124 1127
yellowfive@124 1128 -- make the last operation wait no matter what, before this gets called again to check if everything succeeded
yellowfive@124 1129 _pendingGearOps[#_pendingGearOps].wait = true
yellowfive@124 1130
yellowfive@124 1131 if not _gearOpWaiting then
yellowfive@124 1132 _gearOpWaiting = { inventory = {} }
yellowfive@124 1133 end
yellowfive@124 1134
yellowfive@124 1135 _gearOpPasses = passes
yellowfive@124 1136 _currentGearOp = _pendingGearOps[1]
yellowfive@139 1137 initializeGearOp(_currentGearOp, setupId, 1)
yellowfive@124 1138
yellowfive@124 1139 processCurrentGearOp()
yellowfive@124 1140 else
yellowfive@124 1141 -- TODO: print message that gear set couldn't be equipped
yellowfive@89 1142 end
yellowfive@57 1143
yellowfive@57 1144 else
yellowfive@89 1145 onEquipGearSetComplete()
yellowfive@124 1146 end
yellowfive@57 1147 end
yellowfive@57 1148
yellowfive@57 1149 local function onActiveTalentGroupChanged()
yellowfive@81 1150
yellowfive@57 1151 local auto = Amr.db.profile.options.autoGear
yellowfive@81 1152 local currentSpec = GetSpecialization()
yellowfive@129 1153 local waitingSpec = _waitingForSpec
yellowfive@129 1154 _waitingForSpec = 0
yellowfive@57 1155
yellowfive@139 1156 -- when spec changes, change active setup to first one for this spec (does nothing if they have no setups for this spec)
yellowfive@139 1157 if _activeSetupId then
yellowfive@139 1158 local currentSetup = getSetupById(_activeSetupId)
yellowfive@139 1159 if currentSetup.SpecSlot ~= currentSpec then
yellowfive@139 1160 Amr:PickFirstSetupForSpec()
yellowfive@139 1161 end
yellowfive@139 1162 end
yellowfive@139 1163
yellowfive@129 1164 if currentSpec == waitingSpec or auto then
yellowfive@129 1165 -- spec is what we want, now equip the gear but after a short delay because the game auto-swaps artifact weapons
yellowfive@129 1166 Amr.Wait(2, function()
yellowfive@139 1167 beginEquipGearSet(_activeSetupId, 0)
yellowfive@129 1168 end)
yellowfive@57 1169 end
yellowfive@57 1170 end
yellowfive@57 1171
yellowfive@81 1172 -- activate the specified spec and then equip the saved gear set
yellowfive@139 1173 function Amr:EquipGearSet(setupIndex)
yellowfive@57 1174
yellowfive@139 1175 -- if no argument, then cycle
yellowfive@139 1176 if not setupIndex then
yellowfive@139 1177 if not _activeSetupId then
yellowfive@139 1178 Amr:PickFirstSetupForSpec()
yellowfive@139 1179 end
yellowfive@139 1180 for i,setup in ipairs(Amr.db.char.GearSetups) do
yellowfive@139 1181 if setup.Id == _activeSetupId then
yellowfive@139 1182 setupIndex = i
yellowfive@139 1183 break
yellowfive@139 1184 end
yellowfive@139 1185 end
yellowfive@139 1186 if not setupIndex then
yellowfive@139 1187 setupIndex = 1
yellowfive@139 1188 else
yellowfive@139 1189 setupIndex = setupIndex + 1
yellowfive@139 1190 end
yellowfive@57 1191 end
yellowfive@81 1192
yellowfive@139 1193 setupIndex = tonumber(setupIndex)
yellowfive@81 1194
yellowfive@139 1195 if setupIndex > #Amr.db.char.GearSetups then
yellowfive@139 1196 setupIndex = 1
yellowfive@139 1197 end
yellowfive@139 1198
yellowfive@57 1199 if UnitAffectingCombat("player") then
yellowfive@57 1200 Amr:Print(L.GearEquipErrorCombat)
yellowfive@57 1201 return
yellowfive@57 1202 end
yellowfive@57 1203
yellowfive@139 1204 _activeSetupId = Amr.db.char.GearSetups[setupIndex].Id
yellowfive@139 1205 Amr:RefreshGearDisplay()
yellowfive@139 1206
yellowfive@139 1207 local setup = Amr.db.char.GearSetups[setupIndex]
yellowfive@81 1208 local currentSpec = GetSpecialization()
yellowfive@139 1209 if currentSpec ~= setup.SpecSlot then
yellowfive@139 1210 _waitingForSpec = setup.SpecSlot
yellowfive@139 1211 SetSpecialization(setup.SpecSlot)
yellowfive@57 1212 else
yellowfive@129 1213 -- spec is what we want, now equip the gear
yellowfive@139 1214 beginEquipGearSet(_activeSetupId, 0)
yellowfive@57 1215 end
yellowfive@57 1216 end
yellowfive@57 1217
yellowfive@124 1218 -- moves any gear in bags to the bank if not part of a gear set
yellowfive@57 1219 function Amr:CleanBags()
yellowfive@57 1220 -- TODO: implement
yellowfive@57 1221 end
yellowfive@57 1222
yellowfive@81 1223 --[[
yellowfive@81 1224 local function testfunc(message)
yellowfive@81 1225 print(strsub(message, 13))
yellowfive@81 1226 end
yellowfive@81 1227 ]]
yellowfive@81 1228
yellowfive@57 1229 function Amr:InitializeGear()
yellowfive@87 1230 Amr:AddEventHandler("ACTIVE_TALENT_GROUP_CHANGED", onActiveTalentGroupChanged)
yellowfive@57 1231
yellowfive@81 1232 --Amr:AddEventHandler("CHAT_MSG_CHANNEL", testfunc)
yellowfive@81 1233
yellowfive@57 1234 Amr:AddEventHandler("UNIT_INVENTORY_CHANGED", function(unitID)
yellowfive@57 1235 if unitID and unitID ~= "player" then return end
yellowfive@124 1236
yellowfive@124 1237 -- don't update during a gear operation, wait until it is totally finished
yellowfive@124 1238 if _pendingGearOps then return end
yellowfive@124 1239
yellowfive@139 1240 Amr:RefreshGearDisplay()
yellowfive@57 1241 end)
yellowfive@57 1242
yellowfive@124 1243 Amr:AddEventHandler("ITEM_UNLOCKED", handleItemUnlocked)
yellowfive@122 1244 end