arrow.lua

   1 QuestHelper_File["arrow.lua"] = "Development Version"
   2 QuestHelper_Loadtime["arrow.lua"] = GetTime()
   3
   4 --[[ This entire file is pretty liberally ganked from TomTom (and then modified) under the following license:
   5
   6 -------------------------------------------------------------------------
   7   Copyright (c) 2006-2007, James N. Whitehead II
   8   All rights reserved.
   9
  10   Redistribution and use in source and binary forms, with or without
  11   modification, are permitted provided that the following conditions are
  12   met:
  13
  14       * Redistributions of source code must retain the above copyright
  15         notice, this list of conditions and the following disclaimer.
  16       * Redistributions in binary form must reproduce the above
  17         copyright notice, this list of conditions and the following
  18         disclaimer in the documentation and/or other materials provided
  19         with the distribution.
  20       * The name or alias of the copyright holder may not be used to endorse
  21         or promote products derived from this software without specific prior
  22         written permission.
  23
  24   THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  25   "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  26   LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  27   A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  28   OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  29   SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  30   LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  31   DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  32   THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  33   (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  34   OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  35 ---------------------------------------------------------------------------
  36
  37 Thanks, James! <3   ]]
  38
  39 local function ColorGradient(perc, ...)
  40         local num = select("#", ...)
  41         local hexes = type(select(1, ...)) == "string"
  42
  43         if perc == 1 then
  44                 return select(num-2, ...), select(num-1, ...), select(num, ...)
  45         end
  46
  47         num = num / 3
  48
  49         local segment, relperc = math.modf(perc*(num-1))
  50         local r1, g1, b1, r2, g2, b2
  51         r1, g1, b1 = select((segment*3)+1, ...), select((segment*3)+2, ...), select((segment*3)+3, ...)
  52         r2, g2, b2 = select((segment*3)+4, ...), select((segment*3)+5, ...), select((segment*3)+6, ...)
  53
  54         if not r2 or not g2 or not b2 then
  55                 return r1, g1, b1
  56         else
  57                 return r1 + (r2-r1)*relperc,
  58                 g1 + (g2-g1)*relperc,
  59                 b1 + (b2-b1)*relperc
  60         end
  61 end
  62
  63 local wayframe = CreateFrame("Button", "QHArrowFrame", UIParent)
  64 wayframe:SetPoint("CENTER", -300, 300)
  65 wayframe:EnableMouse(true)
  66 wayframe:SetMovable(true)
  67 wayframe:SetUserPlaced(true)
  68 wayframe:Hide()
  69
  70 wif = wayframe
  71
  72 -- Frame used to control the scaling of the title and friends
  73 local titleframe = CreateFrame("Frame", nil, wayframe)
  74
  75 wayframe.title = titleframe:CreateFontString("OVERLAY", nil, "GameFontNormalSmall")
  76 wayframe.status = titleframe:CreateFontString("OVERLAY", nil, "GameFontNormalSmall")
  77 wayframe.tta = titleframe:CreateFontString("OVERLAY", nil, "GameFontNormalSmall")
  78 wayframe.title:SetPoint("TOP", wayframe, "BOTTOM", 0, 0)
  79 wayframe.status:SetPoint("TOP", wayframe.title, "BOTTOM", 0, 0)
  80 wayframe.tta:SetPoint("TOP", wayframe.status, "BOTTOM", 0, 0)
  81 local default_font_name, default_font_size, default_font_flags = wayframe.title:GetFont()
  82
  83 do
  84   local r, g, b, a = wayframe.status:GetTextColor()
  85   r, g, b = r - 0.2, g - 0.2, b - 0.2
  86   wayframe.status:SetTextColor(r, g, b, a)
  87 end
  88
  89 local OnUpdate
  90
  91 local function OnDragStart(self, button)
  92         if not QuestHelper_Pref.arrow_locked then  -- TODO TWEAKERY
  93                 self:StartMoving()
  94         end
  95 end
  96
  97 local function OnDragStop(self, button)
  98         self:StopMovingOrSizing()
  99 end
 100
 101 QH_Event("ZONE_CHANGED_NEW_AREA", function ()
 102         if QuestHelper_Pref.arrow and not QuestHelper_Pref.hide then -- TODO TWEAKERY
 103                 wayframe:Show()
 104     OnUpdate()
 105         end
 106 end)
 107
 108 wayframe:SetScript("OnDragStart", OnDragStart)
 109 wayframe:SetScript("OnDragStop", OnDragStop)
 110 wayframe:RegisterForDrag("LeftButton")
 111
 112 wayframe.arrow = wayframe:CreateTexture("OVERLAY")
 113 --wayframe.arrow:SetTexture("Interface\\AddOns\\QuestHelper\\arrow_image_down") -- if we don't do this, the image doesn't seem to end up cached. :blizzard:
 114 wayframe.arrow:SetTexture("Interface\\AddOns\\QuestHelper\\arrow_image")
 115 wayframe.arrow:SetAllPoints()
 116
 117 local active_point, arrive_distance, showDownArrow, point_title
 118 active_point = {}
 119
 120 function QH_Arrow_Show()
 121   wayframe:Show()
 122 end
 123
 124 function QH_Arrow_Reset()
 125   QuestHelper_Pref.arrow = true
 126   wayframe:ClearAllPoints()
 127   wayframe:SetPoint("CENTER", 0, 0)
 128   QuestHelper_Pref.arrow_locked = false -- they're probably going to want to move it
 129   QH_Arrow_Show()
 130 end
 131
 132 function QH_Arrow_SetScale(scale)
 133   if scale then
 134     QuestHelper_Pref.arrow_arrowsize = scale
 135   else
 136     scale = QuestHelper_Pref.arrow_arrowsize
 137   end
 138
 139   if not showDownArrow then
 140     wayframe:SetHeight(42 * scale)
 141     wayframe:SetWidth(56 * scale)
 142   else
 143     scale = scale * 0.8
 144     wayframe:SetHeight(70 * scale)
 145     wayframe:SetWidth(53 * scale)
 146   end
 147 end
 148
 149 function QH_Arrow_SetTextScale(scale)
 150   if scale then
 151     QuestHelper_Pref.arrow_textsize = scale
 152   else
 153     scale = QuestHelper_Pref.arrow_textsize
 154   end
 155
 156   wayframe.title:SetFont(default_font_name, default_font_size * scale, default_font_flags)
 157   wayframe.status:SetFont(default_font_name, default_font_size * scale, default_font_flags)
 158   wayframe.tta:SetFont(default_font_name, default_font_size * scale, default_font_flags)
 159 end
 160
 161 local function wpupdate(c, z, x, y, desc)
 162   if c then
 163     QuestHelper: Assert(z)
 164     QuestHelper: Assert(x)
 165     QuestHelper: Assert(y)
 166   else
 167     QuestHelper: Assert(not z)
 168     QuestHelper: Assert(not x)
 169     QuestHelper: Assert(not y)
 170   end
 171   active_point.c, active_point.z, active_point.x, active_point.y = c, z, x, y
 172   wayframe.title:SetText(desc)
 173   wayframe:Show()
 174   OnUpdate()
 175 end
 176
 177 QuestHelper:AddWaypointCallback(wpupdate)
 178
 179 local status = wayframe.status
 180 local tta = wayframe.tta
 181 local arrow = wayframe.arrow
 182 local last_distance = 0
 183 local tta_throttle = 0
 184 local speed = 0
 185 local speed_count = 0
 186
 187 OnUpdate = function()
 188   local self = wayframe
 189   QuestHelper: Assert(self)
 190
 191         if not active_point.c or QuestHelper.collect_rc ~= active_point.c or QuestHelper.collect_delayed or QuestHelper.InBrokenInstance or not QuestHelper_Pref.arrow then
 192                 self:Hide()
 193                 return
 194         end
 195
 196   self:Show()
 197
 198   local dist, dx, dy = QuestHelper.Astrolabe:ComputeDistance(QuestHelper.collect_rc, QuestHelper.collect_rz, QuestHelper.collect_rx, QuestHelper.collect_ry, active_point.c, active_point.z, active_point.x, active_point.y)
 199
 200   if dist then
 201     status:SetText(QHFormat("DISTANCE", math.floor(dist + 0.5)))
 202   else
 203     status:SetText("")
 204   end
 205
 206         -- Showing the arrival arrow?
 207         if dist and dist <= 10 then
 208                 if not showDownArrow then
 209                         showDownArrow = true
 210       QH_Arrow_SetScale()
 211                         arrow:SetTexture("Interface\\AddOns\\QuestHelper\\arrow_image_down")
 212                         arrow:SetVertexColor(0, 1, 0)
 213                 end
 214
 215                 local cell = math.floor(mod(GetTime() * 20, 55)) -- 20 fps seems to be around the right number
 216
 217                 local column = cell % 9
 218                 local row = floor(cell / 9)
 219
 220                 local xstart = (column * 53) / 512
 221                 local ystart = (row * 70) / 512
 222                 local xend = ((column + 1) * 53) / 512
 223                 local yend = ((row + 1) * 70) / 512
 224                 arrow:SetTexCoord(xstart,xend,ystart,yend)
 225         else
 226                 if showDownArrow then
 227                         showDownArrow = false
 228       QH_Arrow_SetScale()
 229                         arrow:SetTexture("Interface\\AddOns\\QuestHelper\\arrow_image")
 230                         showDownArrow = false
 231                 end
 232
 233                 local angle = atan2(-dx, -dy) / 360 * (math.pi * 2) -- degrees. seriously what
 234     --if angle < 0 then angle = angle + math.pi * 2 end
 235                 local player = GetPlayerFacing()
 236                 angle = angle - player
 237
 238                 local perc = math.abs((math.pi - math.abs(angle)) / math.pi)
 239     if perc > 1 then perc = 2 - perc end -- siiigh
 240
 241                 local gr,gg,gb = 0, 1, 0
 242                 local mr,mg,mb = 1, 1, 0
 243                 local br,bg,bb = 1, 0, 0
 244                 local r,g,b = ColorGradient(perc, br, bg, bb, mr, mg, mb, gr, gg, gb)
 245                 arrow:SetVertexColor(r,g,b)
 246
 247
 248                 local cell = floor(angle / (math.pi * 2) * 108 + 0.5) % 108
 249                 local column = cell % 9
 250                 local row = floor(cell / 9)
 251
 252                 local xstart = (column * 56) / 512
 253                 local ystart = (row * 42) / 512
 254                 local xend = ((column + 1) * 56) / 512
 255                 local yend = ((row + 1) * 42) / 512
 256                 arrow:SetTexCoord(xstart,xend,ystart,yend)
 257         end
 258
 259         -- Calculate the TTA every second  (%01d:%02d)
 260
 261   --[[
 262   if elapsed then
 263     tta_throttle = tta_throttle + elapsed
 264
 265     if tta_throttle >= 1.0 then
 266       -- Calculate the speed in yards per sec at which we're moving
 267       local current_speed = (last_distance - dist) / tta_throttle
 268
 269       if last_distance == 0 then
 270         current_speed = 0
 271       end
 272
 273       if speed_count < 2 then
 274         speed = (speed + current_speed) / 2
 275         speed_count = speed_count + 1
 276       else
 277         speed_count = 0
 278         speed = current_speed
 279       end
 280
 281       if speed > 0 then
 282         local eta = math.abs(dist / speed)
 283         tta:SetFormattedText("%01d:%02d", eta / 60, eta % 60)
 284       else
 285         tta:SetText("")
 286       end
 287
 288       last_distance = dist
 289       tta_throttle = 0
 290     end
 291   end
 292   ]]
 293 end
 294
 295 wayframe:SetScript("OnUpdate", OnUpdate)
 296
 297
 298 local function spacer()
 299   local htex = QuestHelper:CreateIconTexture(item, 10)
 300   htex:SetVertexColor(1, 1, 1, 0)
 301   return htex
 302 end
 303
 304 function QH_Arrow_PopulateMenu(menu)
 305   local hide = QuestHelper:CreateMenuItem(menu, QuestHelper_Pref.arrow and QHText("QH_BUTTON_HIDE") or QHText("QH_BUTTON_SHOW"))
 306   hide:SetFunction(function () QuestHelper:ToggleArrow() end)
 307   --hide:AddTexture(spacer(), true)
 308   --hide:AddTexture(spacer(), false)
 309
 310   local lock = QuestHelper:CreateMenuItem(menu, "Lock")
 311   local ltex = QuestHelper:CreateIconTexture(item, 10)
 312   lock:SetFunction(function () QuestHelper_Pref.arrow_locked = not QuestHelper_Pref.arrow_locked end)
 313   lock:AddTexture(ltex, true)
 314   lock:AddTexture(spacer(), false)
 315   ltex:SetVertexColor(1, 1, 1, QuestHelper_Pref.arrow_locked and 1 or 0)
 316
 317   local scale = QuestHelper:CreateMenuItem(menu, "Arrow Scale")
 318   local scale_menu = QuestHelper:CreateMenu()
 319   scale:SetSubmenu(scale_menu)
 320   scale:AddTexture(spacer(), true)
 321   for i = 5, 15 do
 322     local it = QuestHelper:CreateMenuItem(scale_menu, string.format("%d%%", i * 10))
 323     local ix = i
 324     it:SetFunction(function () QH_Arrow_SetScale(ix / 10) end)
 325     local icon = QuestHelper:CreateIconTexture(item, 10)
 326     if QuestHelper_Pref.arrow_arrowsize == ix / 10 then
 327       icon:SetVertexColor(1, 1, 1, 1)
 328     else
 329       icon:SetVertexColor(1, 1, 1, 0)
 330     end
 331     it:AddTexture(icon)
 332   end
 333
 334   local tscale = QuestHelper:CreateMenuItem(menu, "Text Scale")
 335   local tscale_menu = QuestHelper:CreateMenu()
 336   tscale:SetSubmenu(tscale_menu)
 337   tscale:AddTexture(spacer(), true)
 338   for i = 5, 15 do
 339     local it = QuestHelper:CreateMenuItem(tscale_menu, string.format("%d%%", i * 10))
 340     local ix = i
 341     it:SetFunction(function () QH_Arrow_SetTextScale(ix / 10) end)
 342     local icon = QuestHelper:CreateIconTexture(item, 10)
 343     if QuestHelper_Pref.arrow_textsize == ix / 10 then
 344       icon:SetVertexColor(1, 1, 1, 1)
 345     else
 346       icon:SetVertexColor(1, 1, 1, 0)
 347     end
 348     it:AddTexture(icon)
 349   end
 350
 351   local reset = QuestHelper:CreateMenuItem(menu, QHText("SETTINGS_MENU_ARROW_RESET"))
 352   reset:SetFunction(function () QH_Arrow_Reset() end)
 353 end
 354
 355 local function WayFrame_OnClick(self, button)
 356   local menu = QuestHelper:CreateMenu()
 357   QuestHelper:CreateMenuTitle(menu, QHText("SETTINGS_ARROWLINK_ARROW"))
 358
 359   QH_Arrow_PopulateMenu(menu)
 360
 361   menu:ShowAtCursor()
 362 end
 363
 364 wayframe:RegisterForClicks("RightButtonUp")
 365
 366 QH_OnUpdate(OnUpdate)
 367