rimworld-animation-studio/Assets/Scripts/Managers/ApplicationManager.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using UnityEngine;
using System.IO;
using UnityEngine.UI;
using SFB;

namespace RimWorldAnimationStudio
{
    public class ApplicationManager : Singleton<ApplicationManager>
    {
        public DialogBox exitDialog;
        public DialogBox newAnimationDialog;
        public SelectAnimationDialog selectAnimationDialog;

        public void Start()
        {
            LoadPawnRaceDefs();
            LoadCustomArrays();
        }

        public void TryToCloseApplication()
        {
            exitDialog.Pop();
        }

        public void CloseApplication()
        {
            Debug.Log("Exiting application");
            Application.Quit();
        }

        public void TryToLoadAnimation()
        {
            var paths = StandaloneFileBrowser.OpenFilePanel("Open AnimationDef File", "", "xml", false);
            
            if (paths == null || paths.Any() == false)
            { Debug.LogWarning("Selected file was null or invalid"); return; }

            AnimationDefs defs = null;

            try 
            { defs = XmlUtility.ReadXML<AnimationDefs>(paths[0]); }

            catch
            { Debug.LogError("Could not read .xml file '" + paths[0] + "' - this file either uses a schema which is outdated, incorrect, or contains empty fields"); return; }

            if (defs?.animationDefs == null)
            { Debug.LogError("Selected file contains no animation data"); return; }

            if (defs.animationDefs.Count == 1)
            { LoadAnimation(defs.animationDefs[0]); return;  }

            Workspace.animationSavePath = paths[0];

            selectAnimationDialog.Initialize(defs);
            selectAnimationDialog.Pop();
        }

        public void LoadAnimation(AnimationDef animationDef)
        {
            UpdateCustomArrays(animationDef);
            RunPostLoadOperations(animationDef);

            Debug.Log("Loaded AnimationDef: " + animationDef.DefName);

            Workspace.animationDef = animationDef;
            animationDef.Initialize();

            AnimationController.Instance.Reset();
            Workspace.Reset();

            EventsManager.OnAnimationChanged();
            Workspace.RecordEvent("AnimationDef loaded");
        }

        public void RunPostLoadOperations(AnimationDef animationDef)
        {
            if (animationDef.animationTimeTicksQuick <= 0)
            {
                if (animationDef.AnimationStages.Count > 1)
                {
                    for (int i = 0; i < animationDef.AnimationStages.Count; i++)
                    {
                        if (i == 0) continue;
                        animationDef.AnimationStages[i].PlayTimeTicksQuick = animationDef.AnimationStages[i].PlayTimeTicks;
                    }
                }

                else if (animationDef.AnimationStages.Count == 1)
                { animationDef.AnimationStages[0].PlayTimeTicksQuick = animationDef.AnimationStages[0].PlayTimeTicks; }
            }

            foreach (AnimationStage stage in animationDef.AnimationStages)
            {
                stage.OnPostLoad();
            }
        }

        public void TryToSaveAnimation()
        {
            if (Workspace.animationDef == null)
            { return; }

            string path = Workspace.animationSavePath;

            if (path != null && path != "" && File.Exists(path))
            { SaveAnimation(path); }

            else
            { TryToSaveAnimationAs(); }
        }

        public void TryToSaveAnimationAs()
        {
            if (Workspace.animationDef == null)
            { return; }

            string defName = Workspace.animationDef.DefName != null && Workspace.animationDef.DefName != "" ? Workspace.animationDef.DefName : "newAnimationDef";
            string path = StandaloneFileBrowser.SaveFilePanel("Save AnimationDef File", "", defName, "xml");
            
            if (path != null && path != "")
            { SaveAnimation(path); }
        }

        public void SaveAnimation(string path)
        {
            AnimationDef animationDef = Workspace.animationDef.Copy();
            RunPreSaveOperations(animationDef);

            Debug.Log("Saving AnimationDef: " + Workspace.animationDef.DefName);

            AnimationDefs defs = new AnimationDefs();
            defs.animationDefs.Add(animationDef);

            XmlUtility.WriteXML(defs, path);

            Workspace.animationSavePath = path;
        }

