LoTROInterface SVN SequenceBars

Slot = class(Turbine.UI.Control);

function Slot:Constructor(info, wantsMouseWheelEvents)
    Turbine.UI.Control.Constructor(self);
    self:SetMouseVisible(false);
    self:SetSize(36, 36);
    self.advanceFunc = {};
    self:SetInfo(info);
    self:SetBlendMode(Turbine.UI.BlendMode.AlphaBlend);
    Turbine.UI.Control.SetAllowDrop(self, false);
    self.wantsMouseWheelEvents = wantsMouseWheelEvents;
end

function Slot:GetInfo()
    return self.info;
end

-- Get skill name(s) from moebius92's SkillData database
function Slot:GetSkillNames()
    if (not self.skillNames and (self.info.type == Turbine.UI.Lotro.ShortcutType.Skill)) then
        local shortcut = self.object and self.object.GetShortcut and self.object:GetShortcut();
        self.skillNames = shortcut and shortcut:GetSkillName();
        if (type(self.skillNames) == "string") then
            self.skillNames = { self.skillNames };
        end
    end
    return self.skillNames;
end

-- Get canonical skill name (with no fire oil, light oil, alternate stance, etc.)
function Slot:GetSkillName()
    local skillNames = self:GetSkillNames();
    if (skillNames) then
        return skillNames[1];
    end
end

function Slot:GetSkill()
    local names = self:GetSkillNames();
    if (type(names) == "string") then
        return Thurallor.Utils.Watcher.GetSkillByNames(names);
    elseif (type(names) == "table") then
        return Thurallor.Utils.Watcher.GetSkillByNames(names);
    end
end

function Slot:GetItem()
    if (self.object.GetItem) then
        return self.object:GetItem();
    end
end

function Slot:GetItemName()
    local item = self:GetItem();
    return item and item:GetItemInfo():GetName();
end

function Slot:GetNumLabel()
    return self.slotNumber;
end

function Slot:IsQuickslot()
    return self.isQuickslot;
end

function Slot:SetActionEnabled(enabled)
    self.actionEnabled = enabled;
    if (self.object) then
        self.object:SetMouseVisible(enabled);
    end
    if (self.numLabel) then
        self.numLabel:SetMouseVisible(not enabled);
    end
end

function Slot:SetAllowDrop(allowDrop)
    if (self.numLabel) then
        self.numLabel:SetAllowDrop(allowDrop);
    elseif (self.object) then
        self.object:SetAllowDrop(allowDrop);
    end
end

function Slot:SetUseOnRightClick(useOnRightClick)
    self.useOnRightClick = useOnRightClick;
    if (self.object and self.object.SetUseOnRightClick) then
        self.object:SetUseOnRightClick(useOnRightClick);
    end
end

function Slot:SetDraggable(draggable)
    if (draggable) then
        self.numLabel.MouseDown = function(sender, args)
            self:MouseDown(args);
        end
        self.numLabel.MouseUp = function(sender, args)
            self:MouseUp(args);
        end
        self.numLabel.MouseMove = function()
            self:MouseMove();
        end
    elseif (self.numLabel) then
        self.numLabel.MouseDown = nil;
        self.numLabel.MouseUp = nil;
        self.numLabel.MouseMove = nil;
    end
end

