NoM0Re
375 lines
14 KiB
Lua
375 lines
14 KiB
Lua
if not WeakAuras.IsLibsOK() then return end
|
|
local AddonName, OptionsPrivate = ...
|
|
|
|
-- Lua APIs
|
|
local rad = rad
|
|
|
|
-- WoW APIs
|
|
local CreateFrame = CreateFrame
|
|
|
|
local AceGUI = LibStub("AceGUI-3.0")
|
|
|
|
local WeakAuras = WeakAuras
|
|
local L = WeakAuras.L
|
|
|
|
local modelPicker
|
|
|
|
local function GetAll(baseObject, path, property, default)
|
|
local valueFromPath = OptionsPrivate.Private.ValueFromPath
|
|
if not property then
|
|
return default
|
|
end
|
|
|
|
local result = default
|
|
local first = true
|
|
for child in OptionsPrivate.Private.TraverseLeafsOrAura(baseObject) do
|
|
local childObject = valueFromPath(child, path)
|
|
if childObject and childObject[property] then
|
|
if first then
|
|
result = childObject[property]
|
|
first = false
|
|
else
|
|
if result ~= childObject[property] then
|
|
return default
|
|
end
|
|
end
|
|
end
|
|
end
|
|
return result
|
|
end
|
|
|
|
local function ConstructModelPicker(frame)
|
|
local function RecurseSetFilter(tree, filter)
|
|
for k, v in ipairs(tree) do
|
|
if v.children == nil and v.text then
|
|
v.visible = not filter or filter == "" or v.text:find(filter, 1, true) ~= nil
|
|
else
|
|
RecurseSetFilter(v.children, filter)
|
|
end
|
|
end
|
|
end
|
|
|
|
local group = AceGUI:Create("SimpleGroup");
|
|
group.frame:SetParent(frame);
|
|
group.frame:SetPoint("BOTTOMRIGHT", frame, "BOTTOMRIGHT", -17, 87);
|
|
group.frame:SetPoint("TOPLEFT", frame, "TOPLEFT", 17, -63);
|
|
group.frame:Hide();
|
|
group:SetLayout("flow");
|
|
|
|
local filterInput = CreateFrame("EditBox", "WeakAurasModelFilterInput", group.frame, "WA_InputBoxTemplate")
|
|
filterInput:SetAutoFocus(false)
|
|
filterInput:SetTextInsets(16, 20, 0, 0)
|
|
|
|
filterInput.Instructions = filterInput:CreateFontString(nil, "ARTWORK", "GameFontDisableSmall")
|
|
filterInput.Instructions:SetText(SEARCH)
|
|
filterInput.Instructions:SetPoint("TOPLEFT", filterInput, "TOPLEFT", 16, 0)
|
|
filterInput.Instructions:SetPoint("BOTTOMRIGHT", filterInput, "BOTTOMRIGHT", -20, 0)
|
|
filterInput.Instructions:SetTextColor(0.35, 0.35, 0.35)
|
|
filterInput.Instructions:SetJustifyH("LEFT")
|
|
filterInput.Instructions:SetJustifyV("MIDDLE")
|
|
|
|
filterInput.searchIcon = filterInput:CreateTexture(nil, "OVERLAY")
|
|
filterInput.searchIcon:SetTexture("Interface\\Common\\UI-Searchbox-Icon")
|
|
filterInput.searchIcon:SetVertexColor(0.6, 0.6, 0.6)
|
|
filterInput.searchIcon:SetSize(14, 14)
|
|
filterInput.searchIcon:SetPoint("LEFT", 0, -2)
|
|
|
|
filterInput.clearButton = CreateFrame("Button", nil, filterInput)
|
|
filterInput.clearButton:SetSize(14, 14)
|
|
filterInput.clearButton:SetPoint("RIGHT", -3, 0)
|
|
|
|
filterInput.clearButton.texture = filterInput.clearButton:CreateTexture()
|
|
filterInput.clearButton.texture:SetTexture("Interface\\FriendsFrame\\ClearBroadcastIcon")
|
|
filterInput.clearButton.texture:SetAlpha(0.5)
|
|
filterInput.clearButton.texture:SetSize(17, 17)
|
|
filterInput.clearButton.texture:SetPoint("CENTER", 0, 0)
|
|
|
|
filterInput.clearButton:SetScript("OnEnter", function(self) self.texture:SetAlpha(1.0) end)
|
|
filterInput.clearButton:SetScript("OnLeave", function(self) self.texture:SetAlpha(0.5) end)
|
|
filterInput.clearButton:SetScript("OnMouseDown", function(self) if self:IsEnabled() then self.texture:SetPoint("CENTER", 1, -1) end end)
|
|
filterInput.clearButton:SetScript("OnMouseUp", function(self) self.texture:SetPoint("CENTER") end)
|
|
filterInput.clearButton:SetScript("OnClick", function(self)
|
|
local editBox = self:GetParent()
|
|
editBox:SetText("")
|
|
editBox:ClearFocus()
|
|
end)
|
|
filterInput:SetScript("OnEditFocusLost", function(self)
|
|
if self:GetText() == "" then
|
|
self.searchIcon:SetVertexColor(0.6, 0.6, 0.6)
|
|
self.clearButton:Hide()
|
|
end
|
|
end)
|
|
filterInput:SetScript("OnEditFocusGained", function(self)
|
|
self.searchIcon:SetVertexColor(1.0, 1.0, 1.0)
|
|
self.clearButton:Show()
|
|
end)
|
|
filterInput:HookScript("OnTextChanged", function(self)
|
|
if not self:HasFocus() and self:GetText() == "" then
|
|
self.searchIcon:SetVertexColor(0.6, 0.6, 0.6)
|
|
self.clearButton:Hide()
|
|
else
|
|
self.searchIcon:SetVertexColor(1.0, 1.0, 1.0)
|
|
self.clearButton:Show()
|
|
end
|
|
if self:GetText() == "" then
|
|
self.Instructions:Show()
|
|
else
|
|
self.Instructions:Hide()
|
|
end
|
|
local filterText = filterInput:GetText()
|
|
RecurseSetFilter(group.modelTree.tree, filterText)
|
|
group.modelTree.filter = filterText ~= nil and filterText ~= ""
|
|
group.modelTree:RefreshTree()
|
|
end)
|
|
filterInput:SetHeight(15)
|
|
filterInput:SetPoint("BOTTOMRIGHT", group.frame, "TOPRIGHT", -3, 5)
|
|
filterInput:SetWidth(200)
|
|
filterInput:SetFont(STANDARD_TEXT_FONT, 10)
|
|
group.frame.filterInput = filterInput
|
|
|
|
local modelPickerZ = AceGUI:Create("Slider");
|
|
modelPickerZ:SetSliderValues(-20, 20, 0.05);
|
|
modelPickerZ:SetLabel(L["Z Offset"]);
|
|
modelPickerZ.frame:SetParent(group.frame);
|
|
modelPickerZ:SetCallback("OnValueChanged", function()
|
|
group:Pick(nil, modelPickerZ:GetValue());
|
|
end);
|
|
|
|
local modelPickerX = AceGUI:Create("Slider");
|
|
modelPickerX:SetSliderValues(-20, 20, 0.05);
|
|
modelPickerX:SetLabel(L["X Offset"]);
|
|
modelPickerX.frame:SetParent(group.frame);
|
|
modelPickerX:SetCallback("OnValueChanged", function()
|
|
group:Pick(nil, nil, modelPickerX:GetValue());
|
|
end);
|
|
|
|
local modelPickerY = AceGUI:Create("Slider");
|
|
modelPickerY:SetSliderValues(-20, 20, 0.05);
|
|
modelPickerY:SetLabel(L["Y Offset"]);
|
|
modelPickerY.frame:SetParent(group.frame);
|
|
modelPickerY:SetCallback("OnValueChanged", function()
|
|
group:Pick(nil, nil, nil, modelPickerY:GetValue());
|
|
end);
|
|
|
|
local modelPickerRotation = AceGUI:Create("Slider");
|
|
modelPickerRotation:SetSliderValues(0, 360, 0.05);
|
|
modelPickerRotation:SetLabel(L["Rotation"]);
|
|
modelPickerRotation.frame:SetParent(group.frame);
|
|
modelPickerRotation:SetCallback("OnValueChanged", function()
|
|
group:Pick(nil, nil, nil, nil, modelPickerRotation:GetValue());
|
|
end);
|
|
|
|
local modelTree = AceGUI:Create("WeakAurasTreeGroup");
|
|
group.modelTree = modelTree;
|
|
group.frame:SetScript("OnSizeChanged", function()
|
|
local frameWidth = frame:GetWidth();
|
|
local sliderWidth = (frameWidth - 50) / 4;
|
|
local narrowSliderWidth = (frameWidth - 50) / 7;
|
|
|
|
modelTree:SetTreeWidth(frameWidth - 370);
|
|
|
|
modelPickerZ.frame:SetPoint("bottomleft", frame, "bottomleft", 15, 43);
|
|
modelPickerZ.frame:SetPoint("bottomright", frame, "bottomleft", 15 + sliderWidth, 43);
|
|
|
|
modelPickerX.frame:SetPoint("bottomleft", frame, "bottomleft", 25 + sliderWidth, 43);
|
|
modelPickerX.frame:SetPoint("bottomright", frame, "bottomleft", 25 + (2 * sliderWidth), 43);
|
|
|
|
modelPickerY.frame:SetPoint("bottomleft", frame, "bottomleft", 35 + (2 * sliderWidth), 43);
|
|
modelPickerY.frame:SetPoint("bottomright", frame, "bottomleft", 35 + (3 * sliderWidth), 43);
|
|
|
|
modelPickerRotation.frame:SetPoint("bottomleft", frame, "bottomleft", 45 + (3 * sliderWidth), 43);
|
|
modelPickerRotation.frame:SetPoint("bottomright", frame, "bottomleft", 45 + (4 * sliderWidth), 43);
|
|
end);
|
|
group:SetLayout("fill");
|
|
modelTree:SetTree(WeakAuras.ModelPaths);
|
|
modelTree:SetCallback("OnGroupSelected", function(self, event, value, fileId)
|
|
local path = string.gsub(value, "\001", "/");
|
|
if(string.lower(string.sub(path, -3, -1)) == ".m2") then
|
|
group:Pick(path, fileId);
|
|
end
|
|
end);
|
|
group:AddChild(modelTree);
|
|
|
|
local model = CreateFrame("PlayerModel", nil, group.content);
|
|
model:SetAllPoints(modelTree.content);
|
|
model:SetFrameStrata("FULLSCREEN");
|
|
group.model = model;
|
|
|
|
local startX, rotation
|
|
local function OnUpdateScript()
|
|
local uiScale, x = UIParent:GetEffectiveScale(), GetCursorPosition()
|
|
local screenW, screenH = GetScreenWidth(), GetScreenHeight()
|
|
local diffX = startX/uiScale - x/uiScale
|
|
rotation = (rotation + 180 / screenW * diffX) % 360
|
|
model:SetFacing(rad(rotation))
|
|
end
|
|
model:EnableMouse()
|
|
model:SetScript("OnMouseDown", function(self)
|
|
startX = GetCursorPosition()
|
|
rotation = group.selectedValues.rotation or 0
|
|
self:SetScript("OnUpdate", OnUpdateScript)
|
|
end)
|
|
model:SetScript("OnMouseUp", function(self)
|
|
self:SetScript("OnUpdate", nil)
|
|
group:Pick(nil, nil, nil, nil, rotation)
|
|
end)
|
|
|
|
local function SetOnObject(object, model_path, model_z, model_x, model_y, rotation)
|
|
if model_path then
|
|
object.model_path = model_path
|
|
end
|
|
if model_z then
|
|
object.model_z = model_z
|
|
end
|
|
if model_x then
|
|
object.model_x = model_x
|
|
end
|
|
if model_y then
|
|
object.model_y = model_y
|
|
end
|
|
if rotation then
|
|
object.rotation = rotation
|
|
end
|
|
end
|
|
|
|
function group.Pick(self, model_path, model_z, model_x, model_y, rotation)
|
|
local valueFromPath = OptionsPrivate.Private.ValueFromPath
|
|
|
|
self.selectedValues.model_path = model_path or self.selectedValues.model_path
|
|
self.selectedValues.model_x = model_x or self.selectedValues.model_x
|
|
self.selectedValues.model_y = model_y or self.selectedValues.model_y
|
|
self.selectedValues.model_z = model_z or self.selectedValues.model_z
|
|
self.selectedValues.rotation = rotation or self.selectedValues.rotation
|
|
|
|
WeakAuras.SetModel(self.model, self.selectedValues.model_path)
|
|
|
|
self.model:SetPosition(self.selectedValues.model_z, self.selectedValues.model_x, self.selectedValues.model_y);
|
|
self.model:SetFacing(rad(self.selectedValues.rotation));
|
|
|
|
for child in OptionsPrivate.Private.TraverseLeafsOrAura(self.baseObject) do
|
|
local object = valueFromPath(child, self.path)
|
|
if(object) then
|
|
SetOnObject(object, model_path, model_z, model_x, model_y, rotation)
|
|
WeakAuras.Add(child)
|
|
WeakAuras.UpdateThumbnail(child)
|
|
end
|
|
end
|
|
end
|
|
|
|
function group.Open(self, baseObject, path)
|
|
local valueFromPath = OptionsPrivate.Private.ValueFromPath
|
|
|
|
self.baseObject = baseObject
|
|
self.path = path
|
|
self.selectedValues = {}
|
|
|
|
self.selectedValues.model_path = GetAll(baseObject, path, "model_path", "spells/arcanepower_state_chest.m2")
|
|
|
|
WeakAuras.SetModel(self.model, self.selectedValues.model_path)
|
|
|
|
self.selectedValues.model_x = GetAll(baseObject, path, "model_x", 0)
|
|
self.selectedValues.model_y = GetAll(baseObject, path, "model_y", 0)
|
|
self.selectedValues.model_z = GetAll(baseObject, path, "model_z", 0)
|
|
self.selectedValues.rotation = GetAll(baseObject, path, "rotation", 0)
|
|
|
|
self.model:SetPosition(self.selectedValues.model_z, self.selectedValues.model_x, self.selectedValues.model_y);
|
|
self.model:SetFacing(rad(self.selectedValues.rotation));
|
|
modelPickerZ:SetValue(self.selectedValues.model_z);
|
|
modelPickerZ.editbox:SetText(("%.2f"):format(self.selectedValues.model_z));
|
|
modelPickerX:SetValue(self.selectedValues.model_x);
|
|
modelPickerX.editbox:SetText(("%.2f"):format(self.selectedValues.model_x));
|
|
modelPickerY:SetValue(self.selectedValues.model_y);
|
|
modelPickerY.editbox:SetText(("%.2f"):format(self.selectedValues.model_y));
|
|
modelPickerRotation:SetValue(self.selectedValues.rotation);
|
|
modelPickerRotation.editbox:SetText(("%.2f"):format(self.selectedValues.rotation));
|
|
|
|
if(baseObject.controlledChildren) then
|
|
self.givenModel = {};
|
|
self.givenZ = {};
|
|
self.givenX = {};
|
|
self.givenY = {};
|
|
self.givenRotation = {};
|
|
for child in OptionsPrivate.Private.TraverseLeafs(baseObject) do
|
|
local childId = child.id
|
|
local object = valueFromPath(child, path)
|
|
if(object) then
|
|
self.givenModel[childId] = object.model_path;
|
|
self.givenZ[childId] = object.model_z;
|
|
self.givenX[childId] = object.model_x;
|
|
self.givenY[childId] = object.model_y;
|
|
self.givenRotation[childId] = object.rotation;
|
|
end
|
|
end
|
|
else
|
|
local object = valueFromPath(baseObject, path)
|
|
|
|
self.givenModel = object.model_path;
|
|
self.givenZ = object.model_z;
|
|
self.givenX = object.model_x;
|
|
self.givenY = object.model_y;
|
|
self.givenRotation = object.rotation;
|
|
end
|
|
frame.window = "model";
|
|
frame:UpdateFrameVisible()
|
|
end
|
|
|
|
function group.Close()
|
|
frame.window = "default"
|
|
frame:UpdateFrameVisible()
|
|
WeakAuras.FillOptions()
|
|
end
|
|
|
|
function group.CancelClose()
|
|
local valueFromPath = OptionsPrivate.Private.ValueFromPath
|
|
if(group.baseObject.controlledChildren) then
|
|
for child in OptionsPrivate.Private.TraverseLeafs(group.baseObject) do
|
|
local childId = child.id
|
|
local object = valueFromPath(child, group.path)
|
|
if(object) then
|
|
object.model_path = group.givenModel[childId];
|
|
object.model_z = group.givenZ[childId];
|
|
object.model_x = group.givenX[childId];
|
|
object.model_y = group.givenY[childId];
|
|
object.rotation = group.givenRotation[childId];
|
|
WeakAuras.Add(child);
|
|
WeakAuras.UpdateThumbnail(child);
|
|
end
|
|
end
|
|
else
|
|
local object = valueFromPath(group.baseObject, group.path)
|
|
|
|
if(object) then
|
|
object.model_path = group.givenModel
|
|
object.model_z = group.givenZ
|
|
object.model_x = group.givenX
|
|
object.model_y = group.givenY
|
|
object.rotation = group.givenRotation
|
|
WeakAuras.Add(group.baseObject);
|
|
WeakAuras.UpdateThumbnail(group.baseObject);
|
|
end
|
|
end
|
|
group.Close();
|
|
end
|
|
|
|
local cancel = CreateFrame("Button", nil, group.frame, "UIPanelButtonTemplate");
|
|
cancel:SetScript("OnClick", group.CancelClose);
|
|
cancel:SetPoint("BOTTOMRIGHT", frame, "BOTTOMRIGHT", -27, 20);
|
|
cancel:SetHeight(20);
|
|
cancel:SetWidth(100);
|
|
cancel:SetText(L["Cancel"]);
|
|
|
|
local close = CreateFrame("Button", nil, group.frame, "UIPanelButtonTemplate");
|
|
close:SetScript("OnClick", group.Close);
|
|
close:SetPoint("RIGHT", cancel, "LEFT", -10, 0);
|
|
close:SetHeight(20);
|
|
close:SetWidth(100);
|
|
close:SetText(L["Okay"]);
|
|
|
|
return group
|
|
end
|
|
|
|
function OptionsPrivate.ModelPicker(frame, noConstruct)
|
|
modelPicker = modelPicker or (not noConstruct and ConstructModelPicker(frame))
|
|
return modelPicker
|
|
end
|