#ifndef CREATURELIB_TYPELIBRARY_HPP
#define CREATURELIB_TYPELIBRARY_HPP

#include <Arbutils/Collections/Dictionary.hpp>
#include <Arbutils/Collections/List.hpp>
#include <Arbutils/ConstString.hpp>
#include <numeric>
#include <unordered_set>
#include <vector>

namespace CreatureLib::Library {
    class TypeLibrary {
        ArbUt::Dictionary<uint32_t, uint8_t> _types;
        ArbUt::List<ArbUt::List<float>> _effectiveness;

    public:
        TypeLibrary(size_t initialCapacity = 20) : _types(ArbUt::Dictionary<uint32_t, uint8_t>(initialCapacity)) {}

        inline uint8_t GetTypeId(const ArbUt::CaseInsensitiveConstString& key) const { return _types.Get(key); }
        inline uint8_t GetTypeId(uint32_t s) const { return _types.Get(s); }
        [[nodiscard]] inline float GetSingleEffectiveness(uint8_t attacking, uint8_t defensive) const {
            return _effectiveness[attacking][defensive];
        }
        [[nodiscard]] inline float GetEffectiveness(uint8_t attacking,
                                                    const std::unordered_set<uint8_t>& defensive) const {
            return std::accumulate(defensive.begin(), defensive.end(), (float)1,
                                   [this, attacking](float init, uint8_t defense) {
                                       return init * GetSingleEffectiveness(attacking, defense);
                                   });
        }

        uint8_t RegisterType(const ArbUt::CaseInsensitiveConstString& typeName);
        uint8_t RegisterType(uint32_t typeHash);
        void SetEffectiveness(uint8_t attacking, uint8_t defensive, float effectiveness);
    };
}

#endif // CREATURELIB_TYPELIBRARY_HPP