gathererinfo.lua

时间太紧了

--[[
	Gatherer Report/Search UI
	Revision: $Id: GathererInfo.lua 205 2006-05-02 08:33:15Z islorgris $
]]--

-- Scrolling variables
GATHERERINFO_TO_DISPLAY=15;
GATHERERINFO_FRAME_HEIGHT=15;
GATHERERINFO_SUBFRAMES = { "GathererInfo_ReportFrame", "GathererInfo_SearchFrame" };

-- Defaults
currentNode = "";
GathererInfo_LastSortType = "gtype";
GathererInfo_LastSearchSortType= "Cname";
GathererInfo_AutoRegion = true;
GathererInfo_LastSortReverse = false;

GathererInfoContinents = {};
GathererInfoZones = {};

numGathererInfo = 0;
GI_totalCount = 0;


-- show display frame
function showGathererInfo(tab)
	local myTab = 1;
	if ( tab ) then myTab = tab; end;
		
	if ( not GathererInfo_DialogFrame:IsShown() ) then
		GathererInfo_DialogFrame:Show();
		PanelTemplates_SetTab(GathererInfo_DialogFrame, myTab);
		ToggleGathererInfo_Dialog(GATHERERINFO_SUBFRAMES[myTab]);
	else
		GathererInfo_DialogFrame:Hide();
	end
end

function GathererInfo_LoadContinents(...)
	for c=1, arg.n, 1 do
		GathererInfoContinents[c] = {}
		GathererInfoContinents[c] = arg[c];
		GathererInfoContinents[arg[c]] = c;
		GathererInfo_LoadZones(c, GetMapZones(c));
	end
end

function GathererInfo_LoadZones(c, ...)
	for z=1, arg.n, 1 do
		local zname = arg[z];
		if ( not GathererInfoZones[c] ) then GathererInfoZones[c] = {}; end
		GathererInfoZones[c][zname] = { continent=c, zone=z };
		GathererInfoZones[c][z] = { continent=c, zone=zname };
	end
end
-- ********************************************************************
-- dropdown menu functions
-- Continent
function GathererInfo_LocContinentDropDown_Initialize()
	local index;
	local info = {};
	
	for index in GatherItems do
		-- grab zone text from table initiated by Gatherer Main
		info.text = GathererInfoContinents[index];
		info.checked = nil;
		info.func = GathererInfo_LocContinentDropDown_OnClick;
		UIDropDownMenu_AddButton(info);
	end
end

function GathererInfo_LocContinentDropDown_OnClick()
	UIDropDownMenu_SetSelectedID(GathererInfo_LocContinentDropDown, this:GetID());
	UIDropDownMenu_SetSelectedID(GathererInfo_LocZoneDropDown, 0);
	
	UIDropDownMenu_SetText("", GathererInfo_LocZoneDropDown);

	UIDropDownMenu_ClearAll(GathererInfo_LocZoneDropDown);
	UIDropDownMenu_Initialize(GathererInfo_LocZoneDropDown, GathererInfo_LocZoneDropDown_Initialize);

	local j=1;
	while (j < GATHERERINFO_TO_DISPLAY+1) do
		getglobal("GathererInfo_FrameButton"..j):Hide();
		j = j+1;
	end	

end

-- Zone
function GathererInfo_LocZoneDropDown_Initialize()
	local index;
	local info = {};
	local gi_cont, gi_t_cont;

	gi_t_cont = UIDropDownMenu_GetText(GathererInfo_LocContinentDropDown);
	if ( gi_t_cont and gi_t_cont ~= nil and gi_t_cont ~= "") then
		gi_cont = GathererInfoContinents[gi_t_cont];

		for index in GatherItems[gi_cont] do
			-- grab zone text from table initiated by Gatherer Main
			info.text = GathererInfoZones[gi_cont][index].zone;
			info.checked = nil;
			info.func = GathererInfo_LocZoneDropDown_OnClick;
			UIDropDownMenu_AddButton(info);
		end
	end
end

function GathererInfo_LocZoneDropDown_OnClick()
	local gi_cont, gi_zone;
	local l_zone, node_idx, index;
	
	UIDropDownMenu_SetSelectedID(GathererInfo_LocZoneDropDown, this:GetID());
	
	GathererInfo_AutoRegion = false;

	gi_cont = GathererInfoContinents[UIDropDownMenu_GetText(GathererInfo_LocContinentDropDown)];
	gi_zone = GathererInfoZones[gi_cont][UIDropDownMenu_GetText(GathererInfo_LocZoneDropDown)].zone;

	for l_zone in GatherItems[gi_cont] do
		GI_totalCount = 0;
		numGathererInfo=0;
								
		for node_idx in GatherItems[gi_cont][gi_zone] do
			for index in  GatherItems[gi_cont][gi_zone][node_idx] do
				GI_totalCount = GI_totalCount + GatherItems[gi_cont][gi_zone][node_idx][index].count;
				numGathererInfo = numGathererInfo + 1;
			end
		end
	end
		
	GathererInfo_Update();
