Glamourer/Glamourer.GameData/CharacterEquipExtensions.cs

using System;
using System.ComponentModel;
using System.Runtime.InteropServices;
using FFXIVClientStructs.FFXIV.Client.Game.Character;
using Penumbra.GameData.Enums;
using Penumbra.GameData.Structs;
using Penumbra.PlayerWatch;

namespace Glamourer;

public static class WriteExtensions
{
    private static unsafe void Write(IntPtr characterPtr, EquipSlot slot, SetId? id, WeaponType? type, ushort? variant, StainId? stain)
    {
        void WriteWeapon(int offset)
        {
            var address = (byte*)characterPtr + offset;
            if (id.HasValue)
                *(ushort*)address = (ushort)id.Value;

            if (type.HasValue)
                *(ushort*)(address + 2) = (ushort)type.Value;

            if (variant.HasValue)
                *(ushort*)(address + 4) = variant.Value;

            if (*(ushort*)address == 0)
                *(address + 6) = 0;
            else if (stain.HasValue)
                *(address + 6) = (byte)stain.Value;
        }

        void WriteEquip(int offset)
        {
            var address = (byte*)characterPtr + offset;
            if (id.HasValue)
                *(ushort*)address = (ushort)id.Value;

            if (variant < byte.MaxValue)
                *(address + 2) = (byte)variant.Value;

            if (stain.HasValue)
                *(address + 3) = (byte)stain.Value;
        }

        var drawDataOffset = (int) Marshal.OffsetOf<Character>(nameof(Character.DrawData));
        switch (slot)
        {
            case EquipSlot.MainHand:
                WriteWeapon(drawDataOffset + (int)Marshal.OffsetOf<DrawDataContainer>(nameof(DrawDataContainer.MainHandModel)));
                break;
            case EquipSlot.OffHand:
                WriteWeapon(drawDataOffset + (int)Marshal.OffsetOf<DrawDataContainer>(nameof(DrawDataContainer.OffHandModel)));
                break;
            case EquipSlot.Head:
                WriteEquip(drawDataOffset + (int)Marshal.OffsetOf<DrawDataContainer>(nameof(DrawDataContainer.Head)));
                break;
            case EquipSlot.Body:
                WriteEquip(drawDataOffset + (int)Marshal.OffsetOf<DrawDataContainer>(nameof(DrawDataContainer.Top)));
                break;
            case EquipSlot.Hands:
                WriteEquip(drawDataOffset + (int)Marshal.OffsetOf<DrawDataContainer>(nameof(DrawDataContainer.Arms)));
                break;
            case EquipSlot.Legs:
                WriteEquip(drawDataOffset + (int)Marshal.OffsetOf<DrawDataContainer>(nameof(DrawDataContainer.Legs)));
                break;
            case EquipSlot.Feet:
                WriteEquip(drawDataOffset + (int)Marshal.OffsetOf<DrawDataContainer>(nameof(DrawDataContainer.Feet)));
                break;
            case EquipSlot.Ears:
                WriteEquip(drawDataOffset + (int)Marshal.OffsetOf<DrawDataContainer>(nameof(DrawDataContainer.Ear)));
                break;
            case EquipSlot.Neck:
                WriteEquip(drawDataOffset + (int)Marshal.OffsetOf<DrawDataContainer>(nameof(DrawDataContainer.Neck)));
                break;
            case EquipSlot.Wrists:
                WriteEquip(drawDataOffset + (int)Marshal.OffsetOf<DrawDataContainer>(nameof(DrawDataContainer.Wrist)));
                break;
            case EquipSlot.RFinger:
                WriteEquip(drawDataOffset + (int)Marshal.OffsetOf<DrawDataContainer>(nameof(DrawDataContainer.RFinger)));
                break;
            case EquipSlot.LFinger:
                WriteEquip(drawDataOffset + (int)Marshal.OffsetOf<DrawDataContainer>(nameof(DrawDataContainer.LFinger)));
                break;
            default: throw new InvalidEnumArgumentException();
        }
    }

    public static void Write(this Stain stain, IntPtr characterPtr, EquipSlot slot)
        => Write(characterPtr, slot, null, null, null, stain.RowIndex);

    public static void Write(this Item item, IntPtr characterAddress)
    {
        var (id, type, variant) = item.MainModel;
        Write(characterAddress, item.EquippableTo, id, type, variant, null);
        if (item.EquippableTo == EquipSlot.MainHand && item.HasSubModel)
        {
            var (subId, subType, subVariant) = item.SubModel;
            Write(characterAddress, EquipSlot.OffHand, subId, subType, subVariant, null);
        }
    }

    public static void Write(this CharacterArmor armor, IntPtr characterAddress, EquipSlot slot)
        => Write(characterAddress, slot, armor.Set, null, armor.Variant, armor.Stain);

    public static void Write(this CharacterWeapon weapon, IntPtr characterAddress, EquipSlot slot)
        => Write(characterAddress, slot, weapon.Set, weapon.Type, weapon.Variant, weapon.Stain);

