vanilla-wow-addons – Rev 1

Subversion Repositories:
Rev:
-- Handle displaying all the fish in their habitats

FishingBuddy.Locations = {};

local NUM_THINGIES_DISPLAYED = 20;
FishingBuddy.Locations.FRAME_THINGIEHEIGHT = 16;

local Collapsed = false;
local LocationLineSelected = 0;
local LocationLines = {};
local LocationLastLine = 1;

local function MakeInfo(line, level, collapsible, expanded, hasicon, text, extra, index, id)
   if ( not LocationLines[line] ) then
      LocationLines[line] = {};
   end
   LocationLines[line].level = level;
   LocationLines[line].collapsible = collapsible;
   LocationLines[line].expanded = expanded;
   LocationLines[line].hasicon = hasicon;
   LocationLines[line].text = text;
   LocationLines[line].extra = extra;
   if ( index ) then
      LocationLines[line].index = index;
   else
      LocationLines[line].index = text;
   end
   LocationLines[line].id = id;
   LocationLines[line].valid = true;
end

local function CountLocationLines()
   local linecount = 0;
   local j = 1;
   local limit = LocationLastLine;
   while ( j <= limit ) do
      local info = LocationLines[j];
      j = j + 1;
      if ( info and info.valid ) then
         linecount = linecount + 1;
         if ( info.collapsible and not info.expanded ) then
            local i2 = LocationLines[j];
            while ( i2 and (i2.level > info.level) ) do
               j = j + 1;
               i2 = LocationLines[j];
            end
         end     
      end
   end
   -- there's a zero-based vs. one-based bug here, somewhere
   if ( linecount > NUM_THINGIES_DISPLAYED and
       limit >= table.getn(LocationLines)) then
      linecount = linecount + 1;
   end
--   FishingBuddy.Debug("Count "..linecount.." limit "..limit);
   return linecount;
end

local function FishiesChanged()
   local fh = FishingBuddy_Info["FishingHoles"];
   local ff = FishingBuddy_Info["Fishies"];
   local line = 1
   local fishcount = table.getn(FishingBuddy.SortedFishies);
   local zonetotals = {};

   for i=1,fishcount,1 do
      local fishid = FishingBuddy.SortedFishies[i].id;
      local fishname = ff[fishid].name;
      local locsort = {};
      local total = 0;
      for zone in FishingBuddy.ByFishie[fishid] do
         if ( not zonetotals[zone] ) then
            local fi = nil;
            for z in fh do
               if ( fh[z][zone] ) then
                  fi = fh[z][zone];
                  break;
               end
            end
            if ( fi ) then
               local tot = 0;
               for f in fi do
                  tot = tot + fi[f];
               end
               zonetotals[zone] = tot;
            end
         end
         local count = FishingBuddy.ByFishie[fishid][zone];
         local info = {};
         info.text = zone;
         info.count = count;
         info.total = zonetotals[zone];
         if ( not info.total or info.total == 0) then
            info.total = 1;
         end
         tinsert(locsort, info);
         total = total + count;
      end
      local extra = " ("..total.." total";
      if ( ff[fishid].level ) then
         extra = extra..", "..ff[fishid].level;
      end
      extra = extra..")";
      MakeInfo(line, 0, true, true, true, fishname, extra, nil, fishid);
      line = line + 1;
      FishingBuddy.FishSort(locsort);
      for j=1,table.getn(locsort),1 do
         local zone = locsort[j].text;
         local amount = locsort[j].count;
         local total = locsort[j].total;
         local percent = format("%.1f", ( amount / total ) * 100);
         MakeInfo(line, 1, false, false, false, zone, " ("..amount..", "..percent.."%)");
         line = line + 1;
      end
   end
   LocationLastLine = line;
end

local function BothLocationsChanged()
   local fh = FishingBuddy_Info["FishingHoles"];
   local ff = FishingBuddy_Info["Fishies"];
   local sorted = FishingBuddy.SortedZones;
   local line = 1;
   local zonecount = table.getn(sorted);
   for i=1,zonecount,1 do
      local zone = sorted[i];
      local where = zone;
      MakeInfo(line, 0, true, true, false, zone, nil, where);
      line = line + 1;
      local subsorted = FishingBuddy.SortedByZone[zone];
      local subcount = table.getn(subsorted);
      for s=1,subcount,1 do
         local subzone = subsorted[s];
         local count, total = FishingBuddy.FishCount(zone, subzone);
         where = zone.."."..subzone;
         local extra = " ("..count.." types, "..total.." total)";
         if ( FishingBuddy_Info["FishingSkill"][zone] and FishingBuddy_Info["FishingSkill"][zone][subzone] ) then
            extra = extra.." ["..FishingBuddy_Info["FishingSkill"][zone][subzone].."]";
         end
         if ( fh[zone][subzone] ) then
            MakeInfo(line, 1, true, true, false,  subzone, extra, where);
            line = line + 1;
            local fishsort = {};
            for fishid in fh[zone][subzone] do
               local info = {};
               info.id = fishid;
               info.text = ff[fishid].name;
               info.count = fh[zone][subzone][fishid];
               tinsert(fishsort, info);
            end
            FishingBuddy.FishSort(fishsort);
            for j=1,table.getn(fishsort),1 do
               local fishie = fishsort[j].text;
               local id = fishsort[j].id;
               local amount = fishsort[j].count;
               local percent = format("%.1f", ( amount / total ) * 100);
               MakeInfo(line, 2, false, false, true, fishie, " ("..percent.."%)", nil, id);
               line = line + 1;
            end
         end
      end
   end
   LocationLastLine = line;
