NAK_CVR_Mods/AASBufferFix/AASBufferHelper.cs

using ABI.CCK.Components;
using ABI_RC.Core.Player;
using UnityEngine;

namespace NAK.Melons.AASBufferFix
{
    public class AASBufferHelper : MonoBehaviour
    {
        ///public bool DebuggingFlag = false;

        //public stuff
        public bool GameHandlesAAS { get; private set; }

        //internal references
        private PuppetMaster _puppetMaster;

        //outside aas buffers
        private float[] _aasBufferFloat = new float[0];
        private int[] _aasBufferInt = new int[0];
        private byte[] _aasBufferByte = new byte[0];

        //calculated footprints
        private int _aasFootprint = -1;
        private int _avatarFootprint = 0;

        private void Start() => _puppetMaster = GetComponent<PuppetMaster>();

        public void OnAvatarInstantiated(Animator animator)
        {
            //check if avatar uses Avatar Advanced Settings
            ///SendDebug("[OnInit] Remote avatar initialized. Checking for AAS...");
            CVRAvatar avatar = animator.GetComponent<CVRAvatar>();
            if (avatar != null && !avatar.avatarUsesAdvancedSettings)
            {
                GameHandlesAAS = true;
                return;
            }

            //check if AAS footprint is valid
            ///SendDebug("[OnInit] Avatar uses AAS. Generating AAS footprint...");
            _avatarFootprint = Utils.GenerateAnimatorAASFootprint(animator);
            if (_avatarFootprint == 1)
            {
                // we will let the game handle this by setting GameHandlesAAS to true
                ///SendDebug("[OnInit] Avatar does not contain valid AAS. It is likely hidden or blocked.");
                GameHandlesAAS = true;
                return;
            }

            ///SendDebug($"[OnInit] Avatar footprint is : {_avatarFootprint}");

            //check if we received expected AAS while we loaded the avatar, and if so, apply it now
            if (SyncDataMatchesExpected())
            {
                ///SendDebug("[OnInit] Valid buffered AAS found. Applying buffer...");
                ApplyExternalAASBuffer();
                return;
            }

            //we loaded avatar faster than wearer
            ///SendDebug("[OnInit] Remote avatar initialized faster than wearer. Waiting on valid AAS...");
        }

        public void OnAvatarDestroyed()
        {
            GameHandlesAAS = false;
            _aasFootprint = -1;
            _avatarFootprint = 0;
        }

        public void OnReceiveAAS(float[] settingsFloat, int[] settingsInt, byte[] settingsByte)
        {
            // Calculate AAS footprint to compare against.
            _aasFootprint = (settingsFloat.Length + 1) * (settingsInt.Length + 1) * (settingsByte.Length + 1);

            //if it matches, apply the settings and let game take over
            if (SyncDataMatchesExpected())
            {
                ///SendDebug("[OnSync] Avatar values matched and have been applied.");
                ApplyExternalAAS(settingsFloat, settingsInt, settingsByte);
                return;
            }

            //avatar is still loading on our side, we must assume AAS data is correct and store it until we load
            //there is also a chance it errored
            //if (_avatarFootprint == 0)
            //{
            //    ///SendDebug("[OnSync] Avatar is still loading on our end.");
            //    StoreExternalAASBuffer(settingsFloat, settingsInt, settingsByte);
            //    return;
            //}

            //avatar is loaded on our end, and is not blocked by filter
            //this does run if it is manually hidden or distance hidden

            ///SendDebug("[OnSync] Avatar is loaded on our side and is not blocked. Comparing for expected values.");
            ///SendDebug($"[OnSync] Avatar Footprint is : {_avatarFootprint}");

            //if it did not match, that means the avatar we see on our side is different than what the remote user is wearing and syncing
            ///SendDebug("[OnSync] Avatar loaded is different than wearer. The wearer is likely still loading the avatar!");
            StoreExternalAASBuffer(settingsFloat, settingsInt, settingsByte);
        }

        private void ApplyExternalAASBuffer()
        {
            GameHandlesAAS = true;
            _puppetMaster?.ApplyAdvancedAvatarSettings(_aasBufferFloat, _aasBufferInt, _aasBufferByte);
        }

        private void ApplyExternalAAS(float[] settingsFloat, int[] settingsInt, byte[] settingsByte)
        {
            GameHandlesAAS = true;
            _puppetMaster?.ApplyAdvancedAvatarSettings(settingsFloat, settingsInt, settingsByte);
        }

        private void StoreExternalAASBuffer(float[] settingsFloat, int[] settingsInt, byte[] settingsByte)
        {
            Array.Resize(ref _aasBufferFloat, settingsFloat.Length);
            Array.Resize(ref _aasBufferInt, settingsInt.Length);
            Array.Resize(ref _aasBufferByte, settingsByte.Length);
            Array.Copy(settingsFloat, _aasBufferFloat, settingsFloat.Length);
            Array.Copy(settingsInt, _aasBufferInt, settingsInt.Length);
            Array.Copy(settingsByte, _aasBufferByte, settingsByte.Length);
        }

        private bool SyncDataMatchesExpected() => _aasFootprint == _avatarFootprint;

        ///private void SendDebug(string message)
        ///{
        ///    if (!DebuggingFlag) return;
        ///    AASBufferFix.Logger.Msg(message);
        ///}
    }
}