Glamourer/Glamourer.GameData/Customization/CustomizationSet.cs

using System;
using System.Collections.Generic;
using System.Linq;
using Penumbra.GameData.Enums;

namespace Glamourer.Customization;

public class CustomizationSet
{
    public const int DefaultAvailable =
        (1 << (int)CustomizationId.Height)
      | (1 << (int)CustomizationId.Hairstyle)
      | (1 << (int)CustomizationId.SkinColor)
      | (1 << (int)CustomizationId.EyeColorR)
      | (1 << (int)CustomizationId.EyeColorL)
      | (1 << (int)CustomizationId.HairColor)
      | (1 << (int)CustomizationId.HighlightColor)
      | (1 << (int)CustomizationId.FacialFeaturesTattoos)
      | (1 << (int)CustomizationId.TattooColor)
      | (1 << (int)CustomizationId.LipColor)
      | (1 << (int)CustomizationId.Height);

    internal CustomizationSet(SubRace clan, Gender gender)
    {
        Gender = gender;
        Clan   = clan;
        _settingAvailable = clan.ToRace() == Race.Hrothgar && gender == Gender.Female
            ? 0
            : DefaultAvailable;
    }

    public Gender  Gender { get; }
    public SubRace Clan   { get; }

    public Race Race
        => Clan.ToRace();

    private int _settingAvailable;

    internal void SetAvailable(CustomizationId id)
        => _settingAvailable |= 1 << (int)id;

    public bool IsAvailable(CustomizationId id)
        => (_settingAvailable & (1 << (int)id)) != 0;

    public int NumEyebrows    { get; internal set; }
    public int NumEyeShapes   { get; internal set; }
    public int NumNoseShapes  { get; internal set; }
    public int NumJawShapes   { get; internal set; }
    public int NumMouthShapes { get; internal set; }


    public IReadOnlyList<string>                       OptionName      { get; internal set; } = null!;
    public IReadOnlyList<Customization>                Faces           { get; internal set; } = null!;
    public IReadOnlyList<Customization>                HairStyles      { get; internal set; } = null!;
    public IReadOnlyList<Customization>                TailEarShapes   { get; internal set; } = null!;
    public IReadOnlyList<IReadOnlyList<Customization>> FeaturesTattoos { get; internal set; } = null!;
    public IReadOnlyList<Customization>                FacePaints      { get; internal set; } = null!;

    public IReadOnlyList<Customization> SkinColors           { get; internal set; } = null!;
    public IReadOnlyList<Customization> HairColors           { get; internal set; } = null!;
    public IReadOnlyList<Customization> HighlightColors      { get; internal set; } = null!;
    public IReadOnlyList<Customization> EyeColors            { get; internal set; } = null!;
    public IReadOnlyList<Customization> TattooColors         { get; internal set; } = null!;
    public IReadOnlyList<Customization> FacePaintColorsLight { get; internal set; } = null!;
    public IReadOnlyList<Customization> FacePaintColorsDark  { get; internal set; } = null!;
    public IReadOnlyList<Customization> LipColorsLight       { get; internal set; } = null!;
    public IReadOnlyList<Customization> LipColorsDark        { get; internal set; } = null!;

    public IReadOnlyList<CharaMakeParams.MenuType> Types { get; internal set; } = null!;

    public string Option(CustomizationId id)
        => OptionName[(int)id];

    public Customization FacialFeature(int faceIdx, int idx)
        => FeaturesTattoos[faceIdx - 1][idx];

