AAEmu.Game/Core/Managers/AnimationManager.cs - develop - NLObP/AAEmu

AAEmu.Game/Core/Managers/AnimationManager.cs
174 строки6 KB
Roger Barreto
Collection expressions check + blank lines update (#1110)
11 дек 2024, 19:38
11 дек 2024, 19:380ce5643
using System.Collections.Generic;
using System.Linq;

using AAEmu.Commons.Utils;
using AAEmu.Game.IO;
using AAEmu.Game.Models.Game.Animation;
using AAEmu.Game.Utils.DB;

using NLog;

namespace AAEmu.Game.Core.Managers;

public class AnimationManager : Singleton<AnimationManager>
{
    private static Logger Logger { get; } = LogManager.GetCurrentClassLogger();

    private Dictionary<uint, Anim> _animations = [];
    private Dictionary<string, Anim> _animationsByName = [];

    public Anim GetAnimation(uint id)
    {
        return _animations.TryGetValue(id, out var animation) ? animation : null;
    }

    public Anim GetAnimation(string name)
    {
        return _animationsByName.TryGetValue(name, out var value) ? value : null;
    }

    /// <summary>
    /// Parse target .g file into a List<AnimCombatSyncEvent>
    /// </summary>
    /// <param name="gFileName"></param>
    /// <returns>Returns null if there is a error, otherwise returns the list</returns>
    private static List<AnimCombatSyncEvent> ParseGFile(string gFileName)
    {
        var res = new List<AnimCombatSyncEvent>();
        var lines = ClientFileManager.GetFileAsString(gFileName).Split("\r\n");

        AnimCombatSyncEvent lastCombatSyncEvent = null;
        AnimDuration lastAnimDuration = null;
        for (var n = 0; n < lines.Length; n++)
        {
            var line = lines[n];
            var spaceCount = line.TakeWhile(char.IsWhiteSpace).Count();
            var trimmedLine = line.Trim(' ');
            if (spaceCount == 0)
            {
                // Start of new model section
                lastCombatSyncEvent = new AnimCombatSyncEvent();
                lastCombatSyncEvent.ModelName = line.Trim('"');
                res.Add(lastCombatSyncEvent);
                lastAnimDuration = null;
            }
            else if ((lastCombatSyncEvent != null) && (spaceCount == 4))
            {
                // Start of new animation section
                lastAnimDuration = new AnimDuration();
                if (!lastCombatSyncEvent.Animations.TryAdd(trimmedLine, lastAnimDuration))
                {
                    Logger.Warn($"Syntax error in {gFileName} at line {n + 1} : {line}");
                    return null;
                }
            }
            else if ((lastAnimDuration != null) && (spaceCount == 8))
            {
                // This is a actual property
                var props = trimmedLine.Split(' ');
                if (props.Length != 2)
                {
                    Logger.Warn($"Syntax error in {gFileName} at line {n + 1} : {line}");
                    return null;
                }
                else if (props[0] == "total_time")
                {
                    if (int.TryParse(props[1], out var totTime))
                        lastAnimDuration.total_time = totTime;
                    else
                    {
                        Logger.Warn($"int parse error in {gFileName} at line {n + 1} : {line}");
                        return null;
                    }
                }
                else if (props[0] == "combat_sync_time")
                {
                    if (int.TryParse(props[1], out var syncTime))
                        lastAnimDuration.combat_sync_time = syncTime;
                    else
                    {
                        Logger.Warn($"int parse error in {gFileName} at line {n + 1} : {line}");
                        return null;
                    }
                }
                else
                {
                    Logger.Warn($"Unknown property in {gFileName} at line {n + 1} : {line}");
                }
            }
            else
            {
                Logger.Warn($"Unknown Syntax in {gFileName} at line {n + 1} : {line}");
                return null;
            }
        }
        return res;
    }

    public void Load()
    {
        _animations = [];
        _animationsByName = [];

        Logger.Info("Loading animations...");

        using (var connection = SQLite.CreateConnection())
        {
            /* Anims */
            using (var command = connection.CreateCommand())
            {
                command.CommandText = "SELECT * FROM anims";
                command.Prepare();
                using (var reader = new SQLiteWrapperReader(command.ExecuteReader()))
                {
                    while (reader.Read())
                    {
                        var template = new Anim()
                        {
                            Id = reader.GetUInt32("id"),
                            Name = reader.GetString("name"),
                            Loop = reader.GetBoolean("loop"),
                            Category = (AnimCategory)reader.GetUInt32("category_id"),
                            RideUB = reader.GetString("ride_ub"),
                            HangUB = reader.GetString("hang_ub"),
                            SwimUB = reader.GetString("swim_ub"),
                            MoveUB = reader.GetString("move_ub"),
                            RelaxedUB = reader.GetString("relaxed_ub"),
                            SwimMoveUB = reader.GetString("swim_move_ub")
                        };

                        _animations.Add(template.Id, template);
                        _animationsByName.Add(template.Name, template);
                    }
                }
            }
        }

        // Load animation durations from client data
        var gFileName = "game/combat_sync_event_list.g";
        var combatSyncEvents = ParseGFile(gFileName);

        if (combatSyncEvents == null)
        {
            Logger.Fatal($"Error reading {gFileName}");
            return;
        }

        // Apply values to our animation manager (only takes nuian_male into account as a base value)
        foreach (var cse in combatSyncEvents)
        {
            if (cse.ModelName == "nuian_male")
            {
                // Copy stuff
                foreach (var (animKey, animVal) in cse.Animations)
                {
                    if (_animationsByName.TryGetValue(animKey, out var anim))
                    {
                        anim.Duration = animVal.total_time;
                        anim.CombatSyncTime = animVal.combat_sync_time;
                    }
                }
            }
        }
    }
}