end

local function SubZonesChanged()
   local fh = FishingBuddy_Info["FishingHoles"];
   local ff = FishingBuddy_Info["Fishies"];
   local mapping = {};
   for zone in fh do
      for subzone in fh[zone] do
         mapping[subzone] = zone;
      end
   end
   local line = 1;
   local zonecount = table.getn(FishingBuddy.SortedSubZones);
   for i=1,zonecount,1 do
      local subzone = FishingBuddy.SortedSubZones[i];
      local zone = mapping[subzone];
      local extra = nil;
      if ( FishingBuddy_Info["FishingSkill"][zone] and FishingBuddy_Info["FishingSkill"][zone][subzone] ) then
         extra = " ["..FishingBuddy_Info["FishingSkill"][zone][subzone].."]";
      end
      MakeInfo(line, 0, true, true, false, subzone, extra);
      line = line + 1;
      local zone = mapping[subzone];
      local count, total = FishingBuddy.FishCount(zone, subzone);
      local fishsort = {};
      for fishid in fh[zone][subzone] do
         local info = {};
         info.id = fishid;
         info.text = ff[fishid].name;
         info.count = fh[zone][subzone][fishid];
         tinsert(fishsort, info);
      end
      FishingBuddy.FishSort(fishsort);
      for j=1,table.getn(fishsort),1 do
         local id = fishsort[j].id;
         local fishie = fishsort[j].text;
         local amount = fishsort[j].count;
         local percent = format("%.1f", ( amount / total ) * 100);
         MakeInfo(line, 1, false, false, true, fishie, " ("..percent.."%)", nil, id);
         line = line + 1;
      end
   end
   LocationLastLine = line;
end

local function LinesChanged()
   if ( FishingBuddy.GetSetting("GroupByLocation") == 1 ) then
      if ( FishingBuddy.GetSetting("ShowLocationZones") == 1 ) then
         BothLocationsChanged();
      else
         SubZonesChanged();
      end
   else
      FishiesChanged();
   end
   for i=LocationLastLine,table.getn(LocationLines) do
      local info = LocationLines[i];
      if ( info ) then
         info.valid = false;
      end
   end
   FishingLocationsFrame.valid = true;
end

-- local MOUSEWHEEL_DELAY = 0.1;
-- local lastScrollTime = nil;
-- function FishingLocationsFrame_OnMouseWheel(value)
--    local now = GetTime();
--    if ( not lastScrollTime ) then
--       lastScrollTime = now - 0.2;
--    end
--    if ( (now - lastScrollTime) > MOUSEWHEEL_DELAY ) then
--       -- call the old mouse wheel function somehow?
--    end
-- end

function FishingLocationsFrame_SetSelection(id, line)
   local info = LocationLines[line];
   FishingLocationHighlightFrame:Hide();
   if info then
      if ( info.collapsible ) then
         info.expanded = not info.expanded;
      else
         LocationLineSelected = line;
         FishingLocationHighlightFrame:SetPoint ( "TOPLEFT" ,  getglobal("FishingLocations"..id):GetName() , "TOPLEFT" , 5 , 0 )
         FishingLocationHighlightFrame:Show()
      end
   end
end

function FishingLocationsFrame_MoveButtonText(i, what)
   local relativeTo = "FishingLocations"..i..what;
   local textfield = getglobal("FishingLocations"..i.."Text");
   textfield:SetPoint("LEFT", relativeTo, "RIGHT", 2, 0);
   textfield = getglobal("FishingLocations"..i.."HighlightText");
   textfield:SetPoint("LEFT", relativeTo, "RIGHT", 2, 0);
end

