LoTROInterface SVN KragenBars

import "Turbine.UI";
import "Turbine.UI.Lotro";
import "Turbine.Gameplay";
import "KragenPlugs.Utils";
import "KragenPlugs.KragenBars";

ClassBar = class( );

function ClassBar:Constructor()
        
        -- inialize the setupWindow
        self.setup = SetupWindow();
        self.setupbb = self.setup.setupbb;
        self.setupf = self.setup.setupf;
        self.setupqs = self.setup.setupqs;
        
        -- inialize tables
        self.windows = { };
        self.bars = { };
        self.skills = { };
        self.multi = { };
        
        self.barlist = { };
        self.effectlist = {};
        self.barcount = 0;
        
        self.setup:SetWantsKeyEvents(true);
        
        self.setup.KeyDown = function(sender, args)
        if args.Action == 0x100000B3 then
                        self.setup:SetWantsUpdates(true);
                end
        end
        
        self.setup.Update = function(sender, args)
        for k,v in pairs(self.skills) do
                        v:RefreshBorders();
                end
                self.setup:SetWantsUpdates(false);
        end
        
        -- Combat Event
        AddCallback(vars.player, "InCombatChanged", function(sender, args)
                KragenBars:CombatChange(vars.player:IsInCombat());
        end);
        
        -- Effect Events
    AddCallback(vars.effects, "EffectAdded", function(sender, args)
                KragenBars:AddEffect( args.Index );
        end);

        AddCallback(vars.effects, "EffectRemoved", function(sender, args)
                if (args.Effect ~= nil) then
                        KragenBars:RemoveEffect( args.Effect );
                end
        end);
        
end

-- Index and Position Bars
function ClassBar:InitBars()
        self.DefineEffectList();
        
        for k,v in pairs(self.bars) do
                v.index = k;
        end
        
        if(settings.skilldata ~=nil) then
                for k,v in pairs(self.skills) do
                        if (settings.skilldata[k] ~= nil) then
                            local data = settings.skilldata[k];
                            data.parent = self.bars[settings.skilldata[k].parent]
                                v:LoadPosition(data);
                        end
                end
        end
        
        if(settings.multidata ~=nil) then
                for k,v in pairs(self.multi) do
                        if (settings.multidata[k] ~= nil) then
                                local data = settings.multidata[k];
                            data.parent = self.bars[settings.multidata[k].parent]
                                v:LoadPosition(data);
                        end
                end
        end
        
        for k,v in pairs(self.windows) do
                v:SetVisible(v.active);
        end
        
        for k,v in pairs(self.skills) do
                if (v.parent ~= nil) then
                        v.parent:AddSkill(v);
                end
                v:RefreshBorders();
        end
        
        for i = 1, vars.effects:GetCount() do
                self:AddEffect(i);
        end
end

-- Swap Slots
function ClassBar:SwapSlots(ref, code)
        
        local swap = nil;
        local swaptype = 1;
        if (not settings.lockslots) then
                for k,v in pairs(self.skills) do
                        if (v.quickslot:GetShortcut():GetData() == code ) then
                                swap = v;
                                swaptype = 1;
                        end
                end
                for k,v in pairs(self.multi) do
                        if (v.quickslot:GetShortcut():GetData() == code ) then
                                --if ((ref:GetParent() == v:GetParent() or ((v:GetParent().exclusive == nil or ref:GetParent().exclusive == nil) or (not ref:GetParent().exclusive and not v:GetParent().exclusive)))) then
                                        swap = v;
                                        swaptype = 2;
                                --end
                        end
                end
        end
        if (swap ~= nil) then
                if (swap:GetParent() == limbo)  then
                        if (not ref:GetParent().exclusive and ref:GetParent() == ref.parent) then
                                        swap:SwapPosition(ref);
                        end
                else
                        if( (ref:GetParent() == swap:GetParent() or ( not ref:GetParent().exclusive and not swap:GetParent().exclusive)) and (ref:GetParent() == ref.parent and swap:GetParent() == swap.parent)) then
                                if (swaptype == 1) then
                                        ref:SwapPosition(swap);
                                else
                                        swap:SwapPosition(ref);
                                end
                        end
                end
        end
        ref:ResetQuickslot();
end

-- Rotate Window
function ClassBar:RotateWindow(ref)
        local rotatebar = { };
        local counter=1
        for k,v in pairs(self.bars) do
                if(v:GetParent() == ref) then
                        v:RotateBar();
                        rotatebar[counter] = v;
                        counter = counter + 1;
                end
        end
        
        for k,v in pairs(self.skills) do
                for i=1, counter do
                        if(v:GetParent() == rotatebar[i]) then
                                v:RotateBar();
                        end
                end
        end
        
        for k,v in pairs(self.multi) do
                for i=1, counter do
                        if(v:GetParent() == rotatebar[i]) then
                                v:RotateBar();
                        end
                end
        end
end

