rimworld-animation-studio/Assets/Scripts/AnimationComponents/AnimationClips/PawnAnimationClip.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Xml;
using System.Xml.Serialization;
using UnityEngine;

namespace RimWorldAnimationStudio
{
    public class PawnAnimationClip : AnimationClip
    {
        [XmlArray("addons"), XmlArrayItem("li")] public List<ActorAddon> _addons = new List<ActorAddon>();
        [XmlIgnore] public List<ActorAddon> addons = new List<ActorAddon>();

        [XmlAttribute("Class")] public string className = "Rimworld_Animations.PawnAnimationClip";
        [XmlArray("keyframes"), XmlArrayItem("li")] public List<PawnKeyframe> keyframes = new List<PawnKeyframe>();

        [XmlIgnore] public Dictionary<int, bool> quiver = new Dictionary<int, bool>();
        [XmlIgnore] public SimpleCurve GenitalAngle = new SimpleCurve();
        [XmlIgnore] public SimpleCurve BodyAngle = new SimpleCurve();
        [XmlIgnore] public SimpleCurve HeadAngle = new SimpleCurve();
        [XmlIgnore] public SimpleCurve HeadBob = new SimpleCurve();
        [XmlIgnore] public SimpleCurve BodyOffsetX = new SimpleCurve();
        [XmlIgnore] public SimpleCurve BodyOffsetZ = new SimpleCurve();
        [XmlIgnore] public SimpleCurve HeadFacing = new SimpleCurve();
        [XmlIgnore] public SimpleCurve BodyFacing = new SimpleCurve();

        public override int duration { get { return keyframes.Max(x => x.atTick.Value); } }

        public override void BuildSimpleCurves()
        {
            BodyAngle.Clear();
            HeadAngle.Clear();
            BodyOffsetX.Clear();
            BodyOffsetZ.Clear();
            HeadFacing.Clear();
            BodyFacing.Clear();
            HeadBob.Clear();
            GenitalAngle.Clear();

            foreach (ActorAddon addon in addons)
            {
                addon.PosX.Clear();
                addon.PosZ.Clear();
                addon.Rotation.Clear();
            }

            int keyframePosition = 0;
            int duration = 0;

            keyframes[keyframes.Count - 1].tickDuration = 1;

            foreach (PawnKeyframe frame in keyframes)
            { duration += frame.tickDuration; }

            for (int i = 0; i < keyframes.Count; i++)
            {
                PawnKeyframe keyframe = keyframes[i];

                if (keyframe.atTick.HasValue)
                {
                    if (keyframe.HasValidKeyframeID() == false)
                    { keyframe.GenerateKeyframeID(Workspace.animationDef.animationStages[Workspace.stageID].animationClips.IndexOf(this)); }

                    BodyAngle.Add((float)keyframe.atTick / (float)duration, keyframe.bodyAngle, true);
                    HeadAngle.Add((float)keyframe.atTick / (float)duration, keyframe.headAngle, true);
                    BodyOffsetX.Add((float)keyframe.atTick / (float)duration, keyframe.bodyOffsetX, true);
                    BodyOffsetZ.Add((float)keyframe.atTick / (float)duration, keyframe.bodyOffsetZ, true);
                    HeadFacing.Add((float)keyframe.atTick / (float)duration, keyframe.headFacing, true);
                    BodyFacing.Add((float)keyframe.atTick / (float)duration, keyframe.bodyFacing, true);
                    HeadBob.Add((float)keyframe.atTick / (float)duration, keyframe.headBob, true);
                    GenitalAngle.Add((float)keyframe.atTick / (float)duration, keyframe.genitalAngle, true);

                    foreach (ActorAddon addon in addons)
                    {
                        if (keyframe.addonKeyframes.Any(x => x.addonName == addon.addonName) == false)
                        { keyframe.addonKeyframes.Add(new AddonKeyframe(addon.addonName)); }

                        addon.PosX.Add((float)keyframe.atTick / (float)duration, keyframe.GetAddonKeyframe(addon.addonName).posX, true);
                        addon.PosZ.Add((float)keyframe.atTick / (float)duration, keyframe.GetAddonKeyframe(addon.addonName).posZ, true);
                        addon.Rotation.Add((float)keyframe.atTick / (float)duration, keyframe.GetAddonKeyframe(addon.addonName).rotation, true);
                    }

                    if (i + 1 < keyframes.Count)
                    { keyframes[i].tickDuration = keyframes[i + 1].atTick.Value - keyframes[i].atTick.Value; }
                }

                else
                {
                    BodyAngle.Add((float)keyframePosition / (float)duration, keyframe.bodyAngle, true);
                    HeadAngle.Add((float)keyframePosition / (float)duration, keyframe.headAngle, true);
                    BodyOffsetX.Add((float)keyframePosition / (float)duration, keyframe.bodyOffsetX, true);
                    BodyOffsetZ.Add((float)keyframePosition / (float)duration, keyframe.bodyOffsetZ, true);
                    HeadFacing.Add((float)keyframePosition / (float)duration, keyframe.headFacing, true);
                    BodyFacing.Add((float)keyframePosition / (float)duration, keyframe.bodyFacing, true);
                    HeadBob.Add((float)keyframePosition / (float)duration, keyframe.headBob, true);
                    GenitalAngle.Add((float)keyframePosition / (float)duration, keyframe.genitalAngle, true);

                    foreach (ActorAddon addon in addons)
                    {
                        if (keyframe.addonKeyframes.Any(x => x.addonName == addon.addonName) == false)
                        { keyframe.addonKeyframes.Add(new AddonKeyframe(addon.addonName)); }

                        addon.PosX.Add((float)keyframePosition / (float)duration, keyframe.GetAddonKeyframe(addon.addonName).posX, true);
                        addon.PosZ.Add((float)keyframePosition / (float)duration, keyframe.GetAddonKeyframe(addon.addonName).posZ, true);
                        addon.Rotation.Add((float)keyframePosition / (float)duration, keyframe.GetAddonKeyframe(addon.addonName).rotation, true);
                    }

                    if (keyframe.tickDuration != 1 && keyframe.quiver.HasValue)
                    {
                        quiver.Add(keyframePosition, true);
                        quiver.Add(keyframePosition + keyframe.tickDuration - 1, false);
                    }

                    keyframe.atTick = keyframePosition + Constants.minTick;
                    keyframePosition += keyframe.tickDuration;
                }
            }
        }