function Slot:SetInfo(info)

    -- Discard old object, if it's a different type
    if (self.object and (self.info.class ~= info.class)) then
        self.object:SetParent(nil);
        self.object = nil;
    end

    self.info = {};
    DeepTableCopy(info, self.info);
    
    if (not self.parent) then
        return;
    end

    -- Create the quickslot object or command icon
    if ((not info.class) or (info.class == "Turbine.UI.Lotro.Quickslot")) then
        if (not self.object) then
            local object = Turbine.UI.Lotro.Quickslot();
            object:SetParent(self);
            object:SetZOrder(1);
            self.object = object;

            -- Forward events to parent object
            object.MouseClick = function(sender, args)
                DoCallbacks(self, "MouseClick", args);
            end
            object.MouseWheel = function(sender, args)
                DoCallbacks(self, "MouseWheel", args);
            end
        end
        local shortcut = Turbine.UI.Lotro.Shortcut(info.type, ((type(info.Data) == "string") and info.Data:toWindows()) or info.Data);
        self.object:SetShortcut(shortcut);
        self.isQuickslot = true;
        self.skillNames = nil;
        self.object.DragDrop = function(sender)
            sender:SetShortcut(Turbine.UI.Lotro.Shortcut(self.info.type, self.info.Data));
        end
    elseif ((info.class == "Turbine.UI.Control") or (info.class == "Turbine.UI.Lotro.EntityControl")) then
        if (not self.object) then
        
            local object;
            if (info.type == "SelectTarget") then
                object = Turbine.UI.Lotro.EntityControl();
                object:SetSelectionEnabled(true);
                object:SetContextMenuEnabled(false);
                object:SetBackground(resources.Icon.SelectTarget);
                self.watcher = Thurallor.Utils.Watcher;
                self.player = Turbine.Gameplay.LocalPlayer:GetInstance();
                self.getTargetFunc = loadstring(info.getTargetFunc);
            else
                -- MouseWheel events aren't implemented for Controls, only ListBoxes.
                -- But I think they probably use more resources than Controls, so I'll
                -- only use ListBoxes if the mouse wheel option is enabled.
                if (self.wantsMouseWheelEvents) then
                    object = Turbine.UI.ListBox();
                else
                    object = Turbine.UI.Control();
                end
            end
            
            object:SetParent(self);
            object:SetSize(32, 32);
            object:SetPosition(3, 3);
            object:SetZOrder(1);
            self.object = object;
            object.slot = self;

            -- Forward events to parent object
            object.MouseClick = function(obj, args)
                if (obj.slot.info.type == "SelectTarget") then
                    -- Update the round-robin "cursor" to the next member, regardless
                    -- of whether the member was successfully selected; clicking on the
                    -- slot is enough to know the user's intent.
                    Thurallor.Utils.Watcher.GetNextPartyMember(obj.slot.entity);
                end
                DoCallbacks(obj.slot, "MouseClick", args);
            end
            object.MouseEnter = function(obj)
                if (obj.slot.info.type == "SelectTarget") then
                    obj.slot:UpdateEntity();
                    obj.slot:SetWantsUpdates(true);
                end
                DoCallbacks(obj.slot, "MouseEnter");
            end
            object.MouseLeave = function(obj)
                if (obj.slot.info.type == "SelectTarget") then
                    obj:SetBackground(resources.Icon.SelectTarget);
                    obj:SetEntity(nil);
                    obj.slot.entity = nil;
                    obj.slot:SetWantsUpdates(false);
                end
                DoCallbacks(obj.slot, "MouseLeave");
            end
            object.MouseWheel = function(obj, args)
                DoCallbacks(obj.slot, "MouseWheel", args);
            end

            -- Add tooltip
            if (info.type == "SelectTarget") then
                self:SetTooltip(function()
                    local entity = self.object:GetEntity();
                    return entity and entity:GetName() or L:GetText("/SequenceEditor/NoTarget");
                end);
            end
        end
        self.isQuickslot = false;
    end
    self:SetUseOnRightClick(self.useOnRightClick);

    -- Draw text overlay and icon overlay (if any)
    if (not self.textOverlay) then
        local textOverlay = Turbine.UI.Label();
        textOverlay:SetParent(self);
        textOverlay:SetZOrder(2);
        textOverlay:SetSize(32, 32);
        textOverlay:SetPosition(3, 3);
        textOverlay:SetMouseVisible(false);
        textOverlay:SetFont(Turbine.UI.Lotro.Font.Verdana10);
        textOverlay:SetTextAlignment(Turbine.UI.ContentAlignment.MiddleCenter);
        textOverlay:SetFontStyle(Turbine.UI.FontStyle.Outline);
        textOverlay:SetOutlineColor(Turbine.UI.Color(1, 0, 0, 0));
        textOverlay:SetMultiline(true);
        textOverlay:SetMarkupEnabled(true);
        textOverlay:SetBlendMode(Turbine.UI.BlendMode.AlphaBlend);
        self.textOverlay = textOverlay;
    end
    self.background = info.background;
    if (info.altIcon) then
        self.background = info.altIcon;
    end
    self.textOverlay:SetBackground(self.background);
    self.textOverlay:SetText(info.textOverlay);
end

-- For "Select Target" slots only:
function Slot:UpdateEntity()
    local entity = self.getTargetFunc(Thurallor.Utils.Watcher, self.player, self.info.raidMemberName, self.info.number); 
    if (entity ~= self.entity) then
        self.object:SetEntity(entity);
        self.entity = entity;
        self.tooltip:Update();
    end
    if (entity) then
        self.object:SetBackground(resources.Icon.SelectTargetKnown);
    else
        self.object:SetBackground(resources.Icon.SelectTargetUnknown);
    end