        public void RunPreSaveOperations(AnimationDef animationDef)
        {
            animationDef.OnPreSave();

            foreach (Actor actor in animationDef.Actors)
            { actor.OnPreSave(); }

            foreach (AnimationStage stage in animationDef.AnimationStages)
            {
                stage.OnPreSave();

                foreach (PawnAnimationClip clip in stage.AnimationClips)
                {
                    clip.OnPreSave();

                    foreach (PawnKeyframe keyframe in clip.Keyframes)
                    { keyframe.OnPreSave(); }
                }
            }
        }

        public void TryToMakeNewAnimation()
        {
            if (Workspace.animationDef == null)
            { NewAnimation(); return; }

            newAnimationDialog.Pop();
        }

        public void NewAnimation()
        {
            var path = Path.Combine(Application.streamingAssetsPath, "AnimationDefs/newAnimationDef.xml");

            AnimationDefs defs = XmlUtility.ReadXML<AnimationDefs>(path);

            if (defs?.animationDefs == null)
            { Debug.LogError("Default animation def file contains no animation data"); return; }

            LoadAnimation(defs.animationDefs[0]);

            Workspace.animationSavePath = null;
        }

        public void SaveCustomArrays()
        {
            var path = Path.Combine(Application.persistentDataPath, "customTags.xml");

            CustomTagsHelper helper = new CustomTagsHelper();

            helper.defNames = CustomTags.defNames;
            helper.bodyParts = CustomTags.bodyParts;
            helper.bodyDefTypes = CustomTags.bodyDefTypes;
            helper.sexTypes = CustomTags.sexTypes;
            helper.interactionDefTypes = CustomTags.interactionDefTypes;
            helper.soundDefs = CustomTags.soundDefs;

            XmlUtility.WriteXML(helper, path);
        }

        public void LoadCustomArrays()
        {
            string path;

            if (File.Exists(Path.Combine(Application.persistentDataPath, "customTags.xml")))
            { path = Path.Combine(Application.persistentDataPath, "customTags.xml"); }

            else
            { path = Path.Combine(Application.streamingAssetsPath, "customTags.xml"); }
            
            if (File.Exists(path) == false)
            { SaveCustomArrays(); return; }
            
            CustomTagsHelper helper = XmlUtility.ReadXML<CustomTagsHelper>(path);

            CustomTags.defNames = helper.defNames;
            CustomTags.bodyParts = helper.bodyParts;
            CustomTags.bodyDefTypes = helper.bodyDefTypes;
            CustomTags.sexTypes = helper.sexTypes;
            CustomTags.interactionDefTypes = helper.interactionDefTypes;
            CustomTags.soundDefs = helper.soundDefs;
        }

        public void UpdateCustomArrays(AnimationDef animationDef)
        {
            CustomTags.bodyParts.AddRangeDistinct(animationDef.Actors.SelectMany(x => x.RequiredGenitals).Except(DefaultTags.bodyParts));
            CustomTags.bodyDefTypes.AddRangeDistinct(animationDef.Actors.SelectMany(x => x.BodyDefTypes).Except(DefaultTags.bodyDefTypes));
            CustomTags.sexTypes.AddRangeDistinct(animationDef.SexTypes.Except(DefaultTags.sexTypes));
            CustomTags.interactionDefTypes.AddRangeDistinct(animationDef.InteractionDefTypes.Except(DefaultTags.interactionDefTypes));
            CustomTags.soundDefs.AddRangeDistinct(animationDef.AnimationStages.SelectMany(x => x.AnimationClips.SelectMany(y => y.Keyframes.Select(z => z.SoundEffect))).Except(DefaultTags.soundDefs));

            SaveCustomArrays();
        }

        public void LoadPawnRaceDefs()
        {
            string path;

            if (File.Exists(Path.Combine(Application.persistentDataPath, "pawnRaceDefs.xml")))
            { path = Path.Combine(Application.persistentDataPath, "pawnRaceDefs.xml"); }

            else
            { path = Path.Combine(Application.streamingAssetsPath, "pawnRaceDefs.xml"); }

            if (File.Exists(path) == false)
            { SavePawnRaceDefs(); return; }

            PawnRaceDefs.allDefs = XmlUtility.ReadXML<List<PawnRaceDef>>(path);
            PawnRaceDefs.OnLoad();
        }

        public void SavePawnRaceDefs()
        {
            var path = Path.Combine(Application.persistentDataPath, "pawnRaceDefs.xml");

            XmlUtility.WriteXML(PawnRaceDefs.allDefs, path);
        }
    }
}