Moho Animation Forum

Code: Select all

-- cc_reverse_layer_keyframes.lua
-- Adds Point Curvature channel support (curvature/weight/offset)

ScriptName = "CC_REVERSE_SELECTED_KEYFRAMES_FULL"

CC_REVERSE_SELECTED_KEYFRAMES_FULL = {}

function CC_REVERSE_SELECTED_KEYFRAMES_FULL:Name()
    return "CC: Reverse Selected Keyframes (Full Auto)"
end

function CC_REVERSE_SELECTED_KEYFRAMES_FULL:Version()
    return "1.3"
end

function CC_REVERSE_SELECTED_KEYFRAMES_FULL:Description()
    return "Reverse selected keyframes (transform, points, bones, switch, curvature) automatically at current cursor position."
end

function CC_REVERSE_SELECTED_KEYFRAMES_FULL:Creator()
    return "Based on Crashcore | Expanded by OpenAI"
end

function CC_REVERSE_SELECTED_KEYFRAMES_FULL:UILabel()
    return "Reverse Selected Keyframes"
end

function CC_REVERSE_SELECTED_KEYFRAMES_FULL:IsEnabled(moho)
    return moho.layer ~= nil
end

local function AddIfValid(vecList, vec)
    if vec ~= nil then
        table.insert(vecList, vec)
    end
end

-- Collect all relevant anim channels (including curvature)
local function GetAllAnimVecs(moho, layer)
    local vecs = {}

    -- Layer standard transforms / props
    AddIfValid(vecs, layer.fTranslation)
    AddIfValid(vecs, layer.fRotationX)
    AddIfValid(vecs, layer.fRotationY)
    AddIfValid(vecs, layer.fRotationZ)
    AddIfValid(vecs, layer.fScale)
    AddIfValid(vecs, layer.fShear)
    AddIfValid(vecs, layer.fFlipH)
    AddIfValid(vecs, layer.fFlipV)
    AddIfValid(vecs, layer.fVisibility)
    AddIfValid(vecs, layer.fAlpha)
    AddIfValid(vecs, layer.fBlur)

    -- Vector layer: points & curvature
    local mesh = moho:Mesh()
    if mesh then
        -- Point positions
        for i = 0, mesh:CountPoints() - 1 do
            local pt = mesh:Point(i)
            AddIfValid(vecs, pt.fAnimPos)
        end

        -- Add the composite Point Curvature channel (covers curvature/weight/offset subchannels)
        if layer.Channel then
            local curveChan = layer:Channel(MOHO.CHANNEL_CURVE) -- selection-based multi-subchannel
            AddIfValid(vecs, curveChan)
        end

        for c = 0, mesh:CountCurves() - 1 do
            local curve = mesh:Curve(c)
            if curve and curve.CountPoints then
                for pid = 0, curve:CountPoints() - 1 do
                    -- M_Curve:Curvature(ptID) -> AnimVal
                    AddIfValid(vecs, curve:Curvature(pid))
                end
            end
        end
    end

    -- Bone layer
    if layer:IsBoneType() then
        local skel = moho:Skeleton()
        if skel then
            for i = 0, skel:CountBones() - 1 do
                local bone = skel:Bone(i)
                AddIfValid(vecs, bone.fAnimPos)
                AddIfValid(vecs, bone.fAnimAngle)
                AddIfValid(vecs, bone.fAnimScale)
                AddIfValid(vecs, bone.fIKLock)
                AddIfValid(vecs, bone.fIKParentTarget)
                AddIfValid(vecs, bone.fIKGlobalAngle)
            end
        end
    end

    -- Switch layer
    if layer:LayerType() == MOHO.LT_SWITCH then
        local switch = moho:LayerAsSwitch(layer)
        AddIfValid(vecs, switch:SwitchValues())
    end

    return vecs
end

-- Find the range of selected keyframes across all channels
local function GetSelectedKeyframeRange(vecs, moho)
    local first, last
    local doc = moho.document
    local s, e = doc:StartFrame(), doc:EndFrame()
    for _, vec in ipairs(vecs) do
        for f = s, e do
            if vec:HasKey(f) and vec:IsKeySelected(f) then
                if not first or f < first then first = f end
                if not last or f > last then last = f end
            end
        end
    end
    return first, last
end

-- Mirror the selected keys into a reversed copy starting at pasteStart
local function ReverseSelectedKeyframes(vecs, first, last, pasteStart)
    local duration = last - first + 1
    for _, vec in ipairs(vecs) do
        for f = first, last do
            if vec:HasKey(f) and vec:IsKeySelected(f) then
                local offset = f - first
                local destFrame = pasteStart + (duration - offset - 1)
                vec:AddKey(destFrame)
                vec:SetValue(destFrame, vec:GetValue(f))
            end
        end
    end
end

function CC_REVERSE_SELECTED_KEYFRAMES_FULL:Run(moho)
    local layer = moho.layer
    if not layer then return end

    local vecs = GetAllAnimVecs(moho, layer)
    local firstKey, lastKey = GetSelectedKeyframeRange(vecs, moho)
    if not firstKey or not lastKey or firstKey == lastKey then return end

    local pasteStart = moho.frame

    moho.document:PrepUndo(layer)
    ReverseSelectedKeyframes(vecs, firstKey, lastKey, pasteStart)
    moho.document:SetDirty()
end