mirror of
https://github.com/Ottermandias/Glamourer.git
synced 2025-12-12 18:27:24 +01:00
282 lines
11 KiB
C#
282 lines
11 KiB
C#
using System.Collections;
|
|
using System.Collections.Generic;
|
|
using System.Diagnostics;
|
|
using System.Linq;
|
|
using System.Threading.Tasks;
|
|
using Dalamud.Plugin.Services;
|
|
using Dalamud.Utility;
|
|
using FFXIVClientStructs.FFXIV.Client.Game.Object;
|
|
using Lumina.Excel.GeneratedSheets;
|
|
using OtterGui.Services;
|
|
using Penumbra.GameData.DataContainers;
|
|
using Penumbra.GameData.Structs;
|
|
|
|
namespace Glamourer.Customization;
|
|
|
|
public class NpcCustomizeSet : IAsyncDataContainer, IReadOnlyList<NpcData>
|
|
{
|
|
public string Name
|
|
=> nameof(NpcCustomizeSet);
|
|
|
|
private readonly List<NpcData> _data = [];
|
|
|
|
public long Time { get; private set; }
|
|
public long Memory { get; private set; }
|
|
public int TotalCount
|
|
=> _data.Count;
|
|
|
|
public Task Awaiter { get; }
|
|
|
|
public NpcCustomizeSet(IDataManager data, DictENpc eNpcs, DictBNpc bNpcs, DictBNpcNames bNpcNames)
|
|
{
|
|
var waitTask = Task.WhenAll(eNpcs.Awaiter, bNpcs.Awaiter, bNpcNames.Awaiter);
|
|
Awaiter = waitTask.ContinueWith(_ =>
|
|
{
|
|
var watch = Stopwatch.StartNew();
|
|
var eNpcTask = Task.Run(() => CreateEnpcData(data, eNpcs));
|
|
var bNpcTask = Task.Run(() => CreateBnpcData(data, bNpcs, bNpcNames));
|
|
FilterAndOrderNpcData(eNpcTask.Result, bNpcTask.Result);
|
|
Time = watch.ElapsedMilliseconds;
|
|
});
|
|
}
|
|
|
|
private static List<NpcData> CreateEnpcData(IDataManager data, DictENpc eNpcs)
|
|
{
|
|
var enpcSheet = data.GetExcelSheet<ENpcBase>()!;
|
|
var list = new List<NpcData>(eNpcs.Count);
|
|
|
|
foreach (var (id, name) in eNpcs)
|
|
{
|
|
var row = enpcSheet.GetRow(id.Id);
|
|
if (row == null || name.IsNullOrWhitespace())
|
|
continue;
|
|
|
|
var (valid, customize) = FromEnpcBase(row);
|
|
if (!valid)
|
|
continue;
|
|
|
|
var ret = new NpcData
|
|
{
|
|
Name = name,
|
|
Customize = customize,
|
|
Id = id,
|
|
Kind = ObjectKind.EventNpc,
|
|
};
|
|
|
|
if (row.NpcEquip.Row != 0 && row.NpcEquip.Value is { } equip)
|
|
{
|
|
ApplyNpcEquip(ref ret, equip);
|
|
}
|
|
else
|
|
{
|
|
ret.Set(0, row.ModelHead | (row.DyeHead.Row << 24));
|
|
ret.Set(1, row.ModelBody | (row.DyeBody.Row << 24));
|
|
ret.Set(2, row.ModelHands | (row.DyeHands.Row << 24));
|
|
ret.Set(3, row.ModelLegs | (row.DyeLegs.Row << 24));
|
|
ret.Set(4, row.ModelFeet | (row.DyeFeet.Row << 24));
|
|
ret.Set(5, row.ModelEars | (row.DyeEars.Row << 24));
|
|
ret.Set(6, row.ModelNeck | (row.DyeNeck.Row << 24));
|
|
ret.Set(7, row.ModelWrists | (row.DyeWrists.Row << 24));
|
|
ret.Set(8, row.ModelRightRing | (row.DyeRightRing.Row << 24));
|
|
ret.Set(9, row.ModelLeftRing | (row.DyeLeftRing.Row << 24));
|
|
ret.Mainhand = new CharacterWeapon(row.ModelMainHand | ((ulong)row.DyeMainHand.Row << 48));
|
|
ret.Offhand = new CharacterWeapon(row.ModelOffHand | ((ulong)row.DyeOffHand.Row << 48));
|
|
ret.VisorToggled = row.Visor;
|
|
}
|
|
|
|
list.Add(ret);
|
|
}
|
|
|
|
return list;
|
|
}
|
|
|
|
private static List<NpcData> CreateBnpcData(IDataManager data, DictBNpc bNpcs, DictBNpcNames bNpcNames)
|
|
{
|
|
var bnpcSheet = data.GetExcelSheet<BNpcBase>()!;
|
|
var list = new List<NpcData>((int)bnpcSheet.RowCount);
|
|
foreach (var baseRow in bnpcSheet)
|
|
{
|
|
if (baseRow.ModelChara.Value!.Type != 1)
|
|
continue;
|
|
|
|
var bnpcNameIds = bNpcNames[baseRow.RowId];
|
|
if (bnpcNameIds.Count == 0)
|
|
continue;
|
|
|
|
var (valid, customize) = FromBnpcCustomize(baseRow.BNpcCustomize.Value!);
|
|
if (!valid)
|
|
continue;
|
|
|
|
var equip = baseRow.NpcEquip.Value!;
|
|
var ret = new NpcData
|
|
{
|
|
Customize = customize,
|
|
Id = baseRow.RowId,
|
|
Kind = ObjectKind.BattleNpc,
|
|
};
|
|
ApplyNpcEquip(ref ret, equip);
|
|
foreach (var bnpcNameId in bnpcNameIds)
|
|
{
|
|
if (bNpcs.TryGetValue(bnpcNameId.Id, out var name) && !name.IsNullOrWhitespace())
|
|
list.Add(ret with { Name = name });
|
|
}
|
|
}
|
|
|
|
return list;
|
|
}
|
|
|
|
private void FilterAndOrderNpcData(List<NpcData> eNpcEquip, List<NpcData> bNpcEquip)
|
|
{
|
|
_data.Clear();
|
|
_data.EnsureCapacity(eNpcEquip.Count + bNpcEquip.Count);
|
|
var groups = eNpcEquip.Concat(bNpcEquip).GroupBy(d => d.Name).ToDictionary(g => g.Key, g => g.ToList());
|
|
foreach (var (name, duplicates) in groups.OrderBy(kvp => kvp.Key))
|
|
{
|
|
for (var i = 0; i < duplicates.Count; ++i)
|
|
{
|
|
var current = duplicates[i];
|
|
for (var j = 0; j < i; ++j)
|
|
{
|
|
if (current.DataEquals(duplicates[j]))
|
|
{
|
|
duplicates.RemoveAt(i--);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (duplicates.Count == 1)
|
|
{
|
|
_data.Add(duplicates[0]);
|
|
Memory += 96;
|
|
}
|
|
else
|
|
{
|
|
_data.AddRange(duplicates
|
|
.Select(duplicate => duplicate with
|
|
{
|
|
Name = $"{name} ({(duplicate.Kind is ObjectKind.BattleNpc ? 'B' : 'E')}{duplicate.Id})"
|
|
}));
|
|
Memory += 96 * duplicates.Count + duplicates.Sum(d => d.Name.Length * 2);
|
|
}
|
|
}
|
|
|
|
var lastWeird = _data.FindIndex(d => char.IsAsciiLetterOrDigit(d.Name[0]));
|
|
if (lastWeird != -1)
|
|
{
|
|
_data.AddRange(_data.Take(lastWeird));
|
|
_data.RemoveRange(0, lastWeird);
|
|
}
|
|
_data.TrimExcess();
|
|
}
|
|
|
|
private static void ApplyNpcEquip(ref NpcData data, NpcEquip row)
|
|
{
|
|
data.Set(0, row.ModelHead | (row.DyeHead.Row << 24));
|
|
data.Set(1, row.ModelBody | (row.DyeBody.Row << 24));
|
|
data.Set(2, row.ModelHands | (row.DyeHands.Row << 24));
|
|
data.Set(3, row.ModelLegs | (row.DyeLegs.Row << 24));
|
|
data.Set(4, row.ModelFeet | (row.DyeFeet.Row << 24));
|
|
data.Set(5, row.ModelEars | (row.DyeEars.Row << 24));
|
|
data.Set(6, row.ModelNeck | (row.DyeNeck.Row << 24));
|
|
data.Set(7, row.ModelWrists | (row.DyeWrists.Row << 24));
|
|
data.Set(8, row.ModelRightRing | (row.DyeRightRing.Row << 24));
|
|
data.Set(9, row.ModelLeftRing | (row.DyeLeftRing.Row << 24));
|
|
data.Mainhand = new CharacterWeapon(row.ModelMainHand | ((ulong)row.DyeMainHand.Row << 48));
|
|
data.Offhand = new CharacterWeapon(row.ModelOffHand | ((ulong)row.DyeOffHand.Row << 48));
|
|
data.VisorToggled = row.Visor;
|
|
}
|
|
|
|
private static (bool, Customize) FromBnpcCustomize(BNpcCustomize bnpcCustomize)
|
|
{
|
|
var customize = new Customize();
|
|
customize.Data.Set(0, (byte)bnpcCustomize.Race.Row);
|
|
customize.Data.Set(1, bnpcCustomize.Gender);
|
|
customize.Data.Set(2, bnpcCustomize.BodyType);
|
|
customize.Data.Set(3, bnpcCustomize.Height);
|
|
customize.Data.Set(4, (byte)bnpcCustomize.Tribe.Row);
|
|
customize.Data.Set(5, bnpcCustomize.Face);
|
|
customize.Data.Set(6, bnpcCustomize.HairStyle);
|
|
customize.Data.Set(7, bnpcCustomize.HairHighlight);
|
|
customize.Data.Set(8, bnpcCustomize.SkinColor);
|
|
customize.Data.Set(9, bnpcCustomize.EyeHeterochromia);
|
|
customize.Data.Set(10, bnpcCustomize.HairColor);
|
|
customize.Data.Set(11, bnpcCustomize.HairHighlightColor);
|
|
customize.Data.Set(12, bnpcCustomize.FacialFeature);
|
|
customize.Data.Set(13, bnpcCustomize.FacialFeatureColor);
|
|
customize.Data.Set(14, bnpcCustomize.Eyebrows);
|
|
customize.Data.Set(15, bnpcCustomize.EyeColor);
|
|
customize.Data.Set(16, bnpcCustomize.EyeShape);
|
|
customize.Data.Set(17, bnpcCustomize.Nose);
|
|
customize.Data.Set(18, bnpcCustomize.Jaw);
|
|
customize.Data.Set(19, bnpcCustomize.Mouth);
|
|
customize.Data.Set(20, bnpcCustomize.LipColor);
|
|
customize.Data.Set(21, bnpcCustomize.BustOrTone1);
|
|
customize.Data.Set(22, bnpcCustomize.ExtraFeature1);
|
|
customize.Data.Set(23, bnpcCustomize.ExtraFeature2OrBust);
|
|
customize.Data.Set(24, bnpcCustomize.FacePaint);
|
|
customize.Data.Set(25, bnpcCustomize.FacePaintColor);
|
|
|
|
if (customize.BodyType.Value != 1
|
|
|| !CustomizationOptions.Races.Contains(customize.Race)
|
|
|| !CustomizationOptions.Clans.Contains(customize.Clan)
|
|
|| !CustomizationOptions.Genders.Contains(customize.Gender))
|
|
return (false, Customize.Default);
|
|
|
|
return (true, customize);
|
|
}
|
|
|
|
private static (bool, Customize) FromEnpcBase(ENpcBase enpcBase)
|
|
{
|
|
if (enpcBase.ModelChara.Value?.Type != 1)
|
|
return (false, Customize.Default);
|
|
|
|
var customize = new Customize();
|
|
customize.Data.Set(0, (byte)enpcBase.Race.Row);
|
|
customize.Data.Set(1, enpcBase.Gender);
|
|
customize.Data.Set(2, enpcBase.BodyType);
|
|
customize.Data.Set(3, enpcBase.Height);
|
|
customize.Data.Set(4, (byte)enpcBase.Tribe.Row);
|
|
customize.Data.Set(5, enpcBase.Face);
|
|
customize.Data.Set(6, enpcBase.HairStyle);
|
|
customize.Data.Set(7, enpcBase.HairHighlight);
|
|
customize.Data.Set(8, enpcBase.SkinColor);
|
|
customize.Data.Set(9, enpcBase.EyeHeterochromia);
|
|
customize.Data.Set(10, enpcBase.HairColor);
|
|
customize.Data.Set(11, enpcBase.HairHighlightColor);
|
|
customize.Data.Set(12, enpcBase.FacialFeature);
|
|
customize.Data.Set(13, enpcBase.FacialFeatureColor);
|
|
customize.Data.Set(14, enpcBase.Eyebrows);
|
|
customize.Data.Set(15, enpcBase.EyeColor);
|
|
customize.Data.Set(16, enpcBase.EyeShape);
|
|
customize.Data.Set(17, enpcBase.Nose);
|
|
customize.Data.Set(18, enpcBase.Jaw);
|
|
customize.Data.Set(19, enpcBase.Mouth);
|
|
customize.Data.Set(20, enpcBase.LipColor);
|
|
customize.Data.Set(21, enpcBase.BustOrTone1);
|
|
customize.Data.Set(22, enpcBase.ExtraFeature1);
|
|
customize.Data.Set(23, enpcBase.ExtraFeature2OrBust);
|
|
customize.Data.Set(24, enpcBase.FacePaint);
|
|
customize.Data.Set(25, enpcBase.FacePaintColor);
|
|
|
|
if (customize.BodyType.Value != 1
|
|
|| !CustomizationOptions.Races.Contains(customize.Race)
|
|
|| !CustomizationOptions.Clans.Contains(customize.Clan)
|
|
|| !CustomizationOptions.Genders.Contains(customize.Gender))
|
|
return (false, Customize.Default);
|
|
|
|
return (true, customize);
|
|
}
|
|
|
|
public IEnumerator<NpcData> GetEnumerator()
|
|
=> _data.GetEnumerator();
|
|
|
|
IEnumerator IEnumerable.GetEnumerator()
|
|
=> GetEnumerator();
|
|
|
|
public int Count
|
|
=> _data.Count;
|
|
|
|
public NpcData this[int index]
|
|
=> _data[index];
|
|
}
|