Posts by mokaByls

local pluginName    = select(1,...);
local componentName = select(2,...);
local signalTable   = select(3,...);
local pluginHandle  = select(4,...);

require("gma3_helpers")

local VERBOSE = false

local function ltrim(text)
    return text:match'^%s*(.*)'
end

local function split(str, sep)
    local result = {}
    local regex = ("([^%s]+)"):format(sep)
    for each in str:gmatch(regex) do
        table.insert(result, each)
    end
    return result
end
 
local function isSupportAttribute(sfi, attribute)
    local attribIndex = GetAttributeIndex(attribute);
    local uiChannelIndex = GetUIChannelIndex(sfi, attribIndex);
    
    local boolInfo = "No"
    if uiChannelIndex ~= nil then
        boolInfo = "Yes"
    end

    -- Echo(string.format("SubFixtureIndex \"%s\" support the \"%s\" attribute ? %s", sfi, attribute, boolInfo))
    -- Echo(string.format("SubFixtureIndex \"%s\" | attribIndex : %s", sfi, attribIndex))
    -- Echo("---")

    return uiChannelIndex ~= nil
end

local function hasChildren(fixture)
    local children = fixture:Children()
    return #children > 0
end

local function extractPlainSelection(attributes, allowSubFixture)
    local fixtures = {}
    local fixtureIds = {}
    local subFixtureIndexes = {}
    local uniqueValues = {}

    local subfixtureIndex = SelectionFirst(true);
    repeat
        local fixture = GetSubfixture(subfixtureIndex)
        local fixtureId = fixture.No
        
        local inner = fixture:Get('Fixture')
        if inner.No ~= fixture.No then
            fixtureId = string.format("%d.%d", inner.No, fixture.No)
        end

        for i, attribute in ipairs(attributes) do
            local supports = isSupportAttribute(subfixtureIndex, attribute)
            
            if VERBOSE then
                Echo(string.format("Fixture \"%s\" support the \"%s\" attribute ? %s", fixture.name, attribute, (supports and "Yes") or "No"))
            end

            if supports == false and hasChildren(fixture) then
                for _, child in ipairs(fixture:Children()) do
                    local subFixtureId = string.format("%d.%d", fixtureId, child.No)
                    local childSupports = isSupportAttribute(child:Get("SubfixtureIndex"), attribute)

                    if VERBOSE then
                        Echo(string.format("Child Fixture \"%s.%s\" (%s.%d) [%d] support the \"%s\" attribute ? %s", fixture.name, child.name, fixtureId, child.No, child:Get("SubfixtureIndex"), attribute, (childSupports and "Yes") or "No"))
                    end

                    if childSupports then
                        if allowSubFixture then
                            if uniqueValues[subFixtureId] == nil then
                                fixtures[#fixtures + 1] = child
                                fixtureIds[#fixtureIds + 1] = subFixtureId
                                subFixtureIndexes[#subFixtureIndexes + 1] = child:Get("SubfixtureIndex")
                                uniqueValues[subFixtureId] = true
                            end
                        elseif uniqueValues[fixtureId] == nil then
                            fixtures[#fixtures + 1] = fixture
                            fixtureIds[#fixtureIds + 1] = fixtureId
                            subFixtureIndexes[#subFixtureIndexes + 1] = subfixtureIndex
                            uniqueValues[fixtureId] = true
                        end
                    end
                end
            elseif supports and allowSubFixture == false and inner.No ~= fixture.No then
                if uniqueValues[inner.No] == nil then
                    fixtures[#fixtures + 1] = inner
                    fixtureIds[#fixtureIds + 1] = inner.No
                    subFixtureIndexes[#subFixtureIndexes + 1] = inner:Get("SubfixtureIndex")
                    uniqueValues[inner.No] = true
                end
            elseif supports and uniqueValues[fixtureId] == nil then
                fixtures[#fixtures + 1] = fixture
                fixtureIds[#fixtureIds + 1] = fixtureId
                subFixtureIndexes[#subFixtureIndexes + 1] = subfixtureIndex
                uniqueValues[fixtureId] = true
            end
        end

        subfixtureIndex = SelectionNext(subfixtureIndex)
    until not subfixtureIndex;

    return {
        key = table.concat( fixtureIds, "_"),
        fixtures = fixtures,
        fids = fixtureIds,
        subFixtureIndexes = subFixtureIndexes
    }
end

local function extractPlainSelectionIds(attributes, allowSubFixture)
    local values = extractPlainSelection(attributes, allowSubFixture)
    return values.fids
end

local function extractSelection(attributes, allowSubFixture)
    local uniqueValues = {}
    local sections = {
        all = extractPlainSelection(attributes, allowSubFixture),
        sections = {}
    }

    while true do
        Cmd("Next")
        local selection = extractPlainSelection(attributes, allowSubFixture)
        if uniqueValues[selection.key] ~= nil then
            break
        end
        sections['sections'][#sections['sections'] + 1] = selection
        uniqueValues[selection.key] = true
    end

    return sections
end

local function Main()
    if SelectionCount() == 0 then
        MessageBox({
            title = 'Selection',
            message = "You need to select at least one fixture. Try Again !",
            backColor = "Global.Alert",
            timeout = 10000,
            commands = {{name = "Ok", value = 0}}
        })
        return
    end

    local input = TextInput("Please enter attribute", "ColorRGB_R")
    
    local parts = split(input, ',')
    local attributes = {}
    for i, part in ipairs(parts) do
        attributes[#attributes + 1] = ltrim(part)
    end

    if Confirm("Extract Ids", "Would you like the extraction to be grouped by your MATricks configuration ? Tap Ok for yes.", nil, true) then
        local selection = {}
        if Confirm("Extract Ids", "Do you want the extraction to include the sub fixtures or just the parent fixtures ? Tap Ok for yes.", nil, true) then
            selection = extractSelection(attributes, true)
        else
            selection = extractSelection(attributes, false)
        end

        for i, group in ipairs(selection.sections) do
            Echo("===================== Group n°" .. i .. " =====================")
            if group.fids ~= nil then
                for i, fid in ipairs(group.fids) do
                    Echo(string.format("\tFixture \"%s\" (Fid: %s | SubFixtureId: %s)", group.fixtures[i].name, fid, group.subFixtureIndexes[i]))
                end
            end
        end

    else
        local selection = {}
        if Confirm("Extract Ids", "Do you want the extraction to include the sub fixtures or just the parent fixtures ? Tap Ok for yes.", nil, true) then
            selection = extractPlainSelection(attributes, true)
        else
            selection = extractPlainSelection(attributes, false)
        end

        if selection.fids ~= nil then
            for i, fid in ipairs(selection.fids) do
                Echo(string.format("Fixture \"%s\" (Fid: %s | SubFixtureId: %s)", selection.fixtures[i].name, fid, selection.subFixtureIndexes[i]))
            end
        end
    end
end

return Main

Cmd('Next')

local selection = {}
local indexes = {}
local done = false
while true do
    local first = SelectionFirst()
    local currentFixtureIndex = SelectionFirst()

    for _, v in ipairs(indexes) do 
        if (v == first) then
            done = true
        end
    end

    if done then
        break
    end

    local subFixtures = {}

    repeat
        local subFixture = GetSubfixture(currentFixtureIndex);
        if subFixture:GetClass() == 'Fixture' then
            table.insert(subFixtures, subFixture.index)
        end
        currentFixtureIndex = SelectionNext(currentFixtureIndex)
    until not currentFixtureIndex

    table.insert(selection, subFixtures)
    table.insert(indexes, first)
    Cmd('Next')
end

Cmd("ClearSelection")

-- pseudo code to illustrate my goal
for _, fixtureId in ipairs(selection) do
    Cmd(string.format("Fixture %d", fixtureId))
    Cmd("Attribute \"Tilt\" At Absolute +45")
    Cmd("ClearSelection")
end