FishingBuddy.Locations.Update = function(forced)
   if ( not FishingLocationsFrame:IsVisible() ) then
      return;
   end

   if ( forced or not FishingLocationsFrame.valid ) then
      LinesChanged();
   end

   local offset = FauxScrollFrame_GetOffset(FishingLocsScrollFrame);
   FauxScrollFrame_Update( FishingLocsScrollFrame, CountLocationLines(),
                          NUM_THINGIES_DISPLAYED,
                          FishingBuddy.Locations.FRAME_THINGIEHEIGHT );

   local lastlevel = 0;
   FishingLocationHighlightFrame:Hide();
   local j = 1;
   local o = 1;
   while ( o < offset ) do
      local info = LocationLines[j];
      if ( info ) then
         j = j + 1;
         o = o + 1;
         if ( info.collapsible and not info.expanded ) then
            local i2 = LocationLines[j];
            while ( i2 and i2.level > info.level ) do
               j = j + 1;
               i2 = LocationLines[j];
            end
         end
      end
   end
   for i = 1,NUM_THINGIES_DISPLAYED,1 do
      local locButton = getglobal ( "FishingLocations"..i );
      if ( LocationLines[j] ) then
         local icon = getglobal("FishingLocations"..i.."Icon");
         local icontex = getglobal("FishingLocations"..i.."IconTexture");
         local info = LocationLines[j];
         locButton.id = i;
         locButton.line = j;

         local leveloffset = (info.level - lastlevel)*16;
         if ( i == 1 ) then
            locButton:SetPoint("TOPRIGHT", "FishingLocsScrollFrame", "TOPLEFT", leveloffset, 0);
         else
            local t = i - 1;
            locButton:SetPoint("TOPLEFT", "FishingLocations"..t, "BOTTOMLEFT", leveloffset, 0);
         end
         lastlevel = info.level;

         local text = info.text;
         if text and info.extra then
            text = text .. info.extra;
         end

         locButton:SetText( text );
         icon:ClearAllPoints();
         if ( info.collapsible ) then
            icon:SetPoint("LEFT", "FishingLocations"..i, "LEFT", 21, 0);
            locButton:SetTextColor( 1, 0.82, 0 );
            if ( info.expanded ) then
               locButton:SetNormalTexture("Interface\\Buttons\\UI-MinusButton-Up");
            else
               locButton:SetNormalTexture("Interface\\Buttons\\UI-PlusButton-Up");
            end
            getglobal("FishingLocations"..i.."Highlight"):SetTexture("Interface\\Buttons\\UI-PlusButton-Hilight");
            getglobal("FishingLocations"..i):UnlockHighlight();
         else
            icon:SetPoint("LEFT", "FishingLocations"..i, "LEFT", 3, 0);
            locButton:SetTextColor( .5, .5, .5 );
            locButton:SetNormalTexture("");
            getglobal("FishingLocations"..i.."Highlight"):SetTexture("");
            -- Place the highlight and lock the highlight state
            if ( LocationLineSelected == j ) then
               FishingLocationHighlightFrame:SetPoint("TOPLEFT", "FishingLocations"..i, "TOPLEFT", 21, 0);
               FishingLocationHighlightFrame:Show();
               locButton:LockHighlight();
            else
               locButton:UnlockHighlight();
            end
         end

         locButton.tooltip = nil;
         if ( info.hasicon ) then
            local item, texture, _, _, _ = FishingBuddy.GetFishie(info.id);
            locButton.item = item;
            locButton.name = info.text;
            if( texture ) then
               icontex:SetTexture(texture);
               icon:Show();
               icontex:Show();
            end
            FishingLocationsFrame_MoveButtonText(i, "Icon");
         else
            locButton.item = nil;
            locButton.name = nil;
            icontex:SetTexture("");
            icontex:Hide();
            icon:Hide();
            FishingLocationsFrame_MoveButtonText(i, "Highlight");
         end
         locButton:Show();
         j = j + 1;
         if ( info.collapsible and not info.expanded ) then
            local i2 = LocationLines[j];            
            while ( i2 and (i2.level > info.level) ) do
               j = j + 1;
               i2 = LocationLines[j];
            end
         end
      else
         locButton:Hide();
         locButton.id = nil;
         locButton.line = nil;
      end
   end

   if LocationLines then
      -- Set the expand/collapse all button texture
      local numHeaders = 0;
      local notExpanded = 0;
      for i=1,table.getn(LocationLines),1 do
         local j = i + offset;
         local info = LocationLines[j];
         if ( info and info.collapsible ) then
            numHeaders = numHeaders + 1;
            if ( not info.expanded ) then
               notExpanded = notExpanded + 1;
            end
         end
      end
      FishingLocationsCollapseAllButton:Show();
      -- If all headers are not expanded then show collapse button, otherwise show the expand button
      if ( notExpanded ~= numHeaders ) then
         Collapsed = false;
         FishingLocationsCollapseAllButton:SetNormalTexture("Interface\\Buttons\\UI-MinusButton-Up");
      else
         Collapsed = true;
         FishingLocationsCollapseAllButton:SetNormalTexture("Interface\\Buttons\\UI-PlusButton-Up");
      end
   else
      FishingLocationsCollapseAllButton:Hide();
   end
