wow/askmrrobot: Shopping.lua comparison

comparison Shopping.lua @ 89:6bbe64d587b4 v42

Improved artifact/relic reading, gear display. Now creates equipment manager sets when you use button or command to equip a set.

author	yellowfive
date	Sun, 18 Sep 2016 21:24:08 -0700
parents	0515882856f1
children	f1da233629be

comparison

equal deleted inserted replaced

-:b3ff336fad77
+:6bbe64d587b4
 local _chk1
 local _chk2
 local _chk3
 local _chk4
 local _isAhOpen = false
+local function incrementTableItem(tbl, key, inc)
+	tbl[key] = tbl[key] and tbl[key] + inc or inc
+end
 local function onShopFrameClose(widget)
 	AceGUI:Release(widget)
 	_frameShop = nil
 	_cboPlayers = nil
 		panel:SetWidth(w)
 		panel:SetHeight(30)
 		scroll:AddChild(panel)
 		lbl = AceGUI:Create("AmrUiLabel")
-		lbl:SetWidth(35)
+		lbl:SetWidth(40)
 		lbl:SetWordWrap(false)
 		lbl:SetFont(Amr.CreateFont("Bold", 20, Amr.Colors.White))
 		lbl:SetText(count .. "x")
 		lbl:SetPoint("LEFT", panel.content, "LEFT")
 		panel:AddChild(lbl)
 		end, { itemIcon = icon, itemText = btn })
 	end
 end
+-- get the number of a specified gem/enchant/material that the player currently owns
+local function getOwnedCount(itemId)
+	local ret = 0
+	local list = Amr.db.char.BagItemsAndCounts
+	if list and list[itemId] then
+		ret = ret + list[itemId]
+	end
+	list = Amr.db.char.BankItemsAndCounts
+	if list and list[itemId] then
+		ret = ret + list[itemId]
+	end
+	return ret
+end
+local function removeOwned(list, owned)
+	for itemId, count in pairs(list) do
+		-- load up how many of an item we have
+		if not owned.loaded[itemId] then
+			owned.counts[itemId] = getOwnedCount(itemId)
+			owned.loaded[itemId] = true
+		end
+		-- see how many we can remove from the required count
+		local used = math.min(owned.counts[itemId], count)
+		-- update owned count so we can't double-use something
+		owned.counts[itemId] = owned.counts[itemId] - used;
+		-- reduce the requirement, removing entirely if we have it completely covered
+		list[itemId] = list[itemId] - used;
+		if list[itemId] == 0 then
+			list[itemId] = nil
+		end
+	end
+end
 function Amr:RefreshShoppingUi()
-	_chk1:SetVisible(false)
+	local posToCheck = { _chk1, _chk2, _chk3, _chk4 }
-	_chk2:SetVisible(false)
+	local chk
-	_chk3:SetVisible(false)
-	_chk4:SetVisible(false)
+	-- reset spec checkboxes
+	for specPos = 1,4 do
-	_chk1:SetChecked(false)
+		chk = posToCheck[specPos]
-	_chk2:SetChecked(false)
+		chk:SetVisible(false)
-	_chk3:SetChecked(false)
+		chk:SetChecked(false)
-	_chk4:SetChecked(false)
+	end
 	-- clear out any previous data
 	_panelContent:ReleaseChildren()
 	local data = Amr.db.global.Shopping[_selectedPlayer]
 	if not data then
 		lbl:SetText(L.ShopEmpty)
 		lbl:SetJustifyH("CENTER")
 		lbl:SetPoint("TOP", _panelContent.content, "TOP", 0, -30)
 		_panelContent:AddChild(lbl)
 	else
-		-- set labels on checkboxes
+		local allStuff = { gems = {}, enchants = {}, materials = {} }
-		if data.specs[1] and data.specs[1] ~= 0 then
+		local hasStuff = false
-			_chk1:SetText(L.SpecsShort[data.specs[1]])
+		local visited = {}
-			_chk1:SetVisible(true)
-			_chk1:SetChecked(_specs[1])
+		for specPos = 1,4 do
-		end
+			-- set labels on checkboxes
+			if data.specs[specPos] and data.specs[specPos] ~= 0 then
-		if data.specs[2] and data.specs[2] ~= 0 then
+				chk = posToCheck[specPos]
-			_chk2:SetText(L.SpecsShort[data.specs[2]])
+				chk:SetText(L.SpecsShort[data.specs[specPos]])
-			_chk2:SetVisible(true)
+				chk:SetVisible(true)
-			_chk2:SetChecked(_specs[2])
+				chk:SetChecked(_specs[specPos])
-		end
+				-- gather up all stuff for checked specs
-		if data.specs[3] and data.specs[3] ~= 0 then
+				if _specs[specPos] then
-			_chk3:SetText(L.SpecsShort[data.specs[3]])
+					hasStuff = true
-			_chk3:SetVisible(true)
-			_chk3:SetChecked(_specs[3])
+					for inventoryId, stuff in pairs(data.stuff[specPos]) do
-		end
+						if not visited[inventoryId] then
+							if stuff.gems then
-		if data.specs[4] and data.spes[4] ~= 0 then
+								for itemId, count in pairs(stuff.gems) do
-			_chk4:SetText(L.SpecsShort[data.specs[4]])
+									incrementTableItem(allStuff.gems, itemId, count)
-			_chk4:SetVisible(true)
+								end
-			_chk4:SetChecked(_specs[4])
+							end
-		end
+							if stuff.enchants then
-		local spec = 0
+								for itemId, count in pairs(stuff.enchants) do
-		if not _specs[1] and not _specs[2] and not _specs[3] and not _specs[4] then
+									incrementTableItem(allStuff.enchants, itemId, count)
-			-- all unchecked, show nothing
+								end
-		else
+							end
-			-- both is 0, otherwise the one that is selected
-			if not _specs[1] or not _specs[2] then
+							if stuff.materials then
-				spec = _specs[1] and 1 or 2
+								for itemId, count in pairs(stuff.materials) do
+									incrementTableItem(allStuff.materials, itemId, count)
+								end
+							end
+							-- make sure not to count the same physical item twice
+							if inventoryId ~= -1 then
+								visited[inventoryId] = true
+							end
+						end
+					end
+				end
 			end
