FModel/FModel/Views/Snooper/Animations/Animation.cs

using System;
using System.Collections.Generic;
using System.IO;
using System.Numerics;
using CUE4Parse_Conversion;
using CUE4Parse_Conversion.Animations.PSA;
using CUE4Parse.UE4.Assets.Exports;
using CUE4Parse.UE4.Objects.Core.Misc;
using FModel.Settings;
using FModel.Views.Snooper.Models;
using ImGuiNET;

namespace FModel.Views.Snooper.Animations;

public class Animation : IDisposable
{
    private readonly UObject _export;

    public readonly CAnimSet UnrealAnim;
    public readonly string Path;
    public readonly string Name;
    public readonly Sequence[] Sequences;
    public readonly float StartTime;                // Animation Start Time
    public readonly float EndTime;                  // Animation End Time
    public readonly float TotalElapsedTime;         // Animation Max Time
    public readonly Dictionary<int, float> Framing;

    public bool IsActive;
    public bool IsSelected;

    public readonly List<FGuid> AttachedModels;

    public Animation(UObject export)
    {
        _export = export;
        Path = _export.GetPathName();
        Name = _export.Name;
        Sequences = Array.Empty<Sequence>();
        Framing = new Dictionary<int, float>();
        AttachedModels = new List<FGuid>();
    }

    public Animation(UObject export, CAnimSet animSet) : this(export)
    {
        UnrealAnim = animSet;

        Sequences = new Sequence[UnrealAnim.Sequences.Count];
        for (int i = 0; i < Sequences.Length; i++)
        {
            Sequences[i] = new Sequence(UnrealAnim.Sequences[i]);
            EndTime = Sequences[i].EndTime;
        }

        TotalElapsedTime = animSet.TotalAnimTime;
        if (Sequences.Length > 0)
            StartTime = Sequences[0].StartTime;
    }

    public Animation(UObject export, CAnimSet animSet, params FGuid[] animatedModels) : this(export, animSet)
    {
        AttachedModels.AddRange(animatedModels);
    }

    public void TimeCalculation(float elapsedTime)
    {
        for (int i = 0; i < Sequences.Length; i++)
        {
            var sequence = Sequences[i];
            if (elapsedTime <= sequence.EndTime && elapsedTime >= sequence.StartTime)
            {
                Framing[i] = (elapsedTime - sequence.StartTime) / sequence.TimePerFrame;
            }
            else Framing.Remove(i);
        }

        if (elapsedTime >= TotalElapsedTime)
            Framing.Clear();
    }

    public void Dispose()
    {
        AttachedModels.Clear();
    }

    public void ImGuiAnimation(Snooper s, Save saver, ImDrawListPtr drawList, ImFontPtr fontPtr, Vector2 timelineP0, Vector2 treeP0, Vector2 timeStep, Vector2 timeRatio, float y, float t, int i)
    {
        var name = $"{Name}##{i}";
        var p1 = new Vector2(timelineP0.X + StartTime * timeRatio.X + t, y + t);
        var p2 = new Vector2(timelineP0.X + EndTime * timeRatio.X - t, y + timeStep.Y - t);

        ImGui.SetCursorScreenPos(p1);
        ImGui.InvisibleButton($"timeline_sequencetracker_{name}", new Vector2(EndTime * timeRatio.X - t, timeStep.Y - t), ImGuiButtonFlags.MouseButtonLeft);
        IsActive = ImGui.IsItemActive();
        IsSelected = s.Renderer.Options.SelectedAnimation == i;
        if (ImGui.IsItemClicked(ImGuiMouseButton.Left))
        {
            s.Renderer.Options.SelectAnimation(i);
        }
        Popup(s, saver, i);

        drawList.AddRectFilled(p1, p2, IsSelected ? 0xFF48B048 : 0xFF175F17, 5.0f, ImDrawFlags.RoundCornersTop);
        for (int j = 0; j < Sequences.Length; j++)
        {
            Sequences[j].DrawSequence(drawList, fontPtr, timelineP0.X, p2, timeStep, timeRatio, t, IsSelected);
        }

        ImGui.SetCursorScreenPos(treeP0 with { Y = p1.Y });
        if (ImGui.Selectable(name, s.Renderer.Options.SelectedAnimation == i, ImGuiSelectableFlags.SpanAllColumns, new Vector2(p1.X - treeP0.X, timeStep.Y - t - t)))
        {
            s.Renderer.Options.SelectAnimation(i);
        }
        Popup(s, saver, i);
    }

    private void Popup(Snooper s, Save saver, int i)
    {
        SnimGui.Popup(() =>
        {
            s.Renderer.Options.SelectAnimation(i);
            if (ImGui.BeginMenu("Animate"))
            {
                foreach ((var guid, var model) in s.Renderer.Options.Models)
                {
                    var selected = AttachedModels.Contains(guid);
                    if (model is SkeletalModel skeletalModel && ImGui.MenuItem(model.Name, null, selected, !skeletalModel.Skeleton.IsAnimated || selected))
                    {
                        if (selected) AttachedModels.Remove(guid); else AttachedModels.Add(guid);
                        skeletalModel.Skeleton.ResetAnimatedData(true);
                        if (!selected) skeletalModel.Skeleton.Animate(UnrealAnim);
                    }
                }
                ImGui.EndMenu();
            }
            if (ImGui.MenuItem("Save"))
            {
                s.WindowShouldFreeze(true);
                saver.Value = new Exporter(_export).TryWriteToDir(new DirectoryInfo(UserSettings.Default.ModelDirectory), out saver.Label, out saver.Path);
                s.WindowShouldFreeze(false);
            }
            ImGui.Separator();
            if (ImGui.MenuItem("Copy Path to Clipboard")) ImGui.SetClipboardText(Path);
        });
    }
}