PartyDamage/Plugin.cs

591 lines
20 KiB
C#

using System.Diagnostics;
using System.Globalization;
using System.Text;
using Dalamud.Game.Addon.Lifecycle;
using Dalamud.Game.Addon.Lifecycle.AddonArgTypes;
using Dalamud.Game.Gui.ContextMenu;
using Dalamud.Game.Text;
using Dalamud.IoC;
using Dalamud.Memory;
using Dalamud.Plugin;
using Dalamud.Plugin.Services;
using FFXIVClientStructs.FFXIV.Client.Game.Character;
using FFXIVClientStructs.FFXIV.Client.Game.Group;
using FFXIVClientStructs.FFXIV.Client.Graphics;
using FFXIVClientStructs.FFXIV.Client.UI;
using FFXIVClientStructs.FFXIV.Client.UI.Agent;
using FFXIVClientStructs.FFXIV.Component.GUI;
using Lumina.Excel.GeneratedSheets;
namespace PartyDamage;
public class Plugin : IDalamudPlugin {
[PluginService]
internal static IPluginLog Log { get; private set; }
[PluginService]
private IAddonLifecycle AddonLifecycle { get; init; }
[PluginService]
private IClientState ClientState { get; init; }
[PluginService]
private IFramework Framework { get; init; }
[PluginService]
private IPartyList PartyList { get; init; }
[PluginService]
internal IDalamudPluginInterface Interface { get; init; }
[PluginService]
internal ICommandManager CommandManager { get; init; }
[PluginService]
internal IContextMenu ContextMenu { get; init; }
[PluginService]
internal IDataManager DataManager { get; init; }
// [PluginService]
// internal ITextureProvider TextureProvider { get; init; }
internal Configuration Config { get; }
private Client Client { get; }
internal PluginUi Ui { get; }
private Commands Commands { get; }
private Stopwatch AlternateWatch { get; } = Stopwatch.StartNew();
private Stopwatch DelayWatch { get; } = new();
private bool _ranLastTick;
private bool _showDps = true;
private bool _reset;
private bool _wasActive;
private bool _manuallyReset = true;
private readonly byte[] _manaUsers = [
6, // cnj
7, // thm
19, // pld
24, // whm
25, // blm
26, // acn
27, // smn
28, // sch
32, // drk
33, // ast
35, // rdm
36, // blu
40, // sge
42, // pct
];
public Plugin() {
this.Config = this.Interface!.GetPluginConfig() as Configuration ?? new Configuration();
this.Client = new Client();
this.Ui = new PluginUi(this);
this.Commands = new Commands(this);
this.AddonLifecycle!.RegisterListener(AddonEvent.PostUpdate, "_PartyList", this.UpdateList);
this.ContextMenu!.OnMenuOpened += this.MenuOpened;
}
public void Dispose() {
this.ContextMenu.OnMenuOpened -= this.MenuOpened;
this.AddonLifecycle.UnregisterListener(AddonEvent.PostUpdate, "_PartyList", this.UpdateList);
this.Commands.Dispose();
this.Ui.Dispose();
this.Client.Dispose();
}
internal void SaveConfig() {
this.Interface.SavePluginConfig(this.Config);
}
private unsafe void MenuOpened(IMenuOpenedArgs args) {
var add = args.AddonName == "_PartyList";
if (!add) {
var ctx = AgentContext.Instance();
if (args.AgentPtr != (nint) ctx) {
return;
}
var targetId = ctx->TargetObjectId.ObjectId;
if (targetId == 0xE000_0000) {
return;
}
if (this.ClientState.LocalPlayer is { } player && player.GameObjectId == targetId) {
add = true;
} else {
var group = GroupManager.Instance()->GetGroup();
if (group != null && group->GetPartyMemberByEntityId(targetId) != null) {
add = true;
}
}
}
if (!add) {
return;
}
args.AddMenuItem(new MenuItem {
Name = "Clear parse results",
Prefix = SeIconChar.BoxedLetterP,
PrefixColor = 37,
IsEnabled = this.Client.Data?.IsActive == false,
OnClicked = _ => {
this._reset = true;
this._manuallyReset = true;
},
});
}
private unsafe void UpdateList(AddonEvent type, AddonArgs args) {
var ranLast = this._ranLastTick;
this._ranLastTick = false;
var list = (AddonPartyList*) AtkStage.Instance()->RaptureAtkUnitManager->GetAddonByName("_PartyList");
if (list == null) {
return;
}
var shouldUpdate = false;
var wasActive = this._wasActive;
var becameInactive = false;
if (this.Client.Data is { } data) {
if (wasActive && !data.IsActive) {
becameInactive = true;
}
this._wasActive = data.IsActive;
}
if (this.Client.Data is { IsActive: true }) {
shouldUpdate = true;
this._manuallyReset = false;
} else if (becameInactive) {
if (this.Config.ClearResultsOnInactive) {
this.DelayWatch.Restart();
}
this.HandleEvaluation();
}
if (this.Config.ClearResultsOnInactive && this.DelayWatch.IsRunning && this.DelayWatch.Elapsed >= TimeSpan.FromSeconds(this.Config.ClearDelaySeconds)) {
this.DelayWatch.Reset();
this._reset = true;
}
if (!this._reset && !this._manuallyReset) {
// only do these checks if we shouldn't reset and if we're waiting for active data
// keep running if we're in the delay period
if (this.DelayWatch.IsRunning) {
shouldUpdate = true;
}
// keep running if the user wants to manually clear results
if (!this.Config.ClearResultsOnInactive) {
shouldUpdate = true;
}
}
if (this._reset) {
this.DelayWatch.Reset();
this._reset = false;
this._manuallyReset = true;
this.ResetMembers(list);
}
if (shouldUpdate && this.UpdateListInner(list) && ranLast) {
this.ResetMembers(list);
}
}
private unsafe void HandleEvaluation() {
if (
!this.Config.UseEvaluatorNpc
|| this.ClientState.LocalPlayer is not { } player
|| this.Client.Data == null
) {
return;
}
var evaluator = this.GetEvaluator();
if (evaluator == null) {
return;
}
if (this.Client.Data.Combatants.Count < this.Config.EvaluationMinCombatants) {
return;
}
var combatants = this.Client.Data.Combatants.Values
.Where(combatant => {
var job = this.DataManager.GetExcelSheet<ClassJob>()!
.FirstOrDefault(j => j.Abbreviation.RawString.Equals(combatant.JobAbbr, StringComparison.InvariantCultureIgnoreCase));
return job?.Role == player.ClassJob.GameData?.Role;
})
.ToList();
if (combatants.Count < this.Config.EvaluationMinSameRole) {
return;
}
combatants.Sort((a, b) => a.EncDps.CompareTo(b.EncDps));
var youIndex = combatants.FindIndex(combatant => combatant.Name == "YOU");
if (youIndex == -1) {
return;
}
var rank = (float) (youIndex + 1) / combatants.Count;
var thresholds = this.Config.EvaluationThresholds.ToList();
thresholds.Sort((a, b) => b.Value.CompareTo(a.Value));
foreach (var threshold in thresholds) {
if (rank < threshold.Value) {
continue;
}
var msg = evaluator.GetLineFor(threshold.Key, this.Config.BlendEvaluations, youIndex + 1);
if (msg != null) {
var bytes = msg.EncodeWithNullTerminator();
UIModule.Instance()->ShowBattleTalkImage(
Encoding.UTF8.GetBytes(evaluator.Name),
bytes,
this.Config.EvaluationLength,
evaluator.BattleTalkImage,
6
);
}
break;
}
return;
}
internal Evaluator? GetEvaluator() {
if (this.Config.EvaluatorId != Evaluator.RandomId) {
return Evaluator.Evaluators.FirstOrDefault(e => e.Id == this.Config.EvaluatorId);
}
var pool = Evaluator.Evaluators
.Where(e => this.Config.EvaluatorsRandomEnabled.GetValueOrDefault(e.Id))
.ToArray();
return pool.Length == 0
? null
: Random.Shared.GetItems(pool, 1).GetIndexOrDefault(0);
}
/// <summary>
/// Update the party list.
/// </summary>
/// <returns>true if the list should be reset immediately</returns>
private unsafe bool UpdateListInner(AddonPartyList* list) {
if (this.Client.Data is not { } data) {
return false;
}
if (this.ClientState.LocalPlayer is not { } player) {
return false;
}
var chara = (Character*) player.Address;
var playerName = player.Name.TextValue;
if (list->HoveredIndex >= 0 || list->TargetedIndex >= 0) {
return true;
}
var names = new List<(string, int)>();
var members = AgentHUD.Instance()->PartyMembers;
for (var i = 0; i < members.Length && i < list->PartyMembers.Length; i++) {
var member = members[i];
if (member.Name == null) {
continue;
}
// controller soft target not handled by hoveredindex above
if (chara->GetSoftTargetId() == member.EntityId) {
return true;
}
var name = MemoryHelper.ReadStringNullTerminated((nint) member.Name);
names.Add((name, i));
}
this._ranLastTick = true;
if (this.AlternateWatch.Elapsed.TotalMilliseconds > this.Config.AlternateSeconds * 1_000) {
this.AlternateWatch.Restart();
this._showDps ^= true;
}
for (var i = 0; i < names.Count; i++) {
var (name, membersIdx) = names[i];
var lookupName = name == playerName
? "YOU"
: name;
var member = members[membersIdx];
if (!data.Combatants.TryGetValue(lookupName, out var combatant)) {
continue;
}
this.UpdateMember(list, i, member.Object, data.Encounter, combatant, data.Combatants.Values);
}
return false;
}
private unsafe void ResetMember(
AddonPartyList* list,
BattleChara* chara,
int listIndex,
bool includeBar,
bool includeTargeted
) {
var unit = list->PartyMembers[listIndex];
if (includeBar && unit.TargetGlow != null && (includeTargeted || list->TargetedIndex != listIndex)) {
unit.TargetGlow->SetAlpha(255);
unit.TargetGlow->SetScaleX(0);
unit.TargetGlow->AddRed = 0;
unit.TargetGlow->AddGreen = 0;
unit.TargetGlow->AddBlue = 0;
unit.TargetGlow->MultiplyRed = 100;
unit.TargetGlow->MultiplyGreen = 100;
unit.TargetGlow->MultiplyBlue = 100;
unit.TargetGlow->ToggleVisibility(true);
}
var left = (AtkTextNode*) unit.MPGaugeBar->GetTextNodeById(2);
var right = (AtkTextNode*) unit.MPGaugeBar->GetTextNodeById(3);
var hasLeft = left != null && left->IsVisible();
var hasRight = right != null && right->IsVisible();
var hasChara = chara != null;
if (!hasLeft) {
return;
}
left->TextColor = new ByteColor {
RGBA = 0xFFFFFFFF,
};
var manaString = hasChara
? chara->Mana.ToString(CultureInfo.InvariantCulture)
: "???";
if (hasRight) {
if (manaString.Length <= 1) {
left->SetText("");
right->SetText(manaString);
} else {
left->SetText(manaString[..^2]);
right->SetText(manaString[^2..]);
}
} else {
left->SetText(manaString);
}
left->AddRed = 0;
left->AddGreen = 0;
left->AddBlue = 0;
left->MultiplyRed = 100;
left->MultiplyGreen = 100;
left->MultiplyBlue = 100;
if (hasRight) {
right->TextColor = left->TextColor;
right->AddRed = left->AddRed;
right->AddGreen = left->AddGreen;
right->AddBlue = left->AddBlue;
right->MultiplyRed = left->MultiplyRed;
right->MultiplyGreen = left->MultiplyGreen;
right->MultiplyBlue = left->MultiplyBlue;
}
// fixme: need to figure out a way to get the game to repop
// var defaultName = new StringBuilder("\ue06a");
// foreach (var digit in member.Object->Level.ToString(CultureInfo.InvariantCulture)) {
// var offset = digit - '0';
// defaultName.Append((char) ('\ue060' + offset));
// }
// defaultName.Append(' ');
// defaultName.Append(member.Object->NameString);
// unit.Name->SetText(defaultName.ToString());
// unit.Name->TextColor = new ByteColor {
// RGBA = 0xFFFFFFFF,
// };
// unit.Name->AddRed = 0;
// unit.Name->AddGreen = 0;
// unit.Name->AddBlue = 0;
// unit.Name->MultiplyRed = 100;
// unit.Name->MultiplyGreen = 100;
// unit.Name->MultiplyBlue = 100;
}
private unsafe void ResetMembers(AddonPartyList* list) {
var members = AgentHUD.Instance()->PartyMembers;
for (var i = 0; i < members.Length && i < list->PartyMembers.Length; i++) {
this.ResetMember(list, members[i].Object, i, true, false);
}
}
private unsafe string[] GetPartyNames(bool alliance) {
var group = GroupManager.Instance()->GetGroup();
if (group == null) {
return [];
}
var length = group->PartyMembers.Length;
if (alliance) {
length += group->AllianceMembers.Length;
}
var members = new string[length];
int i;
for (i = 0; i < group->PartyMembers.Length; i++) {
var member = group->PartyMembers[i];
members[i] = member.NameString;
}
if (alliance) {
for (var j = 0; j < group->AllianceMembers.Length; j++) {
var member = group->AllianceMembers[j];
members[i + j] = member.NameString;
}
}
return members;
}
private unsafe void UpdateMember(
AddonPartyList* list,
int listIndex,
BattleChara* chara,
Encounter encounter,
Combatant combatant,
IEnumerable<Combatant> combatants
) {
var member = list->PartyMembers[listIndex];
if (this.Config.UseDpsBar && member.TargetGlow != null) {
float denominator;
switch (this.Config.DpsBarMode) {
case DpsBarMode.Encounter: {
denominator = combatants.Select(c => c.EncDps).Max();
break;
}
case DpsBarMode.Party: {
var party = this.GetPartyNames(false);
denominator = combatants
.Where(c => c.Name == "YOU" || party.Contains(c.Name))
.Select(c => c.EncDps)
.Max();
break;
}
case DpsBarMode.Alliance: {
var alliance = this.GetPartyNames(true);
denominator = combatants
.Where(c => c.Name == "YOU" || alliance.Contains(c.Name))
.Select(c => c.EncDps)
.Max();
break;
}
default: {
denominator = encounter.EncDps;
break;
}
}
member.TargetGlow->ToggleVisibility(true);
member.TargetGlow->SetAlpha((byte) Math.Round(this.Config.BarAlpha * 255));
member.TargetGlow->SetScaleX(
denominator == 0
? 0
: combatant.EncDps / denominator
);
member.TargetGlow->AddRed = (byte) Math.Clamp(this.Config.BarAddRed, 0, 255);
member.TargetGlow->AddGreen = (byte) Math.Clamp(this.Config.BarAddGreen, 0, 255);
member.TargetGlow->AddBlue = (byte) Math.Clamp(this.Config.BarAddBlue, 0, 255);
member.TargetGlow->MultiplyRed = (byte) Math.Clamp(this.Config.BarMulRed, 0, 100);
member.TargetGlow->MultiplyGreen = (byte) Math.Clamp(this.Config.BarMulGreen, 0, 100);
member.TargetGlow->MultiplyBlue = (byte) Math.Clamp(this.Config.BarMulBlue, 0, 100);
}
if (this.Config.Mode == MeterMode.Mana) {
var left = (AtkTextNode*) member.MPGaugeBar->GetTextNodeById(2);
var right = (AtkTextNode*) member.MPGaugeBar->GetTextNodeById(3);
var hasLeft = left != null && left->IsVisible();
var hasRight = right != null && right->IsVisible();
var hasChara = chara != null;
if (!hasLeft) {
return;
}
if (this.Config.Alternate && !this._showDps) {
var isCaster = hasChara && Array.IndexOf(this._manaUsers, chara->ClassJob) != -1;
if (!this.Config.ManaModeAlternateOnlyManaUsers || isCaster) {
this.ResetMember(list, chara, listIndex, false, false);
return;
}
}
left->TextColor = new ByteColor {
RGBA = this.Config.TextColour,
};
left->AddRed = (byte) Math.Clamp(this.Config.TextAddRed, 0, 255);
left->AddGreen = (byte) Math.Clamp(this.Config.TextAddGreen, 0, 255);
left->AddBlue = (byte) Math.Clamp(this.Config.TextAddBlue, 0, 255);
left->MultiplyRed = (byte) Math.Clamp(this.Config.TextMulRed, 0, 100);
left->MultiplyGreen = (byte) Math.Clamp(this.Config.TextMulGreen, 0, 100);
left->MultiplyBlue = (byte) Math.Clamp(this.Config.TextMulBlue, 0, 100);
if (hasRight) {
right->TextColor = left->TextColor;
right->AddRed = left->AddRed;
right->AddGreen = left->AddGreen;
right->AddBlue = left->AddBlue;
right->MultiplyRed = left->MultiplyRed;
right->MultiplyGreen = left->MultiplyGreen;
right->MultiplyBlue = left->MultiplyBlue;
}
string leftText, rightText;
if (combatant.EncDps == 0 || float.IsInfinity(combatant.EncDps) || float.IsNaN(combatant.EncDps)) {
leftText = "0.";
rightText = "00";
} else if (combatant.EncDps < 1_000) {
var dps = Math.Round(combatant.EncDps * 100).ToString(CultureInfo.InvariantCulture);
leftText = $"{dps[..^2]}.";
rightText = dps[^2..];
} else if (combatant.EncDps < 1_000_000) {
var dps = Math.Round(combatant.EncDps / 100).ToString(CultureInfo.InvariantCulture);
leftText = $"{dps[..^1]}.";
rightText = $"{dps[^1..]}K";
} else if (combatant.EncDps < 1_000_000_000) {
var dps = Math.Round(combatant.EncDps / 100_000).ToString(CultureInfo.InvariantCulture);
leftText = $"{dps[..^1]}.";
rightText = $"{dps[^1..]}M";
} else {
var dps = Math.Round(combatant.EncDps / 100_000_000).ToString(CultureInfo.InvariantCulture);
leftText = $"{dps[..^1]}.";
rightText = $"{dps[^1..]}B";
}
if (hasRight) {
left->SetText(leftText);
right->SetText(rightText);
} else {
left->SetText($"{leftText}{rightText}");
}
}
}
}