437 lines
16 KiB
C#
437 lines
16 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Diagnostics;
|
|
using System.Linq;
|
|
using System.Threading;
|
|
using Dalamud.Utility;
|
|
using FFXIVClientStructs.FFXIV.Client.Game.Character;
|
|
using FFXIVClientStructs.FFXIV.Client.Graphics.Scene;
|
|
using FFXIVClientStructs.FFXIV.Client.System.Resource;
|
|
using MareSynchronos.API;
|
|
using MareSynchronos.FileCache;
|
|
using MareSynchronos.Interop;
|
|
using MareSynchronos.Managers;
|
|
using MareSynchronos.Models;
|
|
using MareSynchronos.Utils;
|
|
using Penumbra.GameData.ByteString;
|
|
using Penumbra.Interop.Structs;
|
|
using Object = FFXIVClientStructs.FFXIV.Client.Graphics.Scene.Object;
|
|
|
|
namespace MareSynchronos.Factories;
|
|
|
|
public class CharacterDataFactory
|
|
{
|
|
private readonly DalamudUtil _dalamudUtil;
|
|
private readonly IpcManager _ipcManager;
|
|
private readonly TransientResourceManager transientResourceManager;
|
|
private readonly FileCacheManager fileDbManager;
|
|
|
|
public CharacterDataFactory(DalamudUtil dalamudUtil, IpcManager ipcManager, TransientResourceManager transientResourceManager, FileCacheManager fileDbManager)
|
|
{
|
|
Logger.Verbose("Creating " + nameof(CharacterDataFactory));
|
|
this.fileDbManager = fileDbManager;
|
|
_dalamudUtil = dalamudUtil;
|
|
_ipcManager = ipcManager;
|
|
this.transientResourceManager = transientResourceManager;
|
|
}
|
|
|
|
private unsafe bool CheckForPointer(IntPtr playerPointer)
|
|
{
|
|
return playerPointer == IntPtr.Zero || ((Character*)playerPointer)->GameObject.GetDrawObject() == null;
|
|
}
|
|
|
|
public CharacterData BuildCharacterData(CharacterData previousData, PlayerRelatedObject playerRelatedObject, CancellationToken token)
|
|
{
|
|
if (!_ipcManager.Initialized)
|
|
{
|
|
throw new ArgumentException("Penumbra is not connected");
|
|
}
|
|
|
|
bool pointerIsZero = true;
|
|
try
|
|
{
|
|
pointerIsZero = CheckForPointer(playerRelatedObject.Address);
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
Logger.Warn("Could not create data for " + playerRelatedObject.ObjectKind);
|
|
Logger.Warn(ex.Message);
|
|
Logger.Warn(ex.StackTrace ?? string.Empty);
|
|
}
|
|
|
|
if (pointerIsZero)
|
|
{
|
|
Logger.Verbose("Pointer was zero for " + playerRelatedObject.ObjectKind);
|
|
previousData.FileReplacements.Remove(playerRelatedObject.ObjectKind);
|
|
previousData.GlamourerString.Remove(playerRelatedObject.ObjectKind);
|
|
return previousData;
|
|
}
|
|
|
|
var previousFileReplacements = previousData.FileReplacements.ToDictionary(d => d.Key, d => d.Value);
|
|
var previousGlamourerData = previousData.GlamourerString.ToDictionary(d => d.Key, d => d.Value);
|
|
|
|
try
|
|
{
|
|
return CreateCharacterData(previousData, playerRelatedObject, token);
|
|
}
|
|
catch (OperationCanceledException)
|
|
{
|
|
Logger.Debug("Cancelled creating Character data");
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
Logger.Warn("Failed to create " + playerRelatedObject.ObjectKind + " data");
|
|
Logger.Warn(e.Message);
|
|
Logger.Warn(e.StackTrace ?? string.Empty);
|
|
}
|
|
|
|
previousData.FileReplacements = previousFileReplacements;
|
|
previousData.GlamourerString = previousGlamourerData;
|
|
return previousData;
|
|
}
|
|
|
|
private (string, string) GetIndentationForInheritanceLevel(int inheritanceLevel)
|
|
{
|
|
return (string.Join("", Enumerable.Repeat("\t", inheritanceLevel)), string.Join("", Enumerable.Repeat("\t", inheritanceLevel + 2)));
|
|
}
|
|
|
|
private void DebugPrint(FileReplacement fileReplacement, ObjectKind objectKind, string resourceType, int inheritanceLevel)
|
|
{
|
|
var indentation = GetIndentationForInheritanceLevel(inheritanceLevel);
|
|
|
|
if (fileReplacement.HasFileReplacement)
|
|
{
|
|
Logger.Verbose(indentation.Item1 + objectKind + resourceType + " [" + string.Join(", ", fileReplacement.GamePaths) + "]");
|
|
Logger.Verbose(indentation.Item2 + "=> " + fileReplacement.ResolvedPath);
|
|
}
|
|
}
|
|
|
|
private unsafe void AddReplacementsFromRenderModel(RenderModel* mdl, ObjectKind objectKind, CharacterData cache, int inheritanceLevel = 0)
|
|
{
|
|
if (mdl == null || mdl->ResourceHandle == null || mdl->ResourceHandle->Category != ResourceCategory.Chara)
|
|
{
|
|
return;
|
|
}
|
|
|
|
string mdlPath;
|
|
try
|
|
{
|
|
mdlPath = new Utf8String(mdl->ResourceHandle->FileName()).ToString();
|
|
}
|
|
catch
|
|
{
|
|
Logger.Warn("Could not get model data for " + objectKind);
|
|
return;
|
|
}
|
|
Logger.Verbose("Checking File Replacement for Model " + mdlPath);
|
|
|
|
FileReplacement mdlFileReplacement = CreateFileReplacement(mdlPath);
|
|
DebugPrint(mdlFileReplacement, objectKind, "Model", inheritanceLevel);
|
|
|
|
cache.AddFileReplacement(objectKind, mdlFileReplacement);
|
|
|
|
for (var mtrlIdx = 0; mtrlIdx < mdl->MaterialCount; mtrlIdx++)
|
|
{
|
|
var mtrl = (Material*)mdl->Materials[mtrlIdx];
|
|
if (mtrl == null) continue;
|
|
|
|
AddReplacementsFromMaterial(mtrl, objectKind, cache, inheritanceLevel + 1);
|
|
}
|
|
}
|
|
|
|
private unsafe void AddReplacementsFromMaterial(Material* mtrl, ObjectKind objectKind, CharacterData cache, int inheritanceLevel = 0)
|
|
{
|
|
string fileName;
|
|
try
|
|
{
|
|
fileName = new Utf8String(mtrl->ResourceHandle->FileName()).ToString();
|
|
|
|
}
|
|
catch
|
|
{
|
|
Logger.Warn("Could not get material data for " + objectKind);
|
|
return;
|
|
}
|
|
|
|
Logger.Verbose("Checking File Replacement for Material " + fileName);
|
|
var mtrlPath = fileName.Split("|")[2];
|
|
|
|
if (cache.FileReplacements.ContainsKey(objectKind))
|
|
{
|
|
if (cache.FileReplacements[objectKind].Any(c => c.ResolvedPath.Contains(mtrlPath, StringComparison.Ordinal)))
|
|
{
|
|
return;
|
|
}
|
|
}
|
|
|
|
var mtrlFileReplacement = CreateFileReplacement(mtrlPath);
|
|
DebugPrint(mtrlFileReplacement, objectKind, "Material", inheritanceLevel);
|
|
|
|
cache.AddFileReplacement(objectKind, mtrlFileReplacement);
|
|
|
|
var mtrlResourceHandle = (MtrlResource*)mtrl->ResourceHandle;
|
|
for (var resIdx = 0; resIdx < mtrlResourceHandle->NumTex; resIdx++)
|
|
{
|
|
string? texPath = null;
|
|
try
|
|
{
|
|
texPath = new Utf8String(mtrlResourceHandle->TexString(resIdx)).ToString();
|
|
}
|
|
catch
|
|
{
|
|
Logger.Warn("Could not get Texture data for Material " + fileName);
|
|
}
|
|
|
|
if (string.IsNullOrEmpty(texPath)) continue;
|
|
|
|
Logger.Verbose("Checking File Replacement for Texture " + texPath);
|
|
|
|
AddReplacementsFromTexture(texPath, objectKind, cache, inheritanceLevel + 1);
|
|
}
|
|
}
|
|
|
|
private void AddReplacement(string varPath, ObjectKind objectKind, CharacterData cache, int inheritanceLevel = 0, bool doNotReverseResolve = false)
|
|
{
|
|
if (varPath.IsNullOrEmpty()) return;
|
|
|
|
if (cache.FileReplacements.ContainsKey(objectKind))
|
|
{
|
|
if (cache.FileReplacements[objectKind].Any(c => c.GamePaths.Contains(varPath, StringComparer.Ordinal)))
|
|
{
|
|
return;
|
|
}
|
|
}
|
|
|
|
var variousReplacement = CreateFileReplacement(varPath, doNotReverseResolve);
|
|
DebugPrint(variousReplacement, objectKind, "Various", inheritanceLevel);
|
|
|
|
cache.AddFileReplacement(objectKind, variousReplacement);
|
|
}
|
|
|
|
private void AddReplacementsFromTexture(string texPath, ObjectKind objectKind, CharacterData cache, int inheritanceLevel = 0, bool doNotReverseResolve = true)
|
|
{
|
|
if (string.IsNullOrEmpty(texPath)) return;
|
|
|
|
if (cache.FileReplacements.ContainsKey(objectKind))
|
|
{
|
|
if (cache.FileReplacements[objectKind].Any(c => c.GamePaths.Contains(texPath, StringComparer.Ordinal)))
|
|
{
|
|
return;
|
|
}
|
|
}
|
|
|
|
var texFileReplacement = CreateFileReplacement(texPath, doNotReverseResolve);
|
|
DebugPrint(texFileReplacement, objectKind, "Texture", inheritanceLevel);
|
|
|
|
cache.AddFileReplacement(objectKind, texFileReplacement);
|
|
|
|
if (texPath.Contains("/--", StringComparison.Ordinal)) return;
|
|
|
|
var texDx11Replacement =
|
|
CreateFileReplacement(texPath.Insert(texPath.LastIndexOf('/') + 1, "--"), doNotReverseResolve);
|
|
|
|
DebugPrint(texDx11Replacement, objectKind, "Texture (DX11)", inheritanceLevel);
|
|
|
|
cache.AddFileReplacement(objectKind, texDx11Replacement);
|
|
}
|
|
|
|
private unsafe CharacterData CreateCharacterData(CharacterData previousData, PlayerRelatedObject playerRelatedObject, CancellationToken token)
|
|
{
|
|
var objectKind = playerRelatedObject.ObjectKind;
|
|
var charaPointer = playerRelatedObject.Address;
|
|
|
|
Stopwatch st = Stopwatch.StartNew();
|
|
|
|
if (playerRelatedObject.HasUnprocessedUpdate)
|
|
{
|
|
Logger.Debug("Handling unprocessed update for " + objectKind);
|
|
|
|
if (previousData.FileReplacements.ContainsKey(objectKind))
|
|
{
|
|
previousData.FileReplacements[objectKind].Clear();
|
|
}
|
|
else
|
|
{
|
|
previousData.FileReplacements.Add(objectKind, new());
|
|
}
|
|
|
|
var chara = _dalamudUtil.CreateGameObject(charaPointer)!;
|
|
while (!_dalamudUtil.IsObjectPresent(chara))
|
|
{
|
|
Logger.Verbose("Character is null but it shouldn't be, waiting");
|
|
Thread.Sleep(50);
|
|
}
|
|
|
|
_dalamudUtil.WaitWhileCharacterIsDrawing(objectKind.ToString(), charaPointer, 15000);
|
|
|
|
var human = (Human*)((Character*)charaPointer)->GameObject.GetDrawObject();
|
|
for (var mdlIdx = 0; mdlIdx < human->CharacterBase.SlotCount; ++mdlIdx)
|
|
{
|
|
var mdl = (RenderModel*)human->CharacterBase.ModelArray[mdlIdx];
|
|
if (mdl == null || mdl->ResourceHandle == null || mdl->ResourceHandle->Category != ResourceCategory.Chara)
|
|
{
|
|
continue;
|
|
}
|
|
|
|
token.ThrowIfCancellationRequested();
|
|
|
|
AddReplacementsFromRenderModel(mdl, objectKind, previousData, 0);
|
|
}
|
|
|
|
foreach (var item in previousData.FileReplacements[objectKind])
|
|
{
|
|
transientResourceManager.RemoveTransientResource(charaPointer, item);
|
|
}
|
|
|
|
if (objectKind == ObjectKind.Player)
|
|
{
|
|
AddPlayerSpecificReplacements(previousData, objectKind, charaPointer, human);
|
|
}
|
|
|
|
if (objectKind == ObjectKind.Pet)
|
|
{
|
|
foreach (var item in previousData.FileReplacements[objectKind])
|
|
{
|
|
transientResourceManager.AddSemiTransientResource(objectKind, item);
|
|
}
|
|
|
|
previousData.FileReplacements[objectKind].Clear();
|
|
}
|
|
}
|
|
|
|
previousData.ManipulationString = _ipcManager.PenumbraGetMetaManipulations();
|
|
previousData.GlamourerString[objectKind] = _ipcManager.GlamourerGetCharacterCustomization(charaPointer);
|
|
previousData.HeelsOffset = _ipcManager.GetHeelsOffset();
|
|
|
|
Logger.Debug("Handling transient update for " + objectKind);
|
|
ManageSemiTransientData(previousData, objectKind, charaPointer);
|
|
|
|
st.Stop();
|
|
Logger.Verbose("Building " + objectKind + " Data took " + st.Elapsed);
|
|
return previousData;
|
|
}
|
|
|
|
private unsafe void ManageSemiTransientData(CharacterData previousData, ObjectKind objectKind, IntPtr charaPointer)
|
|
{
|
|
transientResourceManager.PersistTransientResources(charaPointer, objectKind, CreateFileReplacement);
|
|
|
|
foreach (var item in transientResourceManager.GetSemiTransientResources(objectKind))
|
|
{
|
|
if (!previousData.FileReplacements.ContainsKey(objectKind))
|
|
{
|
|
previousData.FileReplacements.Add(objectKind, new());
|
|
}
|
|
|
|
if (!previousData.FileReplacements[objectKind].Any(k => k.GamePaths.Any(p => item.GamePaths.Contains(p, StringComparer.OrdinalIgnoreCase))))
|
|
{
|
|
var penumResolve = _ipcManager.PenumbraResolvePath(item.GamePaths.First()).ToLowerInvariant();
|
|
var gamePath = item.GamePaths.First().ToLowerInvariant();
|
|
if (string.Equals(penumResolve, gamePath, StringComparison.Ordinal))
|
|
{
|
|
Logger.Verbose("PenumResolve was same as GamePath, not adding " + item);
|
|
transientResourceManager.RemoveTransientResource(charaPointer, item);
|
|
}
|
|
else
|
|
{
|
|
Logger.Verbose("Found semi transient resource: " + item);
|
|
previousData.FileReplacements[objectKind].Add(item);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
private unsafe void AddPlayerSpecificReplacements(CharacterData previousData, ObjectKind objectKind, IntPtr charaPointer, Human* human)
|
|
{
|
|
var weaponObject = (Weapon*)((Object*)human)->ChildObject;
|
|
|
|
if ((IntPtr)weaponObject != IntPtr.Zero)
|
|
{
|
|
var mainHandWeapon = weaponObject->WeaponRenderModel->RenderModel;
|
|
|
|
AddReplacementsFromRenderModel(mainHandWeapon, objectKind, previousData, 0);
|
|
|
|
foreach (var item in previousData.FileReplacements[objectKind])
|
|
{
|
|
transientResourceManager.RemoveTransientResource(charaPointer, item);
|
|
}
|
|
|
|
foreach (var item in transientResourceManager.GetTransientResources((IntPtr)weaponObject))
|
|
{
|
|
Logger.Verbose("Found transient weapon resource: " + item);
|
|
AddReplacement(item, objectKind, previousData, 1, true);
|
|
}
|
|
|
|
if (weaponObject->NextSibling != (IntPtr)weaponObject)
|
|
{
|
|
var offHandWeapon = ((Weapon*)weaponObject->NextSibling)->WeaponRenderModel->RenderModel;
|
|
|
|
AddReplacementsFromRenderModel(offHandWeapon, objectKind, previousData, 1);
|
|
|
|
foreach (var item in previousData.FileReplacements[objectKind])
|
|
{
|
|
transientResourceManager.RemoveTransientResource((IntPtr)offHandWeapon, item);
|
|
}
|
|
|
|
foreach (var item in transientResourceManager.GetTransientResources((IntPtr)offHandWeapon))
|
|
{
|
|
Logger.Verbose("Found transient offhand weapon resource: " + item);
|
|
AddReplacement(item, objectKind, previousData, 1, true);
|
|
}
|
|
}
|
|
}
|
|
|
|
AddReplacementSkeleton(((HumanExt*)human)->Human.RaceSexId, objectKind, previousData);
|
|
try
|
|
{
|
|
AddReplacementsFromTexture(new Utf8String(((HumanExt*)human)->Decal->FileName()).ToString(), objectKind, previousData, 0, false);
|
|
}
|
|
catch
|
|
{
|
|
Logger.Warn("Could not get Decal data");
|
|
}
|
|
try
|
|
{
|
|
AddReplacementsFromTexture(new Utf8String(((HumanExt*)human)->LegacyBodyDecal->FileName()).ToString(), objectKind, previousData, 0, false);
|
|
}
|
|
catch
|
|
{
|
|
Logger.Warn("Could not get Legacy Body Decal Data");
|
|
}
|
|
|
|
foreach (var item in previousData.FileReplacements[objectKind])
|
|
{
|
|
transientResourceManager.RemoveTransientResource(charaPointer, item);
|
|
}
|
|
}
|
|
|
|
private void AddReplacementSkeleton(ushort raceSexId, ObjectKind objectKind, CharacterData cache)
|
|
{
|
|
string raceSexIdString = raceSexId.ToString("0000");
|
|
|
|
string skeletonPath = $"chara/human/c{raceSexIdString}/skeleton/base/b0001/skl_c{raceSexIdString}b0001.sklb";
|
|
|
|
var replacement = CreateFileReplacement(skeletonPath, true);
|
|
cache.AddFileReplacement(objectKind, replacement);
|
|
|
|
DebugPrint(replacement, objectKind, "SKLB", 0);
|
|
}
|
|
|
|
private FileReplacement CreateFileReplacement(string path, bool doNotReverseResolve = false)
|
|
{
|
|
var fileReplacement = new FileReplacement(fileDbManager);
|
|
if (!doNotReverseResolve)
|
|
{
|
|
fileReplacement.GamePaths =
|
|
_ipcManager.PenumbraReverseResolvePlayer(path).ToList();
|
|
fileReplacement.SetResolvedPath(path);
|
|
}
|
|
else
|
|
{
|
|
fileReplacement.GamePaths = new List<string> { path };
|
|
fileReplacement.SetResolvedPath(_ipcManager.PenumbraResolvePath(path)!);
|
|
}
|
|
|
|
return fileReplacement;
|
|
}
|
|
}
|