end

function Slot:SetTooltip(content)
    if (self.tooltip) then
        self.tooltip:SetVisible(false);
        self.tooltip:Detach();
    end
    self.tooltip = Thurallor.UI.Tooltip(content);
    self.tooltip:Attach(self.object);
end

function Slot:SetNumLabel(slotNumber)
    if (not self.numLabel) then
        local numLabel = Turbine.UI.Label();
        numLabel:SetParent(self);
        numLabel:SetFont(Turbine.UI.Lotro.Font.TrajanPro15);
        numLabel:SetFontStyle(Turbine.UI.FontStyle.Outline);
        numLabel:SetOutlineColor(Turbine.UI.Color(1, 0, 0, 0));
        numLabel:SetTextAlignment(Turbine.UI.ContentAlignment.MiddleCenter);
        numLabel:SetBlendMode(Turbine.UI.BlendMode.AlphaBlend);
        numLabel:SetSize(38, 38);
        numLabel:SetZOrder(3);
        numLabel.DragDrop = function(sender, args)
            DoCallbacks(self, "DragDrop", args);
        end
        numLabel.MouseClick = function(sender, args)
            DoCallbacks(self, "MouseClick", args);
        end
        self.numLabel = numLabel;
    end
    self.numLabel:SetText(slotNumber);
    self.slotNumber = slotNumber;
end

function Slot:StartCooldown(period)
    if (period <= 0) then
        period = 0.001;
    end
    self.coolDown = period;
    self.resetTime = Turbine.Engine.GetGameTime() + period;
    if (self.textOverlay) then
        self.textOverlay:SetBackground(resources.Icon.Delay);
        self.textOverlay:SetText(nil);
    end
    self:ShowTimeRemaining(true);
    self:SetWantsUpdates(true);
end

function Slot:Update()
    if (self.info.type == "SelectTarget") then
        self:UpdateEntity();
    else
        self:ShowTimeRemaining();
    end
end

-- Reuse the numLabel layer for cooldown overlay.
function Slot:ShowTimeRemaining(starting)
    local currentTime = Turbine.Engine.GetGameTime();
    -- Check for abort ("Reset" event handler removed)
    if (not starting and not self.Reset) then
        self.resetTime = currentTime;
    end

    if (currentTime >= self.resetTime) then
        self.numLabel:SetParent(nil);
        self.numLabel = nil;
        self.resetTime = nil;
        self:SetWantsUpdates(false);
        if (self.textOverlay) then
            self.textOverlay:SetBackground(self.background);
            self.textOverlay:SetText(self.info.textOverlay);
        end
        self.hidden = self.hiddenAfterReset;
        DoCallbacks(self, "Reset");
    else
        self:SetNumLabel("");
        self.numLabel.MouseClick = nil;
        self.numLabel:SetPosition(3, 3);
        self.numLabel:SetSize(32, 32);
        local steps = resources.Icon.CoolDownEnd - resources.Icon.CoolDownStart + 1;
        local progress = 1 - (self.resetTime - currentTime) / self.coolDown;
        self.numLabel:SetBackground(resources.Icon.CoolDownStart + math.floor(progress * steps));
    end
end

function Slot:SetSelected(selected, color)
    if (not selected) then
        if (self.marquee) then
            self.marquee:SetParent(nil);
            self.marquee = nil;
        end
        return;
    end

    if (not color) then
        color = Turbine.UI.Color.White;
    end

    if (not self.marquee) then
        local marquee = Thurallor.UI.Marquee();
        marquee:SetParent(self);
        marquee:SetPosition(2, 2);
        marquee:SetSize(34, 34);
        marquee:SetMouseVisible(false);
        self.marquee = marquee;
    end
    
    self.marquee:SetColor(color);
end

function Slot:SetParent(parent)
    self.parent = parent;
    if (parent) then
        self:SetInfo(self.info);
    elseif (self.object) then
        self:MouseLeave();
        self.object:SetParent(nil);
        self.object = nil;
    end
    Turbine.UI.Control.SetParent(self, parent);
end

