mapcolors: module to show class colored circles

init/modules.xml

@@ -44,6 +44,7 @@
   <Include file="..\modules\sellvalue.lua"/>
   <Include file="..\modules\eqcompare.lua"/>
   <Include file="..\modules\map.lua"/>
+  <Include file="..\modules\mapcolors.lua"/>
   <Include file="..\modules\uf_tukui.lua"/>
   <Include file="..\modules\custom.lua"/>
   <Include file="..\modules\gm.lua"/>

modules/mapcolors.lua

@@ -0,0 +1,78 @@
+-- adds class colored circles on world and battlefield map
+pfUI:RegisterModule("mapcolors", function ()
+  local mapcolors = CreateFrame("Frame", nil, UIParent)
+  mapcolors:SetScript("OnUpdate", function()
+    -- throttle to to one item per .1 second
+    if ( this.tick or 1) > GetTime() then return else this.tick = GetTime() + .1 end
+
+    local frame, icon
+
+    -- initialize all button names
+    if not this.buttons then
+      this.buttons = {}
+
+      for i = 1, 4 do
+        icon = string.format("WorldMapParty%d", i)
+        this.buttons[icon] = string.format("party%d", i)
+      end
+
+      for i = 1, 4 do
+        icon = string.format("BattlefieldMinimapParty%d", i)
+        this.buttons[icon] = string.format("party%d", i)
+      end
+
+      for i = 1, 40 do
+        icon = string.format("BattlefieldMinimapRaid%d", i)
+        this.buttons[icon] = string.format("raid%d", i)
+      end
+
+      for i = 1, 40 do
+        icon = string.format("WorldMapRaid%d", i)
+        this.buttons[icon] = string.format("raid%d", i)
+      end
+    end
+
+    -- update all available buttons
+    local ingroup
+    for name, unitstr in pairs(this.buttons) do
+      frame = _G[name]
+
+      if frame and UnitExists(unitstr) then
+        icon = _G[name.."Icon"]
+        icon:SetTexture()
+
+        -- create icon if not yet existing
+        if not frame.pfIcon then
+          frame.pfIcon = frame:CreateTexture(nil, "OVERLAY")
+          SetAllPointsOffset(frame.pfIcon, frame, 3, -3)
+        end
+
+        -- check if unit is in same group
+        ingroup = nil
+        for i=1,5 do -- check if unit is in group
+          if UnitName(string.format("party%d", i)) == UnitName(unitstr) then
+            ingroup = true
+          end
+        end
+
+        -- update texture according to raid/group state
+        if ingroup and frame.pfIcon.ingroup ~= "PARTY" then
+          frame.pfIcon:SetTexture(pfUI.media["img:circleparty"])
+          frame.pfIcon.ingroup = "PARTY"
+        elseif not ingroup and frame.pfIcon.ingroup ~= "RAID" then
+          frame.pfIcon:SetTexture(pfUI.media["img:circleraid"])
+          frame.pfIcon.ingroup = "RAID"
+        end
+
+        -- detect unit class and set color
+        local _, class = UnitClass(unitstr)
+        local color = RAID_CLASS_COLORS[class]
+        if color then
+          frame.pfIcon:SetVertexColor(color.r, color.g, color.b)
+        else
+          frame.pfIcon:SetVertexColor(.5,1,.5)
+        end
+      end
+    end
+  end)
+end)