+		end
+		if hasStuff then
+			-- remove what we already own
+			local owned = { counts = {}, loaded = {} }
+			removeOwned(allStuff.gems, owned)
+			removeOwned(allStuff.enchants, owned)
+			removeOwned(allStuff.materials, owned)
 			_panelContent:SetLayout("Fill")
 			local scroll = AceGUI:Create("AmrUiScrollFrame")
 			scroll:SetLayout("List")
 			_panelContent:AddChild(scroll)
-			renderShopSection(data.gems[spec], scroll, L.ShopHeaderGems)
+			renderShopSection(allStuff.gems, scroll, L.ShopHeaderGems)
-			renderShopSection(data.enchants[spec], scroll, L.ShopHeaderEnchants)
+			renderShopSection(allStuff.enchants, scroll, L.ShopHeaderEnchants)
-			renderShopSection(data.materials[spec], scroll, L.ShopHeaderMaterials)
+			renderShopSection(allStuff.materials, scroll, L.ShopHeaderMaterials)
 		end
 	end
 end
--- compare gear to everything the player owns, and return the minimum gems/enchants/materials needed to optimize
+-- compare gear to everything the player owns, and return the minimum gems/enchants/materials needed to optimize, grouped by inventory ID so that we can combine multiple specs without double-counting
 local function getShoppingData(player, gear, spec)
-	local ret = {
+	local ret = {}
-		gems = {},
-		enchants = {},
-		materials = {}
-	}
 	-- used to prevent considering the same item twice
 	local usedItems = {}
 	for slotId, optimalItem in pairs(gear) do
 		local matchItemLink, matchItem = Amr:FindMatchingItem(optimalItem, player, usedItems)
 		local itemInfo = Amr.db.char.ExtraItemData[spec][optimalItem.id]
+		local inventoryId = optimalItem.inventoryId or -1
 		-- find gem/enchant differences on the best-matching item
-		-- gems
+		-- gems, but skip artifact relics (will have relicBonusIds set)
-		if itemInfo and itemInfo.socketColors then
+		if not optimalItem.relicBonusIds and itemInfo and itemInfo.socketColors then
 			for i = 1, #itemInfo.socketColors do
 				local g = optimalItem.gemIds[i]
 				local isGemEquipped = g ~= 0 and matchItem and matchItem.gemIds and matchItem.gemIds[i] == g
 				if not isGemEquipped then
-					ret.gems[g] = ret.gems[g] and ret.gems[g] + 1 or 1
+					if not ret[inventoryId] then
+						ret[inventoryId] = { gems = {}, enchants = {}, materials = {} }
+					end
+					incrementTableItem(ret[inventoryId].gems, g, 1)
 				end
 			end
 		end
 		-- enchant
 		if optimalItem.enchantId and optimalItem.enchantId ~= 0 then
 			local e = optimalItem.enchantId
 			local isEnchantEquipped = matchItem and matchItem.enchantId and matchItem.enchantId == e
 			if not isEnchantEquipped then
-				ret.enchants[e] = ret.enchants[e] and ret.enchants[e] + 1 or 1
+				-- enchant info, look in all spec extra info cache
+				local enchInfo = nil
+				for specPos = 1,4 do
+					if Amr.db.char.ExtraEnchantData[specPos] then
+						enchInfo = Amr.db.char.ExtraEnchantData[specPos][e]
+						if enchInfo then break end
+					end
+				end
+				if enchInfo then
+					if not ret[inventoryId] then
+						ret[inventoryId] = { gems = {}, enchants = {}, materials = {} }
+					end
+					incrementTableItem(ret[inventoryId].enchants, enchInfo.itemId, 1)
+					if enchInfo.materials then
+						for k, v in pairs(enchInfo.materials) do
+							incrementTableItem(ret[inventoryId].materials, k, v)
+						end
+					end
+				end
 			end
 		end