    public static unsafe void Write(this CharacterEquipment equip, IntPtr characterAddress)
    {
        if (equip.IsSet == 0)
            return;

        Write(characterAddress, EquipSlot.MainHand, equip.MainHand.Set, equip.MainHand.Type, equip.MainHand.Variant, equip.MainHand.Stain);
        Write(characterAddress, EquipSlot.OffHand,  equip.OffHand.Set,  equip.OffHand.Type,  equip.OffHand.Variant,  equip.OffHand.Stain);

        fixed (CharacterArmor* equipment = &equip.Head)
        {
            Buffer.MemoryCopy(equipment, &((Character*)characterAddress)->DrawData.Head,
                CharacterEquipment.EquipmentSlots * sizeof(CharacterArmor), CharacterEquipment.EquipmentSlots * sizeof(CharacterArmor));
        }
    }

    public static void Write(this CharacterEquipment equip, IntPtr characterAddress, CharacterEquipMask models, CharacterEquipMask stains)
    {
        if (models == CharacterEquipMask.All && stains == CharacterEquipMask.All)
        {
            equip.Write(characterAddress);
            return;
        }

        if (models.HasFlag(CharacterEquipMask.MainHand))
            Write(characterAddress, EquipSlot.MainHand, equip.MainHand.Set, equip.MainHand.Type, equip.MainHand.Variant, null);
        if (stains.HasFlag(CharacterEquipMask.MainHand))
            Write(characterAddress, EquipSlot.MainHand, null, null, null, equip.MainHand.Stain);
        if (models.HasFlag(CharacterEquipMask.OffHand))
            Write(characterAddress, EquipSlot.OffHand, equip.OffHand.Set, equip.OffHand.Type, equip.OffHand.Variant, null);
        if (stains.HasFlag(CharacterEquipMask.OffHand))
            Write(characterAddress, EquipSlot.OffHand, null, null, null, equip.OffHand.Stain);

        if (models.HasFlag(CharacterEquipMask.Head))
            Write(characterAddress, EquipSlot.Head, equip.Head.Set, null, equip.Head.Variant, null);
        if (stains.HasFlag(CharacterEquipMask.Head))
            Write(characterAddress, EquipSlot.Head, null, null, null, equip.Head.Stain);
        if (models.HasFlag(CharacterEquipMask.Body))
            Write(characterAddress, EquipSlot.Body, equip.Body.Set, null, equip.Body.Variant, null);
        if (stains.HasFlag(CharacterEquipMask.Body))
            Write(characterAddress, EquipSlot.Body, null, null, null, equip.Body.Stain);
        if (models.HasFlag(CharacterEquipMask.Hands))
            Write(characterAddress, EquipSlot.Hands, equip.Hands.Set, null, equip.Hands.Variant, null);
        if (stains.HasFlag(CharacterEquipMask.Hands))
            Write(characterAddress, EquipSlot.Hands, null, null, null, equip.Hands.Stain);
        if (models.HasFlag(CharacterEquipMask.Legs))
            Write(characterAddress, EquipSlot.Legs, equip.Legs.Set, null, equip.Legs.Variant, null);
        if (stains.HasFlag(CharacterEquipMask.Legs))
            Write(characterAddress, EquipSlot.Legs, null, null, null, equip.Legs.Stain);
        if (models.HasFlag(CharacterEquipMask.Feet))
            Write(characterAddress, EquipSlot.Feet, equip.Feet.Set, null, equip.Feet.Variant, null);
        if (stains.HasFlag(CharacterEquipMask.Feet))
            Write(characterAddress, EquipSlot.Feet, null, null, null, equip.Feet.Stain);

        if (models.HasFlag(CharacterEquipMask.Ears))
            Write(characterAddress, EquipSlot.Ears, equip.Ears.Set, null, equip.Ears.Variant, null);
        if (stains.HasFlag(CharacterEquipMask.Ears))
            Write(characterAddress, EquipSlot.Ears, null, null, null, equip.Ears.Stain);
        if (models.HasFlag(CharacterEquipMask.Neck))
            Write(characterAddress, EquipSlot.Neck, equip.Neck.Set, null, equip.Neck.Variant, null);
        if (stains.HasFlag(CharacterEquipMask.Neck))
            Write(characterAddress, EquipSlot.Neck, null, null, null, equip.Neck.Stain);
        if (models.HasFlag(CharacterEquipMask.Wrists))
            Write(characterAddress, EquipSlot.Wrists, equip.Wrists.Set, null, equip.Wrists.Variant, null);
        if (stains.HasFlag(CharacterEquipMask.Wrists))
            Write(characterAddress, EquipSlot.Wrists, null, null, null, equip.Wrists.Stain);
        if (models.HasFlag(CharacterEquipMask.LFinger))
            Write(characterAddress, EquipSlot.LFinger, equip.LFinger.Set, null, equip.LFinger.Variant, null);
        if (stains.HasFlag(CharacterEquipMask.LFinger))
            Write(characterAddress, EquipSlot.LFinger, null, null, null, equip.LFinger.Stain);
        if (models.HasFlag(CharacterEquipMask.RFinger))
            Write(characterAddress, EquipSlot.RFinger, equip.RFinger.Set, null, equip.RFinger.Variant, null);
        if (stains.HasFlag(CharacterEquipMask.RFinger))
            Write(characterAddress, EquipSlot.RFinger, null, null, null, equip.RFinger.Stain);
    }
}