Glamourer/Glamourer.GameData/Structs/CharacterEquipMask.cs

using System;
using Penumbra.GameData.Enums;

namespace Glamourer.Structs;


// Turn EquipSlot into a bitfield flag enum.
[Flags]
public enum CharacterEquipMask : ushort
{
    None     = 0,
    MainHand = 0b000000000001,
    OffHand  = 0b000000000010,
    Head     = 0b000000000100,
    Body     = 0b000000001000,
    Hands    = 0b000000010000,
    Legs     = 0b000000100000,
    Feet     = 0b000001000000,
    Ears     = 0b000010000000,
    Neck     = 0b000100000000,
    Wrists   = 0b001000000000,
    RFinger  = 0b010000000000,
    LFinger  = 0b100000000000,
    All      = 0b111111111111,
}

public static class CharacterEquipMaskExtensions
{
    public static bool Fits(this CharacterEquipMask mask, EquipSlot slot)
        => slot switch
        {
            EquipSlot.Unknown => false,
            EquipSlot.Head    => mask.HasFlag(CharacterEquipMask.Head),
            EquipSlot.Body    => mask.HasFlag(CharacterEquipMask.Body),
            EquipSlot.Hands   => mask.HasFlag(CharacterEquipMask.Hands),
            EquipSlot.Legs    => mask.HasFlag(CharacterEquipMask.Legs),
            EquipSlot.Feet    => mask.HasFlag(CharacterEquipMask.Feet),
            EquipSlot.Ears    => mask.HasFlag(CharacterEquipMask.Ears),
            EquipSlot.Neck    => mask.HasFlag(CharacterEquipMask.Neck),
            EquipSlot.Wrists  => mask.HasFlag(CharacterEquipMask.Wrists),
            EquipSlot.RFinger => mask.HasFlag(CharacterEquipMask.RFinger),
            EquipSlot.LFinger => mask.HasFlag(CharacterEquipMask.LFinger),
            _                 => false,
        };
}