end

FishingBuddy.Locations.Button_OnClick = function(button)
   if ( button == "LeftButton" ) then
      if( IsShiftKeyDown() and this.item ) then
         FishingBuddy.ChatLink(this.item, this.name, this.color);
      elseif ( this.id and this.line ) then
         FishingLocationsFrame_SetSelection(this.id, this.line);
         FishingBuddy.Locations.Update();
      end
   end
end

function FishingLocationsCollapseAllButton_OnClick()
   if not Collapsed then
      FishingLocsScrollFrameScrollBar:SetValue(0);
      LocationLineSelected = 1;
   end
   for _,info in LocationLines do
      info.expanded = Collapsed;
   end
   Collapsed = not Collapsed;
   FishingBuddy.Locations.Update();
end

FishingBuddy.Locations.Button_OnEnter = function()
   if( GameTooltip.finished ) then
      return;
   end
   if( this.item or this.tooltip ) then
      GameTooltip:SetOwner(this, "ANCHOR_RIGHT");
      if ( this.item and this.item ~= "" ) then
         local link = "item:"..this.item;
         local n,l,_,_,_,_,_,_ = GetItemInfo(link);
         if ( n and l ) then
            GameTooltip:SetHyperlink(link);
         else
            this.tooltip = {}
            this.tooltip[1] = { ["text"] = this.name };
            this.tooltip[2] = { ["text"] = FishingBuddy.NOTLINKABLE, ["r"] = 1.0, ["g"] = 0, ["b"] = 0 };
            FishingBuddy.AddTooltip(this.tooltip);
            this.item = nil;
         end
      elseif ( this.tooltip ) then
         FishingBuddy.AddTooltip(this.tooltip, 1, 1, 1);
      end
      GameTooltip.finished = 1;
      GameTooltip:Show();
   end
end

FishingBuddy.Locations.Button_OnLeave = function()
   GameTooltip.finished = nil;
   if( this.item or this.tooltip ) then
      GameTooltip:Hide();
   end
end

FishingBuddy.Locations.DisplayChanged = function()
   FishingLocsScrollFrameScrollBar:SetValue(0);
   LocationLineSelected = 1;
   FishingBuddy.Locations.Update(true);
end

FishingBuddy.Locations.SwitchDisplay = function()
   -- backwards logic check, we're about to change...
   if ( FishingBuddy.GetSetting("GroupByLocation") == 1 ) then
      FishingLocationsSwitchButton:SetText(FishingBuddy.SHOWLOCATIONS);
      FishingBuddyOptionSLZ:Hide();
      FishingBuddy.SetSetting("GroupByLocation", 0);
   else
      FishingLocationsSwitchButton:SetText(FishingBuddy.SHOWFISHIES);
      FishingBuddyOptionSLZ:Show();
      FishingBuddy.SetSetting("GroupByLocation", 1);
   end
   FishingBuddy.Locations.DisplayChanged();
end

FishingBuddy.Locations.SwitchButton_OnEnter = function()
   if ( FishingBuddy.GetSetting("GroupByLocation") == 1 ) then
      GameTooltip:SetText(FishingBuddy.SHOWFISHIES_INFO);
   else
      GameTooltip:SetText(FishingBuddy.SHOWLOCATIONS_INFO);
   end
   GameTooltip:Show();
end

FishingBuddy.Locations.OnLoad = function()
   this:RegisterEvent("VARIABLES_LOADED");
   FishingLocationsSwitchButton:SetText(FishingBuddy.SHOWFISHIES);
   -- Set up checkbox
   FishingBuddyOptionSLZ.name = "ShowLocationZones";
   FishingBuddyOptionSLZ.text = FishingBuddy.CONFIG_SHOWLOCATIONZONES_ONOFF;
   FishingBuddyOptionSLZ.tooltip = FishingBuddy.CONFIG_SHOWLOCATIONZONES_INFO;
end

FishingBuddy.Locations.OnShow = function()
   if ( FishingBuddy.IsLoaded() ) then
      FishingBuddy.Locations.Update();
   end
end

FishingBuddy.Locations.OnEvent = function()
   -- this crashes the client when enabled
   -- this:EnableMouseWheel(0);
end

FishingBuddy.FishCount = function(zone, subzone)
   local count = 0;
   local total = 0;
   local fh = FishingBuddy_Info["FishingHoles"];
   if( fh[zone] and fh[zone][subzone] ) then
      for fishie in fh[zone][subzone] do
         count = count + 1;
         total = total + fh[zone][subzone][fishie];
      end
   end
   return count, total;
end

FishingBuddy.Locations.DataChanged = function(zone, subzone, fishie)
   FishingLocationsFrame.valid = false;   
end