end

-- ********************************************************************
-- Display frame initialization
function GathererInfo_SetDefaultLocation()
	local gi_t_cont = GathererInfoContinents[GetCurrentMapContinent()];
	local gi_t_zone = GetRealZoneText();
	local gi_cont = GetCurrentMapContinent();
	local gi_zone = nil;
	local l_cnt, l_zone;
	local node_idx, index;
	local found = 0;

	if ( GetRealZoneText() and GathererInfoZones[gi_cont] and GathererInfoZones[gi_cont][GetRealZoneText()] ) then
		gi_zone = GathererInfoZones[gi_cont][GetRealZoneText()].zone;
	else
		gi_zone = "unknown";
	end

	for l_cnt in GatherItems do
		if( l_cnt == gi_cont ) then
			UIDropDownMenu_SetSelectedName(GathererInfo_LocContinentDropDown, gi_t_cont);
			UIDropDownMenu_SetText(gi_t_cont, GathererInfo_LocContinentDropDown);
			
			for l_zone in GatherItems[gi_cont] do
				if ( l_zone == gi_zone and GathererInfo_AutoRegion == true) then
					UIDropDownMenu_SetSelectedName(GathererInfo_LocZoneDropDown, gi_t_zone);
					UIDropDownMenu_SetText(gi_t_zone, GathererInfo_LocZoneDropDown);
					
					GI_totalCount = 0;
					numGathererInfo=0;
					found = 1;
								
					for node_idx in GatherItems[gi_cont][gi_zone] do
						for index in  GatherItems[gi_cont][gi_zone][node_idx] do
							GI_totalCount = GI_totalCount + GatherItems[gi_cont][gi_zone][node_idx][index].count;
							numGathererInfo = numGathererInfo + 1;
						end
					end
					
					GathererInfo_Update();
				end
			end
		end
	end
end