    public int DataByValue(CustomizationId id, byte value, out Customization? custom)
    {
        var type = id.ToType();
        custom = null;
        if (type == CharaMakeParams.MenuType.Percentage || type == CharaMakeParams.MenuType.ListSelector)
        {
            if (value < Count(id))
            {
                custom = new Customization(id, value, 0, value);
                return value;
            }

            return -1;
        }

        int Get(IEnumerable<Customization> list, ref Customization? output)
        {
            var (val, idx) = list.Cast<Customization?>().Select((c, i) => (c, i)).FirstOrDefault(c => c.c!.Value.Value == value);
            if (val == null)
                return -1;

            output = val;
            return idx;
        }

        return id switch
        {
            CustomizationId.SkinColor      => Get(SkinColors,                                       ref custom),
            CustomizationId.EyeColorL      => Get(EyeColors,                                        ref custom),
            CustomizationId.EyeColorR      => Get(EyeColors,                                        ref custom),
            CustomizationId.HairColor      => Get(HairColors,                                       ref custom),
            CustomizationId.HighlightColor => Get(HighlightColors,                                  ref custom),
            CustomizationId.TattooColor    => Get(TattooColors,                                     ref custom),
            CustomizationId.LipColor       => Get(LipColorsDark.Concat(LipColorsLight),             ref custom),
            CustomizationId.FacePaintColor => Get(FacePaintColorsDark.Concat(FacePaintColorsLight), ref custom),

            CustomizationId.Face                  => Get(Faces,              ref custom),
            CustomizationId.Hairstyle             => Get(HairStyles,         ref custom),
            CustomizationId.TailEarShape          => Get(TailEarShapes,      ref custom),
            CustomizationId.FacePaint             => Get(FacePaints,         ref custom),
            CustomizationId.FacialFeaturesTattoos => Get(FeaturesTattoos[0], ref custom),
            _                                     => throw new ArgumentOutOfRangeException(nameof(id), id, null),
        };
    }

    public Customization Data(CustomizationId id, int idx)
    {
        if (idx > Count(id))
            throw new IndexOutOfRangeException();

        switch (id.ToType())
        {
            case CharaMakeParams.MenuType.Percentage:   return new Customization(id, (byte)idx, 0, (ushort)idx);
            case CharaMakeParams.MenuType.ListSelector: return new Customization(id, (byte)idx, 0, (ushort)idx);
        }

        return id switch
        {
            CustomizationId.Face                  => Faces[idx],
            CustomizationId.Hairstyle             => HairStyles[idx],
            CustomizationId.TailEarShape          => TailEarShapes[idx],
            CustomizationId.FacePaint             => FacePaints[idx],
            CustomizationId.FacialFeaturesTattoos => FeaturesTattoos[0][idx],

            CustomizationId.SkinColor      => SkinColors[idx],
            CustomizationId.EyeColorL      => EyeColors[idx],
            CustomizationId.EyeColorR      => EyeColors[idx],
            CustomizationId.HairColor      => HairColors[idx],
            CustomizationId.HighlightColor => HighlightColors[idx],
            CustomizationId.TattooColor    => TattooColors[idx],
            CustomizationId.LipColor       => idx < 96 ? LipColorsDark[idx] : LipColorsLight[idx - 96],
            CustomizationId.FacePaintColor => idx < 96 ? FacePaintColorsDark[idx] : FacePaintColorsLight[idx - 96],
            _                              => new Customization(0, 0),
        };
    }

    public CharaMakeParams.MenuType Type(CustomizationId id)
        => Types[(int)id];


    public int Count(CustomizationId id)
    {
        if (!IsAvailable(id))
            return 0;

        if (id.ToType() == CharaMakeParams.MenuType.Percentage)
            return 101;

        return id switch
        {
            CustomizationId.Face                  => Faces.Count,
            CustomizationId.Hairstyle             => HairStyles.Count,
            CustomizationId.HighlightsOnFlag      => 2,
            CustomizationId.SkinColor             => SkinColors.Count,
            CustomizationId.EyeColorR             => EyeColors.Count,
            CustomizationId.HairColor             => HairColors.Count,
            CustomizationId.HighlightColor        => HighlightColors.Count,
            CustomizationId.FacialFeaturesTattoos => 8,
            CustomizationId.TattooColor           => TattooColors.Count,
            CustomizationId.Eyebrows              => NumEyebrows,
            CustomizationId.EyeColorL             => EyeColors.Count,
            CustomizationId.EyeShape              => NumEyeShapes,
            CustomizationId.Nose                  => NumNoseShapes,
            CustomizationId.Jaw                   => NumJawShapes,
            CustomizationId.Mouth                 => NumMouthShapes,
            CustomizationId.LipColor              => LipColorsLight.Count + LipColorsDark.Count,
            CustomizationId.TailEarShape          => TailEarShapes.Count,
            CustomizationId.FacePaint             => FacePaints.Count,
            CustomizationId.FacePaintColor        => FacePaintColorsLight.Count + FacePaintColorsDark.Count,
            _                                     => throw new ArgumentOutOfRangeException(nameof(id), id, null),
        };
    }
}