442 lines
17 KiB
C#
442 lines
17 KiB
C#
using System.Linq;
|
|
using System.Numerics;
|
|
using Content.Shared.Atmos;
|
|
using Content.Shared.Damage.Components;
|
|
using Content.Shared.Damage.Prototypes;
|
|
using Content.Shared.FixedPoint;
|
|
using Content.Shared.Humanoid;
|
|
using Content.Shared.Humanoid.Prototypes;
|
|
using Content.Shared.IdentityManagement;
|
|
using Content.Shared.MedicalScanner;
|
|
using Content.Shared.Mobs;
|
|
using Content.Shared.Mobs.Components;
|
|
using Robust.Client.AutoGenerated;
|
|
using Robust.Client.GameObjects;
|
|
using Robust.Client.Graphics;
|
|
using Robust.Client.ResourceManagement;
|
|
using Robust.Client.UserInterface.Controls;
|
|
using Robust.Client.UserInterface.XAML;
|
|
using Robust.Shared.Prototypes;
|
|
using Robust.Shared.Utility;
|
|
namespace Content.Client.HealthAnalyzer.UI;
|
|
|
|
using Content.Client._DV.Traits.Assorted; // DeltaV
|
|
using Content.Shared._DV.Traits.Assorted; // DeltaV
|
|
using Content.Shared._DV.Medical; // DeltaV - Uncloneable
|
|
using Content.Shared._DV.MedicalRecords; // DeltaV - Medical Records
|
|
using Content.Shared._Shitmed.Targeting; // Shitmed
|
|
|
|
// Health analyzer UI is split from its window because it's used by both the
|
|
// health analyzer item and the cryo pod UI.
|
|
|
|
[GenerateTypedNameReferences]
|
|
public sealed partial class HealthAnalyzerControl : BoxContainer
|
|
{
|
|
private readonly IEntityManager _entityManager;
|
|
private readonly SpriteSystem _spriteSystem;
|
|
private readonly IPrototypeManager _prototypes;
|
|
private readonly IResourceCache _cache;
|
|
|
|
private readonly UnborgableSystem _unborgable; // DeltaV
|
|
private readonly RedshirtSystem _redshirt; // DeltaV
|
|
private readonly UncloneableSystem _uncloneable; // DeltaV
|
|
|
|
// Shitmed Change Start
|
|
public event Action<TargetBodyPart?, EntityUid>? OnBodyPartSelected;
|
|
private EntityUid _spriteViewEntity;
|
|
|
|
private readonly EntProtoId _bodyView = "AlertSpriteView";
|
|
|
|
private readonly Dictionary<TargetBodyPart, TextureButton> _bodyPartControls;
|
|
private EntityUid? _target;
|
|
// Shitmed Change End
|
|
|
|
// Begin DeltaV - Medical Records
|
|
private readonly ButtonGroup _triageStatusGroup = new();
|
|
private readonly Dictionary<TriageStatus, Button> _triageControls = new();
|
|
|
|
public event Action<TriageStatus>? OnTriageStatusChanged;
|
|
public event Action? OnClaimPatient;
|
|
// End DeltaV - Medical Records
|
|
|
|
public HealthAnalyzerControl()
|
|
{
|
|
RobustXamlLoader.Load(this);
|
|
|
|
var dependencies = IoCManager.Instance!;
|
|
_entityManager = dependencies.Resolve<IEntityManager>();
|
|
_spriteSystem = _entityManager.System<SpriteSystem>();
|
|
_prototypes = dependencies.Resolve<IPrototypeManager>();
|
|
_cache = dependencies.Resolve<IResourceCache>();
|
|
|
|
_unborgable = _entityManager.System<UnborgableSystem>(); // DeltaV
|
|
_redshirt = _entityManager.System<RedshirtSystem>(); // DeltaV
|
|
_uncloneable = _entityManager.System<UncloneableSystem>(); // DeltaV
|
|
// Shitmed Change Start
|
|
_bodyPartControls = new Dictionary<TargetBodyPart, TextureButton>
|
|
{
|
|
{ TargetBodyPart.Head, HeadButton },
|
|
{ TargetBodyPart.Torso, ChestButton },
|
|
{ TargetBodyPart.Groin, GroinButton },
|
|
{ TargetBodyPart.LeftArm, LeftArmButton },
|
|
{ TargetBodyPart.LeftHand, LeftHandButton },
|
|
{ TargetBodyPart.RightArm, RightArmButton },
|
|
{ TargetBodyPart.RightHand, RightHandButton },
|
|
{ TargetBodyPart.LeftLeg, LeftLegButton },
|
|
{ TargetBodyPart.LeftFoot, LeftFootButton },
|
|
{ TargetBodyPart.RightLeg, RightLegButton },
|
|
{ TargetBodyPart.RightFoot, RightFootButton },
|
|
};
|
|
|
|
foreach (var bodyPartButton in _bodyPartControls)
|
|
{
|
|
bodyPartButton.Value.MouseFilter = MouseFilterMode.Stop;
|
|
bodyPartButton.Value.OnPressed += _ => SetActiveBodyPart(bodyPartButton.Key, bodyPartButton.Value);
|
|
}
|
|
ReturnButton.OnPressed += _ => ResetBodyPart();
|
|
// Shitmed Change End
|
|
|
|
// Begin DeltaV - Medical Records
|
|
foreach (var item in Enum.GetValues<TriageStatus>())
|
|
{
|
|
var btn = new Button();
|
|
var ftlKey = item.ToString();
|
|
btn.Group = _triageStatusGroup;
|
|
btn.Text = Loc.GetString($"health-analyzer-window-triage-status-{ftlKey}");
|
|
btn.ToolTip = Loc.GetString($"health-analyzer-window-triage-status-{ftlKey}.ToolTip");
|
|
btn.OnPressed += _ => OnTriageStatusChanged?.Invoke(item);
|
|
btn.AddStyleClass("ButtonSquare");
|
|
StatusBox.AddChild(btn);
|
|
_triageControls[item] = btn;
|
|
}
|
|
StatusBox.Children.First().AddStyleClass("OpenRight");
|
|
StatusBox.Children.First().RemoveStyleClass("ButtonSquare");
|
|
StatusBox.Children.Last().AddStyleClass("OpenLeft");
|
|
StatusBox.Children.Last().RemoveStyleClass("ButtonSquare");
|
|
ClaimButton.OnPressed += _ => OnClaimPatient?.Invoke();
|
|
// End DeltaV - Medical Records
|
|
}
|
|
|
|
public void Populate(HealthAnalyzerUiState state)
|
|
{
|
|
var target = _entityManager.GetEntity(state.TargetEntity);
|
|
// Begin Shitmed
|
|
_target = target;
|
|
EntityUid? part = state.Part != null ? _entityManager.GetEntity(state.Part.Value) : null;
|
|
var isPart = part != null;
|
|
// End Shitmed
|
|
|
|
if (target == null
|
|
|| !_entityManager.TryGetComponent<DamageableComponent>(isPart ? part : target, out var damageable)) // Shitmed
|
|
{
|
|
NoPatientDataText.Visible = true;
|
|
return;
|
|
}
|
|
|
|
// Begin Shitmed
|
|
ReturnButton.Visible = isPart;
|
|
PartNameLabel.Visible = isPart;
|
|
|
|
if (part != null)
|
|
PartNameLabel.Text = _entityManager.HasComponent<MetaDataComponent>(part)
|
|
? Identity.Name(part.Value, _entityManager)
|
|
: Loc.GetString("health-analyzer-window-entity-unknown-value-text");
|
|
// End Shitmed
|
|
|
|
NoPatientDataText.Visible = false;
|
|
|
|
// Scan Mode
|
|
|
|
ScanModeLabel.Text = state.ScanMode.HasValue
|
|
? state.ScanMode.Value
|
|
? Loc.GetString("health-analyzer-window-scan-mode-active")
|
|
: Loc.GetString("health-analyzer-window-scan-mode-inactive")
|
|
: Loc.GetString("health-analyzer-window-entity-unknown-text");
|
|
|
|
ScanModeLabel.FontColorOverride = state.ScanMode.HasValue && state.ScanMode.Value ? Color.Green : Color.Red;
|
|
|
|
// Patient Information
|
|
|
|
SpriteView.SetEntity(SetupIcon(state.Body) ?? target.Value); // Shitmed Change
|
|
SpriteView.Visible = state.ScanMode.HasValue && state.ScanMode.Value;
|
|
PartView.Visible = SpriteView.Visible; // Shitmed Change
|
|
NoDataTex.Visible = !SpriteView.Visible;
|
|
|
|
var name = new FormattedMessage();
|
|
name.PushColor(Color.White);
|
|
name.AddText(_entityManager.HasComponent<MetaDataComponent>(target.Value)
|
|
? Identity.Name(target.Value, _entityManager)
|
|
: Loc.GetString("health-analyzer-window-entity-unknown-text"));
|
|
NameLabel.SetMessage(name);
|
|
|
|
SpeciesLabel.Text =
|
|
_entityManager.TryGetComponent<HumanoidAppearanceComponent>(target.Value,
|
|
out var humanoidAppearanceComponent)
|
|
? Loc.GetString(_prototypes.Index<SpeciesPrototype>(humanoidAppearanceComponent.Species).Name)
|
|
: Loc.GetString("health-analyzer-window-entity-unknown-species-text");
|
|
|
|
// Basic Diagnostic
|
|
|
|
TemperatureLabel.Text = !float.IsNaN(state.Temperature)
|
|
? $"{state.Temperature - Atmospherics.T0C:F1} °C ({state.Temperature:F1} K)"
|
|
: Loc.GetString("health-analyzer-window-entity-unknown-value-text");
|
|
|
|
BloodLabel.Text = !float.IsNaN(state.BloodLevel)
|
|
? $"{state.BloodLevel * 100:F1} %"
|
|
: Loc.GetString("health-analyzer-window-entity-unknown-value-text");
|
|
|
|
StatusLabel.Text =
|
|
_entityManager.TryGetComponent<MobStateComponent>(target.Value, out var mobStateComponent)
|
|
? GetStatus(mobStateComponent.CurrentState)
|
|
: Loc.GetString("health-analyzer-window-entity-unknown-text");
|
|
|
|
// Total Damage
|
|
|
|
DamageLabel.Text = damageable.TotalDamage.ToString();
|
|
|
|
// Alerts
|
|
// DeltaV traits - This is going to be horrid if we just keep adding things like this.
|
|
var unborgable = _unborgable.IsUnborgable(target.Value);
|
|
var redshirt = _redshirt.IsRedshirt(target.Value) && mobStateComponent?.CurrentState == MobState.Dead; // DeltaV - Redshirt
|
|
var uncloneable = _uncloneable.IsUncloneable(target.Value) && mobStateComponent?.CurrentState == MobState.Dead; // DeltaV - Unclonable
|
|
// END DeltaV
|
|
|
|
var showAlerts = state.Unrevivable == true || state.Bleeding == true;
|
|
|
|
AlertsDivider.Visible = showAlerts;
|
|
AlertsContainer.Visible = showAlerts;
|
|
|
|
if (showAlerts)
|
|
AlertsContainer.RemoveAllChildren();
|
|
|
|
if (state.Unrevivable == true)
|
|
AlertsContainer.AddChild(new RichTextLabel
|
|
{
|
|
Text = Loc.GetString("health-analyzer-window-entity-unrevivable-text"),
|
|
Margin = new Thickness(0, 4),
|
|
MaxWidth = 300
|
|
});
|
|
|
|
if (state.Bleeding == true)
|
|
AlertsContainer.AddChild(new RichTextLabel
|
|
{
|
|
Text = Loc.GetString("health-analyzer-window-entity-bleeding-text"),
|
|
Margin = new Thickness(0, 4),
|
|
MaxWidth = 300
|
|
});
|
|
|
|
if (unborgable) // DeltaV
|
|
AlertsContainer.AddChild(new RichTextLabel
|
|
{
|
|
Text = Loc.GetString("health-analyzer-window-entity-unborgable-text"),
|
|
Margin = new Thickness(0, 4),
|
|
MaxWidth = 300
|
|
});
|
|
|
|
if (redshirt) // DeltaV
|
|
AlertsContainer.AddChild(new RichTextLabel
|
|
{
|
|
Text = Loc.GetString("health-analyzer-window-entity-redshirt-text"),
|
|
Margin = new Thickness(0, 4),
|
|
MaxWidth = 300
|
|
});
|
|
|
|
if (uncloneable) // DeltaV - Uncloneable
|
|
AlertsContainer.AddChild(new RichTextLabel
|
|
{
|
|
Text = Loc.GetString("health-analyzer-window-entity-uncloneable-text"),
|
|
Margin = new Thickness(0, 4),
|
|
MaxWidth = 300
|
|
});
|
|
|
|
// Damage Groups
|
|
|
|
var damageSortedGroups =
|
|
damageable.DamagePerGroup.OrderByDescending(damage => damage.Value)
|
|
.ToDictionary(x => x.Key, x => x.Value);
|
|
|
|
IReadOnlyDictionary<string, FixedPoint2> damagePerType = damageable.Damage.DamageDict;
|
|
|
|
DrawDiagnosticGroups(damageSortedGroups, damagePerType);
|
|
|
|
// Begin DeltaV - Medical Records
|
|
if (state.MedicalRecord is not { } records)
|
|
{
|
|
TriageControls.Visible = false;
|
|
return;
|
|
}
|
|
|
|
TriageControls.Visible = true;
|
|
_triageControls[records.Status].Pressed = true;
|
|
|
|
// Update claim button based on claimed status
|
|
if (records.ClaimedName != null)
|
|
{
|
|
ClaimButton.Text = Loc.GetString("health-analyzer-window-triage-unclaim", ("claimedBy", records.ClaimedName));
|
|
}
|
|
else
|
|
{
|
|
ClaimButton.Text = Loc.GetString("health-analyzer-window-triage-claim");
|
|
}
|
|
// End DeltaV - Medical Records
|
|
}
|
|
|
|
private static string GetStatus(MobState mobState)
|
|
{
|
|
return mobState switch
|
|
{
|
|
MobState.Alive => Loc.GetString("health-analyzer-window-entity-alive-text"),
|
|
MobState.Critical => Loc.GetString("health-analyzer-window-entity-critical-text"),
|
|
MobState.Dead => Loc.GetString("health-analyzer-window-entity-dead-text"),
|
|
_ => Loc.GetString("health-analyzer-window-entity-unknown-text"),
|
|
};
|
|
}
|
|
|
|
private void DrawDiagnosticGroups(
|
|
Dictionary<string, FixedPoint2> groups,
|
|
IReadOnlyDictionary<string, FixedPoint2> damageDict)
|
|
{
|
|
GroupsContainer.RemoveAllChildren();
|
|
|
|
foreach (var (damageGroupId, damageAmount) in groups)
|
|
{
|
|
if (damageAmount == 0)
|
|
continue;
|
|
|
|
var groupTitleText = $"{Loc.GetString(
|
|
"health-analyzer-window-damage-group-text",
|
|
("damageGroup", _prototypes.Index<DamageGroupPrototype>(damageGroupId).LocalizedName),
|
|
("amount", damageAmount)
|
|
)}";
|
|
|
|
var groupContainer = new BoxContainer
|
|
{
|
|
Align = AlignMode.Begin,
|
|
Orientation = LayoutOrientation.Vertical,
|
|
};
|
|
|
|
groupContainer.AddChild(CreateDiagnosticGroupTitle(groupTitleText, damageGroupId));
|
|
|
|
GroupsContainer.AddChild(groupContainer);
|
|
|
|
// Show the damage for each type in that group.
|
|
var group = _prototypes.Index<DamageGroupPrototype>(damageGroupId);
|
|
|
|
foreach (var type in group.DamageTypes)
|
|
{
|
|
if (!damageDict.TryGetValue(type, out var typeAmount) || typeAmount <= 0)
|
|
continue;
|
|
|
|
var damageString = Loc.GetString(
|
|
"health-analyzer-window-damage-type-text",
|
|
("damageType", _prototypes.Index<DamageTypePrototype>(type).LocalizedName),
|
|
("amount", typeAmount)
|
|
);
|
|
|
|
groupContainer.AddChild(CreateDiagnosticItemLabel(damageString.Insert(0, " · ")));
|
|
}
|
|
}
|
|
}
|
|
|
|
// Shitmed Change Start
|
|
public void SetActiveBodyPart(TargetBodyPart part, TextureButton button)
|
|
{
|
|
if (_target == null)
|
|
return;
|
|
|
|
// Bit of the ole shitcode until we have Groins in the prototypes.
|
|
OnBodyPartSelected?.Invoke(part == TargetBodyPart.Groin ? TargetBodyPart.Torso : part, _target.Value);
|
|
}
|
|
|
|
public void ResetBodyPart()
|
|
{
|
|
if (_target == null)
|
|
return;
|
|
|
|
OnBodyPartSelected?.Invoke(null, _target.Value);
|
|
}
|
|
|
|
public void SetActiveButtons(bool isHumanoid)
|
|
{
|
|
foreach (var button in _bodyPartControls)
|
|
button.Value.Visible = isHumanoid;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Sets up the Body Doll using Alert Entity to use in Health Analyzer.
|
|
/// </summary>
|
|
private EntityUid? SetupIcon(Dictionary<TargetBodyPart, TargetIntegrity>? body)
|
|
{
|
|
if (body is null)
|
|
return null;
|
|
|
|
if (!_entityManager.Deleted(_spriteViewEntity))
|
|
_entityManager.QueueDeleteEntity(_spriteViewEntity);
|
|
|
|
_spriteViewEntity = _entityManager.Spawn(_bodyView);
|
|
|
|
if (!_entityManager.TryGetComponent<SpriteComponent>(_spriteViewEntity, out var sprite))
|
|
return null;
|
|
|
|
int layer = 0;
|
|
foreach (var (bodyPart, integrity) in body)
|
|
{
|
|
// TODO: PartStatusUIController and make it use layers instead of TextureRects when EE refactors alerts.
|
|
string enumName = Enum.GetName(typeof(TargetBodyPart), bodyPart) ?? "Unknown";
|
|
int enumValue = (int) integrity;
|
|
var rsi = new SpriteSpecifier.Rsi(new ResPath($"/Textures/_Shitmed/Interface/Targeting/Status/{enumName.ToLowerInvariant()}.rsi"), $"{enumName.ToLowerInvariant()}_{enumValue}");
|
|
// Shitcode with love from Russia :)
|
|
if (!_spriteSystem.TryGetLayer(_spriteViewEntity, layer, out _, false))
|
|
_spriteSystem.AddTextureLayer(_spriteViewEntity, _spriteSystem.Frame0(rsi));
|
|
else
|
|
_spriteSystem.LayerSetTexture(_spriteViewEntity, layer, _spriteSystem.Frame0(rsi));
|
|
_spriteSystem.LayerSetScale(_spriteViewEntity, layer, new Vector2(3f, 3f));
|
|
layer++;
|
|
}
|
|
return _spriteViewEntity;
|
|
}
|
|
// Shitmed Change End
|
|
|
|
private Texture GetTexture(string texture)
|
|
{
|
|
var rsiPath = new ResPath("/Textures/Objects/Devices/health_analyzer.rsi");
|
|
var rsiSprite = new SpriteSpecifier.Rsi(rsiPath, texture);
|
|
|
|
var rsi = _cache.GetResource<RSIResource>(rsiSprite.RsiPath).RSI;
|
|
if (!rsi.TryGetState(rsiSprite.RsiState, out var state))
|
|
{
|
|
rsiSprite = new SpriteSpecifier.Rsi(rsiPath, "unknown");
|
|
}
|
|
|
|
return _spriteSystem.Frame0(rsiSprite);
|
|
}
|
|
|
|
private static Label CreateDiagnosticItemLabel(string text)
|
|
{
|
|
return new Label
|
|
{
|
|
Text = text,
|
|
};
|
|
}
|
|
|
|
private BoxContainer CreateDiagnosticGroupTitle(string text, string id)
|
|
{
|
|
var rootContainer = new BoxContainer
|
|
{
|
|
Margin = new Thickness(0, 6, 0, 0),
|
|
VerticalAlignment = VAlignment.Bottom,
|
|
Orientation = LayoutOrientation.Horizontal,
|
|
};
|
|
|
|
rootContainer.AddChild(new TextureRect
|
|
{
|
|
SetSize = new Vector2(30, 30),
|
|
Texture = GetTexture(id.ToLower())
|
|
});
|
|
|
|
rootContainer.AddChild(CreateDiagnosticItemLabel(text));
|
|
|
|
return rootContainer;
|
|
}
|
|
}
|