JisolGame/JEX_GAS/Assets/GAS/Runtime/AttributeSet/AttributeSetContainer.cs

using System;
using System.Collections.Generic;
using GAS.General;

namespace GAS.Runtime
{
    public class AttributeSetContainer
    {
        private readonly AbilitySystemComponent _owner;
        private readonly Dictionary<string, AttributeSet> _attributeSets = new();

        private readonly Dictionary<AttributeBase, AttributeAggregator> _attributeAggregators = new();

        public Dictionary<string, AttributeSet> Sets => _attributeSets;

        public AttributeSetContainer(AbilitySystemComponent owner)
        {
            _owner = owner;
        }

        public void AddAttributeSet<T>() where T : AttributeSet
        {
            AddAttributeSet(typeof(T));
        }

        public void AddAttributeSet(Type attrSetType)
        {
            if (TryGetAttributeSet(attrSetType, out _)) return;
            var setName = AttributeSetUtil.AttributeSetName(attrSetType);
            _attributeSets.Add(setName, Activator.CreateInstance(attrSetType) as AttributeSet);

            var attrSet = _attributeSets[setName];
            foreach (var attr in attrSet.AttributeNames)
            {
                if (!_attributeAggregators.ContainsKey(attrSet[attr]))
                {
                    var attrAggt = new AttributeAggregator(attrSet[attr], _owner);
                    if (_owner.enabled) attrAggt.OnEnable();
                    _attributeAggregators.Add(attrSet[attr], attrAggt);
                }
            }

            attrSet.SetOwner(_owner);
        }

        /// <summary>
        /// Be careful when using this method, it may cause unexpected errors(when using network sync).
        /// </summary>
        /// <typeparam name="T"></typeparam>
        public void RemoveAttributeSet<T>() where T : AttributeSet
        {
            var setName = AttributeSetUtil.AttributeSetName(typeof(T));
            var attrSet = _attributeSets[setName];
            foreach (var attr in attrSet.AttributeNames)
            {
                _attributeAggregators.Remove(attrSet[attr]);
            }

            _attributeSets.Remove(setName);
        }

        public bool TryGetAttributeSet<T>(out T attributeSet) where T : AttributeSet
        {
            if (_attributeSets.TryGetValue(AttributeSetUtil.AttributeSetName(typeof(T)), out var set))
            {
                attributeSet = (T)set;
                return true;
            }

            attributeSet = null;
            return false;
        }

        bool TryGetAttributeSet(Type attrSetType, out AttributeSet attributeSet)
        {
            if (_attributeSets.TryGetValue(AttributeSetUtil.AttributeSetName(attrSetType), out var set))
            {
                attributeSet = set;
                return true;
            }

            attributeSet = null;
            return false;
        }

        public AttributeValue? GetAttributeAttributeValue(string attrSetName, string attrShortName)
        {
            return _attributeSets.TryGetValue(attrSetName, out var set)
                ? set[attrShortName].Value
                : (AttributeValue?)null;
        }

        public CalculateMode? GetAttributeCalculateMode(string attrSetName, string attrShortName)
        {
            return _attributeSets.TryGetValue(attrSetName, out var set)
                ? set[attrShortName].CalculateMode
                : (CalculateMode?)null;
        }

        public float? GetAttributeBaseValue(string attrSetName, string attrShortName)
        {
            return _attributeSets.TryGetValue(attrSetName, out var set) ? set[attrShortName].BaseValue : (float?)null;
        }

        public float? GetAttributeCurrentValue(string attrSetName, string attrShortName)
        {
            return _attributeSets.TryGetValue(attrSetName, out var set)
                ? set[attrShortName].CurrentValue
                : (float?)null;
        }


        public Dictionary<string, float> Snapshot()
        {
            var snapshot = ObjectPool.Instance.Fetch<Dictionary<string, float>>();
            foreach (var kv in _attributeSets)
            {
                var attributeSet = kv.Value;
                foreach (var name in attributeSet.AttributeNames)
                {
                    var attr = attributeSet[name];
                    snapshot.Add(attr.Name, attr.CurrentValue);
                }
            }

            return snapshot;
        }

        public void OnDisable()
        {
            foreach (var aggregator in _attributeAggregators)
                aggregator.Value.OnDisable();
        }

        public void OnEnable()
        {
            foreach (var aggregator in _attributeAggregators)
                aggregator.Value.OnEnable();
        }

        public void OnDestroy()
        {
            foreach (var aggregator in _attributeAggregators)
                aggregator.Value.OnDestroy();
            _attributeAggregators.Clear();
        }
    }
}