        public void AddActorAddon(string addonName, float scale = 1f)
        {
            if (addons.Any(x => x.addonName == addonName) == false)
            { 
                addons.Add(new ActorAddon(addonName, scale));
            }

            foreach (PawnKeyframe keyframe in keyframes)
            {
                if (keyframe.addonKeyframes.Any(x => x.addonName == addonName) == false)
                { keyframe.addonKeyframes.Add(new AddonKeyframe(addonName)); }
            }
        }

        public void ShowOrHideActorAddon(string addonName, bool flag)
        {
            ActorAddon addon = GetActorAddon(addonName);

            if (addon != null)
            { addon.render = flag; }
        }

        public bool IsActorAddonVisible(string addonName)
        {
            ActorAddon addon = GetActorAddon(addonName);

            if (addon != null)
            { return addon.render; }

            return false;
        }

        public ActorAddon GetActorAddon(string addonName)
        {
            return addons.FirstOrDefault(x => x.addonName == addonName);
        }

        public override void ValidateData() 
        {
            _addons.Clear();

            foreach (ActorAddon addon in addons)
            {
                Debug.Log(addon.anchorName);

                if (addon.render)
                {
                    _addons.Add(addon);
                }
            }
        }

        public int GetOwningActorID()
        {
            if (Workspace.animationDef == null) return -1;

            return Workspace.animationDef.animationStages[Workspace.stageID].animationClips.IndexOf(this);
        }

        public bool MakeNew(int actorID = -1)
        {
            PawnKeyframe lastkeyframe = null;

            if (actorID >= 0)
            { lastkeyframe = Workspace.animationDef.animationStages[Workspace.stageID].animationClips[actorID]?.keyframes?.Last(); }

            if (lastkeyframe != null)
            {
                PawnKeyframe keyframeA = lastkeyframe.Copy();
                keyframeA.atTick = null;
                keyframeA.tickDuration = Constants.defaultAnimationClipLength - 1;
                keyframeA.GenerateKeyframeID(actorID);
                keyframes.Add(keyframeA);

                PawnKeyframe keyframeB = lastkeyframe.Copy();
                keyframeB.atTick = null;
                keyframeB.tickDuration = 1;
                keyframeB.GenerateKeyframeID(actorID);
                keyframes.Add(keyframeB);              
            }

            else
            {
                PawnKeyframe keyframeA = new PawnKeyframe();
                keyframeA.tickDuration = Constants.defaultAnimationClipLength - 1;
                keyframes.Add(keyframeA);

                PawnKeyframe keyframeB = new PawnKeyframe();
                keyframes.Add(keyframeB);
            }

            BuildSimpleCurves();
            return true;
        }

        public void OnPostLoad()
        {
            addons = _addons.Copy();

            foreach (PawnKeyframe keyframe in keyframes)
            {
                keyframe.OnPostLoad();
            }

            AddActorAddon("left hand", 0.667f);
            AddActorAddon("right hand", 0.667f);
            AddActorAddon("dildo");
        }
    }
}