-	end
-	return ret
-end
--- get the number of a specified item that the player currently owns
-local function getOwnedCount(itemId)
-	local ret = 0
-	local list = Amr.db.char.BagItemsAndCounts
-	if list and list[itemId] then
-		ret = ret + list[itemId]
-	end
-	list = Amr.db.char.BankItemsAndCounts
-	if list and list[itemId] then
-		ret = ret + list[itemId]
 	end
 	return ret
 end
 -- look at both gear sets and find stuff that a player needs to acquire to gem/enchant their gear
 function Amr:UpdateShoppingData(player)
-	-- TODO: re-enable shopping list when Legion comes out
-	do return end
-	-- 0 is combination of all specs
 	local required = {
-		gems = {
+		stuff = {},
-			[0] = {},
-			[1] = {},
-			[2] = {},
-			[3] = {},
-			[4] = {}
-		},
-		enchants = {
-			[0] = {},
-			[1] = {},
-			[2] = {},
-			[3] = {},
-			[4] = {}
-		},
-		materials = {
-			[0] = {},
-			[1] = {},
-			[2] = {},
-			[3] = {},
-			[4] = {}
-		},
 		specs = player.Specs
 	}
 	local enchantItemIdToId = {}
 	for spec, gear in pairs(Amr.db.char.GearSets) do
-		local obj = getShoppingData(player, gear, spec)
+		required.stuff[spec] = getShoppingData(player, gear, spec)
-		for k, v in pairs(obj.gems) do
-			local gemInfo = Amr.db.char.ExtraGemData[spec][k]
-			if gemInfo then
-				local prev = required.gems[spec][gemInfo.id]
-				required.gems[spec][gemInfo.id] = prev and prev + v or v
-				prev = required.gems[0][gemInfo.id]
-				required.gems[0][gemInfo.id] = prev and prev + v or v
-			end
-		end
-		for k, v in pairs(obj.enchants) do
-			local enchInfo = Amr.db.char.ExtraEnchantData[spec][k]
-			if enchInfo then
-				enchantItemIdToId[enchInfo.itemId] = k
-				local prev = required.enchants[spec][enchInfo.itemId]
-				required.enchants[spec][enchInfo.itemId] = prev and prev + v or v
-				prev = required.enchants[0][enchInfo.itemId]
-				required.enchants[0][enchInfo.itemId] = prev and prev + v or v
-			end
-		end
-	end
-	-- now subtract stuff the player already has, and generate a list of materials as well
-	for spec = 0, 4 do
-		local specId = spec == 0 and 1 or GetSpecializationInfo(spec)
-		if specId then
-			-- now check if the player has any of the gems or enchants in their inventory, and subtract those
-			for itemId, count in pairs(required.gems[spec]) do
-				required.gems[spec][itemId] = math.max(count - getOwnedCount(itemId), 0)
-				if required.gems[spec][itemId] == 0 then
-					required.gems[spec][itemId] = nil
-				end
-			end
-			for itemId, count in pairs(required.enchants[spec]) do
-				-- look in both spec extra info cache
-				local e = enchantItemIdToId[itemId]
-				local enchInfo = nil
-				if Amr.db.char.ExtraEnchantData[1] then
-					enchInfo = Amr.db.char.ExtraEnchantData[1][e]
-				end
-				if not enchInfo then
-					if Amr.db.char.ExtraEnchantData[2] then
-						enchInfo = Amr.db.char.ExtraEnchantData[2][e]
-					end
-				end
-				if enchInfo then
-					required.enchants[spec][itemId] = math.max(count - getOwnedCount(itemId), 0)
-					if required.enchants[spec][itemId] == 0 then
-						required.enchants[spec][itemId] = nil
-					else
-						-- count up required materials
-						if enchInfo.materials then
-							local c = required.enchants[spec][itemId]
-							for k, v in pairs(enchInfo.materials) do
-								local prev = required.materials[spec][k]
-								required.materials[spec][k] = prev and prev + (v * c) or (v * c)
-							end
-						end
-					end
-				end
-			end
-			-- check if player has any of the materials already
-			for itemId, count in pairs(required.materials[spec]) do
-				required.materials[spec][itemId] = math.max(count - getOwnedCount(itemId), 0)
-				if required.materials[spec][itemId] == 0 then
-					required.materials[spec][itemId] = nil
-				end
-			end
-		end
 	end
 	Amr.db.global.Shopping[player.Name .. "-" .. player.Realm] = required
 end
--- TODO: re-enable shopping list with Legion
---[[
 Amr:AddEventHandler("AUCTION_HOUSE_SHOW", function()
 	_isAhOpen = true
 	if Amr.db.profile.options.shopAh then
 		Amr:ShowShopWindow()
 	end
 	_isAhOpen = false
 	if Amr.db.profile.options.shopAh then
 		Amr:HideShopWindow()
 	end
 end)
-]]

Mercurial > wow > askmrrobot

comparison Shopping.lua @ 89:6bbe64d587b4 v42