-- update display frame relative to scrollbar
function GathererInfo_Update()
	local node_idx, index;
	local gatherInfoOffset = FauxScrollFrame_GetOffset(GathererInfo_ListScrollFrame);
	local gi_cont, gi_zone, gi_loc;
	local gatherInfoIndex;
	local showScrollBar = nil;
	local button;
	local i = 1;
	GITT = {};

	-- Get continent, zone
	if ( UIDropDownMenu_GetText(GathererInfo_LocContinentDropDown) and
		 UIDropDownMenu_GetText(GathererInfo_LocContinentDropDown) ~= "") 
	then
		gi_cont = GathererInfoContinents[UIDropDownMenu_GetText(GathererInfo_LocContinentDropDown)];
	else
		gi_cont = GetCurrentMapContinent();
	end

	if ( UIDropDownMenu_GetText(GathererInfo_LocZoneDropDown) and
		 UIDropDownMenu_GetText(GathererInfo_LocZoneDropDown) ~= "") 
	then
		gi_zone = GathererInfoZones[gi_cont][UIDropDownMenu_GetText(GathererInfo_LocZoneDropDown)].zone;
	else
		if ( GetRealZoneText() and GathererInfoZones[gi_cont] and GathererInfoZones[gi_cont][GetRealZoneText()] ) then
			gi_zone = GathererInfoZones[gi_cont][GetRealZoneText()].zone;
		else
			gi_zone = nil;
		end
	end
	-- if loc exists in stored base
	if ( GatherItems and gi_cont and gi_zone and GatherItems[gi_cont] and 
		 GatherItems[gi_cont][gi_zone] ) 
	then
		if ( GatherItems and GatherItems[gi_cont] and GatherItems[gi_cont][gi_zone] )
		then
			local typeNodeCount = {};
			typeNodeCount[0]=0;
			typeNodeCount[1]=0;
			typeNodeCount[2]=0;
			typeNodeCount[3]=0;

			for node_idx in GatherItems[gi_cont][gi_zone] do
				for index in GatherItems[gi_cont][gi_zone][node_idx] do
					local gtype;
					if ( type(GatherItems[gi_cont][gi_zone][node_idx][index].gtype) =="number" )
					then
						gtype = GatherItems[gi_cont][gi_zone][node_idx][index].gtype;
					else
						gtype = Gather_DB_TypeIndex[GatherItems[gi_cont][gi_zone][node_idx][index].gtype]
					end

					typeNodeCount[gtype] = typeNodeCount[gtype] + GatherItems[gi_cont][gi_zone][node_idx][index].count;
				end
			end

			-- assign values
			for node_idx in GatherItems[gi_cont][gi_zone] do
				local nodeCount;
				for index in GatherItems[gi_cont][gi_zone][node_idx] do
					local amount = GatherItems[gi_cont][gi_zone][node_idx][index].count;
					local gtype;
					if ( type(GatherItems[gi_cont][gi_zone][node_idx][index].gtype) == "number" )
					then
						gtype = GatherItems[gi_cont][gi_zone][node_idx][index].gtype;
					else
						gtype = Gather_DB_TypeIndex[GatherItems[gi_cont][gi_zone][node_idx][index].gtype]
					end

					nodeCount = typeNodeCount[gtype];				
				
					local found=0;
					local findex;
					for findex in GITT do
						if ( GITT[findex].name == node_idx ) then
							found = findex;
						end
					end
				
					if ( found == 0 ) then
						GITT[i] = {};
						GITT[i].name     = node_idx;

						if ( type(GatherItems[gi_cont][gi_zone][node_idx][index].gtype) == "number" ) then
							GITT[i].gatherType = Gather_DB_TypeIndex[GatherItems[gi_cont][gi_zone][node_idx][index].gtype];
						else
							GITT[i].gatherType = GatherItems[gi_cont][gi_zone][node_idx][index].gtype;
						end

						GITT[i].typePercent  = tonumber(format("%.1f",( amount / nodeCount ) * 100));
						GITT[i].densityPercent  = tonumber(format("%.1f",( amount / GI_totalCount ) * 100));

						-- divide by zero check
						if ( GITT[i].typePercent == nil ) then GITT[i].typePercent = 0; end
						if ( GITT[i].densityPercent == nil ) then GITT[i].densityPercent = 0; end

						GITT[i].amount   = tonumber(amount);

						if ( type(GatherItems[gi_cont][gi_zone][node_idx][index].icon) == "number" ) then
							GITT[i].texture  = Gather_IconSet["iconic"][GITT[i].gatherType][Gatherer_GetDB_IconIndex(GatherItems[gi_cont][gi_zone][node_idx][index].icon, GITT[i].gatherType)];
						else
							GITT[i].texture  = Gather_IconSet["iconic"][GITT[i].gatherType][GatherItems[gi_cont][gi_zone][node_idx][index].icon];
						end

						GITT[i].nodeCount = nodeCount;
						GITT[i].gi = gatherInfoOffset + i;

						i = i + 1;
					else
						GITT[found].amount = GITT[found].amount + tonumber(amount);
						GITT[found].typePercent = tonumber(format("%.1f",( GITT[found].amount / GITT[found].nodeCount ) * 100));
						GITT[found].densityPercent = tonumber(format("%.1f",( GITT[found].amount / GI_totalCount ) * 100));

						-- divide by zero check
						if ( GITT[found].typePercent == nil ) then GITT[found].typePercent = 0; end
						if ( GITT[found].densityPercent == nil ) then GITT[found].densityPercent = 0; end
						found = 0;
					end
				end
			end
		end
		SortGathererInfo(GathererInfo_LastSortType, false);
		GathererInfo_totalcount:SetText(string.gsub(string.gsub(GATHERER_REPORT_SUMMARY, "#", GI_totalCount), "&", numGathererInfo));
	end
end	

-- Add items in display frame
function GathererInfo_additem_table(gatherName, gatherNumber, gatherTexture, gatherType, typePercent, densityPercent, GIlocation, GIIndex)

	if ( GIlocation < GATHERERINFO_TO_DISPLAY +1) then
		getglobal("GathererInfo_FrameButton"..GIlocation).gatherIndex = GIIndex;
		getglobal("GathererInfo_FrameButton"..GIlocation.."Icon"):SetTexture(gatherTexture);
		getglobal("GathererInfo_FrameButton"..GIlocation.."Type"):SetText(gatherType);
		getglobal("GathererInfo_FrameButton"..GIlocation.."Gatherable"):SetText(gatherName);		
		getglobal("GathererInfo_FrameButton"..GIlocation.."Number"):SetText(gatherNumber);
		getglobal("GathererInfo_FrameButton"..GIlocation.."TypePercent"):SetText(typePercent);
		getglobal("GathererInfo_FrameButton"..GIlocation.."DensityPercent"):SetText(densityPercent);
		getglobal("GathererInfo_FrameButton"..GIlocation):Show();
	end
	
end

-- **************************************************************************
-- Sort functions (common to Report and Search)
function SortGathererInfo(sortType, reverse, search)
	local compstr = {};
	local ref_ind = 0;
	local ind, newind;
	local displayed=1;
	local offSet = FauxScrollFrame_GetOffset(GathererInfo_ListScrollFrame);
	local key1, key2, key3;
	local indexToShow=0;
	local search_stub;
	
	if (search) then
		search_stub = search;
	else
		search_stub = "";
	end

	for ind in GITT do
		indexToShow = indexToShow + 1;
		compstr[ind]= {}

		-- report sort types
		if ( sortType == "gtype" ) then -- keys: type, % per type, name (default sort)
			key1 = GITT[ind].gatherType;
			key2 = format("%.6f", GITT[ind].typePercent / 10000);
			key3 = GITT[ind].name;