Mercurial > wow > askmrrobot

annotate ui/GearComparisonTab.lua @ 17:e77e01abce98

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Warlords of Draenor pre-patch
author Adam tegen <adam.tegen@gmail.com>
date Mon, 13 Oct 2014 21:28:32 -0500
parents
children 44c285acfff0
rev   line source
adam@17 1 local _, AskMrRobot = ...
adam@17 2 local L = AskMrRobot.L;
adam@17 3
adam@17 4 -- initialize the GearComparisonTab class
adam@17 5 AskMrRobot.GearComparisonTab = AskMrRobot.inheritsFrom(AskMrRobot.Frame)
adam@17 6
adam@17 7 -- stores the results of the last gear comparison
adam@17 8 AskMrRobot.ComparisonResult = {
adam@17 9 items = {},
adam@17 10 gems = {},
adam@17 11 enchants = {}
adam@17 12 }
adam@17 13
adam@17 14 local function createTabButton(tab, text, index)
adam@17 15 local button = CreateFrame("Button", "GearComparisonTab" .. index, tab, "TabButtonTemplate")
adam@17 16 if index == 1 then
adam@17 17 button:SetPoint("TOPLEFT")
adam@17 18 else
adam@17 19 button:SetPoint("LEFT", "GearComparisonTab" .. (index - 1), "RIGHT")
adam@17 20 end
adam@17 21
adam@17 22 button:SetText(text)
adam@17 23 button:SetWidth(50)
adam@17 24 button:SetHeight(20)
adam@17 25 button:SetID(index)
adam@17 26 button:SetScript("OnClick", AskMrRobot.GearComparisonTab.tabButtonClick)
adam@17 27 return button
adam@17 28 end
adam@17 29
adam@17 30 function AskMrRobot.GearComparisonTab:tabButtonClick()
adam@17 31 local tab = self:GetParent()
adam@17 32 local id = self:GetID()
adam@17 33 PanelTemplates_SetTab(tab, id)
adam@17 34 for i = 1, #tab.tabs do
adam@17 35 local t = tab.tabs[i]
adam@17 36 if t:GetID() == id then
adam@17 37 t:Show()
adam@17 38 else
adam@17 39 t:Hide()
adam@17 40 end
adam@17 41 end
adam@17 42 end
adam@17 43
adam@17 44 function AskMrRobot.GearComparisonTab:new(parent)
adam@17 45
adam@17 46 local tab = AskMrRobot.Frame:new("GearComparison", parent)
adam@17 47 setmetatable(tab, { __index = AskMrRobot.GearComparisonTab })
adam@17 48 tab:SetPoint("TOPLEFT")
adam@17 49 tab:SetPoint("BOTTOMRIGHT")
adam@17 50 tab:Hide()
adam@17 51
adam@17 52 tab.initialized = false
adam@17 53
adam@17 54 tab.tabButtons = {
adam@17 55 createTabButton(tab, L.AMR_UI_BUTTON_IMPORT, 1),
adam@17 56 createTabButton(tab, L.AMR_UI_BUTTON_SUMMARY, 2),
adam@17 57 createTabButton(tab, L.AMR_UI_BUTTON_GEMS, 3),
adam@17 58 createTabButton(tab, L.AMR_UI_BUTTON_ENCHANTS, 4),
adam@17 59 createTabButton(tab, L.AMR_UI_BUTTON_SHOPPING_LIST, 5)
adam@17 60 }
adam@17 61
adam@17 62 PanelTemplates_SetNumTabs(tab, 5)
adam@17 63 PanelTemplates_SetTab(tab, 1)
adam@17 64 for i = 1, #tab.tabButtons do
adam@17 65 PanelTemplates_TabResize(tab.tabButtons[i], 0)
adam@17 66 end
adam@17 67
adam@17 68 PanelTemplates_DisableTab(tab, 2)
adam@17 69 PanelTemplates_DisableTab(tab, 3)
adam@17 70 PanelTemplates_DisableTab(tab, 4)
adam@17 71 PanelTemplates_DisableTab(tab, 5)
adam@17 72
adam@17 73 -- create the import tab
adam@17 74 tab.importTab = AskMrRobot.ImportTab:new(tab)
adam@17 75 tab.importTab:SetID(1)
adam@17 76
adam@17 77 -- set the tab left of the tab
adam@17 78 tab.importTab:SetPoint("TOPLEFT", tab, "TOPLEFT", 0, -30)
adam@17 79
adam@17 80 tab.summaryTab = AskMrRobot.SummaryTab:new(tab)
adam@17 81 tab.summaryTab:SetID(2)
adam@17 82 tab.summaryTab:SetPoint("TOPLEFT", tab, "TOPLEFT", 0, -30)
adam@17 83
adam@17 84 tab.gemTab = AskMrRobot.GemTab:new(tab)
adam@17 85 tab.gemTab:SetID(3)
adam@17 86 tab.gemTab:SetPoint("TOPLEFT", tab, "TOPLEFT", 0, -30)
adam@17 87
adam@17 88 tab.enchantTab = AskMrRobot.EnchantTab:new(tab)
adam@17 89 tab.enchantTab:SetID(4)
adam@17 90 tab.enchantTab:SetPoint("TOPLEFT", tab, "TOPLEFT", 0, -30)
adam@17 91
adam@17 92 tab.shoppingTab = AskMrRobot.ShoppingListTab:new(tab)
adam@17 93 tab.shoppingTab:SetID(5)
adam@17 94 tab.shoppingTab:SetPoint("TOPLEFT", tab, "TOPLEFT", 0, -30)
adam@17 95
adam@17 96 tab.tabs = {tab.importTab, tab.summaryTab, tab.gemTab, tab.enchantTab, tab.shoppingTab}
adam@17 97
adam@17 98 -- show the first tab
adam@17 99 tab.importTab:Show()
adam@17 100
adam@17 101 -- setup the import button to run the import
adam@17 102 tab.importTab.button:SetScript("OnClick", function()
adam@17 103 tab:Import()
adam@17 104 tab.tabButtonClick(tab.tabButtons[2])
adam@17 105 end)
adam@17 106
adam@17 107 tab:SetScript("OnShow", AskMrRobot.GearComparisonTab.OnShow)
adam@17 108
adam@17 109 --tab:RegisterEvent("ITEM_PUSH")
adam@17 110 --tab:RegisterEvent("DELETE_ITEM_CONFIRM")
adam@17 111 tab:RegisterEvent("SOCKET_INFO_CLOSE")
adam@17 112 tab:RegisterEvent("SOCKET_INFO_UPDATE")
adam@17 113 tab:RegisterEvent("PLAYER_SPECIALIZATION_CHANGED")
adam@17 114 tab:RegisterEvent("BAG_UPDATE")
adam@17 115
adam@17 116 tab:SetScript("OnEvent", AskMrRobot.GearComparisonTab.OnEvent)
adam@17 117
adam@17 118 return tab
adam@17 119 end
adam@17 120
adam@17 121 function AskMrRobot.GearComparisonTab:On_SOCKET_INFO_CLOSE()
adam@17 122 self:Import()
adam@17 123 end
adam@17 124
adam@17 125 function AskMrRobot.GearComparisonTab:On_SOCKET_INFO_UPDATE()
adam@17 126 self:Import()
adam@17 127 end
adam@17 128
adam@17 129 function AskMrRobot.GearComparisonTab:On_PLAYER_SPECIALIZATION_CHANGED()
adam@17 130 self:Import()
adam@17 131 end
adam@17 132
adam@17 133 function AskMrRobot.GearComparisonTab:On_BAG_UPDATE()
adam@17 134 self:Import()
adam@17 135 end
adam@17 136
adam@17 137 function AskMrRobot.GearComparisonTab:On_ITEM_PUSH()
adam@17 138 self:Import()
adam@17 139 end
adam@17 140
adam@17 141 function AskMrRobot.GearComparisonTab:On_DELETE_ITEM_CONFIRM()
adam@17 142 self:Import()
adam@17 143 end
adam@17 144
adam@17 145 function AskMrRobot.GearComparisonTab:OnShow()
adam@17 146 if not self.initialized then
adam@17 147 self.initialized = true
adam@17 148
adam@17 149 -- on first show, load the last import
adam@17 150 if AmrDb.LastCharacterImport and AmrDb.LastCharacterImport ~= "" then
adam@17 151 self.importTab:SetImportText(AmrDb.LastCharacterImport)
adam@17 152 self:Import()
adam@17 153 else
adam@17 154 self:Update()
adam@17 155 end
adam@17 156 else
adam@17 157 self:Update()
adam@17 158 end
adam@17 159 end
adam@17 160
adam@17 161 function AskMrRobot.GearComparisonTab:Import()
adam@17 162
adam@17 163 -- example string
adam@17 164 -- $2;Brill (EU);Yellowfive;Twisted Legion;11;2;90;7:600,9:600;1;s1;34;2123320;68164,47782,7833;q1;99195s7u493x4647y0c22e4823;1s3u0x0y0c11e90;6s10u0x0y0c11e-481;1s1u0x383y-383c41;3047s15u12x0c1e-8;3109s5u-12x0y0z0c112e445;55s8u12x0c1e-442;5371s16u0b450x0c1e18;1691s13u-14b-1;238s11u0b0x-60c3;28s2u0b0;21s6u2b0x60y0z0c130;29s14u0b0;1s9u-2b0e-30;62s12u0b0x0c1;95s17u2b0x0c1$g\4647\76697\76631,76697,83146\20 _CriticalStrike_@g\5030\95344\\Indomitable@g\4587\76636\76570,76636,83144\20 _CriticalStrike_@e\4823\83765\122388\19 _Strength_, 11 _CriticalStrike_\72163=1,76061=1@e\4913\87585\113047\20 _Strength_, 5 _CriticalStrike_\39354=1,79254=3@e\4432\74721\104419\11 _Strength_\74249=3,74250=1,74247=1@e\4424\74713\104404\12 _CriticalStrike_\74250=1@e\4869\85559\124091\9 _Stamina_\72120=4@e\4427\74716\104408\11 _CriticalStrike_\74249=2,74250=1@e\4445\74727\104440\Colossus\74247=3@e\4415\74704\104390\18 _Strength_\74248=3
adam@17 165
adam@17 166 local err = AskMrRobot.ImportCharacter(self.importTab:GetImportText())
adam@17 167 -- goto the summary tab
adam@17 168 self.summaryTab:showImportError(err)
adam@17 169 PanelTemplates_EnableTab(self, 2)
adam@17 170 if err then
adam@17 171 PanelTemplates_DisableTab(self, 3)
adam@17 172 PanelTemplates_DisableTab(self, 4)
adam@17 173 PanelTemplates_DisableTab(self, 5)
adam@17 174 else
adam@17 175 PanelTemplates_EnableTab(self, 3)
adam@17 176 PanelTemplates_EnableTab(self, 4)
adam@17 177 PanelTemplates_EnableTab(self, 5)
adam@17 178 self:Update()
adam@17 179 end
adam@17 180 end
adam@17 181
adam@17 182 -- update the panel and state
adam@17 183 function AskMrRobot.GearComparisonTab:Update()
adam@17 184 -- update the comparison
adam@17 185 if self.summaryTab then
adam@17 186 AskMrRobot.GearComparisonTab.Compare()
adam@17 187 self.summaryTab:showBadItems()
adam@17 188 self.gemTab:Update()
adam@17 189 self.enchantTab:Update()
adam@17 190 self.shoppingTab:Update()
adam@17 191 end
adam@17 192 end
adam@17 193
adam@17 194
adam@17 195 -- Helper for checking for swapped items, returns a number indicating how different two items are (0 means the same, higher means more different)
adam@17 196 local function countItemDifferences(item1, item2)
adam@17 197 if item1 == nil and item2 == nil then return 0 end
adam@17 198
adam@17 199 -- different items (id + bonus ids + suffix, constitutes a different physical drop)
adam@17 200 if AskMrRobot.getItemUniqueId(item1, true) ~= AskMrRobot.getItemUniqueId(item2, true) then
adam@17 201 if item1 == nil or item2 == nil then return 1000 end
adam@17 202 -- do a check to deal with SoO item variants, see if we have duplicate ID information
adam@17 203 local info = AskMrRobot.ExtraItemData[item2.id]
adam@17 204 if info == nil then
adam@17 205 info = AskMrRobot.ExtraItemData[item1.id]
adam@17 206 if info == nil or info.duplicateId ~= item2.id then
adam@17 207 return 1000
adam@17 208 end
adam@17 209 elseif info.duplicateId ~= item1.id then
adam@17 210 return 1000
adam@17 211 end
adam@17 212 end
adam@17 213
adam@17 214 -- different upgrade levels of the same item (only for older gear, player has control over upgrade level)
adam@17 215 if item1.upgradeId ~= item2.upgradeId then
adam@17 216 return 100
adam@17 217 end
adam@17 218
adam@17 219 -- different gems
adam@17 220 local gemDiffs = 0
adam@17 221 for i = 1, 3 do
adam@17 222 if item1.gemIds[i] ~= item2.gemIds[i] then
adam@17 223 gemDiffs = gemDiffs + 1
adam@17 224 end
adam@17 225 end
adam@17 226
adam@17 227 local enchantDiff = 0
adam@17 228 if item1.enchantId ~= item2.enchantId then
adam@17 229 enchantDiff = 1
adam@17 230 end
adam@17 231
adam@17 232 return gemDiffs + enchantDiff
adam@17 233 end
adam@17 234
adam@17 235 function AskMrRobot.GearComparisonTab:OnEvent(event, ...)
adam@17 236 local handler = self["On_" .. event]
adam@17 237 if handler then
adam@17 238 handler(self, ...)
adam@17 239 end
adam@17 240 end
adam@17 241
adam@17 242 -- modifies data2 such that differences between data1 and data2 in the two specified slots is the smallest
adam@17 243 local function checkSwappedItems(data1, data2, slot1, slot2)
adam@17 244 local diff = countItemDifferences(data1[slot1], data2[slot1]) + countItemDifferences(data1[slot2], data2[slot2])
adam@17 245 local swappedDiff
adam@17 246 if diff > 0 then
adam@17 247 swappedDiff = countItemDifferences(data1[slot1], data2[slot2]) + countItemDifferences(data1[slot2], data2[slot1])
adam@17 248 if swappedDiff < diff then
adam@17 249 local temp = data2[slot1]
adam@17 250 data2[slot1] = data2[slot2]
adam@17 251 data2[slot2] = temp
adam@17 252 end
adam@17 253 end
adam@17 254 end
adam@17 255
adam@17 256 -- compare the last import data to the player's current state
adam@17 257 function AskMrRobot.GearComparisonTab.Compare()
adam@17 258 if not AskMrRobot.ImportData then return end
adam@17 259
adam@17 260 AskMrRobot.SaveAll()
adam@17 261
adam@17 262 -- first parse the player's equipped gear into item objects
adam@17 263 local equipped = {}
adam@17 264 for k, v in pairs(AmrDb.Equipped[AmrDb.ActiveSpec]) do
adam@17 265 equipped[k] = AskMrRobot.parseItemLink(v)
adam@17 266 end
adam@17 267
adam@17 268 -- swap finger/trinket in AskMrRobot.ImportData such that the number of differences is the smallest
adam@17 269 checkSwappedItems(equipped, AskMrRobot.ImportData, INVSLOT_FINGER1, INVSLOT_FINGER2)
adam@17 270 checkSwappedItems(equipped, AskMrRobot.ImportData, INVSLOT_TRINKET1, INVSLOT_TRINKET2)
adam@17 271
adam@17 272 -- clear previous comparison result
adam@17 273 AskMrRobot.ComparisonResult = {
adam@17 274 items = {},
adam@17 275 gems = {},
adam@17 276 enchants = {}
adam@17 277 }
adam@17 278
adam@17 279 local result = {
adam@17 280 items = {},
adam@17 281 gems = {},
adam@17 282 enchants = {}
adam@17 283 }
adam@17 284
adam@17 285 -- determine specific differences
adam@17 286 for i,slotId in ipairs(AskMrRobot.slotIds) do
adam@17 287 local itemEquipped = equipped[slotId]
adam@17 288 local itemImported = AskMrRobot.ImportData[slotId]
adam@17 289
adam@17 290 local itemsDifferent = AskMrRobot.getItemUniqueId(itemEquipped) ~= AskMrRobot.getItemUniqueId(itemImported)
adam@17 291 if itemsDifferent and itemEquipped ~= nil and itemImported ~= nil then
adam@17 292 -- do an extra check for old versions of SoO items, our server code always converts to new, equivalent version, but need to check backwards for the addon
adam@17 293 local info = AskMrRobot.ExtraItemData[itemImported.id]
adam@17 294 if info and info.duplicateId == itemEquipped.id then
adam@17 295 itemsDifferent = false
adam@17 296 end
adam@17 297 end
adam@17 298
adam@17 299 if itemsDifferent then
adam@17 300 -- the items are different
adam@17 301 local needsUpgrade = false
adam@17 302 if itemEquipped and itemImported and itemEquipped.id == itemImported.id and itemImported.upgradeId > itemEquipped.upgradeId then
adam@17 303 needsUpgrade = true
adam@17 304 end
adam@17 305 result.items[slotId] = {
adam@17 306 current = itemEquipped,
adam@17 307 optimized = itemImported,
adam@17 308 needsUpgrade = needsUpgrade
adam@17 309 }
adam@17 310 elseif itemEquipped then
adam@17 311 if AskMrRobot.ExtraItemData[itemEquipped.id] and AskMrRobot.ExtraItemData[itemEquipped.id].socketColors then
adam@17 312 -- items are same, check for gem/enchant differences
adam@17 313 -- NOTE: we used to do a bunch of fancy gem checks, but we can ditch all that logic b/c WoD gems are much simpler (no socket bonuses, gem/socket colors to worry about)
adam@17 314 local hasBadGems = false
adam@17 315 for g = 1, #AskMrRobot.ExtraItemData[itemEquipped.id].socketColors do
adam@17 316 if not AskMrRobot.AreGemsCompatible(itemEquipped.gemIds[g], itemImported.gemIds[g]) then
adam@17 317 hasBadGems = true
adam@17 318 break
adam@17 319 end
adam@17 320 end
adam@17 321
adam@17 322 if hasBadGems then
adam@17 323 result.gems[slotId] = {
adam@17 324 current = {},
adam@17 325 optimized = {}
adam@17 326 }
adam@17 327
adam@17 328 for g = 1, #AskMrRobot.ExtraItemData[itemEquipped.id].socketColors do
adam@17 329 result.gems[slotId].current[g] = itemEquipped.gemIds[g]
adam@17 330 result.gems[slotId].optimized[g] = itemImported.gemIds[g]
adam@17 331 end
adam@17 332 end
adam@17 333 end
adam@17 334
adam@17 335 if itemEquipped.enchantId ~= itemImported.enchantId then
adam@17 336 result.enchants[slotId] = {
adam@17 337 current = itemEquipped.enchantId,
adam@17 338 optimized = itemImported.enchantId
adam@17 339 }
adam@17 340 end
adam@17 341 end
adam@17 342 end
adam@17 343
adam@17 344 -- only set the new result if it is completely successful
adam@17 345 AskMrRobot.ComparisonResult = result
adam@17 346 end
adam@17 347
adam@17 348 -- checks our extra gem information to see if the two gems are functionally equivalent
adam@17 349 function AskMrRobot.AreGemsCompatible(gemId1, gemId2)
adam@17 350 if gemId1 == gemId2 then return true end
adam@17 351
adam@17 352 -- see if we have extra gem information
adam@17 353 local extraInfo = AskMrRobot.ExtraGemData[gemId1]
adam@17 354 if not extraInfo then
adam@17 355 extraInfo = AskMrRobot.ExtraGemData[gemId2]
adam@17 356 end
adam@17 357 if extraInfo == nil or extraInfo.identicalGroup == nil then return false end
adam@17 358
adam@17 359 -- if identicalGroup contains both gem ids, they are equivalent
adam@17 360 return extraInfo.identicalGroup[gemId1] and extraInfo.identicalGroup[gemId2]
adam@17 361 end