function Slot:DragDrop(args)
    local info = {};
    info.class = "Turbine.UI.Lotro.Quickslot";
    local newShortcut = args.DragDropInfo:GetShortcut();
    if (newShortcut) then
        info.type = newShortcut:GetType();
        info.Data = newShortcut:GetData();
        self:SetInfo(info);
        DoCallbacks(self, "ShortcutChanged", info.type, info.Data);
    end

    -- Don't remove the item/skill from the location where it was dragged from.
    args.DragDropInfo:SetSuccessful(false);
end

function Slot:MouseEnter()
    if ((self.actionEnabled) and (not self.highlight)) then
        local highlight = Turbine.UI.Control();
        highlight:SetParent(self);
        highlight:SetSize(32, 32);
        highlight:SetPosition(3, 3);
        highlight:SetZOrder(4);
        highlight:SetBlendMode(Turbine.UI.BlendMode.Overlay);
        highlight:SetBackground(resources.Icon.Highlight);
        highlight:SetMouseVisible(false);
        self.highlight = highlight;
    end
end

function Slot:MouseLeave()
    if (self.highlight) then
        self.highlight:SetParent(nil);
        self.highlight = nil;
    end
end

function Slot:MouseDown(args)
    if (args.Button == Turbine.UI.MouseButton.Left) then
        -- Drag begin.  Save the current position of the slot and mouse.
        self.mouseDown = true;
        self.mouseMoved = false;
        self.originalLeft, self.originalTop = self:PointToScreen(0, 0);
        self.originalMouseX, self.originalMouseY = Turbine.UI.Display.GetMouseX(), Turbine.UI.Display.GetMouseY();
    end
end

function Slot:MouseMove()
    if (not self.mouseDown) then
        return;
    end
    
    -- Detach from parent if we haven't already done so.
    if (not self.mouseMoved) then
        self.numLabel:SetText(nil);
        self.parent = self:GetParent();
        self.tempContainer = Turbine.UI.Window();
        self.tempContainer:SetVisible(true);
        self.tempContainer:SetSize(self:GetSize());
        self.tempContainer:SetZOrder(2147483647);
        self.tempContainer:SetOpacity(0.75);
        self:SetParent(self.tempContainer);
        self:SetPosition(0, 0);
        self.mouseMoved = true;
    end
    
    -- Drag continuing.  Find out how much the mouse has moved.
    local newMouseX, newMouseY = Turbine.UI.Display.GetMouseX(), Turbine.UI.Display.GetMouseY();
    local deltaX, deltaY = newMouseX - self.originalMouseX, newMouseY - self.originalMouseY;

    -- Move the icon and overlay.
    local left, top = self.originalLeft + deltaX, self.originalTop + deltaY;
    self.tempContainer:SetPosition(left, top);
    
    -- Show a red "X" if invalid drag location is selected.
    local centerX, centerY = left + 16, top + 16;
    if (self:DropPositionValid(centerX, centerY)) then
        self.numLabel:SetBackground(nil);
    else
        self.numLabel:SetBackground(resources.Icon.DragFail);
    end
end

function Slot:MouseUp(args)
    if (args.Button ~= Turbine.UI.MouseButton.Left) then
        return;
    end
    self.mouseDown = false;
    if (not self.mouseMoved) then
        return;
    end
    
    -- Reattach to parent.
    local left, top = self.tempContainer:GetPosition();
    self:SetParent(self.parent);
    self.tempContainer:Close();
    self.tempContainer = nil;
    self.numLabel:SetBackground(nil);
    self.numLabel:SetText(self.slotNumber);

    local centerX, centerY = left + 16, top + 16;
    self:DragDropComplete(centerX, centerY);
end

-- Event handlers that should be overridden
-- 
-- function Slot:ShortcutChanged(shortcutType, shortcutData)
--     Puts("ShortcutChanged(" .. Serialize(shortcutType) .. ", " .. Serialize(shortcutData) .. ")");
-- end
-- 
-- function Slot:MouseClick(args)
--     Puts("MouseClick(" .. Serialize(args) .. ")");
-- end
-- 
-- function Slot:MouseWheel(args)
--     Puts("MouseWheel(" .. Serialize(args) .. ")");
-- end
-- 
-- function Slot:DropPositionValid(left, top)
--     Puts("DropPositionValid(" .. tostring(left) .. ", " .. tostring(top) .. ")");
--     return true;
-- end
-- 
-- function Slot:DragDropComplete(left, top)
--     Puts("DragDropComplete(" .. tostring(left) .. ", " .. tostring(top) .. ")");
--     return true;
-- end