-- Save Settings
function ClassBar:SaveSettings()

        if (self.settings.preconf ~= nil) then
                for k,v in pairs(self.settings.preconf) do
                        v.x, v.y = self.windows[k]:GetPosition();
                        v.rotate = self.windows[k].rotate;
                end
                
                for k,v in pairs(self.settings.userconf) do
                        v.name = self.windows[k].heading;
                        v.x, v.y = self.windows[k]:GetPosition();
                        v.size[1] = self.windows[k].width;
                        v.size[2] = self.windows[k].height;
                        v.rotate = self.windows[k].rotate;
                end
        else
                for k,v in pairs(self.windows) do
                        self.settings[k].x, self.settings[k].y = self.windows[k]:GetPosition();
                        self.settings[k].rotate = self.windows[k].rotate;
                end
        end
    
    local skilldata = { };
    for k,v in pairs(self.skills) do
                if (v.GetParent() == limbo) then
                        skilldata[k] = { pos = {v.x, v.y}, parent = nil };
                else
                        skilldata[k] = { pos = {v.x, v.y}, parent = v.parent.index };
                end
        end
        
        local multidata = { };
        for k,v in pairs(self.multi) do
                if (v.GetParent() == limbo) then
                        multidata[k] = { current = v.currentshortcut, layer = v.currentlayer, lock = v.lock, pos = { v.x, v.y }, parent = nil };
                else
                        multidata[k] = { current = v.currentshortcut, layer = v.currentlayer, lock = v.lock, pos = { v.x, v.y }, parent = v.parent.index };
                end
                
        end
        
        for k,v in pairs(self.windows) do
                v:SetVisible(v.active);
        end

        --Turbine.Shell.WriteLine("KragenBars: " .. settings.class .. " Bars settings saved successfully.");

        settings.multidata = multidata;
        settings.skilldata = skilldata;
        settings.classinfo = self.settings;
        SaveSettings();
end

-- Refresh Player Level and Window and Skill States
function ClassBar:RefreshBars()
        -- Refresh Player Level
        vars.playerlevel = vars.player:GetLevel();
        
        if (self.settings.preconf ~= nil) then
                -- Refresh Windows
                for k,v in pairs(self.settings.preconf) do
                        self.windows[k]:Refresh(v.active);
                end
                
                for k,v in pairs(self.settings.userconf) do
                        self.windows[k]:Refresh(v.active);
                end
        else
                -- Refresh Windows
                for k,v in pairs(self.windows) do
                        self.windows[k]:Refresh(self.settings[k].active);
                end
        end
        
        -- Refresh Skills
        for k,v in pairs(self.skills) do
                self.skills[k]:Refresh();
        end
end

function ClassBar:DefaultPosition()
    -- Refresh Windows
        for k,v in pairs(self.windows) do
        self.windows[k]:DefaultPosition();
        end
end

function ClassBar:RecalculateBorders()
        -- Refresh Skills
        for k,v in pairs(self.skills) do
                v.border:SetPosition(v:PointToScreen(1, 1));
        end
end

function ClassBar:CombatChange(combat)
        for k,v in pairs(self.skills) do
                self.skills[k]:RefreshCombat();
        end
        
        for k,v in pairs(self.multi) do
                self.multi[k]:RefreshCombat();
        end
end

function ClassBar:CreateBar()
        local name = "UserBar"..tostring(settings.userbars + 1);
        local v = nil;
        
        settings.userbars = settings.userbars + 1;
        self.barcount = self.barcount + 1;
        
        v = { id = self.barcount, name = name, defpos = {0,0}, size = {1, 1} ,active = true, x = 800, y = 200+(35*settings.userbars), rotate = 0 };
        self.windows[name] = SkillWindow(v.name, v.defpos[1], v.defpos[2], v.size[1], v.size[2], {active = v.active, x = v.x, y = v.y, rotate = v.rotate});
        self.bars[name] = SkillBar(self.windows[name], 1, 1, 1, 1, true, false);
        
        self.settings.userconf[name] = v;
end

function ClassBar:RenameBar(key, name)
        if (self.windows[key] ~= nil) then
                self.windows[key]:SetName(name);
                self.settings.userconf[key].name = name;
        end
end

function ClassBar:DeleteBar(key)
        local id = nil
        local counter = 1;
        
        if (self.windows[key] ~= nil) then
                id = self.settings.userconf[key].id;
                self.bars[key]:ChangeSize(0,0);
                self.bars[key]:SetVisible(false);
                self.windows[key]:SetVisible(false);
                self.windows[key].dragBar:SetAllowsDragging(false);
                self.windows[key].dragBar:SetAllowsHUDHiding(false);
                self.windows[key].dragBar:SetVisible(false);
                self.windows[key] = nil;
                self.bars[key] = nil;
                self.settings.userconf[key] = nil;
                
                self.barlist = { };
        
                for k,v in pairs (self.settings.userconf) do
                        if (v.id > id) then
                                v.id = id - 1;
                                self.barlist[v.id] = k;
                        end
                        counter = counter + 1;
                end
                
                self.setupbb["checkboxbb"..tostring(counter)]:SetVisible(false);
                self.barcount = self.barcount - 1;
        end
end