st-perks.h

Go to the documentation of this file.

#pragma once
#include <unordered_set>
 
namespace StyyxUtil
{
struct PerkUtil
{

    static constexpr std::array<RE::ActorValue, 18> kPlayableSkills = {
        RE::ActorValue::kOneHanded,  RE::ActorValue::kTwoHanded,   RE::ActorValue::kArchery,
        RE::ActorValue::kBlock,      RE::ActorValue::kSmithing,    RE::ActorValue::kHeavyArmor,
        RE::ActorValue::kLightArmor, RE::ActorValue::kPickpocket,  RE::ActorValue::kLockpicking,
        RE::ActorValue::kSneak,      RE::ActorValue::kAlchemy,     RE::ActorValue::kSpeech,
        RE::ActorValue::kAlteration, RE::ActorValue::kConjuration, RE::ActorValue::kDestruction,
        RE::ActorValue::kIllusion,   RE::ActorValue::kRestoration, RE::ActorValue::kEnchanting};

    static void TraversePerkNode(RE::BGSSkillPerkTreeNode* node, std::unordered_set<RE::BGSPerk*>& set_out,
                                 std::unordered_set<RE::BGSSkillPerkTreeNode*>& visited)
    {
        if (!node || !visited.insert(node).second)
            return;
 
        if (node->perk)
        {
            set_out.insert(node->perk);
        }
 
        for (auto* child : node->children)
        {
            TraversePerkNode(child, set_out, visited);
        }
    };

    static std::unordered_set<RE::BGSPerk*> GetPerksForSkill(RE::ActorValue av)
    {
        std::unordered_set<RE::BGSPerk*> out;
        if (!std::ranges::contains(kPlayableSkills.begin(), kPlayableSkills.end(), av))
            return out;
 
        std::unordered_set<RE::BGSSkillPerkTreeNode*> visited;
        if (const auto avif = RE::ActorValueList::GetActorValueInfo(av); avif && avif->perkTree)
            TraversePerkNode(avif->perkTree, out, visited);
        return out;
    }

    static std::unordered_set<RE::BGSPerk*> GetAllPlayablePerks() { return GetPerksForSkills(kPlayableSkills); }

    static std::unordered_set<RE::BGSPerk*> GetAllMagicPerks() { return GetPerksForSkills(kMageSkills); }

    static std::unordered_set<RE::BGSPerk*> GetAllThiefPerks() { return GetPerksForSkills(kThiefSkills); }

    static std::unordered_set<RE::BGSPerk*> GetAllWarriorPerks() { return GetPerksForSkills(kWarriorSkills); }
 
  private:
    static constexpr std::array<RE::ActorValue, 6> kWarriorSkills = {
        RE::ActorValue::kOneHanded, RE::ActorValue::kTwoHanded, RE::ActorValue::kHeavyArmor,
        RE::ActorValue::kArchery,   RE::ActorValue::kSmithing,  RE::ActorValue::kBlock};
    static constexpr std::array<RE::ActorValue, 6> kThiefSkills = {
        RE::ActorValue::kSpeech, RE::ActorValue::kAlchemy,    RE::ActorValue::kLockpicking,
        RE::ActorValue::kSneak,  RE::ActorValue::kPickpocket, RE::ActorValue::kLightArmor};
    static constexpr std::array<RE::ActorValue, 6> kMageSkills = {
        RE::ActorValue::kAlteration, RE::ActorValue::kConjuration, RE::ActorValue::kDestruction,
        RE::ActorValue::kIllusion,   RE::ActorValue::kEnchanting,  RE::ActorValue::kRestoration};
 
    static std::unordered_set<RE::BGSPerk*> GetPerksForSkills(std::span<const RE::ActorValue> skills)
    {
        std::unordered_set<RE::BGSPerk*> out;
        std::unordered_set<RE::BGSSkillPerkTreeNode*> visited;
 
        for (const auto av : skills)
        {
            if (const auto avif = RE::ActorValueList::GetActorValueInfo(av); avif && avif->perkTree)
                TraversePerkNode(avif->perkTree, out, visited);
        }
        return out;
    }
};
} // namespace StyyxUtil